Fixed a bug in Am2900; created dlatch8/80; relayouted some components
[Mograsim.git] / net.mograsim.logic.model / src / net / mograsim / logic / model / snippets / highlevelstatehandlers / standard / StandardHighLevelStateHandler.java
index ec3b04d..58e8e64 100644 (file)
@@ -1,9 +1,15 @@
 package net.mograsim.logic.model.snippets.highlevelstatehandlers.standard;
 
+import java.util.Collections;
 import java.util.HashMap;
 import java.util.Map;
+import java.util.Map.Entry;
+import java.util.TreeMap;
+import java.util.function.BiFunction;
+import java.util.function.Function;
 
-import net.mograsim.logic.model.model.components.GUIComponent;
+import net.mograsim.logic.model.model.components.submodels.SubmodelComponent;
+import net.mograsim.logic.model.serializing.IdentifyParams;
 import net.mograsim.logic.model.snippets.HighLevelStateHandler;
 import net.mograsim.logic.model.snippets.SnippetDefinintion;
 import net.mograsim.logic.model.snippets.SubmodelComponentSnippetSuppliers;
@@ -14,24 +20,51 @@ import net.mograsim.logic.model.snippets.highlevelstatehandlers.standard.subcomp
 
 public class StandardHighLevelStateHandler implements HighLevelStateHandler
 {
-       private final GUIComponent component;
+       private final SubmodelComponent component;
        private final Map<String, SubcomponentHighLevelStateHandler> subcomponentHighLevelStateHandlers;
+       private final Map<String, SubcomponentHighLevelStateHandler> subcomponentHighLevelStateHandlersUnmodifiable;
        private final Map<String, AtomicHighLevelStateHandler> atomicHighLevelStateHandlers;
+       private final Map<String, AtomicHighLevelStateHandler> atomicHighLevelStateHandlersUnmodifiable;
 
-       public StandardHighLevelStateHandler(GUIComponent component, SimpleHighLevelStateHandlerParams params)
+       public StandardHighLevelStateHandler(SubmodelComponent component)
+       {
+               this(component, null);
+       }
+
+       public StandardHighLevelStateHandler(SubmodelComponent component, StandardHighLevelStateHandlerParams params)
        {
                this.component = component;
                this.subcomponentHighLevelStateHandlers = new HashMap<>();
+               this.subcomponentHighLevelStateHandlersUnmodifiable = Collections.unmodifiableMap(subcomponentHighLevelStateHandlers);
                this.atomicHighLevelStateHandlers = new HashMap<>();
-               params.subcomponentHighLevelStates.forEach(this::addSubcomponentHighLevelState);
-               params.atomicHighLevelStates.forEach(this::addAtomicHighLevelState);
+               this.atomicHighLevelStateHandlersUnmodifiable = Collections.unmodifiableMap(atomicHighLevelStateHandlers);
+               if (params != null)
+               {
+                       params.subcomponentHighLevelStates.forEach(this::addSubcomponentHighLevelState);
+                       params.atomicHighLevelStates.forEach(this::addAtomicHighLevelState);
+               }
+       }
+
+       public SubcomponentHighLevelStateHandler addSubcomponentHighLevelState(String subcomponentStateID,
+                       SubcomponentHighLevelStateHandlerParams handlerParams)
+       {
+               return addSubcomponentHighLevelState(subcomponentStateID,
+                               StandardHighLevelStateHandlerSnippetSuppliers.subcomponentHandlerSupplier.getSnippetSupplier(handlerParams.id)::create,
+                               handlerParams.params);
+       }
+
+       public <P, H extends SubcomponentHighLevelStateHandler> H addSubcomponentHighLevelState(String subcomponentStateID,
+                       BiFunction<SubmodelComponent, P, H> handlerConstructor, P handlerParams)
+       {
+               return addSubcomponentHighLevelState(subcomponentStateID, c -> handlerConstructor.apply(c, handlerParams));
        }
 
-       public void addSubcomponentHighLevelState(String subcomponentStateID, SubcomponentHighLevelStateHandlerParams handlerParams)
+       public <H extends SubcomponentHighLevelStateHandler> H addSubcomponentHighLevelState(String subcomponentStateID,
+                       Function<SubmodelComponent, H> handlerConstructor)
        {
-               HighLevelStateHandlerContext context = new HighLevelStateHandlerContext(component, subcomponentStateID);
-               addSubcomponentHighLevelState(subcomponentStateID, StandardHighLevelStateHandlerSnippetSuppliers.subcomponentHandlerSupplier
-                               .getSnippetSupplier(handlerParams.id).create(context, handlerParams.params));
+               H handler = handlerConstructor.apply(component);
+               addSubcomponentHighLevelState(subcomponentStateID, handler);
+               return handler;
        }
 
        public void addSubcomponentHighLevelState(String subcomponentStateID, SubcomponentHighLevelStateHandler handler)
@@ -40,11 +73,36 @@ public class StandardHighLevelStateHandler implements HighLevelStateHandler
                subcomponentHighLevelStateHandlers.put(subcomponentStateID, handler);
        }
 
