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