mirror of
https://github.com/storytold/photocraft.git
synced 2026-10-31 11:42:13 +01:00
77 lines
3.2 KiB
Rust
77 lines
3.2 KiB
Rust
|
//! Colour-managed canvas display cost on a 36 MP document (#46).
|
|||
|
|
//!
|
|||
|
|
//! ```sh
|
|||
|
|
//! cargo run --release -p photocraft-engine --example bench_display -- [--size 7360x4912] [--reps 5]
|
|||
|
|
//! ```
|
|||
|
|
//!
|
|||
|
|
//! Times, per profile: the composite (shared by every path), the CPU canvas conversion before
|
|||
|
|
//! (`Buffer::to_rgba8`, no colour management) and after (`CanvasDisplay::to_rgba8`), the
|
|||
|
|
//! per-frame display lookup (`display_signature`, what the GPU canvas checks each frame) and
|
|||
|
|
//! building the 33³ GPU display LUT (once per profile/monitor change).
|
|||
|
|
|
|||
|
|
use std::time::Instant;
|
|||
|
|
|
|||
|
|
use photocraft_cms::Builtin;
|
|||
|
|
use photocraft_color::{Color, ColorMode, SampleType};
|
|||
|
|
use photocraft_doc::{Document, Size};
|
|||
|
|
use photocraft_engine::Session;
|
|||
|
|
|
|||
|
|
fn arg(args: &[String], name: &str) -> Option<String> {
|
|||
|
|
args.iter().position(|a| a == name).and_then(|i| args.get(i + 1).cloned())
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fn median(mut v: Vec<f64>) -> f64 {
|
|||
|
|
v.sort_by(|a, b| a.total_cmp(b));
|
|||
|
|
v[v.len() / 2]
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fn time(reps: usize, mut f: impl FnMut()) -> f64 {
|
|||
|
|
median(
|
|||
|
|
(0..reps)
|
|||
|
|
.map(|_| {
|
|||
|
|
let t = Instant::now();
|
|||
|
|
f();
|
|||
|
|
t.elapsed().as_secs_f64() * 1000.0
|
|||
|
|
})
|
|||
|
|
.collect(),
|
|||
|
|
)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fn main() {
|
|||
|
|
let args: Vec<String> = std::env::args().collect();
|
|||
|
|
let (w, h) =
|
|||
|
|
arg(&args, "--size").and_then(|s| s.split_once('x').map(|(a, b)| (a.parse().unwrap_or(7360), b.parse().unwrap_or(4912)))).unwrap_or((7360, 4912));
|
|||
|
|
let reps: usize = arg(&args, "--reps").and_then(|v| v.parse().ok()).unwrap_or(5).max(1);
|
|||
|
|
println!("document {w}×{h} ({:.1} MP), median of {reps}", (w * h) as f64 / 1e6);
|
|||
|
|
println!("{:<22} {:>10} {:>14} {:>14} {:>12} {:>10}", "profile", "composite", "CPU before", "CPU after", "per frame", "LUT build");
|
|||
|
|
for (name, profile, depth) in [
|
|||
|
|
("sRGB (untagged)", None, SampleType::U8),
|
|||
|
|
("Display P3", Some(Builtin::DisplayP3), SampleType::U8),
|
|||
|
|
("ProPhoto", Some(Builtin::ProPhotoCompat), SampleType::U16),
|
|||
|
|
("linear sRGB (EXR)", Some(Builtin::LinearSrgb), SampleType::F32),
|
|||
|
|
] {
|
|||
|
|
let mut d = Document::with_background("b", Size::new(w, h), ColorMode::Rgb, depth, Color::rgb(0.7, 0.4, 0.2));
|
|||
|
|
d.icc_profile = profile.map(|b| b.profile().to_bytes());
|
|||
|
|
let s = Session::new();
|
|||
|
|
let mut buf = None;
|
|||
|
|
let composite = time(reps, || buf = Some(photocraft_compose::flatten(&d)));
|
|||
|
|
let buf = buf.expect("composited");
|
|||
|
|
let before = time(reps, || {
|
|||
|
|
std::hint::black_box(buf.to_rgba8());
|
|||
|
|
});
|
|||
|
|
let display = s.color.canvas_display(&d).expect("display");
|
|||
|
|
let after = time(reps, || {
|
|||
|
|
let d = s.color.canvas_display(&d).expect("display");
|
|||
|
|
std::hint::black_box(if d.is_identity() { buf.to_rgba8() } else { d.to_rgba8(&buf) });
|
|||
|
|
});
|
|||
|
|
let frame = time(reps * 200, || {
|
|||
|
|
std::hint::black_box(s.color.display_signature(&d));
|
|||
|
|
});
|
|||
|
|
let lut = time(reps, || {
|
|||
|
|
std::hint::black_box(s.color.canvas_lut(&d, 33).expect("lut"));
|
|||
|
|
});
|
|||
|
|
let tag = if display.is_identity() { " (identity)" } else { "" };
|
|||
|
|
println!("{:<22} {composite:>8.1}ms {before:>12.1}ms {after:>12.1}ms {:>10.1}µs {lut:>8.1}ms{tag}", name, frame * 1000.0);
|
|||
|
|
}
|
|||
|
|
}
|