Completes additional generic overloads feature mentioned in #20

genaray · Nov 27, 2022 · db51cb9 · db51cb9
1 parent cbdbce6
commit db51cb9
Show file tree

Hide file tree

Showing 2 changed files with 52 additions and 0 deletions.
diff --git a/Arch.SourceGenerator/Fundamentals/StructuralChangesExtensions.cs b/Arch.SourceGenerator/Fundamentals/StructuralChangesExtensions.cs
@@ -90,6 +90,56 @@ public static StringBuilder AppendWorldRemove(this StringBuilder sb, int amount)
 
     Move(in entity, oldArchetype, newArchetype);
 }}
+";
+
+        return sb.AppendLine(template);
+    }
+
+    public static StringBuilder AppendEntityAdds(this StringBuilder sb, int amount)
+    {
+        for (var index = 1; index < amount; index++)
+            sb.AppendEntityAdd(index);
+
+        return sb;
+    }
+
+    public static StringBuilder AppendEntityAdd(this StringBuilder sb, int amount)
+    {
+
+        var generics = new StringBuilder().GenericWithoutBrackets(amount);
+        var parameters = new StringBuilder().GenericInDefaultParams(amount);
+
+        var template = $@"
+[MethodImpl(MethodImplOptions.AggressiveInlining)]
+public static void Add<{generics}>(this in Entity entity, {parameters})
+{{
+    var world = World.Worlds[entity.WorldId];
+    world.Add<{generics}>(in entity);
+}}
+";
+
+        return sb.AppendLine(template);
+    }
+
+    public static StringBuilder AppendEntityRemoves(this StringBuilder sb, int amount)
+    {
+        for (var index = 1; index < amount; index++)
+            sb.AppendEntityRemove(index);
+
+        return sb;
+    }
+
+    public static StringBuilder AppendEntityRemove(this StringBuilder sb, int amount)
+    {
+
+        var generics = new StringBuilder().GenericWithoutBrackets(amount);
+        var template = $@"
+[MethodImpl(MethodImplOptions.AggressiveInlining)]
+public static void Remove<{generics}>(this in Entity entity)
+{{
+    var world = World.Worlds[entity.WorldId];
+    world.Remove<{generics}>(in entity);
+}}
 ";
 
         return sb.AppendLine(template);

diff --git a/Arch.SourceGenerator/QueryGenerator.cs b/Arch.SourceGenerator/QueryGenerator.cs
@@ -104,6 +104,8 @@ public static partial class EntityExtensions{{
                     {new StringBuilder().AppendEntityHases(10)}
                     {new StringBuilder().AppendEntitySets(10)}
                     {new StringBuilder().AppendEntityGets(10)}
+                    {new StringBuilder().AppendEntityAdds(10)}
+                    {new StringBuilder().AppendEntityRemoves(10)}
                 }}
             ");