From 7b04930aa07ca8aee44b2c7911a9258845e78958 Mon Sep 17 00:00:00 2001
From: Ben Clayton <bclayton@google.com>
Date: Tue, 28 Sep 2021 14:09:40 +0000
Subject: [PATCH] dawn_node: Add binding/GPUShaderModule.cpp

Bug: dawn:1123
Change-Id: I0b0bd50314c656a527913d5dffa73032e6cdcdf2
Reviewed-on: https://dawn-review.googlesource.com/c/dawn/+/64910
Commit-Queue: Ben Clayton <bclayton@google.com>
Reviewed-by: Austin Eng <enga@chromium.org>
---
 src/dawn_node/binding/CMakeLists.txt      |   1 +
 src/dawn_node/binding/GPUShaderModule.cpp | 125 ++++++++++++++++++++++
 2 files changed, 126 insertions(+)
 create mode 100644 src/dawn_node/binding/GPUShaderModule.cpp

diff --git a/src/dawn_node/binding/CMakeLists.txt b/src/dawn_node/binding/CMakeLists.txt
index 4287e13206..e1f7de4018 100644
--- a/src/dawn_node/binding/CMakeLists.txt
+++ b/src/dawn_node/binding/CMakeLists.txt
@@ -46,6 +46,7 @@ add_library(dawn_node_binding STATIC
     "GPURenderPipeline.h"
     "GPUSampler.cpp"
     "GPUSampler.h"
+    "GPUShaderModule.cpp"
     "GPUShaderModule.h"
     "GPUSupportedLimits.cpp"
     "GPUSupportedLimits.h"
diff --git a/src/dawn_node/binding/GPUShaderModule.cpp b/src/dawn_node/binding/GPUShaderModule.cpp
new file mode 100644
index 0000000000..3323ba5c83
--- /dev/null
+++ b/src/dawn_node/binding/GPUShaderModule.cpp
@@ -0,0 +1,125 @@
+// Copyright 2021 The Dawn Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#include "src/dawn_node/binding/GPUShaderModule.h"
+
+#include <memory>
+
+#include "src/dawn_node/utils/Debug.h"
+
+namespace wgpu { namespace binding {
+
+    ////////////////////////////////////////////////////////////////////////////////
+    // wgpu::bindings::GPUShaderModule
+    ////////////////////////////////////////////////////////////////////////////////
+    GPUShaderModule::GPUShaderModule(wgpu::ShaderModule shader, std::shared_ptr<AsyncRunner> async)
+        : shader_(std::move(shader)), async_(std::move(async)) {
+    }
+
+    interop::Promise<interop::Interface<interop::GPUCompilationInfo>>
+    GPUShaderModule::compilationInfo(Napi::Env env) {
+        struct GPUCompilationMessage : public interop::GPUCompilationMessage {
+            WGPUCompilationMessage message;
+
+            GPUCompilationMessage(const WGPUCompilationMessage& m) : message(m) {
+            }
+            std::string getMessage(Napi::Env) override {
+                return message.message;
+            }
+            interop::GPUCompilationMessageType getType(Napi::Env) override {
+                switch (message.type) {
+                    case WGPUCompilationMessageType_Error:
+                        return interop::GPUCompilationMessageType::kError;
+                    case WGPUCompilationMessageType_Warning:
+                        return interop::GPUCompilationMessageType::kWarning;
+                    case WGPUCompilationMessageType_Info:
+                        return interop::GPUCompilationMessageType::kInfo;
+                    default:
+                        UNIMPLEMENTED();
+                }
+            }
+            uint64_t getLineNum(Napi::Env) override {
+                return message.lineNum;
+            }
+            uint64_t getLinePos(Napi::Env) override {
+                return message.linePos;
+            }
+            uint64_t getOffset(Napi::Env) override {
+                return message.offset;
+            }
+            uint64_t getLength(Napi::Env) override {
+                return message.length;
+            }
+        };
+
+        using Messages = std::vector<interop::Interface<interop::GPUCompilationMessage>>;
+
+        struct GPUCompilationInfo : public interop::GPUCompilationInfo {
+            std::vector<Napi::ObjectReference> messages;
+
+            GPUCompilationInfo(Napi::Env env, Messages msgs) {
+                messages.reserve(msgs.size());
+                for (auto& msg : msgs) {
+                    messages.emplace_back(Napi::Persistent(Napi::Object(env, msg)));
+                }
+            }
+            Messages getMessages(Napi::Env) override {
+                Messages out;
+                out.reserve(messages.size());
+                for (auto& msg : messages) {
+                    out.emplace_back(msg.Value());
+                }
+                return out;
+            }
+        };
+
+        using Promise = interop::Promise<interop::Interface<interop::GPUCompilationInfo>>;
+
+        struct Context {
+            Napi::Env env;
+            Promise promise;
+            AsyncTask task;
+        };
+        auto ctx = new Context{env, env, async_};
+        auto promise = ctx->promise;
+
+        shader_.GetCompilationInfo(
+            [](WGPUCompilationInfoRequestStatus status, WGPUCompilationInfo const* compilationInfo,
+               void* userdata) {
+                auto c = std::unique_ptr<Context>(static_cast<Context*>(userdata));
+
+                Messages messages(compilationInfo->messageCount);
+                for (uint32_t i = 0; i < compilationInfo->messageCount; i++) {
+                    auto& msg = compilationInfo->messages[i];
+                    messages[i] =
+                        interop::GPUCompilationMessage::Create<GPUCompilationMessage>(c->env, msg);
+                }
+
+                c->promise.Resolve(interop::GPUCompilationInfo::Create<GPUCompilationInfo>(
+                    c->env, c->env, std::move(messages)));
+            },
+            ctx);
+
+        return promise;
+    }
+
+    std::optional<std::string> GPUShaderModule::getLabel(Napi::Env) {
+        UNIMPLEMENTED();
+    }
+
+    void GPUShaderModule::setLabel(Napi::Env, std::optional<std::string> value) {
+        UNIMPLEMENTED();
+    }
+
+}}  // namespace wgpu::binding