Move script gpu files into webgpu folder (#34415)

Signed-off-by: atbrakhi <atbrakhi@igalia.com>
2025-08-06 06:00:15 +01:00 · 2024-11-28 15:24:15 +01:00 · 2024-11-28 15:24:15 +01:00 · d2d3407501
commit d2d3407501
parent a37ccc3e64
44 changed files with 108 additions and 118 deletions
--- a/components/script/dom/webgpu/gputexture.rs
+++ b/components/script/dom/webgpu/gputexture.rs
@ -0,0 +1,285 @@
+/* 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::string::String;
+
+use dom_struct::dom_struct;
+use webgpu::wgc::resource;
+use webgpu::{wgt, WebGPU, WebGPURequest, WebGPUTexture, WebGPUTextureView};
+
+use super::gpuconvert::convert_texture_descriptor;
+use crate::dom::bindings::cell::DomRefCell;
+use crate::dom::bindings::codegen::Bindings::WebGPUBinding::{
+    GPUTextureAspect, GPUTextureDescriptor, GPUTextureDimension, GPUTextureFormat,
+    GPUTextureMethods, GPUTextureViewDescriptor,
+};
+use crate::dom::bindings::error::Fallible;
+use crate::dom::bindings::reflector::{reflect_dom_object, DomObject, Reflector};
+use crate::dom::bindings::root::{Dom, DomRoot};
+use crate::dom::bindings::str::USVString;
+use crate::dom::globalscope::GlobalScope;
+use crate::dom::gpudevice::GPUDevice;
+use crate::dom::gputextureview::GPUTextureView;
+
+#[dom_struct]
+pub struct GPUTexture {
+    reflector_: Reflector,
+    #[no_trace]
+    texture: WebGPUTexture,
+    label: DomRefCell<USVString>,
+    device: Dom<GPUDevice>,
+    #[ignore_malloc_size_of = "channels are hard"]
+    #[no_trace]
+    channel: WebGPU,
+    #[ignore_malloc_size_of = "defined in wgpu"]
+    #[no_trace]
+    texture_size: wgt::Extent3d,
+    mip_level_count: u32,
+    sample_count: u32,
+    dimension: GPUTextureDimension,
+    format: GPUTextureFormat,
+    texture_usage: u32,
+}
+
+impl GPUTexture {
+    #[allow(clippy::too_many_arguments)]
+    fn new_inherited(
+        texture: WebGPUTexture,
+        device: &GPUDevice,
+        channel: WebGPU,
+        texture_size: wgt::Extent3d,
+        mip_level_count: u32,
+        sample_count: u32,
+        dimension: GPUTextureDimension,
+        format: GPUTextureFormat,
+        texture_usage: u32,
+        label: USVString,
+    ) -> Self {
+        Self {
+            reflector_: Reflector::new(),
+            texture,
+            label: DomRefCell::new(label),
+            device: Dom::from_ref(device),
+            channel,
+            texture_size,
+            mip_level_count,
+            sample_count,
+            dimension,
+            format,
+            texture_usage,
+        }
+    }
+
+    #[allow(clippy::too_many_arguments)]
+    pub fn new(
+        global: &GlobalScope,
+        texture: WebGPUTexture,
+        device: &GPUDevice,
+        channel: WebGPU,
+        texture_size: wgt::Extent3d,
+        mip_level_count: u32,
+        sample_count: u32,
+        dimension: GPUTextureDimension,
+        format: GPUTextureFormat,
+        texture_usage: u32,
+        label: USVString,
+    ) -> DomRoot<Self> {
+        reflect_dom_object(
+            Box::new(GPUTexture::new_inherited(
+                texture,
+                device,
+                channel,
+                texture_size,
+                mip_level_count,
+                sample_count,
+                dimension,
+                format,
+                texture_usage,
+                label,
+            )),
+            global,
+        )
+    }
+}
+
+impl Drop for GPUTexture {
+    fn drop(&mut self) {
+        if let Err(e) = self
+            .channel
+            .0
+            .send(WebGPURequest::DropTexture(self.texture.0))
+        {
+            warn!(
+                "Failed to send WebGPURequest::DropTexture({:?}) ({})",
+                self.texture.0, e
+            );
+        };
+    }
+}
+
+impl GPUTexture {
+    pub fn id(&self) -> WebGPUTexture {
+        self.texture
+    }
+
+    /// <https://gpuweb.github.io/gpuweb/#dom-gpudevice-createtexture>
+    pub fn create(
+        device: &GPUDevice,
+        descriptor: &GPUTextureDescriptor,
+    ) -> Fallible<DomRoot<GPUTexture>> {
+        let (desc, size) = convert_texture_descriptor(descriptor, device)?;
+
+        let texture_id = device.global().wgpu_id_hub().create_texture_id();
+
+        device
+            .channel()
+            .0
+            .send(WebGPURequest::CreateTexture {
+                device_id: device.id().0,
+                texture_id,
+                descriptor: desc,
+            })
+            .expect("Failed to create WebGPU Texture");
+
+        let texture = WebGPUTexture(texture_id);
+
+        Ok(GPUTexture::new(
+            &device.global(),
+            texture,
+            device,
+            device.channel().clone(),
+            size,
+            descriptor.mipLevelCount,
+            descriptor.sampleCount,
+            descriptor.dimension,
+            descriptor.format,
+            descriptor.usage,
+            descriptor.parent.label.clone(),
+        ))
+    }
+}
+
+impl GPUTextureMethods<crate::DomTypeHolder> for GPUTexture {
+    /// <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;
+    }
+
+    /// <https://gpuweb.github.io/gpuweb/#dom-gputexture-createview>
+    fn CreateView(
+        &self,
+        descriptor: &GPUTextureViewDescriptor,
+    ) -> Fallible<DomRoot<GPUTextureView>> {
+        let desc = if !matches!(descriptor.mipLevelCount, Some(0)) &&
+            !matches!(descriptor.arrayLayerCount, Some(0))
+        {
+            Some(resource::TextureViewDescriptor {
+                label: (&descriptor.parent).into(),
+                format: descriptor
+                    .format
+                    .map(|f| self.device.validate_texture_format_required_features(&f))
+                    .transpose()?,
+                dimension: descriptor.dimension.map(|dimension| dimension.into()),
+                range: wgt::ImageSubresourceRange {
+                    aspect: match descriptor.aspect {
+                        GPUTextureAspect::All => wgt::TextureAspect::All,
+                        GPUTextureAspect::Stencil_only => wgt::TextureAspect::StencilOnly,
+                        GPUTextureAspect::Depth_only => wgt::TextureAspect::DepthOnly,
+                    },
+                    base_mip_level: descriptor.baseMipLevel,
+                    mip_level_count: descriptor.mipLevelCount,
+                    base_array_layer: descriptor.baseArrayLayer,
+                    array_layer_count: descriptor.arrayLayerCount,
+                },
+            })
+        } else {
+            self.device
+                .dispatch_error(webgpu::Error::Validation(String::from(
+                    "arrayLayerCount and mipLevelCount cannot be 0",
+                )));
+            None
+        };
+
+        let texture_view_id = self.global().wgpu_id_hub().create_texture_view_id();
+
+        self.channel
+            .0
+            .send(WebGPURequest::CreateTextureView {
+                texture_id: self.texture.0,
+                texture_view_id,
+                device_id: self.device.id().0,
+                descriptor: desc,
+            })
+            .expect("Failed to create WebGPU texture view");
+
+        let texture_view = WebGPUTextureView(texture_view_id);
+
+        Ok(GPUTextureView::new(
+            &self.global(),
+            self.channel.clone(),
+            texture_view,
+            self,
+            descriptor.parent.label.clone(),
+        ))
+    }
+
+    /// <https://gpuweb.github.io/gpuweb/#dom-gputexture-destroy>
+    fn Destroy(&self) {
+        if let Err(e) = self
+            .channel
+            .0
+            .send(WebGPURequest::DestroyTexture(self.texture.0))
+        {
+            warn!(
+                "Failed to send WebGPURequest::DestroyTexture({:?}) ({})",
+                self.texture.0, e
+            );
+        };
+    }
+
+    /// <https://gpuweb.github.io/gpuweb/#dom-gputexture-width>
+    fn Width(&self) -> u32 {
+        self.texture_size.width
+    }
+
+    /// <https://gpuweb.github.io/gpuweb/#dom-gputexture-height>
+    fn Height(&self) -> u32 {
+        self.texture_size.height
+    }
+
+    /// <https://gpuweb.github.io/gpuweb/#dom-gputexture-depthorarraylayers>
+    fn DepthOrArrayLayers(&self) -> u32 {
+        self.texture_size.depth_or_array_layers
+    }
+
+    /// <https://gpuweb.github.io/gpuweb/#dom-gputexture-miplevelcount>
+    fn MipLevelCount(&self) -> u32 {
+        self.mip_level_count
+    }
+
+    /// <https://gpuweb.github.io/gpuweb/#dom-gputexture-samplecount>
+    fn SampleCount(&self) -> u32 {
+        self.sample_count
+    }
+
+    /// <https://gpuweb.github.io/gpuweb/#dom-gputexture-dimension>
+    fn Dimension(&self) -> GPUTextureDimension {
+        self.dimension
+    }
+
+    /// <https://gpuweb.github.io/gpuweb/#dom-gputexture-format>
+    fn Format(&self) -> GPUTextureFormat {
+        self.format
+    }
+
+    /// <https://gpuweb.github.io/gpuweb/#dom-gputexture-usage>
+    fn Usage(&self) -> u32 {
+        self.texture_usage
+    }
+}