Chams: preload style materials outside the model draw

FindMaterial inside DrawModelExecute did not load the style VMTs, so every
style fell back to the solid material. Preload the six style materials from
outer_draw (throttled) and have the render hook use the cached pointers only.
world_find_material now returns only the exact requested material; the caller
falls back to the solid style. Add a vgui/white probe and per-style preload
logs.

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)