export renders in separate thread

This commit is contained in:
Crispy 2023-07-14 13:37:46 +02:00
parent 67168f919a
commit c5b961a33a

View file

@ -3,6 +3,8 @@ use std::{
fs::{self, File}, fs::{self, File},
io::Write, io::Write,
path::PathBuf, path::PathBuf,
sync::mpsc::{self, Receiver, Sender},
thread::{self, JoinHandle},
time::SystemTime, time::SystemTime,
}; };
@ -51,6 +53,19 @@ struct JuliaGUI {
#[serde(skip)] #[serde(skip)]
settings_changed: bool, settings_changed: bool,
preview_point: bool, preview_point: bool,
#[serde(skip)]
render_thread_handle: Option<JoinHandle<()>>,
#[serde(skip)]
render_thread: Option<Sender<RenderJob>>,
#[serde(skip)]
render_result: Option<Receiver<f64>>,
#[serde(skip)]
waiting: bool,
}
enum RenderJob {
Render(PathBuf, RenderOptions, (u8, u8, u8)),
Exit,
} }
impl Default for JuliaGUI { impl Default for JuliaGUI {
@ -65,7 +80,11 @@ impl Default for JuliaGUI {
export_iterations: 512, export_iterations: 512,
export_path: "".into(), export_path: "".into(),
settings_changed: true, settings_changed: true,
preview_point: true, preview_point: false,
render_thread_handle: None,
render_thread: None,
render_result: None,
waiting: false,
} }
} }
} }
@ -84,9 +103,34 @@ impl JuliaGUI {
TextureOptions::default(), TextureOptions::default(),
); );
let (job_sender, job_receiver) = mpsc::channel::<RenderJob>();
let (result_sender, result_receiver) = mpsc::channel::<f64>();
let render_thread = thread::Builder::new()
.name("render".into())
.spawn(move || loop {
if let Ok(job) = job_receiver.recv() {
match job {
RenderJob::Exit => break,
RenderJob::Render(path, options, color) => {
let start_time = SystemTime::now();
let image = render(&options, color);
if let Err(err) = image.save(&path) {
println!("Failed to save render: {err}");
}
let time = start_time.elapsed().unwrap().as_micros() as f64 / 1000.0;
result_sender.send(time).unwrap();
}
}
}
})
.unwrap();
n.preview = Some(preview); n.preview = Some(preview);
n.settings_changed = true; n.settings_changed = true;
n.export_path = "julia_fractal.png".into(); n.export_path = "julia_fractal.png".into();
n.render_thread_handle = Some(render_thread);
n.render_thread = Some(job_sender);
n.render_result = Some(result_receiver);
n n
} }
@ -116,7 +160,8 @@ impl JuliaGUI {
} }
fn export_render(&mut self) { fn export_render(&mut self) {
let start_time = SystemTime::now(); self.save_settings();
if let Some(channel) = &self.render_thread {
let res_mul = 1 << self.export_res_power; let res_mul = 1 << self.export_res_power;
let settings = RenderOptions { let settings = RenderOptions {
width: self.render_options.width * res_mul, width: self.render_options.width * res_mul,
@ -124,12 +169,16 @@ impl JuliaGUI {
max_iterations: self.export_iterations, max_iterations: self.export_iterations,
..self.render_options.clone() ..self.render_options.clone()
}; };
let image = render(&settings, self.color);
if let Err(err) = image.save(&self.export_path) { channel
println!("Error exporting render: {err}"); .send(RenderJob::Render(
self.export_path.clone(),
settings,
self.color,
))
.expect("failed to start render job");
self.waiting = true;
} }
self.export_render_ms = Some(start_time.elapsed().unwrap().as_micros() as f64 / 1000.0);
self.save_settings();
} }
fn export_render_new_path(&mut self) { fn export_render_new_path(&mut self) {
@ -152,6 +201,11 @@ impl eframe::App for JuliaGUI {
self.settings_changed = false; self.settings_changed = false;
} }
if let Some(result) = self.render_result.as_mut().and_then(|r| r.try_recv().ok()) {
self.export_render_ms = Some(result);
self.waiting = false;
}
egui::SidePanel::left("main_left_panel") egui::SidePanel::left("main_left_panel")
.resizable(false) .resizable(false)
.exact_width(200.0) .exact_width(200.0)
@ -280,6 +334,7 @@ impl eframe::App for JuliaGUI {
)); ));
ui.horizontal(|ui| { ui.horizontal(|ui| {
ui.add_enabled_ui(!self.waiting, |ui| {
let export_text = if self.export_path.is_file() { let export_text = if self.export_path.is_file() {
"Overwrite" "Overwrite"
} else { } else {
@ -292,7 +347,10 @@ impl eframe::App for JuliaGUI {
self.export_render_new_path(); self.export_render_new_path();
} }
}); });
if self.waiting {
ui.spinner();
}
});
ui.label( ui.label(
self.export_path self.export_path
.file_name() .file_name()
@ -323,6 +381,10 @@ impl eframe::App for JuliaGUI {
} }
fn on_exit(&mut self, _gl: Option<&eframe::glow::Context>) { fn on_exit(&mut self, _gl: Option<&eframe::glow::Context>) {
if let Some(channel) = &self.render_thread {
channel.send(RenderJob::Exit).unwrap();
}
self.render_thread_handle.take().unwrap().join().unwrap();
self.save_settings(); self.save_settings();
} }
} }