ContextD3D11.h

#pragma once
 
#include <d3d11_1.h>
 
#include "../Base/ContextCommon.h"
 
#include "EffectsD3D11.h"
 
namespace Cogs
{
  struct BufferD3D11;
  struct BuffersD3D11;
  struct EffectsD3D11;
  struct GraphicsDeviceD3D11;
  struct RenderTargetsD3D11;
  struct SyncObjectsD3D11;
  struct TexturesD3D11;
 
  struct InputAssemblerStateD3D11 : public InputAssemblerStateCommon
  {
    ID3D11Buffer * currentVertexBuffers[kMaxVertexInputSlots] = {};
    BufferD3D11 * currentIndexBuffer = nullptr;
  };
 
  struct ContextD3D11 : public ContextCommon
  {
    ContextD3D11();
    ~ContextD3D11();
 
    void initialize(GraphicsDeviceD3D11 * graphicsDevice, BuffersD3D11 * buffers, TexturesD3D11 * textures, EffectsD3D11 * effects, RenderTargetsD3D11 * renderTargets, SyncObjectsD3D11* sync);
 
    void setDefaults();
 
    void pushCommandGroupAnnotation(const StringView & name) override;
    void popCommandGroupAnnotation() override;
    void setAnnotationMarker(const StringView & name) override;
 
    void setEffect(EffectHandle handle);
 
    void signal(FenceHandle fenceHandle) override;
 
    void beginRenderPass(const RenderPassInfo &info) override;
    void endRenderPass() override;
 
    void setRenderTarget(const RenderTargetHandle handle, const DepthStencilHandle depthStencilHandle) override;
    void setViewport(const float x, const float y, const float width, const float height);
    void setScissor(const int x, const int y, const int width, const int height);
 
    void clearRenderTarget(const float * value);
    void clearRenderTarget(const float ** values, const int numValue);
    void clearDepth(const float depth = 1.0f);
 
    void setDepthStencilState(const DepthStencilStateHandle handle);
    void setRasterizerState(const RasterizerStateHandle handle);
    void setBlendState(const BlendStateHandle handle, const float* constant) override;
 
    void resolveResource(TextureHandle sourceTexture, TextureHandle destination);
 
    void setInputLayout(const InputLayoutHandle inputLayoutHandle);
    void setVertexBuffers(const VertexBufferHandle * vertexBufferHandles, const size_t count, const uint32_t * strides, const uint32_t * offsets);
    void setVertexBuffers(const VertexBufferHandle * vertexBufferHandles, const size_t count);
    void setIndexBuffer(IndexBufferHandle indexBufferHandle, uint32_t stride, uint32_t offset);
    void setConstantBuffer(const ConstantBufferBindingHandle bufferBinding, const BufferHandle bufferHandle, const uint32_t offset, const uint32_t size);
 
    void setupConstantBuffers();
    void endFrame();
    
    void draw(PrimitiveType::EPrimitiveType primitiveType, const size_t startVertex, const size_t numVertexes);
    void drawIndexed(PrimitiveType::EPrimitiveType primitiveType, const size_t startIndex, const size_t numIndexes, const size_t startVertex = 0);
 
    void drawInstanced(PrimitiveType::EPrimitiveType primitiveType, const size_t startVertex, const size_t numVertexes, const size_t startInstance, const size_t numInstances);
    void drawInstancedIndexed(PrimitiveType::EPrimitiveType primitiveType, const size_t startInstance, const size_t numInstances, const size_t startIndex, const size_t numIndexes);
    
    void dispatchCompute(const unsigned int threadGroupsX, const unsigned int threadGroupsY, const unsigned int threadGroupsZ);
 
    void readDepthBuffer(BufferHandle bufferHandle, int x, int y, int width, int height, Framebuffer::EFrameBuffer framebuffer);
    void readColorBuffer(BufferHandle bufferHandle, int x, int y, int width, int height, Framebuffer::EFrameBuffer framebuffer);
    
