1 files changed, 45 insertions, 0 deletions
diff --git a/Client/Assets/Behavior Designer/Runtime/Basic Tasks/Rigidbody2D/GetDrag.cs b/Client/Assets/Behavior Designer/Runtime/Basic Tasks/Rigidbody2D/GetDrag.cs
new file mode 100644
index 00000000..b01bf9a2
--- /dev/null
+++ b/Client/Assets/Behavior Designer/Runtime/Basic Tasks/Rigidbody2D/GetDrag.cs
@@ -0,0 +1,45 @@
+using UnityEngine;
+
+namespace BehaviorDesigner.Runtime.Tasks.Basic.UnityRigidbody2D
+{
+    [TaskCategory("Basic/Rigidbody2D")]
+    [TaskDescription("Stores the drag of the Rigidbody2D. Returns Success.")]
+    public class GetDrag : Action
+    {
+        [Tooltip("The GameObject that the task operates on. If null the task GameObject is used.")]
+        public SharedGameObject targetGameObject;
+        [Tooltip("The drag of the Rigidbody2D")]
+        [RequiredField]
+        public SharedFloat storeValue;
+
+        private Rigidbody2D rigidbody2D;
+        private GameObject prevGameObject;
+
+        public override void OnStart()
+        {
+            var currentGameObject = GetDefaultGameObject(targetGameObject.Value);
+            if (currentGameObject != prevGameObject) {
+                rigidbody2D = currentGameObject.GetComponent<Rigidbody2D>();
+                prevGameObject = currentGameObject;
+            }
+        }
+
+        public override TaskStatus OnUpdate()
+        {
+            if (rigidbody2D == null) {
+                Debug.LogWarning("Rigidbody2D is null");
+                return TaskStatus.Failure;
+            }
+
+            storeValue.Value = rigidbody2D.drag;
+
+            return TaskStatus.Success;
+        }
+
+        public override void OnReset()
+        {
+            targetGameObject = null;
+            storeValue = 0;
+        }
+    }
+}