save/load added operation IDs

sc420 · Feb 27, 2024 · 6cdaccc · 6cdaccc
1 parent 92ab245
commit 6cdaccc
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/interactive-computational-graph/src/components/GraphContainer.tsx b/interactive-computational-graph/src/components/GraphContainer.tsx
@@ -541,6 +541,7 @@ const GraphContainer: FunctionComponent<GraphContainerProps> = ({
       nextNodeId,
       nodeNameBuilderState,
       nextOperationId,
+      operationIdsAddedAtLeastOnce: Array.from(operationIdsAddedAtLeastOnce),
       reactFlowState: reactFlowInstance.toObject(),
     };
   }, [
@@ -550,6 +551,7 @@ const GraphContainer: FunctionComponent<GraphContainerProps> = ({
     nextNodeId,
     nextOperationId,
     nodeNameBuilder,
+    operationIdsAddedAtLeastOnce,
     reactFlowInstance,
     saveFeatureOperations,
   ]);
@@ -630,6 +632,9 @@ const GraphContainer: FunctionComponent<GraphContainerProps> = ({
       setNextNodeId(graphContainerState.nextNodeId);
       nodeNameBuilder.load(graphContainerState.nodeNameBuilderState);
       setNextOperationId(graphContainerState.nextOperationId);
+      setOperationIdsAddedAtLeastOnce(
+        new Set(graphContainerState.operationIdsAddedAtLeastOnce),
+      );
       loadReactFlow(graphContainerState.reactFlowState);
     },
     [coreGraphAdapter, loadFeatureOperations, loadReactFlow, nodeNameBuilder],

diff --git a/interactive-computational-graph/src/states/GraphContainerState.ts b/interactive-computational-graph/src/states/GraphContainerState.ts
@@ -3,7 +3,6 @@ import type FeatureOperationState from "./FeatureOperationState";
 import type NodeNameBuilderState from "./NodeNameBuilderState";
 
 interface GraphContainerState {
-  // TODO(sc420): Uncomment other states
   // Core graph
   coreGraphAdapterState: CoreGraphAdapterState;
 
@@ -16,7 +15,7 @@ interface GraphContainerState {
   nextOperationId: number;
 
   // Feature panel states
-  // operationIdsAddedAtLeastOnce: Set<string>;
+  operationIdsAddedAtLeastOnce: string[];
 
   // React Flow states
   reactFlowState: object;