Add setting for tech node discount

KSP-RO · Apr 8, 2023 · 83691b0 · 83691b0
1 parent ad6ff4c
commit 83691b0
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/TestFlightCore/TestFlightCore/TestFlight.cs b/TestFlightCore/TestFlightCore/TestFlight.cs
@@ -405,8 +405,10 @@ public class TestFlightManagerScenario : ScenarioModule
 
         [KSPField(isPersistant = true)] public bool settingsEnabled = true;
         [KSPField(isPersistant = true)] public bool settingsAlwaysMaxData = false;
+        [KSPField(isPersistant = true)] public bool settingsTechDiscountEnabled = false;
         public bool SettingsEnabled { get { return settingsEnabled; } set { settingsEnabled = value; } }
         public bool SettingsAlwaysMaxData { get { return settingsAlwaysMaxData; } set { settingsAlwaysMaxData = value; } }
+        public bool SettingsTechDiscountEnabled { get { return settingsTechDiscountEnabled; } set { settingsTechDiscountEnabled = value; } }
 
         private bool rp1Available = false;
         private bool careerLogging = false;

diff --git a/TestFlightCore/TestFlightCore/TestFlightKSC.cs b/TestFlightCore/TestFlightCore/TestFlightKSC.cs
@@ -308,6 +308,13 @@ internal override void DrawWindow(int id)
                                 tfScenario.SettingsAlwaysMaxData = saveMaxData;
                             }
                             GUILayout.EndHorizontal();
+                            GUILayout.BeginHorizontal();
+                            bool techDiscountEnabled = tfScenario.SettingsTechDiscountEnabled;
+                            if (DrawToggle(ref techDiscountEnabled, "Apply discount to tech node cost for flown parts", Styles.styleToggle))
+                            {
+                                tfScenario.SettingsTechDiscountEnabled = techDiscountEnabled;
+                            }
+                            GUILayout.EndHorizontal();
                             break;
                     }
                     break;

diff --git a/TestFlightCore/TestFlightCore/TestFlightRnD.cs b/TestFlightCore/TestFlightCore/TestFlightRnD.cs
@@ -22,7 +22,8 @@ public void Start()
 
         public void OnTreeSpawn(RDController controller)
         {
-            if (TestFlightManagerScenario.Instance == null || controller.nodes == null)
+            if (TestFlightManagerScenario.Instance == null || controller.nodes == null ||
+                !TestFlightManagerScenario.Instance.SettingsTechDiscountEnabled)
             {
                 return;
             }