feat: non-pbr tank lighting and model cache

src/hooks/useArmor.ts

@@ -1,7 +1,11 @@
-import { useLoader } from '@react-three/fiber';
-import { GLTFLoader } from 'three-stdlib';
+import { ObjectMap, useLoader } from '@react-three/fiber';
+import { GLTF, GLTFLoader } from 'three-stdlib';
 import { asset } from '../core/blitzkrieg/asset';
 
+const cache: Record<number, GLTF & ObjectMap> = {};
+
 export function useArmor(id: number) {
-  return useLoader(GLTFLoader, asset(`3d/tanks/armor/${id}.glb`));
+  const gltf = useLoader(GLTFLoader, asset(`3d/tanks/armor/${id}.glb`));
+  if (!cache[id]) cache[id] = gltf;
+  return cache[id];
 }

src/hooks/useModel.ts

@@ -1,7 +1,28 @@
-import { useLoader } from '@react-three/fiber';
-import { GLTFLoader } from 'three-stdlib';
+import { ObjectMap, useLoader } from '@react-three/fiber';
+import { Mesh, MeshStandardMaterial } from 'three';
+import { GLTF, GLTFLoader } from 'three-stdlib';
 import { asset } from '../core/blitzkrieg/asset';
 
+const cache: Record<
+  number,
+  {
+    gltf: GLTF & ObjectMap;
+    hasPbr: boolean;
+  }
+> = {};
+
 export function useModel(id: number) {
-  return useLoader(GLTFLoader, asset(`3d/tanks/models/${id}.glb`));
+  const gltf = useLoader(GLTFLoader, asset(`3d/tanks/models/${id}.glb`));
+  if (!cache[id]) {
+    cache[id] = {
+      gltf,
+      hasPbr: Object.values(gltf.nodes).some(
+        (node) =>
+          node instanceof Mesh &&
+          node.material instanceof MeshStandardMaterial &&
+          node.material.roughnessMap !== null,
+      ),
+    };
+  }
+  return cache[id];
 }