Fix an issue with the new generator, switch to using entity references

CodeGenerator/CodeGenerator.csproj

@@ -8,7 +8,7 @@
         <PackageProjectUrl>https://git.exmods.org/modtainers/GamecraftModdingAPI</PackageProjectUrl>
         <NeutralLanguage>en-CA</NeutralLanguage>
         <AllowUnsafeBlocks>true</AllowUnsafeBlocks>
-        <LangVersion>9</LangVersion>
+        <LangVersion>latest</LangVersion>
         <OutputType>Exe</OutputType>
     </PropertyGroup>
     <PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'Debug|AnyCPU' ">

CodeGenerator/ECSAnalyzer.cs

@@ -5,21 +5,23 @@ using System.Linq;
 using System.Linq.Expressions;
 using System.Reflection;
 using Gamecraft.Tweaks;
+using HarmonyLib;
 using Svelto.ECS;
 
 namespace CodeGenerator
 {
-    public class ECSAnalyzer
+    public static class ECSAnalyzer
     {
         public static ECSClassInfo AnalyzeEntityDescriptor(Type entityDescriptorType)
         {
             // TODO: Add support for creating/deleting entities (getting an up to date server/client engines root)
             var templateType = typeof(EntityDescriptorTemplate<>).MakeGenericType(entityDescriptorType);
-            var getTemplateClass = Expression.Constant(templateType);
-            var getDescriptorExpr = Expression.PropertyOrField(getTemplateClass, "descriptor");
+            var templateDescriptor = AccessTools.Property(templateType, "descriptor");
+            var getDescriptorExpr = Expression.MakeMemberAccess(null, templateDescriptor ?? throw new InvalidOperationException());
             var getTemplateDescriptorExpr =
                 Expression.Lambda<Func<IEntityDescriptor>>(getDescriptorExpr);
             var getTemplateDescriptor = getTemplateDescriptorExpr.Compile();
+            // TODO: Crashes on constructing the descriptor type. Maybe move the analysis part to a mod.
             var builders = getTemplateDescriptor().componentsToBuild;
             return new ECSClassInfo
             {

CodeGenerator/Program.cs

@@ -15,7 +15,8 @@ namespace CodeGenerator
     {
         public static void Main(string[] args)
         {
-            GenerateBlockClasses();
+            //GenerateBlockClasses();
+            ECSClassGenerator.Generate(typeof(EngineBlockEntityDescriptor));
         }
 
         private static void GenerateBlockClasses()

TechbloxModdingAPI/App/GameBuildSimEventEngine.cs

@@ -1,6 +1,6 @@
 using RobocraftX.StateSync;
 using Svelto.ECS;
-using TechbloxModdingAPI.Common;
+using TechbloxModdingAPI.Common.Engines;
 using Unity.Jobs;
 using TechbloxModdingAPI.Utility;
 

TechbloxModdingAPI/Block.cs

@@ -8,13 +8,13 @@ using Svelto.ECS.EntityStructs;
 using RobocraftX.Common;
 using RobocraftX.Blocks;
 using Unity.Mathematics;
-using HarmonyLib;
 using RobocraftX.PilotSeat;
 using RobocraftX.Rendering;
 using Techblox.BlockLabelsServer;
 using TechbloxModdingAPI.Blocks;
 using TechbloxModdingAPI.Blocks.Engines;
 using TechbloxModdingAPI.Client.App;
+using TechbloxModdingAPI.Common;
 using TechbloxModdingAPI.Common.Engines;
 using TechbloxModdingAPI.Tests;
 using TechbloxModdingAPI.Utility;

TechbloxModdingAPI/BlockGroup.cs

@@ -9,8 +9,8 @@ using UnityEngine;
 
 using TechbloxModdingAPI.Blocks;
 using TechbloxModdingAPI.Blocks.Engines;
+using TechbloxModdingAPI.Common;
 using TechbloxModdingAPI.Common.Engines;
-using TechbloxModdingAPI.Utility;
 
 namespace TechbloxModdingAPI
 {

TechbloxModdingAPI/Blocks/Engines/BlockCloneEngine.cs

@@ -5,11 +5,9 @@ using Gamecraft.Wires;
 using HarmonyLib;
 using RobocraftX.Blocks;
 using RobocraftX.Character;
-using RobocraftX.Common;
 using Svelto.DataStructures;
 using Svelto.ECS;
-using TechbloxModdingAPI.Common;
-using TechbloxModdingAPI.Engines;
+using TechbloxModdingAPI.Common.Engines;
 
 namespace TechbloxModdingAPI.Blocks.Engines
 {

TechbloxModdingAPI/Blocks/Engines/BlockEngine.cs

@@ -18,9 +18,8 @@ using Svelto.ECS.Hybrid;
 using Techblox.BuildingDrone;
 using Techblox.ObjectIDBlockServer;
 using TechbloxModdingAPI.Common;
+using TechbloxModdingAPI.Common.Engines;
 using Unity.Mathematics;
-
-using TechbloxModdingAPI.Engines;
 using TechbloxModdingAPI.Utility;
 using TechbloxModdingAPI.Utility.ECS;
 using PrefabsID = RobocraftX.Common.PrefabsID;

TechbloxModdingAPI/Blocks/Engines/MovementEngine.cs

@@ -2,7 +2,7 @@
 using RobocraftX.DOTS;
 using Svelto.ECS;
 using Svelto.ECS.EntityStructs;
-using TechbloxModdingAPI.Common;
+using TechbloxModdingAPI.Common.Engines;
 using Unity.Mathematics;
 using Unity.Transforms;
 using TechbloxModdingAPI.Utility;

TechbloxModdingAPI/Blocks/Engines/PlacementEngine.cs

@@ -12,7 +12,7 @@ using RobocraftX.Rendering;
 using RobocraftX.Rendering.GPUI;
 using Svelto.ECS;
 using Svelto.ECS.EntityStructs;
-using TechbloxModdingAPI.Common;
+using TechbloxModdingAPI.Common.Engines;
 using Unity.Mathematics;
 using TechbloxModdingAPI.Utility;
 using TechbloxModdingAPI.Utility.ECS;

TechbloxModdingAPI/Blocks/Engines/RemovalEngine.cs

@@ -9,12 +9,10 @@ using RobocraftX.StateSync;
 using Svelto.ECS;
 using Svelto.ECS.Native;
 using Techblox.Blocks.Connections;
-using TechbloxModdingAPI.Common;
+using TechbloxModdingAPI.Common.Engines;
 using Unity.Collections;
 using Unity.Jobs;
 using Allocator = Unity.Collections.Allocator;
-
-using TechbloxModdingAPI.Engines;
 using TechbloxModdingAPI.Utility;
 
 namespace TechbloxModdingAPI.Blocks.Engines

TechbloxModdingAPI/Blocks/Engines/RotationEngine.cs

@@ -2,11 +2,9 @@
 using RobocraftX.DOTS;
 using Svelto.ECS;
 using Svelto.ECS.EntityStructs;
-using TechbloxModdingAPI.Common;
+using TechbloxModdingAPI.Common.Engines;
 using Unity.Mathematics;
 using UnityEngine;
-
-using TechbloxModdingAPI.Engines;
 using TechbloxModdingAPI.Utility;
 using TechbloxModdingAPI.Utility.ECS;
 

TechbloxModdingAPI/Blocks/Engines/ScalingEngine.cs

@@ -4,10 +4,8 @@ using HarmonyLib;
 using RobocraftX.Common;
 using RobocraftX.DOTS;
 using Svelto.ECS;
-using TechbloxModdingAPI.Common;
+using TechbloxModdingAPI.Common.Engines;
 using Unity.Entities;
-
-using TechbloxModdingAPI.Engines;
 using TechbloxModdingAPI.Utility;
 
 namespace TechbloxModdingAPI.Blocks.Engines

TechbloxModdingAPI/Blocks/Wire.cs

@@ -1,11 +1,7 @@
-using System;
-
 using Gamecraft.Wires;
 using Svelto.ECS;
-using Svelto.ECS.Experimental;
-
 using TechbloxModdingAPI.Blocks.Engines;
-using TechbloxModdingAPI.Utility;
+using TechbloxModdingAPI.Common;
 
 namespace TechbloxModdingAPI.Blocks
 {

TechbloxModdingAPI/Client/App/ClientEngine.cs

@@ -3,7 +3,7 @@ using RobocraftX.Multiplayer;
 using Svelto.ECS;
 using Techblox.GameSelection;
 using TechbloxModdingAPI.Client.Game;
-using TechbloxModdingAPI.Common;
+using TechbloxModdingAPI.Common.Engines;
 using TechbloxModdingAPI.Utility;
 
 namespace TechbloxModdingAPI.Client.App;

TechbloxModdingAPI/Client/Game/ClientMachine.cs

@@ -1,6 +1,6 @@
-using System;
 using RobocraftX.GUI.MyGamesScreen;
 using Svelto.ECS;
+using TechbloxModdingAPI.Common;
 
 namespace TechbloxModdingAPI.Client.Game;
 

TechbloxModdingAPI/Cluster.cs

@@ -1,6 +1,7 @@
 using Svelto.ECS;
 using Techblox.Destruction;
 using Techblox.TimeRunning.Clusters;
+using TechbloxModdingAPI.Common;
 
 namespace TechbloxModdingAPI
 {

TechbloxModdingAPI/Commands/ICustomCommandEngine.cs

@@ -1,4 +1,4 @@
-using TechbloxModdingAPI.Common;
+using TechbloxModdingAPI.Common.Engines;
 
 namespace TechbloxModdingAPI.Commands
 {

TechbloxModdingAPI/Common/EcsObjectBase.cs

@@ -1,21 +1,27 @@
 using System;
 using System.Collections.Generic;
-using System.Linq.Expressions;
 using Svelto.DataStructures;
 using Svelto.ECS;
 using Svelto.ECS.Internal;
+using TechbloxModdingAPI.Common.Engines;
 using TechbloxModdingAPI.Common.Utils;
 using TechbloxModdingAPI.Utility;
 
-namespace TechbloxModdingAPI
-{
+namespace TechbloxModdingAPI.Common;
+
 public abstract class EcsObjectBase
 {
-        public EGID Id { get; }
+    public EGID Id => _engine.GetEgid(Reference);
+    /// <summary>
+    /// A reference to a specific entity that persists through group swaps and such.
+    /// May be an invalid reference, in that case operations do not have any effect.
+    /// </summary>
+    public EntityReference Reference { get; }
         
-        private static readonly Dictionary<Type, WeakDictionary<EGID, EcsObjectBase>> _instances = new();
+    private static readonly Dictionary<Type, WeakDictionary<EntityReference, EcsObjectBase>> _instances = new();
+    private static readonly EcsObjectBaseEngine _engine = new();
 
-        internal static WeakDictionary<EGID, EcsObjectBase> GetInstances(Type type)
+    private static WeakDictionary<EntityReference, EcsObjectBase> GetInstances(Type type)
     {
         return _instances.TryGetValue(type, out var dict) ? dict : null;
     }
@@ -31,21 +37,25 @@ namespace TechbloxModdingAPI
     internal static T GetInstance<T>(EGID egid, Func<EGID, T> constructor, Type type = null) where T : EcsObjectBase
     {
         var instances = GetInstances(type ?? typeof(T));
-            if (instances == null || !instances.TryGetValue(egid, out var instance))
+        if (instances == null || !instances.TryGetValue(_engine.GetEntityReference(egid), out var instance))
             return constructor(egid); // It will be added by the constructor
         return (T)instance;
     }
 
-        protected EcsObjectBase(EGID id, Type entityDescriptorType)
+    protected EcsObjectBase(EGID id, Type entityDescriptorType) : this(_engine.GetEntityReference(id), entityDescriptorType)
+    {
+    }
+
+    protected EcsObjectBase(EntityReference reference, Type entityDescriptorType)
     {
         if (!_instances.TryGetValue(GetType(), out var dict))
         {
-                dict = new WeakDictionary<EGID, EcsObjectBase>();
+            dict = new();
             _instances.Add(GetType(), dict);
         }
-            if (!dict.ContainsKey(id)) // Multiple instances may be created
-                dict.Add(id, this);
-            Id = id;
+        if (!dict.ContainsKey(reference)) // Multiple instances may be created
+            dict.Add(reference, this);
+        Reference = reference;
     }
 
     #region ECS initializer stuff
@@ -77,5 +87,9 @@ namespace TechbloxModdingAPI
     private static readonly GetInitGroupFunc GetInitGroup = Reflections.CreateAccessor<GetInitGroupFunc>("_group");
 
     #endregion
+
+    public static void Init()
+    {
+        EngineManager.AddEngine(_engine, ApiEngineType.Build, ApiEngineType.Menu, ApiEngineType.PlayClient, ApiEngineType.PlayServer);
     }
 }

TechbloxModdingAPI/Common/EcsObjectBaseEngine.cs

@@ -0,0 +1,26 @@
+using Svelto.ECS;
+using TechbloxModdingAPI.Common.Engines;
+
+namespace TechbloxModdingAPI.Common;
+
+public class EcsObjectBaseEngine : IApiEngine
+{
+    public void Ready()
+    {
+    }
+
+    public EntitiesDB entitiesDB { get; set; }
+    public void Dispose()
+    {
+    }
+
+    public EntityReference GetEntityReference(EGID egid)
+    {
+        return entitiesDB is not null && egid != default ? egid.ToEntityReference(entitiesDB) : EntityReference.Invalid;
+    }
+
+    public EGID GetEgid(EntityReference reference)
+    {
+        return entitiesDB is not null && reference.ToEGID(entitiesDB, out var egid) ? egid : default;
+    }
+}

TechbloxModdingAPI/Common/Engines/IApiEngine.cs

@@ -1,7 +1,7 @@
 using System;
 using Svelto.ECS;
 
-namespace TechbloxModdingAPI.Common;
+namespace TechbloxModdingAPI.Common.Engines;
 
 /// <summary>
 /// Base engine interface used by all TechbloxModdingAPI engines

TechbloxModdingAPI/Common/Engines/INamedApiEngine.cs

@@ -1,4 +1,4 @@
-namespace TechbloxModdingAPI.Common;
+namespace TechbloxModdingAPI.Common.Engines;
 
 public interface INamedApiEngine : IApiEngine
 {

TechbloxModdingAPI/Common/Engines/IReactionaryEngine.cs

@@ -1,5 +1,5 @@
 using Svelto.ECS;
-using TechbloxModdingAPI.Common;
+using TechbloxModdingAPI.Common.Engines;
 
 namespace TechbloxModdingAPI.Engines
 {

TechbloxModdingAPI/Common/Utils/AsyncUtilsEngine.cs

@@ -1,9 +1,9 @@
 using System.Collections;
 using System.Threading.Tasks;
-using RobocraftX.Schedulers;
 using Svelto.ECS;
 using Svelto.Tasks;
 using Svelto.Tasks.ExtraLean;
+using TechbloxModdingAPI.Common.Engines;
 
 namespace TechbloxModdingAPI.Common.Utils;
 

TechbloxModdingAPI/Input/FakeInputEngine.cs

@@ -2,7 +2,7 @@
 using RobocraftX.Common.Input;
 using RobocraftX.Players;
 using Svelto.ECS;
-using TechbloxModdingAPI.Common;
+using TechbloxModdingAPI.Common.Engines;
 
 namespace TechbloxModdingAPI.Input
 {

TechbloxModdingAPI/Player.cs

@@ -13,9 +13,9 @@ using Techblox.Camera;
 using Techblox.Character;
 using TechbloxModdingAPI.Blocks;
 using TechbloxModdingAPI.Client.App;
+using TechbloxModdingAPI.Common;
 using TechbloxModdingAPI.Common.Engines;
 using TechbloxModdingAPI.Players;
-using TechbloxModdingAPI.Utility;
 using UnityEngine;
 
 namespace TechbloxModdingAPI

TechbloxModdingAPI/Players/PlayerEventsEngine.cs

@@ -2,7 +2,7 @@ using RobocraftX.Character;
 using RobocraftX.Character.Movement;
 using RobocraftX.Common.Input;
 using Svelto.ECS;
-using TechbloxModdingAPI.Common;
+using TechbloxModdingAPI.Common.Engines;
 
 namespace TechbloxModdingAPI.Players
 {

TechbloxModdingAPI/SimBody.cs

@@ -2,11 +2,10 @@
 using Svelto.ECS;
 using Unity.Mathematics;
 using UnityEngine;
-
-using Gamecraft.Damage;
 using RobocraftX.Common;
 using RobocraftX.Physics;
 using Techblox.TimeRunning.Clusters;
+using TechbloxModdingAPI.Common;
 
 namespace TechbloxModdingAPI
 {

TechbloxModdingAPI/Utility/DebugInterfaceEngine.cs

@@ -5,7 +5,7 @@ using System.Reflection.Emit;
 using System.Text;
 using HarmonyLib;
 using Svelto.ECS;
-using TechbloxModdingAPI.Common;
+using TechbloxModdingAPI.Common.Engines;
 
 namespace TechbloxModdingAPI.Utility
 {

TechbloxModdingAPI/Utility/ECS/ManagedApiExtensions.cs

@@ -1,5 +1,6 @@
 using Svelto.ECS;
 using Svelto.ECS.Hybrid;
+using TechbloxModdingAPI.Common;
 
 namespace TechbloxModdingAPI.Utility.ECS
 {

TechbloxModdingAPI/Utility/ECS/NativeApiExtensions.cs

@@ -1,9 +1,9 @@
 using System;
 using System.Collections.Generic;
-using Svelto.DataStructures;
 using Svelto.ECS;
 using Svelto.Tasks;
 using Svelto.Tasks.Lean;
+using TechbloxModdingAPI.Common;
 using TechbloxModdingAPI.Tasks;
 
 namespace TechbloxModdingAPI.Utility.ECS

TechbloxModdingAPI/Utility/GameStateEngine.cs

@@ -1,6 +1,6 @@
 using Svelto.ECS;
 using RobocraftX.SimulationModeState;
-using TechbloxModdingAPI.Common;
+using TechbloxModdingAPI.Common.Engines;
 
 namespace TechbloxModdingAPI.Utility
 {

TechbloxModdingAPI/Utility/OptionalRef.cs

@@ -1,6 +1,7 @@
 using System;
 using Svelto.DataStructures;
 using Svelto.ECS;
+using TechbloxModdingAPI.Common;
 
 namespace TechbloxModdingAPI.Utility
 {