diff --git a/prism/src/parser/ast/ModulesFile.java b/prism/src/parser/ast/ModulesFile.java index 63cc3c2b..db9fea74 100644 --- a/prism/src/parser/ast/ModulesFile.java +++ b/prism/src/parser/ast/ModulesFile.java @@ -143,6 +143,11 @@ public class ModulesFile extends ASTElement systemDefn = s; } + public void clearRewardStructs() + { + rewardStructs.clear(); + } + public void addRewardStruct(RewardStruct r) { rewardStructs.add(r); diff --git a/prism/src/pta/Modules2PTA.java b/prism/src/pta/Modules2PTA.java index d896b40b..f1c1beba 100644 --- a/prism/src/pta/Modules2PTA.java +++ b/prism/src/pta/Modules2PTA.java @@ -85,9 +85,13 @@ public class Modules2PTA // and simplify any expressions as much as possible. modulesFile = (ModulesFile) modulesFile.deepCopy().replaceConstants(constantValues).simplify(); - // Remove labels from (cloned) model - these are not translated. + // Remove formulas/labels from (cloned) model - these are not translated. + modulesFile.setFormulaList(new FormulaList()); modulesFile.setLabelList(new LabelList()); - + + // Also remove reward structures - these are not currently used + modulesFile.clearRewardStructs(); + // Go through list of modules numModules = modulesFile.getNumModules(); pcStates = new ArrayList>(numModules);