1
0
Fork 0
mirror of https://codeberg.org/demostf/parser.git synced 2026-06-03 10:14:06 +02:00

criterion

This commit is contained in:
Robin Appelman 2022-04-18 17:13:20 +02:00
commit d18a75e34e
5 changed files with 491 additions and 44 deletions

View file

@ -1,15 +1,12 @@
#![allow(dead_code)]
#![allow(unused_imports)]
#![allow(unused_variables)]
#![feature(test)]
extern crate test;
use criterion::{black_box, criterion_group, criterion_main, Criterion};
use pretty_assertions::assert_eq;
use std::fs;
use std::collections::{HashMap, HashSet};
use test::Bencher;
use tf_demo_parser::demo::message::Message;
use tf_demo_parser::demo::packet::datatable::{ParseSendTable, SendTableName};
use tf_demo_parser::demo::packet::stringtable::StringTableEntry;
@ -39,23 +36,27 @@ impl MessageHandler for SendPropAnalyser {
}
}
fn flatten_bench(input_file: &str, b: &mut Bencher) {
fn flatten_bench(input_file: &str, b: &mut Criterion) {
let file = fs::read(input_file).expect("Unable to read file");
let demo = Demo::new(&file);
let stream = demo.get_stream();
let (_, send_tables) = DemoParser::new_with_analyser(stream.clone(), SendPropAnalyser)
.parse()
.unwrap();
b.iter(|| {
let flat: Vec<_> = send_tables
.iter()
.map(|table| table.flatten_props(&send_tables))
.collect();
test::black_box(flat);
b.bench_function(&format!("flatten {}", input_file), |b| {
b.iter(|| {
let flat: Vec<_> = send_tables
.iter()
.map(|table| table.flatten_props(&send_tables))
.collect();
black_box(flat);
})
});
}
#[bench]
fn sendprop_test_gully(b: &mut Bencher) {
fn sendprop_test_gully(b: &mut Criterion) {
flatten_bench("test_data/gully.dem", b);
}
criterion_group!(benches, sendprop_test_gully);
criterion_main!(benches);