1 files changed, 46 insertions, 0 deletions
diff --git a/Assets/ThirdParty/Behavior Designer/Runtime/Basic Tasks/ParticleSystem/SetGravityModifier.cs b/Assets/ThirdParty/Behavior Designer/Runtime/Basic Tasks/ParticleSystem/SetGravityModifier.cs
new file mode 100644
index 00000000..e11dcd85
--- /dev/null
+++ b/Assets/ThirdParty/Behavior Designer/Runtime/Basic Tasks/ParticleSystem/SetGravityModifier.cs
@@ -0,0 +1,46 @@
+#if UNITY_5_1 || UNITY_5_2 || UNITY_5_3 || UNITY_5_4
+using UnityEngine;
+
+namespace BehaviorDesigner.Runtime.Tasks.Basic.UnityParticleSystem
+{
+    [TaskCategory("Basic/ParticleSystem")]
+    [TaskDescription("Sets the gravity modifier of the Particle System.")]
+    public class SetGravityModifier : Action
+    {
+        [Tooltip("The GameObject that the task operates on. If null the task GameObject is used.")]
+        public SharedGameObject targetGameObject;
+        [Tooltip("The gravity modifier of the ParticleSystem")]
+        public SharedFloat gravityModifier;
+
+        private ParticleSystem particleSystem;
+        private GameObject prevGameObject;
+
+        public override void OnStart()
+        {
+            var currentGameObject = GetDefaultGameObject(targetGameObject.Value);
+            if (currentGameObject != prevGameObject) {
+                particleSystem = currentGameObject.GetComponent<ParticleSystem>();
+                prevGameObject = currentGameObject;
+            }
+        }
+
+        public override TaskStatus OnUpdate()
+        {
+            if (particleSystem == null) {
+                Debug.LogWarning("ParticleSystem is null");
+                return TaskStatus.Failure;
+            }
+
+            particleSystem.gravityModifier = gravityModifier.Value;
+
+            return TaskStatus.Success;
+        }
+
+        public override void OnReset()
+        {
+            targetGameObject = null;
+            gravityModifier = 0;
+        }
+    }
+}
+#endif
+\ No newline at end of file