chore: Clean up

thiagomvas · Jun 18, 2024 · 7c4d854 · 7c4d854
1 parent 00bc197
commit 7c4d854
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/Basalt/Common/Components/BaseParticleSystem.cs b/Basalt/Common/Components/BaseParticleSystem.cs
@@ -1,5 +1,6 @@
 using Basalt.Common.Entities;
 using Basalt.Types;
+using System.Drawing;
 using System.Numerics;
 
 namespace Basalt.Common.Components
@@ -94,7 +95,8 @@ public override void OnUpdate()
 		{
 			var dt = Time.DeltaTime;
 			_systemLifetime += dt;
-			if (!Looping && _systemLifetime > SystemDuration)
+
+			if (!Looping && _systemLifetime > SystemDuration + ParticleLifetime)
 				return;
 
 			for (int i = 0; i < _particles.Length; i++)
@@ -104,6 +106,12 @@ public override void OnUpdate()
 				_particleUpdate?.Invoke(ref _particles[i]);
 				if (_particles[i].Lifetime > ParticleLifetime)
 				{
+					if (!Looping && _systemLifetime > SystemDuration)
+					{
+						_particles[i] = defaults;
+						_particles[i].Color = Color.FromArgb(0x00000000);
+						continue;
+					}
 					_particles[i] = defaults;
 					_onParticleReset?.Invoke(ref _particles[i]);
 				}