Possible colors patch fix?

SpaceWarpDev · Oct 31, 2023 · b48db52 · b48db52
1 parent 9b27c29
commit b48db52
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/SpaceWarp.Core/Patching/ColorsPatch.cs b/SpaceWarp.Core/Patching/ColorsPatch.cs
@@ -6,6 +6,7 @@
 using KSP.Game;
 using KSP.Modules;
 using KSP.OAB;
+using KSP.Sim.impl;
 using SpaceWarp.API.Assets;
 using UnityEngine;
 
@@ -42,6 +43,24 @@ public static void Prefix(IObjectAssemblyAvailablePart obj, ref GameObject prefa
         }
     }
 
+    [HarmonyPrefix]
+    [HarmonyPatch(typeof(SimulationObjectView), nameof(SimulationObjectView.InitializeView))]
+    public static void UpdateColorsInFlight(GameObject instance)
+    {
+        foreach (var renderer in instance.GetComponentsInChildren<Renderer>())
+        {
+            var shaderName = renderer.material.shader.name;
+            if (shaderName is not ("Parts Replace" or "KSP2/Parts/Paintable")) continue;
+            Material material;
+            var mat = new Material(Shader.Find(Ksp2OpaquePath))
+            {
+                name = (material = renderer.material).name
+            };
+            mat.CopyPropertiesFromMaterial(material);
+            renderer.material = mat;
+        }
+    }
+
     //Everything below this point will be removed in the next patch
     private const int Diffuse = 0;
     private const int Metallic = 1;