From ee1b879b9bb25635abc2ed37f6daa57ccd6d799b Mon Sep 17 00:00:00 2001 From: Jan Gruteser <jan.gruteser@hhu.de> Date: Mon, 23 Dec 2024 16:30:18 +0100 Subject: [PATCH] fix map creation --- .../de/tla2b/analysis/PredicateVsExpression.java | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/src/main/java/de/tla2b/analysis/PredicateVsExpression.java b/src/main/java/de/tla2b/analysis/PredicateVsExpression.java index 672b281..2e4c8fe 100644 --- a/src/main/java/de/tla2b/analysis/PredicateVsExpression.java +++ b/src/main/java/de/tla2b/analysis/PredicateVsExpression.java @@ -5,27 +5,27 @@ import de.tla2b.global.BBuiltInOPs; import tla2sany.semantic.*; import tlc2.tool.BuiltInOPs; -import java.util.Arrays; +import java.util.HashMap; import java.util.Map; -import java.util.stream.Collectors; import static de.tla2b.global.TranslationGlobals.STANDARD_MODULES; public class PredicateVsExpression extends BuiltInOPs implements BBuildIns { - private final Map<OpDefNode, DefinitionType> definitionsTypeMap; + private final Map<OpDefNode, DefinitionType> definitionsTypeMap = new HashMap<>(); public enum DefinitionType { PREDICATE, EXPRESSION } - public DefinitionType getDefinitionType(OpDefNode def) { - return definitionsTypeMap.get(def); + public PredicateVsExpression(ModuleNode moduleNode) { + for (OpDefNode def : moduleNode.getOpDefs()) { + definitionsTypeMap.put(def, visitSemanticNode(def.getBody())); + } } - public PredicateVsExpression(ModuleNode moduleNode) { - this.definitionsTypeMap = Arrays.stream(moduleNode.getOpDefs()) - .collect(Collectors.toMap(def -> def, def -> visitSemanticNode(def.getBody()))); + public DefinitionType getDefinitionType(OpDefNode def) { + return definitionsTypeMap.get(def); } private DefinitionType visitSemanticNode(SemanticNode s) { -- GitLab