Unreal Engine 4 (UE4) その2 Advent Calendar 2017の12日目です

やりたいこと

オブジェクトの可視判定を正確に取りたい！
ゲーム用の判定であればコリジョンを設定してRayを飛ばせば完了だが、もっと厳密な判定をしたい！
ひらひら舞うスカートから下着が見えていないか、厳密にチェックしたい！

ということで、ピクセル単位での可視をリアルタイムで判断できるようなものを実装してみます。
利用しているのはUE4.18.1です。

実装したもの　↓
ユニティちゃんの画面描画ピクセル数と、座標平均値のマーカがが表示されています。

こはく先輩の描画ピクセル数を数えた pic.twitter.com/oWY3SX3Cn7
— はるべえ (@ruyo_h) 2017年12月12日

※伝わらないですがリアルタイムで滑らかに動作しています。

ざっくり方針

対象オブジェクトのCustomDepthへ書き込みを有効にして、そのバッファをComputeShaderでチェックします。
CPUでも実現できると思いますが、勉強がてらShaderでやってみます。あとCPUだとなんとなく処理が重そうな気がします。

概要は以下です。

Shaderプラグインのサンプルを導入して、ComputeShaderのタスクを呼び出せるようにする
ComputeShaderからCustomDepthを参照できるようにする
バッファをチェックする

UE4ShaderPluginDemoの導入

こちらを利用させて頂きました。
https://github.com/Temaran/UE4ShaderPluginDemo

本家の更新は止まっていますが、PullRequestにシェーダプラグイン対応(UE4.17以降)を上げている方がいます。ありがたく取り込みます。
ComputeShaderUsageExample.hにある、ExecuteComputeShader を呼べば入り口は完成です。
Blueprintから呼べるようノードを追加しておきます。

UE4ShaderPluginDemoからの変更

改造していきます。

Depthを参照できるようにする

主にcpp側の対応です。
PixelShaderサンプルをコピペ元にして、TextureParameter、TextureParameterSRV の処理を追加します。

可視判定に利用するバッファは以下のコードで参照できます。
また、ComputeShader.Build.csにはRendererへの依存リストも追加します。

※今回の実装では、Depth,CustomDepthあたりしか参照できません。
深追いしてないですが、GBufferを参照するにはComputeShaderの実行タイミングを変更するか、なんらか前フレームの絵を保存しておく必要がありそうです。

ComputeShaderUsageExample.cpp

// 略
FSceneRenderTargets& SceneContext = FSceneRenderTargets::Get(RHICmdList);
if (SceneContext.GBufferA) {
    TextureParameter = (const FTexture2DRHIRef&)(SceneContext.GBufferA->GetRenderTargetItem().TargetableTexture);
}
if (SceneContext.SceneDepthZ) {
    TextureParameter = (const FTexture2DRHIRef&)(SceneContext.SceneDepthZ->GetRenderTargetItem().TargetableTexture);
}
if (SceneContext.CustomDepth) {
    TextureParameter = (const FTexture2DRHIRef&)(SceneContext.CustomDepth->GetRenderTargetItem().TargetableTexture);
}

if (NULL != TextureParameterSRV)
{
    TextureParameterSRV.SafeRelease();
    TextureParameterSRV = NULL;
}
TextureParameterSRV = RHICreateShaderResourceView(TextureParameter, 0);
ComputeShader->SetSurfaces(RHICmdList, TextureParameterSRV);
// 略

ComputeShader.Build.cs

    // -- 略
    PublicDependencyModuleNames.AddRange(
        new string[]
        {
            "Core",
            "CoreUObject",
            "Engine",
            "RenderCore",
            "ShaderCore",
            "RHI",
            "Renderer", // <-- 追加
        }
    );
    // -- 略

ピクセルカウンタの追加

主にusf側の対応です。
元のComputeShaderサンプルはOutputSurfaceに画像を書き込むプログラムです。
OutputSurfaceはUAVなので、今回はこれをそのままピクセル数の記録先にします。
CustomDepthが書き込まれていたらカウンタ+1します。（SceneDepthより奥にあるピクセルはカウントしません）

ComputeShaderExample.usf

    float sizeX, sizeY;
    TextureParameter.GetDimensions(sizeX, sizeY);

    float2 iResolution = float2(sizeX, sizeY);
    float2 uv = (ThreadId.xy / iResolution.xy) - 0.5;

    float f1 = (float)TextureParameterDepth.Load(int3(sizeX * uv.x, sizeY * uv.y, 0));
    float f2 = (float)TextureParameterCustomDepth.Load(int3(sizeX * uv.x, sizeY * uv.y, 0));

    if (f1>0 && f1<=f2){
        float2 v = {0,0};
        uint temp;
        InterlockedAdd(OutputSurface[v], 1U, temp);
    }

これをCPU側で取り出して、BlueprintなりUMGに渡して完成です。

最後に

UE4ShaderPluginDemo はシェーダまわりの一通りの実装が入っており、とても参考になりました。
解説を省略しましたが、実装したものは描画ピクセル座標の平均値にマーカーが出るようにしてあります。
OutputSurfaceに情報を埋め込めば他にも色々できそうです。
オブジェクトの可視判定としては、特定のマテリアルやShaderModelIDで判別できれば実用的かなと思うところです。

以上です。お疲れ様でした。

この作品はユニティちゃんライセンス条項の元に提供されています

シェーダでピクセルを数える