1use cuneus::compute::*;
2use cuneus::prelude::*;
3
4cuneus::uniform_params! {
5 struct LichParams {
6 cloud_density: f32,
7 lightning_intensity: f32,
8 branch_count: f32,
9 feedback_decay: f32,
10 base_color: [f32; 3],
11 _pad1: f32,
12 color_shift: f32,
13 spectrum_mix: f32,
14 _pad2: [f32; 2]}
15}
16
17struct LichShader {
18 base: RenderKit,
19 compute_shader: ComputeShader,
20 current_params: LichParams}
21
22impl LichShader {
23 fn clear_buffers(&mut self, core: &Core) {
24 self.compute_shader.clear_all_buffers(core);
26 }
27}
28
29impl ShaderManager for LichShader {
30 fn init(core: &Core) -> Self {
31 let base = RenderKit::new(core);
32
33 let passes = vec![
34 PassDescription::new("lightning", &[]),
35 PassDescription::new("feedback", &["lightning", "feedback"]), PassDescription::new("main_image", &["feedback"]),
37 ];
38
39 let config = ComputeShader::builder()
40 .with_multi_pass(&passes)
41 .with_custom_uniforms::<LichParams>()
42 .with_workgroup_size([16, 16, 1])
43 .with_texture_format(cuneus::compute::COMPUTE_TEXTURE_FORMAT_RGBA16)
44 .with_label("Lich Lightning")
45 .build();
46
47 let compute_shader = cuneus::compute_shader!(core, "shaders/lich.wgsl", config);
48
49 let initial_params = LichParams {
50 cloud_density: 3.0,
51 lightning_intensity: 1.0,
52 branch_count: 1.0,
53 feedback_decay: 0.98,
54 base_color: [1.0, 1.0, 1.0],
55 _pad1: 0.0,
56 color_shift: 2.0,
57 spectrum_mix: 0.5,
58 _pad2: [0.0; 2]};
59
60 compute_shader.set_custom_params(initial_params, &core.queue);
62
63 Self {
64 base,
65 compute_shader,
66 current_params: initial_params}
67 }
68
69 fn update(&mut self, core: &Core) {
70 self.compute_shader.handle_export(core, &mut self.base);
72
73 let current_time = self.base.controls.get_time(&self.base.start_time);
75 let delta = 1.0 / 60.0;
76 self.compute_shader
77 .set_time(current_time, delta, &core.queue);
78 }
79
80 fn resize(&mut self, core: &Core) {
81 self.compute_shader
82 .resize(core, core.size.width, core.size.height);
83 }
84
85 fn render(&mut self, core: &Core) -> Result<(), cuneus::SurfaceError> {
86 let mut frame = self.base.begin_frame(core)?;
87
88 let mut params = self.current_params;
89 let mut changed = false;
90 let mut should_start_export = false;
91 let mut export_request = self.base.export_manager.get_ui_request();
92 let mut controls_request = self
93 .base
94 .controls
95 .get_ui_request(&self.base.start_time, &core.size, self.base.fps_tracker.fps());
96
97 let full_output = if self.base.key_handler.show_ui {
98 self.base.render_ui(core, |ctx| {
99 RenderKit::apply_default_style(ctx);
100
101 egui::Window::new("Lich Lightning")
102 .collapsible(true)
103 .resizable(true)
104 .default_width(300.0)
105 .show(ctx, |ui| {
106 egui::CollapsingHeader::new("Lightning Parameters")
107 .default_open(true)
108 .show(ui, |ui| {
109 changed |= ui
110 .add(
111 egui::Slider::new(&mut params.cloud_density, 0.0..=24.0)
112 .text("Seed"),
113 )
114 .changed();
115 changed |= ui
116 .add(
117 egui::Slider::new(
118 &mut params.lightning_intensity,
119 0.1..=6.0,
120 )
121 .text("Lightning"),
122 )
123 .changed();
124 changed |= ui
125 .add(
126 egui::Slider::new(&mut params.branch_count, 0.0..=2.0)
127 .text("Branch"),
128 )
129 .changed();
130 changed |= ui
131 .add(
132 egui::Slider::new(&mut params.feedback_decay, 0.1..=1.5)
133 .text("Decay"),
134 )
135 .changed();
136 });
137
138 egui::CollapsingHeader::new("Color Settings")
139 .default_open(false)
140 .show(ui, |ui| {
141 let mut color = params.base_color;
142 if ui.color_edit_button_rgb(&mut color).changed() {
143 params.base_color = color;
144 changed = true;
145 }
146 changed |= ui
147 .add(
148 egui::Slider::new(&mut params.color_shift, 0.1..=20.0)
149 .text("Temperature"),
150 )
151 .changed();
152 changed |= ui
153 .add(
154 egui::Slider::new(&mut params.spectrum_mix, 0.0..=1.0)
155 .text("Spectral"),
156 )
157 .changed();
158 });
159
160 ui.separator();
161 ShaderControls::render_controls_widget(ui, &mut controls_request);
162
163 ui.separator();
164 should_start_export =
165 ExportManager::render_export_ui_widget(ui, &mut export_request);
166
167 ui.separator();
168 ui.label("Electric lightning with atomic buffer accumulation");
169 });
170 })
171 } else {
172 self.base.render_ui(core, |_ctx| {})
173 };
174
175 self.compute_shader.dispatch(&mut frame.encoder, core);
176
177 self.base.renderer.render_to_view(&mut frame.encoder, &frame.view, &self.compute_shader.get_output_texture().bind_group);
178
179 if controls_request.should_clear_buffers {
181 self.clear_buffers(core);
182 }
183 self.base.apply_control_request(controls_request.clone());
184
185 self.base.export_manager.apply_ui_request(export_request);
186 if should_start_export {
187 self.base.export_manager.start_export();
188 }
189
190 if changed {
191 self.current_params = params;
192 self.compute_shader.set_custom_params(params, &core.queue);
193 }
194
195 self.base.end_frame(core, frame, full_output);
196
197 Ok(())
198 }
199
200 fn handle_input(&mut self, core: &Core, event: &WindowEvent) -> bool {
201 self.base.default_handle_input(core, event)
202 }
203}
204
205fn main() -> Result<(), Box<dyn std::error::Error>> {
206 env_logger::init();
207 let (app, event_loop) = ShaderApp::new("Lich Lightning", 800, 600);
208 app.run(event_loop, LichShader::init)
209}