From 21e186f75b504d832d9c7bef0456edd7d5d3155e Mon Sep 17 00:00:00 2001
From: chai <chaifix@163.com>
Date: Wed, 8 Sep 2021 10:52:35 +0800
Subject: +behavior design

---
 .../Runtime/Basic Tasks/Rigidbody/SetMass.cs       | 45 ++++++++++++++++++++++
 1 file changed, 45 insertions(+)
 create mode 100644 Assets/ThirdParty/Behavior Designer/Runtime/Basic Tasks/Rigidbody/SetMass.cs

(limited to 'Assets/ThirdParty/Behavior Designer/Runtime/Basic Tasks/Rigidbody/SetMass.cs')

diff --git a/Assets/ThirdParty/Behavior Designer/Runtime/Basic Tasks/Rigidbody/SetMass.cs b/Assets/ThirdParty/Behavior Designer/Runtime/Basic Tasks/Rigidbody/SetMass.cs
new file mode 100644
index 00000000..48c1da8f
--- /dev/null
+++ b/Assets/ThirdParty/Behavior Designer/Runtime/Basic Tasks/Rigidbody/SetMass.cs	
@@ -0,0 +1,45 @@
+using UnityEngine;
+
+namespace BehaviorDesigner.Runtime.Tasks.Basic.UnityRigidbody
+{
+    [TaskCategory("Basic/Rigidbody")]
+    [TaskDescription("Sets the mass of the Rigidbody. Returns Success.")]
+    public class SetMass : Action
+    {
+        [Tooltip("The GameObject that the task operates on. If null the task GameObject is used.")]
+        public SharedGameObject targetGameObject;
+        [Tooltip("The mass of the Rigidbody")]
+        public SharedFloat mass;
+
+        // cache the rigidbody component
+        private Rigidbody rigidbody;
+        private GameObject prevGameObject;
+
+        public override void OnStart()
+        {
+            var currentGameObject = GetDefaultGameObject(targetGameObject.Value);
+            if (currentGameObject != prevGameObject) {
+                rigidbody = currentGameObject.GetComponent<Rigidbody>();
+                prevGameObject = currentGameObject;
+            }
+        }
+
+        public override TaskStatus OnUpdate()
+        {
+            if (rigidbody == null) {
+                Debug.LogWarning("Rigidbody is null");
+                return TaskStatus.Failure;
+            }
+
+            rigidbody.mass = mass.Value;
+
+            return TaskStatus.Success;
+        }
+
+        public override void OnReset()
+        {
+            targetGameObject = null;
+            mass = 0;
+        }
+    }
+}
\ No newline at end of file
-- 
cgit v1.1-26-g67d0