wgpu_core/indirect_validation/
mod.rs

1use 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    /// `Ok(None)` will only be returned if `buffer_size` is `0`.
70    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}