ShaderSystem.h

浏览该文件的文档.

#pragma once
 
#include "gpgpu/ComputeShader.h"
 
#include <cstdint>
#include <string>
#include <vector>
 
namespace eve::gpgpu {
 
class Gpgpu;
class GpuBuffer;
 
class ShaderSystem {
public:
    static constexpr int kMaxBindings = ComputeShader::kMaxBindings;
    static constexpr int kDefaultLocalSize = 64;
 
    ShaderSystem() = default;
    ~ShaderSystem();
 
    ShaderSystem(const ShaderSystem &) = delete;
    ShaderSystem &operator=(const ShaderSystem &) = delete;
 
    void setGpgpu(Gpgpu *gpu);
    Gpgpu *getGpgpu() const { return gpu_; }
 
    void setShaderSource(const std::string &glsl);
 
    void setShader(ComputeShader *shader, bool takeOwnership = false);
 
    ComputeShader *getShader() const { return shader_; }
 
    void setLocalSize(int localSize);
    int getLocalSize() const { return localSize_; }
 
    GpuBuffer *ensureBuffer(int binding, int floatCount);
 
    GpuBuffer *getBuffer(int binding) const;
 
    void upload(int binding, const float *data, int floatCount);
    void download(int binding, float *out, int floatCount) const;
 
    void upload(int binding, const std::vector<float> &data);
    std::vector<float> download(int binding, int floatCount) const;
 
    void setFloat(int index, float value);
    float getFloat(int index) const;
 
    void dispatch(int entityCount, float dt = 0.f);
 
    void clearBuffers();
 
private:
    Gpgpu *gpu_ = nullptr;
    ComputeShader *shader_ = nullptr;
    bool ownsShader_ = false;
    int localSize_ = kDefaultLocalSize;
    GpuBuffer *buffers_[kMaxBindings]{};
    bool ownsBuffer_[kMaxBindings]{};
};
 
}  // namespace eve::gpgpu