Add getControlFlowGraph() to relevant node impls

- RoutineImplementationNodeImpl
- AnonymousMethodNodeImpl

Author: jgardn3r

delphi-frontend/src/main/java/au/com/integradev/delphi/antlr/ast/node/AnonymousMethodNodeImpl.java

@@ -19,8 +19,12 @@
 package au.com.integradev.delphi.antlr.ast.node;
 
 import au.com.integradev.delphi.antlr.ast.visitors.DelphiParserVisitor;
+import au.com.integradev.delphi.cfg.ControlFlowGraphFactory;
+import au.com.integradev.delphi.cfg.api.ControlFlowGraph;
 import au.com.integradev.delphi.type.factory.TypeFactoryImpl;
 import au.com.integradev.delphi.type.parameter.FormalParameter;
+import com.google.common.base.Supplier;
+import com.google.common.base.Suppliers;
 import java.util.Collections;
 import java.util.Set;
 import java.util.stream.Collectors;
@@ -42,6 +46,16 @@
 public final class AnonymousMethodNodeImpl extends ExpressionNodeImpl
     implements AnonymousMethodNode {
   private String image;
+  private final Supplier<ControlFlowGraph> cfgSupplier =
+      Suppliers.memoize(
+          () -> {
+            CompoundStatementNode compoundStatementNode =
+                getFirstChildOfType(CompoundStatementNode.class);
+            if (compoundStatementNode == null) {
+              return null;
+            }
+            return ControlFlowGraphFactory.create(compoundStatementNode);
+          });
 
   public AnonymousMethodNodeImpl(Token token) {
     super(token);
@@ -154,4 +168,9 @@ protected Type createType() {
                 : returnTypeNode.getTypeNode().getType(),
             getDirectives());
   }
+
+  @Nullable
+  public ControlFlowGraph getControlFlowGraph() {
+    return cfgSupplier.get();
+  }
 }

delphi-frontend/src/main/java/au/com/integradev/delphi/antlr/ast/node/RoutineImplementationNodeImpl.java

@@ -19,6 +19,10 @@
 package au.com.integradev.delphi.antlr.ast.node;
 
 import au.com.integradev.delphi.antlr.ast.visitors.DelphiParserVisitor;
+import au.com.integradev.delphi.cfg.ControlFlowGraphFactory;
+import au.com.integradev.delphi.cfg.api.ControlFlowGraph;
+import com.google.common.base.Supplier;
+import com.google.common.base.Suppliers;
 import java.util.function.Function;
 import javax.annotation.Nonnull;
 import javax.annotation.Nullable;
@@ -36,6 +40,19 @@
 public final class RoutineImplementationNodeImpl extends RoutineNodeImpl
     implements RoutineImplementationNode {
   private TypeNameDeclaration typeDeclaration;
+  private final Supplier<ControlFlowGraph> cfgSupplier =
+      Suppliers.memoize(
+          () -> {
+            RoutineBodyNode routineBody = getRoutineBody();
+            if (routineBody == null) {
+              return null;
+            }
+            CompoundStatementNode block = routineBody.getStatementBlock();
+            if (block == null) {
+              return null;
+            }
+            return ControlFlowGraphFactory.create(block);
+          });
 
   public RoutineImplementationNodeImpl(Token token) {
     super(token);
@@ -127,6 +144,11 @@ public NameReferenceNode getNameReferenceNode() {
     return getRoutineNameNode().getNameReferenceNode();
   }
 
+  @Nullable
+  public ControlFlowGraph getControlFlowGraph() {
+    return cfgSupplier.get();
+  }
+
   @Override
   public VisibilityType createVisibility() {
     return VisibilityType.PUBLIC;