Kinda sloppy placeholder for program counter for program printing

calyxir · Jul 17, 2024 · 9d4589e · 9d4589e
1 parent 2aca971
commit 9d4589e
Show file tree

Hide file tree

Showing 2 changed files with 45 additions and 2 deletions.
diff --git a/interp/src/debugger/debugger_core.rs b/interp/src/debugger/debugger_core.rs
@@ -267,7 +267,6 @@ impl<C: AsRef<Context> + Clone> Debugger<C> {
                     .debugging_context
                     .print_watchpoints(self.interpreter.env()),
                 Command::PrintPC(_override_flag) => {
-                    println!("{}", "PLACEHOLDER".bold().underline().italic());
                     self.interpreter.print_pc();
                 }
 

diff --git a/interp/src/flatten/structures/environment/env.rs b/interp/src/flatten/structures/environment/env.rs
@@ -37,6 +37,7 @@ use ahash::HashMap;
 use ahash::HashSet;
 use ahash::HashSetExt;
 use itertools::Itertools;
+use owo_colors::OwoColorize;
 use slog::warn;
 use std::fmt::Debug;
 use std::fmt::Write;
@@ -567,7 +568,50 @@ impl<C: AsRef<Context> + Clone> Environment<C> {
     }
 
     pub fn print_pc(&self) {
-        println!("{:?}", self.pc)
+        let current_nodes = self.pc.iter().filter(|point| {
+            let node = &self.ctx.as_ref().primary[point.control_node_idx];
+            match node {
+                ControlNode::Enable(_) | ControlNode::Invoke(_) => {
+                    let comp_go = self.get_comp_go(point.comp);
+                    self.ports[comp_go].as_bool().unwrap_or_default()
+                }
+
+                _ => false,
+            }
+        });
+
+        let ctx = &self.ctx.as_ref();
+
+        for point in current_nodes {
+            let node = &ctx.primary[point.control_node_idx];
+            match node {
+                ControlNode::Enable(x) => {
+                    println!(
+                        "{}::{}",
+                        self.get_full_name(point.comp),
+                        ctx.lookup_name(x.group()).underline()
+                    );
+                }
+                ControlNode::Invoke(x) => {
+                    let invoked_name = match x.cell {
+                        CellRef::Local(l) => self.get_full_name(
+                            &self.cells[point.comp].unwrap_comp().index_bases
+                                + l,
+                        ),
+                        CellRef::Ref(_) => {
+                            "<REF CELLS NOT YET SUPPORTED>".to_string()
+                        }
+                    };
+
+                    println!(
+                        "{}: invoke {}",
+                        self.get_full_name(point.comp),
+                        invoked_name.underline()
+                    );
+                }
+                _ => unreachable!(),
+            }
+        }
     }
 
     fn get_name_from_cell_and_parent(