Bunch better layout options

1 files changed, 3 insertions, 1 deletions

diff --git a/chapter18/src/main/java/com/seaofnodes/simple/print/SimpleGraphAdapter.java b/chapter18/src/main/java/com/seaofnodes/simple/print/SimpleGraphAdapter.java
index 442bfd50..17dd1131 100644
--- a/chapter18/src/main/java/com/seaofnodes/simple/print/SimpleGraphAdapter.java
+++ b/chapter18/src/main/java/com/seaofnodes/simple/print/SimpleGraphAdapter.java
@@ -24,7 +24,9 @@ public class SimpleGraphAdapter extends GraphAdapter<Node> {
         for( int i = 0; i < n.nIns(); i++ ) {
             String name = n instanceof ScopeNode scope && i < scope._vars.size()
                 ? scope._vars.get(i)._name : null;
-            edges.add(new Edge(i, ref(n.in(i)), role(n, i), name));
+            int jump = n instanceof CallEndNode && i>0 && n.in(i) instanceof ReturnNode ret
+                ? ref(ret.fun()) : 0;
+            edges.add(new Edge(i, ref(n.in(i)), role(n, i), name, jump));
         }
         return edges;
     }