com.unity.perception/com.unity.perception/Runtime/GroundTruth/RenderPasses/HdrpPasses/SemanticSegmentationPass.cs

#if HDRP_PRESENT
using System;
using UnityEngine.Rendering;
using UnityEngine.Rendering.HighDefinition;

namespace UnityEngine.Perception.GroundTruth
{
    /// <summary>
    /// Custom Pass which renders labeled images where each object with a Labeling component is drawn with the value
    /// specified by the given LabelingConfiguration.
    /// </summary>
    public class SemanticSegmentationPass : CustomPass
    {
        public RenderTexture targetTexture;
        public SemanticSegmentationLabelConfig semanticSegmentationLabelConfig;
        public Camera targetCamera;

        SemanticSegmentationCrossPipelinePass m_SemanticSegmentationCrossPipelinePass;

        public SemanticSegmentationPass(Camera targetCamera, RenderTexture targetTexture, SemanticSegmentationLabelConfig semanticSegmentationLabelConfig)
        {
            this.targetTexture = targetTexture;
            this.semanticSegmentationLabelConfig = semanticSegmentationLabelConfig;
            this.targetCamera = targetCamera;
            EnsureInit();
        }

        void EnsureInit()
        {
            if (m_SemanticSegmentationCrossPipelinePass == null)
            {
                m_SemanticSegmentationCrossPipelinePass = new SemanticSegmentationCrossPipelinePass(targetCamera, semanticSegmentationLabelConfig);
                m_SemanticSegmentationCrossPipelinePass.EnsureActivated();
            }
        }

        protected override void Setup(ScriptableRenderContext renderContext, CommandBuffer cmd)
        {
            EnsureInit();
            m_SemanticSegmentationCrossPipelinePass.Setup();
        }

        //overrides obsolete member in HDRP on 2020.1+. Re-address when removing 2019.4 support or the API is dropped
#if HDRP_9_OR_NEWER
        protected override void Execute(CustomPassContext ctx)
        {
            ScriptableRenderContext renderContext = ctx.renderContext;
            var cmd = ctx.cmd;
            var hdCamera = ctx.hdCamera;
            var cullingResult = ctx.cullingResults;
#else
        protected override void Execute(ScriptableRenderContext renderContext, CommandBuffer cmd, HDCamera hdCamera, CullingResults cullingResult)
        {
#endif
            if (targetCamera != hdCamera.camera)
                return;

            CoreUtils.SetRenderTarget(cmd, targetTexture);
            m_SemanticSegmentationCrossPipelinePass.Execute(renderContext, cmd, hdCamera.camera, cullingResult);
        }
    }
}
#endif