mirror-github-bspeice/speice.io
1
0
Fork 0
mirror of https://github.com/bspeice/speice.io synced 2025-04-26 15:51:31 -04:00
Code Issues Releases Wiki Activity

Implement image scaling

Browse Source
This commit is contained in:
Bradlee Speice 2025-04-06 20:59:53 -04:00
parent 5806ebd051
commit 98d9111688
2 changed files with 19 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
blog/2025-03-30-draw-compute-shader/draw-compute/shader/src/lib.rs
View File

@ -54,12 +54,16 @@ pub fn main_fs(
#[spirv(storage_buffer, descriptor_set = 0, binding = 1)] image: &mut [glam::Vec4],
output: &mut glam::Vec4,
) {
let pixel_coordinate = frag_coord.xy().as_usizevec2();
let (pixel_x, pixel_y) = (pixel_coordinate.x, pixel_coordinate.y);
let vp_size = viewport.size.as_vec2();
let img_size = viewport.image.as_vec2();
*output = if pixel_x < viewport.image.x as usize && pixel_y < viewport.image.y as usize {
image[image_index(pixel_x, pixel_y, viewport.image.x as usize)]
let scale = (vp_size / img_size).min_element();
let img_offset = (vp_size / scale - img_size) / 2.0;
let img_coord = (frag_coord.xy() - viewport.offset.as_vec2()) / scale - img_offset;
*output = if img_coord.cmpge(glam::Vec2::ZERO).all() && img_coord.cmple(img_size).all() {
image[image_index(img_coord.x as usize, img_coord.y as usize, viewport.image.x as usize)]
} else {
BLACK
};
}
}

12
blog/2025-03-30-draw-compute-shader/draw-compute/src/main.rs
View File

@ -172,6 +172,7 @@ impl DrawResources {
fn resize(&mut self, width: u64, height: u64) {
self.image_buffer = Self::image_buffer(&self.device, width, height);
self.image_size = glam::uvec2(width as u32, height as u32);
self.bind_group = Self::bind_group(
&self.device,
&self.bind_group_layout,
@ -206,7 +207,7 @@ impl CallbackTrait for DrawCallback {
);
}
let mut viewport = shader::Viewport {
let viewport = shader::Viewport {
image: resources.image_size,
offset: glam::uvec2(self.draw_rect.min.x as u32, self.draw_rect.min.y as u32),
size: glam::uvec2(
@ -273,6 +274,13 @@ impl eframe::App for ComputeDraw {
callback_resources.insert(DrawResources::new(&device, &format, 800, 600));
}
egui::TopBottomPanel::bottom("bottom").show(ctx, |ui| {
let wgpu_render_state = frame.wgpu_render_state().expect("missing WGPU state");
let image_size = wgpu_render_state.renderer.as_ref().read().callback_resources.get::<DrawResources>().unwrap().image_size;
ui.label(format!("Viewport: image={image_size}"))
});
egui::CentralPanel::default().show(ctx, |ui| {
egui::Frame::canvas(ui.style()).show(ui, |ui| {
let interact_rect = ui.available_rect_before_wrap();
@ -301,7 +309,7 @@ fn main() {
eframe::run_native(
"Compute Draw",
native_options,
Box::new(|cc| Ok(Box::new(ComputeDraw { initial_draw: true }))),
Box::new(|_cc| Ok(Box::new(ComputeDraw { initial_draw: true }))),
)
.unwrap()
}