45 lines
2.3 KiB
Rust
45 lines
2.3 KiB
Rust
//! Times the example Invert plug-in on a large image:
|
||
//! `cargo run -p photocraft-plugins --release --example bench_invert -- [width] [height]`
|
||
//! (default 6000 × 4000 = 24 MP, RGBA 8-bit).
|
||
|
||
use std::time::Instant;
|
||
|
||
use photocraft_color::PixelFormat;
|
||
use photocraft_geom::Rect;
|
||
use photocraft_plugins::{Limits, Plugin};
|
||
use photocraft_raster::Surface;
|
||
|
||
fn main() {
|
||
let mut args = std::env::args().skip(1).map(|a| a.parse::<i32>().unwrap_or(0));
|
||
let w = args.next().filter(|v| *v > 0).unwrap_or(6000);
|
||
let h = args.next().filter(|v| *v > 0).unwrap_or(4000);
|
||
let mut s = Surface::new(PixelFormat::RGBA8);
|
||
let row: Vec<f32> = (0..w).flat_map(|x| [(x % 256) as f32 / 255.0, 0.5, 0.25, 1.0]).collect();
|
||
for y in 0..h {
|
||
s.write_region(Rect::new(0, y, w, y + 1), &row);
|
||
}
|
||
let p = Plugin::load(include_bytes!("../tests/fixtures/invert.wasm"), Limits::default()).expect("load");
|
||
if std::env::var("NOOP").is_ok() {
|
||
let noop = wat::parse_str(NOOP).unwrap();
|
||
let p = Plugin::load(&noop, Limits::default()).unwrap();
|
||
let t = Instant::now();
|
||
p.apply(&s, Rect::new(0, 0, w, h), None, &serde_json::json!({})).unwrap();
|
||
println!("noop: {:.0} ms", t.elapsed().as_secs_f64() * 1e3);
|
||
}
|
||
let canvas = Rect::new(0, 0, w, h);
|
||
for run in 0..3 {
|
||
let t = Instant::now();
|
||
let out = p.apply(&s, canvas, None, &serde_json::json!({})).expect("apply");
|
||
println!("run {run}: {w}x{h} ({:.1} MP) in {:.0} ms", (w as f64 * h as f64) / 1e6, t.elapsed().as_secs_f64() * 1e3);
|
||
assert!((out.pixel(1, 1)[0] - (1.0 - 1.0 / 255.0)).abs() < 1e-6);
|
||
}
|
||
}
|
||
|
||
const NOOP: &str = r#"(module (memory (export "memory") 1) (data (i32.const 16) "{\"id\":\"n\",\"name\":\"n\",\"kind\":\"filter\"}")
|
||
(func (export "pc_abi_version") (result i32) (i32.const 1))
|
||
(func (export "pc_manifest") (result i64) (i64.const 0x0000002500000010))
|
||
(func (export "pc_alloc") (param $n i32) (result i32) (local $old i32)
|
||
(local.set $old (memory.grow (i32.shr_u (i32.add (local.get $n) (i32.const 65535)) (i32.const 16))))
|
||
(if (result i32) (i32.eq (local.get $old) (i32.const -1)) (then (i32.const 0)) (else (i32.mul (local.get $old) (i32.const 65536)))))
|
||
(func (export "pc_filter") (param i32 i32 i32 i32 i32 i32 i32 i32) (result i32) (i32.const 0)))"#;
|