[WebGPU] Improve GPUBindGroup performance using one device-shared argument MTLBuffer
[WebKit-https.git] / Source / WebCore / platform / graphics / gpu / GPUDevice.h
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 #pragma once
27
28 #if ENABLE(WEBGPU)
29
30 #include "GPUBindGroupAllocator.h"
31 #include "GPUQueue.h"
32 #include "GPUSwapChain.h"
33 #include <wtf/Function.h>
34 #include <wtf/Optional.h>
35 #include <wtf/Ref.h>
36 #include <wtf/RefCounted.h>
37 #include <wtf/RefPtr.h>
38 #include <wtf/RetainPtr.h>
39 #include <wtf/WeakPtr.h>
40
41 OBJC_PROTOCOL(MTLDevice);
42
43 namespace WebCore {
44
45 class GPUBindGroup;
46 class GPUBindGroupLayout;
47 class GPUBuffer;
48 class GPUCommandBuffer;
49 class GPUComputePipeline;
50 class GPUErrorScopes;
51 class GPUPipelineLayout;
52 class GPURenderPipeline;
53 class GPUSampler;
54 class GPUShaderModule;
55 class GPUTexture;
56
57 struct GPUBindGroupDescriptor;
58 struct GPUBindGroupLayoutDescriptor;
59 struct GPUBufferDescriptor;
60 struct GPUComputePipelineDescriptor;
61 struct GPUPipelineLayoutDescriptor;
62 struct GPURenderPipelineDescriptor;
63 struct GPURequestAdapterOptions;
64 struct GPUSamplerDescriptor;
65 struct GPUShaderModuleDescriptor;
66 struct GPUSwapChainDescriptor;
67 struct GPUTextureDescriptor;
68
69 enum class GPUBufferMappedOption;
70
71 using PlatformDevice = MTLDevice;
72 using PlatformDeviceSmartPtr = RetainPtr<MTLDevice>;
73
74 class GPUDevice : public RefCounted<GPUDevice>, public CanMakeWeakPtr<GPUDevice> {
75 public:
76     static RefPtr<GPUDevice> tryCreate(const Optional<GPURequestAdapterOptions>&);
77
78     RefPtr<GPUBuffer> tryCreateBuffer(const GPUBufferDescriptor&, GPUBufferMappedOption, GPUErrorScopes&);
79     RefPtr<GPUTexture> tryCreateTexture(const GPUTextureDescriptor&) const;
80     RefPtr<GPUSampler> tryCreateSampler(const GPUSamplerDescriptor&) const;
81
82     RefPtr<GPUBindGroupLayout> tryCreateBindGroupLayout(const GPUBindGroupLayoutDescriptor&) const;
83     Ref<GPUPipelineLayout> createPipelineLayout(GPUPipelineLayoutDescriptor&&) const;
84     RefPtr<GPUBindGroup> tryCreateBindGroup(const GPUBindGroupDescriptor&, GPUErrorScopes&) const;
85
86     RefPtr<GPUShaderModule> tryCreateShaderModule(const GPUShaderModuleDescriptor&) const;
87     RefPtr<GPURenderPipeline> tryCreateRenderPipeline(const GPURenderPipelineDescriptor&, GPUErrorScopes&) const;
88     RefPtr<GPUComputePipeline> tryCreateComputePipeline(const GPUComputePipelineDescriptor&, GPUErrorScopes&) const;
89
90     RefPtr<GPUCommandBuffer> tryCreateCommandBuffer() const;
91
92     RefPtr<GPUQueue> tryGetQueue() const;
93     
94     PlatformDevice* platformDevice() const { return m_platformDevice.get(); }
95     GPUSwapChain* swapChain() const { return m_swapChain.get(); }
96     void setSwapChain(RefPtr<GPUSwapChain>&&);
97
98 private:
99     explicit GPUDevice(PlatformDeviceSmartPtr&&);
100
101     PlatformDeviceSmartPtr m_platformDevice;
102     mutable RefPtr<GPUQueue> m_queue;
103     RefPtr<GPUSwapChain> m_swapChain;
104     mutable RefPtr<GPUBindGroupAllocator> m_bindGroupAllocator;
105 };
106
107 } // namespace WebCore
108
109 #endif // ENABLE(WEBGPU)