CSE machine: support for StreamViz drawing arrows

src/createContext.ts

@@ -139,6 +139,8 @@
     runtime: createEmptyRuntime(),
     numberOfOuterEnvironments: 1,
     prelude: null,
+    pendingStreamFnStack: [],
+    streamLineage: new Map<string, string[]>(),
     debugger: createEmptyDebugger(),
     nativeStorage: createNativeStorage(),
     executionMethod: 'auto',
@@ -363,7 +365,7 @@
         'call_cc(f)',
         context.variant === Variant.EXPLICIT_CONTROL
           ? call_with_current_continuation
           : (f: any) => {
               throw new Error('call_cc is only available in Explicit-Control variant');
             },
       );

src/cse-machine/interpreter.ts

@@ -359,6 +359,25 @@ export function* generateCSEMachineStateStream(
       context.runtime.changepointSteps.push(steps + 1);
     }
 
+    const evalResult = context.runtime.stash?.peek();
+    const mostRecentControlHeight =
+      context.pendingStreamFnStack[context.pendingStreamFnStack.length - 1]?.[1];
+
+    if (
+      Array.isArray(evalResult) &&
+      evalResult.length === 2 &&
+      mostRecentControlHeight !== undefined &&
+      context.runtime.control?.size() === mostRecentControlHeight - 1
+    ) {
+      const mostRecentNullaryFnId = context.pendingStreamFnStack.pop()?.[0];
+      if (mostRecentNullaryFnId !== undefined) {
+        if (!context.streamLineage.get(mostRecentNullaryFnId)) {
+          context.streamLineage.set(mostRecentNullaryFnId, []);
+        }
+        context.streamLineage.get(mostRecentNullaryFnId)?.push((evalResult as any).id);
+      }
+    }
+
     control.pop();
     if (isNode(command)) {
       context.runtime.nodes.shift();
@@ -834,6 +853,16 @@ const cmdEvaluators: CommandEvaluators = {
         context.runtime.environments.unshift(func.environment);
       }
 
+      // Streams Visualisation:
+      // If CALL0, push the nullary fn to pendingStreamFnStack
+      // (to map which nullary fn produces which pairs; for nested nullary fn)
+      if (func.node.params.length === 0) {
+        context.pendingStreamFnStack.push([
+          func.id,
+          context.runtime.control !== null ? context.runtime.control.size() : 0,
+        ]);
+      }
+
       // Handle special case if function is simple
       if (isSimpleFunction(func.node)) {
         // Closures convert ArrowExpressionStatements to BlockStatements

src/types.ts

@@ -65,6 +65,11 @@ export interface Context<T = any> {
     changepointSteps: number[];
   };
 
+  // STREAM VISUALISATION
+  streamLineage: Map<string, string[]>;
+
+  pendingStreamFnStack: [string, number][];
+
   numberOfOuterEnvironments: number;
 
   prelude: string | null;