test: add wasm bench in rust

rust/Makefile

@@ -11,3 +11,7 @@ test:
 
 clean:
  cargo clean
+
+bench-wasm:
+ cd crates/wasm && \
+ wasm-pack test --firefox --headless --release

rust/crates/tidy-tree/tests/layout_bench.rs

@@ -2,10 +2,11 @@
 
 mod aesthetic_rules;
 mod gen;
+
 use rand::{prelude::StdRng, SeedableRng};
 use tidy_tree::{BasicLayout, Layout, TidyLayout};
 extern crate test;
-use test::Bencher;
+use test::{black_box, Bencher};
 
 #[bench]
 fn bench_tidy_layout(bench: &mut Bencher) {
@@ -16,9 +17,23 @@ fn bench_tidy_layout(bench: &mut Bencher) {
  peer_margin: 10.,
  };
 
- bench.iter(|| {
+ bench.iter(black_box(|| {
+ layout.layout(&mut tree);
+ }));
+}
+
+#[bench]
+fn bench_tidy_layout_large(bench: &mut Bencher) {
+ let mut rng = StdRng::seed_from_u64(1001);
+ let mut tree = gen::gen_tree(&mut rng, 1_000_000);
+ let layout = TidyLayout {
+ parent_child_margin: 10.,
+ peer_margin: 10.,
+ };
+
+ bench.iter(black_box(|| {
  layout.layout(&mut tree);
- });
+ }));
 }
 
 #[bench]
@@ -30,7 +45,7 @@ fn bench_naive_layout(bench: &mut Bencher) {
  peer_margin: 10.,
  };
 
- bench.iter(|| {
+ bench.iter(black_box(|| {
  layout.layout(&mut tree);
- });
+ }));
 }

rust/crates/wasm/Cargo.toml

@@ -5,11 +5,21 @@ edition = "2021"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 [lib]
-crate-type = ["cdylib"]
+crate-type = ["cdylib", "rlib"]
 
 [dependencies]
 wasm-bindgen = "0.2.74"
 tidy-tree = { path = "../tidy-tree" }
 
 [dev-dependencies]
 getrandom = { version = "0.2", features = ["js"] }
+rand = "0.8.5"
+wasm-bindgen-test = "0.3.30"
+
+[dependencies.web-sys]
+version = "0.3.4"
+features = [
+ 'Window',
+ 'Performance',
+ 'PerformanceTiming'
+]

rust/crates/wasm/tests/bench.rs

@@ -0,0 +1,44 @@
+use rand::{prelude::StdRng, SeedableRng};
+use wasm::Tidy;
+use web_sys::{self, Performance};
+
+mod gen;
+use gen::gen_tree;
+extern crate wasm_bindgen_test;
+use wasm_bindgen_test::*;
+
+wasm_bindgen_test_configure!(run_in_browser);
+
+#[wasm_bindgen_test]
+fn bench_tidy() {
+ let performance = get_performance();
+ let mut rng = StdRng::seed_from_u64(1001);
+ let tree = gen_tree(&mut rng, 100_000);
+ let mut layout = Tidy::with_tidy_layout();
+ tree.pre_order_traversal(|node| {
+ if node.parent.is_some() {
+ layout.add_node(
+ node.id,
+ node.width,
+ node.height,
+ unsafe { node.parent.unwrap().as_ref() }.id,
+ );
+ } else {
+ layout.add_node(node.id, node.width, node.height, Tidy::null_id());
+ }
+ });
+ let start = performance.now();
+ let times = 8;
+ for _ in 0..times {
+ layout.layout();
+ }
+ console_log!("{}ms", (performance.now() - start) / (times as f64));
+}
+
+fn get_performance() -> Performance {
+ let window = web_sys::window().expect("should have a window in this context");
+ let performance = window
+ .performance()
+ .expect("performance should be available");
+ performance
+}

rust/crates/wasm/tests/gen.rs

@@ -0,0 +1,39 @@
+use std::ptr::NonNull;
+
+use rand::prelude::*;
+use tidy_tree::{geometry::Coord, Node};
+
+pub fn gen_node(rng: &mut StdRng) -> Node {
+ Node {
+ id: rng.gen(),
+ width: rng.gen_range(5..50) as Coord,
+ height: rng.gen_range(5..50) as Coord,
+ x: 0.,
+ y: 0.,
+ relative_x: 0.,
+ relative_y: 0.,
+ bbox: Default::default(),
+ parent: None,
+ children: vec![],
+ tidy: None,
+ }
+}
+
+pub fn gen_tree(rng: &mut StdRng, num: usize) -> Node {
+ let root = gen_node(rng);
+ let mut nodes: Vec<NonNull<Node>> = vec![(&root).into()];
+ let max_chosen_size = 100;
+ for _ in 0..num {
+ let parent_index = if nodes.len() < max_chosen_size {
+ rng.gen_range(0..nodes.len())
+ } else {
+ rng.gen_range((nodes.len() - max_chosen_size)..nodes.len())
+ };
+ let parent = unsafe { nodes[parent_index].as_mut() };
+ let node = gen_node(rng);
+ parent.append_child(node);
+ nodes.push(parent.children.last().unwrap().as_ref().into());
+ }
+
+ root
+}