ContextVK.h

#pragma once
 
#include "../Base/ContextCommon.h"
 
#include "CommonVK.h"
 
#include "BuffersVK.h"
#include "TexturesVK.h"
#include "EffectsVK.h"
#include "RenderTargetsVK.h"
 
#include <array>
 
#include "ContextStateVK.h"
 
namespace Cogs
{
  class ContextVK : public ContextCommon
  {
  public:
    void initialize(class GraphicsDeviceVK * graphicsDevice);
    void beginFrame();
    void endFrame();
 
    void setRenderTarget(const RenderTargetHandle handle, const DepthStencilHandle depthStencilHandle) override;
    void setViewport(const float x, const float y, const float width, const float height) override;
    void setScissor(const int x, const int y, const int width, const int height) override;
 
    void clearRenderTarget(const float * color) override;
    void clearRenderTarget(const float ** colors, const int count) override;
    void clearDepth(const float depth = 1.0f) override;
    
    void setDepthStencilState(const DepthStencilStateHandle handle) override;
    void setBlendState(const BlendStateHandle handle, const float* constant = nullptr) override;
    void setRasterizerState(const RasterizerStateHandle handle) override;
    
    void setEffect(EffectHandle handle) override;
 
    void setTexture(const TextureBindingHandle textureBindingHandle, const TextureHandle textureHandle) override;
    void setTexture(const TextureBindingHandle textureBindingHandle, TextureViewHandle textureViewHandle) override;
 
    void setSamplerState(const SamplerStateBindingHandle samplerStateBindingHandle, const SamplerStateHandle samplerStateHandle) override;
 
    void setInputLayout(const InputLayoutHandle inputLayoutHandle) override;
    void setVertexBuffers(const VertexBufferHandle * vertexBufferHandles, const size_t count, const uint32_t * strides, const uint32_t * offsets) override;
    void setVertexBuffers(const VertexBufferHandle * vertexBufferHandles, const size_t count) override;
    void setIndexBuffer(IndexBufferHandle bufferHandle, uint32_t stride = 4, uint32_t offset = 0) override;
    void setVertexArrayObject(VertexArrayObjectHandle vertexArrayObject) override;
 
    void setConstantBuffer(const ConstantBufferBindingHandle bufferBindingHandle, const BufferHandle bufferHandle, const uint32_t offset = 0, const uint32_t size = ~0u) override;
 
    void setBuffer(const BufferBindingHandle bindingHandle, BufferHandle bufferHandle) override;
 
    void setBufferCounter(BufferHandle bufferHandle, uint32_t value) override;
    void setBufferCounter(BufferHandle bufferHandle, BufferHandle sourceBufferHandle) override;
    void getBufferCounter(BufferHandle bufferHandle, BufferHandle destinationBufferHandle) override;
    uint32_t getBufferCounter(BufferHandle bufferHandle) override;
 
    void draw(PrimitiveType primitiveType, const size_t startVertex, const size_t numVertexes) override;
    void drawIndexed(PrimitiveType primitiveType, const size_t startIndex, const size_t numIndexes, const size_t startVertex) override;
    void drawInstanced(PrimitiveType primitiveType, const size_t startVertex, const size_t numVertexes, const size_t startInstance, const size_t numInstances) override;
    void drawInstancedIndexed(PrimitiveType primitiveType, const size_t startInstance, const size_t numInstances, const size_t startIndex, const size_t numIndexes) override;
    void dispatchCompute(const unsigned int threadGroupsX, const unsigned int threadGroupsY, const unsigned int threadGroupsZ) override;
    
    void resolveResource(TextureHandle source, TextureHandle destination) override;
    void readDepthBuffer(BufferHandle bufferHandle, int x, int y, int width, int height, Framebuffer framebuffer) override;
    void readColorBuffer(BufferHandle bufferHandle, int x, int y, int width, int height, Framebuffer framebuffer) override;
    
    void * map(BufferHandle bufferHandle, MapMode::EMapMode mapMode, uint32_t * stride = nullptr) override;
    void unmap(BufferHandle bufferHandle) override;
 
    void updateBuffer(BufferHandle bufferHandle, const void* data, size_t size) override;
    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) override;
    void updateSubBuffer(BufferHandle bufferHandle, const size_t offset, const size_t size, const void * data) override;
    
    void copyResource(BufferHandle destinationHandle, BufferHandle sourceHandle) override;
    void copyResource(TextureHandle destinationHandle, TextureHandle sourceHandle) override;
 
    void copyTexture(TextureHandle dstHandle, unsigned dstSub, unsigned dstX, unsigned dstY, unsigned dstZ, TextureHandle sourceHandle, unsigned srcSub) override;
 
    void clearResource(BufferHandle destinationHandle, uint32_t *Values) override {}
    void clearResource(BufferHandle destinationHandle, float *Values) override {}
 
    bool updateDescriptorSet();
 
    bool setupPipeline(PrimitiveType primitiveType);
 
    void updatePipelineState(PrimitiveType primitiveType);
 
    void beginRenderPass();
    void endRenderPass();
 
    void updateConstantBuffers();
 
    InputAssemblerStateCommon * getIAState() override { return &iaState; }
    EffectHandle getCurrentEffect() override { return currentEffect; }
    IEffects * getEffects() { return effects; }
 
    TextureHandle defaultTexture;
    SamplerStateHandle defaultSampler;
 
    RasterizerStateHandle defaultRasterizerState;
    DepthStencilStateHandle defaultDepthStencilState;
    BlendStateHandle defaultBlendState;
 
    EffectsVK * effects;
    BuffersVK * buffers;
    TexturesVK * textures;
    RenderTargetsVK * renderTargets;
 
    EffectHandle currentEffect;
    VkRenderPass renderPass = VK_NULL_HANDLE;
 
    Vulkan::InputAssemblerState iaState;
    Vulkan::PipelineState state;
 
    struct FrameResources
    {
      VkCommandPool commandPool = VK_NULL_HANDLE;
      VkCommandBuffer commandBuffer = VK_NULL_HANDLE;
 
      VkFence fence = VK_NULL_HANDLE;
      VkSemaphore presentSemaphore = VK_NULL_HANDLE;
 
      std::vector<PoolBufferVK *> orhpanedBuffers;
      std::vector<TextureHandle> orphanedTextures;
 
      VkDescriptorPool descriptorPool = VK_NULL_HANDLE;
    };
 
    RenderTargetHandle defaultRenderTarget;
 
    std::vector<VkClearValue> clearValues;
 
    std::array<FrameResources, 3> frameResources;
    FrameResources * currentFrameResources = nullptr;
 
    VkPipelineCache pipelineCache = VK_NULL_HANDLE;
    std::unordered_map<size_t, VkPipeline> PSOs;
 
    VkDescriptorPool descriptorPool = VK_NULL_HANDLE;
    std::vector<VkDescriptorSet> descriptorSets;
 
    uint64_t currentFrame = 0;
 
    struct ShaderState
    {
      EffectSignature effectSignature;
 
      Binding<BufferVK *, 16> CBVs[ShaderType::NumShaderTypes];
      Binding<TextureVK *, 16> SRVs[ShaderType::NumShaderTypes];
      Binding<SamplerVK *, 16> Samplers[ShaderType::NumShaderTypes];
      Binding<BufferVK *, 16> UAVs[ShaderType::NumShaderTypes];
    } shaderState;
 
    private:
      GraphicsDeviceVK * graphicsDevice;
      VkDevice device = VK_NULL_HANDLE;
  };  
}