1 files changed, 44 insertions, 0 deletions
diff --git a/Client/Assets/Behavior Designer/Runtime/Basic Tasks/ParticleSystem/SetGravityModifier.cs b/Client/Assets/Behavior Designer/Runtime/Basic Tasks/ParticleSystem/SetGravityModifier.cs
new file mode 100644
index 00000000..1082989d
--- /dev/null
+++ b/Client/Assets/Behavior Designer/Runtime/Basic Tasks/ParticleSystem/SetGravityModifier.cs
@@ -0,0 +1,44 @@
+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.MainModule main = particleSystem.main;
+            main.gravityModifierMultiplier = gravityModifier.Value;
+            //particleSystem.main = main;
+            return TaskStatus.Success;
+        }
+
+        public override void OnReset()
+        {
+            targetGameObject = null;
+            gravityModifier = 0;
+        }
+    }
+}
+\ No newline at end of file