try to save core graph state

interactive-computational-graph/src/components/SaveLoadPanel.test.tsx

@@ -26,6 +26,12 @@ test("should trigger the event when clicking the load button", async () => {
   const contents = `\
 {
     "coreGraphAdapterState": {
+        "coreGraphState": {
+            "nodeIdToNodes": {},
+            "differentiationMode": "REVERSE",
+            "targetNodeId": null,
+            "nodeIdToDerivatives": {}
+        },
         "nodeIdToNames": {},
         "dummyInputNodeIdToNodeIds": {}
     },

interactive-computational-graph/src/core/ConstantNode.ts

@@ -1,3 +1,4 @@
+import type CoreNodeState from "../states/CoreNodeState";
 import type CoreNode from "./CoreNode";
 import NodeRelationship from "./NodeRelationship";
 import type NodeType from "./NodeType";
@@ -44,6 +45,12 @@ class ConstantNode implements CoreNode {
   getRelationship(): NodeRelationship {
     return this.nodeRelationship;
   }
+
+  save(): CoreNodeState {
+    return {
+      nodeType: "CONSTANT",
+    };
+  }
 }
 
 export default ConstantNode;

interactive-computational-graph/src/core/CoreNode.ts

@@ -1,3 +1,4 @@
+import type CoreNodeState from "../states/CoreNodeState";
 import type NodeRelationship from "./NodeRelationship";
 import type NodeType from "./NodeType";
 
@@ -25,6 +26,8 @@ interface CoreNode {
   calculateDfdx: (x: CoreNode) => string;
 
   getRelationship: () => NodeRelationship;
+
+  save: () => CoreNodeState;
 }
 
 export default CoreNode;

interactive-computational-graph/src/core/Graph.ts

@@ -1,3 +1,5 @@
+import type CoreGraphState from "../states/CoreGraphState";
+import type CoreNodeState from "../states/CoreNodeState";
 import type ChainRuleTerm from "./ChainRuleTerm";
 import {
   CycleError,
@@ -27,7 +29,7 @@ class Graph {
    * the current target node. For forward mode, it propagates from left to
    * right. For reverse mode, it propagates from right to left.
    */
-  private readonly nodeIdToDerivatives = new Map<string, string>();
+  private nodeIdToDerivatives = new Map<string, string>();
 
   getNodes(): CoreNode[] {
     return Array.from(this.nodeIdToNodes.values());
@@ -306,6 +308,30 @@ multiple edges`,
     });
   }
 
+  save(): CoreGraphState {
+    const nodeIdToNodes: Record<string, CoreNodeState> = {};
+    this.nodeIdToNodes.forEach((node, nodeId) => {
+      nodeIdToNodes[nodeId] = node.save();
+    });
+
+    return {
+      nodeIdToNodes,
+      differentiationMode: this.differentiationMode,
+      targetNodeId: this.targetNodeId,
+      nodeIdToDerivatives: Object.fromEntries(this.nodeIdToDerivatives),
+    };
+  }
+
+  load(state: CoreGraphState): void {
+    // TODO(sc420): Build the core nodes
+
+    this.differentiationMode = state.differentiationMode;
+    this.targetNodeId = state.targetNodeId;
+    this.nodeIdToDerivatives = new Map(
+      Object.entries(state.nodeIdToDerivatives),
+    );
+  }
+
   /**
    * Checks if we can visit the node y from node x in an acyclic graph.
    *

interactive-computational-graph/src/core/OperationNode.ts

@@ -1,3 +1,4 @@
+import type CoreNodeState from "../states/CoreNodeState";
 import type CoreNode from "./CoreNode";
 import NodeRelationship from "./NodeRelationship";
 import type NodeType from "./NodeType";
@@ -90,6 +91,14 @@ class OperationNode implements CoreNode {
     });
     return fInputNodeToValues;
   }
+
+  save(): CoreNodeState {
+    return {
+      nodeType: "OPERATION",
+      value: this.value,
+      operationId: "", // TODO(sc420): Return operation ID
+    };
+  }
 }
 
 export default OperationNode;

interactive-computational-graph/src/core/VariableNode.ts

@@ -1,3 +1,4 @@
+import type CoreNodeState from "../states/CoreNodeState";
 import type CoreNode from "./CoreNode";
 import NodeRelationship from "./NodeRelationship";
 import type NodeType from "./NodeType";
@@ -50,6 +51,13 @@ class VariableNode implements CoreNode {
   getRelationship(): NodeRelationship {
     return this.nodeRelationship;
   }
+
+  save(): CoreNodeState {
+    return {
+      nodeType: "VARIABLE",
+      value: this.value,
+    };
+  }
 }
 
 export default VariableNode;

interactive-computational-graph/src/features/CoreGraphAdapter.ts

@@ -548,6 +548,7 @@ cycle`;
 
   save(): CoreGraphAdapterState {
     return {
+      coreGraphState: this.graph.save(),
       nodeIdToNames: Object.fromEntries(this.nodeIdToNames),
       dummyInputNodeIdToNodeIds: Object.fromEntries(
         this.dummyInputNodeIdToNodeIds,
@@ -556,6 +557,7 @@ cycle`;
   }
 
   load(state: CoreGraphAdapterState): void {
+    this.graph.load(state.coreGraphState);
     this.nodeIdToNames = new Map(Object.entries(state.nodeIdToNames));
     this.dummyInputNodeIdToNodeIds = new Map(
       Object.entries(state.dummyInputNodeIdToNodeIds),

interactive-computational-graph/src/states/CoreGraphAdapterState.ts

@@ -1,8 +1,7 @@
-// import type CoreGraphState from "./CoreGraphState";
+import type CoreGraphState from "./CoreGraphState";
 
 interface CoreGraphAdapterState {
-  // TODO(sc420): Uncomment
-  // coreGraphState: CoreGraphState;
+  coreGraphState: CoreGraphState;
   nodeIdToNames: Record<string, string>;
   dummyInputNodeIdToNodeIds: Record<string, string>;
 }

interactive-computational-graph/src/states/CoreGraphState.ts

@@ -1,10 +1,11 @@
 import type DifferentiationMode from "../core/DifferentiationMode";
+import type CoreNodeState from "./CoreNodeState";
 
 interface CoreGraphState {
-  // TODO(sc420): Add nodeIdToNodes state
+  nodeIdToNodes: Record<string, CoreNodeState>;
   differentiationMode: DifferentiationMode;
   targetNodeId: string | null;
-  nodeIdToDerivatives: Map<string, string>;
+  nodeIdToDerivatives: Record<string, string>;
 }
 
 export default CoreGraphState;

interactive-computational-graph/src/states/CoreNodeState.ts

@@ -0,0 +1,20 @@
+// TODO(sc420): Add nodeRelationship states
+
+interface ConstantNodeState {
+  nodeType: "CONSTANT";
+}
+
+interface VariableNodeState {
+  nodeType: "VARIABLE";
+  value: string;
+}
+
+interface OperationNodeState {
+  nodeType: "OPERATION";
+  value: string;
+  operationId: string;
+}
+
+type CoreNodeState = ConstantNodeState | VariableNodeState | OperationNodeState;
+
+export default CoreNodeState;