Automation/gen_csproj.py

@@ -5,7 +5,7 @@ from pathlib import Path, PurePath
 import re
 import os
 
-DLL_EXCLUSIONS_REGEX = r"(System|Microsoft|Mono|IronPython|DiscordRPC|IllusionInjector|IllusionPlugin|netstandard)\."
+DLL_EXCLUSIONS_REGEX = r"(System|Microsoft|Mono|IronPython|DiscordRPC)\."
 
 def getAssemblyReferences(path):
     asmDir = Path(path)
@@ -15,12 +15,10 @@ def getAssemblyReferences(path):
         addedPath = "../"
         asmDir = Path(addedPath + path)
     for child in asmDir.iterdir():
-        if child.is_file() and re.search(DLL_EXCLUSIONS_REGEX, str(child)) is None and str(child).lower().endswith(".dll"):
+        if child.is_file() and re.search(DLL_EXCLUSIONS_REGEX, str(child), re.I) is None and str(child).lower().endswith(".dll"):
             childstr = str(child)
             childstr = os.path.relpath(childstr, addedPath).replace("\\", "/")
             result.append(childstr)
-    result.sort(key=str.lower)
-    result = [path + "/IllusionInjector.dll", path + "/IllusionPlugin.dll"] + result # Always put it on top
     return result
 
 def buildReferencesXml(path):
@@ -41,7 +39,7 @@ if __name__ == "__main__":
     args = parser.parse_args()
     
     print("Building Assembly references")
-    asmXml = buildReferencesXml("../ref_TB/Techblox_Data/Managed")
+    asmXml = buildReferencesXml("../ref/Techblox_Data/Managed")
     # print(asmXml)
     
     with open("../TechbloxModdingAPI/TechbloxModdingAPI.csproj", "r") as xmlFile:
@@ -53,7 +51,7 @@ if __name__ == "__main__":
         if depsStart is None or depsEnd is None:
             print("Unable to find dependency XML comments, aborting!")
             exit(1)
-        newFileStr = fileStr[:depsStart.start() - 1] + "\n" + asmXml + "\n" + fileStr[depsEnd.end() + 1:]
+        newFileStr = fileStr[:depsStart.start()] + "\n" + asmXml + "\n" + fileStr[depsEnd.end() + 1:]
     with open("../TechbloxModdingAPI/TechbloxModdingAPI.csproj", "w") as xmlFile:
         print("Writing Assembly references")
         xmlFile.write(newFileStr)

CodeGenerator/BlockClassGenerator.cs

@@ -22,7 +22,7 @@ namespace CodeGenerator
                     group = GetGroup(name) + "_BLOCK_BUILD_GROUP";
             }
 
-            if (!group.Contains('.'))
+            if (!group.Contains("."))
                 group = "CommonExclusiveGroups." + group;
 
             var codeUnit = new CodeCompileUnit();
@@ -67,7 +67,7 @@ namespace CodeGenerator
             codeUnit.Namespaces.Add(ns);
 
             var provider = CodeDomProvider.CreateProvider("CSharp");
-            var path = $@"../../../../TechbloxModdingAPI/Blocks/{name}.cs";
+            var path = $@"..\..\..\TechbloxModdingAPI\Blocks\{name}.cs";
             using (var sw = new StreamWriter(path))
             {
                 provider.GenerateCodeFromCompileUnit(codeUnit, sw, new CodeGeneratorOptions {BracingStyle = "C"});

CodeGenerator/Program.cs

@@ -1,12 +1,8 @@
 using System.Collections.Generic;
 using HarmonyLib;
 using RobocraftX.Blocks;
-using RobocraftX.Common;
-using RobocraftX.GroupTags;
 using RobocraftX.PilotSeat;
-using Svelto.ECS;
 using Techblox.EngineBlock;
-using Techblox.ServoBlocksServer;
 using Techblox.WheelRigBlock;
 
 namespace CodeGenerator
@@ -14,11 +10,6 @@ namespace CodeGenerator
     internal class Program
     {
         public static void Main(string[] args)
-        {
-            GenerateBlockClasses();
-        }
-
-        private static void GenerateBlockClasses()
         {
             var bcg = new BlockClassGenerator();
             bcg.Generate("Engine", null, new Dictionary<string, string>
@@ -32,7 +23,7 @@ namespace CodeGenerator
                 },
                 typeof(TweakableJointDampingComponent), typeof(DampedSpringReadOnlyStruct));
             bcg.Generate("LogicGate", "LOGIC_BLOCK_GROUP");
-            bcg.Generate("Servo", types: typeof(ServoReadOnlyTweakableComponent), renames: new Dictionary<string, string>
+            bcg.Generate("Servo", types: typeof(ServoReadOnlyStruct), renames: new Dictionary<string, string>
             {
                 {"minDeviation", "MinimumAngle"},
                 {"maxDeviation", "MaximumAngle"},
@@ -47,7 +38,6 @@ namespace CodeGenerator
                 {"pistonVelocity", "MaximumForce"}
             }, typeof(PistonReadOnlyStruct));
             bcg.Generate("Motor", null, null, typeof(MotorReadOnlyStruct));
-            //bcg.Generate("ObjectID", "ObjectIDBlockExclusiveGroups.OBJECT_ID_BLOCK_GROUP", null, typeof(ObjectIDTweakableComponent));
         }
     }
 }

CodeGenerator/Properties/AssemblyInfo.cs

@@ -0,0 +1,35 @@
+using System.Reflection;
+using System.Runtime.InteropServices;
+
+// General Information about an assembly is controlled through the following 
+// set of attributes. Change these attribute values to modify the information
+// associated with an assembly.
+[assembly: AssemblyTitle("CodeGenerator")]
+[assembly: AssemblyDescription("")]
+[assembly: AssemblyConfiguration("")]
+[assembly: AssemblyCompany("")]
+[assembly: AssemblyProduct("CodeGenerator")]
+[assembly: AssemblyCopyright("Copyright © ExMods 2021")]
+[assembly: AssemblyTrademark("")]
+[assembly: AssemblyCulture("")]
+
+// Setting ComVisible to false makes the types in this assembly not visible 
+// to COM components.  If you need to access a type in this assembly from 
+// COM, set the ComVisible attribute to true on that type.
+[assembly: ComVisible(false)]
+
+// The following GUID is for the ID of the typelib if this project is exposed to COM
+[assembly: Guid("0EBB6400-95A7-4A3D-B2ED-BF31E364CC10")]
+
+// Version information for an assembly consists of the following four values:
+//
+//      Major Version
+//      Minor Version 
+//      Build Number
+//      Revision
+//
+// You can specify all the values or you can default the Build and Revision Numbers 
+// by using the '*' as shown below:
+// [assembly: AssemblyVersion("1.0.*")]
+[assembly: AssemblyVersion("1.0.0.0")]
+[assembly: AssemblyFileVersion("1.0.0.0")]

MakeEverythingPublicInGame/Program.cs

@@ -1,25 +0,0 @@
-using System.Text.RegularExpressions;
-using Mono.Cecil;
-
-Console.WriteLine("Starting assembly editing...");
-var fileRegex =
-    new Regex(".*(Techblox|Gamecraft|RobocraftX|FullGame|RobocraftECS|DataLoader|RCX|GameState|Svelto\\.ECS)[^/]*(\\.dll)");
-foreach (var file in Directory.EnumerateFiles(@"../../../../../ref/Techblox_Data/Managed"))
-{
-    if (!fileRegex.IsMatch(file)) continue;
-    Console.WriteLine(file);
-    ProcessAssembly(file);
-}
-
-void ProcessAssembly(string path)
-{
-    using var mod = ModuleDefinition.ReadModule(path, new(ReadingMode.Immediate) { ReadWrite = true });
-    foreach (var typeDefinition in mod.Types)
-    {
-        typeDefinition.IsPublic = true;
-        foreach (var method in typeDefinition.Methods) method.IsPublic = true;
-        foreach (var field in typeDefinition.Fields) field.IsPublic = true;
-    }
-
-    mod.Write();
-}

TechbloxModdingAPI.sln

@@ -7,8 +7,6 @@ Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "TechbloxModdingAPI", "Techb
 EndProject
 Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "CodeGenerator", "CodeGenerator\CodeGenerator.csproj", "{0EBB6400-95A7-4A3D-B2ED-BF31E364CC10}"
 EndProject
-Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "MakeEverythingPublicInGame", "MakeEverythingPublicInGame\MakeEverythingPublicInGame.csproj", "{391A3107-E5C6-4A04-9467-6D868AA9A8B4}"
-EndProject
 Global
 	GlobalSection(SolutionConfigurationPlatforms) = preSolution
 		Debug|Any CPU = Debug|Any CPU
@@ -28,12 +26,6 @@ Global
 		{0EBB6400-95A7-4A3D-B2ED-BF31E364CC10}.Release|Any CPU.Build.0 = Release|Any CPU
 		{0EBB6400-95A7-4A3D-B2ED-BF31E364CC10}.Test|Any CPU.ActiveCfg = Debug|Any CPU
 		{0EBB6400-95A7-4A3D-B2ED-BF31E364CC10}.Test|Any CPU.Build.0 = Debug|Any CPU
-		{391A3107-E5C6-4A04-9467-6D868AA9A8B4}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
-		{391A3107-E5C6-4A04-9467-6D868AA9A8B4}.Debug|Any CPU.Build.0 = Debug|Any CPU
-		{391A3107-E5C6-4A04-9467-6D868AA9A8B4}.Release|Any CPU.ActiveCfg = Release|Any CPU
-		{391A3107-E5C6-4A04-9467-6D868AA9A8B4}.Release|Any CPU.Build.0 = Release|Any CPU
-		{391A3107-E5C6-4A04-9467-6D868AA9A8B4}.Test|Any CPU.ActiveCfg = Debug|Any CPU
-		{391A3107-E5C6-4A04-9467-6D868AA9A8B4}.Test|Any CPU.Build.0 = Debug|Any CPU
 	EndGlobalSection
 	GlobalSection(SolutionProperties) = preSolution
 		HideSolutionNode = FALSE

TechbloxModdingAPI/App/AntiAntiCheatPatch.cs

@@ -22,7 +22,7 @@ namespace TechbloxModdingAPI.App
             harmony.Patch(AccessTools.Method(type, "StartProtectedSession"), new HarmonyMethod(((AntiAnticheatDelegate) AntiAntiCheat).Method));
             harmony.Patch(AccessTools.Method(type, "StopProtectedSession"), new HarmonyMethod(((AntiAnticheatDelegateBool) AntiAntiCheat).Method));
             harmony.Patch(AccessTools.Method("Techblox.Services.Eos.Anticheat.Client.EosGetPendingMessagesToSendServiceRequest:Execute"), new HarmonyMethod(((AntiAnticheatDelegateTask)AntiAntiCheatTask).Method));
-            harmony.Patch(AccessTools.Method("Techblox.Anticheat.Client.Engines.ProcessEACViolationEngine:PollAnticheatStatus"), new HarmonyMethod(((AntiAnticheatDelegateTask)AntiAntiCheatTask).Method));
+            harmony.Patch(AccessTools.Method("Techblox.Anticheat.Client.Engines.ShowFeedbackDialogEngine:PollAnticheatStatus"), new HarmonyMethod(((AntiAnticheatDelegateTask)AntiAntiCheatTask).Method));
             harmony.Patch(AccessTools.Method(typeof(AnticheatClientCompositionRoot), "ClientComposeTimeRunning"), new HarmonyMethod(((Func<bool>)AntiAntiCheat).Method));
         }
 

TechbloxModdingAPI/App/Client.cs

@@ -114,6 +114,11 @@ namespace TechbloxModdingAPI.App
 	        popup.SecondButton();
         }
 
+        internal void CloseBetaPopup()
+        {
+	        Game.menuEngine.CloseBetaPopup();
+        }
+
         internal static void Init()
 		{
 			// this would have been so much simpler if this didn't involve a bunch of internal fields & classes

TechbloxModdingAPI/App/GameGameEngine.cs

@@ -140,19 +140,19 @@ namespace TechbloxModdingAPI.App
         {
 	        var allBlocks = entitiesDB.QueryEntities<BlockTagEntityStruct>();
 	        List<EGID> blockEGIDs = new List<EGID>();
-	        foreach (var ((_, ids, count), group) in allBlocks)
+	        foreach (var (blocks, _) in allBlocks)
 	        {
+		        var (buffer, count) = blocks.ToBuffer();
 		        for (int i = 0; i < count; i++)
 		        {
-			        var id = new EGID(ids[i], group);
 			        uint dbid;
 			        if (filter == BlockIDs.Invalid)
 				        dbid = (uint)filter;
 			        else
-				        dbid = entitiesDB.QueryEntity<DBEntityStruct>(id).DBID;
+				        dbid = entitiesDB.QueryEntity<DBEntityStruct>(buffer[i].ID).DBID;
-			        var ownership = entitiesDB.QueryEntity<BlockOwnershipComponent>(id).BlockOwnership;
+			        var ownership = entitiesDB.QueryEntity<BlockOwnershipComponent>(buffer[i].ID).BlockOwnership;
 			        if ((ownership & BlockOwnership.User) != 0 && dbid == (ulong)filter)
-				        blockEGIDs.Add(id);
+				        blockEGIDs.Add(buffer[i].ID);
 		        }
 	        }
 

TechbloxModdingAPI/App/GameMenuEngine.cs

@@ -55,12 +55,13 @@ namespace TechbloxModdingAPI.App
 
 		public Game[] GetMyGames()
 		{
-			var (mgsevs, count) = entitiesDB.QueryEntities<MyGameDataEntityStruct>(MyGamesScreenExclusiveGroups.MyGames);
+			EntityCollection<MyGameDataEntityStruct> mgsevs = entitiesDB.QueryEntities<MyGameDataEntityStruct>(MyGamesScreenExclusiveGroups.MyGames);
-			Game[] games = new Game[count];
+			var mgsevsB = mgsevs.ToBuffer().buffer;
-			for (int i = 0; i < count; i++)
+			Game[] games = new Game[mgsevs.count];
+			for (int i = 0; i < mgsevs.count; i++)
 			{
-				Utility.Logging.MetaDebugLog($"Found game named {mgsevs[i].GameName}");
+				Utility.Logging.MetaDebugLog($"Found game named {mgsevsB[i].GameName}");
-				games[i] = new Game(mgsevs[i].ID);
+				games[i] = new Game(mgsevsB[i].ID);
 			}
 			return games;
 		}
@@ -83,13 +84,14 @@ namespace TechbloxModdingAPI.App
 
 		public uint HighestID()
 		{
-			var (games, count) = entitiesDB.QueryEntities<MyGameDataEntityStruct>(MyGamesScreenExclusiveGroups.MyGames);
+			EntityCollection<MyGameDataEntityStruct> games = entitiesDB.QueryEntities<MyGameDataEntityStruct>(MyGamesScreenExclusiveGroups.MyGames);
+			var gamesB = games.ToBuffer().buffer;
 			uint max = 0;
-			for (int i = 0; i < count; i++)
+			for (int i = 0; i < games.count; i++)
 			{
-				if (games[i].ID.entityID > max)
+				if (gamesB[i].ID.entityID > max)
 				{
-					max = games[i].ID.entityID;
+					max = gamesB[i].ID.entityID;
 				}
 			}
 			return max;
@@ -154,6 +156,16 @@ namespace TechbloxModdingAPI.App
 		{
 			return ref entitiesDB.QueryEntity<T>(id);
 		}
+
+		internal void CloseBetaPopup()
+		{
+			var (buffer, count) = entitiesDB.QueryEntities<TogglePanelButtonEntityViewStruct>(ExclusiveGroup.Search("BetaPopup"));
+			for (int index = 0; index < count; ++index)
+			{
+				entitiesDB.QueryEntity<GUIEntityViewStruct>(buffer[index].TogglePanelButtonComponent.targetPanel)
+					.guiRoot.enabled = false;
+			}
+		}
 	}
 
 	internal class MyGameDataEntityDescriptor_DamnItFJWhyDidYouMakeThisInternal : GenericEntityDescriptor<MyGameDataEntityStruct> { }

TechbloxModdingAPI/Block.cs

@@ -11,7 +11,8 @@ using Unity.Mathematics;
 using HarmonyLib;
 using RobocraftX.PilotSeat;
 using RobocraftX.Rendering;
-using Techblox.BlockLabelsServer;
+using Techblox.BlockLabels;
+
 using TechbloxModdingAPI.Blocks;
 using TechbloxModdingAPI.Blocks.Engines;
 using TechbloxModdingAPI.Tests;
@@ -67,7 +68,7 @@ namespace TechbloxModdingAPI
 		/// <returns>The block object or null if doesn't exist</returns>
 		public static Block GetLastPlacedBlock()
 		{
-			uint lastBlockID = CommonExclusiveGroups.blockIDGeneratorClient.Peek() - 1;
+			uint lastBlockID = (uint) AccessTools.Field(typeof(CommonExclusiveGroups), "_nextBlockEntityID").GetValue(null) - 1;
 			EGID? egid = BlockEngine.FindBlockEGID(lastBlockID);
 			return egid.HasValue ? New(egid.Value) : null;
 		}
@@ -366,12 +367,8 @@ namespace TechbloxModdingAPI
 			get
 			{
 				if (blockGroup != null) return blockGroup;
-				if (!GameState.IsBuildMode()) return null; // Breaks in simulation
 				var bgec = BlockEngine.GetBlockInfo<BlockGroupEntityComponent>(this);
-				return blockGroup = bgec.currentBlockGroup == -1
+				return blockGroup = bgec.currentBlockGroup == -1 ? null : new BlockGroup(bgec.currentBlockGroup, this);
-					? null
-					: GetInstance(new EGID((uint)bgec.currentBlockGroup, BlockGroupExclusiveGroups.BlockGroupEntityGroup),
-						egid => new BlockGroup((int)egid.entityID, this));
 			}
 			set
 			{
@@ -399,23 +396,6 @@ namespace TechbloxModdingAPI
 			set => BlockEngine.GetBlockInfo<BlockStaticComponent>(this).isStatic = value;
 		}
 
-		/// <summary>
-		/// The mass of the block.
-		/// </summary>
-		public float Mass
-		{
-			get => BlockEngine.GetBlockInfo<MassStruct>(this).mass;
-		}
-
-		/// <summary>
-		/// Block complexity used for build rules. Determines the 'cost' of the block.
-		/// </summary>
-		public BlockComplexity Complexity
-		{
-			get => new(BlockEngine.GetBlockInfo<BlockComplexityComponent>(this));
-			set => BlockEngine.GetBlockInfo<BlockComplexityComponent>(this) = value;
-		}
-
 		/// <summary>
 		/// Whether the block exists. The other properties will return a default value if the block doesn't exist.
 		/// If the block was just placed, then this will also return false but the properties will work correctly.
@@ -441,10 +421,9 @@ namespace TechbloxModdingAPI
 		public SimBody GetSimBody()
 		{
 			var st = BlockEngine.GetBlockInfo<GridConnectionsEntityStruct>(this);
-			/*return st.machineRigidBodyId != uint.MaxValue
+			return st.machineRigidBodyId != uint.MaxValue
-				? new SimBody(st.machineRigidBodyId, st.clusterId) - TODO:
+				? new SimBody(st.machineRigidBodyId, st.clusterId)
-				: null;*/
+				: null;
-			return null;
 		}
 
 		/// <summary>

TechbloxModdingAPI/Blocks/BlockComplexity.cs

@@ -1,17 +0,0 @@
-using RobocraftX.Blocks;
-
-namespace TechbloxModdingAPI.Blocks
-{
-    public record BlockComplexity(int Cpu, int Power)
-    {
-        public BlockComplexity(BlockComplexityComponent component) : this(component.cpu, component.power)
-        {
-        }
-
-        public int Cpu { get; } = Cpu;
-        public int Power { get; } = Power;
-
-        public static implicit operator BlockComplexityComponent(BlockComplexity complexity) =>
-            new() { cpu = complexity.Cpu, power = complexity.Power };
-    }
-}

TechbloxModdingAPI/Blocks/BlockIDs.cs

@@ -274,110 +274,11 @@ namespace TechbloxModdingAPI.Blocks
         /// The grid block used by the world editor, named Small Grid like the other one
         /// </summary>
         SmallGridInWorldEditor,
-        CityDoubleCrossing,
+        SegoeUITextblock = 376,
-        CityDoubleCrossroads,
-        CitySmallDoubleJunction,
-        CityDoubleJunction,
-        CityDoubleToSingleJunction,
-        CitySmallDoubleRoad,
-        CityDoubleRoad,
-        CitySmallDoubleTurn,
-        CityLargeDoubleTurn,
-        CitySmallSingleTurn,
-        CityLargeSingleTurn,
-        CitySingleJunction,
-        CitySingleRoad,
-        SegoeUITextblock,
         GravtracTextblock,
         HauserTextblock,
         TechnopollasTextblock,
-        CityDoubleHillRoad,
+        BitBlock = 385,
-        DiagonalTrackTile,
+        Timer
-        DiagonalTrackTile2,
-        DiagonalTransitionTile,
-        SplitLane,
-        BitBlock,
-        Timer,
-        CityNightAtmosphere,
-        FloodLight,
-        SoccerBall,
-        CircularWallLight,
-        BlueSkyAtmos,
-        DirtToGrassTransitionTile = 393,
-        DirtToGrassTransitionInnerTile,
-        DirtToGrassTransitionOuterTile,
-        DirtToGrassTransitionHillTile,
-        DirtToGrassTransitionRoadTile,
-        DirtHill2 = 399,
-        DirtHill3,
-        DirtInnerCorner2 = 402,
-        DirtInnerCorner3,
-        DirtOuterCorner2 = 405,
-        DirtOuterCorner3,
-        CityTarmacEdgeInner,
-        CityTarmacEdgeOuter,
-        CityTarmacEdgeRoad,
-        CityTarmac,
-        SmallGrassQuarterTile,
-        CityToRacetrackTransition,
-        HUDTimer,
-        CentreHUD,
-        Checkpoint,
-        ScoreboardHUD,
-        GameplaySFX,
-        SpawnPoint,
-        AreaSensor,
-        WorldResetter,
-        SmallJet,
-        MediumJet,
-        LargeJet,
-        DistanceSensor,
-        Stabilizer,
-        ObjectID,
-        ScoreToTechpointConversion,
-        TeamScore,
-        ScorePickupBlock,
-        SportyHatchbackDriverSeat,
-        SportyHatchbackPassengerSeat,
-        FlamingExhaust = 433,
-        SmokingExhaust,
-        StreetLamp,
-        Vector7HatchbackWheel,
-        Vector7HatchbackWheelWideProfile,
-        Vector7SedanWheel,
-        Vector7SedanWideProfile,
-        Vector7FormulaWheel,
-        Vector7FormulaWheelRear,
-        Vector7MonsterTruckWheel,
-        Vector7TruckWheel,
-        Vector7TruckWheelDouble,
-        BusSeat,
-        XLJet,
-        XXLJet,
-        ElectricSedanEngine,
-        HeadlampIndicator,
-        HeadlampSrip,
-        HeadlampStripEdge,
-        ConstantBlock,
-        CounterBlock,
-        SmallGridHill,
-        SmallGridHillInnerCorner,
-        SmallGridHillOuterCorner,
-        AimingAxleServo,
-        AimingHingeServo,
-        WeaponDisabler,
-        Vector7SmallJet,
-        Vector7MediumJet,
-        Vector7LargeJet,
-        Vector7XLJet,
-        Vector7XXLJet,
-        APCWheelRigNoSteering,
-        APCWheelRigWithSteering,
-        APCWheel,
-        APCSeat,
-        APCEngine,
-        DamageScoreBlock,
-        KillScoreBlock,
-        Autocannon = 480
     }
 }

