BuffersD3D11.h

#pragma once
 
#include "../Base/BuffersCommon.h"
 
#include <map>
 
#include "CommonD3D11.h"
 
namespace Cogs
{
  struct BufferD3D11
  {
    ResourcePointer<ID3D11Buffer> buffer;
    
    size_t size = 0;
 
    union
    {
      struct
      {
        const VertexFormat * vertexFormat;
      } vertexBuffer;
 
      struct
      {
        uint32_t count;
        DXGI_FORMAT indexFormat;
      } indexBuffer;
 
      struct
      {
        uint32_t count;
        uint32_t stride;
      } structuredBuffer;
    };
 
    uint16_t bindFlags = 0;
 
    ResourcePointer<ID3D11UnorderedAccessView> uav;
    ResourcePointer<ID3D11ShaderResourceView> srv;
 
    ResourcePointer<ID3D11Texture2D> aliasedTexture;
 
    uint32_t uavCount = 0;
    int aliasedWidth = 0;
    int aliasedHeight = 0;
    bool aliasMapped = false;
  };
 
  struct InputLayoutD3D11
  {
    ResourcePointer<ID3D11InputLayout> resource;
 
    const VertexFormat * formats[16] = {};
    size_t numFormats = 0;
  };
 
  struct BuffersD3D11 : public BuffersCommon
  {
    BuffersD3D11(struct GraphicsDeviceD3D11 * device);
 
    void setDevice(const ResourcePointer<ID3D11Device> & device_set)
    {
      this->device = device_set;
    }
 
    void initialize(struct ContextD3D11 * context_set, struct EffectsD3D11 * effects_set)
    {
      this->context = context_set;
      this->effects = effects_set;
    }
 
    void annotate(BufferHandle handle, const StringView& name) override;
 
    void annotate(VertexBufferHandle handle, const StringView& name) override;
 
    VertexBufferHandle loadVertexBuffer(const void * vertexData, const size_t count, const VertexFormat & vertexFormat);
    VertexBufferHandle loadVertexBuffer(const void * vertexData, const size_t count, VertexFormatHandle vertexFormatHandle);
    void releaseVertexBuffer(VertexBufferHandle vertexBufferHandle);
 
    IndexBufferHandle loadIndexBuffer(const void * vertexData, const size_t count, const size_t indexSize);
    void releaseIndexBuffer(IndexBufferHandle indexBufferHandle);
 
    void retrieveSubBuffer(void * /*data*/, BufferHandle /*source*/, const size_t /*offset*/, const size_t /*size*/) {}
 
    InputLayoutHandle loadInputLayout(const VertexFormatHandle * vertexFormats, const size_t count, EffectHandle effectHandle) override;
    void releaseInputLayout(InputLayoutHandle vertexFormatHandle);
 
    BufferHandle loadBuffer(const void * data, const size_t size, Usage::EUsage usage, uint32_t accessMode, uint32_t bindFlags, uint32_t stride = 0);
    void releaseBuffer(BufferHandle bufferHandle);
 
    void * getNativeHandle(BufferHandle bufferHandle) override;
 
    void releaseResources();
 
    SafeResourceMap<BufferHandle, BufferD3D11> buffers;
    size_t bufferMemoryConsumption = 0;
 
    SafeResourceMap<InputLayoutHandle, InputLayoutD3D11> inputLayouts;
    
  private:
    ResourcePointer<ID3D11Device> device;
 
    ContextD3D11 * context = nullptr;
    EffectsD3D11 * effects = nullptr;
  };
}