#pragma once
#include "Containers/Array.h"
#include "Initializer.h"
#include "MinimalEngine.h"
#include "RenderTarget.h"
#include "Resources.h"


namespace Seele {
namespace Gfx {
DECLARE_REF(Window)
DECLARE_REF(Viewport)
DECLARE_REF(ShaderCompiler)
DECLARE_REF(Texture)
DECLARE_REF(Texture2D)
DECLARE_REF(Texture3D)
DECLARE_REF(TextureCube)
DECLARE_REF(DescriptorLayout)
DECLARE_REF(VertexShader)
DECLARE_REF(FragmentShader)
DECLARE_REF(ComputeShader)
DECLARE_REF(TaskShader)
DECLARE_REF(MeshShader)
DECLARE_REF(ShaderBuffer)
DECLARE_REF(VertexBuffer)
DECLARE_REF(IndexBuffer)
DECLARE_REF(UniformBuffer)
DECLARE_REF(PipelineLayout)
DECLARE_REF(GraphicsPipeline)
DECLARE_REF(RayTracingPipeline)
DECLARE_REF(ComputePipeline)
DECLARE_REF(RenderCommand)
DECLARE_REF(ComputeCommand)
DECLARE_REF(OcclusionQuery)
DECLARE_REF(PipelineStatisticsQuery)
DECLARE_REF(TimestampQuery)
DECLARE_REF(BottomLevelAS)
DECLARE_REF(TopLevelAS)
DECLARE_REF(RayGenShader)
DECLARE_REF(AnyHitShader)
DECLARE_REF(ClosestHitShader)
DECLARE_REF(MissShader)
DECLARE_REF(IntersectionShader)
DECLARE_REF(CallableShader)
class Graphics {
  public:
    Graphics();
    virtual ~Graphics();
    virtual void init(GraphicsInitializer initializer) = 0;

    const QueueFamilyMapping getFamilyMapping() const { return queueMapping; }

    PShaderCompiler getShaderCompiler() { return shaderCompiler; }

    virtual OWindow createWindow(const WindowCreateInfo& createInfo) = 0;
    virtual OViewport createViewport(PWindow owner, const ViewportCreateInfo& createInfo) = 0;

    virtual ORenderPass createRenderPass(RenderTargetLayout layout, Array<SubPassDependency> dependencies, URect renderArea,
                                         std::string name = "", Array<uint32> viewMasks = {0}, Array<uint32> correlationMasks = {}) = 0;
    virtual void beginRenderPass(PRenderPass renderPass) = 0;
    virtual void endRenderPass() = 0;
    virtual void waitDeviceIdle() = 0;

    virtual void executeCommands(ORenderCommand commands) = 0;
    virtual void executeCommands(Array<ORenderCommand> commands) = 0;
    virtual void executeCommands(OComputeCommand commands) = 0;
    virtual void executeCommands(Array<OComputeCommand> commands) = 0;

    virtual OTexture2D createTexture2D(const TextureCreateInfo& createInfo) = 0;
    virtual OTexture2DArray createTexture2DArray(const TextureCreateInfo& createInfo) = 0;
    virtual OTexture3D createTexture3D(const TextureCreateInfo& createInfo) = 0;
    virtual OTextureCube createTextureCube(const TextureCreateInfo& createInfo) = 0;
    virtual OUniformBuffer createUniformBuffer(const UniformBufferCreateInfo& bulkData) = 0;
    virtual OShaderBuffer createShaderBuffer(const ShaderBufferCreateInfo& bulkData) = 0;
    virtual OVertexBuffer createVertexBuffer(const VertexBufferCreateInfo& bulkData) = 0;
    virtual OIndexBuffer createIndexBuffer(const IndexBufferCreateInfo& bulkData) = 0;

    virtual ORenderCommand createRenderCommand(const std::string& name = "") = 0;
    virtual OComputeCommand createComputeCommand(const std::string& name = "") = 0;

    virtual void beginShaderCompilation(const ShaderCompilationInfo& compileInfo) = 0;
    virtual OVertexShader createVertexShader(const ShaderCreateInfo& createInfo) = 0;
    virtual OFragmentShader createFragmentShader(const ShaderCreateInfo& createInfo) = 0;
    virtual OComputeShader createComputeShader(const ShaderCreateInfo& createInfo) = 0;
    virtual OMeshShader createMeshShader(const ShaderCreateInfo& createInfo) = 0;
    virtual OTaskShader createTaskShader(const ShaderCreateInfo& createInfo) = 0;
    virtual PGraphicsPipeline createGraphicsPipeline(LegacyPipelineCreateInfo createInfo) = 0;
    virtual PGraphicsPipeline createGraphicsPipeline(MeshPipelineCreateInfo createInfo) = 0;
    virtual PRayTracingPipeline createRayTracingPipeline(RayTracingPipelineCreateInfo createInfo) = 0;
    virtual PComputePipeline createComputePipeline(ComputePipelineCreateInfo createInfo) = 0;
    virtual OSampler createSampler(const SamplerCreateInfo& createInfo) = 0;

    virtual ODescriptorLayout createDescriptorLayout(const std::string& name = "") = 0;
    virtual OPipelineLayout createPipelineLayout(const std::string& name = "", PPipelineLayout baseLayout = nullptr) = 0;

    virtual OVertexInput createVertexInput(VertexInputStateCreateInfo createInfo) = 0;

    virtual Gfx::OOcclusionQuery createOcclusionQuery(const std::string& name = "") = 0;
    virtual Gfx::OPipelineStatisticsQuery createPipelineStatisticsQuery(const std::string& name = "") = 0;
    virtual Gfx::OTimestampQuery createTimestampQuery(uint64 numTimestamps, const std::string& name = "") = 0;

    virtual void beginDebugRegion(const std::string& name) = 0;
    virtual void endDebugRegion() = 0;

    virtual void resolveTexture(PTexture source, PTexture destination) = 0;
    virtual void copyTexture(PTexture src, PTexture dst) = 0;

    virtual void copyBuffer(Gfx::PShaderBuffer src, Gfx::PShaderBuffer dst) = 0;

    constexpr bool supportMeshShading() const { return meshShadingEnabled; }
    constexpr bool supportRayTracing() const { return rayTracingEnabled; }

    // Ray Tracing
    virtual OBottomLevelAS createBottomLevelAccelerationStructure(const BottomLevelASCreateInfo& createInfo) = 0;
    virtual OTopLevelAS createTopLevelAccelerationStructure(const TopLevelASCreateInfo& createInfo) = 0;
    virtual void buildBottomLevelAccelerationStructures(Array<Gfx::PBottomLevelAS> data) = 0;

    virtual ORayGenShader createRayGenShader(const ShaderCreateInfo& createInfo) = 0;
    virtual OAnyHitShader createAnyHitShader(const ShaderCreateInfo& createInfo) = 0;
    virtual OClosestHitShader createClosestHitShader(const ShaderCreateInfo& createInfo) = 0;
    virtual OMissShader createMissShader(const ShaderCreateInfo& createInfo) = 0;
    virtual OIntersectionShader createIntersectionShader(const ShaderCreateInfo& createInfo) = 0;
    virtual OCallableShader createCallableShader(const ShaderCreateInfo& createInfo) = 0;
  protected:
    QueueFamilyMapping queueMapping;
    OShaderCompiler shaderCompiler;
    bool meshShadingEnabled = false;
    bool rayTracingEnabled = false;
    friend class Window;
};
DEFINE_REF(Graphics)
} // namespace Gfx
} // namespace Seele