TechbloxModdingAPI/Blocks/BlockMaterial.cs

@@ -31,16 +31,5 @@ namespace TechbloxModdingAPI.Blocks
         SteelBodyworkPaintedChipped,
         WoodPainted,
         WoodRoughGrungy,
-        Boundary,
-        Emissive,
-        AircraftPanelingRivetedPainted,
-        AircraftPanelingRivetedMetallic,
-        SteelBodyworkPearlescent,
-        SteelBodyworkRadWrap,
-        SteelBodyworkGlitter,
-        BouncyRubber,
-        BouncyRubberTieDye,
-        BrickPainted,
-        FuturisticPanelingRivetedPainted,
     }
 }

TechbloxModdingAPI/Blocks/BlockTests.cs

@@ -5,8 +5,11 @@ using System.Reflection;
 
 using DataLoader;
 using Svelto.Tasks;
+using Svelto.Tasks.Enumerators;
 using Unity.Mathematics;
+using UnityEngine;
 
+using TechbloxModdingAPI.App;
 using TechbloxModdingAPI.Tests;
 using TechbloxModdingAPI.Utility;
 
@@ -79,50 +82,35 @@ namespace TechbloxModdingAPI.Blocks
 		        yield break;
 	        for (var index = 0; index < blocks.Length; index++)
 	        {
+		        if (index % 50 == 0) yield return new WaitForSecondsEnumerator(0.2f).Continue(); //The material or flipped status can only be changed 130 times per submission
 		        var block = blocks[index];
 		        if (!block.Exists) continue;
 		        foreach (var property in block.GetType().GetProperties())
 		        {
 			        //Includes specialised block properties
 			        if (property.SetMethod == null) continue;
-
+			        var testValues = new (Type, object, Predicate<object>)[]
-			        bool3 Float3Compare(float3 a, float3 b)
-			        { // From Unity reference code
-				        return math.abs(b - a) < math.max(
-					        0.000001f * math.max(math.abs(a), math.abs(b)),
-					        float.Epsilon * 8
-				        );
-			        }
-			        bool4 Float4Compare(float4 a, float4 b)
-			        { // From Unity reference code
-				        return math.abs(b - a) < math.max(
-					        0.000001f * math.max(math.abs(a), math.abs(b)),
-					        float.Epsilon * 8
-				        );
-			        }
-			        
-			        var testValues = new (Type, object, Predicate<(object Value, object Default)>)[]
 			        {
 				        //(type, default value, predicate or null for equality)
 				        (typeof(long), 3, null),
 				        (typeof(int), 4, null),
-				        (typeof(double), 5.2f, t => Math.Abs((double) t.Value - (double) t.Default) < float.Epsilon),
+				        (typeof(double), 5.2f, obj => Math.Abs((double) obj - 5.2f) < float.Epsilon),
-				        (typeof(float), 5.2f, t => Math.Abs((float) t.Value - (float) t.Default) < float.Epsilon),
+				        (typeof(float), 5.2f, obj => Math.Abs((float) obj - 5.2f) < float.Epsilon),
-				        (typeof(bool), true, t => (bool) t.Value),
+				        (typeof(bool), true, obj => (bool) obj),
-				        (typeof(string), "Test", t => (string) t.Value == "Test"), //String equality check
+				        (typeof(string), "Test", obj => (string) obj == "Test"), //String equality check
-				        (typeof(float3), (float3) 20, t => math.all(Float3Compare((float3)t.Value, (float3)t.Default))),
+				        (typeof(float3), (float3) 2, obj => math.all((float3) obj - 2 < (float3) float.Epsilon)),
 				        (typeof(BlockColor), new BlockColor(BlockColors.Aqua, 2), null),
-				        (typeof(float4), (float4) 5, t => math.all(Float4Compare((float4)t.Value, (float4)t.Default)))
+				        (typeof(float4), (float4) 5, obj => math.all((float4) obj - 5 < (float4) float.Epsilon))
 			        };
 			        var propType = property.PropertyType;
 			        if (!propType.IsValueType) continue;
-			        (object valueToUse, Predicate<(object Value, object Default)> predicateToUse) = (null, null);
+			        (object valueToUse, Predicate<object> predicateToUse) = (null, null);
 			        foreach (var (type, value, predicate) in testValues)
 			        {
 				        if (type.IsAssignableFrom(propType))
 				        {
 					        valueToUse = value;
-					        predicateToUse = predicate ?? (t => Equals(t.Value, t.Default));
+					        predicateToUse = predicate ?? (obj => Equals(obj, value));
 					        break;
 				        }
 			        }
@@ -131,7 +119,7 @@ namespace TechbloxModdingAPI.Blocks
 			        {
 				        var values = propType.GetEnumValues();
 				        valueToUse = values.GetValue(values.Length / 2);
-				        predicateToUse = t => Equals(t.Value, t.Default);
+				        predicateToUse = val => Equals(val, valueToUse);
 			        }
 
 			        if (valueToUse == null)
@@ -159,7 +147,7 @@ namespace TechbloxModdingAPI.Blocks
 				        continue;
 			        }
 			        var attr = property.GetCustomAttribute<TestValueAttribute>();
-			        if (!predicateToUse((got, valueToUse)) && (attr == null || !Equals(attr.PossibleValue, got)))
+			        if (!predicateToUse(got) && (attr == null || !Equals(attr.PossibleValue, got)))
 			        {
 				        Assert.Fail($"Property {block.GetType().Name}.{property.Name} value {got} does not equal {valueToUse} for block {block}.");
 				        yield break;

TechbloxModdingAPI/Blocks/Engine.cs

@@ -23,18 +23,18 @@ namespace TechbloxModdingAPI.Blocks
         {
         }
         
-        /*/// <summary> - TODO: Internal struct access
+        /// <summary>
         /// Gets or sets the Engine's On property. May not be saved.
         /// </summary>
         public bool On
         {
             get
             {
-                return BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).engineOn;
+                return ((bool)(BlockEngine.GetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "engineOn")));
             }
             set
             {
-                BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).engineOn = value;
+                BlockEngine.SetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "engineOn", value);
             }
         }
         
@@ -45,11 +45,11 @@ namespace TechbloxModdingAPI.Blocks
         {
             get
             {
-                return BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).currentGear;
+                return ((int)(BlockEngine.GetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "currentGear")));
             }
             set
             {
-                BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).currentGear = value;
+                BlockEngine.SetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "currentGear", value);
             }
         }
         
@@ -60,11 +60,11 @@ namespace TechbloxModdingAPI.Blocks
         {
             get
             {
-                return BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).gearChangeCountdown;
+                return ((float)(BlockEngine.GetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "gearChangeCountdown")));
             }
             set
             {
-                BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).gearChangeCountdown = value;
+                BlockEngine.SetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "gearChangeCountdown", value);
             }
         }
         
@@ -75,11 +75,11 @@ namespace TechbloxModdingAPI.Blocks
         {
             get
             {
-                return BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).currentRpmAV;
+                return ((float)(BlockEngine.GetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "currentRpmAV")));
             }
             set
             {
-                BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).currentRpmAV = value;
+                BlockEngine.SetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "currentRpmAV", value);
             }
         }
         
@@ -90,11 +90,11 @@ namespace TechbloxModdingAPI.Blocks
         {
             get
             {
-                return BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).currentRpmLV;
+                return ((float)(BlockEngine.GetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "currentRpmLV")));
             }
             set
             {
-                BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).currentRpmLV = value;
+                BlockEngine.SetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "currentRpmLV", value);
             }
         }
         
@@ -105,11 +105,11 @@ namespace TechbloxModdingAPI.Blocks
         {
             get
             {
-                return BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).targetRpmAV;
+                return ((float)(BlockEngine.GetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "targetRpmAV")));
             }
             set
             {
-                BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).targetRpmAV = value;
+                BlockEngine.SetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "targetRpmAV", value);
             }
         }
         
@@ -120,11 +120,11 @@ namespace TechbloxModdingAPI.Blocks
         {
             get
             {
-                return BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).targetRpmLV;
+                return ((float)(BlockEngine.GetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "targetRpmLV")));
             }
             set
             {
-                BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).targetRpmLV = value;
+                BlockEngine.SetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "targetRpmLV", value);
             }
         }
         
@@ -135,11 +135,11 @@ namespace TechbloxModdingAPI.Blocks
         {
             get
             {
-                return BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).currentTorque;
+                return ((float)(BlockEngine.GetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "currentTorque")));
             }
             set
             {
-                BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).currentTorque = value;
+                BlockEngine.SetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "currentTorque", value);
             }
         }
         
@@ -150,11 +150,11 @@ namespace TechbloxModdingAPI.Blocks
         {
             get
             {
-                return BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).totalWheelVelocityAV;
+                return ((float)(BlockEngine.GetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "totalWheelVelocityAV")));
             }
             set
             {
-                BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).totalWheelVelocityAV = value;
+                BlockEngine.SetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "totalWheelVelocityAV", value);
             }
         }
         
@@ -165,11 +165,11 @@ namespace TechbloxModdingAPI.Blocks
         {
             get
             {
-                return BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).totalWheelVelocityLV;
+                return ((float)(BlockEngine.GetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "totalWheelVelocityLV")));
             }
             set
             {
-                BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).totalWheelVelocityLV = value;
+                BlockEngine.SetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "totalWheelVelocityLV", value);
             }
         }
         
@@ -180,11 +180,11 @@ namespace TechbloxModdingAPI.Blocks
         {
             get
             {
-                return BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).totalWheelCount;
+                return ((int)(BlockEngine.GetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "totalWheelCount")));
             }
             set
             {
-                BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).totalWheelCount = value;
+                BlockEngine.SetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "totalWheelCount", value);
             }
         }
         
@@ -195,11 +195,11 @@ namespace TechbloxModdingAPI.Blocks
         {
             get
             {
-                return BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).lastGearUpInput;
+                return ((bool)(BlockEngine.GetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "lastGearUpInput")));
             }
             set
             {
-                BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).lastGearUpInput = value;
+                BlockEngine.SetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "lastGearUpInput", value);
             }
         }
         
@@ -210,11 +210,11 @@ namespace TechbloxModdingAPI.Blocks
         {
             get
             {
-                return BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).lastGearDownInput;
+                return ((bool)(BlockEngine.GetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "lastGearDownInput")));
             }
             set
             {
-                BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).lastGearDownInput = value;
+                BlockEngine.SetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "lastGearDownInput", value);
             }
         }
         
@@ -225,11 +225,11 @@ namespace TechbloxModdingAPI.Blocks
         {
             get
             {
-                return BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).manualToAutoGearCoolOffCounter;
+                return ((float)(BlockEngine.GetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "manualToAutoGearCoolOffCounter")));
             }
             set
             {
-                BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).manualToAutoGearCoolOffCounter = value;
+                BlockEngine.SetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "manualToAutoGearCoolOffCounter", value);
             }
         }
         
@@ -240,11 +240,11 @@ namespace TechbloxModdingAPI.Blocks
         {
             get
             {
-                return BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).load;
+                return ((float)(BlockEngine.GetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "load")));
             }
             set
             {
-                BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockComponent>(this).load = value;
+                BlockEngine.SetBlockInfo(this, HarmonyLib.AccessTools.TypeByName("Techblox.EngineBlock.EngineBlockComponent"), "load", value);
             }
         }
         
@@ -377,6 +377,6 @@ namespace TechbloxModdingAPI.Blocks
             {
                 BlockEngine.GetBlockInfo<Techblox.EngineBlock.EngineBlockReadonlyComponent>(this).manualToAutoGearCoolOffTime = value;
             }
-        }*/
+        }
     }
 }

TechbloxModdingAPI/Blocks/Engines/BlockEngine.cs

@@ -1,9 +1,11 @@
 using System;
 using System.Collections.Generic;
 using System.Linq;
+using System.Reflection;
 using HarmonyLib;
 
 using Gamecraft.ColourPalette;
+using Gamecraft.TimeRunning;
 using Gamecraft.Wires;
 using RobocraftX.Blocks;
 using RobocraftX.Common;
@@ -13,16 +15,12 @@ using RobocraftX.Rendering.GPUI;
 using Svelto.DataStructures;
 using Svelto.ECS;
 using Svelto.ECS.EntityStructs;
-using Svelto.ECS.Experimental;
 using Svelto.ECS.Hybrid;
 using Techblox.BuildingDrone;
-using Techblox.ObjectIDBlockServer;
 using Unity.Mathematics;
 
 using TechbloxModdingAPI.Engines;
 using TechbloxModdingAPI.Utility;