-       public void addAtomicHighLevelState(String atomicStateID, AtomicHighLevelStateHandlerParams handlerParams)
+       public void removeSubcomponentHighLevelState(String subcomponentStateID)
        {
-               HighLevelStateHandlerContext context = new HighLevelStateHandlerContext(component, atomicStateID);
-               addSubcomponentHighLevelState(atomicStateID, StandardHighLevelStateHandlerSnippetSuppliers.subcomponentHandlerSupplier
-                               .getSnippetSupplier(handlerParams.id).create(context, handlerParams.params));
+               checkHighLevelStateIDPart(subcomponentStateID);
+               subcomponentHighLevelStateHandlers.remove(subcomponentStateID);
+       }
+
+       public Map<String, SubcomponentHighLevelStateHandler> getSubcomponentHighLevelStates()
+       {
+               return subcomponentHighLevelStateHandlersUnmodifiable;
+       }
+
+       public AtomicHighLevelStateHandler addAtomicHighLevelState(String atomicStateID, AtomicHighLevelStateHandlerParams handlerParams)
+       {
+               return addAtomicHighLevelState(atomicStateID,
+                               StandardHighLevelStateHandlerSnippetSuppliers.atomicHandlerSupplier.getSnippetSupplier(handlerParams.id)::create,
+                               handlerParams.params);
+       }
+
+       public <P, H extends AtomicHighLevelStateHandler> H addAtomicHighLevelState(String subcomponentStateID,
+                       BiFunction<SubmodelComponent, P, H> handlerConstructor, P handlerParams)
+       {
+               return addAtomicHighLevelState(subcomponentStateID, c -> handlerConstructor.apply(c, handlerParams));
+       }
+
+       public <H extends AtomicHighLevelStateHandler> H addAtomicHighLevelState(String subcomponentStateID,
+                       Function<SubmodelComponent, H> handlerConstructor)
+       {
+               H handler = handlerConstructor.apply(component);
+               addAtomicHighLevelState(subcomponentStateID, handler);
+               return handler;
        }
 
        public void addAtomicHighLevelState(String atomicStateID, AtomicHighLevelStateHandler handler)
@@ -53,6 +111,17 @@ public class StandardHighLevelStateHandler implements HighLevelStateHandler
                atomicHighLevelStateHandlers.put(atomicStateID, handler);
        }
 
+       public void removeAtomicHighLevelState(String atomicStateID)
+       {
+               checkHighLevelStateIDPart(atomicStateID);
+               atomicHighLevelStateHandlers.remove(atomicStateID);
+       }
+
+       public Map<String, AtomicHighLevelStateHandler> getAtomicHighLevelStates()
+       {
+               return atomicHighLevelStateHandlersUnmodifiable;
+       }
+
        private static void checkHighLevelStateIDPart(String stateIDPart)
        {
                if (stateIDPart.indexOf('.') != -1)
@@ -98,7 +167,40 @@ public class StandardHighLevelStateHandler implements HighLevelStateHandler
                }
        }
 
-       public static class SimpleHighLevelStateHandlerParams
+       @Override
+       public String getIDForSerializing(IdentifyParams idParams)
+       {
+               return "standard";
+       }
+
+       @Override
+       public StandardHighLevelStateHandlerParams getParamsForSerializing(IdentifyParams idParams)
+       {
+               StandardHighLevelStateHandlerParams params = new StandardHighLevelStateHandlerParams();
+               params.subcomponentHighLevelStates = new TreeMap<>();
+               params.atomicHighLevelStates = new TreeMap<>();
+               for (Entry<String, SubcomponentHighLevelStateHandler> e : subcomponentHighLevelStateHandlers.entrySet())
+               {
+                       String stateID = e.getKey();
+                       SubcomponentHighLevelStateHandler handler = e.getValue();
+                       SubcomponentHighLevelStateHandlerParams handlerParams = new SubcomponentHighLevelStateHandlerParams();
+                       handlerParams.id = handler.getIDForSerializing(idParams);
+                       handlerParams.params = handler.getParamsForSerializingJSON(idParams);
+                       params.subcomponentHighLevelStates.put(stateID, handlerParams);
+               }
+               for (Entry<String, AtomicHighLevelStateHandler> e : atomicHighLevelStateHandlers.entrySet())
+               {
+                       String stateID = e.getKey();
+                       AtomicHighLevelStateHandler handler = e.getValue();
+                       AtomicHighLevelStateHandlerParams handlerParams = new AtomicHighLevelStateHandlerParams();
+                       handlerParams.id = handler.getIDForSerializing(idParams);
+                       handlerParams.params = handler.getParamsForSerializingJSON(idParams);
+                       params.atomicHighLevelStates.put(stateID, handlerParams);
+               }
+               return params;
+       }
+
+       public static class StandardHighLevelStateHandlerParams
        {
                public Map<String, SubcomponentHighLevelStateHandlerParams> subcomponentHighLevelStates;
                public Map<String, AtomicHighLevelStateHandlerParams> atomicHighLevelStates;
@@ -108,6 +210,6 @@ public class StandardHighLevelStateHandler implements HighLevelStateHandler
        {
                SubmodelComponentSnippetSuppliers.highLevelStateHandlerSupplier.setSnippetSupplier(
                                StandardHighLevelStateHandler.class.getCanonicalName(),
-                               SnippetDefinintion.create(SimpleHighLevelStateHandlerParams.class, StandardHighLevelStateHandler::new));
+                               SnippetDefinintion.create(StandardHighLevelStateHandlerParams.class, StandardHighLevelStateHandler::new));
        }
 }
\ No newline at end of file