live_debugger.py

import time
from graphviz import Digraph


class LiveDebugger:
    def __init__(self):
        self.graph = Digraph()
        self.nodes = {}

    def add_nodes(self, program):
        for i, instr in enumerate(program):
            nid = f"N{i}"
            self.nodes[i] = nid
            self.graph.node(nid, instr)

        for i in range(len(program) - 1):
            self.graph.edge(self.nodes[i], self.nodes[i + 1])

    def highlight(self, active_index, filename="live_cfg"):
        g = Digraph()

        for i, nid in self.nodes.items():
            if i == active_index:
                g.node(nid, style="filled", fillcolor="yellow")
            else:
                g.node(nid)

        for i in range(len(self.nodes) - 1):
            g.edge(self.nodes[i], self.nodes[i + 1])

        g.render(f"{filename}_{active_index}", format="png", cleanup=True)

    def animate(self, trace, program):
        self.add_nodes(program)

        for step, instr, regs in trace:
            print(f"STEP {step}: {instr} | REG: {regs}")

            self.highlight(step)
            time.sleep(0.4)