    void * map(BufferHandle bufferHandle, MapMode::EMapMode accessMode, uint32_t * stride);
    void unmap(BufferHandle bufferHandle);
 
    void * map(TextureHandle textureHandle, MapMode::EMapMode accessMode, uint32_t * rowPitch, uint32_t * depthPitch) override;
    void unmap(TextureHandle textureHandle) override;
 
    void updateSubTexture(TextureHandle textureHandle, const size_t level, const void * data);
    void updateSubBuffer(BufferHandle bufferHandle, const size_t offset, const size_t size, const void * data) override;
 
    void copyResource(BufferHandle destinationHandle, BufferHandle sourceHandle);
    void copyResource(TextureHandle destinationHandle, TextureHandle sourceHandle);
 
    void copyTexture(TextureHandle dstHandle, unsigned dstSub, unsigned dstX, unsigned dstY, unsigned dstZ, TextureHandle sourceHandle, unsigned srcSub) override;
 
    void clearResource(BufferHandle destinationHandle, uint32_t *Values);
    void clearResource(BufferHandle destinationHandle, float *Values);
 
    void setTexture(const TextureBindingHandle textureBindingHandle, const TextureHandle textureHandle);
    void setTexture(const TextureBindingHandle textureBindingHandle, TextureViewHandle textureViewHandle) override;
    void setSamplerState(const SamplerStateBindingHandle samplerStateBindingHandle, const SamplerStateHandle samplerStateHandle);
 
    void setBuffer(const BufferBindingHandle bufferBindingHandle, BufferHandle bufferHandle);
 
    void setBufferCounter(BufferHandle bufferHandle, uint32_t value);
    void setBufferCounter(BufferHandle bufferHandle, BufferHandle sourceBufferHandle);
    void getBufferCounter(BufferHandle bufferHandle, BufferHandle destinationBufferHandle);
 
    void setShaderResources(uint32_t * slots, ID3D11ShaderResourceView *const *ppShaderResourceViews);
 
    uint32_t getBufferCounter(BufferHandle bufferHandle);
 
    void setDevice(ResourcePointer<ID3D11Device>& device, ResourcePointer<ID3D11Device1>& device1);
 
    void resetSRVBindings();
 
    void updateRenderTargets();
 
    EffectHandle currentEffect;
    bool isCompute;
 
    ResourcePointer<ID3D11Device> device;
    ResourcePointer<ID3D11Device1> device1;
    ResourcePointer<ID3D11DeviceContext> immediateContext;
    ResourcePointer<ID3D11DeviceContext1> immediateContext1;
    ResourcePointer<ID3D11DeviceContext4> immediateContext4;
 
    ResourcePointer<ID3DUserDefinedAnnotation> annotationInterface;
 
    InputAssemblerStateCommon * getIAState() override { return &iaState; }
    EffectHandle getCurrentEffect() override { return currentEffect; }
    IEffects * getEffects() override { return effects; }
 
    InputAssemblerStateD3D11 iaState;
 
    ID3D11UnorderedAccessView * uavs[D3D11_PS_CS_UAV_REGISTER_COUNT] = {};
    uint32_t uavCounts[16] = {};
    uint32_t minUavSlot = ~0u;
    uint32_t maxUavSlot = 0;
    bool uavDirty = false;
 
    GraphicsDeviceD3D11 * graphicsDevice = nullptr;
    BuffersD3D11 * buffers = nullptr;
    TexturesD3D11 * textures = nullptr;
    EffectsD3D11 * effects = nullptr;
    RenderTargetsD3D11 * renderTargets = nullptr;
    SyncObjectsD3D11* syncObjects = nullptr;
 
    DepthStencilStateHandle defaultDepthStencilState;
    RasterizerStateHandle defaultRasterizerState;
    BlendStateHandle defaultBlendState;
    SamplerStateHandle defaultSamplerState;
 
    bool inRenderPass = false;
    RenderPassInfo renderPassInfo = {};
 
    RenderTargetHandle currentRenderTarget;
    DepthStencilHandle currentDepthStencil;
 
    InputLayoutHandle currentInputLayout;
  };
}