wgpu_core/indirect_validation/
mod.rs1use crate::{
2 device::DeviceError,
3 pipeline::{CreateComputePipelineError, CreateShaderModuleError},
4};
5use alloc::boxed::Box;
6use thiserror::Error;
7
8mod dispatch;
9mod draw;
10mod utils;
11
12pub(crate) use dispatch::Dispatch;
13pub(crate) use draw::{Draw, DrawBatcher, DrawResources};
14
15#[derive(Clone, Debug, Error)]
16#[non_exhaustive]
17enum CreateIndirectValidationPipelineError {
18 #[error(transparent)]
19 DeviceError(#[from] DeviceError),
20 #[error(transparent)]
21 ShaderModule(#[from] CreateShaderModuleError),
22 #[error(transparent)]
23 ComputePipeline(#[from] CreateComputePipelineError),
24}
25
26pub(crate) struct IndirectValidation {
27 pub(crate) dispatch: Dispatch,
28 pub(crate) draw: Draw,
29}
30
31impl IndirectValidation {
32 pub(crate) fn new(
33 device: &dyn hal::DynDevice,
34 required_limits: &wgt::Limits,
35 required_features: &wgt::Features,
36 ) -> Result<Self, DeviceError> {
37 let dispatch = match Dispatch::new(device, required_limits) {
38 Ok(dispatch) => dispatch,
39 Err(e) => {
40 log::error!("indirect-validation error: {e:?}");
41 return Err(DeviceError::Lost);
42 }
43 };
44 let draw = match Draw::new(device, required_features) {
45 Ok(draw) => draw,
46 Err(e) => {
47 log::error!("indirect-draw-validation error: {e:?}");
48 return Err(DeviceError::Lost);
49 }
50 };
51 Ok(Self { dispatch, draw })
52 }
53
54 pub(crate) fn dispose(self, device: &dyn hal::DynDevice) {
55 let Self { dispatch, draw } = self;
56
57 dispatch.dispose(device);
58 draw.dispose(device);
59 }
60}
61
62#[derive(Debug)]
63pub(crate) struct BindGroups {
64 pub(crate) dispatch: Box<dyn hal::DynBindGroup>,
65 draw: Box<dyn hal::DynBindGroup>,
66}
67
68impl BindGroups {
69 pub(crate) fn new(
71 indirect_validation: &IndirectValidation,
72 device: &crate::device::Device,
73 buffer_size: u64,
74 buffer: &dyn hal::DynBuffer,
75 ) -> Result<Option<Self>, DeviceError> {
76 let dispatch = indirect_validation.dispatch.create_src_bind_group(
77 device.raw(),
78 &device.limits,
79 buffer_size,
80 buffer,
81 )?;
82 let draw = indirect_validation.draw.create_src_bind_group(
83 device.raw(),
84 &device.adapter.limits(),
85 buffer_size,
86 buffer,
87 )?;
88
89 match (dispatch, draw) {
90 (None, None) => Ok(None),
91 (None, Some(_)) => unreachable!(),
92 (Some(_), None) => unreachable!(),
93 (Some(dispatch), Some(draw)) => Ok(Some(Self { dispatch, draw })),
94 }
95 }
96
97 pub(crate) fn dispose(self, device: &dyn hal::DynDevice) {
98 let Self { dispatch, draw } = self;
99
100 unsafe {
101 device.destroy_bind_group(dispatch);
102 device.destroy_bind_group(draw);
103 }
104 }
105}