mirror of
https://github.com/RGBCube/cstree
synced 2025-07-27 09:07:44 +00:00
add tree construction benchmark
This commit is contained in:
parent
87699336bb
commit
c539674fc7
2 changed files with 89 additions and 0 deletions
|
@ -34,6 +34,11 @@ m_lexer = "0.0.4"
|
|||
serde_json = "1.0.61"
|
||||
serde_test = "1.0.119"
|
||||
crossbeam-utils = "0.8"
|
||||
criterion = "0.3"
|
||||
|
||||
[[bench]]
|
||||
name = "main"
|
||||
harness = false
|
||||
|
||||
[features]
|
||||
default = []
|
||||
|
|
84
benches/main.rs
Normal file
84
benches/main.rs
Normal file
|
@ -0,0 +1,84 @@
|
|||
use criterion::{criterion_group, criterion_main, Criterion, Throughput};
|
||||
use cstree::*;
|
||||
use lasso::{Interner, Rodeo};
|
||||
|
||||
#[derive(Debug)]
|
||||
pub enum Element<'s> {
|
||||
Node(Vec<Element<'s>>),
|
||||
Token(&'s str),
|
||||
}
|
||||
|
||||
#[derive(Debug, Copy, Clone, Eq, PartialEq, Ord, PartialOrd, Hash)]
|
||||
pub enum TestLang {}
|
||||
impl Language for TestLang {
|
||||
type Kind = SyntaxKind;
|
||||
|
||||
fn kind_from_raw(raw: SyntaxKind) -> Self::Kind {
|
||||
raw
|
||||
}
|
||||
|
||||
fn kind_to_raw(kind: Self::Kind) -> SyntaxKind {
|
||||
kind
|
||||
}
|
||||
}
|
||||
|
||||
pub fn build_tree_with_cache<'c, 'i, I>(root: &Element<'_>, cache: &'c mut NodeCache<'i, I>) -> GreenNode
|
||||
where
|
||||
I: Interner,
|
||||
{
|
||||
let mut builder = GreenNodeBuilder::with_cache(cache);
|
||||
build_recursive(root, &mut builder, 0);
|
||||
let (node, interner) = builder.finish();
|
||||
assert!(interner.is_none());
|
||||
node
|
||||
}
|
||||
|
||||
pub fn build_recursive<'c, 'i, I>(root: &Element<'_>, builder: &mut GreenNodeBuilder<'c, 'i, I>, mut from: u16) -> u16
|
||||
where
|
||||
I: Interner,
|
||||
{
|
||||
match root {
|
||||
Element::Node(children) => {
|
||||
builder.start_node(SyntaxKind(from));
|
||||
for child in children {
|
||||
from = build_recursive(child, builder, from + 1);
|
||||
}
|
||||
builder.finish_node();
|
||||
}
|
||||
Element::Token(text) => {
|
||||
builder.token(SyntaxKind(from), *text);
|
||||
}
|
||||
}
|
||||
from
|
||||
}
|
||||
|
||||
fn two_level_tree() -> Element<'static> {
|
||||
use Element::*;
|
||||
Node(vec![
|
||||
Node(vec![Token("0.0"), Token("0.1")]),
|
||||
Node(vec![Token("1.0")]),
|
||||
Node(vec![Token("2.0"), Token("2.1"), Token("2.2")]),
|
||||
])
|
||||
}
|
||||
|
||||
pub fn create(c: &mut Criterion) {
|
||||
let mut group = c.benchmark_group("qualification");
|
||||
group.throughput(Throughput::Elements(1));
|
||||
|
||||
let mut interner = Rodeo::new();
|
||||
let mut cache = NodeCache::with_interner(&mut interner);
|
||||
let tree = two_level_tree();
|
||||
|
||||
group.bench_function("two-level tree", |b| {
|
||||
b.iter(|| {
|
||||
for _ in 0..100_000 {
|
||||
let _tree = build_tree_with_cache(&tree, &mut cache);
|
||||
}
|
||||
})
|
||||
});
|
||||
|
||||
group.finish();
|
||||
}
|
||||
|
||||
criterion_group!(benches, create);
|
||||
criterion_main!(benches);
|
Loading…
Add table
Add a link
Reference in a new issue