From 8625d3a2708e6332083a1b780b4f016beacf67d3 Mon Sep 17 00:00:00 2001
From: bd-912 <bdunahu@colostate.edu>
Date: Fri, 19 Apr 2024 23:21:01 -0600
Subject: CFG captures both VBranch paths

---
 vaporize/library/ControlFlowGraph.java | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

(limited to 'vaporize/library/ControlFlowGraph.java')

diff --git a/vaporize/library/ControlFlowGraph.java b/vaporize/library/ControlFlowGraph.java
index e7057ac..c0eff10 100644
--- a/vaporize/library/ControlFlowGraph.java
+++ b/vaporize/library/ControlFlowGraph.java
@@ -15,7 +15,7 @@ public class ControlFlowGraph {
         this.start = null;
     }
 
-    protected CFGNode getNode(Node a) {
+    protected CFGNode getNode(Object a) {
         CFGNode ret = null;
         for (CFGNode n : this.nodes) {
             if (n.equals(a)) {
@@ -24,9 +24,12 @@ public class ControlFlowGraph {
             }
         }
 
-        if (ret == null)
+        if (ret == null) {
+            String str = (a instanceof Node) ? ((Node) a).sourcePos.toString() :
+                a.toString();
             MinimalLogger.severe(String.format("Could not find a node matching %s",
-                                               a.sourcePos.toString()));
+                                               str));
+        }
         return ret;
     }
 
-- 
cgit v1.2.3