pub const SOURCE: &str = include_str!("shader.wgsl");
pub fn create_shader_module(device: &wgpu::Device) -> wgpu::ShaderModule {
let source = std::borrow::Cow::Borrowed(SOURCE);
device.create_shader_module(wgpu::ShaderModuleDescriptor {
label: None,
source: wgpu::ShaderSource::Wgsl(source),
})
}
pub fn create_pipeline_layout(device: &wgpu::Device) -> wgpu::PipelineLayout {
device.create_pipeline_layout(&wgpu::PipelineLayoutDescriptor {
label: None,
bind_group_layouts: &[],
immediate_size: 0,
})
}
pub const ENTRY_VS_MAIN_MULTIPLE: &str = "vs_main_multiple";
pub const ENTRY_VS_MAIN_NONE: &str = "vs_main_none";
pub const ENTRY_VS_MAIN_SINGLE: &str = "vs_main_single";
#[repr(C)]
#[derive(Debug, Copy, Clone, PartialEq)]
pub struct Input0 {
pub in0: [f32; 4],
pub in1: [f32; 4],
pub in2: [f32; 4],
}
impl Input0 {
pub const VERTEX_ATTRIBUTES: [wgpu::VertexAttribute; 3] = [
wgpu::VertexAttribute {
format: wgpu::VertexFormat::Float32x4,
offset: std::mem::offset_of!(Input0, in0) as u64,
shader_location: 0,
},
wgpu::VertexAttribute {
format: wgpu::VertexFormat::Float32x4,
offset: std::mem::offset_of!(Input0, in1) as u64,
shader_location: 1,
},
wgpu::VertexAttribute {
format: wgpu::VertexFormat::Float32x4,
offset: std::mem::offset_of!(Input0, in2) as u64,
shader_location: 2,
},
];
pub const fn vertex_buffer_layout(
step_mode: wgpu::VertexStepMode,
) -> wgpu::VertexBufferLayout<'static> {
wgpu::VertexBufferLayout {
array_stride: std::mem::size_of::<Input0>() as u64,
step_mode,
attributes: &Input0::VERTEX_ATTRIBUTES,
}
}
}
#[repr(C)]
#[derive(Debug, Copy, Clone, PartialEq)]
pub struct Input1 {
pub in3: [f32; 4],
pub in4: [f32; 4],
pub in5: [f32; 4],
pub in6: [u32; 4],
}
impl Input1 {
pub const VERTEX_ATTRIBUTES: [wgpu::VertexAttribute; 4] = [
wgpu::VertexAttribute {
format: wgpu::VertexFormat::Float32x4,
offset: std::mem::offset_of!(Input1, in3) as u64,
shader_location: 3,
},
wgpu::VertexAttribute {
format: wgpu::VertexFormat::Float32x4,
offset: std::mem::offset_of!(Input1, in4) as u64,
shader_location: 4,
},
wgpu::VertexAttribute {
format: wgpu::VertexFormat::Float32x4,
offset: std::mem::offset_of!(Input1, in5) as u64,
shader_location: 5,
},
wgpu::VertexAttribute {
format: wgpu::VertexFormat::Uint32x4,
offset: std::mem::offset_of!(Input1, in6) as u64,
shader_location: 6,
},
];
pub const fn vertex_buffer_layout(
step_mode: wgpu::VertexStepMode,
) -> wgpu::VertexBufferLayout<'static> {
wgpu::VertexBufferLayout {
array_stride: std::mem::size_of::<Input1>() as u64,
step_mode,
attributes: &Input1::VERTEX_ATTRIBUTES,
}
}
}
#[derive(Debug)]
pub struct VertexEntry<const N: usize> {
pub entry_point: &'static str,
pub buffers: [wgpu::VertexBufferLayout<'static>; N],
pub constants: Vec<(&'static str, f64)>,
}
pub fn vertex_state<'a, const N: usize>(
module: &'a wgpu::ShaderModule,
entry: &'a VertexEntry<N>,
) -> wgpu::VertexState<'a> {
wgpu::VertexState {
module,
entry_point: Some(entry.entry_point),
buffers: &entry.buffers,
compilation_options: wgpu::PipelineCompilationOptions {
constants: &entry.constants,
..Default::default()
},
}
}
pub fn vs_main_multiple_entry(
in0_step_mode: wgpu::VertexStepMode,
in1_step_mode: wgpu::VertexStepMode,
in3_step_mode: wgpu::VertexStepMode,
) -> VertexEntry<3> {
VertexEntry {
entry_point: ENTRY_VS_MAIN_MULTIPLE,
buffers: [
Input0::vertex_buffer_layout(in0_step_mode),
Input1::vertex_buffer_layout(in1_step_mode),
wgpu::VertexBufferLayout {
array_stride: 16,
step_mode: in3_step_mode,
attributes: &[wgpu::VertexAttribute {
format: wgpu::VertexFormat::Float32x4,
offset: 0,
shader_location: 7u32,
}],
},
],
constants: Default::default(),
}
}
pub fn vs_main_none_entry() -> VertexEntry<0> {
VertexEntry {
entry_point: ENTRY_VS_MAIN_NONE,
buffers: [],
constants: Default::default(),
}
}
pub fn vs_main_single_entry(in0_step_mode: wgpu::VertexStepMode) -> VertexEntry<1> {
VertexEntry {
entry_point: ENTRY_VS_MAIN_SINGLE,
buffers: [Input0::vertex_buffer_layout(in0_step_mode)],
constants: Default::default(),
}
}