Move script gpu files into webgpu folder (#34415)

Signed-off-by: atbrakhi <atbrakhi@igalia.com>

1 files changed, 154 insertions, 0 deletions

diff --git a/components/script/dom/webgpu/gpushadermodule.rs b/components/script/dom/webgpu/gpushadermodule.rs
new file mode 100644
index 00000000000..44cc8fa8115
--- /dev/null
+++ b/components/script/dom/webgpu/gpushadermodule.rs
@@ -0,0 +1,154 @@
+/* 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::rc::Rc;
+
+use dom_struct::dom_struct;
+use webgpu::{WebGPU, WebGPURequest, WebGPUResponse, WebGPUShaderModule};
+
+use super::gpu::AsyncWGPUListener;
+use super::gpucompilationinfo::GPUCompilationInfo;
+use crate::dom::bindings::cell::DomRefCell;
+use crate::dom::bindings::codegen::Bindings::WebGPUBinding::{
+    GPUShaderModuleDescriptor, GPUShaderModuleMethods,
+};
+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::bindings::trace::RootedTraceableBox;
+use crate::dom::globalscope::GlobalScope;
+use crate::dom::promise::Promise;
+use crate::dom::types::GPUDevice;
+use crate::dom::webgpu::gpu::response_async;
+use crate::realms::InRealm;
+use crate::script_runtime::CanGc;
+
+#[dom_struct]
+pub struct GPUShaderModule {
+    reflector_: Reflector,
+    #[ignore_malloc_size_of = "defined in webgpu"]
+    #[no_trace]
+    channel: WebGPU,
+    label: DomRefCell<USVString>,
+    #[no_trace]
+    shader_module: WebGPUShaderModule,
+    #[ignore_malloc_size_of = "promise"]
+    compilation_info_promise: Rc<Promise>,
+}
+
+impl GPUShaderModule {
+    fn new_inherited(
+        channel: WebGPU,
+        shader_module: WebGPUShaderModule,
+        label: USVString,
+        promise: Rc<Promise>,
+    ) -> Self {
+        Self {
+            reflector_: Reflector::new(),
+            channel,
+            label: DomRefCell::new(label),
+            shader_module,
+            compilation_info_promise: promise,
+        }
+    }
+
+    pub fn new(
+        global: &GlobalScope,
+        channel: WebGPU,
+        shader_module: WebGPUShaderModule,
+        label: USVString,
+        promise: Rc<Promise>,
+    ) -> DomRoot<Self> {
+        reflect_dom_object(
+            Box::new(GPUShaderModule::new_inherited(
+                channel,
+                shader_module,
+                label,
+                promise,
+            )),
+            global,
+        )
+    }
+}
+
+impl GPUShaderModule {
+    pub fn id(&self) -> WebGPUShaderModule {
+        self.shader_module
+    }
+
+    /// <https://gpuweb.github.io/gpuweb/#dom-gpudevice-createshadermodule>
+    pub fn create(
+        device: &GPUDevice,
+        descriptor: RootedTraceableBox<GPUShaderModuleDescriptor>,
+        comp: InRealm,
+        can_gc: CanGc,
+    ) -> DomRoot<GPUShaderModule> {
+        let program_id = device.global().wgpu_id_hub().create_shader_module_id();
+        let promise = Promise::new_in_current_realm(comp, can_gc);
+        let shader_module = GPUShaderModule::new(
+            &device.global(),
+            device.channel().clone(),
+            WebGPUShaderModule(program_id),
+            descriptor.parent.label.clone(),
+            promise.clone(),
+        );
+        let sender = response_async(&promise, &*shader_module);
+        device
+            .channel()
+            .0
+            .send(WebGPURequest::CreateShaderModule {
+                device_id: device.id().0,
+                program_id,
+                program: descriptor.code.0.clone(),
+                label: None,
+                sender,
+            })
+            .expect("Failed to create WebGPU ShaderModule");
+        shader_module
+    }
+}
+
+impl GPUShaderModuleMethods<crate::DomTypeHolder> for GPUShaderModule {
+    /// <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-gpushadermodule-getcompilationinfo>
+    fn GetCompilationInfo(&self) -> Rc<Promise> {
+        self.compilation_info_promise.clone()
+    }
+}
+
+impl AsyncWGPUListener for GPUShaderModule {
+    fn handle_response(&self, response: WebGPUResponse, promise: &Rc<Promise>, can_gc: CanGc) {
+        match response {
+            WebGPUResponse::CompilationInfo(info) => {
+                let info = GPUCompilationInfo::from(&self.global(), info, can_gc);
+                promise.resolve_native(&info);
+            },
+            _ => unreachable!("Wrong response received on AsyncWGPUListener for GPUShaderModule"),
+        }
+    }
+}
+
+impl Drop for GPUShaderModule {
+    fn drop(&mut self) {
+        if let Err(e) = self
+            .channel
+            .0
+            .send(WebGPURequest::DropShaderModule(self.shader_module.0))
+        {
+            warn!(
+                "Failed to send DropShaderModule ({:?}) ({})",
+                self.shader_module.0, e
+            );
+        }
+    }
+}