VulkanGpuBuffer.h

浏览该文件的文档.

#pragma once
 
#include "gpgpu/GpuBuffer.h"
#include "vkbuilder.hpp"
 
#include <string>
 
namespace eve::gpgpu {
 
class VulkanGpuBuffer final : public GpuBuffer {
public:
    ~VulkanGpuBuffer() override;
 
    int getSize() const override { return int(size_); }
    std::string getUsage() const override { return usage_; }
 
    void writeData(data::ByteData *data, int dstOffset = 0) override;
    data::ByteData *readData(int srcOffset = 0, int size = -1) override;
    void writeFloat32(int floatIndex, float value) override;
    float readFloat32(int floatIndex) override;
    void fillFloat32(float value) override;
    void writeFloat32s(const float *data, int count, int startIndex = 0) override;
    void readFloat32s(float *out, int count, int startIndex = 0) const override;
    void uploadBytes(const void *src, uint64_t nbytes, uint64_t dstOffset = 0) override;
    void downloadBytes(void *dst, uint64_t nbytes, uint64_t srcOffset = 0) const override;
    GpuResidentBufferView residentView() const override;
 
    vkb::Device *device_ = nullptr;
    vk::Buffer buffer_{};
    vk::DeviceMemory memory_{};
#if defined(VKB_ENABLE_VMA)
    VmaAllocation vmaAllocation_ = VK_NULL_HANDLE;
#endif
    vk::DeviceSize size_ = 0;
    std::string usage_ = "storage";
    bool hostVisible_ = false;
};
 
}  // namespace eve::gpgpu