use crate::deferred::DeferredState;
use crate::pipeline::{load_shader, SceneState};
pub struct SsrState {
pub ssr_texture: wgpu::Texture,
pub ssr_view: wgpu::TextureView,
pub ssr_pipeline: wgpu::RenderPipeline,
ssr_bgl: wgpu::BindGroupLayout,
pub ssr_bind_group: wgpu::BindGroup,
pub apply_pipeline: wgpu::RenderPipeline,
apply_bgl: wgpu::BindGroupLayout,
pub apply_bind_group: wgpu::BindGroup,
nearest_sampler: wgpu::Sampler,
pub width: u32,
pub height: u32,
}
impl SsrState {
pub fn new(
device: &wgpu::Device,
scene: &SceneState,
deferred: &DeferredState,
hdr_view: &wgpu::TextureView,
width: u32,
height: u32,
) -> Self {
let linear_sampler = device.create_sampler(&wgpu::SamplerDescriptor {
address_mode_u: wgpu::AddressMode::ClampToEdge,
address_mode_v: wgpu::AddressMode::ClampToEdge,
mag_filter: wgpu::FilterMode::Linear,
min_filter: wgpu::FilterMode::Linear,
..Default::default()
});
let half_w = (width / 2).max(1);
let half_h = (height / 2).max(1);
let (ssr_texture, ssr_view) = Self::mk_ssr_tex(device, half_w, half_h);
let ssr_bgl = Self::mk_ssr_bgl(device);
let apply_bgl = Self::mk_apply_bgl(device);
let ssr_bind_group = Self::mk_ssr_bg(
device,
&ssr_bgl,
hdr_view,
&deferred.normal_roughness_view,
&deferred.world_position_view,
&linear_sampler,
);
let apply_bind_group = Self::mk_apply_bg(device, &apply_bgl, &ssr_view, &linear_sampler);
let ssr_pipeline = Self::mk_ssr_pipeline(device, scene, &ssr_bgl);
let apply_pipeline = Self::mk_apply_pipeline(device, &apply_bgl);
Self {
ssr_texture,
ssr_view,
ssr_pipeline,
ssr_bgl,
ssr_bind_group,
apply_pipeline,
apply_bgl,
apply_bind_group,
nearest_sampler: linear_sampler,
width,
height,
}
}
pub fn resize(
&mut self,
device: &wgpu::Device,
deferred: &DeferredState,
hdr_view: &wgpu::TextureView,
width: u32,
height: u32,
) {
let half_w = (width / 2).max(1);
let half_h = (height / 2).max(1);
let (ssr_texture, ssr_view) = Self::mk_ssr_tex(device, half_w, half_h);
self.ssr_bind_group = Self::mk_ssr_bg(
device,
&self.ssr_bgl,
hdr_view,
&deferred.normal_roughness_view,
&deferred.world_position_view,
&self.nearest_sampler,
);
self.apply_bind_group =
Self::mk_apply_bg(device, &self.apply_bgl, &ssr_view, &self.nearest_sampler);
self.ssr_texture = ssr_texture;
self.ssr_view = ssr_view;
self.width = width;
self.height = height;
}
fn mk_ssr_tex(device: &wgpu::Device, w: u32, h: u32) -> (wgpu::Texture, wgpu::TextureView) {
let t = device.create_texture(&wgpu::TextureDescriptor {
label: Some("ssr_texture"),
size: wgpu::Extent3d {
width: w,
height: h,
depth_or_array_layers: 1,
},
mip_level_count: 1,
sample_count: 1,
dimension: wgpu::TextureDimension::D2,
format: wgpu::TextureFormat::Rgba16Float,
usage: wgpu::TextureUsages::RENDER_ATTACHMENT | wgpu::TextureUsages::TEXTURE_BINDING,
view_formats: &[],
});
let v = t.create_view(&wgpu::TextureViewDescriptor::default());
(t, v)
}
fn mk_ssr_bgl(device: &wgpu::Device) -> wgpu::BindGroupLayout {
device.create_bind_group_layout(&wgpu::BindGroupLayoutDescriptor {
label: Some("ssr_bgl"),
entries: &[
wgpu::BindGroupLayoutEntry {
binding: 0,
visibility: wgpu::ShaderStages::FRAGMENT,
ty: wgpu::BindingType::Texture {
multisampled: false,
view_dimension: wgpu::TextureViewDimension::D2,
sample_type: wgpu::TextureSampleType::Float { filterable: false },
},
count: None,
},
wgpu::BindGroupLayoutEntry {
binding: 1,
visibility: wgpu::ShaderStages::FRAGMENT,
ty: wgpu::BindingType::Texture {
multisampled: false,
view_dimension: wgpu::TextureViewDimension::D2,
sample_type: wgpu::TextureSampleType::Float { filterable: false },
},
count: None,
},
wgpu::BindGroupLayoutEntry {
binding: 2,
visibility: wgpu::ShaderStages::FRAGMENT,
ty: wgpu::BindingType::Texture {
multisampled: false,
view_dimension: wgpu::TextureViewDimension::D2,
sample_type: wgpu::TextureSampleType::Float { filterable: false },
},
count: None,
},
wgpu::BindGroupLayoutEntry {
binding: 3,
visibility: wgpu::ShaderStages::FRAGMENT,
ty: wgpu::BindingType::Sampler(wgpu::SamplerBindingType::Filtering),
count: None,
},
],
})
}
fn mk_apply_bgl(device: &wgpu::Device) -> wgpu::BindGroupLayout {
device.create_bind_group_layout(&wgpu::BindGroupLayoutDescriptor {
label: Some("ssr_apply_bgl"),
entries: &[
wgpu::BindGroupLayoutEntry {
binding: 0,
visibility: wgpu::ShaderStages::FRAGMENT,
ty: wgpu::BindingType::Texture {
multisampled: false,
view_dimension: wgpu::TextureViewDimension::D2,
sample_type: wgpu::TextureSampleType::Float { filterable: true },
},
count: None,
},
wgpu::BindGroupLayoutEntry {
binding: 1,
visibility: wgpu::ShaderStages::FRAGMENT,
ty: wgpu::BindingType::Sampler(wgpu::SamplerBindingType::Filtering),
count: None,
},
],
})
}
fn mk_ssr_bg(
device: &wgpu::Device,
layout: &wgpu::BindGroupLayout,
hdr_view: &wgpu::TextureView,
normal_view: &wgpu::TextureView,
pos_view: &wgpu::TextureView,
sampler: &wgpu::Sampler,
) -> wgpu::BindGroup {
device.create_bind_group(&wgpu::BindGroupDescriptor {
label: Some("ssr_bg"),
layout,
entries: &[
wgpu::BindGroupEntry {
binding: 0,
resource: wgpu::BindingResource::TextureView(hdr_view),
},
wgpu::BindGroupEntry {
binding: 1,
resource: wgpu::BindingResource::TextureView(normal_view),
},
wgpu::BindGroupEntry {
binding: 2,
resource: wgpu::BindingResource::TextureView(pos_view),
},
wgpu::BindGroupEntry {
binding: 3,
resource: wgpu::BindingResource::Sampler(sampler),
},
],
})
}
fn mk_apply_bg(
device: &wgpu::Device,
layout: &wgpu::BindGroupLayout,
ssr_view: &wgpu::TextureView,
sampler: &wgpu::Sampler,
) -> wgpu::BindGroup {
device.create_bind_group(&wgpu::BindGroupDescriptor {
label: Some("ssr_apply_bg"),
layout,
entries: &[
wgpu::BindGroupEntry {
binding: 0,
resource: wgpu::BindingResource::TextureView(ssr_view),
},
wgpu::BindGroupEntry {
binding: 1,
resource: wgpu::BindingResource::Sampler(sampler),
},
],
})
}
fn mk_ssr_pipeline(
device: &wgpu::Device,
scene: &SceneState,
bgl: &wgpu::BindGroupLayout,
) -> wgpu::RenderPipeline {
let shader = load_shader(
device,
"demo/assets/shaders/ssr.wgsl",
include_str!("shaders/ssr.wgsl"),
"SSR Shader",
);
let layout = device.create_pipeline_layout(&wgpu::PipelineLayoutDescriptor {
label: Some("ssr_layout"),
bind_group_layouts: &[&scene.global_bind_group_layout, bgl],
push_constant_ranges: &[],
});
device.create_render_pipeline(&wgpu::RenderPipelineDescriptor {
label: Some("ssr_pipeline"),
layout: Some(&layout),
vertex: wgpu::VertexState {
module: &shader,
entry_point: "vs_main",
compilation_options: Default::default(),
buffers: &[],
},
fragment: Some(wgpu::FragmentState {
module: &shader,
entry_point: "fs_main",
compilation_options: Default::default(),
targets: &[Some(wgpu::ColorTargetState {
format: wgpu::TextureFormat::Rgba16Float,
blend: None,
write_mask: wgpu::ColorWrites::ALL,
})],
}),
primitive: wgpu::PrimitiveState {
topology: wgpu::PrimitiveTopology::TriangleList,
cull_mode: None,
..Default::default()
},
depth_stencil: None,
multisample: wgpu::MultisampleState::default(),
multiview: None,
})
}
fn mk_apply_pipeline(
device: &wgpu::Device,
bgl: &wgpu::BindGroupLayout,
) -> wgpu::RenderPipeline {
let shader = load_shader(
device,
"demo/assets/shaders/ssr_apply.wgsl",
include_str!("shaders/ssr_apply.wgsl"),
"SSR Apply Shader",
);
let layout = device.create_pipeline_layout(&wgpu::PipelineLayoutDescriptor {
label: Some("ssr_apply_layout"),
bind_group_layouts: &[bgl],
push_constant_ranges: &[],
});
device.create_render_pipeline(&wgpu::RenderPipelineDescriptor {
label: Some("ssr_apply_pipeline"),
layout: Some(&layout),
vertex: wgpu::VertexState {
module: &shader,
entry_point: "vs_main",
compilation_options: Default::default(),
buffers: &[],
},
fragment: Some(wgpu::FragmentState {
module: &shader,
entry_point: "fs_main",
compilation_options: Default::default(),
targets: &[Some(wgpu::ColorTargetState {
format: wgpu::TextureFormat::Rgba16Float,
blend: Some(wgpu::BlendState {
color: wgpu::BlendComponent {
src_factor: wgpu::BlendFactor::One,
dst_factor: wgpu::BlendFactor::One,
operation: wgpu::BlendOperation::Add,
},
alpha: wgpu::BlendComponent::REPLACE,
}),
write_mask: wgpu::ColorWrites::ALL,
})],
}),
primitive: wgpu::PrimitiveState {
topology: wgpu::PrimitiveTopology::TriangleList,
cull_mode: None,
..Default::default()
},
depth_stencil: None,
multisample: wgpu::MultisampleState::default(),
multiview: None,
})
}
}