feat: add particle system getter and setter for attractor

close #253
mob-sakai · Aug 15, 2023 · a4bcf93 · a4bcf93
1 parent a499f0c
commit a4bcf93
Showing 1 changed file with 31 additions and 11 deletions.
diff --git a/Scripts/UIParticleAttractor.cs b/Scripts/UIParticleAttractor.cs
@@ -72,22 +72,25 @@ public Movement movement
             }
         }
 
-        private UIParticle _uiParticle;
-
-        private void OnEnable()
+        public ParticleSystem particleSystem
         {
-            if (m_ParticleSystem == null)
+            get
             {
-                Debug.LogError("No particle system attached to particle attractor script", this);
-                enabled = false;
-                return;
+                return m_ParticleSystem;
             }
-
-            _uiParticle = m_ParticleSystem.GetComponentInParent<UIParticle>();
-            if (_uiParticle && !_uiParticle.particles.Contains(m_ParticleSystem))
+            set
             {
-                _uiParticle = null;
+                m_ParticleSystem = value;
+                if (!ApplyParticleSystem()) return;
+                enabled = true;
             }
+        }
+
+        private UIParticle _uiParticle;
+
+        private void OnEnable()
+        {
+            if (!ApplyParticleSystem()) return;
             UIParticleUpdater.Register(this);
         }
 
@@ -200,5 +203,22 @@ private Vector3 GetAttractedPosition(Vector3 current, Vector3 target, float dura
             return Vector3.MoveTowards(current, target, speed);
         }
 
+        private bool ApplyParticleSystem()
+        {
+            if (m_ParticleSystem == null)
+            {
+                Debug.LogError("No particle system attached to particle attractor script", this);
+                enabled = false;
+                return false;
+            }
+
+            _uiParticle = m_ParticleSystem.GetComponentInParent<UIParticle>();
+            if (_uiParticle && !_uiParticle.particles.Contains(m_ParticleSystem))
+            {
+                _uiParticle = null;
+            }
+
+            return true;
+        }
     }
 }