support creating task with custom factory function

lunditoph-siago · Nov 22, 2023 · 8afbe1f · 8afbe1f
1 parent 948909b
commit 8afbe1f
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/Sia/Systems/SystemChain.cs b/Sia/Systems/SystemChain.cs
@@ -53,6 +53,21 @@ public SystemChain Add<TSystem>()
         }
         return new(Sequence.Add(typeof(TSystem)), newEntries);
     }
+
+    public SystemChain Add<TSystem>(Func<TSystem> creator)
+        where TSystem : ISystem
+    {
+        var newEntries = Entries.Add(
+            typeof(TSystem), new(
+                (sysLib, scheduler, taskGraphNodes) => sysLib.Register(scheduler, creator, taskGraphNodes),
+                GetAttributedSystemTypes<TSystem>(typeof(AfterSystemAttribute<>)),
+                GetAttributedSystemTypes<TSystem>(typeof(BeforeSystemAttribute<>))
+            ));
+        if (newEntries == Entries) {
+            return this;
+        }
+        return new(Sequence.Add(typeof(TSystem)), newEntries);
+    }
 
     public SystemChain Remove<TSystem>()
         where TSystem : ISystem, new()

diff --git a/Sia/Systems/SystemLibrary.cs b/Sia/Systems/SystemLibrary.cs
@@ -225,8 +225,12 @@ internal Entry Acquire(Type systemType)
 
     public SystemHandle Register<TSystem>(Scheduler scheduler, IEnumerable<Scheduler.TaskGraphNode>? dependedTasks = null)
         where TSystem : ISystem, new()
+        => Register<TSystem>(scheduler, () => new(), dependedTasks);
+
+    public SystemHandle Register<TSystem>(Scheduler scheduler, Func<TSystem> creator, IEnumerable<Scheduler.TaskGraphNode>? dependedTasks = null)
+        where TSystem : ISystem
     {
-        var system = new TSystem();
+        var system = creator();
         var sysEntry = Acquire(system.GetType());
 
         [MethodImpl(MethodImplOptions.AggressiveInlining)]