summaryrefslogtreecommitdiff
path: root/tools/shadertoy/template.cc
blob: 81c39ba4d316dcf9b261d7cdc3c7df29c9c45b3d (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
// This file is part of the 64k demo project.
// ShaderToy effect implementation - REPLACE THIS LINE
// TODO: Update description, rename class

#include "effects/shadertoy_effect.h"
#include "effects/shaders.h"
#include "gpu/gpu.h"
#include "gpu/post_process_helper.h"
#include "util/fatal_error.h"

// TODO: Rename class
ShaderToyEffect::ShaderToyEffect(const GpuContext& ctx,
                                 const std::vector<std::string>& inputs,
                                 const std::vector<std::string>& outputs,
                                 float start_time, float end_time)
    : Effect(ctx, inputs, outputs, start_time, end_time) {
  HEADLESS_RETURN_IF_NULL(ctx_.device);
  create_nearest_sampler();
  create_dummy_scene_texture();

  // TODO: Update shader name to match declaration in shaders.h
  pipeline_.set(create_post_process_pipeline(
      ctx_.device, WGPUTextureFormat_RGBA8Unorm, shadertoy_shader_wgsl));
}

void ShaderToyEffect::render(WGPUCommandEncoder encoder,
                             const UniformsSequenceParams& params,
                             NodeRegistry& nodes) {
  WGPUTextureView output_view = nodes.get_view(output_nodes_[0]);

  // uniforms_buffer_ is auto-updated by base class dispatch_render()
  pp_update_bind_group(ctx_.device, pipeline_.get(), bind_group_.get_address(),
                       dummy_texture_view_.get(), uniforms_buffer_.get(),
                       {nullptr, 0});

  WGPURenderPassColorAttachment color_attachment = {};
  gpu_init_color_attachment(color_attachment, output_view);

  WGPURenderPassDescriptor pass_desc = {};
  pass_desc.colorAttachmentCount = 1;
  pass_desc.colorAttachments = &color_attachment;

  WGPURenderPassEncoder pass =
      wgpuCommandEncoderBeginRenderPass(encoder, &pass_desc);
  wgpuRenderPassEncoderSetPipeline(pass, pipeline_.get());
  wgpuRenderPassEncoderSetBindGroup(pass, 0, bind_group_.get(), 0, nullptr);
  wgpuRenderPassEncoderDraw(pass, 3, 1, 0, 0);
  wgpuRenderPassEncoderEnd(pass);
  wgpuRenderPassEncoderRelease(pass);
}