// Copyright Epic Games, Inc. All Rights Reserved.

#pragma once

#include "NiagaraAsyncGpuTraceProvider.h"
#include "ShaderParameterMacros.h"

class FViewUniformShaderParameters;

#if RHI_RAYTRACING

class FNiagaraAsyncGpuTraceProviderHwrt : public FNiagaraAsyncGpuTraceProvider
{
public:
	static const EProviderType Type;

	FNiagaraAsyncGpuTraceProviderHwrt(EShaderPlatform InShaderPlatform, FNiagaraGpuComputeDispatchInterface* Dispatcher);

	static bool IsSupported();
	virtual void PostRenderOpaque(FRHICommandList& RHICmdList, TConstStridedView<FSceneView> Views, TUniformBufferRef<FSceneUniformParameters> SceneUniformBufferRHI, FCollisionGroupHashMap* CollisionGroupHash) override;
	virtual bool IsAvailable() const override;
	virtual void IssueTraces(FRHICommandList& RHICmdList, const FDispatchRequest& Request, TUniformBufferRef<FSceneUniformParameters> SceneUniformBufferRHI, FCollisionGroupHashMap* CollisionGroupHash) override;
	virtual void Reset() override;
	virtual EProviderType GetType() const override { return Type; }

private:
	FRayTracingPipelineState* RayTracingPipelineState = nullptr;
	FRHIShaderBindingTable* RayTracingSBT = nullptr; // TODO: Hold reference?
	FRHIShaderResourceView* TLASSRV = nullptr;
	TUniformBufferRef<FViewUniformShaderParameters> ViewUniformBuffer;
};

#endif