com.unity.perception/com.unity.perception/Runtime/GroundTruth/LensDistortionPass.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 LensDistortionPass : CustomPass
    {
        public RenderTexture targetTexture;
        public Camera targetCamera;

        internal LensDistortionCrossPipelinePass m_LensDistortionCrossPipelinePass;

        public LensDistortionPass(Camera targetCamera, RenderTexture targetTexture)
        {
            this.targetTexture = targetTexture;
            this.targetCamera = targetCamera;
            EnsureInit();
        }

        public void EnsureInit()
        {
            if (m_LensDistortionCrossPipelinePass == null)
            {
                m_LensDistortionCrossPipelinePass = new LensDistortionCrossPipelinePass(targetCamera, targetTexture);
                m_LensDistortionCrossPipelinePass.EnsureActivated();
            }
        }

        public LensDistortionPass()
        {
            //
        }

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

        //protected override void Execute(ScriptableRenderContext renderContext, CommandBuffer cmd, HDCamera hdCamera, CullingResults cullingResult)
        protected override void Execute(CustomPassContext customPassContext)
        {
            CoreUtils.SetRenderTarget(customPassContext.cmd, targetTexture);
            m_LensDistortionCrossPipelinePass.Execute(customPassContext.renderContext, customPassContext.cmd, customPassContext.hdCamera.camera, customPassContext.cullingResults);
        }
    }
}
#endif