feat: add event trace visualization tools and enhance simulator with trace file support

Author: shirohasuki

.gitignore

@@ -1,3 +1,5 @@
 target/
 Cargo.lock
 build/
+
+.json

README.md

@@ -21,7 +21,11 @@ cargo build --release --bin bebop
 ```
 
 3. Run the simulation
+4. 
 ```
 cd bebop
 ./target/release/bebop
+
+# run in quiet with only workload logs
+cargo run --release --bin bebop -- -q
 ```

bebop/Cargo.toml

@@ -11,3 +11,5 @@ path = "src/bin/bebop.rs"
 sim = "0.13"
 serde = { version = "1.0", features = ["derive"] }
 serde_json = "1.0"
+rustyline = "14.0"
+clap = { version = "4.5", features = ["derive"] }

bebop/src/bin/bebop.rs

@@ -1,20 +1,40 @@
 use bebop::simulator::sim::mode::{SimConfig, StepMode};
 use bebop::simulator::Simulator;
+use clap::Parser;
 
-use std::env;
+/// Bebop - A RISC-V NPU simulator
+#[derive(Parser, Debug)]
+#[command(name = "bebop")]
+#[command(version = "0.1.0")]
+#[command(about = "Bebop simulator developed by buckyball", long_about = None)]
+struct Args {
+  /// Enable step mode (interactive stepping)
+  #[arg(short, long)]
+  step: bool,
+
+  /// Quiet mode (suppress log messages)
+  #[arg(short, long)]
+  quiet: bool,
+
+  /// Output trace file path
+  #[arg(long, value_name = "FILE")]
+  trace_file: Option<String>,
+}
 
 fn main() -> std::io::Result<()> {
-  let args: Vec<String> = env::args().collect();
+  let args = Args::parse();
 
-  let step_mode = if args.iter().any(|arg| arg == "--step" || arg == "-s") {
+  let step_mode = if args.step {
     StepMode::Step
   } else {
     StepMode::Continuous
   };
 
-  let quiet = args.iter().any(|arg| arg == "--quiet" || arg == "-q");
-
-  let config = SimConfig { quiet, step_mode };
+  let config = SimConfig {
+    quiet: args.quiet,
+    step_mode,
+    trace_file: args.trace_file,
+  };
 
   let mut simulator = Simulator::new(config)?;
   simulator.run()

bebop/src/buckyball/inject.rs

@@ -20,7 +20,7 @@ pub fn inject_message(
   content: &str,
 ) {
   let msg = Message::new(
-    source_id.unwrap_or("default").to_string(),
+    source_id.unwrap_or("host").to_string(),
     source_port.unwrap_or("default").to_string(),
     target_model.to_string(),
     target_port.unwrap_or("default").to_string(),

bebop/src/buckyball/tdma.rs

@@ -198,6 +198,7 @@ impl DevsModel for Tdma {
         port_name: self.commit_to_rob_port.clone(),
       });
       self.all_bank_read_iter = 0; // Reset to avoid re-sending
+      self.until_next_event = 1.0;
       has_work = true;
     }
 
@@ -211,6 +212,7 @@ impl DevsModel for Tdma {
         port_name: self.commit_to_rob_port.clone(),
       });
       self.all_bank_write_iter = 0; // Reset to avoid re-sending
+      self.until_next_event = 1.0;
       has_work = true;
     }
 

bebop/src/prof/etrace/README.md

@@ -0,0 +1,15 @@
+# Event Trace Visualization
+
+Event trace visualization tools for the Bebop simulator. Analyzes JSON Lines trace files and generates interactive HTML visualizations.
+
+**graph.py**: Generates an interactive network graph showing module connections and event statistics.
+**timeline.py**: Creates a timeline visualization displaying events for each module over time.
+
+## Usage
+
+```bash
+python graph.py trace.jsonl [output.html]
+python timeline.py trace.jsonl [output.html]
+```
+
+Both tools read JSON Lines format and produce standalone HTML files viewable in any browser.