bool RayTracer::VInit(DXDriver* _driver)
{
if (m_pPrimRayStage || m_pIntersectionStage || m_pColorStage)
{
PrintError(AT, "shader stages already created");
return false;
}
m_pPrimRayStage = VNEW ComputeShader();
if (!m_pPrimRayStage->Init(L"Assets/Shaders/RTPrimaryRays.hlsl", _driver->GetDevice()))
{
PrintError(AT, "failed to create compute shader stage");
return false;
}
m_pIntersectionStage = VNEW ComputeShader();
if (!m_pIntersectionStage->Init(L"Assets/Shaders/RTIntersect.hlsl", _driver->GetDevice()))
{
PrintError(AT, "failed to create compute shader stage");
return false;
}
m_pColorStage = VNEW ComputeShader();
//if (!m_pColorStage->Init(L"Assets/Shaders/RCSinglePass.hlsl", _driver->GetDevice()))
if (!m_pColorStage->Init(L"Assets/Shaders/RTComputeColor.hlsl", _driver->GetDevice()))
{
PrintError(AT, "failed to create compute shader stage");
return false;
}
Resolution res = g_pApplication->VGetResolution();
/* UAVS */
m_pRays = VNEW D3DBuffer();
if (!m_pRays->Init(_driver->GetDevice(), BT_STRUCTURED, BB_UAV, res.height * res.width, sizeof(Ray)))
return false;
m_pIntersections = VNEW D3DBuffer();
if (!m_pIntersections->Init(_driver->GetDevice(), BT_STRUCTURED, BB_UAV, res.height *res.width, sizeof(Intersection)))
return false;
m_pColorAccumulation = VNEW D3DBuffer();
if (!m_pColorAccumulation->Init(_driver->GetDevice(), BT_STRUCTURED, BB_UAV, res.height * res.width, sizeof(FVEC4)))
return false;
return true;
}
/*-------------------------------------------
main関数
--------------------------------------------*/
int wmain(int argc, WCHAR* argv[])
{
HRESULT hr;
// ロケールを設定
_wsetlocale(LC_ALL, L"Japanese");
// **********************************************************
// Direct3D11デバイスの作成
hr = CreateDevice();
// **********************************************************
// コンピュート・シェーダの作成
if (SUCCEEDED(hr))
hr = CreateShader();
// **********************************************************
// 定数バッファの作成
if (SUCCEEDED(hr))
hr = CreateCBuffer();
// **********************************************************
// リソースの作成
if (SUCCEEDED(hr))
hr = CreateResource();
// **********************************************************
// シェーダ リソース ビューの作成
if (SUCCEEDED(hr))
hr = CreateSRV();
// **********************************************************
// アンオーダード・アクセス・ビューの作成
if (SUCCEEDED(hr))
hr = CreateUAV();
// **********************************************************
// コンピュート・シェーダを使った演算
if (SUCCEEDED(hr))
ComputeShader();
// **********************************************************
// 開放
SAFE_RELEASE(g_pUAV[1]);
SAFE_RELEASE(g_pUAV[0]);
SAFE_RELEASE(g_pSRV[1]);
SAFE_RELEASE(g_pSRV[0]);
SAFE_RELEASE(g_pReadBackBuffer);
SAFE_RELEASE(g_pBuffer[1]);
SAFE_RELEASE(g_pBuffer[0]);
SAFE_RELEASE(g_pCBuffer);
SAFE_RELEASE(g_pComputeShader);
SAFE_RELEASE(g_pImmediateContext);
SAFE_RELEASE(g_pD3DDevice);
return 0;
}
void FRCPassPostProcessHistogram::Process(FRenderingCompositePassContext& Context)
{
SCOPED_DRAW_EVENT(PostProcessHistogram, DEC_SCENE_ITEMS);
const FPooledRenderTargetDesc* InputDesc = GetInputDesc(ePId_Input0);
if(!InputDesc)
{
// input is not hooked up correctly
return;
}
const FSceneView& View = Context.View;
const FSceneViewFamily& ViewFamily = *(View.Family);
FIntPoint SrcSize = InputDesc->Extent;
FIntRect DestRect = View.ViewRect;
const FSceneRenderTargetItem& DestRenderTarget = PassOutputs[0].RequestSurface(Context);
TShaderMapRef<FPostProcessHistogramCS> ComputeShader(GetGlobalShaderMap());
RHISetComputeShader(ComputeShader->GetComputeShader());
RHISetRenderTarget(FTextureRHIRef(), FTextureRHIRef());
// set destination
check(DestRenderTarget.UAV);
RHISetUAVParameter(ComputeShader->GetComputeShader(), ComputeShader->HistogramRWTexture.GetBaseIndex(), DestRenderTarget.UAV);
// we currently assume the input is half res, one full res pixel less to avoid getting bilinear filtered input
FIntPoint GatherExtent = (DestRect.Size() - FIntPoint(1, 1)) / 2;
FIntPoint ThreadGroupCountValue = ComputeThreadGroupCount(GatherExtent);
ComputeShader->SetCS(Context, ThreadGroupCountValue, (DestRect.Min + FIntPoint(1, 1)) / 2, GatherExtent);
DispatchComputeShader(*ComputeShader, ThreadGroupCountValue.X, ThreadGroupCountValue.Y, 1);
// un-set destination
RHISetUAVParameter(ComputeShader->GetComputeShader(), ComputeShader->HistogramRWTexture.GetBaseIndex(), NULL);
RHICopyToResolveTarget(DestRenderTarget.TargetableTexture, DestRenderTarget.ShaderResourceTexture, false, FResolveParams());
}
void FRCPassPostProcessHistogram::Process(FRenderingCompositePassContext& Context)
{
SCOPED_DRAW_EVENT(Context.RHICmdList, PostProcessHistogram);
const FPooledRenderTargetDesc* InputDesc = GetInputDesc(ePId_Input0);
if(!InputDesc)
{
// input is not hooked up correctly
return;
}
const FSceneView& View = Context.View;
const FSceneViewFamily& ViewFamily = *(View.Family);
FIntPoint SrcSize = InputDesc->Extent;
FIntRect DestRect = View.ViewRect;
const FSceneRenderTargetItem& DestRenderTarget = PassOutputs[0].RequestSurface(Context);
TShaderMapRef<FPostProcessHistogramCS> ComputeShader(Context.GetShaderMap());
SetRenderTarget(Context.RHICmdList, FTextureRHIRef(), FTextureRHIRef());
Context.RHICmdList.SetComputeShader(ComputeShader->GetComputeShader());
// set destination
check(DestRenderTarget.UAV);
Context.RHICmdList.TransitionResource(EResourceTransitionAccess::ERWBarrier, EResourceTransitionPipeline::EGfxToCompute, DestRenderTarget.UAV);
Context.RHICmdList.SetUAVParameter(ComputeShader->GetComputeShader(), ComputeShader->HistogramRWTexture.GetBaseIndex(), DestRenderTarget.UAV);
FIntPoint GatherExtent = ComputeGatherExtent(View);
FIntPoint ThreadGroupCountValue = ComputeThreadGroupCount(GatherExtent);
ComputeShader->SetCS(Context.RHICmdList, Context, ThreadGroupCountValue, (DestRect.Min + FIntPoint(1, 1)) / 2, GatherExtent);
DispatchComputeShader(Context.RHICmdList, *ComputeShader, ThreadGroupCountValue.X, ThreadGroupCountValue.Y, 1);
// un-set destination
Context.RHICmdList.SetUAVParameter(ComputeShader->GetComputeShader(), ComputeShader->HistogramRWTexture.GetBaseIndex(), NULL);
Context.RHICmdList.TransitionResource(EResourceTransitionAccess::EReadable, EResourceTransitionPipeline::EComputeToGfx, DestRenderTarget.UAV);
ensureMsgf(DestRenderTarget.TargetableTexture == DestRenderTarget.ShaderResourceTexture, TEXT("%s should be resolved to a separate SRV"), *DestRenderTarget.TargetableTexture->GetName().ToString());
}
void FRCPassPostProcessVelocityFlatten::Process(FRenderingCompositePassContext& Context)
{
SCOPED_DRAW_EVENT(Context.RHICmdList, PostProcessVelocityFlatten);
const FPooledRenderTargetDesc* InputDesc = GetInputDesc(ePId_Input0);
if(!InputDesc)
{
// input is not hooked up correctly
return;
}
const FSceneView& View = Context.View;
const FSceneViewFamily& ViewFamily = *(View.Family);
const FSceneRenderTargetItem& DestRenderTarget0 = PassOutputs[0].RequestSurface(Context);
const FSceneRenderTargetItem& DestRenderTarget1 = PassOutputs[1].RequestSurface(Context);
//const FSceneRenderTargetItem& DestRenderTarget2 = PassOutputs[2].RequestSurface(Context);
TShaderMapRef< FPostProcessVelocityFlattenCS > ComputeShader( Context.GetShaderMap() );
Context.RHICmdList.SetComputeShader(ComputeShader->GetComputeShader());
SetRenderTarget(Context.RHICmdList, FTextureRHIRef(), FTextureRHIRef());
// set destination
Context.RHICmdList.SetUAVParameter( ComputeShader->GetComputeShader(), ComputeShader->OutVelocityFlat.GetBaseIndex(), DestRenderTarget0.UAV );
//Context.RHICmdList.SetUAVParameter( ComputeShader->GetComputeShader(), ComputeShader->OutPackedVelocityDepth.GetBaseIndex(), DestRenderTarget1.UAV );
Context.RHICmdList.SetUAVParameter( ComputeShader->GetComputeShader(), ComputeShader->OutMaxTileVelocity.GetBaseIndex(), DestRenderTarget1.UAV );
ComputeShader->SetCS(Context.RHICmdList, Context, View );
FIntPoint ThreadGroupCountValue = ComputeThreadGroupCount( View.ViewRect.Size() );
DispatchComputeShader(Context.RHICmdList, *ComputeShader, ThreadGroupCountValue.X, ThreadGroupCountValue.Y, 1);
// un-set destination
Context.RHICmdList.SetUAVParameter( ComputeShader->GetComputeShader(), ComputeShader->OutVelocityFlat.GetBaseIndex(), NULL );
//Context.RHICmdList.SetUAVParameter( ComputeShader->GetComputeShader(), ComputeShader->OutPackedVelocityDepth.GetBaseIndex(), NULL );
Context.RHICmdList.SetUAVParameter( ComputeShader->GetComputeShader(), ComputeShader->OutMaxTileVelocity.GetBaseIndex(), NULL );
Context.RHICmdList.CopyToResolveTarget(DestRenderTarget0.TargetableTexture, DestRenderTarget0.ShaderResourceTexture, false, FResolveParams());
Context.RHICmdList.CopyToResolveTarget(DestRenderTarget1.TargetableTexture, DestRenderTarget1.ShaderResourceTexture, false, FResolveParams());
//Context.RHICmdList.CopyToResolveTarget(DestRenderTarget1.TargetableTexture, DestRenderTarget2.ShaderResourceTexture, false, FResolveParams());
}
DefaultResources::DefaultResources(DevicePtr dptr) :
_spirv(std::make_unique<SpirVData[]>(spirv_count)),
_computes(std::make_unique<ComputeProgram[]>(compute_count)),
_materials(std::make_unique<AssetPtr<Material>[]>(material_count)) {
for(usize i = 0; i != spirv_count; ++i) {
_spirv[i] = load_spirv(spirv_names[i]);
}
for(usize i = 0; i != compute_count; ++i) {
_computes[i] = ComputeProgram(ComputeShader(dptr, _spirv[usize(compute_spirvs[i])]));
}
for(usize i = 0; i != material_count; ++i) {
const auto& data = material_datas[i];
_materials[i] = make_asset<Material>(dptr, MaterialData()
.set_frag_data(_spirv[data.frag])
.set_vert_data(_spirv[data.vert])
.set_depth_tested(data.depth_tested)
.set_culled(data.culled)
.set_blended(data.blended)
);
}
}
void FRCPassPostProcessDeferredDecals::DecodeRTWriteMask(FRenderingCompositePassContext& Context)
{
// @todo: get these values from the RHI?
const uint32 MaskTileSizeX = 8;
const uint32 MaskTileSizeY = 8;
check(GSupportsRenderTargetWriteMask);
FRHICommandListImmediate& RHICmdList = Context.RHICmdList;
FSceneRenderTargets& SceneContext = FSceneRenderTargets::Get(RHICmdList);
FTextureRHIRef DBufferTex = SceneContext.DBufferA->GetRenderTargetItem().TargetableTexture;
FIntPoint RTWriteMaskDims(
FMath::DivideAndRoundUp(DBufferTex->GetTexture2D()->GetSizeX(), MaskTileSizeX),
FMath::DivideAndRoundUp(DBufferTex->GetTexture2D()->GetSizeY(), MaskTileSizeY));
// allocate the DBufferMask from the render target pool.
FPooledRenderTargetDesc MaskDesc(FPooledRenderTargetDesc::Create2DDesc(RTWriteMaskDims,
PF_R8_UINT,
FClearValueBinding::White,
TexCreate_None,
TexCreate_UAV | TexCreate_RenderTargetable,
false));
GRenderTargetPool.FindFreeElement(Context.RHICmdList, MaskDesc, SceneContext.DBufferMask, TEXT("DBufferMask"));
FIntRect ViewRect(0, 0, DBufferTex->GetTexture2D()->GetSizeX(), DBufferTex->GetTexture2D()->GetSizeY());
TShaderMapRef< FRTWriteMaskDecodeCS > ComputeShader(Context.GetShaderMap());
SetRenderTarget(Context.RHICmdList, FTextureRHIRef(), FTextureRHIRef());
Context.SetViewportAndCallRHI(ViewRect);
Context.RHICmdList.SetComputeShader(ComputeShader->GetComputeShader());
// set destination
Context.RHICmdList.SetUAVParameter(ComputeShader->GetComputeShader(), ComputeShader->OutCombinedRTWriteMask.GetBaseIndex(), SceneContext.DBufferMask->GetRenderTargetItem().UAV);
ComputeShader->SetCS(Context.RHICmdList, Context, Context.View, RTWriteMaskDims);
RHICmdList.TransitionResource(EResourceTransitionAccess::EWritable, EResourceTransitionPipeline::EGfxToCompute, SceneContext.DBufferMask->GetRenderTargetItem().UAV);
{
SCOPED_DRAW_EVENTF(Context.RHICmdList, DeferredDecals, TEXT("Combine DBuffer RTWriteMasks"));
FIntPoint ThreadGroupCountValue(
FMath::DivideAndRoundUp((uint32)RTWriteMaskDims.X, FRTWriteMaskDecodeCS::ThreadGroupSizeX),
FMath::DivideAndRoundUp((uint32)RTWriteMaskDims.Y, FRTWriteMaskDecodeCS::ThreadGroupSizeY));
DispatchComputeShader(Context.RHICmdList, *ComputeShader, ThreadGroupCountValue.X, ThreadGroupCountValue.Y, 1);
}
// void FD3D11DynamicRHI::RHIGraphicsWaitOnAsyncComputeJob( uint32 FenceIndex )
Context.RHICmdList.FlushComputeShaderCache();
RHICmdList.TransitionResource(EResourceTransitionAccess::EReadable, EResourceTransitionPipeline::EComputeToGfx, SceneContext.DBufferMask->GetRenderTargetItem().UAV);
RHICmdList.TransitionResource(EResourceTransitionAccess::EMetaData, SceneContext.DBufferA->GetRenderTargetItem().TargetableTexture);
RHICmdList.TransitionResource(EResourceTransitionAccess::EMetaData, SceneContext.DBufferB->GetRenderTargetItem().TargetableTexture);
RHICmdList.TransitionResource(EResourceTransitionAccess::EMetaData, SceneContext.DBufferC->GetRenderTargetItem().TargetableTexture);
// un-set destination
Context.RHICmdList.SetUAVParameter(ComputeShader->GetComputeShader(), ComputeShader->OutCombinedRTWriteMask.GetBaseIndex(), NULL);
}
