noteuclid/minitracing
1
0
Fork 0
Code Issues 5 Pull Requests Actions Packages Projects Releases Wiki Activity

Extract image savers to function

Browse Source
This commit is contained in:
numzero 2024-12-31 00:37:07 +03:00
parent d125635f80
commit 946617f988
1 changed files with 23 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
src/bin/rec.rs
View File

@ -1,7 +1,7 @@
use std::env::args;
use std::error::Error;
use std::io::{stdout, Write};
use std::path::PathBuf;
use std::path::{Path, PathBuf};
use std::sync::Arc;
use std::{fs, io};
@ -50,26 +50,35 @@ fn main() {
}
let n_threads = std::thread::available_parallelism().map(|n| n.get()).unwrap_or(2);
let (img_sender, img_receiver) = async_channel::bounded::<Frame>(2 * n_threads);
std::thread::scope(|s| {
let (img_sender, img_receiver) = async_channel::bounded::<(u32, Arc<wgpu::Buffer>)>(2 * n_threads);
for _ in 0..n_threads {
let img_receiver = img_receiver.clone();
let path = &path;
s.spawn(move || {
while let Ok((frame, buffer)) = img_receiver.recv_blocking() {
let img = image::RgbaImage::from_raw(SIZE.x, SIZE.y, buffer.slice(..).get_mapped_range().to_vec())
.expect("read failure!");
let img: image::RgbImage = img.convert();
img.save(path.join(&format!("frame{frame:06}.webp")))
.expect("save failure!");
}
});
start_saver(s, &path, &img_receiver);
}
do_work(img_sender, start_frame, stop_frame);
});
}
fn do_work(img_sender: async_channel::Sender<(u32, Arc<wgpu::Buffer>)>, start_frame: u32, stop_frame: u32) {
type Frame = (u32, Arc<wgpu::Buffer>);
fn start_saver<'a>(
scope: &'a std::thread::Scope<'a, '_>,
path: &'a Path,
img_receiver: &'a async_channel::Receiver<Frame>,
) {
scope.spawn(move || {
while let Ok((frame, buffer)) = img_receiver.recv_blocking() {
let img = image::RgbaImage::from_raw(SIZE.x, SIZE.y, buffer.slice(..).get_mapped_range().to_vec())
.expect("read failure!");
let img: image::RgbImage = img.convert();
img.save(path.join(&format!("frame{frame:06}.webp")))
.expect("save failure!");
}
});
}
fn do_work(img_sender: async_channel::Sender<Frame>, start_frame: u32, stop_frame: u32) {
println!("Loading...");
let img_sender = Arc::new(img_sender);
let (device, queue) = pollster::block_on(init_gpu()).unwrap();