Benchmark with example, and raw rendering

1 files changed, 10 insertions, 11 deletions

diff --git a/benches/rendering.rs b/benches/rendering.rs
index 549ad21..51e1b13 100644
--- a/benches/rendering.rs
+++ b/benches/rendering.rs
@@ -1,23 +1,22 @@
 use criterion::{black_box, criterion_group, criterion_main, Criterion};
 use std::fs;
-use wmap_renderer::{render_to_png, Configuration, StageType};
+use wmap_renderer::{render_to_surface, Configuration, StageType};
 
 fn benchmark_rendering(c: &mut Criterion) {
-    // Read the huge.wmap file for realistic benchmarking
-    let source = fs::read_to_string("benches/huge.wmap").expect("Failed to read benches/huge.wmap");
-    let map = wmap_parser::parse(&source);
     let config = Configuration::default();
 
-    c.bench_function("render huge map", |b| {
-        b.iter(|| render_to_png(black_box(&map), StageType::default(), &config))
+    let example_source = fs::read_to_string("benches/example.wmap").expect("Failed to read benches/example.wmap");
+    let example_map = wmap_parser::parse(&example_source);
+
+    c.bench_function("render example map", |b| {
+        b.iter(|| render_to_surface(black_box(&example_map), StageType::default(), &config))
     });
 
-    // Also benchmark a simple map
-    let simple_source = "Tea (90, 50)\nCup (80, 40)\nTea -> Cup";
-    let simple_map = wmap_parser::parse(simple_source);
+    let huge_source = fs::read_to_string("benches/huge.wmap").expect("Failed to read benches/huge.wmap");
+    let huge_map = wmap_parser::parse(&huge_source);
 
-    c.bench_function("render simple map", |b| {
-        b.iter(|| render_to_png(black_box(&simple_map), StageType::default(), &config))
+    c.bench_function("render huge map", |b| {
+        b.iter(|| render_to_surface(black_box(&huge_map), StageType::default(), &config))
     });
 }