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

using System;
using System.Collections.Generic;

using UnityEngine;
using UnityEditor.Experimental.VFX.Utility;


namespace UnityEditor.VFX.Operator
{
    [VFXHelpURL("Operator-SamplePointCache")]
    [VFXInfo(category = "Sampling", experimental = true)]
    class SamplePointCache : VFXOperator
    {
        override public string name { get { return "Sample Point Cache"; } }

        [VFXSetting, Tooltip("Specifies the Point Cache Asset to sample from.")]
        public PointCacheAsset asset;

        public class InputProperties
        {
            [Tooltip("Sets the index of the point to sample.")]
            public uint index = 0u;
        }

        [VFXSetting, SerializeField, Tooltip("Specifies how Unity handles the sample when the particleId is out of the point cache bounds.")]
        private VFXOperatorUtility.SequentialAddressingMode mode = VFXOperatorUtility.SequentialAddressingMode.Wrap;

        private static Type GetOutputType(Texture2D surface)
        {
            switch (surface.format)
            {
                case TextureFormat.Alpha8:
                    return typeof(float);
                case TextureFormat.RGBA32:
                    return typeof(Vector3);
                case TextureFormat.RHalf:
                    return typeof(float);
                case TextureFormat.RGBAHalf:
                    return typeof(Vector3);

                default:
                    throw new InvalidOperationException("Unsupported texture format  : " + surface.format);
            }
        }

        protected override IEnumerable<VFXPropertyWithValue> outputProperties
        {
            get
            {
                if (!object.ReferenceEquals(asset, null))
                {
                    if (asset == null)
                        asset = EditorUtility.InstanceIDToObject(asset.GetInstanceID()) as PointCacheAsset;
                    yield return new VFXPropertyWithValue(new VFXProperty(typeof(uint), "Point Count"));
                    foreach (var surface in asset.surfaces)
                        yield return new VFXPropertyWithValue(new VFXProperty(GetOutputType(surface), surface.name));
                }
            }
        }

        protected override sealed VFXExpression[] BuildExpression(VFXExpression[] inputExpression)
        {
            VFXExpression[] expressions = new VFXExpression[asset.surfaces.Length + 1];
            expressions[0] = VFXValue.Constant((uint)asset.PointCount);

            for (int i = 0; i < asset.surfaces.Length; i++)
            {
                var surfaceExpr = VFXValue.Constant(asset.surfaces[i]);
                VFXExpression height = new VFXExpressionTextureHeight(surfaceExpr);
                VFXExpression width = new VFXExpressionTextureWidth(surfaceExpr);
                VFXExpression u_index = VFXOperatorUtility.ApplyAddressingMode(inputExpression[0], new VFXExpressionMin(height * width, expressions[0]), mode);
                VFXExpression y = u_index / width;
                VFXExpression x = u_index - (y * width);

                Type outputType = GetOutputType(asset.surfaces[i]);
                var type = typeof(VFXExpressionSampleAttributeMap<>).MakeGenericType(outputType);
                var outputExpr = Activator.CreateInstance(type, new object[] { surfaceExpr, x, y });

                expressions[i + 1] = (VFXExpression)outputExpr;
            }

            return expressions;
        }
    }
}