Add missing def statement to leaf statement visitor

nickbattle · Dec 20, 2021 · 8fcfbcd · 8fcfbcd
1 parent 27bd905
commit 8fcfbcd
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/vdmj/src/main/java/com/fujitsu/vdmj/tc/statements/visitors/TCLeafStatementVisitor.java b/vdmj/src/main/java/com/fujitsu/vdmj/tc/statements/visitors/TCLeafStatementVisitor.java
@@ -39,6 +39,7 @@
 import com.fujitsu.vdmj.tc.statements.TCCaseStmtAlternative;
 import com.fujitsu.vdmj.tc.statements.TCCasesStatement;
 import com.fujitsu.vdmj.tc.statements.TCCyclesStatement;
+import com.fujitsu.vdmj.tc.statements.TCDefStatement;
 import com.fujitsu.vdmj.tc.statements.TCDurationStatement;
 import com.fujitsu.vdmj.tc.statements.TCElseIfStatement;
 import com.fujitsu.vdmj.tc.statements.TCErrorCase;
@@ -316,6 +317,20 @@ public C caseLetDefStatement(TCLetDefStatement node, S arg)
 		all.addAll(node.statement.apply(this, arg));
 		return all;
 	}
+
+ 	@Override
+ 	public C caseDefStatement(TCDefStatement node, S arg)
+ 	{
+		C all = newCollection();
+
+		for (TCDefinition def: node.equalsDefs)
+		{
+			all.addAll(visitorSet.applyDefinitionVisitor(def, arg));
+		}
+
+		all.addAll(node.statement.apply(this, arg));
+		return all;
+ 	}
 
  	@Override
 	public C casePeriodicStatement(TCPeriodicStatement node, S arg)