X-Git-Url: https://mograsim.net/gitweb/?a=blobdiff_plain;f=plugins%2Fnet.mograsim.logic.model%2Fsrc%2Fnet%2Fmograsim%2Flogic%2Fmodel%2Fsnippets%2Fhighlevelstatehandlers%2Fstandard%2Fatomic%2FWireForcingAtomicHighLevelStateHandler.java;h=4f7cd4579e4078c68d1167f6c45dfcf01a14a02b;hb=69ec19d54ceb6d5abbb8b4faa55284af22174859;hp=aa509c9ddca61d5ed5cbb0570536631dab0ef49e;hpb=f1933b06b5fe800902131e4dc34f002ac3fa17f0;p=Mograsim.git

diff --git a/plugins/net.mograsim.logic.model/src/net/mograsim/logic/model/snippets/highlevelstatehandlers/standard/atomic/WireForcingAtomicHighLevelStateHandler.java b/plugins/net.mograsim.logic.model/src/net/mograsim/logic/model/snippets/highlevelstatehandlers/standard/atomic/WireForcingAtomicHighLevelStateHandler.java
index aa509c9d..4f7cd457 100644
--- a/plugins/net.mograsim.logic.model/src/net/mograsim/logic/model/snippets/highlevelstatehandlers/standard/atomic/WireForcingAtomicHighLevelStateHandler.java
+++ b/plugins/net.mograsim.logic.model/src/net/mograsim/logic/model/snippets/highlevelstatehandlers/standard/atomic/WireForcingAtomicHighLevelStateHandler.java
@@ -2,11 +2,16 @@ package net.mograsim.logic.model.snippets.highlevelstatehandlers.standard.atomic
 
 import java.util.ArrayList;
 import java.util.Collections;
+import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
+import java.util.Objects;
+import java.util.concurrent.atomic.AtomicReference;
+import java.util.function.Consumer;
 import java.util.function.Function;
 import java.util.stream.Collectors;
 
+import net.mograsim.logic.core.LogicObserver;
 import net.mograsim.logic.core.types.Bit;
 import net.mograsim.logic.core.types.BitVector;
 import net.mograsim.logic.model.model.components.submodels.SubmodelComponent;
@@ -24,6 +29,8 @@ public class WireForcingAtomicHighLevelStateHandler implements AtomicHighLevelSt
 	private final List<ModelWire> wiresToForceInverted;
 	private final List<ModelWire> wiresToForceInvertedUnmodifiable;
 
+	private final Map<Consumer<Object>, LogicObserver> wireObsPerListener;
+
 	public WireForcingAtomicHighLevelStateHandler(SubmodelComponent component)
 	{
 		this(component, null);
@@ -36,6 +43,9 @@ public class WireForcingAtomicHighLevelStateHandler implements AtomicHighLevelSt
 		this.wiresToForceUnmodifiable = Collections.unmodifiableList(wiresToForce);
 		this.wiresToForceInverted = new ArrayList<>();
 		this.wiresToForceInvertedUnmodifiable = Collections.unmodifiableList(wiresToForceInverted);
+
+		this.wireObsPerListener = new HashMap<>();
+
 		if (params != null)
 		{
 			Map<String, ModelWire> wiresByName = component.submodel.getWiresByName();
@@ -133,6 +143,37 @@ public class WireForcingAtomicHighLevelStateHandler implements AtomicHighLevelSt
 				wire.forceWireValues(vector);
 	}
 
+	@Override
+	public void addListener(Consumer<Object> stateChanged)
+	{
+		if (wireObsPerListener.containsKey(stateChanged))
+			return;
+		AtomicReference<Object> lastStateRef = new AtomicReference<>(getHighLevelState());
+		LogicObserver obs = w ->
+		{
+			Object newState = getHighLevelState();
+			if (!Objects.equals(lastStateRef.getAndSet(newState), newState))
+				stateChanged.accept(newState);
+		};
+		wireObsPerListener.put(stateChanged, obs);
+		for (ModelWire w : wiresToForce)
+			w.addObserver(obs);
+		for (ModelWire w : wiresToForceInverted)
+			w.addObserver(obs);
+	}
+
+	@Override
+	public void removeListener(Consumer<Object> stateChanged)
+	{
+		LogicObserver obs = wireObsPerListener.remove(stateChanged);
+		if (obs == null)
+			return;
+		for (ModelWire w : wiresToForce)
+			w.removeObserver(obs);
+		for (ModelWire w : wiresToForceInverted)
+			w.removeObserver(obs);
+	}
+
 	@Override
 	public String getIDForSerializing(IdentifyParams idParams)
 	{