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 index b06800279..16907e0a8 100644 --- 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)