-using TechbloxModdingAPI.Utility.ECS;
-using PrefabsID = RobocraftX.Common.PrefabsID;
 
 namespace TechbloxModdingAPI.Blocks.Engines
 {
@@ -47,21 +45,22 @@ namespace TechbloxModdingAPI.Blocks.Engines
 
 		public Block[] GetConnectedBlocks(EGID blockID)
 		{
-			if (!BlockExists(blockID)) return Array.Empty<Block>();
+			if (!BlockExists(blockID)) return new Block[0];
 			Stack<EGID> cubeStack = new Stack<EGID>();
 			FasterList<EGID> cubes = new FasterList<EGID>(10);
 			var coll = entitiesDB.QueryEntities<GridConnectionsEntityStruct>();
-			foreach (var ((ecoll, count), _) in coll)
+			foreach (var (ecoll, _) in coll)
 			{
-				for(int i = 0; i < count; i++)
+				var ecollB = ecoll.ToBuffer();
+				for(int i = 0; i < ecoll.count; i++)
 				{
-					ecoll[i].areConnectionsAssigned = false;
+					ref var conn = ref ecollB.buffer[i];
+					conn.isProcessed = false;
 				}
 			}
 
-			//TODO: GetConnectedCubesUtility
+			ConnectedCubesUtility.TreeTraversal.GetConnectedCubes(entitiesDB, blockID, cubeStack, cubes,
-			/*ConnectedCubesUtility.TreeTraversal.GetConnectedCubes(entitiesDB, blockID, cubeStack, cubes,
+				(in GridConnectionsEntityStruct g) => { return false; });
-				(in GridConnectionsEntityStruct _) => false);*/
 
 			var ret = new Block[cubes.count];
 			for (int i = 0; i < cubes.count; i++)
@@ -72,7 +71,7 @@ namespace TechbloxModdingAPI.Blocks.Engines
 		public float4 ConvertBlockColor(byte index) => index == byte.MaxValue
 			? new float4(-1f, -1f, -1f, -1f)
 			: entitiesDB.QueryEntity<PaletteEntryEntityStruct>(index,
-				ColourPaletteExclusiveGroups.COLOUR_PALETTE_GROUP).Colour;
+				CommonExclusiveGroups.COLOUR_PALETTE_GROUP).Colour;
 
 		public OptionalRef<T> GetBlockInfoOptional<T>(Block block) where T : unmanaged, IEntityComponent
 		{
@@ -100,21 +99,19 @@ namespace TechbloxModdingAPI.Blocks.Engines
 
 		internal object GetBlockInfo(Block block, Type type, string name)
 		{
-			var opt = AccessTools.Method(typeof(NativeApiExtensions), "QueryEntityOptional",
+			/*var opt = AccessTools.Method(typeof(BlockEngine), "GetBlockInfoOptional", generics: new[] { type })
-					new[] { typeof(EntitiesDB), typeof(EcsObjectBase), typeof(ExclusiveGroupStruct) }, new[] { type })
+				.Invoke(this, new object[] { block }); - TODO: Cannot call method with by-ref return value
-				.Invoke(null, new object[] { entitiesDB, block, null });
+			var str = AccessTools.Method(opt.GetType(), "Get").Invoke(opt, Array.Empty<object>());
-			var str = AccessTools.Property(opt.GetType(), "Value").GetValue(opt);
+			return AccessTools.Field(str.GetType(), name).GetValue(str);*/
-			return AccessTools.Field(str.GetType(), name).GetValue(str);
+			return AccessTools.Field(type, name).GetValue(Activator.CreateInstance(type));
 		}
 
 		internal void SetBlockInfo(Block block, Type type, string name, object value)
 		{
-			var opt = AccessTools.Method(typeof(BlockEngine), "GetBlockInfoOptional", generics: new[] { type })
+			/*var opt = AccessTools.Method(typeof(BlockEngine), "GetBlockInfoOptional", generics: new[] { type })
 				.Invoke(this, new object[] { block });
-			var prop = AccessTools.Property(opt.GetType(), "Value");
+			var str = AccessTools.Method(opt.GetType(), "Get").Invoke(opt, Array.Empty<object>());
-			var str = prop.GetValue(opt);
+			AccessTools.Field(str.GetType(), name).SetValue(str, value);*/
-			AccessTools.Field(str.GetType(), name).SetValue(str, value);
-			prop.SetValue(opt, str);
 		}
 
 		public void UpdateDisplayedBlock(EGID id)
@@ -143,7 +140,7 @@ namespace TechbloxModdingAPI.Blocks.Engines
 			}
 
 			uint prefabId =
-				PrefabsID.GetOrAddPrefabID((ushort) prefabAssetID, material, 1, flipped);
+				PrefabsID.GetOrCreatePrefabID((ushort) prefabAssetID, material, 1, flipped);
 			entitiesDB.QueryEntityOrDefault<GFXPrefabEntityStructGPUI>(block).prefabID = prefabId;
 			if (block.Exists)
 			{
@@ -162,7 +159,7 @@ namespace TechbloxModdingAPI.Blocks.Engines
 
 		public void UpdateBlockColor(EGID id)
 		{
-			entitiesDB.PublishEntityChangeDelayed<ColourParameterEntityStruct>(id);
+			entitiesDB.PublishEntityChange<ColourParameterEntityStruct>(id);
 		}
 
 		public bool BlockExists(EGID blockID)
@@ -173,22 +170,26 @@ namespace TechbloxModdingAPI.Blocks.Engines
 		public SimBody[] GetSimBodiesFromID(byte id)
 		{
 			var ret = new FasterList<SimBody>(4);
-			var oids = entitiesDB.QueryEntitiesOptional<ObjectIdEntityStruct>(ObjectIDBlockExclusiveGroups.OBJECT_ID_BLOCK_GROUP);
+			var oide = entitiesDB.QueryEntities<ObjectIdEntityStruct>();
 			EGIDMapper<GridConnectionsEntityStruct>? connections = null;
-			foreach (var oid in oids)
+			foreach (var ((oids, count), _) in oide)
 			{
-				if (oid.Get().objectId != id) continue;
+				for (int i = 0; i < count; i++)
+				{	
+					ref ObjectIdEntityStruct oid = ref oids[i];
+					if (oid.objectId != id) continue;
 					if (!connections.HasValue) //Would need reflection to get the group from the build group otherwise
-					connections = entitiesDB.QueryMappedEntities<GridConnectionsEntityStruct>(oid.EGID.groupID);
+						connections = entitiesDB.QueryMappedEntities<GridConnectionsEntityStruct>(oid.ID.groupID);
-				//var rid = connections.Value.Entity(tag.ID.entityID).machineRigidBodyId;
+					var rid = connections.Value.Entity(oid.ID.entityID).machineRigidBodyId;
-				/*foreach (var rb in ret) - TODO
+					foreach (var rb in ret)
 					{
 						if (rb.Id.entityID == rid)
 							goto DUPLICATE; //Multiple Object Identifiers on one rigid body
 					}
 
 					ret.Add(new SimBody(rid));
-				DUPLICATE: ;*/
+					DUPLICATE: ;
+				}
 			}
 
 			return ret.ToArray();
@@ -196,14 +197,14 @@ namespace TechbloxModdingAPI.Blocks.Engines
 
 		public SimBody[] GetConnectedSimBodies(uint id)
 		{
-			var (joints, count) = entitiesDB.QueryEntities<JointEntityStruct>(MachineSimulationGroups.JOINTS_GROUP);
+			var joints = entitiesDB.QueryEntities<JointEntityStruct>(MachineSimulationGroups.JOINTS_GROUP).ToBuffer();
 			var list = new FasterList<SimBody>(4);
-			for (int i = 0; i < count; i++)
+			for (int i = 0; i < joints.count; i++)
 			{
-				ref var joint = ref joints[i];
+				ref var joint = ref joints.buffer[i];
 				if (joint.isBroken) continue;
-				/*if (joint.connectedEntityA == id) list.Add(new SimBody(joint.connectedEntityB)); - TODO:
+				if (joint.connectedEntityA == id) list.Add(new SimBody(joint.connectedEntityB));
-				else if (joint.connectedEntityB == id) list.Add(new SimBody(joint.connectedEntityA));*/
+				else if (joint.connectedEntityB == id) list.Add(new SimBody(joint.connectedEntityA));
 			}
 
 			return list.ToArray();
@@ -213,13 +214,14 @@ namespace TechbloxModdingAPI.Blocks.Engines
 		{
 			var groups = entitiesDB.QueryEntities<GridConnectionsEntityStruct>();
 			var bodies = new HashSet<uint>();
-			foreach (var ((coll, count), _) in groups)
+			foreach (var (coll, _) in groups)
 			{
-				for (var index = 0; index < count; index++)
+				var array = coll.ToBuffer().buffer;
+				for (var index = 0; index < array.capacity; index++)
 				{
-					var conn = coll[index];
+					var conn = array[index];
-					/*if (conn.clusterId == cid) - TODO
+					if (conn.clusterId == cid)
-						bodies.Add(conn.machineRigidBodyId);*/
+						bodies.Add(conn.machineRigidBodyId);
 				}
 			}
 
@@ -241,14 +243,15 @@ namespace TechbloxModdingAPI.Blocks.Engines
 		public Cluster GetCluster(uint sbid)
 		{
 			var groups = entitiesDB.QueryEntities<GridConnectionsEntityStruct>();
-			foreach (var ((coll, count), _) in groups)
+			foreach (var (coll, _) in groups)
 			{
-				for (var index = 0; index < count; index++)
+				var array = coll.ToBuffer().buffer;
+				for (var index = 0; index < array.capacity; index++)
 				{
-					var conn = coll[index];
+					var conn = array[index];
 					//Static blocks don't have a cluster ID but the cluster destruction manager should have one
-					/*if (conn.machineRigidBodyId == sbid && conn.clusterId != uint.MaxValue) - TODO:
+					if (conn.machineRigidBodyId == sbid && conn.clusterId != uint.MaxValue)
-						return new Cluster(conn.clusterId);*/
+						return new Cluster(conn.clusterId);
 				}
 			}
 
@@ -257,36 +260,49 @@ namespace TechbloxModdingAPI.Blocks.Engines
 
 		public Block[] GetBodyBlocks(uint sbid)
 		{
-			var groups = entitiesDB.FindGroups<GridConnectionsEntityStruct>();
+			var groups = entitiesDB.QueryEntities<GridConnectionsEntityStruct>();
-			groups = new QueryGroups(groups).Except(CommonExclusiveGroups.DISABLED_JOINTS_IN_SIM_GROUP).Evaluate().result;
 			var set = new HashSet<Block>();
-			foreach (var ((coll, tags, count), _) in entitiesDB.QueryEntities<GridConnectionsEntityStruct, BlockTagEntityStruct>(groups))
+			foreach (var (coll, _) in groups)
 			{
-				for (var index = 0; index < count; index++)
+				var array = coll.ToBuffer().buffer;
+				for (var index = 0; index < array.capacity; index++)
 				{
-					var conn = coll[index];
+					var conn = array[index];
-					/*if (conn.machineRigidBodyId == sbid) - TODO
+					if (conn.machineRigidBodyId == sbid)
-						set.Add(Block.New(tags[index].ID));*/
+						set.Add(Block.New(conn.ID));
 				}
 			}
 
 			return set.ToArray();
 		}
 
-		public ObjectID[] GetObjectIDsFromID(byte id)
+#if DEBUG
+		public EntitiesDB GetEntitiesDB()
 		{
-			if (!entitiesDB.HasAny<ObjectIDTweakableComponent>(ObjectIDBlockExclusiveGroups.OBJECT_ID_BLOCK_GROUP))
+			return entitiesDB;
-				return Array.Empty<ObjectID>();
+		}
+#endif
 
-			var ret = new FasterList<ObjectID>(4);
+		[HarmonyPatch]
-			var oids = entitiesDB.QueryEntitiesOptional<ObjectIDTweakableComponent>(ObjectIDBlockExclusiveGroups.OBJECT_ID_BLOCK_GROUP);
+		public static class RenderingPatch
-			foreach (var oid in oids)
 		{
-				if (oid.Get().objectIDToTrigger == id)
+			private static ComputeRenderingEntitiesMatricesEngine Engine;
-					ret.Add(new ObjectID(oid.EGID));
+
+			public static void Postfix(ComputeRenderingEntitiesMatricesEngine __instance)
+			{
+				Engine = __instance;
 			}
 
-			return ret.ToArray();
+			public static MethodBase TargetMethod()
+			{
+				return typeof(ComputeRenderingEntitiesMatricesEngine).GetConstructors()[0];
+			}
+
+			public static void UpdateBlocks()
+			{
+				var data = new RenderingDataStruct();
+				Engine.Add(ref data, new EGID(0, CommonExclusiveGroups.BUTTON_BLOCK_GROUP));
+			}
 		}
 	}
 }

TechbloxModdingAPI/Blocks/Engines/BlueprintEngine.cs

@@ -18,10 +18,9 @@ using Svelto.ECS.DataStructures;
 using Svelto.ECS.EntityStructs;
 using Svelto.ECS.Native;
 using Svelto.ECS.Serialization;
-using Techblox.Blocks.Connections;
+using Techblox.Blocks;
 using TechbloxModdingAPI.Engines;
 using TechbloxModdingAPI.Utility;
-using TechbloxModdingAPI.Utility.ECS;
 using Unity.Collections;
 using Unity.Mathematics;
 using UnityEngine;
@@ -49,8 +48,7 @@ namespace TechbloxModdingAPI.Blocks.Engines
         private static readonly MethodInfo SerializeGhostBlueprint =
             AccessTools.Method(SerializeGhostBlueprintType, "SerializeClipboardGhostEntities");
 
-        private static NativeEntityRemove nativeBlockRemove;
+        private static NativeEntityRemove nativeRemove;
-        private static NativeEntityRemove nativeConnectionRemove;
         private static MachineGraphConnectionEntityFactory connectionFactory;
         private static IEntityFunctions entityFunctions;
         private static ClipboardSerializationDataResourceManager clipboardManager;
@@ -90,8 +88,8 @@ namespace TechbloxModdingAPI.Blocks.Engines
 
         public void RemoveBlockGroup(int id)
         {
-            BlockGroupUtility.RemoveAllBlocksInBlockGroup(id, entitiesDB, removedConnections, nativeBlockRemove,
+            BlockGroupUtility.RemoveAllBlocksInBlockGroup(id, entitiesDB, removedConnections, nativeRemove,
-                nativeConnectionRemove, connectionFactory, default).Complete();
+                connectionFactory, default).Complete();
         }
 
         public int CreateBlockGroup(float3 position, quaternion rotation)
@@ -279,7 +277,7 @@ namespace TechbloxModdingAPI.Blocks.Engines
                 FullGameFields._managers.blockLabelResourceManager);
             entityInitializer.Init(dbStruct);
             entityInitializer.Init(new GFXPrefabEntityStructGPUI(
-                PrefabsID.GetOrAddPrefabID((ushort)entityInitializer.Get<PrefabAssetIDComponent>().prefabAssetID,
+                PrefabsID.GetOrCreatePrefabID((ushort)entityInitializer.Get<PrefabAssetIDComponent>().prefabAssetID,
                     entitiesDB.QueryEntity<CubeMaterialStruct>(sourceId).materialId, 7,
                     FlippedBlockUtils.IsFlipped(in scalingEntityStruct.scale)), true));
             entityInitializer.Init(entitiesDB.QueryEntity<CubeMaterialStruct>(sourceId));
@@ -341,8 +339,7 @@ namespace TechbloxModdingAPI.Blocks.Engines
             public static void Prefix(IEntityFunctions entityFunctions,
                 MachineGraphConnectionEntityFactory machineGraphConnectionEntityFactory)
             {
-                nativeBlockRemove = entityFunctions.ToNativeRemove<BlockEntityDescriptor>("TBAPI" + nameof(BlueprintEngine));
+                nativeRemove = entityFunctions.ToNativeRemove<BlockEntityDescriptor>("GCAPI" + nameof(BlueprintEngine));
-                nativeConnectionRemove = entityFunctions.ToNativeRemove<MachineConnectionEntityDescriptor>("TBAPI" + nameof(BlueprintEngine));
                 connectionFactory = machineGraphConnectionEntityFactory;
                 BlueprintEngine.entityFunctions = entityFunctions;
             }

TechbloxModdingAPI/Blocks/Engines/MovementEngine.cs

@@ -7,7 +7,6 @@ using Unity.Transforms;
 
 using TechbloxModdingAPI.Engines;
 using TechbloxModdingAPI.Utility;
-using TechbloxModdingAPI.Utility.ECS;
 
 namespace TechbloxModdingAPI.Blocks.Engines
 {
@@ -41,7 +40,7 @@ namespace TechbloxModdingAPI.Blocks.Engines
             ref PositionEntityStruct posStruct = ref this.entitiesDB.QueryEntityOrDefault<PositionEntityStruct>(block);
             ref GridRotationStruct gridStruct = ref this.entitiesDB.QueryEntityOrDefault<GridRotationStruct>(block);
             ref LocalTransformEntityStruct transStruct = ref this.entitiesDB.QueryEntityOrDefault<LocalTransformEntityStruct>(block);
-            var phyStruct = this.entitiesDB.QueryEntityOptional<DOTSPhysicsEntityStruct>(block);
+            ref DOTSPhysicsEntityStruct phyStruct = ref this.entitiesDB.QueryEntityOrDefault<DOTSPhysicsEntityStruct>(block);
             // main (persistent) position
             posStruct.position = vector;
             // placement grid  position
@@ -49,15 +48,15 @@ namespace TechbloxModdingAPI.Blocks.Engines
             // rendered position
             transStruct.position = vector;
             // collision position
-            if (phyStruct)
+            if (phyStruct.ID != default)
             { //It exists
-                FullGameFields._physicsWorld.EntityManager.SetComponentData(phyStruct.Get().dotsEntity, new Translation
+                FullGameFields._physicsWorld.EntityManager.SetComponentData(phyStruct.dotsEntity, new Translation
                 {
                     Value = posStruct.position
                 });
             }
 
-            entitiesDB.QueryEntityOrDefault<GridConnectionsEntityStruct>(block).areConnectionsAssigned = false;
+            entitiesDB.QueryEntityOrDefault<GridConnectionsEntityStruct>(block).isProcessed = false;
             return posStruct.position;
         }
 

TechbloxModdingAPI/Blocks/Engines/PlacementEngine.cs

@@ -16,7 +16,6 @@ using Unity.Mathematics;
 
 using TechbloxModdingAPI.Engines;
 using TechbloxModdingAPI.Utility;
-using TechbloxModdingAPI.Utility.ECS;
 
 namespace TechbloxModdingAPI.Blocks.Engines
 {
@@ -59,7 +58,7 @@ namespace TechbloxModdingAPI.Blocks.Engines
             uint prefabAssetID = structInitializer.Has<PrefabAssetIDComponent>()
                 ? structInitializer.Get<PrefabAssetIDComponent>().prefabAssetID
                 : throw new BlockException("Prefab asset ID not found!"); //Set by the game
-            uint prefabId = PrefabsID.GetOrAddPrefabID((ushort) prefabAssetID, (byte) BlockMaterial.SteelBodywork, 1, false);
+            uint prefabId = PrefabsID.GetOrCreatePrefabID((ushort) prefabAssetID, (byte) BlockMaterial.SteelBodywork, 1, false);
             structInitializer.Init(new GFXPrefabEntityStructGPUI(prefabId));
             structInitializer.Init(dbEntity);
             structInitializer.Init(new PositionEntityStruct {position = position});

TechbloxModdingAPI/Blocks/Engines/RemovalEngine.cs

@@ -1,61 +1,47 @@
 using System.Reflection;
 
-using Gamecraft.Blocks.BlockGroups;
 using HarmonyLib;
 using RobocraftX.Blocks;
 using RobocraftX.Common;
-using RobocraftX.GroupTags;
+using Svelto.Common;
-using RobocraftX.StateSync;
 using Svelto.ECS;
 using Svelto.ECS.Native;
-using Techblox.Blocks.Connections;
-using Unity.Collections;
-using Unity.Jobs;
-using Allocator = Unity.Collections.Allocator;
 
 using TechbloxModdingAPI.Engines;
 using TechbloxModdingAPI.Utility;
 
 namespace TechbloxModdingAPI.Blocks.Engines
 {
-    public class RemovalEngine : IApiEngine, IDeterministicTimeStopped
+    public class RemovalEngine : IApiEngine
     {
         private static IEntityFunctions _entityFunctions;
         private static MachineGraphConnectionEntityFactory _connectionFactory;
-        private NativeHashSet<ulong> removedConnections;
 
         public bool RemoveBlock(EGID target)
         {
             if (!entitiesDB.Exists<MachineGraphConnectionsEntityStruct>(target))
                 return false;
-            using var connStructMapper =
+            var connections = entitiesDB.QueryEntity<MachineGraphConnectionsEntityStruct>(target);
-                entitiesDB.QueryNativeMappedEntities<MachineGraphConnectionsEntityStruct>(GroupTag<BLOCK_TAG>.Groups,
+            var groups = entitiesDB.FindGroups<MachineGraphConnectionsEntityStruct>();
-                    Svelto.Common.Allocator.Temp);
+            using var connStructMapper = //The allocator needs to be persistent because that's what is used in the Dispose() method
-            if (entitiesDB.TryQueryNativeMappedEntities<MachineConnectionComponent>(
+                entitiesDB.QueryNativeMappedEntities<MachineGraphConnectionsEntityStruct>(groups, Allocator.Persistent);
-                    ConnectionsExclusiveGroups.MACHINE_CONNECTION_GROUP, out var mapper))
+            for (int i = connections.connections.Count<MachineConnectionStruct>() - 1; i >= 0; i--)
-            {
+                _connectionFactory.RemoveConnection(connections, i, connStructMapper);
-                BlockGroupUtility.RemoveBlockConnections(target, removedConnections, _connectionFactory,
-                    connStructMapper, mapper, entitiesDB.GetEntityReferenceMap(), _entityFunctions);
-            }
-
             _entityFunctions.RemoveEntity<BlockEntityDescriptor>(target);
             return true;
         }
 
         public void Ready()
         {
-            removedConnections = new(2000, Allocator.Persistent);
         }
 
         public EntitiesDB entitiesDB { get; set; }
 
         public void Dispose()
         {
-            removedConnections.Dispose();
         }
 
-        public string Name => "TechbloxModdingAPIRemovalGameEngine";
+        public string Name { get; } = "TechbloxModdingAPIRemovalGameEngine";
-        public string name => Name;
 
 		public bool isRemovable => false;
 
@@ -75,12 +61,5 @@ namespace TechbloxModdingAPI.Blocks.Engines
                 return AccessTools.TypeByName("RobocraftX.CR.MachineEditing.RemoveBlockEngine").GetConstructors()[0];
             }
         }
-
-        public JobHandle DeterministicStep(in float deltaTime, JobHandle inputDeps)
-        {
-            if (removedConnections.IsCreated)
-                removedConnections.Clear();
-            return default;
-        }
     }
 }

TechbloxModdingAPI/Blocks/Engines/RotationEngine.cs

@@ -7,7 +7,6 @@ using UnityEngine;
 
 using TechbloxModdingAPI.Engines;
 using TechbloxModdingAPI.Utility;
-using TechbloxModdingAPI.Utility.ECS;
 
 namespace TechbloxModdingAPI.Blocks.Engines
 {
@@ -41,7 +40,7 @@ namespace TechbloxModdingAPI.Blocks.Engines
             ref RotationEntityStruct rotStruct = ref this.entitiesDB.QueryEntityOrDefault<RotationEntityStruct>(block);
             ref GridRotationStruct gridStruct = ref this.entitiesDB.QueryEntityOrDefault<GridRotationStruct>(block);
             ref LocalTransformEntityStruct transStruct = ref this.entitiesDB.QueryEntityOrDefault<LocalTransformEntityStruct>(block);
-            var phyStruct = this.entitiesDB.QueryEntityOptional<DOTSPhysicsEntityStruct>(block);
+            ref DOTSPhysicsEntityStruct phyStruct = ref this.entitiesDB.QueryEntityOrDefault<DOTSPhysicsEntityStruct>(block);
             // main (persistent) rotation
             Quaternion newRotation = rotStruct.rotation;
             newRotation.eulerAngles = vector;
@@ -51,18 +50,16 @@ namespace TechbloxModdingAPI.Blocks.Engines
             // rendered rotation
             transStruct.rotation = newRotation;
             // collision rotation
-            if (phyStruct)
+            if (phyStruct.ID != default)
             { //It exists
-                FullGameFields._physicsWorld.EntityManager.SetComponentData(phyStruct.Get().dotsEntity,
+                FullGameFields._physicsWorld.EntityManager.SetComponentData(phyStruct.dotsEntity,
                     new Unity.Transforms.Rotation
                     {
                         Value = rotStruct.rotation
                     });
             }
 
-            // TODO: Connections probably need to be assigned (maybe)
+            entitiesDB.QueryEntityOrDefault<GridConnectionsEntityStruct>(block).isProcessed = false;
-            // They are assigned during machine processing anyway
-            entitiesDB.QueryEntityOrDefault<GridConnectionsEntityStruct>(block).areConnectionsAssigned = false;
             return ((Quaternion)rotStruct.rotation).eulerAngles;
 
         }

TechbloxModdingAPI/Blocks/Engines/SignalEngine.cs

@@ -6,7 +6,6 @@ using Svelto.ECS;
 
 using TechbloxModdingAPI.Engines;
 using TechbloxModdingAPI.Utility;
-using TechbloxModdingAPI.Utility.ECS;
 
 namespace TechbloxModdingAPI.Blocks.Engines
 {
@@ -42,7 +41,7 @@ namespace TechbloxModdingAPI.Blocks.Engines
 
         // implementations for block wiring
 
-        public (WireEntityStruct Wire, EGID ID) CreateNewWire(EGID startBlock, byte startPort, EGID endBlock, byte endPort)
+        public WireEntityStruct CreateNewWire(EGID startBlock, byte startPort, EGID endBlock, byte endPort)
         {
 	        EGID wireEGID = new EGID(BuildModeWiresGroups.NewWireEntityId, BuildModeWiresGroups.WiresGroup.Group);
 	        EntityInitializer wireInitializer = Factory.BuildEntity<WireEntityDescriptor>(wireEGID);
@@ -51,9 +50,10 @@ namespace TechbloxModdingAPI.Blocks.Engines
 				sourceBlockEGID = startBlock,
 				sourcePortUsage = startPort,
 				destinationBlockEGID = endBlock,
-				destinationPortUsage = endPort
+				destinationPortUsage = endPort,
+				ID = wireEGID
 			});
-	        return (wireInitializer.Get<WireEntityStruct>(), wireEGID);
+	        return wireInitializer.Get<WireEntityStruct>();
         }
 
         public ref WireEntityStruct GetWire(EGID wire)
@@ -116,8 +116,9 @@ namespace TechbloxModdingAPI.Blocks.Engines
 
         public bool SetSignal(uint signalID, float signal, bool input = true)
         {
-	        var (array, count) = GetSignalStruct(signalID, out uint index, input);
+	        var array = GetSignalStruct(signalID, out uint index, input);
-	        if (count > 0) array[index].valueAsFloat = signal;
+	        var arrayB = array.ToBuffer();
+	        if (array.count > 0) arrayB.buffer[index].valueAsFloat = signal;
             return false;
         }
 
@@ -129,10 +130,11 @@ namespace TechbloxModdingAPI.Blocks.Engines
 
         public float AddSignal(uint signalID, float signal, bool clamp = true, bool input = true)
         {
-	        var (array, count) = GetSignalStruct(signalID, out uint index, input);
+	        var array = GetSignalStruct(signalID, out uint index, input);
-	        if (count > 0)
+	        var arrayB = array.ToBuffer();
+	        if (array.count > 0)
 	        {
-		        ref var channelData = ref array[index];
+		        ref var channelData = ref arrayB.buffer[index];
 		        channelData.valueAsFloat += signal;
 		        if (clamp)
 		        {
@@ -160,8 +162,9 @@ namespace TechbloxModdingAPI.Blocks.Engines
 
         public float GetSignal(uint signalID, bool input = true)
         {
-	        var (array, count) = GetSignalStruct(signalID, out uint index, input);
+	        var array = GetSignalStruct(signalID, out uint index, input);
-	        return count > 0 ? array[index].valueAsFloat : 0f;
+	        var arrayB = array.ToBuffer();
+	        return array.count > 0 ? arrayB.buffer[index].valueAsFloat : 0f;
         }
 
         public uint[] GetSignalIDs(EGID blockID, bool input = true)
@@ -226,27 +229,28 @@ namespace TechbloxModdingAPI.Blocks.Engines
 				uint entityID = (output ? ports.firstOutputID : ports.firstInputID) + i;
 				if (!mapper.TryGetArrayAndEntityIndex(entityID, out var index, out var array) ||
 				    array[index].usage != portUsage) continue;
-				return new OptionalRef<PortEntityStruct>(array, index, new EGID(entityID, group));
+				return new OptionalRef<PortEntityStruct>(array, index);
 			}
 
 			return default;
 		}
 
-		public OptionalRef<WireEntityStruct> MatchPortToWire(PortEntityStruct port, EGID blockID, out EGID wireID)
+		public ref WireEntityStruct MatchPortToWire(PortEntityStruct port, EGID blockID, out bool exists)
 		{
-			var (wires, ids, count) = entitiesDB.QueryEntities<WireEntityStruct>(NamedExclusiveGroup<BuildModeWiresGroups.WiresGroup>.Group);
+			var wires = entitiesDB.QueryEntities<WireEntityStruct>(NamedExclusiveGroup<BuildModeWiresGroups.WiresGroup>.Group);
-			for (uint i = 0; i < count; i++)
+			var wiresB = wires.ToBuffer().buffer;
+			for (uint i = 0; i < wires.count; i++)
 			{
-				if ((wires[i].destinationPortUsage == port.usage && wires[i].destinationBlockEGID == blockID)
+				if ((wiresB[i].destinationPortUsage == port.usage && wiresB[i].destinationBlockEGID == blockID)
-				    || (wires[i].sourcePortUsage == port.usage && wires[i].sourceBlockEGID == blockID))
+				    || (wiresB[i].sourcePortUsage == port.usage && wiresB[i].sourceBlockEGID == blockID))
 				{
-					wireID = new EGID(ids[i], BuildModeWiresGroups.WiresGroup.Group);
+					exists = true;
-					return new OptionalRef<WireEntityStruct>(wires, i);
+					return ref wiresB[i];
 				}
 			}
-
+			exists = false;
-			wireID = default;
+			WireEntityStruct[] defRef = new WireEntityStruct[1];
-			return default;
+            return ref defRef[0];
 		}
 
 		public EGID MatchBlocksToWire(EGID startBlock, EGID endBlock, byte startPort = byte.MaxValue, byte endPort = byte.MaxValue)
@@ -275,20 +279,20 @@ namespace TechbloxModdingAPI.Blocks.Engines
 				endPorts = new EGID[] {new EGID(ports.firstInputID + endPort, NamedExclusiveGroup<BuildModeWiresGroups.InputPortsGroup>.Group) };
 			}
 			
+			EntityCollection<WireEntityStruct> wires = entitiesDB.QueryEntities<WireEntityStruct>(NamedExclusiveGroup<BuildModeWiresGroups.WiresGroup>.Group);
+			var wiresB = wires.ToBuffer().buffer;
 			for (int endIndex = 0; endIndex < endPorts.Length; endIndex++)
 			{
 				PortEntityStruct endPES = entitiesDB.QueryEntity<PortEntityStruct>(endPorts[endIndex]);
 				for (int startIndex = 0; startIndex < startPorts.Length; startIndex++)
 				{
 					PortEntityStruct startPES = entitiesDB.QueryEntity<PortEntityStruct>(startPorts[startIndex]);
-					foreach (var wireOpt in entitiesDB.QueryEntitiesOptional<WireEntityStruct>(
+					for (int w = 0; w < wires.count; w++)
-						         NamedExclusiveGroup<BuildModeWiresGroups.WiresGroup>.Group))
 					{
-						var wire = wireOpt.Get();
+						if ((wiresB[w].destinationPortUsage == endPES.usage && wiresB[w].destinationBlockEGID == endBlock)
-						if ((wire.destinationPortUsage == endPES.usage && wire.destinationBlockEGID == endBlock)
+						    && (wiresB[w].sourcePortUsage == startPES.usage && wiresB[w].sourceBlockEGID == startBlock))
-						    && (wire.sourcePortUsage == startPES.usage && wire.sourceBlockEGID == startBlock))
 						{
-							return wireOpt.EGID;
+							return wiresB[w].ID;
 						}
 					}
 				}
@@ -300,21 +304,23 @@ namespace TechbloxModdingAPI.Blocks.Engines
         public OptionalRef<ChannelDataStruct> GetChannelDataStruct(EGID portID)
         {
 	        var port = GetPort(portID);
-			var (channels, count) = entitiesDB.QueryEntities<ChannelDataStruct>(NamedExclusiveGroup<BuildModeWiresGroups.ChannelDataGroup>.Group);
+			var channels = entitiesDB.QueryEntities<ChannelDataStruct>(NamedExclusiveGroup<BuildModeWiresGroups.ChannelDataGroup>.Group);
-			return port.firstChannelIndexCachedInSim < count
+			var channelsB = channels.ToBuffer();
-				? new OptionalRef<ChannelDataStruct>(channels, port.firstChannelIndexCachedInSim)
+			return port.firstChannelIndexCachedInSim < channels.count
+				? new OptionalRef<ChannelDataStruct>(channelsB.buffer, port.firstChannelIndexCachedInSim)
 				: default;
 		}
 
         public EGID[] GetElectricBlocks()
         {
 	        var res = new FasterList<EGID>();
-            foreach (var ((coll, ids, count), _) in entitiesDB.QueryEntities<BlockPortsStruct>())
+            foreach (var (coll, _) in entitiesDB.QueryEntities<BlockPortsStruct>())
             {
-	            for (int i = 0; i < count; i++)
+	            var collB = coll.ToBuffer();
+	            for (int i = 0; i < coll.count; i++)
 	            {
-		            ref BlockPortsStruct s = ref coll[i];
+		            ref BlockPortsStruct s = ref collB.buffer[i];
-		            //res.Add(s.ID); - TODO: Would need to search for the groups for each block
+		            res.Add(s.ID);
 	            }
             }
 
@@ -323,18 +329,43 @@ namespace TechbloxModdingAPI.Blocks.Engines
 
         public EGID[] WiredToInput(EGID block, byte port)
         {
-	        return entitiesDB
+	        WireEntityStruct[] wireEntityStructs = Search(NamedExclusiveGroup<BuildModeWiresGroups.WiresGroup>.Group,
-		        .QueryEntitiesOptional<WireEntityStruct>(NamedExclusiveGroup<BuildModeWiresGroups.WiresGroup>.Group)
+		        (WireEntityStruct wes) => wes.destinationPortUsage == port && wes.destinationBlockEGID == block);
-		        .ToArray(wire => wire.ID,
+	        EGID[] result = new EGID[wireEntityStructs.Length];
-			        wire => wire.Component.destinationPortUsage == port && wire.Component.destinationBlockEGID == block);
+	        for (uint i = 0; i < wireEntityStructs.Length; i++)
+	        {
+		        result[i] = wireEntityStructs[i].ID;
+	        }
+
+	        return result;
         }
         
         public EGID[] WiredToOutput(EGID block, byte port)
         {
-	        return entitiesDB
+	        WireEntityStruct[] wireEntityStructs = Search(NamedExclusiveGroup<BuildModeWiresGroups.WiresGroup>.Group,
-		        .QueryEntitiesOptional<WireEntityStruct>(NamedExclusiveGroup<BuildModeWiresGroups.WiresGroup>.Group)
+		        (WireEntityStruct wes) => wes.sourcePortUsage == port && wes.sourceBlockEGID == block);
-		        .ToArray(wire => wire.ID,
+	        EGID[] result = new EGID[wireEntityStructs.Length];
-			        wire => wire.Component.sourcePortUsage == port && wire.Component.sourceBlockEGID == block);
+	        for (uint i = 0; i < wireEntityStructs.Length; i++)
+	        {
+		        result[i] = wireEntityStructs[i].ID;
+	        }
+
+	        return result;
+        }
+
+        private T[] Search<T>(ExclusiveGroup group, Func<T, bool> isMatch) where T : unmanaged, IEntityComponent
+        {
+	        FasterList<T> results = new FasterList<T>();
+	        EntityCollection<T> components = entitiesDB.QueryEntities<T>(group);
+	        var componentsB = components.ToBuffer();
+	        for (uint i = 0; i < components.count; i++)
+	        {
+		        if (isMatch(componentsB.buffer[i]))
+		        {
+			        results.Add(componentsB.buffer[i]);
+		        }
+	        }
+	        return results.ToArray();
         }
 
         private EntityCollection<ChannelDataStruct> GetSignalStruct(uint signalID, out uint index, bool input = true)
@@ -344,7 +375,7 @@ namespace TechbloxModdingAPI.Blocks.Engines
 		        : NamedExclusiveGroup<BuildModeWiresGroups.OutputPortsGroup>.Group;
 	        if (entitiesDB.Exists<PortEntityStruct>(signalID, group))
 	        {
-		        index = entitiesDB.QueryEntity<PortEntityStruct>(signalID, group).firstChannelIndexCachedInSim;
+		        index = entitiesDB.QueryEntity<PortEntityStruct>(signalID, group).anyChannelIndex;
 		        var channelData =
 			        entitiesDB.QueryEntities<ChannelDataStruct>(NamedExclusiveGroup<BuildModeWiresGroups.ChannelDataGroup>.Group);
 		        return channelData;

TechbloxModdingAPI/Blocks/ObjectID.cs

@@ -1,50 +0,0 @@
-using System;
-using Techblox.ObjectIDBlockServer;
-
-namespace TechbloxModdingAPI.Blocks
-{
-    using Svelto.ECS;
-    
-    
-    public class ObjectID : SignalingBlock
-    {
-        
-        /// <summary>
-        /// Constructs a(n) ObjectID object representing an existing block.
-        /// </summary>
-        public ObjectID(EGID egid) : 
-                base(egid)
-        {
-        }
-        
-        /// <summary>
-        /// Constructs a(n) ObjectID object representing an existing block.
-        /// </summary>
-        public ObjectID(uint id) : 
-                base(new EGID(id, ObjectIDBlockExclusiveGroups.OBJECT_ID_BLOCK_GROUP))
-        {
-        }
-        
-        /// <summary>
-        /// Gets or sets the ObjectID's Identifier property. Tweakable stat.
-        /// </summary>
-        public char Identifier
-        {
-            get => (char) (BlockEngine.GetBlockInfo<ObjectIDTweakableComponent>(this).objectIDToTrigger + 'A');
-            set
-            {
-                if(value is < 'A' or > 'Z')
-                    throw new ArgumentOutOfRangeException(nameof(value), "ObjectIdentifier must be set to a letter between A and Z.");
-                BlockEngine.GetBlockInfo<ObjectIDTweakableComponent>(this).objectIDToTrigger = (byte) (value - 'A');
-                Label = value + ""; //The label isn't updated automatically
-            }
-        }
-
-        /// <summary>
-        /// Finds the identifier blocks with the given ID.
-        /// </summary>
-        /// <param name="id">The ID to look for</param>
-        /// <returns>An array that may be empty</returns>
-        public static ObjectID[] GetByID(char id) => BlockEngine.GetObjectIDsFromID((byte) (id - 'A'));
-    }
-}

TechbloxModdingAPI/Blocks/Servo.cs

@@ -30,11 +30,11 @@ namespace TechbloxModdingAPI.Blocks
         {
             get
             {
-                return BlockEngine.GetBlockInfo<Techblox.ServoBlocksServer.ServoReadOnlyTweakableComponent>(this).servoVelocity;
+                return BlockEngine.GetBlockInfo<RobocraftX.Blocks.ServoReadOnlyStruct>(this).servoVelocity;
             }
             set
             {
-                BlockEngine.GetBlockInfo<Techblox.ServoBlocksServer.ServoReadOnlyTweakableComponent>(this).servoVelocity = value;
+                BlockEngine.GetBlockInfo<RobocraftX.Blocks.ServoReadOnlyStruct>(this).servoVelocity = value;
             }
         }
         
@@ -45,11 +45,11 @@ namespace TechbloxModdingAPI.Blocks
         {
             get
             {
-                return BlockEngine.GetBlockInfo<Techblox.ServoBlocksServer.ServoReadOnlyTweakableComponent>(this).minDeviation;
+                return BlockEngine.GetBlockInfo<RobocraftX.Blocks.ServoReadOnlyStruct>(this).minDeviation;
             }
             set
             {
-                BlockEngine.GetBlockInfo<Techblox.ServoBlocksServer.ServoReadOnlyTweakableComponent>(this).minDeviation = value;
+                BlockEngine.GetBlockInfo<RobocraftX.Blocks.ServoReadOnlyStruct>(this).minDeviation = value;
             }
         }
         
@@ -60,11 +60,11 @@ namespace TechbloxModdingAPI.Blocks
         {
             get
             {
-                return BlockEngine.GetBlockInfo<Techblox.ServoBlocksServer.ServoReadOnlyTweakableComponent>(this).maxDeviation;
+                return BlockEngine.GetBlockInfo<RobocraftX.Blocks.ServoReadOnlyStruct>(this).maxDeviation;
             }
             set
             {
-                BlockEngine.GetBlockInfo<Techblox.ServoBlocksServer.ServoReadOnlyTweakableComponent>(this).maxDeviation = value;
+                BlockEngine.GetBlockInfo<RobocraftX.Blocks.ServoReadOnlyStruct>(this).maxDeviation = value;
             }
         }
         
@@ -75,11 +75,11 @@ namespace TechbloxModdingAPI.Blocks
         {
             get
             {
-                return BlockEngine.GetBlockInfo<Techblox.ServoBlocksServer.ServoReadOnlyTweakableComponent>(this).reverse;
+                return BlockEngine.GetBlockInfo<RobocraftX.Blocks.ServoReadOnlyStruct>(this).reverse;
             }
             set
             {
-                BlockEngine.GetBlockInfo<Techblox.ServoBlocksServer.ServoReadOnlyTweakableComponent>(this).reverse = value;
+                BlockEngine.GetBlockInfo<RobocraftX.Blocks.ServoReadOnlyStruct>(this).reverse = value;
             }
         }
         
@@ -90,11 +90,11 @@ namespace TechbloxModdingAPI.Blocks
         {
             get
             {
-                return BlockEngine.GetBlockInfo<Techblox.ServoBlocksServer.ServoReadOnlyTweakableComponent>(this).hasProportionalInput;
+                return BlockEngine.GetBlockInfo<RobocraftX.Blocks.ServoReadOnlyStruct>(this).hasProportionalInput;
             }
             set
             {
-                BlockEngine.GetBlockInfo<Techblox.ServoBlocksServer.ServoReadOnlyTweakableComponent>(this).hasProportionalInput = value;
+                BlockEngine.GetBlockInfo<RobocraftX.Blocks.ServoReadOnlyStruct>(this).hasProportionalInput = value;
             }
         }
         
@@ -105,11 +105,11 @@ namespace TechbloxModdingAPI.Blocks
         {
             get
             {
-                return BlockEngine.GetBlockInfo<Techblox.ServoBlocksServer.ServoReadOnlyTweakableComponent>(this).directionVector;
+                return BlockEngine.GetBlockInfo<RobocraftX.Blocks.ServoReadOnlyStruct>(this).directionVector;
             }
             set
             {
-                BlockEngine.GetBlockInfo<Techblox.ServoBlocksServer.ServoReadOnlyTweakableComponent>(this).directionVector = value;
+                BlockEngine.GetBlockInfo<RobocraftX.Blocks.ServoReadOnlyStruct>(this).directionVector = value;
             }
         }
         
@@ -120,11 +120,11 @@ namespace TechbloxModdingAPI.Blocks
         {
             get
             {
-                return BlockEngine.GetBlockInfo<Techblox.ServoBlocksServer.ServoReadOnlyTweakableComponent>(this).rotationAxis;
+                return BlockEngine.GetBlockInfo<RobocraftX.Blocks.ServoReadOnlyStruct>(this).rotationAxis;
             }
             set
             {
-                BlockEngine.GetBlockInfo<Techblox.ServoBlocksServer.ServoReadOnlyTweakableComponent>(this).rotationAxis = value;
+                BlockEngine.GetBlockInfo<RobocraftX.Blocks.ServoReadOnlyStruct>(this).rotationAxis = value;
             }
         }
         
@@ -135,11 +135,11 @@ namespace TechbloxModdingAPI.Blocks
         {
             get
             {
-                return BlockEngine.GetBlockInfo<Techblox.ServoBlocksServer.ServoReadOnlyTweakableComponent>(this).forceAxis;
+                return BlockEngine.GetBlockInfo<RobocraftX.Blocks.ServoReadOnlyStruct>(this).forceAxis;
             }
             set
             {
-                BlockEngine.GetBlockInfo<Techblox.ServoBlocksServer.ServoReadOnlyTweakableComponent>(this).forceAxis = value;
+                BlockEngine.GetBlockInfo<RobocraftX.Blocks.ServoReadOnlyStruct>(this).forceAxis = value;
             }
         }
     }

TechbloxModdingAPI/Blocks/SignalingBlock.cs

@@ -46,9 +46,9 @@ namespace TechbloxModdingAPI.Blocks
         /// <returns>The connected wire.</returns>
         /// <param name="portId">Port identifier.</param>
         /// <param name="connected">Whether the port has a wire connected to it.</param>
-        protected OptionalRef<WireEntityStruct> GetConnectedWire(PortEntityStruct port, out EGID egid)
+        protected ref WireEntityStruct GetConnectedWire(PortEntityStruct port, out bool connected)
 		{
-			return SignalEngine.MatchPortToWire(port, Id, out egid);
+			return ref SignalEngine.MatchPortToWire(port, Id, out connected);
 		}
 
         /// <summary>

TechbloxModdingAPI/Blocks/WheelRig.cs

@@ -40,21 +40,6 @@ namespace TechbloxModdingAPI.Blocks
             }
         }
         
-        /// <summary>
-        /// Gets or sets the WheelRig's FlipDirection property. Tweakable stat.
-        /// </summary>
-        public bool FlipDirection
-        {
-            get
-            {
-                return BlockEngine.GetBlockInfo<Techblox.WheelRigBlock.WheelRigTweakableStruct>(this).flipDirection;
-            }
-            set
-            {
-                BlockEngine.GetBlockInfo<Techblox.WheelRigBlock.WheelRigTweakableStruct>(this).flipDirection = value;
-            }
-        }
-        
         /// <summary>
         /// Gets or sets the WheelRig's MaxVelocity property. May not be saved.
         /// </summary>
@@ -86,22 +71,6 @@ namespace TechbloxModdingAPI.Blocks
             }
         }
         
-        /// <summary>
-        /// Gets or sets the WheelRig's FlipSteering property. Tweakable stat.
-        /// </summary>
-        [TestValue(false)]
-        public bool FlipSteering
-        {
-            get
-            {
-                return BlockEngine.GetBlockInfo<Techblox.WheelRigBlock.WheelRigSteerableTweakableStruct>(this).flipSteering;
-            }
-            set
-            {
-                BlockEngine.GetBlockInfo<Techblox.WheelRigBlock.WheelRigSteerableTweakableStruct>(this).flipSteering = value;
-            }
-        }
-        
         /// <summary>
         /// Gets or sets the WheelRig's VelocityForMinAngle property. May not be saved.
         /// </summary>

TechbloxModdingAPI/Blocks/Wire.cs

@@ -5,7 +5,6 @@ using Svelto.ECS;
 using Svelto.ECS.Experimental;
 
 using TechbloxModdingAPI.Blocks.Engines;
-using TechbloxModdingAPI.Utility;
 
 namespace TechbloxModdingAPI.Blocks
 {
@@ -31,8 +30,8 @@ namespace TechbloxModdingAPI.Blocks
 
         public static Wire Connect(SignalingBlock start, byte startPort, SignalingBlock end, byte endPort)
         {
-            var (wire, id) = signalEngine.CreateNewWire(start.Id, startPort, end.Id, endPort);
+            WireEntityStruct wire = signalEngine.CreateNewWire(start.Id, startPort, end.Id, endPort);
-            return new Wire(wire, start, end, id);
+            return new Wire(wire, start, end);
         }
 
         /// <summary>
@@ -46,9 +45,9 @@ namespace TechbloxModdingAPI.Blocks
         {
             var port = signalEngine.MatchBlockIOToPort(end, endPort, false);
             if (!port) return null;
-            var wire = signalEngine.MatchPortToWire(port, end.Id, out var egid);
+            WireEntityStruct wire = signalEngine.MatchPortToWire(port, end.Id, out var exists);
-            return wire
+            return exists
-                ? new Wire(wire.Get().sourceBlockEGID, end.Id, wire.Get().sourcePortUsage, endPort, egid, false)
+                ? new Wire(wire.sourceBlockEGID, end.Id, wire.sourcePortUsage, endPort, wire.ID, false)
                 : null;
         }
 
@@ -63,9 +62,9 @@ namespace TechbloxModdingAPI.Blocks
         {
             var port = signalEngine.MatchBlockIOToPort(start, startPort, true);
             if (!port) return null;
-            var wire = signalEngine.MatchPortToWire(port, start.Id, out var egid);
+            WireEntityStruct wire = signalEngine.MatchPortToWire(port, start.Id, out var exists);
-            return wire
+            return exists
-                ? new Wire(start.Id, wire.Get().destinationBlockEGID, startPort, wire.Get().destinationPortUsage, egid, false)
+                ? new Wire(start.Id, wire.destinationBlockEGID, startPort, wire.destinationPortUsage, wire.ID, false)
                 : null;
         }
 
@@ -133,9 +132,9 @@ namespace TechbloxModdingAPI.Blocks
             this.endBlockEGID = endBlock;
             this.inputToOutput = inputToOutput;
             this.wireEGID = wire;
-            endPortEGID = signalEngine.MatchBlockIOToPort(startBlock, startPort, inputToOutput).EGID; 
+            endPortEGID = signalEngine.MatchBlockIOToPort(startBlock, startPort, inputToOutput).Nullable()?.ID ?? default; 
             if (endPortEGID == default) throw new WireInvalidException("Wire end port not found");
-            startPortEGID = signalEngine.MatchBlockIOToPort(endBlock, endPort, !inputToOutput).EGID;
+            startPortEGID = signalEngine.MatchBlockIOToPort(endBlock, endPort, !inputToOutput).Nullable()?.ID ?? default;
             if (startPortEGID == default) throw new WireInvalidException("Wire start port not found");
             this.startPort = startPort;
             this.endPort = endPort;
@@ -152,8 +151,8 @@ namespace TechbloxModdingAPI.Blocks
                 wireEgid, false);
         }
 
-        private Wire(WireEntityStruct wire, SignalingBlock src, SignalingBlock dest, EGID wireEgid)
+        private Wire(WireEntityStruct wire, SignalingBlock src, SignalingBlock dest)
-            : this(src, dest, wire.sourcePortUsage, wire.destinationPortUsage, wireEgid, false)
+            : this(src, dest, wire.sourcePortUsage, wire.destinationPortUsage, wire.ID, false)
         {
         }
 

TechbloxModdingAPI/Cluster.cs

@@ -1,5 +1,6 @@
-using Svelto.ECS;
+using Gamecraft.Damage;
-using Techblox.TimeRunning.Clusters;
+using RobocraftX.Common;
+using Svelto.ECS;
 
 namespace TechbloxModdingAPI
 {
@@ -13,33 +14,28 @@ namespace TechbloxModdingAPI
         {
         }
 
-        public Cluster(uint id) : this(new EGID(id, ClustersExclusiveGroups.SIMULATION_CLUSTERS_GROUP))
+        public Cluster(uint id) : this(new EGID(id, CommonExclusiveGroups.SIMULATION_CLUSTERS_GROUP))
         {
         }
 
-        public float InitialHealth //TODO
+        public float InitialHealth
         {
-            get => 0f;
+            get => Block.BlockEngine.GetBlockInfo<HealthEntityComponent>(this).initialHealth;
-            set { }
+            set => Block.BlockEngine.GetBlockInfo<HealthEntityComponent>(this).initialHealth = value;
         }
 
         public float CurrentHealth
         {
-            get => 0f;
+            get => Block.BlockEngine.GetBlockInfo<HealthEntityComponent>(this).currentHealth;
-            set { }
+            set => Block.BlockEngine.GetBlockInfo<HealthEntityComponent>(this).currentHealth = value;
         }
 
         public float HealthMultiplier
         {
-            get => 0f;
+            get => Block.BlockEngine.GetBlockInfo<HealthEntityComponent>(this).healthMultiplier;
-            set { }
+            set => Block.BlockEngine.GetBlockInfo<HealthEntityComponent>(this).healthMultiplier = value;
         }
 
-        /// <summary>
-        /// The mass of the cluster.
-        /// </summary>
-        public float Mass => Block.BlockEngine.GetBlockInfo<ClusterMassComponent>(this).mass;
-
         /// <summary>
         /// Returns the simulation-time rigid bodies for the chunks in this cluster.
         /// </summary>

TechbloxModdingAPI/Interface/IMGUI/Button.cs

@@ -24,7 +24,7 @@ namespace TechbloxModdingAPI.Interface.IMGUI
         /// </summary>
         public event EventHandler<bool> OnClick;
         
-        public override void OnGUI()
+        public void OnGUI()
         {
             if (automaticLayout)
             {
@@ -42,15 +42,34 @@ namespace TechbloxModdingAPI.Interface.IMGUI
             }
         }
 
+        /// <summary>
+        /// The button's unique name.
+        /// </summary>
+        public string Name { get; private set; }
+
+        /// <summary>
+        /// Whether to display the button.
+        /// </summary>
+        public bool Enabled { get; set; } = true;
+
         /// <summary>
         /// Initialize a new button with automatic layout.
         /// </summary>
         /// <param name="text">The text to display on the button.</param>
         /// <param name="name">The button's name.</param>
-        public Button(string text, string name = null) : base(text, name)
+        public Button(string text, string name = null)
         {
             automaticLayout = true;
             this.text = text;
+            if (name == null)
+            {
+                this.Name = typeof(Button).FullName + "::" + text;
+            }
+            else
+            {
+                this.Name = name;
+            }
+            IMGUIManager.AddElement(this);
         }
 
         /// <summary>
@@ -64,5 +83,10 @@ namespace TechbloxModdingAPI.Interface.IMGUI
             automaticLayout = false;
             this.Box = box;
         }
+        
+        ~Button()
+        {
+            IMGUIManager.RemoveElement(this);
+        }
     }
 }

TechbloxModdingAPI/Interface/IMGUI/Group.cs

@@ -20,7 +20,7 @@ namespace TechbloxModdingAPI.Interface.IMGUI
         /// </summary>
         public Rect Box { get; set; }
 
-        public override void OnGUI()
+        public void OnGUI()
         {
             /*if (Constants.Default == null) return;
             if (Constants.Default.box == null) return;*/
@@ -61,6 +61,16 @@ namespace TechbloxModdingAPI.Interface.IMGUI
             }
         }
 
