ShaderSystem.h

浏览该文件的文档.

#pragma once
#include "common/Export.h"
 
 
#include "gpgpu/ComputeShader.h"
 
#include <cstdint>
#include <string>
#include <vector>
 
namespace eve::gpgpu {
 
class Gpgpu;
class GpuBuffer;
class Sequence;
 
class EVENGINE_API_WORLD 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 attachBuffer(int binding, GpuBuffer *buffer);
 
    void upload(int binding, const float *data, int floatCount);
    void download(int binding, float *out, int floatCount) const;
 
    void uploadRange(int binding, const float *data, int floatCount, int startFloat);
    void downloadRange(int binding, float *out, int floatCount, int startFloat) const;
 
    uint64_t getUploadCount() const { return uploadCount_; }
    uint64_t getDownloadCount() const { return downloadCount_; }
    uint64_t getDispatchCount() const { return dispatchCount_; }
    void resetStatistics();
 
    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 recordDispatch(Sequence *sequence, 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]{};
    uint64_t         uploadCount_   = 0;
    mutable uint64_t downloadCount_ = 0;
    uint64_t         dispatchCount_ = 0;
};
 
}  // namespace eve::gpgpu