src/gpu/uniform_helper.h


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

// This file is part of the 64k demo project.
// Generic uniform buffer helper.

#pragma once

#include "gpu/gpu.h"

// Usage:
//   UniformBuffer<MyUniforms> uniforms_;
//   uniforms_.init(device);
//   uniforms_.update(queue, my_data);
template <typename T> class UniformBuffer {
 public:
  UniformBuffer() = default;

  void init(WGPUDevice device) {
    buffer_ = gpu_create_buffer(
        device, sizeof(T), WGPUBufferUsage_Uniform | WGPUBufferUsage_CopyDst);
  }

  void update(WGPUQueue queue, const T& data) {
    wgpuQueueWriteBuffer(queue, buffer_.buffer, 0, &data, sizeof(T));
  }

  GpuBuffer& get() {
    return buffer_;
  }

  const GpuBuffer& get() const {
    return buffer_;
  }

 private:
  GpuBuffer buffer_;
};