examples/neck_pivot_probe.rs

neck_pivot_probe.rs 9.1 KiB Download Raw

1	//! Real Mesa/Xvfb proof that neck pivot differs from center-only orbit at pitched yaw.
2	//! Usage: OUTDIR=... xvfb-run -a cargo run --example neck_pivot_probe --locked
3	use paperdoll_renderer::{camera, preview, production, project::CameraFraming};
4	use std::{
5	path::{Path, PathBuf},
6	sync::{Arc, Mutex},
7	};
8
9	struct Quit;
10	impl eframe::App for Quit {
11	fn update(&mut self, ctx: &eframe::egui::Context, _: &mut eframe::Frame) {
12	ctx.send_viewport_cmd(eframe::egui::ViewportCommand::Close);
13	}
14	}
15
16	fn write_rgba(path: &Path, width: u32, height: u32, rgba: &[u8]) -> Result<(), String> {
17	let file = std::fs::File::create(path).map_err(\|e\| e.to_string())?;
18	let mut encoder = png::Encoder::new(file, width, height);
19	encoder.set_color(png::ColorType::Rgba);
20	encoder.set_depth(png::BitDepth::Eight);
21	encoder
22	.write_header()
23	.map_err(\|e\| e.to_string())?
24	.write_image_data(rgba)
25	.map_err(\|e\| e.to_string())
26	}
27
28	fn downsample(src: &[u8], sw: u32, sh: u32, dw: u32, dh: u32) -> Vec<u8> {
29	let mut out = vec![0u8; (dw * dh * 4) as usize];
30	for y in 0..dh {
31	for x in 0..dw {
32	let sx = x * sw / dw;
33	let sy = y * sh / dh;
34	let si = ((sy * sw + sx) * 4) as usize;
35	let di = ((y * dw + x) * 4) as usize;
36	out[di..di + 4].copy_from_slice(&src[si..si + 4]);
37	}
38	}
39	out
40	}
41
42	fn main() -> Result<(), Box<dyn std::error::Error>> {
43	let out = PathBuf::from(
44	std::env::var("OUTDIR").unwrap_or_else(\|_\| "/workspace/paperdoll-neck-artifacts".into()),
45	);
46	std::fs::create_dir_all(&out)?;
47	let outcome = Arc::new(Mutex::new(None));
48	let result = outcome.clone();
49	let options = eframe::NativeOptions {
50	viewport: eframe::egui::ViewportBuilder::default().with_inner_size([64., 64.]),
51	..paperdoll_renderer::preview_gl::native_options()
52	};
53	eframe::run_native(
54	"Paperdoll neck pivot probe",
55	options,
56	Box::new(move \|creation\| {
57	let test = (\|\| -> Result<serde_json::Value, String> {
58	let gl = creation
59	.gl
60	.as_ref()
61	.ok_or_else(\|\| "geen Glow-context".to_string())?;
62	let body = preview::load_gltf_file(
63	&Path::new(env!("CARGO_MANIFEST_DIR"))
64	.join("tests/fixtures/bodies/test-body-male.glb"),
65	)?;
66	let resolved = camera::resolve(&body, CameraFraming::default());
67	let neck_y = resolved.framing.pivot_y_milli.ok_or("nekpivot ontbreekt")?;
68	if neck_y == CameraFraming::default().center_y_milli {
69	return Err("nekpivot mag niet gelijk zijn aan default centrum".into());
70	}
71	let mut renderer = production::GpuRenderer::default();
72	let mut cache = paperdoll_renderer::equipment::TextureCache::default();
73	let pitch = 30.;
74	let yaw = 90.;
75	let mut job = production::RenderJob {
76	character: body,
77	sex: "male".into(),
78	preset: None,
79	target_width: 64,
80	target_height: 96,
81	fps: 12,
82	framing: CameraFraming {
83	height_milli: 2600,
84	center_y_milli: 1000,
85	pivot_y_milli: Some(1000),
86	},
87	output: out.clone(),
88	};
89	let old_scene = production::scene(&job, "walk", 0.0, pitch, yaw, &mut cache)?;
90	let old = unsafe { renderer.render(gl, &old_scene) }?;
91	job.framing.pivot_y_milli = None;
92	let neck_scene = production::scene(&job, "walk", 0.0, pitch, yaw, &mut cache)?;
93	let neck = unsafe { renderer.render(gl, &neck_scene) }?;
94	unsafe { renderer.destroy(gl) };
95	assert_eq!((old.width, old.height), (neck.width, neck.height));
96	let mut changed = 0usize;
97	for (a, b) in old
98	.diffuse
99	.chunks_exact(4)
100	.zip(neck.diffuse.chunks_exact(4))
101	{
102	if a != b {
103	changed += 1;
104	}
105	}
106	if changed < 200 {
107	return Err(format!(
108	"te weinig pixelverschil tussen centrum-pivot en nekpivot: {changed}"
109	));
110	}
111	// Zero-pitch image center / feet framing must stay close after raising pivot.
112	job.framing.pivot_y_milli = Some(1000);
113	let old0 = production::scene(&job, "walk", 0.0, 0., 0., &mut cache)?;
114	job.framing.pivot_y_milli = None;
115	let neck0 = production::scene(&job, "walk", 0.0, 0., 0., &mut cache)?;
116	let old0b = unsafe {
117	let mut r = production::GpuRenderer::default();
118	let b = r.render(gl, &old0)?;
119	r.destroy(gl);
120	b
121	};
122	let neck0b = unsafe {
123	let mut r = production::GpuRenderer::default();
124	let b = r.render(gl, &neck0)?;
125	r.destroy(gl);
126	b
127	};
128	let mid = ((old0b.height / 2) * old0b.width + old0b.width / 2) as usize;
129	let feet = ((old0b.height * 9 / 10) * old0b.width + old0b.width / 2) as usize;
130	let mid_delta = old0b.diffuse[mid * 4..mid * 4 + 4]
131	.iter()
132	.zip(&neck0b.diffuse[mid * 4..mid * 4 + 4])
133	.map(\|(a, b)\| a.abs_diff(*b) as u32)
134	.sum::<u32>();
135	let feet_delta = old0b.diffuse[feet * 4..feet * 4 + 4]
136	.iter()
137	.zip(&neck0b.diffuse[feet * 4..feet * 4 + 4])
138	.map(\|(a, b)\| a.abs_diff(*b) as u32)
139	.sum::<u32>();
140	write_rgba(
141	&out.join("center-pivot-p30-y90.png"),
142	old.width,
143	old.height,
144	&old.diffuse,
145	)?;
146	write_rgba(
147	&out.join("neck-pivot-p30-y90.png"),
148	neck.width,
149	neck.height,
150	&neck.diffuse,
151	)?;
152	let contact_w = 256;
153	let contact_h = 192;
154	let left = downsample(
155	&old.diffuse,
156	old.width,
157	old.height,
158	contact_w / 2,
159	contact_h,
160	);
161	let right = downsample(
162	&neck.diffuse,
163	neck.width,
164	neck.height,
165	contact_w / 2,
166	contact_h,
167	);
168	let mut contact = vec![0u8; (contact_w * contact_h * 4) as usize];
169	for y in 0..contact_h {
170	for x in 0..contact_w / 2 {
171	let si = ((y * (contact_w / 2) + x) * 4) as usize;
172	let li = ((y * contact_w + x) * 4) as usize;
173	let ri = ((y * contact_w + contact_w / 2 + x) * 4) as usize;
174	contact[li..li + 4].copy_from_slice(&left[si..si + 4]);
175	contact[ri..ri + 4].copy_from_slice(&right[si..si + 4]);
176	}
177	}
178	write_rgba(
179	&out.join("neck-vs-center-contact.png"),
180	contact_w,
181	contact_h,
182	&contact,
183	)?;
184	let report = serde_json::json!({
185	"success": true,
186	"pitch": pitch,
187	"yaw": yaw,
188	"center_y_milli": 1000,
189	"neck_y_milli": neck_y,
190	"pivot_source": resolved.source,
191	"highres": [old.width, old.height],
192	"changed_pixels_pitch30": changed,
193	"zero_pitch_mid_channel_delta": mid_delta,
194	"zero_pitch_feet_channel_delta": feet_delta,
195	"artifacts": [
196	"center-pivot-p30-y90.png",
197	"neck-pivot-p30-y90.png",
198	"neck-vs-center-contact.png"
199	]
200	});
201	std::fs::write(
202	out.join("neck-pivot-probe.json"),
203	serde_json::to_vec_pretty(&report).unwrap(),
204	)
205	.map_err(\|e\| e.to_string())?;
206	println!("NECK_PIVOT_PROBE {report}");
207	Ok(report)
208	})();
209	*result.lock().unwrap() = Some(test);
210	Ok(Box::new(Quit))
211	}),
212	)?;
213	let value = outcome
214	.lock()
215	.unwrap()
216	.take()
217	.ok_or("GPU-probe is niet uitgevoerd")??;
218	if value["success"] != true {
219	return Err("probe failed".into());
220	}
221	Ok(())
222	}