[WebGPU] Map WebGPUBindGroupLayoutBindings from the BindGroupLayoutDescriptor for...
[WebKit-https.git] / Source / WebCore / Modules / webgpu / WebGPUDevice.cpp
1 /*
2  * Copyright (C) 2018 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #include "config.h"
27 #include "WebGPUDevice.h"
28
29 #if ENABLE(WEBGPU)
30
31 #include "GPUBindGroup.h"
32 #include "GPUBindGroupBinding.h"
33 #include "GPUBindGroupDescriptor.h"
34 #include "GPUBufferBinding.h"
35 #include "GPUCommandBuffer.h"
36 #include "GPUPipelineStageDescriptor.h"
37 #include "GPURenderPipelineDescriptor.h"
38 #include "GPUShaderModuleDescriptor.h"
39 #include "Logging.h"
40 #include "WebGPUBindGroup.h"
41 #include "WebGPUBindGroupBinding.h"
42 #include "WebGPUBindGroupDescriptor.h"
43 #include "WebGPUBindGroupLayout.h"
44 #include "WebGPUBuffer.h"
45 #include "WebGPUBufferBinding.h"
46 #include "WebGPUCommandBuffer.h"
47 #include "WebGPUPipelineLayout.h"
48 #include "WebGPUPipelineLayoutDescriptor.h"
49 #include "WebGPUPipelineStageDescriptor.h"
50 #include "WebGPUQueue.h"
51 #include "WebGPURenderPipeline.h"
52 #include "WebGPURenderPipelineDescriptor.h"
53 #include "WebGPUShaderModule.h"
54 #include "WebGPUShaderModuleDescriptor.h"
55 #include <wtf/Variant.h>
56
57 namespace WebCore {
58
59 RefPtr<WebGPUDevice> WebGPUDevice::create(Ref<WebGPUAdapter>&& adapter)
60 {
61     if (auto device = GPUDevice::create()) // FIXME: Take adapter into account when creating m_device.
62         return adoptRef(new WebGPUDevice(WTFMove(adapter), device.releaseNonNull()));
63     return nullptr;
64 }
65
66 WebGPUDevice::WebGPUDevice(Ref<WebGPUAdapter>&& adapter, Ref<GPUDevice>&& device)
67     : m_adapter(WTFMove(adapter))
68     , m_device(WTFMove(device))
69 {
70     UNUSED_PARAM(m_adapter);
71 }
72
73 RefPtr<WebGPUBuffer> WebGPUDevice::createBuffer(WebGPUBufferDescriptor&& descriptor) const
74 {
75     // FIXME: Validation on descriptor needed?
76     if (auto buffer = m_device->createBuffer(GPUBufferDescriptor { descriptor.size, descriptor.usage }))
77         return WebGPUBuffer::create(buffer.releaseNonNull());
78     return nullptr;
79 }
80
81 Ref<WebGPUBindGroupLayout> WebGPUDevice::createBindGroupLayout(WebGPUBindGroupLayoutDescriptor&& descriptor) const
82 {
83     auto layout = m_device->tryCreateBindGroupLayout(GPUBindGroupLayoutDescriptor { descriptor.bindings });
84     return WebGPUBindGroupLayout::create(WTFMove(layout));
85 }
86
87 Ref<WebGPUPipelineLayout> WebGPUDevice::createPipelineLayout(WebGPUPipelineLayoutDescriptor&& descriptor) const
88 {
89     // FIXME: Is an empty pipelineLayout an error?
90     auto bindGroupLayouts = descriptor.bindGroupLayouts.map([] (const auto& layout) -> RefPtr<const GPUBindGroupLayout> {
91         return layout->bindGroupLayout();
92     });
93     auto layout = m_device->createPipelineLayout(GPUPipelineLayoutDescriptor { WTFMove(bindGroupLayouts) });
94     return WebGPUPipelineLayout::create(WTFMove(layout));
95 }
96
97 Ref<WebGPUBindGroup> WebGPUDevice::createBindGroup(WebGPUBindGroupDescriptor&& descriptor) const
98 {
99     if (!descriptor.layout || !descriptor.layout->bindGroupLayout()) {
100         LOG(WebGPU, "WebGPUDevice::createBindGroup(): Invalid WebGPUBindGroupLayout!");
101         return WebGPUBindGroup::create(nullptr);
102     }
103
104     if (descriptor.bindings.size() != descriptor.layout->bindGroupLayout()->bindingsMap().size()) {
105         LOG(WebGPU, "WebGPUDevice::createBindGroup(): Mismatched number of WebGPUBindGroupLayoutBindings and WebGPUBindGroupBindings!");
106         return WebGPUBindGroup::create(nullptr);
107     }
108
109     auto bindingResourceVisitor = WTF::makeVisitor([] (RefPtr<WebGPUTextureView> view) -> Optional<GPUBindingResource> {
110         if (view)
111             return static_cast<GPUBindingResource>(view->texture());
112         return WTF::nullopt;
113     }, [] (const WebGPUBufferBinding& binding) -> Optional<GPUBindingResource> {
114         if (binding.buffer)
115             return static_cast<GPUBindingResource>(GPUBufferBinding { binding.buffer->buffer(), binding.offset, binding.size });
116         return WTF::nullopt;
117     });
118
119     Vector<GPUBindGroupBinding> bindGroupBindings;
120     bindGroupBindings.reserveCapacity(descriptor.bindings.size());
121
122     for (const auto& binding : descriptor.bindings) {
123         if (!descriptor.layout->bindGroupLayout()->bindingsMap().contains(binding.binding)) {
124             LOG(WebGPU, "WebGPUDevice::createBindGroup(): WebGPUBindGroupBinding %lu not found in WebGPUBindGroupLayout!", binding.binding);
125             return WebGPUBindGroup::create(nullptr);
126         }
127
128         auto bindingResource = WTF::visit(bindingResourceVisitor, binding.resource);
129         if (bindingResource)
130             bindGroupBindings.uncheckedAppend(GPUBindGroupBinding { binding.binding, WTFMove(bindingResource.value()) });
131         else {
132             LOG(WebGPU, "WebGPUDevice::createBindGroup(): Invalid WebGPUBindingResource for binding %lu in WebGPUBindGroupBindings!", binding.binding);
133             return WebGPUBindGroup::create(nullptr);
134         }
135     }
136     auto bindGroup = GPUBindGroup::create(GPUBindGroupDescriptor { descriptor.layout->bindGroupLayout().releaseNonNull(), WTFMove(bindGroupBindings) });
137     return WebGPUBindGroup::create(WTFMove(bindGroup));
138 }
139
140 RefPtr<WebGPUShaderModule> WebGPUDevice::createShaderModule(WebGPUShaderModuleDescriptor&& descriptor) const
141 {
142     // FIXME: What can be validated here?
143     if (auto module = m_device->createShaderModule(GPUShaderModuleDescriptor { descriptor.code }))
144         return WebGPUShaderModule::create(module.releaseNonNull());
145     return nullptr;
146 }
147
148 static Optional<GPUPipelineStageDescriptor> validateAndConvertPipelineStage(const WebGPUPipelineStageDescriptor& descriptor)
149 {
150     if (!descriptor.module || !descriptor.module->module() || descriptor.entryPoint.isEmpty())
151         return WTF::nullopt;
152
153     return GPUPipelineStageDescriptor { descriptor.module->module(), descriptor.entryPoint };
154 }
155
156 RefPtr<WebGPURenderPipeline> WebGPUDevice::createRenderPipeline(WebGPURenderPipelineDescriptor&& descriptor) const
157 {
158     auto pipelineLayout = descriptor.layout ? descriptor.layout->pipelineLayout() : nullptr;
159
160     auto vertexStage = validateAndConvertPipelineStage(descriptor.vertexStage);
161     auto fragmentStage = validateAndConvertPipelineStage(descriptor.fragmentStage);
162
163     if (!vertexStage || !fragmentStage) {
164         LOG(WebGPU, "WebGPUDevice::createRenderPipeline(): Invalid WebGPUPipelineStageDescriptor!");
165         return nullptr;
166     }
167
168     if (auto pipeline = m_device->createRenderPipeline(GPURenderPipelineDescriptor { WTFMove(pipelineLayout), WTFMove(*vertexStage), WTFMove(*fragmentStage), descriptor.primitiveTopology, WTFMove(descriptor.inputState) }))
169         return WebGPURenderPipeline::create(pipeline.releaseNonNull());
170     return nullptr;
171 }
172
173 RefPtr<WebGPUCommandBuffer> WebGPUDevice::createCommandBuffer() const
174 {
175     if (auto commandBuffer = m_device->createCommandBuffer())
176         return WebGPUCommandBuffer::create(commandBuffer.releaseNonNull());
177     return nullptr;
178 }
179
180 RefPtr<WebGPUQueue> WebGPUDevice::getQueue()
181 {
182     if (!m_queue)
183         m_queue = WebGPUQueue::create(m_device->getQueue());
184
185     return m_queue;
186 }
187
188 } // namespace WebCore
189
190 #endif // ENABLE(WEBGPU)