1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
// WGPU bind group builder - reduces boilerplate for bind group creation
#pragma once
#include <vector>
// Forward declarations (users must include gpu.h)
struct WGPUBindGroupLayoutEntry;
struct WGPUBindGroupEntry;
struct WGPUDeviceImpl;
typedef struct WGPUDeviceImpl* WGPUDevice;
struct WGPUBindGroupLayoutImpl;
typedef struct WGPUBindGroupLayoutImpl* WGPUBindGroupLayout;
struct WGPUBindGroupImpl;
typedef struct WGPUBindGroupImpl* WGPUBindGroup;
struct WGPUBufferImpl;
typedef struct WGPUBufferImpl* WGPUBuffer;
struct WGPUTextureViewImpl;
typedef struct WGPUTextureViewImpl* WGPUTextureView;
struct WGPUSamplerImpl;
typedef struct WGPUSamplerImpl* WGPUSampler;
typedef uint32_t WGPUShaderStageFlags;
#include "platform/platform.h"
class BindGroupLayoutBuilder {
std::vector<WGPUBindGroupLayoutEntry> entries_;
public:
BindGroupLayoutBuilder& uniform(uint32_t binding, WGPUShaderStageFlags vis, size_t min_size = 0) {
WGPUBindGroupLayoutEntry e{};
e.binding = binding;
e.visibility = vis;
e.buffer.type = WGPUBufferBindingType_Uniform;
if (min_size) e.buffer.minBindingSize = min_size;
entries_.push_back(e);
return *this;
}
BindGroupLayoutBuilder& storage(uint32_t binding, WGPUShaderStageFlags vis, size_t min_size = 0) {
WGPUBindGroupLayoutEntry e{};
e.binding = binding;
e.visibility = vis;
e.buffer.type = WGPUBufferBindingType_ReadOnlyStorage;
if (min_size) e.buffer.minBindingSize = min_size;
entries_.push_back(e);
return *this;
}
BindGroupLayoutBuilder& texture(uint32_t binding, WGPUShaderStageFlags vis) {
WGPUBindGroupLayoutEntry e{};
e.binding = binding;
e.visibility = vis;
e.texture.sampleType = WGPUTextureSampleType_Float;
e.texture.viewDimension = WGPUTextureViewDimension_2D;
entries_.push_back(e);
return *this;
}
BindGroupLayoutBuilder& sampler(uint32_t binding, WGPUShaderStageFlags vis) {
WGPUBindGroupLayoutEntry e{};
e.binding = binding;
e.visibility = vis;
e.sampler.type = WGPUSamplerBindingType_Filtering;
entries_.push_back(e);
return *this;
}
WGPUBindGroupLayout build(WGPUDevice device) {
WGPUBindGroupLayoutDescriptor desc{};
desc.entryCount = entries_.size();
desc.entries = entries_.data();
return wgpuDeviceCreateBindGroupLayout(device, &desc);
}
};
class BindGroupBuilder {
std::vector<WGPUBindGroupEntry> entries_;
public:
BindGroupBuilder& buffer(uint32_t binding, WGPUBuffer buf, size_t size) {
WGPUBindGroupEntry e{};
e.binding = binding;
e.buffer = buf;
e.size = size;
entries_.push_back(e);
return *this;
}
BindGroupBuilder& texture(uint32_t binding, WGPUTextureView view) {
WGPUBindGroupEntry e{};
e.binding = binding;
e.textureView = view;
entries_.push_back(e);
return *this;
}
BindGroupBuilder& sampler(uint32_t binding, WGPUSampler samp) {
WGPUBindGroupEntry e{};
e.binding = binding;
e.sampler = samp;
entries_.push_back(e);
return *this;
}
WGPUBindGroup build(WGPUDevice device, WGPUBindGroupLayout layout) {
WGPUBindGroupDescriptor desc{};
desc.layout = layout;
desc.entryCount = entries_.size();
desc.entries = entries_.data();
return wgpuDeviceCreateBindGroup(device, &desc);
}
};
|
