more bench

benches/bench.rs

@@ -1,5 +1,6 @@
 use criterion::{black_box, criterion_group, criterion_main, Criterion};
 use std::fs;
+use tf_demo_parser::demo::parser::gamestateanalyser::GameStateAnalyser;
 use tf_demo_parser::{Demo, DemoParser, Stream};
 
 fn bench_file(input_file: &str, b: &mut Criterion) {
@@ -15,13 +16,41 @@ fn bench_file(input_file: &str, b: &mut Criterion) {
     });
 }
 
+fn bench_gamestate(input_file: &str, b: &mut Criterion) {
+    let file = fs::read(input_file).expect("Unable to read file");
+    let demo = Demo::new(&file);
+    let stream: Stream = demo.get_stream();
+
+    b.bench_function(&format!("bench gamestate {}", input_file), |b| {
+        b.iter(|| {
+            let (_, state) =
+                DemoParser::new_with_analyser(stream.clone(), GameStateAnalyser::default())
+                    .parse()
+                    .unwrap();
+            black_box(state);
+        })
+    });
+}
+
 fn bench_gully(b: &mut Criterion) {
     bench_file("test_data/gully.dem", b);
 }
-
 fn bench_comp(b: &mut Criterion) {
     bench_file("test_data/comp.dem", b);
 }
 
-criterion_group!(benches, bench_comp, bench_gully);
+fn bench_gamestate_gully(b: &mut Criterion) {
+    bench_gamestate("test_data/gully.dem", b);
+}
+fn bench_gamestate_comp(b: &mut Criterion) {
+    bench_gamestate("test_data/comp.dem", b);
+}
+
+criterion_group!(
+    benches,
+    bench_comp,
+    bench_gully,
+    bench_gamestate_comp,
+    bench_gamestate_gully
+);
 criterion_main!(benches);