fungus/Assets/Fungus/Scripts/Commands/Rigidbody2D/AddTorque2D.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>
    /// Add Torque to a Rigidbody2D
    /// </summary>
    [CommandInfo("Rigidbody2D",
                 "AddTorque2D",
                 "Add Torque to a Rigidbody2D")]
    [AddComponentMenu("")]
    public class AddTorque2D : Command
    {
        [SerializeField]
        protected Rigidbody2DData rb;

        [SerializeField]
        protected ForceMode2D forceMode = ForceMode2D.Force;

        [Tooltip("Amount of torque to be added")]
        [SerializeField]
        protected FloatData force;

        public override void OnEnter()
        {
            rb.Value.AddTorque(force.Value, forceMode);

            Continue();
        }

        public override string GetSummary()
        {
            if (rb.Value == null)
            {
                return "Error: rb not set";
            }

            return forceMode.ToString() + ": " + force.Value.ToString() + (force.floatRef != null ? " (" + force.floatRef.Key + ")" : "");
        }

        public override Color GetButtonColor()
        {
            return new Color32(235, 191, 217, 255);
        }
        
        public override bool HasReference(Variable variable)
        {
            if (rb.rigidbody2DRef == variable || force.floatRef == variable)
                return true;

            return false;
        }
    }
}