SkytechThrusters setup

Author: ari-steas

Utility Mods/SkytechThrusters/Data/Scripts/Skytech.Thrusters/AssemblyBase.cs

@@ -0,0 +1,84 @@
+using Sandbox.ModAPI;
+using System.Collections.Generic;
+using VRage.Game.ModAPI;
+using Skytech.Thrusters.ModularAssemblies.Communication;
+
+namespace Skytech.Thrusters
+{
+    /// <summary>
+    /// Single assembly instance logic class. You should put your assembly logic here!
+    /// </summary>
+    internal abstract class AssemblyBase
+    {
+        private static ModularDefinitionApi ModularApi => ModularAssemblies.ModularDefinition.ModularApi;
+
+        public int AssemblyId { get; private set; }
+        public IMyCubeGrid Grid { get; private set; }
+
+        private HashSet<IMyCubeBlock> _blocks = new HashSet<IMyCubeBlock>();
+
+        protected AssemblyBase()
+        {
+            // fuckery to get around generic constructor limitations
+        }
+
+        public static TAssembly Create<TAssembly>(int assemblyId) where TAssembly : AssemblyBase, new()
+        {
+            TAssembly asm = new TAssembly
+            {
+                AssemblyId = assemblyId,
+                Grid = ModularApi.GetAssemblyGrid(assemblyId)
+            };
+
+            asm.Init();
+            return asm;
+        }
+
+        protected virtual void Init()
+        {
+        }
+
+        /// <summary>
+        /// Triggered once per game tick (60ups).
+        /// </summary>
+        public virtual void UpdateTick()
+        {
+        }
+
+        /// <summary>
+        /// Triggered when the assembly is closed (zero blocks contained).
+        /// </summary>
+        public virtual void Unload()
+        {
+        }
+
+        /// <summary>
+        /// Triggers whenever a new part is added to an assembly.
+        /// </summary>
+        /// <param name="block"></param>
+        /// <param name="isBasePart"></param>
+        public virtual void OnPartAdd(IMyCubeBlock block, bool isBasePart)
+        {
+            _blocks.Add(block);
+        }
+
+        /// <summary>
+        /// Triggers whenever a part is removed from an assembly.
+        /// </summary>
+        /// <param name="block"></param>
+        /// <param name="isBasePart"></param>
+        public virtual void OnPartRemove(IMyCubeBlock block, bool isBasePart)
+        {
+            _blocks.Remove(block);
+        }
+
+        /// <summary>
+        /// Triggers whenever a part is destroyed, just after OnPartRemove.
+        /// </summary>
+        /// <param name="block"></param>
+        /// <param name="isBasePart"></param>
+        public virtual void OnPartDestroy(IMyCubeBlock block, bool isBasePart)
+        {
+        }
+    }
+}

Utility Mods/SkytechThrusters/Data/Scripts/Skytech.Thrusters/AssemblyManager.cs

@@ -0,0 +1,91 @@
+using System.Collections.Generic;
+using VRage.Game.ModAPI;
+using Skytech.Thrusters.ModularAssemblies.Communication;
+namespace Skytech.Thrusters
+{
+    internal class AssemblyManager<TAssembly> : IAssemblyManager
+        where TAssembly : AssemblyBase, new()
+    {
+        // This class is a singleton.
+        protected static AssemblyManager<TAssembly> I { get; private set; } = null;
+        private static ModularDefinitionApi ModularApi => ModularAssemblies.ModularDefinition.ModularApi;
+
+
+        private Dictionary<int, AssemblyBase> _assemblies = new Dictionary<int, AssemblyBase>();
+
+
+        public static void Load()
+        {
+            if (I != null)
+                return;
+            I = new AssemblyManager<TAssembly>();
+        }
+
+        public void Unload()
+        {
+            foreach (var system in _assemblies.Values)
+            {
+                system.Unload();
+            }
+            I = null;
+        }
+
+        public void Update()
+        {
+            foreach (var assembly in _assemblies.Values)
+            {
+                assembly.UpdateTick();
+            }
+        }
+
+        public static void OnPartAdd(int assemblyId, IMyCubeBlock block, bool isBaseBlock)
+        {
+            if (I == null)
+                return;
+
+            // find assembly, and register new one if not present.
+            // ID is unique per-session
+            AssemblyBase assemblyBase;
+            if (!I._assemblies.TryGetValue(assemblyId, out assemblyBase))
+            {
+                assemblyBase = AssemblyBase.Create<TAssembly>(assemblyId);
+                I._assemblies.Add(assemblyId, assemblyBase);
+                ModularApi.Log($"AssemblyManager created new assembly {assemblyId}.");
+            }
+
+            assemblyBase.OnPartAdd(block, isBaseBlock);
+        }
+
+        public static void OnPartRemove(int assemblyId, IMyCubeBlock block, bool isBaseBlock)
+        {
+            // find assembly, and skip if not present.
+            AssemblyBase assemblyBase;
+            if (I == null || !I._assemblies.TryGetValue(assemblyId, out assemblyBase))
+                return;
+
+            assemblyBase.OnPartRemove(block, isBaseBlock);
+        }
+
+        public static void OnPartDestroy(int assemblyId, IMyCubeBlock block, bool isBaseBlock)
+        {
+            // find assembly, and skip if not present.
+            AssemblyBase assemblyBase;
+            if (I == null || !I._assemblies.TryGetValue(assemblyId, out assemblyBase))
+                return;
+
+            assemblyBase.OnPartDestroy(block, isBaseBlock);
+        }
+
+        public static void OnAssemblyClose(int assemblyId)
+        {
+            // find assembly, and skip if not present.
+            AssemblyBase assemblyBase;
+            if (I == null || !I._assemblies.TryGetValue(assemblyId, out assemblyBase))
+                return;
+
+            assemblyBase.Unload();
+            I._assemblies.Remove(assemblyId);
+            ModularApi.Log($"AssemblyManager removed assembly {assemblyId}.");
+        }
+    }
+}

Utility Mods/SkytechThrusters/Data/Scripts/Skytech.Thrusters/IAssemblyManager.cs

@@ -0,0 +1,8 @@
+namespace Skytech.Thrusters
+{
+    internal interface IAssemblyManager
+    {
+        void Unload();
+        void Update();
+    }
+}

Utility Mods/SkytechThrusters/Data/Scripts/Skytech.Thrusters/ModularAssemblies/Communication/DefinitionCollector.cs

@@ -0,0 +1,26 @@
+using Skytech.Thrusters.ModularAssemblies.Communication;
+using static Skytech.Thrusters.ModularAssemblies.Communication.DefinitionDefs;
+
+// ReSharper disable once CheckNamespace
+namespace Skytech.Thrusters.ModularAssemblies
+{
+    internal partial class ModularDefinition
+    {
+        internal static ModularDefinitionApi ModularApi = new ModularDefinitionApi();
+        internal ModularDefinitionContainer Container = new ModularDefinitionContainer();
+
+        internal void LoadDefinitions(params ModularPhysicalDefinition[] defs)
+        {
+            Container.PhysicalDefs = defs;
+        }
+
+        /// <summary>
+        ///     Load all definitions for DefinitionSender
+        /// </summary>
+        /// <param name="baseDefs"></param>
+        internal static ModularDefinitionContainer GetBaseDefinitions()
+        {
+            return new Skytech.Thrusters.ModularAssemblies.ModularDefinition().Container;
+        }
+    }
+}