VisualEffectGraph-Samples_backup/Assets/Samples/Visual Effect Graph/10.2.0/OutputEvent Helpers/Runtime/VFXOutputEventRigidBody.cs

#if VFX_OUTPUTEVENT_PHYSICS
using UnityEngine.Events;

namespace UnityEngine.VFX.Utility
{
    [ExecuteAlways]
    [RequireComponent(typeof(VisualEffect))]
    class VFXOutputEventRigidBody : VFXOutputEventAbstractHandler
    {
        public override bool canExecuteInEditor => false;
        public enum Space
        {
            Local,
            World
        }

        static readonly int k_Position = Shader.PropertyToID("position");
        static readonly int k_Size = Shader.PropertyToID("size");
        static readonly int k_Velocity = Shader.PropertyToID("velocity");

        [Tooltip("The Rigid body to apply a force on.")]
        public Rigidbody rigidBody;
        [Tooltip("The Space VFX Attributes values are expressed.")]
        public Space attributeSpace;

        public enum RigidBodyEventType
        {
            Impulse,
            Explosion,
            VelocityChange
        }

        [Tooltip("Type of Instantaneous Force to apply on the RigidBody upon event:\n - Impulse using the Velocity attribute \n - Explosion at given Position attribute, using the Size for radius and the magnitude of Velocity Attribute for intensity\n - Velocity Change using Velocity Attribute")]
        public RigidBodyEventType eventType;

        public override void OnVFXOutputEvent(VFXEventAttribute eventAttribute)
        {
            if (rigidBody == null)
                return;

            var position = eventAttribute.GetVector3(k_Position);
            var size = eventAttribute.GetFloat(k_Size);
            var velocity = eventAttribute.GetVector3(k_Velocity);
            if (attributeSpace == Space.Local)
            {
                position = transform.localToWorldMatrix.MultiplyPoint(position);
                velocity = transform.localToWorldMatrix.MultiplyVector(velocity);
                // We assume that the size is bound to the X component of the transform scale
                // and that the transform is uniform.
                size = transform.localToWorldMatrix.MultiplyVector(Vector3.right * size).magnitude;
            }

            switch (eventType)
            {
                case RigidBodyEventType.Impulse:
                    rigidBody.AddForce(velocity, ForceMode.Impulse);
                    break;
                case RigidBodyEventType.Explosion:
                    rigidBody.AddExplosionForce(velocity.magnitude,  position, size);
                    break;
                case RigidBodyEventType.VelocityChange:
                    rigidBody.AddForce(velocity, ForceMode.VelocityChange);
                    break;
                default:
                    break;
            }
        }
    }
}
#endif