fungus/Assets/Fungus/Scripts/Commands/Rigidbody2D/StopMotionRigidBody2D.cs

// This code is part of the Fungus library (https://github.com/snozbot/fungus)
// It is released for free under the MIT open source license (https://github.com/snozbot/fungus/blob/master/LICENSE)

using System.Collections;
using System.Collections.Generic;
using UnityEngine;

namespace Fungus
{
    // <summary>
    /// Stop velocity and angular velocity on a Rigidbody2D
    /// </summary>
    [CommandInfo("Rigidbody2D",
                 "StopMotion2D",
                 "Stop velocity and angular velocity on a Rigidbody2D")]
    [AddComponentMenu("")]
    public class StopMotionRigidBody2D : Command
    {
        [SerializeField]
        protected Rigidbody2DData rb;

        public enum Motion
        {
            Velocity,
            AngularVelocity,
            AngularAndLinearVelocity
        }

        [SerializeField]
        protected Motion motionToStop = Motion.AngularAndLinearVelocity;

        public override void OnEnter()
        {
            switch (motionToStop)
            {
                case Motion.Velocity:
                    rb.Value.velocity = Vector2.zero;
                    break;
                case Motion.AngularVelocity:
                    rb.Value.angularVelocity = 0;
                    break;
                case Motion.AngularAndLinearVelocity:
                    rb.Value.angularVelocity = 0;
                    rb.Value.velocity = Vector2.zero;
                    break;
                default:
                    break;
            }

            Continue();
        }

        public override string GetSummary()
        {
            return motionToStop.ToString();
        }

        public override Color GetButtonColor()
        {
            return new Color32(235, 191, 217, 255);
        }

        public override bool HasReference(Variable variable)
        {
            if (rb.rigidbody2DRef == variable)
                return true;

            return false;
        }

    }
}