raytracingcontext.h

Go to the documentation of this file.

#pragma once
//------------------------------------------------------------------------------
//------------------------------------------------------------------------------
#include "graphics/graphicscontext.h"
#include "coregraphics/accelerationstructure.h"
#include "coregraphics/buffer.h"
#include "coregraphics/shader.h"
#include "materials/gpulang/materialtemplatesgpulang.h"
#include "materials/shaderconfig.h"
 
namespace Raytracing
{
 
struct RaytracingSetupSettings
{
    SizeT maxNumAllowedInstances;
};
 
enum ObjectType
{
    BRDFObject,
    BSDFObject,
    GLTFObject,
    TerrainObject,
    ParticleObject,
    NumObjectTypes
};
 
static constexpr const char* HitShaderPaths[NumObjectTypes] =
{
    "shd:raytracing/shaders/brdfhit.gplb",
    "shd:raytracing/shaders/bsdfhit.gplb",
    "shd:raytracing/shaders/gltfhit.gplb",
    "shd:raytracing/shaders/terrainhit.gplb",
    "shd:raytracing/shaders/particlehit.gplb"
};
 
enum UpdateType
{
    Dynamic,
    Static
};
 
class RaytracingContext : public Graphics::GraphicsContext
{
    __DeclareContext();
public:
    RaytracingContext();
    ~RaytracingContext();

    static void Create(const RaytracingSetupSettings& settings);
    static void Discard();
    static const CoreGraphics::ShaderProgramId* GetHitPrograms();

    static void SetupModel(const Graphics::GraphicsEntityId id, CoreGraphics::BlasInstanceFlags flags, uchar mask);
    static void SetupMesh(
        const Graphics::GraphicsEntityId id
        , const UpdateType objectType
        , const CoreGraphics::VertexComponent::Format format
        , const CoreGraphics::IndexType::Code indexType
        , const CoreGraphics::VertexAlloc& vertices
        , const CoreGraphics::VertexAlloc& indices
        , const CoreGraphics::PrimitiveGroup& patchPrimGroup
        , const size_t vertexOffsetStride
        , const size_t patchVertexStride
        , const Util::Array<Math::mat4> transforms
        , const uint materialTableOffset
        , const MaterialTemplatesGPULang::MaterialProperties shader
        , const CoreGraphics::VertexLayoutType vertexLayout
    );
    static void InvalidateBLAS(const Graphics::GraphicsEntityId id);

    static void ReconstructTopLevelAcceleration(const Graphics::FrameContext& ctx);
    static void RenderUI(const Graphics::FrameContext& ctx);
    static void UpdateTransforms(const Graphics::FrameContext& ctx);
    static void WaitForJobs(const Graphics::FrameContext& ctx);
    static void UpdateResources(const Graphics::FrameContext& ctx);

    static CoreGraphics::ResourceTableId GetLightGridResourceTable(IndexT bufferIndex);
    static CoreGraphics::TlasId GetTLAS(const IndexT bufferIndex);
    static CoreGraphics::BufferId GetObjectBindingBuffer();
    static CoreGraphics::ResourceTableId GetRaytracingTable(const IndexT bufferIndex);
 
#ifndef PUBLIC_DEBUG    
    static void OnRenderDebug(uint32_t flags);
#endif
 
private:

    static Graphics::ContextEntityId Alloc();
    static void Dealloc(Graphics::ContextEntityId id);
 
 
 
    enum
    {
        Raytracing_Allocation,
        Raytracing_UpdateType,
        Raytracing_NumStructures,
        Raytracing_Blases
    };
    typedef Ids::IdAllocator<
        Memory::RangeAllocation,
        Raytracing::UpdateType,
        uint,
        Util::FixedArray<CoreGraphics::BlasId>
    > RaytracingAllocator;
    static RaytracingAllocator raytracingContextAllocator;
};
 
//------------------------------------------------------------------------------
inline Graphics::ContextEntityId
RaytracingContext::Alloc()
{
    return raytracingContextAllocator.Alloc();
}
} // namespace Raytracing