From a32e83c9d81cfdb3759a2961d8b192ca44d75935 Mon Sep 17 00:00:00 2001
From: Cliff Click <cliffc@acm.org>
Date: Fri, 25 Sep 2026 15:00:00 -0700
Subject: Bunch more animation work

---
 chapter19/src/main/java/com/seaofnodes/simple/CodeGen.java   |  7 ++++++-
 .../main/java/com/seaofnodes/simple/node/CallEndNode.java    |  1 +
 .../src/main/java/com/seaofnodes/simple/node/FunNode.java    |  1 +
 .../java/com/seaofnodes/simple/print/SimpleGraphAdapter.java | 12 +++++++++++-
 .../com/seaofnodes/simple/print/SimpleGraphObserver.java     |  2 +-
 5 files changed, 20 insertions(+), 3 deletions(-)

(limited to 'chapter19/src/main/java/com')

diff --git a/chapter19/src/main/java/com/seaofnodes/simple/CodeGen.java b/chapter19/src/main/java/com/seaofnodes/simple/CodeGen.java
index 8465d692..5baa05a6 100644
--- a/chapter19/src/main/java/com/seaofnodes/simple/CodeGen.java
+++ b/chapter19/src/main/java/com/seaofnodes/simple/CodeGen.java
@@ -169,9 +169,14 @@ public class CodeGen {
         assert _phase.ordinal() <= Phase.Opto.ordinal();
         _phase = Phase.TypeCheck;
 
-        Parser.ParseException err = _stop.walk( Node::err );
+        Parser.ParseException err = _stop.walk(n -> {
+            Parser.ParseException msg = n.err();
+            if( msg != null && _obs != null ) _obs.error(n, msg.getMessage());
+            return msg;
+        });
         if( err != null )
             throw err;
+        if( _obs != null ) _obs.phase("TypeCheck");
         return this;
     }
 
diff --git a/chapter19/src/main/java/com/seaofnodes/simple/node/CallEndNode.java b/chapter19/src/main/java/com/seaofnodes/simple/node/CallEndNode.java
index e6aa9fe9..2205db9f 100644
--- a/chapter19/src/main/java/com/seaofnodes/simple/node/CallEndNode.java
+++ b/chapter19/src/main/java/com/seaofnodes/simple/node/CallEndNode.java
@@ -11,6 +11,7 @@ public class CallEndNode extends CFGNode implements MultiNode {
 
     // When set true, this Call/CallEnd/Fun/Return is being trivially inlined
     private boolean _folding;
+    public boolean folding() { return _folding; }
 
     public CallEndNode(CallNode call) { super(new Node[]{call}); }
     public CallEndNode(CallEndNode cend) { super(cend); }
diff --git a/chapter19/src/main/java/com/seaofnodes/simple/node/FunNode.java b/chapter19/src/main/java/com/seaofnodes/simple/node/FunNode.java
index b514eba0..d24f4829 100644
--- a/chapter19/src/main/java/com/seaofnodes/simple/node/FunNode.java
+++ b/chapter19/src/main/java/com/seaofnodes/simple/node/FunNode.java
@@ -10,6 +10,7 @@ public class FunNode extends RegionNode {
 
     // When set true, this Call/CallEnd/Fun/Return is being trivially inlined
     boolean _folding;
+    public boolean folding() { return _folding; }
 
     private TypeFunPtr _sig;    // Initial signature
     private ReturnNode _ret;    // Return pointer
diff --git a/chapter19/src/main/java/com/seaofnodes/simple/print/SimpleGraphAdapter.java b/chapter19/src/main/java/com/seaofnodes/simple/print/SimpleGraphAdapter.java
index ee424501..41725fcf 100644
--- a/chapter19/src/main/java/com/seaofnodes/simple/print/SimpleGraphAdapter.java
+++ b/chapter19/src/main/java/com/seaofnodes/simple/print/SimpleGraphAdapter.java
@@ -36,7 +36,17 @@ public class SimpleGraphAdapter extends GraphAdapter<Node> {
             : n instanceof CProjNode p ? new Projection(ref(n.in(0)), p._idx) : null;
         String label = n.label();
         return new GraphSnapshot.Node(n._nid, label == null ? n.getClass().getSimpleName() : label,
-                                      n._type == null ? null : n._type.toString(), kind(n), edges(n), proj);
+                                      n._type == null ? null : n._type.toString(), kind(n), edges(n), proj, folding(n));
+    }
+
+    private boolean folding(Node n) {
+        if( n instanceof FunNode fun ) return fun.folding();
+        if( n instanceof ReturnNode ret ) return ret.fun().folding();
+        if( n instanceof CallEndNode cend ) return cend.folding();
+        if( n instanceof CallNode )
+            for( int i=0; i<n.nOuts(); i++ )
+                if( n.out(i) instanceof CallEndNode cend && cend.folding() ) return true;
+        return false;
     }
 
     private Kind kind(Node n) {
diff --git a/chapter19/src/main/java/com/seaofnodes/simple/print/SimpleGraphObserver.java b/chapter19/src/main/java/com/seaofnodes/simple/print/SimpleGraphObserver.java
index 04c9298f..4ec55f4d 100644
--- a/chapter19/src/main/java/com/seaofnodes/simple/print/SimpleGraphObserver.java
+++ b/chapter19/src/main/java/com/seaofnodes/simple/print/SimpleGraphObserver.java
@@ -17,7 +17,7 @@ public class SimpleGraphObserver extends GraphCapture<Node> {
     @Override protected void compile(String src) {
         _code = new CodeGen(src);
         _code._obs = this;
-        _code.parse().opto();
+        _code.parse().opto().typeCheck();
     }
     @Override protected String phase() { return _code._phase.name(); }
     @Override protected int pos() { return _code._phase == CodeGen.Phase.Parse ? _code.P.pos() : -1; }
-- 
cgit v1.2.3