+        /// <summary>
+        /// The group's unique name.
+        /// </summary>
+        public string Name { get; }
+        
+        /// <summary>
+        /// Whether to display the group and everything in it.
+        /// </summary>
+        public bool Enabled { set; get; } = true;
+        
         /// <summary>
         /// The amount of elements in the group.
         /// </summary>
@@ -75,10 +85,19 @@ namespace TechbloxModdingAPI.Interface.IMGUI
         /// <param name="box">The rectangular area to use in the window.</param>
         /// <param name="name">Name of the group.</param>
         /// <param name="automaticLayout">Whether to use automatic UI layout.</param>
-        public Group(Rect box, string name = null, bool automaticLayout = false) : base(box.ToString().Replace(" ", ""), name)
+        public Group(Rect box, string name = null, bool automaticLayout = false)
         {
             Box = box;
+            if (name == null)
+            {
+                this.Name = typeof(Group).FullName + "::" + box.ToString().Replace(" ", "");
+            }
+            else
+            {
+                this.Name = name;
+            }
             this.automaticLayout = automaticLayout;
+            IMGUIManager.AddElement(this);
         }
 
         /// <summary>

TechbloxModdingAPI/Interface/IMGUI/IMGUIManager.cs

@@ -2,7 +2,6 @@ using System.Collections.Generic;
 using Svelto.Tasks;
 using Svelto.Tasks.ExtraLean;
 using TechbloxModdingAPI.Tasks;
