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        instance_flags: wgt::InstanceFlags,
37        backend: wgt::Backend,
38    ) -> Result<Self, DeviceError> {
39        let dispatch = match Dispatch::new(device, instance_flags, required_limits) {
40            Ok(dispatch) => dispatch,
41            Err(e) => {
42                log::error!("indirect-validation error: {e:?}");
43                return Err(DeviceError::Lost);
44            }
45        };
46        let draw = match Draw::new(device, required_features, instance_flags, backend) {
47            Ok(draw) => draw,
48            Err(e) => {
49                log::error!("indirect-draw-validation error: {e:?}");
50                return Err(DeviceError::Lost);
51            }
52        };
53        Ok(Self { dispatch, draw })
54    }
55
56    pub(crate) fn dispose(self, device: &dyn hal::DynDevice) {
57        let Self { dispatch, draw } = self;
58
59        dispatch.dispose(device);
60        draw.dispose(device);
61    }
62}
63
64#[derive(Debug)]
65pub(crate) struct BindGroups {
66    pub(crate) dispatch: Box<dyn hal::DynBindGroup>,
67    draw: Box<dyn hal::DynBindGroup>,
68}
69
70impl BindGroups {
71    /// `Ok(None)` will only be returned if `buffer_size` is `0`.
72    pub(crate) fn new(
73        indirect_validation: &IndirectValidation,
74        device: &crate::device::Device,
75        buffer_size: u64,
76        buffer: &dyn hal::DynBuffer,
77    ) -> Result<Option<Self>, DeviceError> {
78        let dispatch = indirect_validation.dispatch.create_src_bind_group(
79            device.raw(),
80            &device.limits,
81            buffer_size,
82            buffer,
83            device.instance_flags,
84        )?;
85        let draw = indirect_validation.draw.create_src_bind_group(
86            device.raw(),
87            &device.adapter.limits(),
88            buffer_size,
89            buffer,
90            device.instance_flags,
91        )?;
92
93        match (dispatch, draw) {
94            (None, None) => Ok(None),
95            (None, Some(_)) => unreachable!(),
96            (Some(_), None) => unreachable!(),
97            (Some(dispatch), Some(draw)) => Ok(Some(Self { dispatch, draw })),
98        }
99    }
100
101    pub(crate) fn dispose(self, device: &dyn hal::DynDevice) {
102        let Self { dispatch, draw } = self;
103
104        unsafe {
105            device.destroy_bind_group(dispatch);
106            device.destroy_bind_group(draw);
107        }
108    }
109}
wgpu_core/indirect_validation/mod.rs

wgpu_core/indirect_validation/
mod.rs
