1 files changed, 19 insertions, 6 deletions

diff --git a/src/studio.cpp b/src/studio.cpp
index 1b7b77d..d472fbc 100644
--- a/src/studio.cpp
+++ b/src/studio.cpp
@@ -211,12 +211,26 @@ void *chams_material() {
   return g_chams_mat;
 }
 
-void chams_apply(void *mat, bool ignorez) {
+void chams_color(void *mat, float r, float g, float b) {
   using ColorFn = void (*)(void *, float, float, float);
+  using FindVarFn = void *(*)(void *, const char *, bool *, bool);
+  using SetVec3Fn = void (*)(void *, float, float, float);
+  bool found = false;
+  void *var = vfunc<FindVarFn>(mat, 11)(mat, "$color", &found, false);
+  if (var)
+    vfunc<SetVec3Fn>(var, 11)(var, r, g, b);
+  found = false;
+  var = vfunc<FindVarFn>(mat, 11)(mat, "$color2", &found, false);
+  if (var)
+    vfunc<SetVec3Fn>(var, 11)(var, r, g, b);
+  vfunc<ColorFn>(mat, 28)(mat, r, g, b);
+}
+
+void chams_apply(void *mat, bool ignorez) {
   using FlagFn = void (*)(void *, int, bool);
-  vfunc<ColorFn>(mat, 28)(mat, (float)g_chams_rgb[0] / 255.0f,
-                          (float)g_chams_rgb[1] / 255.0f,
-                          (float)g_chams_rgb[2] / 255.0f);
+  chams_color(mat, (float)g_chams_rgb[0] / 255.0f,
+              (float)g_chams_rgb[1] / 255.0f,
+              (float)g_chams_rgb[2] / 255.0f);
   vfunc<FlagFn>(mat, 29)(mat, 1 << 15 /* MATERIAL_VAR_IGNOREZ */, ignorez);
 }
 
@@ -240,10 +254,9 @@ void hkdModelExecute(void *self, const void *state, const void *info,
         g_orig_model_exec(self, state, info, bone);
         g_force_override(self, nullptr, 0);
         // debugwhite is shared: restore colour and depth behaviour.
-        using ColorFn = void (*)(void *, float, float, float);
         using FlagFn = void (*)(void *, int, bool);
         vfunc<FlagFn>(mat, 29)(mat, 1 << 15, false);
-        vfunc<ColorFn>(mat, 28)(mat, 1.0f, 1.0f, 1.0f);
+        chams_color(mat, 1.0f, 1.0f, 1.0f);
         return;
       }
     }