1 files changed, 50 insertions, 15 deletions

diff --git a/src/studio.cpp b/src/studio.cpp
index 1ca8ad7..44d1bb4 100644
--- a/src/studio.cpp
+++ b/src/studio.cpp
@@ -19,6 +19,7 @@
 #include <cstdio>
 #include <cstring>
 #include <dlfcn.h>
+#include <time.h>
 
 namespace {
 
@@ -196,34 +197,37 @@ using ForceOverrideFn = void (*)(void *, void *, int);
 void *g_modelrender = nullptr;
 ModelExecFn g_orig_model_exec = nullptr;
 ForceOverrideFn g_force_override = nullptr;
-void *g_chams_mat = nullptr;
-int g_chams_mat_idx = -1;
+void *g_style_mats[6] = {};
+
+void *viewmodel_material();
 
 const char *chams_material_name(int idx) {
-  (void)idx;
-  // Only this debug material is reliably resident in the base game; the other
-  // debug VMTs are not loaded and FindMaterial does not make them available.
-  // Visual styles are applied as material flags instead.
-  return "debug/debugtranslucentsinglecolor";
+  switch (idx) {
+  case 1: return "debug/debugwireframe";
+  case 2: return "debug/additivevertexcolorvertexalpha";
+  case 3: return "debug/debugtranslucentsinglecolor";
+  case 4: return "debug/debugtranslucentvertexcolor";
+  case 5: return "debug/debugmrmfullbright2";
+  default: return "models/debug/debugwhite";
+  }
 }
 
 void *chams_material() {
   int idx = g_chams_mat_index;
   if (idx < 0) idx = 0;
   if (idx > 5) idx = 5;
-  if (g_chams_mat && g_chams_mat_idx == idx)
-    return g_chams_mat;
-  g_chams_mat = world_find_material(chams_material_name(idx));
-  if (!g_chams_mat)
-    g_chams_mat = world_find_material("debug/debugtranslucentsinglecolor");
-  g_chams_mat_idx = idx;
-  logf("chams: style %d -> %p", idx, g_chams_mat);
-  return g_chams_mat;
+  if (g_style_mats[idx])
+    return g_style_mats[idx];
+  if (g_style_mats[3])
+    return g_style_mats[3];
+  return viewmodel_material();
 }
 
 // Viewmodels need a model shader; the VGUI white materials used by some
 // styles do not draw skinned models and make the hands/weapon disappear.
 void *viewmodel_material() {
+  if (g_style_mats[3])
+    return g_style_mats[3];
   static void *mat = nullptr;
   if (!mat)
     mat = world_find_material("debug/debugtranslucentsinglecolor");
@@ -307,6 +311,37 @@ void hkdModelExecute(void *self, const void *state, const void *info,
 
 } // namespace
 
+static double chams_now() {
+  timespec ts;
+  clock_gettime(CLOCK_MONOTONIC, &ts);
+  return (double)ts.tv_sec + (double)ts.tv_nsec * 1e-9;
+}
+
+// Called from outer_draw (outside the model draw) so FindMaterial can load
+// the style VMTs. The render hook only reads the cached pointers.
+void chams_preload() {
+  static double last = 0.0;
+  static bool probed = false;
+  double now = chams_now();
+  if (now - last < 1.5)
+    return;
+  last = now;
+  if (!probed) {
+    probed = true;
+    void *probe = world_find_material("vgui/white");
+    logf("chams: probe vgui/white = %p", probe);
+  }
+  for (int i = 0; i < 6; i++) {
+    if (g_style_mats[i])
+      continue;
+    void *m = world_find_material(chams_material_name(i));
+    if (m) {
+      g_style_mats[i] = m;
+      logf("chams: preloaded style %d %s = %p", i, chams_material_name(i), m);
+    }
+  }
+}
+
 void chams_init() {
   if (g_modelrender || !g_iface.modelrender) {
     if (!g_iface.modelrender)