UnityCACAO-HDRP17/com.unity.visualeffectgraph/Editor/Models/Operators/Implementations/Normalize.cs

using System.Collections.Generic;
using UnityEngine;

namespace UnityEditor.VFX.Operator
{
    class SafeNormalizationVariantProvider : VariantProvider
    {
        public override IEnumerable<Variant> GetVariants()
        {
            yield return new Variant("Safe Normalize", "Math/Vector", typeof(Normalize), new[] {new KeyValuePair<string, object>("safeNormalize", true)});
            yield return new Variant("Normalize", "Math/Vector", typeof(Normalize), new[] {new KeyValuePair<string, object>("safeNormalize", false)});
        }
    }

    [VFXHelpURL("Operator-Normalize")]
    [VFXInfo(variantProvider = typeof(SafeNormalizationVariantProvider))]
    class Normalize : VFXOperatorNumericUniform
    {
        public class InputProperties
        {
            public Vector3 x = Vector3.one;
        }

        [VFXSetting(VFXSettingAttribute.VisibleFlags.InInspector), SerializeField, Tooltip("Specifies if the operator should check if the vector to be normalized is a vero vector.")]
        bool safeNormalize = false;


        protected override sealed string operatorName { get { return safeNormalize ? "Safe Normalize" : "Normalize"; } }

        protected override sealed VFXExpression[] BuildExpression(VFXExpression[] inputExpression)
        {
            if (safeNormalize)
                return new[] { VFXOperatorUtility.SafeNormalize(inputExpression[0]) };
            else
                return new[] { VFXOperatorUtility.Normalize(inputExpression[0]) };
        }

        protected override sealed ValidTypeRule typeFilter
        {
            get
            {
                return ValidTypeRule.allowVectorType;
            }
        }
    }
}