Add code to populate the basic block predecessors

BuildIt-lang · May 16, 2023 · c6141d5 · c6141d5
1 parent 51a3213
commit c6141d5
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/src/blocks/basic_blocks.cpp b/src/blocks/basic_blocks.cpp
@@ -153,5 +153,12 @@ std::vector<std::shared_ptr<basic_block>> generate_basic_blocks(block::stmt_bloc
         }
     }
 
+    // step 5: populate the predecessors
+    for (auto bb: return_list) {
+        for (auto succ: bb->successor) {
+            succ->predecessor.push_back(bb);
+        }
+    }
+
     return return_list;
 }
diff --git a/src/blocks/loop_finder.cpp b/src/blocks/loop_finder.cpp
@@ -133,7 +133,11 @@ void loop_finder::visit(stmt_block::Ptr a) {
 
 	std::cout << "++++++ basic blocks ++++++ \n";
 	for (auto bb: BBs) {
-		std::cout << bb->name << ":" << "\n";
+		std::cout << bb->name << ":" << "  ; ";
+		for (auto pred: bb->predecessor) {
+			std::cout << pred->name << ", ";
+		}
+		std::cout << "\n";
 		if (bb->branch_expr) {
 			std::cout << "  ";
 			bb->branch_expr->dump(std::cout, 0);