diff options
Diffstat (limited to 'components/script/dom/webgpu/gpupipelinelayout.rs')
-rw-r--r-- | components/script/dom/webgpu/gpupipelinelayout.rs | 142 |
1 files changed, 142 insertions, 0 deletions
diff --git a/components/script/dom/webgpu/gpupipelinelayout.rs b/components/script/dom/webgpu/gpupipelinelayout.rs new file mode 100644 index 00000000000..c1cbbdb72ab --- /dev/null +++ b/components/script/dom/webgpu/gpupipelinelayout.rs @@ -0,0 +1,142 @@ +/* This Source Code Form is subject to the terms of the Mozilla Public + * License, v. 2.0. If a copy of the MPL was not distributed with this + * file, You can obtain one at https://mozilla.org/MPL/2.0/. */ + +use std::borrow::Cow; + +use dom_struct::dom_struct; +use webgpu::wgc::binding_model::PipelineLayoutDescriptor; +use webgpu::{WebGPU, WebGPUBindGroupLayout, WebGPUPipelineLayout, WebGPURequest}; + +use crate::dom::bindings::cell::DomRefCell; +use crate::dom::bindings::codegen::Bindings::WebGPUBinding::{ + GPUPipelineLayoutDescriptor, GPUPipelineLayoutMethods, +}; +use crate::dom::bindings::reflector::{reflect_dom_object, DomObject, Reflector}; +use crate::dom::bindings::root::DomRoot; +use crate::dom::bindings::str::USVString; +use crate::dom::globalscope::GlobalScope; +use crate::dom::gpudevice::GPUDevice; + +#[dom_struct] +pub struct GPUPipelineLayout { + reflector_: Reflector, + #[ignore_malloc_size_of = "defined in webgpu"] + #[no_trace] + channel: WebGPU, + label: DomRefCell<USVString>, + #[no_trace] + pipeline_layout: WebGPUPipelineLayout, + #[no_trace] + bind_group_layouts: Vec<WebGPUBindGroupLayout>, +} + +impl GPUPipelineLayout { + fn new_inherited( + channel: WebGPU, + pipeline_layout: WebGPUPipelineLayout, + label: USVString, + bgls: Vec<WebGPUBindGroupLayout>, + ) -> Self { + Self { + reflector_: Reflector::new(), + channel, + label: DomRefCell::new(label), + pipeline_layout, + bind_group_layouts: bgls, + } + } + + pub fn new( + global: &GlobalScope, + channel: WebGPU, + pipeline_layout: WebGPUPipelineLayout, + label: USVString, + bgls: Vec<WebGPUBindGroupLayout>, + ) -> DomRoot<Self> { + reflect_dom_object( + Box::new(GPUPipelineLayout::new_inherited( + channel, + pipeline_layout, + label, + bgls, + )), + global, + ) + } +} + +impl GPUPipelineLayout { + pub fn id(&self) -> WebGPUPipelineLayout { + self.pipeline_layout + } + + pub fn bind_group_layouts(&self) -> Vec<WebGPUBindGroupLayout> { + self.bind_group_layouts.clone() + } + + /// <https://gpuweb.github.io/gpuweb/#dom-gpudevice-createpipelinelayout> + pub fn create( + device: &GPUDevice, + descriptor: &GPUPipelineLayoutDescriptor, + ) -> DomRoot<GPUPipelineLayout> { + let bgls = descriptor + .bindGroupLayouts + .iter() + .map(|each| each.id()) + .collect::<Vec<_>>(); + + let desc = PipelineLayoutDescriptor { + label: (&descriptor.parent).into(), + bind_group_layouts: Cow::Owned(bgls.iter().map(|l| l.0).collect::<Vec<_>>()), + push_constant_ranges: Cow::Owned(vec![]), + }; + + let pipeline_layout_id = device.global().wgpu_id_hub().create_pipeline_layout_id(); + device + .channel() + .0 + .send(WebGPURequest::CreatePipelineLayout { + device_id: device.id().0, + pipeline_layout_id, + descriptor: desc, + }) + .expect("Failed to create WebGPU PipelineLayout"); + + let pipeline_layout = WebGPUPipelineLayout(pipeline_layout_id); + GPUPipelineLayout::new( + &device.global(), + device.channel().clone(), + pipeline_layout, + descriptor.parent.label.clone(), + bgls, + ) + } +} + +impl GPUPipelineLayoutMethods<crate::DomTypeHolder> for GPUPipelineLayout { + /// <https://gpuweb.github.io/gpuweb/#dom-gpuobjectbase-label> + fn Label(&self) -> USVString { + self.label.borrow().clone() + } + + /// <https://gpuweb.github.io/gpuweb/#dom-gpuobjectbase-label> + fn SetLabel(&self, value: USVString) { + *self.label.borrow_mut() = value; + } +} + +impl Drop for GPUPipelineLayout { + fn drop(&mut self) { + if let Err(e) = self + .channel + .0 + .send(WebGPURequest::DropPipelineLayout(self.pipeline_layout.0)) + { + warn!( + "Failed to send DropPipelineLayout ({:?}) ({})", + self.pipeline_layout.0, e + ); + } + } +} |