PointGPUBuffer.hpp

Go to the documentation of this file.

#pragma once
 
#include "component/Mesh.hpp"
#include "entity/Entity.hpp"
#include "exception/NonexistentComponentError.hpp"
#include "exception/UpdateBufferError.hpp"
#include "resource/AGPUBuffer.hpp"
 
namespace DefaultPipeline::Resource {
class PointGPUBuffer : public Graphic::Resource::AGPUBuffer {
  public:
    explicit PointGPUBuffer(Engine::Entity entity) : _entity(entity) {}
    ~PointGPUBuffer() override { Destroy(); }
    void Create(Engine::Core &core) override
    {
 
        auto meshComponent = _entity.TryGetComponent<Object::Component::Mesh>();
 
        if (!meshComponent)
        {
            throw Graphic::Exception::NonexistentComponentError(
                "Cannot create a GPU buffer for an entity without a Mesh component.");
        }
 
        if (meshComponent->GetNormals().size() != meshComponent->GetVertices().size() ||
            meshComponent->GetTexCoords().size() != meshComponent->GetVertices().size())
        {
            throw Graphic::Exception::UpdateBufferError(
                "Cannot create GPU buffer: normals or texCoords size mismatch with vertices.");
        }
 
        wgpu::BufferDescriptor bufferDesc(wgpu::Default);
        bufferDesc.usage = wgpu::BufferUsage::CopyDst | wgpu::BufferUsage::Vertex;
        bufferDesc.size = sizeof(float) * meshComponent->GetVertices().size() * 8;
        std::string label = fmt::format("PointGPUBuffer_{}", _entity);
        bufferDesc.label = wgpu::StringView(label);
 
        const auto &context = core.GetResource<Graphic::Resource::Context>();
        _buffer = context.deviceContext.GetDevice()->createBuffer(bufferDesc);
 
        _isCreated = true;
        Update(core);
    };
    void Destroy(Engine::Core &core) override { Destroy(); };
 
    void Destroy()
    {
        if (_isCreated)
        {
            _isCreated = false;
            _buffer.release();
        }
    }
 
    bool IsCreated(Engine::Core &core) const override { return _isCreated; };
    void Update(Engine::Core &core) override
    {
        if (!_isCreated)
        {
            throw Graphic::Exception::UpdateBufferError("Cannot update a GPU buffer that is not created.");
        }
 
        const auto &meshComponent = _entity.GetComponents<Object::Component::Mesh>();
        const auto &vertices = meshComponent.GetVertices();
 
        if (vertices.empty())
        {
            throw Graphic::Exception::UpdateBufferError(
                "Cannot update a GPU buffer from a Mesh component with no vertices.");
        }
 
        const auto &normals = meshComponent.GetNormals();
        const auto &texCoords = meshComponent.GetTexCoords();
 
        if (normals.size() != vertices.size() || texCoords.size() != vertices.size())
        {
            throw Graphic::Exception::UpdateBufferError(
                "Cannot update GPU buffer: normals or texCoords size mismatch with vertices.");
        }
 
        std::vector<float> pointData;
        pointData.reserve(vertices.size() * 8u);
 
        for (uint64_t i = 0u; i < vertices.size(); ++i)
        {
            pointData.emplace_back(vertices[i].x);
            pointData.emplace_back(vertices[i].y);
            pointData.emplace_back(vertices[i].z);
            pointData.emplace_back(normals[i].x);
            pointData.emplace_back(normals[i].y);
            pointData.emplace_back(normals[i].z);
            pointData.emplace_back(texCoords[i].x);
            pointData.emplace_back(texCoords[i].y);
        }
 
        core.GetResource<Graphic::Resource::Context>().queue->writeBuffer(_buffer, 0, pointData.data(),
                                                                          sizeof(float) * pointData.size());
    };
 
    const wgpu::Buffer &GetBuffer() const override { return _buffer; };
 
  private:
    wgpu::Buffer _buffer;
    bool _isCreated = false;
    Engine::Entity _entity;
};
} // namespace DefaultPipeline::Resource