-using TechbloxModdingAPI.Utility;
 using UnityEngine;
 
 namespace TechbloxModdingAPI.Interface.IMGUI
@@ -15,9 +14,9 @@ namespace TechbloxModdingAPI.Interface.IMGUI
     /// </summary>
     public static class IMGUIManager
     {
-        internal static OnGuiRunner ImguiScheduler = new("TechbloxModdingAPI_IMGUIScheduler");
+        internal static OnGuiRunner ImguiScheduler = new OnGuiRunner("TechbloxModdingAPI_IMGUIScheduler");
         
-        private static readonly WeakDictionary<string, UIElement> _activeElements = new();
+        private static Dictionary<string, UIElement> _activeElements = new Dictionary<string,UIElement>();
 
         /// <summary>
         /// Add an UIElement instance to be managed by IMGUIManager.

TechbloxModdingAPI/Interface/IMGUI/Image.cs

@@ -20,7 +20,7 @@ namespace TechbloxModdingAPI.Interface.IMGUI
         /// </summary>
         public Rect Box { get; set; } = Rect.zero;
         
-        public override void OnGUI()
+        public void OnGUI()
         {
             //if (Texture == null) return;
             if (automaticLayout)
@@ -33,15 +33,42 @@ namespace TechbloxModdingAPI.Interface.IMGUI
             }
         }
 
+        /// <summary>
+        /// The image element's unique name.
+        /// </summary>
+        public string Name { get; }
+        
+        /// <summary>
+        /// Whether to display the image and everything in it.
+        /// </summary>
+        public bool Enabled { set; get; } = true;
+        
         /// <summary>
         /// Initializes a new instance of the <see cref="T:TechbloxModdingAPI.Interface.IMGUI.Image"/> class with automatic layout.
         /// </summary>
         /// <param name="texture">Image to display.</param>
         /// <param name="name">The element's name.</param>
-        public Image(Texture texture = null, string name = null) : base(texture == null ? "" : $"{texture.name}({texture.width}x{texture.height})", name)
+        public Image(Texture texture = null, string name = null)
         {
             automaticLayout = true;
             Texture = texture;
+            if (name == null)
+            {
+                if (texture == null)
+                {
+                    this.Name = typeof(Image).FullName + "::" + texture;
+                }
+                else
+                {
+                    this.Name = typeof(Image).FullName + "::" + texture.name + "(" + texture.width + "x" + texture.height + ")";
+                }
+                
+            }
+            else
+            {
+                this.Name = name;
+            }
+            IMGUIManager.AddElement(this);
         }
 
         /// <summary>

TechbloxModdingAPI/Interface/IMGUI/Label.cs

@@ -20,7 +20,7 @@ namespace TechbloxModdingAPI.Interface.IMGUI
         /// </summary>
         public Rect Box { get; set; } = Rect.zero;
         
-        public override void OnGUI()
+        public void OnGUI()
         {
             if (automaticLayout)
             {
@@ -32,15 +32,34 @@ namespace TechbloxModdingAPI.Interface.IMGUI
             }
         }
 
+        /// <summary>
+        /// The label's unique name.
+        /// </summary>
+        public string Name { get; }
+        
+        /// <summary>
+        /// Whether to display the label.
+        /// </summary>
+        public bool Enabled { set; get; } = true;
+
         /// <summary>
         /// Initializes a new instance of the <see cref="T:TechbloxModdingAPI.Interface.IMGUI.Label"/> class with automatic layout.
         /// </summary>
         /// <param name="initialText">Initial string to display on the label.</param>
         /// <param name="name">The element's name.</param>
-        public Label(string initialText = null, string name = null) : base(initialText, name)
+        public Label(string initialText = null, string name = null)
         {
             automaticLayout = true;
             Text = initialText;
+            if (name == null)
+            {
+                this.Name = typeof(Label).FullName + "::" + initialText;
+            }
+            else
+            {
+                this.Name = name;
+            }
+            IMGUIManager.AddElement(this);
         }
 
         /// <summary>

TechbloxModdingAPI/Interface/IMGUI/Text.cs

@@ -28,7 +28,7 @@ namespace TechbloxModdingAPI.Interface.IMGUI
         /// </summary>
         public event EventHandler<string> OnEdit;
         
-        public override void OnGUI()
+        public void OnGUI()
         {
             string editedText = null;
             if (automaticLayout)
@@ -61,17 +61,36 @@ namespace TechbloxModdingAPI.Interface.IMGUI
             }
         }
 
+        /// <summary>
+        /// The text field's unique name.
+        /// </summary>
+        public string Name { get; }
+        
+        /// <summary>
+        /// Whether to display the text field.
+        /// </summary>
+        public bool Enabled { set; get; } = true;
+
         /// <summary>
         /// Initialize the text input field with automatic layout.
         /// </summary>
         /// <param name="initialText">Initial text in the input field.</param>
         /// <param name="name">The text field's name.</param>
         /// <param name="multiline">Allow multiple lines?</param>
-        public Text(string initialText = null, string name = null, bool multiline = false) : base(initialText, name)
+        public Text(string initialText = null, string name = null, bool multiline = false)
         {
             this.Multiline = multiline;
             automaticLayout = true;
             text = initialText ?? "";
+            if (name == null)
+            {
+                this.Name = typeof(Text).FullName + "::" + text;
+            }
+            else
+            {
+                this.Name = name;
+            }
+            IMGUIManager.AddElement(this);
         }
 
         /// <summary>

TechbloxModdingAPI/Interface/IMGUI/UIElement.cs

@@ -6,34 +6,23 @@ namespace TechbloxModdingAPI.Interface.IMGUI
     /// GUI Element like a text field, button or picture.
     /// This interface is used to wrap many elements from Unity's IMGUI system.
     /// </summary>
-    public abstract class UIElement
+    public interface UIElement
     {
-        protected UIElement(string text, string name)
-        {
-            Name = name ?? GetType().FullName + "::" + text;
-            IMGUIManager.AddElement(this);
-        }
-
-        ~UIElement()
-        {
-            IMGUIManager.RemoveElement(this);
-        }
-        
         /// <summary>
         /// GUI operations to perform in the OnGUI scope.
         /// This is basically equivalent to a MonoBehaviour's OnGUI method.
         /// </summary>
-        public abstract void OnGUI();
+        void OnGUI();
         
         /// <summary>
         /// The element's name.
         /// This should be unique for every instance of the class.
         /// </summary>
-        public string Name { get; }
+        string Name { get; }
         
         /// <summary>
         /// Whether to display the UI element or not.
         /// </summary>
-        public bool Enabled { get; set; } = true;
+        bool Enabled { get; }
     }
 }

TechbloxModdingAPI/Main.cs

@@ -104,7 +104,7 @@ namespace TechbloxModdingAPI
                 }
                 Scheduler.Dispose();
                 var currentAssembly = Assembly.GetExecutingAssembly();
-                harmony.UnpatchSelf();
+                harmony.UnpatchAll(currentAssembly.GetName().Name);
                 harmony = null;
                 Logging.MetaLog($"{currentAssembly.GetName().Name} v{currentAssembly.GetName().Version} shutdown");
             }

TechbloxModdingAPI/Persistence/SaveAndLoadCompositionRootPatch.cs

@@ -4,18 +4,17 @@ using RobocraftX.SaveAndLoad;
 using Svelto.ECS;
 
 using HarmonyLib;
