blob: 42fb0ecb8598c889ee2f2256c5ffa39b9827053f (
plain)
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
|
// GBufferEffect: Multi-pass G-buffer rendering for CNN v3 input
// Outputs: gbuf_feat0, gbuf_feat1 (packed rgba32uint feature textures, 32 bytes/pixel)
#pragma once
#include "3d/camera.h"
#include "3d/scene.h"
#include "gpu/effect.h"
#include "gpu/sequence.h"
#include "gpu/uniform_helper.h"
#include "gpu/wgpu_resource.h"
#include "util/mini_math.h"
// Uniform for the pack compute shader
struct GBufResUniforms {
vec2 resolution;
float _pad0;
float _pad1;
};
static_assert(sizeof(GBufResUniforms) == 16,
"GBufResUniforms must be 16 bytes");
class GBufferEffect : public Effect {
public:
GBufferEffect(const GpuContext& ctx, const std::vector<std::string>& inputs,
const std::vector<std::string>& outputs, float start_time,
float end_time);
void declare_nodes(NodeRegistry& registry) override;
void render(WGPUCommandEncoder encoder, const UniformsSequenceParams& params,
NodeRegistry& nodes) override;
void set_scene(const Scene* scene, const Camera* camera);
private:
// Internal G-buffer node names
std::string node_albedo_;
std::string node_normal_mat_;
std::string node_depth_;
std::string node_shadow_;
std::string node_transp_;
std::string node_feat0_;
std::string node_feat1_;
const Scene* scene_ = nullptr;
const Camera* camera_ = nullptr;
// Pass 1: MRT rasterization pipeline
RenderPipeline raster_pipeline_;
BindGroup raster_bind_group_;
// Pass 4: Pack compute pipeline
ComputePipeline pack_pipeline_;
BindGroup pack_bind_group_;
UniformBuffer<GBufResUniforms> pack_res_uniform_;
// Placeholder textures for shadow/transp (white/black cleared once)
Texture shadow_placeholder_tex_;
TextureView shadow_placeholder_view_;
Texture transp_placeholder_tex_;
TextureView transp_placeholder_view_;
// GPU-side object data buffers (global uniforms + objects storage)
// These mirror the layout expected by gbuf_raster.wgsl
GpuBuffer global_uniforms_buf_;
GpuBuffer objects_buf_;
int objects_buf_capacity_ = 0; // number of ObjectData slots allocated
void create_raster_pipeline();
void create_pack_pipeline();
void update_raster_bind_group(NodeRegistry& nodes);
void update_pack_bind_group(NodeRegistry& nodes);
void upload_scene_data(const Scene& scene, const Camera& camera,
float time);
void ensure_objects_buffer(int num_objects);
};
|
