Replaced offscreen_blit.wgsl with existing blit.wgsl.

This commit is contained in:
Bingus 2023-06-14 11:10:09 -07:00
parent af099fa6d7
commit d955b3444d
No known key found for this signature in database
GPG key ID: 5F84D2AA40A9F170
2 changed files with 60 additions and 130 deletions

View file

@ -1,16 +1,12 @@
use crate::graphics::color; use crate::graphics::color;
use std::borrow::Cow; use std::borrow::Cow;
use wgpu::util::DeviceExt;
use wgpu::vertex_attr_array;
/// A simple compute pipeline to convert any texture to Rgba8UnormSrgb. /// A simple compute pipeline to convert any texture to Rgba8UnormSrgb.
#[derive(Debug)] #[derive(Debug)]
pub struct Pipeline { pub struct Pipeline {
pipeline: wgpu::RenderPipeline, pipeline: wgpu::RenderPipeline,
vertices: wgpu::Buffer, sampler_bind_group: wgpu::BindGroup,
indices: wgpu::Buffer, texture_layout: wgpu::BindGroupLayout,
sampler: wgpu::Sampler,
layout: wgpu::BindGroupLayout,
} }
#[derive(Debug, Copy, Clone, bytemuck::Pod, bytemuck::Zeroable)] #[derive(Debug, Copy, Clone, bytemuck::Pod, bytemuck::Zeroable)]
@ -22,88 +18,67 @@ struct Vertex {
impl Pipeline { impl Pipeline {
pub fn new(device: &wgpu::Device) -> Self { pub fn new(device: &wgpu::Device) -> Self {
let shader =
device.create_shader_module(wgpu::ShaderModuleDescriptor {
label: Some("iced_wgpu.offscreen.blit.shader"),
source: wgpu::ShaderSource::Wgsl(Cow::Borrowed(include_str!(
"shader/offscreen_blit.wgsl"
))),
});
let vertices =
device.create_buffer_init(&wgpu::util::BufferInitDescriptor {
label: Some("iced_wgpu.offscreen.vertex_buffer"),
contents: bytemuck::cast_slice(&[
//bottom left
Vertex {
ndc: [-1.0, -1.0],
uv: [0.0, 1.0],
},
//bottom right
Vertex {
ndc: [1.0, -1.0],
uv: [1.0, 1.0],
},
//top right
Vertex {
ndc: [1.0, 1.0],
uv: [1.0, 0.0],
},
//top left
Vertex {
ndc: [-1.0, 1.0],
uv: [0.0, 0.0],
},
]),
usage: wgpu::BufferUsages::VERTEX,
});
let indices =
device.create_buffer_init(&wgpu::util::BufferInitDescriptor {
label: Some("iced_wgpu.offscreen.index_buffer"),
contents: bytemuck::cast_slice(&[0u16, 1, 2, 2, 3, 0]),
usage: wgpu::BufferUsages::INDEX,
});
let sampler = device.create_sampler(&wgpu::SamplerDescriptor { let sampler = device.create_sampler(&wgpu::SamplerDescriptor {
label: Some("iced_wgpu.offscreen.sampler"), label: Some("iced_wgpu.offscreen.sampler"),
..Default::default() ..Default::default()
}); });
let bind_group_layout = //sampler in 0
let sampler_layout =
device.create_bind_group_layout(&wgpu::BindGroupLayoutDescriptor { device.create_bind_group_layout(&wgpu::BindGroupLayoutDescriptor {
label: Some("iced_wgpu.offscreen.blit.bind_group_layout"), label: Some("iced_wgpu.offscreen.blit.sampler_layout"),
entries: &[ entries: &[wgpu::BindGroupLayoutEntry {
wgpu::BindGroupLayoutEntry { binding: 0,
binding: 0, visibility: wgpu::ShaderStages::FRAGMENT,
visibility: wgpu::ShaderStages::FRAGMENT, ty: wgpu::BindingType::Sampler(
ty: wgpu::BindingType::Texture { wgpu::SamplerBindingType::NonFiltering,
sample_type: wgpu::TextureSampleType::Float { ),
filterable: false, count: None,
}, }],
view_dimension: wgpu::TextureViewDimension::D2, });
multisampled: false,
let sampler_bind_group =
device.create_bind_group(&wgpu::BindGroupDescriptor {
label: Some("iced_wgpu.offscreen.sampler.bind_group"),
layout: &sampler_layout,
entries: &[wgpu::BindGroupEntry {
binding: 0,
resource: wgpu::BindingResource::Sampler(&sampler),
}],
});
let texture_layout =
device.create_bind_group_layout(&wgpu::BindGroupLayoutDescriptor {
label: Some("iced_wgpu.offscreen.blit.texture_layout"),
entries: &[wgpu::BindGroupLayoutEntry {
binding: 0,
visibility: wgpu::ShaderStages::FRAGMENT,
ty: wgpu::BindingType::Texture {
sample_type: wgpu::TextureSampleType::Float {
filterable: false,
}, },
count: None, view_dimension: wgpu::TextureViewDimension::D2,
multisampled: false,
}, },
wgpu::BindGroupLayoutEntry { count: None,
binding: 1, }],
visibility: wgpu::ShaderStages::FRAGMENT,
ty: wgpu::BindingType::Sampler(
wgpu::SamplerBindingType::NonFiltering,
),
count: None,
},
],
}); });
let pipeline_layout = let pipeline_layout =
device.create_pipeline_layout(&wgpu::PipelineLayoutDescriptor { device.create_pipeline_layout(&wgpu::PipelineLayoutDescriptor {
label: Some("iced_wgpu.offscreen.blit.pipeline_layout"), label: Some("iced_wgpu.offscreen.blit.pipeline_layout"),
bind_group_layouts: &[&bind_group_layout], bind_group_layouts: &[&sampler_layout, &texture_layout],
push_constant_ranges: &[], push_constant_ranges: &[],
}); });
let shader =
device.create_shader_module(wgpu::ShaderModuleDescriptor {
label: Some("iced_wgpu.offscreen.blit.shader"),
source: wgpu::ShaderSource::Wgsl(Cow::Borrowed(include_str!(
"shader/blit.wgsl"
))),
});
let pipeline = let pipeline =
device.create_render_pipeline(&wgpu::RenderPipelineDescriptor { device.create_render_pipeline(&wgpu::RenderPipelineDescriptor {
label: Some("iced_wgpu.offscreen.blit.pipeline"), label: Some("iced_wgpu.offscreen.blit.pipeline"),
@ -111,14 +86,7 @@ impl Pipeline {
vertex: wgpu::VertexState { vertex: wgpu::VertexState {
module: &shader, module: &shader,
entry_point: "vs_main", entry_point: "vs_main",
buffers: &[wgpu::VertexBufferLayout { buffers: &[],
array_stride: std::mem::size_of::<Vertex>() as u64,
step_mode: wgpu::VertexStepMode::Vertex,
attributes: &vertex_attr_array![
0 => Float32x2, // quad ndc pos
1 => Float32x2, // texture uv
],
}],
}, },
fragment: Some(wgpu::FragmentState { fragment: Some(wgpu::FragmentState {
module: &shader, module: &shader,
@ -156,10 +124,8 @@ impl Pipeline {
Self { Self {
pipeline, pipeline,
vertices, sampler_bind_group,
indices, texture_layout,
sampler,
layout: bind_group_layout,
} }
} }
@ -189,20 +155,15 @@ impl Pipeline {
let view = let view =
&texture.create_view(&wgpu::TextureViewDescriptor::default()); &texture.create_view(&wgpu::TextureViewDescriptor::default());
let bind = device.create_bind_group(&wgpu::BindGroupDescriptor { let texture_bind_group =
label: Some("iced_wgpu.offscreen.blit.bind_group"), device.create_bind_group(&wgpu::BindGroupDescriptor {
layout: &self.layout, label: Some("iced_wgpu.offscreen.blit.texture_bind_group"),
entries: &[ layout: &self.texture_layout,
wgpu::BindGroupEntry { entries: &[wgpu::BindGroupEntry {
binding: 0, binding: 0,
resource: wgpu::BindingResource::TextureView(frame), resource: wgpu::BindingResource::TextureView(frame),
}, }],
wgpu::BindGroupEntry { });
binding: 1,
resource: wgpu::BindingResource::Sampler(&self.sampler),
},
],
});
let mut pass = encoder.begin_render_pass(&wgpu::RenderPassDescriptor { let mut pass = encoder.begin_render_pass(&wgpu::RenderPassDescriptor {
label: Some("iced_wgpu.offscreen.blit.render_pass"), label: Some("iced_wgpu.offscreen.blit.render_pass"),
@ -218,13 +179,9 @@ impl Pipeline {
}); });
pass.set_pipeline(&self.pipeline); pass.set_pipeline(&self.pipeline);
pass.set_bind_group(0, &bind, &[]); pass.set_bind_group(0, &self.sampler_bind_group, &[]);
pass.set_vertex_buffer(0, self.vertices.slice(..)); pass.set_bind_group(1, &texture_bind_group, &[]);
pass.set_index_buffer( pass.draw(0..6, 0..1);
self.indices.slice(..),
wgpu::IndexFormat::Uint16,
);
pass.draw_indexed(0..6u32, 0, 0..1);
texture texture
} }

View file

@ -1,27 +0,0 @@
@group(0) @binding(0) var frame_texture: texture_2d<f32>;
@group(0) @binding(1) var frame_sampler: sampler;
struct VertexInput {
@location(0) v_pos: vec2<f32>,
@location(1) texel_coord: vec2<f32>,
}
struct VertexOutput {
@builtin(position) clip_pos: vec4<f32>,
@location(0) uv: vec2<f32>,
}
@vertex
fn vs_main(input: VertexInput) -> VertexOutput {
var output: VertexOutput;
output.clip_pos = vec4<f32>(input.v_pos, 0.0, 1.0);
output.uv = input.texel_coord;
return output;
}
@fragment
fn fs_main(input: VertexOutput) -> @location(0) vec4<f32> {
return textureSample(frame_texture, frame_sampler, input.uv);
}