2022-09-07 02:45:33 +00:00
|
|
|
use crate::layouts::BindLayouts;
|
2022-12-23 22:06:19 +00:00
|
|
|
use crate::pipelines::VERTEX_BUFFERS_DESCRIPTION;
|
2022-09-07 01:41:23 +00:00
|
|
|
use crate::shaders::Shaders;
|
2022-12-27 22:18:03 +00:00
|
|
|
use crate::{
|
2023-01-06 17:41:24 +00:00
|
|
|
create_buffer_with_data, BitmapSamplers, Pipelines, TextureTransforms, Transforms, Vertex,
|
2022-12-27 22:18:03 +00:00
|
|
|
DEFAULT_COLOR_ADJUSTMENTS,
|
|
|
|
};
|
2022-09-08 08:05:34 +00:00
|
|
|
use fnv::FnvHashMap;
|
2023-01-06 17:41:24 +00:00
|
|
|
use std::mem;
|
2022-09-08 08:05:34 +00:00
|
|
|
use std::sync::{Arc, Mutex};
|
2022-09-03 00:14:24 +00:00
|
|
|
|
|
|
|
pub struct Descriptors {
|
2022-10-07 10:17:41 +00:00
|
|
|
pub adapter: wgpu::Adapter,
|
2022-09-03 00:14:24 +00:00
|
|
|
pub device: wgpu::Device,
|
|
|
|
pub limits: wgpu::Limits,
|
|
|
|
pub queue: wgpu::Queue,
|
|
|
|
pub bitmap_samplers: BitmapSamplers,
|
2022-09-07 02:45:33 +00:00
|
|
|
pub bind_layouts: BindLayouts,
|
2022-09-07 11:41:56 +00:00
|
|
|
pub quad: Quad,
|
2022-09-21 00:44:56 +00:00
|
|
|
copy_pipeline: Mutex<FnvHashMap<wgpu::TextureFormat, Arc<wgpu::RenderPipeline>>>,
|
2022-09-08 08:05:34 +00:00
|
|
|
copy_srgb_pipeline: Mutex<FnvHashMap<wgpu::TextureFormat, Arc<wgpu::RenderPipeline>>>,
|
|
|
|
shaders: Shaders,
|
|
|
|
pipelines: Mutex<FnvHashMap<(u32, wgpu::TextureFormat), Arc<Pipelines>>>,
|
2022-12-27 22:18:03 +00:00
|
|
|
pub default_color_bind_group: wgpu::BindGroup,
|
2023-01-06 17:41:24 +00:00
|
|
|
pub empty_bind_group: wgpu::BindGroup,
|
2022-09-03 00:14:24 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
impl Descriptors {
|
2022-10-07 10:17:41 +00:00
|
|
|
pub fn new(adapter: wgpu::Adapter, device: wgpu::Device, queue: wgpu::Queue) -> Self {
|
2022-09-03 00:14:24 +00:00
|
|
|
let limits = device.limits();
|
2022-09-07 02:45:33 +00:00
|
|
|
let bind_layouts = BindLayouts::new(&device);
|
2022-09-08 20:13:16 +00:00
|
|
|
let bitmap_samplers = BitmapSamplers::new(&device);
|
2023-01-06 17:41:24 +00:00
|
|
|
let shaders = Shaders::new(&device, limits.max_push_constant_size > 0);
|
2022-09-07 11:41:56 +00:00
|
|
|
let quad = Quad::new(&device);
|
2022-12-27 22:18:03 +00:00
|
|
|
let default_color_transform = create_buffer_with_data(
|
|
|
|
&device,
|
|
|
|
bytemuck::cast_slice(&[DEFAULT_COLOR_ADJUSTMENTS]),
|
|
|
|
wgpu::BufferUsages::UNIFORM,
|
|
|
|
create_debug_label!("Default colors"),
|
|
|
|
);
|
|
|
|
let default_color_bind_group = device.create_bind_group(&wgpu::BindGroupDescriptor {
|
|
|
|
label: create_debug_label!("Default colors").as_deref(),
|
|
|
|
layout: &bind_layouts.color_transforms,
|
|
|
|
entries: &[wgpu::BindGroupEntry {
|
|
|
|
binding: 0,
|
|
|
|
resource: default_color_transform.as_entire_binding(),
|
|
|
|
}],
|
|
|
|
});
|
2022-09-03 00:14:24 +00:00
|
|
|
|
2023-01-06 17:41:24 +00:00
|
|
|
let empty_bind_group = device.create_bind_group(&wgpu::BindGroupDescriptor {
|
|
|
|
label: None,
|
|
|
|
layout: &bind_layouts.transforms,
|
|
|
|
entries: &[],
|
|
|
|
});
|
|
|
|
|
2022-09-03 00:14:24 +00:00
|
|
|
Self {
|
2022-10-07 10:17:41 +00:00
|
|
|
adapter,
|
2022-09-03 00:14:24 +00:00
|
|
|
device,
|
|
|
|
limits,
|
|
|
|
queue,
|
|
|
|
bitmap_samplers,
|
2022-09-07 02:45:33 +00:00
|
|
|
bind_layouts,
|
2022-09-07 11:41:56 +00:00
|
|
|
quad,
|
2022-09-21 00:44:56 +00:00
|
|
|
copy_pipeline: Default::default(),
|
2022-09-08 08:05:34 +00:00
|
|
|
copy_srgb_pipeline: Default::default(),
|
|
|
|
shaders,
|
|
|
|
pipelines: Default::default(),
|
2022-12-27 22:18:03 +00:00
|
|
|
default_color_bind_group,
|
2023-01-06 17:41:24 +00:00
|
|
|
empty_bind_group,
|
2022-09-07 11:41:56 +00:00
|
|
|
}
|
|
|
|
}
|
2022-09-08 08:05:34 +00:00
|
|
|
|
|
|
|
pub fn copy_srgb_pipeline(&self, format: wgpu::TextureFormat) -> Arc<wgpu::RenderPipeline> {
|
|
|
|
let mut pipelines = self
|
|
|
|
.copy_srgb_pipeline
|
|
|
|
.lock()
|
|
|
|
.expect("Pipelines should not be already locked");
|
|
|
|
pipelines
|
|
|
|
.entry(format)
|
|
|
|
.or_insert_with(|| {
|
|
|
|
let copy_texture_pipeline_layout =
|
|
|
|
&self
|
|
|
|
.device
|
|
|
|
.create_pipeline_layout(&wgpu::PipelineLayoutDescriptor {
|
|
|
|
label: create_debug_label!("Copy sRGB pipeline layout").as_deref(),
|
|
|
|
bind_group_layouts: &[
|
|
|
|
&self.bind_layouts.globals,
|
|
|
|
&self.bind_layouts.transforms,
|
|
|
|
&self.bind_layouts.bitmap,
|
|
|
|
],
|
2023-01-06 17:41:24 +00:00
|
|
|
push_constant_ranges: if self.device.limits().max_push_constant_size > 0
|
|
|
|
{
|
|
|
|
&[wgpu::PushConstantRange {
|
|
|
|
stages: wgpu::ShaderStages::VERTEX,
|
|
|
|
range: 0..(mem::size_of::<Transforms>() as u32),
|
|
|
|
}]
|
|
|
|
} else {
|
|
|
|
&[]
|
|
|
|
},
|
2022-09-08 08:05:34 +00:00
|
|
|
});
|
|
|
|
Arc::new(
|
|
|
|
self.device
|
|
|
|
.create_render_pipeline(&wgpu::RenderPipelineDescriptor {
|
|
|
|
label: create_debug_label!("Copy sRGB pipeline").as_deref(),
|
|
|
|
layout: Some(©_texture_pipeline_layout),
|
|
|
|
vertex: wgpu::VertexState {
|
|
|
|
module: &self.shaders.copy_srgb_shader,
|
|
|
|
entry_point: "main_vertex",
|
|
|
|
buffers: &VERTEX_BUFFERS_DESCRIPTION,
|
|
|
|
},
|
|
|
|
fragment: Some(wgpu::FragmentState {
|
|
|
|
module: &self.shaders.copy_srgb_shader,
|
|
|
|
entry_point: "main_fragment",
|
|
|
|
targets: &[Some(wgpu::ColorTargetState {
|
|
|
|
format,
|
2022-09-21 00:44:56 +00:00
|
|
|
blend: Some(wgpu::BlendState::PREMULTIPLIED_ALPHA_BLENDING),
|
|
|
|
write_mask: Default::default(),
|
|
|
|
})],
|
|
|
|
}),
|
|
|
|
primitive: wgpu::PrimitiveState {
|
|
|
|
topology: wgpu::PrimitiveTopology::TriangleList,
|
|
|
|
strip_index_format: None,
|
|
|
|
front_face: wgpu::FrontFace::Ccw,
|
|
|
|
cull_mode: None,
|
|
|
|
polygon_mode: wgpu::PolygonMode::default(),
|
|
|
|
unclipped_depth: false,
|
|
|
|
conservative: false,
|
|
|
|
},
|
|
|
|
depth_stencil: None,
|
|
|
|
multisample: wgpu::MultisampleState {
|
|
|
|
count: 1,
|
|
|
|
mask: !0,
|
|
|
|
alpha_to_coverage_enabled: false,
|
|
|
|
},
|
|
|
|
multiview: None,
|
|
|
|
}),
|
|
|
|
)
|
|
|
|
})
|
|
|
|
.clone()
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn copy_pipeline(&self, format: wgpu::TextureFormat) -> Arc<wgpu::RenderPipeline> {
|
|
|
|
let mut pipelines = self
|
|
|
|
.copy_pipeline
|
|
|
|
.lock()
|
|
|
|
.expect("Pipelines should not be already locked");
|
|
|
|
pipelines
|
|
|
|
.entry(format)
|
|
|
|
.or_insert_with(|| {
|
|
|
|
let copy_texture_pipeline_layout =
|
|
|
|
&self
|
|
|
|
.device
|
|
|
|
.create_pipeline_layout(&wgpu::PipelineLayoutDescriptor {
|
|
|
|
label: create_debug_label!("Copy pipeline layout").as_deref(),
|
|
|
|
bind_group_layouts: &[
|
|
|
|
&self.bind_layouts.globals,
|
|
|
|
&self.bind_layouts.transforms,
|
|
|
|
&self.bind_layouts.bitmap,
|
|
|
|
],
|
2023-01-06 17:41:24 +00:00
|
|
|
push_constant_ranges: if self.device.limits().max_push_constant_size > 0
|
|
|
|
{
|
|
|
|
&[wgpu::PushConstantRange {
|
|
|
|
stages: wgpu::ShaderStages::VERTEX,
|
|
|
|
range: 0..(mem::size_of::<Transforms>() as u32),
|
|
|
|
}]
|
|
|
|
} else {
|
|
|
|
&[]
|
|
|
|
},
|
2022-09-21 00:44:56 +00:00
|
|
|
});
|
|
|
|
Arc::new(
|
|
|
|
self.device
|
|
|
|
.create_render_pipeline(&wgpu::RenderPipelineDescriptor {
|
|
|
|
label: create_debug_label!("Copy pipeline").as_deref(),
|
|
|
|
layout: Some(©_texture_pipeline_layout),
|
|
|
|
vertex: wgpu::VertexState {
|
|
|
|
module: &self.shaders.copy_shader,
|
|
|
|
entry_point: "main_vertex",
|
|
|
|
buffers: &VERTEX_BUFFERS_DESCRIPTION,
|
|
|
|
},
|
|
|
|
fragment: Some(wgpu::FragmentState {
|
|
|
|
module: &self.shaders.copy_shader,
|
|
|
|
entry_point: "main_fragment",
|
|
|
|
targets: &[Some(wgpu::ColorTargetState {
|
|
|
|
format,
|
|
|
|
blend: Some(wgpu::BlendState::PREMULTIPLIED_ALPHA_BLENDING),
|
2022-09-08 08:05:34 +00:00
|
|
|
write_mask: Default::default(),
|
|
|
|
})],
|
|
|
|
}),
|
|
|
|
primitive: wgpu::PrimitiveState {
|
|
|
|
topology: wgpu::PrimitiveTopology::TriangleList,
|
|
|
|
strip_index_format: None,
|
|
|
|
front_face: wgpu::FrontFace::Ccw,
|
|
|
|
cull_mode: None,
|
|
|
|
polygon_mode: wgpu::PolygonMode::default(),
|
|
|
|
unclipped_depth: false,
|
|
|
|
conservative: false,
|
|
|
|
},
|
|
|
|
depth_stencil: None,
|
|
|
|
multisample: wgpu::MultisampleState {
|
|
|
|
count: 1,
|
|
|
|
mask: !0,
|
|
|
|
alpha_to_coverage_enabled: false,
|
|
|
|
},
|
|
|
|
multiview: None,
|
|
|
|
}),
|
|
|
|
)
|
|
|
|
})
|
|
|
|
.clone()
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn pipelines(&self, msaa_sample_count: u32, format: wgpu::TextureFormat) -> Arc<Pipelines> {
|
|
|
|
let mut pipelines = self
|
|
|
|
.pipelines
|
|
|
|
.lock()
|
|
|
|
.expect("Pipelines should not be already locked");
|
|
|
|
pipelines
|
|
|
|
.entry((msaa_sample_count, format))
|
|
|
|
.or_insert_with(|| {
|
|
|
|
Arc::new(Pipelines::new(
|
|
|
|
&self.device,
|
|
|
|
&self.shaders,
|
|
|
|
format,
|
|
|
|
msaa_sample_count,
|
|
|
|
&self.bind_layouts,
|
|
|
|
))
|
|
|
|
})
|
|
|
|
.clone()
|
|
|
|
}
|
2022-09-07 11:41:56 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
pub struct Quad {
|
|
|
|
pub vertices: wgpu::Buffer,
|
|
|
|
pub indices: wgpu::Buffer,
|
|
|
|
pub texture_transforms: wgpu::Buffer,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Quad {
|
|
|
|
pub fn new(device: &wgpu::Device) -> Self {
|
|
|
|
let vertices = [
|
|
|
|
Vertex {
|
|
|
|
position: [0.0, 0.0],
|
|
|
|
color: [1.0, 1.0, 1.0, 1.0],
|
|
|
|
},
|
|
|
|
Vertex {
|
|
|
|
position: [1.0, 0.0],
|
|
|
|
color: [1.0, 1.0, 1.0, 1.0],
|
|
|
|
},
|
|
|
|
Vertex {
|
|
|
|
position: [1.0, 1.0],
|
|
|
|
color: [1.0, 1.0, 1.0, 1.0],
|
|
|
|
},
|
|
|
|
Vertex {
|
|
|
|
position: [0.0, 1.0],
|
|
|
|
color: [1.0, 1.0, 1.0, 1.0],
|
|
|
|
},
|
|
|
|
];
|
|
|
|
let indices: [u32; 6] = [0, 1, 2, 0, 2, 3];
|
|
|
|
|
|
|
|
let vbo = create_buffer_with_data(
|
|
|
|
device,
|
|
|
|
bytemuck::cast_slice(&vertices),
|
|
|
|
wgpu::BufferUsages::VERTEX,
|
|
|
|
create_debug_label!("Quad vbo"),
|
|
|
|
);
|
|
|
|
|
|
|
|
let ibo = create_buffer_with_data(
|
|
|
|
device,
|
|
|
|
bytemuck::cast_slice(&indices),
|
|
|
|
wgpu::BufferUsages::INDEX,
|
|
|
|
create_debug_label!("Quad ibo"),
|
|
|
|
);
|
|
|
|
|
|
|
|
let tex_transforms = create_buffer_with_data(
|
|
|
|
device,
|
|
|
|
bytemuck::cast_slice(&[TextureTransforms {
|
|
|
|
u_matrix: [
|
|
|
|
[1.0, 0.0, 0.0, 0.0],
|
|
|
|
[0.0, 1.0, 0.0, 0.0],
|
|
|
|
[0.0, 0.0, 1.0, 0.0],
|
|
|
|
[0.0, 0.0, 0.0, 1.0],
|
|
|
|
],
|
|
|
|
}]),
|
|
|
|
wgpu::BufferUsages::UNIFORM,
|
|
|
|
create_debug_label!("Quad tex transforms"),
|
|
|
|
);
|
|
|
|
|
|
|
|
Self {
|
|
|
|
vertices: vbo,
|
|
|
|
indices: ibo,
|
|
|
|
texture_transforms: tex_transforms,
|
2022-09-03 00:14:24 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|