-using RobocraftX.StateSync;
 
 namespace TechbloxModdingAPI.Persistence
 {
-	[HarmonyPatch(typeof(SaveAndLoadCompositionRoot), "ClientComposeTimeStopped")]
+	[HarmonyPatch(typeof(SaveAndLoadCompositionRoot), "ServerCompose")]
     class SaveAndLoadCompositionRootPatch
     {
 		public static EnginesRoot currentEnginesRoot;
 
-		public static void Prefix(StateSyncRegistrationHelper stateSyncHelper)
+		public static void Prefix(EnginesRoot enginesRoot)
 		{
-			currentEnginesRoot = stateSyncHelper.enginesRoot;
+			currentEnginesRoot = enginesRoot;
 			//SerializerManager.RegisterSerializers(enginesRoot);
 		}
     }

TechbloxModdingAPI/Player.cs

@@ -1,5 +1,6 @@
 using System;
 using Gamecraft.Wires;
+using RobocraftX.Blocks.Ghost;
 using RobocraftX.Character;
 using RobocraftX.Character.Movement;
 using Unity.Mathematics;
@@ -10,7 +11,6 @@ using RobocraftX.Physics;
 using Svelto.ECS;
 using Techblox.BuildingDrone;
 using Techblox.Camera;
-using Techblox.Character;
 using TechbloxModdingAPI.Blocks;
 using TechbloxModdingAPI.Players;
 using TechbloxModdingAPI.Utility;
@@ -66,16 +66,13 @@ namespace TechbloxModdingAPI
 
         /// <summary>
         /// Returns the current player belonging to this client. It will be different after entering/leaving simulation.
-        /// May return null if the local player doesn't exist.
         /// </summary>
         public static Player LocalPlayer
         {
 	        get
 	        {
-		        var playerId = playerEngine.GetLocalPlayer();
+		        if (localPlayer == null || localPlayer.Id != playerEngine.GetLocalPlayer())
-		        if (playerId == uint.MaxValue) return null;
+			        localPlayer = GetInstance(playerEngine.GetLocalPlayer());
-		        if (localPlayer == null || localPlayer.Id != playerId)
-			        localPlayer = GetInstance(playerId);
 		        return localPlayer;
 	        }
         }
@@ -195,8 +192,8 @@ namespace TechbloxModdingAPI
         /// The player's mass.
         /// </summary>
         /// <value>The mass.</value>
-        [Obsolete] // We cannot get it clientside or something
+        public float Mass =>
-        public float Mass => 0;
+	        1f / playerEngine.GetCharacterStruct<RigidBodyEntityStruct>(Id).Get().physicsMass.InverseMass;
 
         /// <summary>
         /// The player's latest network ping time.
@@ -214,16 +211,15 @@ namespace TechbloxModdingAPI
         /// The player's initial health when entering Simulation (aka Time Running) mode.
         /// </summary>
         /// <value>The initial health.</value>
-        [Obsolete("We can no longer get initial health, returns max health.")]
         public float InitialHealth
         {
 	        get
 	        {
-		        var opt = playerEngine.GetCharacterStruct<CharacterHealthEntityComponent>(Id);
+		        var opt = playerEngine.GetCharacterStruct<CharacterHealthEntityStruct>(Id);
-		        return opt ? opt.Get().maxHealth : -1f;
+		        return opt ? opt.Get().initialHealth : -1f;
 	        }
 
-	        set => playerEngine.GetCharacterStruct<CharacterHealthEntityComponent>(Id).Get().maxHealth = value;
+	        set => playerEngine.GetCharacterStruct<CharacterHealthEntityStruct>(Id).Get().initialHealth = value;
         }
 
         /// <summary>
@@ -234,25 +230,30 @@ namespace TechbloxModdingAPI
         {
 	        get
 	        {
-		        var opt = playerEngine.GetCharacterStruct<CharacterHealthEntityComponent>(Id);
+		        var opt = playerEngine.GetCharacterStruct<CharacterHealthEntityStruct>(Id);
 		        return opt ? opt.Get().currentHealth : -1f;
 	        }
 
-	        set => playerEngine.GetCharacterStruct<CharacterHealthEntityComponent>(Id).Get().currentHealth = value;
+	        set => playerEngine.GetCharacterStruct<CharacterHealthEntityStruct>(Id).Get().currentHealth = value;
         }
 
         /// <summary>
         /// Whether this <see cref="T:TechbloxModdingAPI.Player"/> is damageable.
         /// </summary>
         /// <value><c>true</c> if damageable; otherwise, <c>false</c>.</value>
-        [Obsolete("Players are probably always damageable")]
         public bool Damageable
         {
-	        get => true;
+	        get
+	        {
+		        var opt = playerEngine.GetCharacterStruct<CharacterHealthEntityStruct>(Id);
+		        return opt.Get().canTakeDamageStat;
+	        }
 
-	        // ReSharper disable once ValueParameterNotUsed
 	        set
 	        {
+		        ref var healthStruct = ref playerEngine.GetCharacterStruct<CharacterHealthEntityStruct>(Id).Get();
+		        healthStruct.canTakeDamage = value;
+		        healthStruct.canTakeDamageStat = value;
 	        }
         }
 
@@ -260,26 +261,30 @@ namespace TechbloxModdingAPI
         /// The player's lives when initially entering Simulation (aka Time Running) mode.
         /// </summary>
         /// <value>The initial lives.</value>
-        [Obsolete("The player has infinite lives")]
         public uint InitialLives
 		{
-	        get => uint.MaxValue;
+	        get
+	        {
+		        var opt = playerEngine.GetCharacterStruct<CharacterLivesEntityComponent>(Id);
+		        return opt ? opt.Get().initialLives : uint.MaxValue;
+	        }
 
-	        // ReSharper disable once ValueParameterNotUsed
+	        set => playerEngine.GetCharacterStruct<CharacterLivesEntityComponent>(Id).Get().initialLives = value;
-	        set { }
 		}
 
         /// <summary>
         /// The player's current lives in Simulation (aka Time Running) mode.
         /// </summary>
         /// <value>The current lives.</value>
-        [Obsolete("The player has infinite lives")]
 		public uint CurrentLives
         {
-	        get => uint.MaxValue;
+	        get
+	        {
+		        var opt = playerEngine.GetCharacterStruct<CharacterLivesEntityComponent>(Id);
+		        return opt ? opt.Get().currentLives : uint.MaxValue;
+	        }
 
-	        // ReSharper disable once ValueParameterNotUsed
+	        set => playerEngine.GetCharacterStruct<CharacterLivesEntityComponent>(Id).Get().currentLives = value;
-	        set { }
         }
 
         /*/// <summary>
@@ -365,9 +370,6 @@ namespace TechbloxModdingAPI
 				var group when group == CharacterExclusiveGroups.MachineSpawningGroup => PlayerState.HoldingMachine,
 				var group when group == CharacterExclusiveGroups.OnFootGroup => PlayerState.OnFoot,
 				var group when group == CharacterExclusiveGroups.InPilotSeatGroup => PlayerState.InSeat,
-				var group when group == CharacterExclusiveGroups.DyingOnFootGroup => PlayerState.OnFoot,
-				var group when group == CharacterExclusiveGroups.DyingInPilotSeatGroup => PlayerState.InSeat,
-				var group when group == CharacterExclusiveGroups.DeadGroup => PlayerState.OnFoot,
 				_ => throw new ArgumentOutOfRangeException("", "Unknown player state")
 			};
 

TechbloxModdingAPI/Players/PlayerEngine.cs

@@ -8,6 +8,7 @@ using RobocraftX.Common.Input;
 using RobocraftX.CR.MachineEditing.BoxSelect;
 using RobocraftX.Physics;
 using RobocraftX.Blocks.Ghost;
+using Gamecraft.GUI.HUDFeedbackBlocks;
 using RobocraftX.Common;
 using RobocraftX.Multiplayer;
 using RobocraftX.SimulationModeState;
@@ -21,7 +22,6 @@ using Techblox.Character;
 using TechbloxModdingAPI.Engines;
 using TechbloxModdingAPI.Input;
 using TechbloxModdingAPI.Utility;
-using TechbloxModdingAPI.Utility.ECS;
 
 namespace TechbloxModdingAPI.Players
 {
@@ -50,10 +50,10 @@ namespace TechbloxModdingAPI.Players
         public uint GetLocalPlayer()
 		{
 			if (!isReady) return uint.MaxValue;
-			var (localPlayers, count) = entitiesDB.QueryEntities<PlayerIDStruct>(PlayersExclusiveGroups.LocalPlayers);
+			var localPlayers = entitiesDB.QueryEntities<PlayerIDStruct>(PlayersExclusiveGroups.LocalPlayers).ToBuffer();
-			if (count > 0)
+			if (localPlayers.count > 0)
 			{
-				return localPlayers[0].ID.entityID;
+				return localPlayers.buffer[0].ID.entityID;
 			}
 			return uint.MaxValue;
 		}
@@ -61,10 +61,10 @@ namespace TechbloxModdingAPI.Players
         public uint GetRemotePlayer()
 		{
 			if (!isReady) return uint.MaxValue;
-			var (localPlayers, count) = entitiesDB.QueryEntities<PlayerIDStruct>(PlayersExclusiveGroups.RemotePlayers);
+			var localPlayers = entitiesDB.QueryEntities<PlayerIDStruct>(PlayersExclusiveGroups.RemotePlayers).ToBuffer();
-            if (count > 0)
+            if (localPlayers.count > 0)
             {
-                return localPlayers[0].ID.entityID;
+                return localPlayers.buffer[0].ID.entityID;
             }
 			return uint.MaxValue;
 		}
@@ -113,10 +113,18 @@ namespace TechbloxModdingAPI.Players
 			return true;
 		}
 
+		public bool GetGameOverScreen(uint playerId)
+		{
+			if (entitiesDB == null) return false;
+			ref HudActivatedBlocksEntityStruct habes = ref entitiesDB.QueryEntity<HudActivatedBlocksEntityStruct>(HUDFeedbackBlocksGUIExclusiveGroups.GameOverHudEgid);
+			NativeDynamicArrayCast<EGID> nativeDynamicArrayCast = new NativeDynamicArrayCast<EGID>(habes.activatedBlocksOrdered);
+			return nativeDynamicArrayCast.count > 0;
+		}
+
 		public bool IsDead(uint playerId)
 		{
 			if (entitiesDB == null) return true;
-			return entitiesDB.Exists<RigidBodyEntityStruct>(playerId, CharacterExclusiveGroups.DeadGroup);
+			return entitiesDB.Exists<RigidBodyEntityStruct>(playerId, CharacterExclusiveGroups.DeadCharacters);
 		}
 
         // reusable methods
@@ -165,9 +173,8 @@ namespace TechbloxModdingAPI.Players
 	        var opt = GetCameraStruct<PhysicCameraRayCastEntityStruct>(playerId);
 	        if (!opt) return default;
 	        PhysicCameraRayCastEntityStruct rayCast = opt;
-	        EGID physicCameraEgid = new EGID(playerId, CameraExclusiveGroups.PhysicCameraGroup);
 	        float distance = maxDistance < 0
-		        ? GhostBlockUtils.GetBuildInteractionDistance(entitiesDB, rayCast, physicCameraEgid,
+		        ? GhostBlockUtils.GetBuildInteractionDistance(entitiesDB, rayCast,
 			        GhostBlockUtils.GhostCastMethod.GhostCastProportionalToBlockSize)
 		        : maxDistance;
 	        if (rayCast.hit && rayCast.distance <= distance)

TechbloxModdingAPI/SimBody.cs

@@ -6,7 +6,6 @@ using UnityEngine;
 using Gamecraft.Damage;
 using RobocraftX.Common;
 using RobocraftX.Physics;
-using Techblox.TimeRunning.Clusters;
 
 namespace TechbloxModdingAPI
 {
@@ -21,7 +20,7 @@ namespace TechbloxModdingAPI
         /// </summary>
         public Cluster Cluster => cluster ??= clusterId == uint.MaxValue // Return cluster or if it's null then set it
             ? Block.BlockEngine.GetCluster(Id.entityID) // If we don't have a clusterId set then get it from the game
-            : GetInstance(new EGID(clusterId, ClustersExclusiveGroups.SIMULATION_CLUSTERS_GROUP),
+            : GetInstance(new EGID(clusterId, CommonExclusiveGroups.SIMULATION_CLUSTERS_GROUP),
                 egid => new Cluster(egid)); // Otherwise get the cluster from the ID
 
         private Cluster cluster;
@@ -74,16 +73,15 @@ namespace TechbloxModdingAPI
             }
         }
 
-        [Obsolete] //Cannot get mass even from UECS
         public float Mass
         {
-            get => 0f;
+            get => math.rcp(GetStruct().physicsMass.InverseMass);
             //set => GetStruct().physicsMass.InverseMass = math.rcp(value);
         }
 
         public float3 CenterOfMass
         {
-            get => 0f; //TODO
+            get => GetStruct().physicsMass.CenterOfMass;
             //set => GetStruct().physicsMass.CenterOfMass = value;
         }
 
@@ -94,20 +92,20 @@ namespace TechbloxModdingAPI
 
         public float InitialHealth
         {
-            get => 0f;
+            get => Block.BlockEngine.GetBlockInfo<HealthEntityComponent>(this).initialHealth;
-            set { }
+            set => Block.BlockEngine.GetBlockInfo<HealthEntityComponent>(this).initialHealth = value;
         }
 
         public float CurrentHealth
         {
-            get => 0f;
+            get => Block.BlockEngine.GetBlockInfo<HealthEntityComponent>(this).currentHealth;
-            set { }
+            set => Block.BlockEngine.GetBlockInfo<HealthEntityComponent>(this).currentHealth = value;
         }
 
         public float HealthMultiplier
         {
-            get => 0f;
+            get => Block.BlockEngine.GetBlockInfo<HealthEntityComponent>(this).healthMultiplier;
-            set { }
+            set => Block.BlockEngine.GetBlockInfo<HealthEntityComponent>(this).healthMultiplier = value;
         }
 
         /// <summary>

TechbloxModdingAPI/Tasks/OnGuiRunner.cs

@@ -1,12 +1,11 @@
 using System.Collections;
-using RobocraftX.Schedulers;
 using Svelto.Tasks;
 using Svelto.Tasks.ExtraLean;
 using Svelto.Tasks.Unity.Internal;
 
 namespace TechbloxModdingAPI.Tasks
 {
-    public class OnGuiRunner : SteppableRunner<ExtraLeanSveltoTask<IEnumerator>>
+    public class OnGuiRunner : BaseRunner<ExtraLeanSveltoTask<IEnumerator>>
     {
         public OnGuiRunner(string name, uint runningOrder = 0)
             : base(name)

TechbloxModdingAPI/Tests/TechbloxModdingAPIPluginTest.cs

@@ -1,23 +1,60 @@
 using System;
+using System.Collections.Generic;
+using System.Diagnostics;
 using System.Reflection;
 using System.Text;
-using BepInEx;
-using BepInEx.Bootstrap;
-using HarmonyLib;
-using RobocraftX.FrontEnd;
 using TechbloxModdingAPI.App;
+using HarmonyLib;
+using IllusionInjector;
+// test
+using RobocraftX.FrontEnd;
+using Unity.Mathematics;
+using UnityEngine;
+using Svelto.Tasks;
+using Svelto.Tasks.Lean;
+using TechbloxModdingAPI.Blocks;
 using TechbloxModdingAPI.Commands;
+using TechbloxModdingAPI.Players;
+using TechbloxModdingAPI.Tasks;
+using TechbloxModdingAPI.Utility;
 
 namespace TechbloxModdingAPI.Tests
 {
-    #if DEBUG // The API should be loaded by other plugins, but it can be used by itself for testing
+#if DEBUG
-    [BepInPlugin("org.exmods.TechbloxModdingAPIPluginTest", PluginInfo.PLUGIN_NAME, PluginInfo.PLUGIN_VERSION)]
+    // unused by design
-    [BepInProcess("Techblox.exe")]
+    /// <summary>
-    public class TechbloxModdingAPIPluginTest : BaseUnityPlugin
+    /// Modding API implemented as a standalone IPA Plugin.
+    /// Ideally, TechbloxModdingAPI should be loaded by another mod; not itself
+    /// </summary>
+    class TechbloxModdingAPIPluginTest : IllusionPlugin.IEnhancedPlugin
     {
-        private void Awake()
+
+        private static Harmony harmony { get; set; }
+
+        public override string Name { get; } = Assembly.GetExecutingAssembly().GetName().Name;
+
+        public override string Version { get; } = Assembly.GetExecutingAssembly().GetName().Version.ToString();
+
+        public string HarmonyID { get; } = "org.git.exmods.modtainers.techbloxmoddingapi";
+
+        public override void OnApplicationQuit()
         {
+            Main.Shutdown();
+        }
+
+		public override void OnApplicationStart()
+		{
+			FileLog.Reset();
+			Harmony.DEBUG = true;
 			Main.Init();
+			Logging.MetaDebugLog($"Version group id {ApiExclusiveGroups.versionGroup}");
+			// disable background music
+			Logging.MetaDebugLog("Audio Mixers: " + string.Join(",", AudioTools.GetMixers()));
+			//AudioTools.SetVolume(0.0f, "Music"); // The game now sets this from settings again after this is called :(
+
+			//Utility.VersionTracking.Enable();//(very) unstable
+
+			// debug/test handlers
 			Client.EnterMenu += (sender, args) => throw new Exception("Test handler always throws an exception!");
 			Client.EnterMenu += (sender, args) => Console.WriteLine("EnterMenu handler after erroring handler");
 			Game.Enter += (s, a) =>
@@ -28,6 +65,9 @@ namespace TechbloxModdingAPI.Tests
 					Console.WriteLine($"Player {Player.LocalPlayer} exited seat {args.Seat}");
 			};
 
+			// debug/test commands
+			if (Dependency.Hell("ExtraCommands"))
+			{
 				CommandBuilder.Builder()
 							  .Name("Exit")
 							  .Description("Close Techblox immediately, without any prompts")
@@ -40,31 +80,310 @@ namespace TechbloxModdingAPI.Tests
 							  .Action((float d) => { UnityEngine.Camera.main.fieldOfView = d; })
 							  .Build();
 
+				CommandBuilder.Builder()
+					.Name("MoveLastBlock")
+					.Description("Move the most-recently-placed block, and any connected blocks by the given offset")
+					.Action((float x, float y, float z) =>
+					{
+						if (GameState.IsBuildMode())
+							foreach (var block in Block.GetLastPlacedBlock().GetConnectedCubes())
+								block.Position += new Unity.Mathematics.float3(x, y, z);
+						else
+							Logging.CommandLogError("Blocks can only be moved in Build mode!");
+					}).Build();
+
+				CommandBuilder.Builder()
+							  .Name("PlaceAluminium")
+							  .Description("Place a block of aluminium at the given coordinates")
+							  .Action((float x, float y, float z) =>
+							  {
+								  var block = Block.PlaceNew(BlockIDs.Cube, new float3(x, y, z));
+								  Logging.CommandLog("Block placed with type: " + block.Type);
+							  })
+							  .Build();
+
+				CommandBuilder.Builder()
+					.Name("PlaceAluminiumLots")
+					.Description("Place a lot of blocks of aluminium at the given coordinates")
+					.Action((float x, float y, float z) =>
+					{
+						Logging.CommandLog("Starting...");
+						var sw = Stopwatch.StartNew();
+						for (int i = 0; i < 100; i++)
+							for (int j = 0; j < 100; j++)
+								Block.PlaceNew(BlockIDs.Cube, new float3(x + i, y, z + j));
+						sw.Stop();
+						Logging.CommandLog("Finished in " + sw.ElapsedMilliseconds + "ms");
+					})
+					.Build();
+
+				Block b = null;
+				CommandBuilder.Builder("moveBlockInSim", "Run in build mode first while looking at a block, then in sim to move it up")
+					.Action(() =>
+					{
+						if (b == null)
+						{
+							b = new Player(PlayerType.Local).GetBlockLookedAt();
+							Logging.CommandLog("Block saved: " + b);
+						}
+						else
+							Logging.CommandLog("Block moved to: " + (b.GetSimBody().Position += new float3(0, 2, 0)));
+					}).Build();
+
+				CommandBuilder.Builder("Error", "Throw an error to make sure SimpleCustomCommandEngine's wrapper catches it.")
+					.Action(() => { throw new Exception("Error Command always throws an error"); })
+					.Build();
+
+				CommandBuilder.Builder("ColorBlock",
+						"Change color of the block looked at if there's any.")
+					.Action<string>(str =>
+					{
+						if (!Enum.TryParse(str, out BlockColors color))
+						{
+							Logging.CommandLog("Color " + str + " not found! Interpreting as 4 color values.");
+							var s = str.Split(' ');
+							new Player(PlayerType.Local).GetBlockLookedAt().CustomColor = new float4(float.Parse(s[0]),
+								float.Parse(s[1]), float.Parse(s[2]), float.Parse(s[3]));
+							return;
+						}
+
+						new Player(PlayerType.Local).GetBlockLookedAt().Color = color;
+						Logging.CommandLog("Colored block to " + color);
+
+					}).Build();
+
+				CommandBuilder.Builder("MoveBlockByID", "Gets a block based on its object identifier and teleports it up.")
+					.Action<char>(ch =>
+					{
+						foreach (var body in SimBody.GetFromObjectID(ch))
+						{
+							Logging.CommandLog("SimBody: " + body);
+							body.Position += new float3(0, 10, 0);
+							foreach (var bodyConnectedBody in body.GetConnectedBodies())
+							{
+								Logging.CommandLog("Moving " + bodyConnectedBody);
+								bodyConnectedBody.Position += new float3(0, 10, 0);
+							}
+						}
+					}).Build();
+
+				CommandBuilder.Builder("TestChunkHealth", "Sets the chunk looked at to the given health.")
+					.Action((float val, float max) =>
+					{
+						var body = new Player(PlayerType.Local).GetSimBodyLookedAt();
+						if (body == null) return;
+						body.CurrentHealth = val;
+						body.InitialHealth = max;
+						Logging.CommandLog("Health set to: " + val);
+					}).Build();
+
+				CommandBuilder.Builder("placeBlockGroup", "Places some blocks in a group")
+					.Action((float x, float y, float z) =>
+					{
+						var pos = new float3(x, y, z);
+						var group = BlockGroup.Create(new Block(BlockIDs.Cube, pos) {Color = BlockColors.Aqua});
+						new Block(BlockIDs.Cube, pos += new float3(1, 0, 0))
+							{Color = BlockColors.Blue, BlockGroup = group};
+						new Block(BlockIDs.Cube, pos += new float3(1, 0, 0))
+							{Color = BlockColors.Green, BlockGroup = group};
+						new Block(BlockIDs.Cube, pos + new float3(1, 0, 0))
+							{Color = BlockColors.Lime, BlockGroup = group};
+					}).Build();
+
+				CommandBuilder.Builder("placeCustomBlock", "Places a custom block, needs a custom catalog and assets.")
+					.Action((float x, float y, float z) =>
+					{
+						Logging.CommandLog("Block placed: " +
+						                   Block.PlaceNew((BlockIDs) 500, new float3(0, 0, 0)));
+					}).Build();
+
+				CommandBuilder.Builder("toggleTimeMode", "Enters or exits simulation.")
+					.Action((float x, float y, float z) =>
+					{
+						Game.CurrentGame().ToggleTimeMode();
+					}).Build();
+
+				CommandBuilder.Builder("testColorBlock", "Tests coloring a block to default color")
+					.Action(() => Player.LocalPlayer.GetBlockLookedAt().Color = BlockColors.Default).Build();
+				CommandBuilder.Builder("testMaterialBlock", "Tests materialing a block to default material")
+					.Action(() => Player.LocalPlayer.GetBlockLookedAt().Material = BlockMaterial.Default).Build();
+				CommandBuilder.Builder("testGameName", "Tests changing the game name")
+					.Action(() => Game.CurrentGame().Name = "Test").Build();
+				CommandBuilder.Builder("makeBlockStatic", "Makes a block you look at static")
+					.Action(() => Player.LocalPlayer.GetBlockLookedAt().Static = true).Build();
+
 				Game.AddPersistentDebugInfo("InstalledMods", InstalledMods);
+				/*Block.Placed += (sender, args) =>
+					Logging.MetaDebugLog("Placed block " + args.Block);
+				Block.Removed += (sender, args) =>
+					Logging.MetaDebugLog("Removed block " + args.Block);*/
+			}
 
