72 lines
2.2 KiB
C#
72 lines
2.2 KiB
C#
using System.Collections;
|
|
using System.Collections.Generic;
|
|
|
|
using Syntriax.Engine.Core;
|
|
|
|
namespace Syntriax.Engine.Physics2D;
|
|
|
|
public class PhysicsCoroutineManager : Behaviour
|
|
{
|
|
private readonly Event<IUniverse, IUniverse.UpdateArguments>.EventHandler delegateOnUpdate = null!;
|
|
|
|
private readonly List<IEnumerator> enumerators = [];
|
|
private IPhysicsEngine2D? physicsEngine = null;
|
|
|
|
public IEnumerator CreateCoroutine(IEnumerator enumerator)
|
|
{
|
|
enumerators.Add(enumerator);
|
|
return enumerator;
|
|
}
|
|
|
|
public void StopCoroutine(IEnumerator enumerator)
|
|
{
|
|
enumerators.Remove(enumerator);
|
|
}
|
|
|
|
protected override void OnEnteredUniverse(IUniverse universe)
|
|
{
|
|
physicsEngine = universe.FindRequiredBehaviour<IPhysicsEngine2D>();
|
|
if (physicsEngine is IPhysicsEngine2D foundPhysicsEngine)
|
|
foundPhysicsEngine.OnPhysicsStep.RemoveListener(OnPhysicsStep);
|
|
else
|
|
universe.OnUpdate.AddListener(OnUpdate);
|
|
}
|
|
|
|
private void OnPhysicsStep(IPhysicsEngine2D sender, float stepDeltaTime)
|
|
{
|
|
for (int i = enumerators.Count - 1; i >= 0; i--)
|
|
{
|
|
if (enumerators[i].Current is ICoroutineYield coroutineYield && coroutineYield.Yield())
|
|
continue;
|
|
|
|
if (!enumerators[i].MoveNext())
|
|
enumerators.RemoveAt(i);
|
|
}
|
|
}
|
|
|
|
protected override void OnExitedUniverse(IUniverse universe)
|
|
{
|
|
if (physicsEngine is IPhysicsEngine2D existingPhysicsEngine)
|
|
existingPhysicsEngine.OnPhysicsStep.RemoveListener(OnPhysicsStep);
|
|
universe.OnUpdate.RemoveListener(delegateOnUpdate);
|
|
}
|
|
|
|
private void OnUpdate(IUniverse sender, IUniverse.UpdateArguments args)
|
|
{
|
|
if (Universe is not IUniverse universe)
|
|
return;
|
|
|
|
physicsEngine = universe.FindRequiredBehaviour<IPhysicsEngine2D>();
|
|
if (physicsEngine is IPhysicsEngine2D foundPhysicsEngine)
|
|
{
|
|
foundPhysicsEngine.OnPhysicsStep.AddListener(OnPhysicsStep);
|
|
universe.OnUpdate.RemoveListener(delegateOnUpdate);
|
|
}
|
|
}
|
|
|
|
public PhysicsCoroutineManager()
|
|
{
|
|
delegateOnUpdate = OnUpdate;
|
|
}
|
|
}
|