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
|
// WgslEffect: generic shader-only post-process effect.
#include "gpu/wgsl_effect.h"
#include "gpu/gpu.h"
#include "gpu/post_process_helper.h"
#include "util/fatal_error.h"
WgslEffect::WgslEffect(const GpuContext& ctx,
const std::vector<std::string>& inputs,
const std::vector<std::string>& outputs,
float start_time, float end_time,
const char* shader_code, WGPULoadOp load_op,
WgslEffectParams initial_params)
: Effect(ctx, inputs, outputs, start_time, end_time),
effect_params(initial_params),
load_op_(load_op) {
HEADLESS_RETURN_IF_NULL(ctx_.device);
create_linear_sampler();
params_buffer_.init(ctx_.device);
pipeline_.set(create_post_process_pipeline(ctx_.device,
WGPUTextureFormat_RGBA8Unorm,
shader_code));
}
void WgslEffect::render(WGPUCommandEncoder encoder,
const UniformsSequenceParams& params,
NodeRegistry& nodes) {
WGPUTextureView input_view = nodes.get_view(input_nodes_[0]);
WGPUTextureView output_view = nodes.get_view(output_nodes_[0]);
params_buffer_.update(ctx_.queue, effect_params);
pp_update_bind_group(ctx_.device, pipeline_.get(), bind_group_.get_address(),
input_view, uniforms_buffer_.get(),
params_buffer_.get());
run_fullscreen_pass(encoder, pipeline_.get(), bind_group_.get(), output_view,
load_op_);
}
|