-            // Plugin startup logic
+			// dependency test
-            Logger.LogInfo($"Plugin {PluginInfo.PLUGIN_GUID} is loaded!");
+			if (Dependency.Hell("TechbloxScripting", new Version("0.0.1.0")))
+			{
+				Logging.LogWarning("You're in TechbloxScripting dependency hell");
+			}
+			else
+			{
+				Logging.Log("Compatible TechbloxScripting detected");
+			}
+			// Interface test
+			/*Group uiGroup = new Group(new Rect(20, 20, 200, 500), "TechbloxModdingAPI_UITestGroup", true);
+			var button = new Button("TEST");
+			button.OnClick += (b, __) => { Logging.MetaDebugLog($"Click on {((Interface.IMGUI.Button)b).Name}");};
+			var button2 = new Button("TEST2");
+			button2.OnClick += (b, __) => { Logging.MetaDebugLog($"Click on {((Interface.IMGUI.Button)b).Name}");};
+			Text uiText = new Text("<Input!>", multiline: true);
+			uiText.OnEdit += (t, txt) => { Logging.MetaDebugLog($"Text in {((Text)t).Name} is now '{txt}'"); };
+			Label uiLabel = new Label("Label!");
+			Image uiImg = new Image(name:"Behold this texture!");
+			uiImg.Enabled = false;
+			uiGroup.AddElement(button);
+			uiGroup.AddElement(button2);
+			uiGroup.AddElement(uiText);
+			uiGroup.AddElement(uiLabel);
+			uiGroup.AddElement(uiImg);*/
 
+			/*Addressables.LoadAssetAsync<Texture2D>("Assets/Art/Textures/UI/FrontEndMap/RCX_Blue_Background_5k.jpg")
+					.Completed +=
+				handle =>
+				{
+					uiImg.Texture = handle.Result;
+					uiImg.Enabled = true;
+					Logging.MetaDebugLog($"Got blue bg asset {handle.Result}");
+				};*/
+            
+            /*((FasterList<GuiInputMap.GuiInputMapElement>)AccessTools.Property(typeof(GuiInputMap), "GuiInputsButtonDown").GetValue(null))
+	            .Add(new GuiInputMap.GuiInputMapElement(RewiredConsts.Action.ToggleCommandLine, GuiIn))*/
+
+            /*Game.Enter += (sender, e) =>
+            {
+	            ushort lastKey = ushort.MaxValue;
+	            foreach (var kv in FullGameFields._dataDb.GetValues<CubeListData>()
+	                .OrderBy(kv=>ushort.Parse(kv.Key)))
+	            {
+	                var data = (CubeListData) kv.Value;
+	                ushort currentKey = ushort.Parse(kv.Key);
+	                var toReplace = new Dictionary<string, string>
+	                {
+		                {"Scalable", ""}, {"Qtr", "Quarter"}, {"RNeg", "Rounded Negative"},
+		                {"Neg", "Negative"}, {"Tetra", "Tetrahedron"},
+		                {"RWedge", "Rounded Wedge"}, {"RTetra", "Rounded Tetrahedron"}
+	                };
+	                string name = LocalizationService.Localize(data.CubeNameKey).Split(' ').Select(str =>
+			                str.Length > 0 ? char.ToUpper(str[0]) + str.Substring(1) : str).Aggregate((a, b) => a + b)
+		                .Replace("-", "");
+	                foreach (var rkv in toReplace)
+	                {
+		                name = Regex.Replace(name, rkv.Key + "([A-Z]|$)", rkv.Value + "$1");
+	                }
+	                Console.WriteLine($"{name}{(currentKey != lastKey + 1 ? $" = {currentKey}" : "")},");
+	                lastKey = currentKey;
+	            }
+            };*/
+            /*Game.Enter += (sender, e) =>
+            {
+	            ushort lastKey = ushort.MaxValue;
+	            Console.WriteLine("Materials:\n" + FullGameFields._dataDb.GetValues<MaterialPropertiesData>()
+		            .OrderBy(kv => ushort.Parse(kv.Key))
+		            .Select(kv =>
+		            {
+			            ushort currentKey = ushort.Parse(kv.Key);
+			            string result = $"{((MaterialPropertiesData)kv.Value).Name}{(currentKey != lastKey + 1 ? $" = {kv.Key}" : "")},";
+			            lastKey = currentKey;
+			            return result;
+		            })
+		            .Aggregate((a, b) => a + "\n" + b));
+            };*/
+
+            CommandBuilder.Builder("takeScreenshot", "Enables the screenshot taker")
+	            .Action(() =>
+	            {
+		            Game.CurrentGame().EnableScreenshotTaker();
+	            }).Build();
+
+            CommandBuilder.Builder("testPositionDefault", "Tests the Block.Position property's default value.")
+	            .Action(() =>
+	            {
+		            IEnumerator<TaskContract> Loop()
+		            {
+			            for (int i = 0; i < 2; i++)
+			            {
+				            Console.WriteLine("A");
+				            var block = Block.PlaceNew(BlockIDs.Cube, 1);
+				            Console.WriteLine("B");
+				            while (!block.Exists)
+					            yield return Yield.It;
+				            Console.WriteLine("C");
+				            block.Remove();
+				            Console.WriteLine("D");
+				            while (block.Exists)
+					            yield return Yield.It;
+				            Console.WriteLine("E - Pos: " + block.Position);
+				            block.Position = 4;
+				            Console.WriteLine("F - Pos: " + block.Position);
+			            }
+		            }
+
+		            Loop().RunOn(Scheduler.leanRunner);
+	            }).Build();
+
+            CommandBuilder.Builder("importAssetBundle")
+	            .Action(() =>
+	            {
+		            Logging.CommandLog("Importing asset bundle...");
+		            var ab = AssetBundle.LoadFromFile(
+			            @"filepath");
+		            Logging.CommandLog("Imported asset bundle: " + ab);
+		            var assets = ab.LoadAllAssets();
+		            Logging.CommandLog("Loaded " + assets.Length + " assets");
+		            foreach (var asset in assets)
+		            {
+			            Logging.CommandLog(asset);
+		            }
+	            }).Build();
+
+            bool shouldTestGhostBlock = false;
+            CommandBuilder.Builder("testGhostBlock")
+	            .Action(() =>
+	            {
+		            if (shouldTestGhostBlock)
+		            {
+			            shouldTestGhostBlock = false;
+			            Logging.CommandLog("Test disabled");
+		            }
+
+		            shouldTestGhostBlock = true;
+		            Scheduler.Schedule(new Repeatable(() =>
+		            {
+			            var ghostBlock = Player.LocalPlayer.GetGhostBlock();
+			            if (ghostBlock == null) return;
+			            ghostBlock.Position = Player.LocalPlayer.Position + 2;
+			            ghostBlock.Color = new BlockColor(BlockColors.Lime);
+		            }, () => shouldTestGhostBlock));
+		            Logging.CommandLog("Test enabled");
+	            }).Build();
+            
+            Client.EnterMenu += (sender, args) => Scheduler.Schedule(new Once(() => Client.Instance.CloseBetaPopup()));
+            
+            Game.Enter += (sender, args) =>
+	            Console.WriteLine(
+		            $"Current game selection data: {FullGameFields._gameSelectionData.gameMode} - {FullGameFields._gameSelectionData.saveType}");
 #if TEST
 			TestRoot.RunTests();
 #endif
 		}
 
-        private void OnDestroy()
-        {
-            Main.Shutdown();
-        }
-
 		private string modsString;
         private string InstalledMods()
         {
 	        if (modsString != null) return modsString;
 	        StringBuilder sb = new StringBuilder("Installed mods:");
-            foreach (var (_, plugin) in Chainloader.PluginInfos)
+	        foreach (var plugin in PluginManager.Plugins)
-                sb.Append("\n" + plugin.Metadata.Name + " - " + plugin.Metadata.Version);
+		        sb.Append("\n" + plugin.Name + " - " + plugin.Version);
 	        return modsString = sb.ToString();
         }
-    }
 
         [HarmonyPatch]
         public class MinimumSpecsPatch
@@ -80,5 +399,6 @@ namespace TechbloxModdingAPI.Tests
 		        return ((Func<bool>) MinimumSpecsCheck.CheckRequirementsMet).Method;
 	        }
         }
+    }
 #endif
 }

TechbloxModdingAPI/Utility/Dependency.cs

@@ -1,5 +1,7 @@
 using System;
-using BepInEx.Bootstrap;
+
+using IllusionInjector;
+using IllusionPlugin;
 
 namespace TechbloxModdingAPI.Utility
 {
@@ -13,11 +15,11 @@ namespace TechbloxModdingAPI.Utility
         /// </summary>
         /// <returns>The plugin.</returns>
         /// <param name="name">The plugin's name.</param>
-        public static BepInEx.PluginInfo GetPlugin(string name)
+        public static IPlugin GetPlugin(string name)
 		{
-			foreach(var plugin in Chainloader.PluginInfos.Values)
+			foreach(IPlugin plugin in PluginManager.Plugins)
 			{
-                if (plugin.Metadata.Name == name)
+                if (plugin.Name == name)
 				{
 					return plugin;
 				}
@@ -33,15 +35,45 @@ namespace TechbloxModdingAPI.Utility
         /// <param name="name">The plugin's name.</param>
         public static Version GetPluginVersion(string name)
 		{
-			var plugin = GetPlugin(name);
+			IPlugin plugin = GetPlugin(name);
 			if (plugin != null) {
 				try
 				{
-					return plugin.Metadata.Version;
+					return new Version(plugin.Version);
-				} catch (Exception e) when (e is ArgumentException or FormatException or OverflowException) {}
+				} catch (Exception e) when (
+				    e is ArgumentException
+					|| e is ArgumentNullException
+				    || e is ArgumentOutOfRangeException
+					|| e is FormatException
+				    || e is OverflowException) {}
 				return plugin.GetType().Assembly.GetName().Version;
 			}
 			return null;
 		}
+        
+        // (I'm leaving the auto-generated version)
+        // <summary>
+        // Hell the specified name and version.
+        // </summary>
+        // <returns>The hell.</returns>
+        // <param name="name">Name.</param>
+        // <param name="version">Version.</param>
+        /// <summary>
+        /// Detect if you're in dependency hell with respect to the plugin.
+		/// ie Check if the plugin doesn't exist or is out of date.
+		/// When version is null, this only checks if the plugin exists.
+		/// The version is retrieved using GetPluginVersion(string name).
+        /// </summary>
+        /// <returns>Are you in dependency hell?</returns>
+        /// <param name="name">The plugin's name'</param>
+        /// <param name="version">The target version.</param>
+		public static bool Hell(string name, Version version = null)
+		{
+			Version pluginVersion = GetPluginVersion(name);
+			if (version == null) {
+				return pluginVersion == null;
+			}
+			return (pluginVersion == null || pluginVersion < version);
+		}
     }
 }

TechbloxModdingAPI/Utility/ECS/NativeApiExtensions.EntitiesDBHelper.cs

@@ -1,56 +0,0 @@
-using System;
-using System.Diagnostics.CodeAnalysis;
-using System.Linq;
-using System.Reflection;
-using HarmonyLib;
-using Svelto.DataStructures;
-using Svelto.ECS;
-
-namespace TechbloxModdingAPI.Utility.ECS
-{
-    public static partial class NativeApiExtensions
-    {
-        [SuppressMessage("ReSharper", "StaticMemberInGenericType")]
-        private static class EntitiesDBHelper<T> where T : unmanaged, IEntityComponent
-        { // Each type gets a new set of fields here (that's what the ReSharper warning is about too)
-            public static readonly Lazy<MethodInfo> EntityStream =
-                new(() => AccessTools.PropertyGetter(typeof(EntitiesDB), "_entityStream"));
-
-            public static readonly Lazy<FieldInfo> Streams = new(() =>
-                AccessTools.Field(EntityStream.Value.ReturnType, "_streams"));
-
-            public static readonly Lazy<FieldInfo> Consumers = new(() =>
-                AccessTools.Field(typeof(EntityStream<T>), "_consumers"));
-
-            public static readonly Lazy<MethodInfo> TryGetValue =
-                new(AccessTools.Method(Streams.Value.FieldType, "TryGetValue"));
-
-            public static readonly Lazy<FieldInfo> RingBuffer =
-                new(() => AccessTools.Field(typeof(Consumer<T>), "_ringBuffer"));
-        }
-
-        private static EntityStream<T> GetEntityStream<T>(this EntitiesDB entitiesDB) where T : unmanaged, IEntityComponent
-        {
-            // EntitiesStreams (internal)
-            var entitiesStreams = EntitiesDBHelper<T>.EntityStream.Value.Invoke(entitiesDB, Array.Empty<object>());
-            // FasterDictionary<RefWrapperType, ITypeSafeStream> (interface is internal)
-            var streams = EntitiesDBHelper<T>.Streams.Value.GetValue(entitiesStreams);
-            
-            var parameters = new object[] { TypeRefWrapper<T>.wrapper, null };
-            var success = EntitiesDBHelper<T>.TryGetValue.Value.Invoke(streams, parameters);
-            if (!(bool)success)
-                return null; // There is no entity stream for this type
-            return (EntityStream<T>)parameters[1];
-        }
-        
-        private static ThreadSafeFasterList<Consumer<T>> GetConsumers<T>(this EntityStream<T> stream) where T : unmanaged, IEntityComponent
-        {
-            return (ThreadSafeFasterList<Consumer<T>>)EntitiesDBHelper<T>.Consumers.Value.GetValue(stream);
-        }
-
-        private static RingBuffer<(T, EGID)> GetRingBuffer<T>(this Consumer<T> consumer) where T : unmanaged, IEntityComponent
-        {
-            return (RingBuffer<(T, EGID)>)EntitiesDBHelper<T>.RingBuffer.Value.GetValue(consumer);
-        }
-    }
-}

TechbloxModdingAPI/Utility/ManagedApiExtensions.cs

@@ -1,7 +1,7 @@
 using Svelto.ECS;
 using Svelto.ECS.Hybrid;
 
