AlicizaX/DragonECS
1
0
Fork 0
mirror of https://github.com/DCFApixels/DragonECS.git synced 2025-09-18 18:14:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add IEcsInjectCallbacks

Browse Source
This commit is contained in:
Mikhail 2023-03-29 06:19:24 +08:00
parent e77c20a09f
commit 899cd9f40b
1 changed files with 117 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

125
src/Builtin/InjectSystem.cs
View File

@ -1,5 +1,55 @@
namespace DCFApixels.DragonECS using System.Collections.Generic;
using DCFApixels.DragonECS.Internal;
using System.Linq;
namespace DCFApixels.DragonECS
{ {
namespace Internal
{
internal class InjectController
{
private EcsSystems _source;
private InjectSystemBase[] _injectSystems;
private int _injectCount;
public InjectController(EcsSystems source)
{
_injectCount = 0;
_source = source;
_injectSystems = _source.AllSystems.OfType<InjectSystemBase>().ToArray();
}
public bool OnInject()
{
_injectCount++;
return IsInjectionEnd;
}
public bool IsInjectionEnd => _injectSystems.Length <= _injectCount;
public void Destroy()
{
_source = null;
_injectSystems = null;
}
}
}
public interface IEcsInject : IEcsSystem
{
public void Inject(object obj);
}
[DebugColor(DebugColor.Gray)]
public sealed class InjectRunner : EcsRunner<IEcsInject>, IEcsInject
{
void IEcsInject.Inject(object obj)
{
foreach (var item in targets) item.Inject(obj);
}
}
public interface IEcsInject<T> : IEcsSystem public interface IEcsInject<T> : IEcsSystem
{ {
public void Inject(T obj); public void Inject(T obj);
@ -9,18 +59,50 @@
{ {
void IEcsInject<T>.Inject(T obj) void IEcsInject<T>.Inject(T obj)
{ {
foreach (var item in targets) foreach (var item in targets) item.Inject(obj);
{ }
item.Inject(obj); }
}
public interface IEcsInjectCallbacks : IEcsSystem
{
public void OnInjectionBefore();
public void OnInjectionAfter();
}
[DebugColor(DebugColor.Gray)]
public sealed class InjectCallbacksRunner : EcsRunner<IEcsInjectCallbacks>, IEcsInjectCallbacks
{
public void OnInjectionAfter()
{
foreach (var item in targets) item.OnInjectionAfter();
}
public void OnInjectionBefore()
{
foreach (var item in targets) item.OnInjectionBefore();
}
}
public class InjectSystemBase
{
private static int _injectSystemID = EcsDebug.RegisterMark("InjectSystem");
protected static void ProfileMarkerBegin()
{
EcsDebug.ProfileMarkBegin(_injectSystemID);
}
protected static void ProfileMarkerEnd()
{
EcsDebug.ProfileMarkEnd(_injectSystemID);
} }
} }
[DebugColor(DebugColor.Gray)] [DebugColor(DebugColor.Gray)]
public class InjectSystem<T> : IEcsPreInitSystem public class InjectSystem<T> : InjectSystemBase, IEcsPreInitSystem, IEcsInject<InjectController>, IEcsInjectCallbacks
{ {
private T _injectedData; private T _injectedData;
private InjectController _injectController;
void IEcsInject<InjectController>.Inject(InjectController obj) => _injectController = obj;
public InjectSystem(T injectedData) public InjectSystem(T injectedData)
{ {
_injectedData = injectedData; _injectedData = injectedData;
@ -28,8 +110,35 @@
public void PreInit(EcsSystems systems) public void PreInit(EcsSystems systems)
{ {
var injector = systems.GetRunner<IEcsInject<T>>(); if(_injectController == null)
injector.Inject(_injectedData); {
ProfileMarkerBegin();
_injectController = new InjectController(systems);
var injectMapRunner = systems.GetRunner<IEcsInject<InjectController>>();
systems.GetRunner<IEcsInjectCallbacks>().OnInjectionBefore();
injectMapRunner.Inject(_injectController);
}
var injectRunnerGeneric = systems.GetRunner<IEcsInject<T>>();
var injectRunner = systems.GetRunner<IEcsInject>();
injectRunnerGeneric.Inject(_injectedData);
injectRunner.Inject(_injectedData);
if (_injectController.OnInject())
{
_injectController.Destroy();
var injectCallbacksRunner = systems.GetRunner<IEcsInjectCallbacks>();
injectCallbacksRunner.OnInjectionAfter();
ProfileMarkerEnd();
}
}
public void OnInjectionBefore() { }
public void OnInjectionAfter()
{
_injectController = null;
} }
} }