Merge branch 'release/2022.7.22'

Author: Leopotam

README.md

@@ -59,7 +59,7 @@ class TestSystem : IEcsRunSystem {
     // Поле будет содержать ссылку на мир "events".
     readonly EcsWorldInject _eventsWorld = "events";
 
-    public void Run (EcsSystems systems) {
+    public void Run (IEcsSystems systems) {
         // Все поля заполнены и могут быть использованы:
         // _defaultWorld.Value.xxx
         // _eventsWorld.Value.xxx
@@ -76,7 +76,7 @@ class TestSystem : IEcsRunSystem {
     // Поле будет содержать ссылку на пул из мира "events".
     readonly EcsPoolInject<C1> _c1EventsPool = "events";
 
-    public void Run (EcsSystems systems) {
+    public void Run (IEcsSystems systems) {
         // Все поля заполнены и могут быть использованы:
         // _c1Pool.Value.xxx
         // _c1EventsPool.Value.xxx
@@ -99,7 +99,7 @@ class TestSystem : IEcsRunSystem {
     // Поле будет содержать ссылку на фильтр (с C1, но без C2) из мира "events".
     readonly EcsFilter<Inc<C1>, Exc<C2>> _eventsFilter11 = "events";
 
-    public void Run (EcsSystems systems) {
+    public void Run (IEcsSystems systems) {
         // Все поля заполнены и могут быть использованы:
         // _filter1.Value.xxx
         // _filter2.Value.xxx
@@ -188,7 +188,7 @@ class TestSystem : IEcsRunSystem {
     // Поле будет содержать ссылку на GetShared() объект.
     readonly EcsSharedInject<Shared> _shared = default;
 
-    public void Run (EcsSystems systems) {
+    public void Run (IEcsSystems systems) {
         // Все поля заполнены и могут быть использованы:
         // _shared.Value.xxx
     }
@@ -209,7 +209,7 @@ class TestSystem : IEcsRunSystem {
     // Поле будет содержать ссылку на объект совместимого типа, переданого в вызов EcsSystems.Inject(xxx).
     readonly EcsCustomInject<CustomData2> _custom2 = default;
 
-    public void Run (EcsSystems systems) {
+    public void Run (IEcsSystems systems) {
         // Все поля заполнены и могут быть использованы:
         // _custom1.Value.xxx
         // _custom2.Value.xxx

package.json

@@ -4,7 +4,7 @@
     "displayName": "LeoECS Lite DI",
     "description": "LeoECS Lite DI - Поддержка автоматической инъекции данных в поля ECS-систем.",
     "unity": "2020.3",
-    "version": "2022.3.22",
+    "version": "2022.7.22",
     "keywords": [
         "leoecslite",
         "leoecs",

src/extensions.cs

@@ -7,28 +7,39 @@
 using System.Reflection;
 
 namespace Leopotam.EcsLite.Di {
+#if LEOECSLITE_DI
+    public interface IEcsInjectSystem : IEcsSystem {
+        void Inject (IEcsSystems systems, params object[] injects);
+    }
+#endif
     public static class Extensions {
-        public static EcsSystems Inject (this EcsSystems systems, params object[] injects) {
+        public static IEcsSystems Inject (this IEcsSystems systems, params object[] injects) {
             if (injects == null) { injects = Array.Empty<object> (); }
-            IEcsSystem[] allSystems = null;
-            var systemsCount = systems.GetAllSystems (ref allSystems);
-
-            for (var i = 0; i < systemsCount; i++) {
-                var system = allSystems[i];
-                foreach (var f in system.GetType ().GetFields (BindingFlags.Public | BindingFlags.NonPublic | BindingFlags.Instance)) {
-                    // skip statics.
-                    if (f.IsStatic) { continue; }
-                    // EcsWorldInject, EcsFilterInject, EcsPoolInject, EcsSharedInject.
-                    if (InjectBuiltIns (f, system, systems)) { continue; }
-                    // EcsDataInject.
-                    if (InjectCustoms (f, system, injects)) { continue; }
+            foreach (var system in systems.GetAllSystems ()) {
+#if LEOECSLITE_DI
+                if (system is IEcsInjectSystem injectSystem) {
+                    injectSystem.Inject (systems, injects);
+                    continue;
                 }
+#endif
+                InjectToSystem (system, systems, injects);
             }
 
             return systems;
         }
 
-        static bool InjectBuiltIns (FieldInfo fieldInfo, IEcsSystem system, EcsSystems systems) {
+        public static void InjectToSystem (IEcsSystem system, IEcsSystems systems, object[] injects) {
+            foreach (var f in system.GetType ().GetFields (BindingFlags.Public | BindingFlags.NonPublic | BindingFlags.Instance)) {
+                // skip statics.
+                if (f.IsStatic) { continue; }
+                // EcsWorldInject, EcsFilterInject, EcsPoolInject, EcsSharedInject.
+                if (InjectBuiltIns (f, system, systems)) { continue; }
+                // EcsDataInject.
+                if (InjectCustoms (f, system, injects)) { }
+            }
+        }
+
+        static bool InjectBuiltIns (FieldInfo fieldInfo, IEcsSystem system, IEcsSystems systems) {
             if (typeof (IEcsDataInject).IsAssignableFrom (fieldInfo.FieldType)) {
                 var instance = (IEcsDataInject) fieldInfo.GetValue (system);
                 instance.Fill (systems);
@@ -50,7 +61,7 @@ static bool InjectCustoms (FieldInfo fieldInfo, IEcsSystem system, object[] inje
     }
 
     public interface IEcsDataInject {
-        void Fill (EcsSystems systems);
+        void Fill (IEcsSystems systems);
     }
 
     public interface IEcsCustomDataInject {
@@ -75,7 +86,7 @@ public static implicit operator EcsFilterInject<TInc> (string worldName) {
             return new EcsFilterInject<TInc> { _worldName = worldName };
         }
 
-        void IEcsDataInject.Fill (EcsSystems systems) {
+        void IEcsDataInject.Fill (IEcsSystems systems) {
             Pools = default;
             Value = Pools.Fill (systems.GetWorld (_worldName)).End ();
         }
@@ -92,7 +103,7 @@ public static implicit operator EcsFilterInject<TInc, TExc> (string worldName) {
             return new EcsFilterInject<TInc, TExc> { _worldName = worldName };
         }
 
-        void IEcsDataInject.Fill (EcsSystems systems) {
+        void IEcsDataInject.Fill (IEcsSystems systems) {
             Pools = default;
             TExc exc = default;
             Value = exc.Fill (Pools.Fill (systems.GetWorld (_worldName))).End ();
@@ -107,15 +118,15 @@ public static implicit operator EcsPoolInject<T> (string worldName) {
             return new EcsPoolInject<T> { _worldName = worldName };
         }
 
-        void IEcsDataInject.Fill (EcsSystems systems) {
+        void IEcsDataInject.Fill (IEcsSystems systems) {
             Value = systems.GetWorld (_worldName).GetPool<T> ();
         }
     }
 
     public struct EcsSharedInject<T> : IEcsDataInject where T : class {
         public T Value;
 
-        void IEcsDataInject.Fill (EcsSystems systems) {
+        void IEcsDataInject.Fill (IEcsSystems systems) {
             Value = systems.GetShared<T> ();
         }
     }
@@ -144,7 +155,7 @@ public static implicit operator EcsWorldInject (string worldName) {
             return new EcsWorldInject { _worldName = worldName };
         }
 
-        void IEcsDataInject.Fill (EcsSystems systems) {
+        void IEcsDataInject.Fill (IEcsSystems systems) {
             Value = systems.GetWorld (_worldName);
         }
     }