omnitech

BattletechModders · Dec 9, 2018 · cef7dd2 · cef7dd2
1 parent ebbb847
commit cef7dd2
Show file tree

Hide file tree

Showing 5 changed files with 41 additions and 10 deletions.
diff --git a/mod.json b/mod.json
@@ -23,15 +23,18 @@
 			"G": 180,
 			"B": 180,
 			"A": 10
-		 },
-		 "DefaultFlagOverlayCColor": {
+		},
+		"DefaultFlagOverlayCColor": {
 			"R": 180,
 			"G": 255,
 			"B": 180,
 			"A": 10
-		 },
+		},
 
-		 "TestEnableAllTags": false
+		"TestEnableAllTags": false,
+		"OmniTechFlag" = "cc_omnitech",
+        "OmniTechCostBySize" = false,
+        "OmniTechInstallCost" = 1
 	},	
 
 	"Manifest": [ ]

diff --git a/source/CCLight/Registry.cs b/source/CCLight/Registry.cs
@@ -94,9 +94,10 @@ internal static void ProcessCustomFactories(object target, Dictionary<string, ob
                 return;
             }
 
-            //Control.Logger.LogDebug($"ProcessCustomCompontentFactories for {target.GetType()} ({target.GetHashCode()})");
-            //Control.Logger.LogDebug($"- {identifier}");
-
+#if CCDEBUG
+            Control.Logger.LogDebug($"ProcessCustomCompontentFactories for {target.GetType()} ({target.GetHashCode()})");
+            Control.Logger.LogDebug($"- {identifier}");
+#endif
             foreach (var preProcessor in PreProcessors)
             {
                 preProcessor.PreProcess(target, values);
@@ -110,17 +111,24 @@ internal static void ProcessCustomFactories(object target, Dictionary<string, ob
                     continue;
                 }
 
+#if CCDEBUG
                 Control.Logger.Log($"Created {factory} for {identifier}");
+
+#endif
                 Database.SetCustomWithIdentifier(identifier, component);
 
                 if (component is IAfterLoad load)
                 {
-                    //Control.Logger.LogDebug($"IAfterLoad: {identifier}");
+#if CCDEBUG
+                    Control.Logger.LogDebug($"IAfterLoad: {identifier}");
+#endif
                     load.OnLoaded(values);
                 }
             }
 
-            //Control.Logger.LogDebug("- done");
+#if CCDEBUG
+            Control.Logger.LogDebug("- done");
+#endif
         }
     }
 }
diff --git a/source/Control.cs b/source/Control.cs
@@ -54,7 +54,7 @@ public static void Init(string directory, string settingsJSON)
                 Registry.RegisterSimpleCustomComponents(Assembly.GetExecutingAssembly());
                 Validator.RegisterMechValidator(CategoryController.ValidateMech, CategoryController.ValidateMechCanBeFielded);
 
-                Logger.Log("Loaded CustomComponents v0.8.3.1 for bt 1.3");
+                Logger.Log("Loaded CustomComponents v0.8.4.0 for bt 1.3");
 #if CCDEBUG
                 Logger.LogDebug("Loading Categories");
 #endif  

diff --git a/source/CustomComponentSettings.cs b/source/CustomComponentSettings.cs
@@ -57,6 +57,9 @@ public class CustomComponentSettings
         [JsonIgnore] public Dictionary<string, Color> ColorTagsDictionary;
 
         public bool TestEnableAllTags = false;
+        public string OmniTechFlag = "cc_omnitech";
+        public bool OmniTechCostBySize = false;
+        public int OmniTechInstallCost = 1;
 
         public void Complete()
         {

diff --git a/source/Fixes/SimGameState_CreateComponentInstallWorkOrder.cs b/source/Fixes/SimGameState_CreateComponentInstallWorkOrder.cs
@@ -13,6 +13,7 @@ public static void FixCost(
             MechComponentRef mechComponent,
             ChassisLocations newLocation,
             ChassisLocations previousLocation,
+            string mechSimGameUID,
             WorkOrderEntry_InstallComponent __result)
         {
 #if CCDEBUG
@@ -44,6 +45,22 @@ public static void FixCost(
                     return;
 
                 }
+                MechDef mechByID = __instance.GetMechByID(mechSimGameUID);
+#if CCDEBUG
+                if (mechByID == null)
+                    Control.Logger.LogDebug("-- no mech found!");
+#endif
+                if (mechByID != null && mechByID.Chassis.ChassisTags.Contains(Control.Settings.OmniTechFlag))
+                {
+#if CCDEBUG
+                    if (mechByID == null)
+                        Control.Logger.LogDebug("-- mech is omni!");
+#endif
+
+                    tr.Value = (Control.Settings.OmniTechCostBySize ? mechComponent.Def.InventorySize / 2 : 1) * Control.Settings.OmniTechInstallCost;
+                }
+
+
                 if (tr.Value == 0)
                     tr.Value = 1;
             }