-namespace TechbloxModdingAPI.Utility.ECS
+namespace TechbloxModdingAPI.Utility
 {
     public static class ManagedApiExtensions
     {
@@ -29,8 +29,7 @@ namespace TechbloxModdingAPI.Utility.ECS
         /// <param name="group">The group of the entity if the object can have multiple</param>
         /// <typeparam name="T">The component to query</typeparam>
         /// <returns>A reference to the component or a dummy value</returns>
-        public static OptionalRef<T> QueryEntityOptional<T>(this EntitiesDB entitiesDB, EcsObjectBase obj,
+        public static OptionalRef<T> QueryEntityOptional<T>(this EntitiesDB entitiesDB, EcsObjectBase obj, ExclusiveGroupStruct group = default)
-            ExclusiveGroupStruct group = default)
             where T : struct, IEntityViewComponent
         {
             EGID id = group == ExclusiveGroupStruct.Invalid ? obj.Id : new EGID(obj.Id.entityID, group);
@@ -46,30 +45,14 @@ namespace TechbloxModdingAPI.Utility.ECS
         /// <param name="group">The group of the entity if the object can have multiple</param>
         /// <typeparam name="T">The component to query</typeparam>
         /// <returns>A reference to the component or a dummy value</returns>
-        public static ref T QueryEntityOrDefault<T>(this EntitiesDB entitiesDB, EcsObjectBase obj,
+        public static ref T QueryEntityOrDefault<T>(this EntitiesDB entitiesDB, EcsObjectBase obj, ExclusiveGroupStruct group = default)
-            ExclusiveGroupStruct group = default)
             where T : struct, IEntityViewComponent
         {
             EGID id = group == ExclusiveGroupStruct.Invalid ? obj.Id : new EGID(obj.Id.entityID, group);
             var opt = QueryEntityOptional<T>(entitiesDB, id);
             if (opt) return ref opt.Get();
-            if (obj.InitData.Valid) return ref obj.InitData.Initializer(id).GetOrAdd<T>();
+            if (obj.InitData.Valid) return ref obj.InitData.Initializer(id).GetOrCreate<T>();
             return ref opt.Get(); //Default value
         }
-
-        /// <summary>
-        /// Query entities as OptionalRefs. The elements always exist, it's just a nice way to encapsulate the data.
-        /// </summary>
-        /// <param name="entitiesDB"></param>
-        /// <param name="group"></param>
-        /// <param name="select"></param>
-        /// <typeparam name="T"></typeparam>
-        /// <typeparam name="TR"></typeparam>
-        /// <returns></returns>
-        public static RefCollection<T> QueryEntitiesOptional<T>(this EntitiesDB entitiesDB, ExclusiveGroupStruct group) where T : struct, IEntityViewComponent
-        {
-            var (buffer, ids, count) = entitiesDB.QueryEntities<T>(group);
-            return new RefCollection<T>(count, buffer, ids, group);
-        }
     }
 }

TechbloxModdingAPI/Utility/NativeApiExtensions.cs

@@ -1,14 +1,13 @@
 using System;
 using System.Collections.Generic;
-using Svelto.DataStructures;
 using Svelto.ECS;
 using Svelto.Tasks;
 using Svelto.Tasks.Lean;
 using TechbloxModdingAPI.Tasks;
 
-namespace TechbloxModdingAPI.Utility.ECS
+namespace TechbloxModdingAPI.Utility
 {
-    public static partial class NativeApiExtensions
+    public static class NativeApiExtensions
     {
         /// <summary>
         /// Attempts to query an entity and returns an optional that contains the result if succeeded.
@@ -34,8 +33,7 @@ namespace TechbloxModdingAPI.Utility.ECS
         /// <param name="group">The group of the entity if the object can have multiple</param>
         /// <typeparam name="T">The component to query</typeparam>
         /// <returns>A reference to the component or a dummy value</returns>
-        public static OptionalRef<T> QueryEntityOptional<T>(this EntitiesDB entitiesDB, EcsObjectBase obj,
+        public static OptionalRef<T> QueryEntityOptional<T>(this EntitiesDB entitiesDB, EcsObjectBase obj, ExclusiveGroupStruct group = default)
-            ExclusiveGroupStruct group = default)
             where T : unmanaged, IEntityComponent
         {
             EGID id = group == ExclusiveGroupStruct.Invalid ? obj.Id : new EGID(obj.Id.entityID, group);
@@ -51,14 +49,13 @@ namespace TechbloxModdingAPI.Utility.ECS
         /// <param name="group">The group of the entity if the object can have multiple</param>
         /// <typeparam name="T">The component to query</typeparam>
         /// <returns>A reference to the component or a dummy value</returns>
-        public static ref T QueryEntityOrDefault<T>(this EntitiesDB entitiesDB, EcsObjectBase obj,
+        public static ref T QueryEntityOrDefault<T>(this EntitiesDB entitiesDB, EcsObjectBase obj, ExclusiveGroupStruct group = default)
-            ExclusiveGroupStruct group = default)
             where T : unmanaged, IEntityComponent
         {
             EGID id = group == ExclusiveGroupStruct.Invalid ? obj.Id : new EGID(obj.Id.entityID, group);
             var opt = QueryEntityOptional<T>(entitiesDB, id);
             if (opt) return ref opt.Get();
-            if (obj.InitData.Valid) return ref obj.InitData.Initializer(id).GetOrAdd<T>();
+            if (obj.InitData.Valid) return ref obj.InitData.Initializer(id).GetOrCreate<T>();
             /*if (!obj.InitData.Valid) return ref opt.Get(); //Default value
             var init = obj.InitData.Initializer(id);
             // Do not create the component if missing, as that can trigger Add() listeners that, in some cases, may be
@@ -67,63 +64,39 @@ namespace TechbloxModdingAPI.Utility.ECS
             return ref opt.Get(); //Default value
         }
 
+        private static readonly Dictionary<Type, (int PublishedCount, HashSet<EGID> Changes)> ChangesToPublish = new();
+
         /// <summary>
         /// Publishes an entity change, ignoring duplicate publishes and delaying changes as necessary.
         /// It will only publish in the next frame.
         /// </summary>
         /// <param name="entitiesDB">The entities DB to publish to</param>
-        /// <param name="id">The ECS object that got changed</param>
+        /// <param name="obj">The ECS object that got changed</param>
+        /// <param name="limit">Limits how many changes to publish - should be no more than the consumers' capacity that process this component</param>
         /// <typeparam name="T">The component that changed</typeparam>
-        public static void PublishEntityChangeDelayed<T>(this EntitiesDB entitiesDB, EGID id)
+        public static void PublishEntityChangeDelayed<T>(this EntitiesDB entitiesDB, EGID id, int limit = 80)
             where T : unmanaged, IEntityComponent
-        {
+        { //TODO: Doesn't seem to help
-            PublishChanges<T>(entitiesDB, id).RunOn(Scheduler.leanRunner);
+            if(!ChangesToPublish.ContainsKey(typeof(T)))
+                ChangesToPublish.Add(typeof(T), (0, new HashSet<EGID>()));
+            var changes = ChangesToPublish[typeof(T)].Changes;
+            if (changes.Contains(id)) return;
+            changes.Add(id);
+            PublishChanges<T>(entitiesDB, id, limit).RunOn(Scheduler.leanRunner);
         }
 
-        private static IEnumerator<TaskContract> PublishChanges<T>(EntitiesDB entitiesDB, EGID id)
+        private static IEnumerator<TaskContract> PublishChanges<T>(EntitiesDB entitiesDB, EGID id, int limit)
             where T : unmanaged, IEntityComponent
         {
             yield return Yield.It;
-            var entityStream = entitiesDB.GetEntityStream<T>();
+            while (ChangesToPublish[typeof(T)].PublishedCount >= limit)
-            if (entityStream is null)
+                yield return Yield.It;
-                yield break; // There is no entity stream for this type
-            var consumers = entityStream.GetConsumers();
-            if (consumers == null)
-            {
-                Console.WriteLine("Consumers is null");
-                yield break;
-            }
-
-            bool waitForConsumers;
-            do
-            {
-                waitForConsumers = false;
-                for (int i = 0; i < consumers.count; i++)
-                {
-                    var buffer = consumers[i].GetRingBuffer();
-                    if (buffer.Count + 1 <= buffer.Capacity) continue;
-                    waitForConsumers = true;
-                    break;
-                }
-
-                if (waitForConsumers) yield return Yield.It;
-            } while (waitForConsumers);
             entitiesDB.PublishEntityChange<T>(id);
-        }
+            var (count, changes) = ChangesToPublish[typeof(T)];
-
+            changes.Remove(id);
-        /// <summary>
+            ChangesToPublish[typeof(T)] = (count + 1, changes);
-        /// Query entities as OptionalRefs. The elements always exist, it's just a nice way to encapsulate the data.
+            yield return Yield.It;
-        /// </summary>
+            ChangesToPublish[typeof(T)] = (0, changes);
-        /// <param name="entitiesDB"></param>
-        /// <param name="group"></param>
-        /// <param name="select"></param>
-        /// <typeparam name="T"></typeparam>
-        /// <typeparam name="TR"></typeparam>
-        /// <returns></returns>
-        public static RefCollection<T> QueryEntitiesOptional<T>(this EntitiesDB entitiesDB, ExclusiveGroupStruct group) where T : unmanaged, IEntityComponent
-        {
-            var (buffer, ids, count) = entitiesDB.QueryEntities<T>(group);
-            return new RefCollection<T>(count, buffer, ids, group);
         }
     }
 }

TechbloxModdingAPI/Utility/OptionalRef.cs

@@ -4,9 +4,8 @@ using Svelto.ECS;
 
 namespace TechbloxModdingAPI.Utility
 {
-    public ref struct OptionalRef<T> where T : struct, IBaseEntityComponent
+    public ref struct OptionalRef<T> where T : struct, IEntityComponent
     {
-        private readonly EGID entityId;
         private readonly State state;
         private readonly uint index;
         private NB<T> array;
@@ -14,22 +13,19 @@ namespace TechbloxModdingAPI.Utility
         private readonly EntityInitializer initializer;
         //The possible fields are: (index && (array || managedArray)) || initializer
 
-        public OptionalRef(NB<T> array, uint index, EGID entityId = default)
+        public OptionalRef(NB<T> array, uint index)
         {
             state = State.Native;
             this.array = array;
             this.index = index;
-            this.entityId = entityId;
             initializer = default;
-            managedArray = default;
         }
 
-        public OptionalRef(MB<T> array, uint index, EGID entityId = default)
+        public OptionalRef(MB<T> array, uint index)
         {
             state = State.Managed;
             managedArray = array;
             this.index = index;
-            this.entityId = entityId;
             initializer = default;
             this.array = default;
         }
@@ -39,9 +35,8 @@ namespace TechbloxModdingAPI.Utility
         /// </summary>
         /// <param name="obj">The object with the initializer</param>
         /// <param name="unmanaged">Whether the struct is unmanaged</param>
-        public OptionalRef(EcsObjectBase obj, bool unmanaged, EGID entityId = default)
+        public OptionalRef(EcsObjectBase obj, bool unmanaged)
         {
-            this.entityId = entityId;
             if (obj.InitData.Valid)
             {
                 initializer = obj.InitData.Initializer(obj.Id);
@@ -54,7 +49,6 @@ namespace TechbloxModdingAPI.Utility
             }
             array = default;
             index = default;
-            managedArray = default;
         }
 
         /// <summary>
@@ -65,25 +59,11 @@ namespace TechbloxModdingAPI.Utility
         {
             CompRefCache.Default = default; //The default value can be changed by mods
             if (state == State.Empty) return ref CompRefCache.Default;
-            if ((state & State.Initializer) != State.Empty) return ref initializer.GetOrAdd<T>();
+            if ((state & State.Initializer) != State.Empty) return ref initializer.GetOrCreate<T>();
             if ((state & State.Native) != State.Empty) return ref array[index];
             return ref managedArray[index];
         }
 
-        /// <summary>
-        /// A non-by-ref view that allows getting and setting the value.
-        /// </summary>
-        public T Value
-        {
-            get => Get();
-            set => Get() = value;
-        }
-
-        /// <summary>
-        /// The ID of the entity this component belongs to or default if it doesn't exist.
-        /// </summary>
-        public EGID EGID => entityId;
-
         public bool Exists => state != State.Empty;
         public T? Nullable() => this ? Get() : default;
 
@@ -91,8 +71,6 @@ namespace TechbloxModdingAPI.Utility
 
         public static implicit operator bool(OptionalRef<T> opt) => opt.state != State.Empty;
 
-        public static implicit operator EGID(OptionalRef<T> opt) => opt.entityId;
-
         /// <summary>
         /// Creates an instance of a struct T that can be referenced.
         /// </summary>

TechbloxModdingAPI/Utility/RefCollection.cs

@@ -1,94 +0,0 @@
-using System;
-using Svelto.DataStructures;
-using Svelto.ECS;
-using Svelto.ECS.Hybrid;
-using Svelto.ECS.Internal;
-
-namespace TechbloxModdingAPI.Utility
-{
-    public readonly ref struct RefCollection<T> where T : struct, IBaseEntityComponent
-    {
-        private readonly bool managed;
-        private readonly int count;
-        private readonly NB<T> nativeArray;
-        private readonly MB<T> managedArray;
-        private readonly NativeEntityIDs nativeIDs;
-        private readonly ManagedEntityIDs managedIDs;
-        private readonly ExclusiveGroupStruct group;
-
-        public RefCollection(int count, MB<T> managedArray, ManagedEntityIDs managedIDs, ExclusiveGroupStruct group)
-        {
-            this.count = count;
-            this.managedArray = managedArray;
-            this.managedIDs = managedIDs;
-            this.group = group;
-            managed = true;
-            nativeArray = default;
-            nativeIDs = default;
-        }
-
-        public RefCollection(int count, NB<T> nativeArray, NativeEntityIDs nativeIDs, ExclusiveGroupStruct group)
-        {
-            this.count = count;
-            this.nativeArray = nativeArray;
-            this.nativeIDs = nativeIDs;
-            this.group = group;
-            managed = false;
-            managedArray = default;
-            managedIDs = default;
-        }
-
-        public Enumerator GetEnumerator() => new(this);
-        
-        /// <summary>
-        /// The amount of items in the collection.
-        /// </summary>
-        public int Count => count;
-
-        public T[] ToArray() => ToArray(a => a.Component);
-
-        public TA[] ToArray<TA>(Func<(T Component, EGID ID), TA> transformFunction, Predicate<(T Component, EGID ID)> predicateFunction = null)
-        {
-            var result = new TA[Count];
-            int i = 0;
-            foreach (var opt in this)
-            {
-                if (predicateFunction != null && !predicateFunction((opt.Get(), opt.EGID))) continue;
-                result[i] = transformFunction((opt.Get(), opt.EGID));
-                i++;
-            }
-
-            return result;
-        }
-
-        public ref struct Enumerator
-        {
-            private RefCollection<T> coll;
-            private int index;
-
-            public Enumerator(RefCollection<T> collection)
-            {
-                index = -1;
-                coll = collection;
-            }
-
-            public OptionalRef<T> Current
-            {
-                get
-                {
-                    if (coll.count <= index && index >= 0) return default;
-                    if (coll.managed)
-                        return new OptionalRef<T>(coll.managedArray, (uint)index,
-                            new EGID(coll.managedIDs[index], coll.group));
-                    return new OptionalRef<T>(coll.nativeArray, (uint)index,
-                        new EGID(coll.nativeIDs[index], coll.group));
-                }
-            }
-
-            public bool MoveNext()
-            {
-                return ++index < coll.count;
-            }
-        }
-    }
-}

TechbloxModdingAPI/Utility/WeakDictionary.cs

@@ -7,7 +7,7 @@ namespace TechbloxModdingAPI.Utility
 {
     public class WeakDictionary<TKey, TValue> : IDictionary<TKey, TValue> where TValue : class
     {
-        private readonly Dictionary<TKey, WeakReference<TValue>> _dictionary = new();
+        private Dictionary<TKey, WeakReference<TValue>> _dictionary = new Dictionary<TKey, WeakReference<TValue>>();
 
         public IEnumerator<KeyValuePair<TKey, TValue>> GetEnumerator()
         {
@@ -19,19 +19,53 @@ namespace TechbloxModdingAPI.Utility
             }
         }
 
-        IEnumerator IEnumerable.GetEnumerator() => GetEnumerator();
+        IEnumerator IEnumerable.GetEnumerator()
-        public void Add(KeyValuePair<TKey, TValue> item) => Add(item.Key, item.Value);
+        {
-        public void Clear() => _dictionary.Clear();
+            return GetEnumerator();
-        public bool Contains(KeyValuePair<TKey, TValue> item) =>
+        }
-            TryGetValue(item.Key, out var value) && item.Value == value;
+
-        public void CopyTo(KeyValuePair<TKey, TValue>[] array, int arrayIndex) =>
+        public void Add(KeyValuePair<TKey, TValue> item)
+        {
+            Add(item.Key, item.Value);
+        }
+
+        public void Clear()
+        {
+            _dictionary.Clear();
+        }
+
+        public bool Contains(KeyValuePair<TKey, TValue> item)
+        {
+            return TryGetValue(item.Key, out var value) && item.Value == value;
+        }
+
+        public void CopyTo(KeyValuePair<TKey, TValue>[] array, int arrayIndex)
+        {
             throw new System.NotImplementedException();
-        public bool Remove(KeyValuePair<TKey, TValue> item) => Contains(item) && Remove(item.Key);
+        }
+
+        public bool Remove(KeyValuePair<TKey, TValue> item)
+        {
+            return Contains(item) && Remove(item.Key);
+        }
+
         public int Count => _dictionary.Count;
         public bool IsReadOnly => false;
-        public bool ContainsKey(TKey key) => TryGetValue(key, out _);
+
-        public void Add(TKey key, TValue value) => _dictionary.Add(key, new WeakReference<TValue>(value));
+        public bool ContainsKey(TKey key)
-        public bool Remove(TKey key) => _dictionary.Remove(key);
+        {
+            return TryGetValue(key, out _);
+        }
+
+        public void Add(TKey key, TValue value)
+        {
+            _dictionary.Add(key, new WeakReference<TValue>(value));
+        }
+
+        public bool Remove(TKey key)
+        {
+            return _dictionary.Remove(key);
+        }
 
         public bool TryGetValue(TKey key, out TValue value)
         {
@@ -49,37 +83,12 @@ namespace TechbloxModdingAPI.Utility
             set => _dictionary[key] = new WeakReference<TValue>(value);
         }
 
-        public ICollection<TKey> Keys => new KeyCollection(this);
+        public ICollection<TKey> Keys => _dictionary.Keys;
         public ICollection<TValue> Values => new ValueCollection(this);
 
-        public class KeyCollection : ICollection<TKey>, IReadOnlyCollection<TKey>
-        {
-            private readonly WeakDictionary<TKey, TValue> _dictionary;
-            internal KeyCollection(WeakDictionary<TKey, TValue> dictionary)
-            {
-                _dictionary = dictionary;
-            }
-
-            public IEnumerator<TKey> GetEnumerator()
-            {
-                using var enumerator = _dictionary.GetEnumerator();
-                while (enumerator.MoveNext())
-                    yield return enumerator.Current.Key;
-            }
-            
-            IEnumerator IEnumerable.GetEnumerator() => GetEnumerator();
-            public void Add(TKey item) => throw new NotSupportedException("The key collection is read only.");
-            public void Clear() => throw new NotSupportedException("The key collection is read only.");
-            public bool Contains(TKey item) => _dictionary.ContainsKey(item);
-            public void CopyTo(TKey[] array, int arrayIndex) => throw new NotImplementedException();
-            public bool Remove(TKey item) => throw new NotSupportedException("The key collection is read only.");
-            public int Count => _dictionary.Count;
-            public bool IsReadOnly => true;
-        }
-
         public class ValueCollection : ICollection<TValue>, IReadOnlyCollection<TValue>
         {
-            private readonly WeakDictionary<TKey, TValue> _dictionary;
+            private WeakDictionary<TKey, TValue> _dictionary;
             internal ValueCollection(WeakDictionary<TKey, TValue> dictionary)
             {
                 _dictionary = dictionary;
@@ -89,15 +98,41 @@ namespace TechbloxModdingAPI.Utility
             {
                 using var enumerator = _dictionary.GetEnumerator();
                 while (enumerator.MoveNext())
+                {
                     yield return enumerator.Current.Value;
                 }
+            }
+
+            IEnumerator IEnumerable.GetEnumerator()
+            {
+                return GetEnumerator();
+            }
+
+            public void Add(TValue item)
+            {
+                throw new NotSupportedException("The value collection is read only.");
+            }
+
+            public void Clear()
+            {
+                throw new NotSupportedException("The value collection is read only.");
+            }
+
+            public bool Contains(TValue item)
+            {
+                return _dictionary.Any(kv => kv.Value == item);
+            }
+
+            public void CopyTo(TValue[] array, int arrayIndex)
+            {
+                throw new NotImplementedException();
+            }
+
+            public bool Remove(TValue item)
+            {
+                throw new NotSupportedException("The value collection is read only.");
+            }
 
-            IEnumerator IEnumerable.GetEnumerator() => GetEnumerator();
-            public void Add(TValue item) => throw new NotSupportedException("The value collection is read only.");
-            public void Clear() => throw new NotSupportedException("The value collection is read only.");
-            public bool Contains(TValue item) => _dictionary.Any(kv => kv.Value == item);
-            public void CopyTo(TValue[] array, int arrayIndex) => throw new NotImplementedException();
-            public bool Remove(TValue item) => throw new NotSupportedException("The value collection is read only.");
             public int Count => _dictionary.Count;
             public bool IsReadOnly => true;
         }