#include "config.h"
#include "WebGPUDevice.h"
#if ENABLE(WEBGPU)
#include "GPUCommandBuffer.h"
#include "GPUPipelineStageDescriptor.h"
#include "GPURenderPipelineDescriptor.h"
#include "GPUShaderModuleDescriptor.h"
#include "Logging.h"
#include "WebGPUBindGroupLayout.h"
#include "WebGPUBuffer.h"
#include "WebGPUCommandBuffer.h"
#include "WebGPUPipelineLayout.h"
#include "WebGPUPipelineLayoutDescriptor.h"
#include "WebGPUPipelineStageDescriptor.h"
#include "WebGPUQueue.h"
#include "WebGPURenderPipeline.h"
#include "WebGPURenderPipelineDescriptor.h"
#include "WebGPUShaderModule.h"
#include "WebGPUShaderModuleDescriptor.h"
namespace WebCore {
RefPtr<WebGPUDevice> WebGPUDevice::create(Ref<WebGPUAdapter>&& adapter)
{
if (auto device = GPUDevice::create()) return adoptRef(new WebGPUDevice(WTFMove(adapter), device.releaseNonNull()));
return nullptr;
}
WebGPUDevice::WebGPUDevice(Ref<WebGPUAdapter>&& adapter, Ref<GPUDevice>&& device)
: m_adapter(WTFMove(adapter))
, m_device(WTFMove(device))
{
UNUSED_PARAM(m_adapter);
}
RefPtr<WebGPUBuffer> WebGPUDevice::createBuffer(WebGPUBufferDescriptor&& descriptor) const
{
if (auto buffer = m_device->createBuffer(GPUBufferDescriptor { descriptor.size, descriptor.usage }))
return WebGPUBuffer::create(buffer.releaseNonNull());
return nullptr;
}
Ref<WebGPUBindGroupLayout> WebGPUDevice::createBindGroupLayout(WebGPUBindGroupLayoutDescriptor&& descriptor) const
{
auto layout = m_device->tryCreateBindGroupLayout(GPUBindGroupLayoutDescriptor { descriptor.bindings });
return WebGPUBindGroupLayout::create(WTFMove(layout));
}
Ref<WebGPUPipelineLayout> WebGPUDevice::createPipelineLayout(WebGPUPipelineLayoutDescriptor&& descriptor) const
{
auto bindGroupLayouts = descriptor.bindGroupLayouts.map([] (const auto& layout) -> RefPtr<const GPUBindGroupLayout> {
return layout->bindGroupLayout();
});
auto layout = m_device->createPipelineLayout(GPUPipelineLayoutDescriptor { WTFMove(bindGroupLayouts) });
return WebGPUPipelineLayout::create(WTFMove(layout));
}
RefPtr<WebGPUShaderModule> WebGPUDevice::createShaderModule(WebGPUShaderModuleDescriptor&& descriptor) const
{
if (auto module = m_device->createShaderModule(GPUShaderModuleDescriptor { descriptor.code }))
return WebGPUShaderModule::create(module.releaseNonNull());
return nullptr;
}
static Optional<GPUPipelineStageDescriptor> validateAndConvertPipelineStage(const WebGPUPipelineStageDescriptor& descriptor)
{
if (!descriptor.module || !descriptor.module->module() || descriptor.entryPoint.isEmpty())
return WTF::nullopt;
return GPUPipelineStageDescriptor { descriptor.module->module(), descriptor.entryPoint };
}
RefPtr<WebGPURenderPipeline> WebGPUDevice::createRenderPipeline(WebGPURenderPipelineDescriptor&& descriptor) const
{
auto vertexStage = validateAndConvertPipelineStage(descriptor.vertexStage);
auto fragmentStage = validateAndConvertPipelineStage(descriptor.fragmentStage);
if (!vertexStage || !fragmentStage) {
LOG(WebGPU, "WebGPUDevice::createRenderPipeline(): Invalid WebGPUPipelineStageDescriptor!");
return nullptr;
}
if (auto pipeline = m_device->createRenderPipeline(GPURenderPipelineDescriptor { WTFMove(*vertexStage), WTFMove(*fragmentStage), descriptor.primitiveTopology, descriptor.inputState }))
return WebGPURenderPipeline::create(pipeline.releaseNonNull());
return nullptr;
}
RefPtr<WebGPUCommandBuffer> WebGPUDevice::createCommandBuffer() const
{
if (auto commandBuffer = m_device->createCommandBuffer())
return WebGPUCommandBuffer::create(commandBuffer.releaseNonNull());
return nullptr;
}
RefPtr<WebGPUQueue> WebGPUDevice::getQueue()
{
if (!m_queue)
m_queue = WebGPUQueue::create(m_device->getQueue());
return m_queue;
}
}
#endif // ENABLE(WEBGPU)