flatten armor dir

packages/website/src/components/PrimaryArmorScene.tsx

@@ -0,0 +1,137 @@
+import { memo, useRef } from "react";
+import { Group } from "three";
+import { correctZYTuple } from "../core/blitz/correctZYTuple";
+import { nameToArmorId } from "../core/blitzkit/nameToArmorId";
+import { resolveArmor } from "../core/blitzkit/resolveThickness";
+import { useArmor } from "../hooks/useArmor";
+import { DuelModule, useDuel } from "../hooks/useDuel";
+import { ModelTankWrapper } from "./ModelTankWrapper";
+import { PrimaryArmorSceneComponent } from "./PrimaryArmorSceneComponent";
+
+export const PrimaryArmorScene = memo(() => {
+  const wrapper = useRef<Group>(null);
+  const turretContainer = useRef<Group>(null!);
+  const gunContainer = useRef<Group>(null!);
+  const tank = useDuel("protagonist", DuelModule.Tank);
+  const track = useDuel("protagonist", DuelModule.Track);
+  const turret = useDuel("protagonist", DuelModule.Turret);
+  const gun = useDuel("protagonist", DuelModule.Gun);
+  const armorGltf = useArmor(tank.id).gltf;
+  const armorNodes = Object.values(armorGltf.nodes);
+  const tankModelDefinition = useTankModelDefinition();
+  const trackModelDefinition = tankModelDefinition.tracks[track.id];
+  const turretModelDefinition = tankModelDefinition.turrets[turret.id];
+  const gunModelDefinition = turretModelDefinition.guns[gun.id];
+  const hullOrigin = correctZYTuple(trackModelDefinition.origin!);
+  const turretOrigin = correctZYTuple(tankModelDefinition.turret_origin!);
+  const gunOrigin = correctZYTuple(turretModelDefinition.gun_origin!);
+  const isDynamicArmorActive = Duel.use((state) =>
+    state.protagonist.consumables.includes(73),
+  );
+
+  useTankTransform(track, turret, turretContainer, gunContainer);
+
+  return (
+    <ModelTankWrapper ref={wrapper}>
+      <group position={hullOrigin}>
+        {armorNodes.map((node) => {
+          const isHull = node.name.startsWith("hull_");
+          const isVisible = isHull;
+          const armorId = nameToArmorId(node.name);
+          const { spaced, thickness } = resolveArmor(
+            tankModelDefinition.armor!,
+            armorId,
+          );
+
+          if (
+            !isVisible ||
+            spaced ||
+            thickness === undefined ||
+            (isDynamicArmorActive && node.name.includes("state_01")) ||
+            (!isDynamicArmorActive && node.name.includes("state_00"))
+          )
+            return null;
+
+          return (
+            <PrimaryArmorSceneComponent
+              key={node.uuid}
+              thickness={thickness}
+              node={node}
+            />
+          );
+        })}
+      </group>
+
+      <group ref={turretContainer}>
+        {armorNodes.map((node) => {
+          const isCurrentTurret = node.name.startsWith(
+            `turret_${turretModelDefinition.model_id
+              .toString()
+              .padStart(2, "0")}`,
+          );
+          const isVisible = isCurrentTurret;
+          const armorId = nameToArmorId(node.name);
+          const { spaced, thickness } = resolveArmor(
+            turretModelDefinition.armor!,
+            armorId,
+          );
+
+          if (
+            !isVisible ||
+            spaced ||
+            thickness === undefined ||
+            (isDynamicArmorActive && node.name.includes("state_01")) ||
+            (!isDynamicArmorActive && node.name.includes("state_00"))
+          )
+            return null;
+
+          return (
+            <group position={hullOrigin} key={node.uuid}>
+              <group position={turretOrigin}>
+                <PrimaryArmorSceneComponent
+                  key={node.uuid}
+                  thickness={thickness}
+                  node={node}
+                />
+              </group>
+            </group>
+          );
+        })}
+
+        <group ref={gunContainer}>
+          {armorNodes.map((node) => {
+            const isCurrentGun = node.name.startsWith(
+              `gun_${gunModelDefinition.model_id.toString().padStart(2, "0")}`,
+            );
+            const isVisible = isCurrentGun;
+            const armorId = nameToArmorId(node.name);
+            const { spaced, thickness } = resolveArmor(
+              gunModelDefinition.armor!,
+              armorId,
+            );
+
+            if (
+              !isVisible ||
+              spaced ||
+              thickness === undefined ||
+              (isDynamicArmorActive && node.name.includes("state_01")) ||
+              (!isDynamicArmorActive && node.name.includes("state_00"))
+            )
+              return null;
+
+            return (
+              <group position={hullOrigin} key={node.uuid}>
+                <group position={turretOrigin.clone().add(gunOrigin)}>
+                  <PrimaryArmorSceneComponent
+                    thickness={thickness}
+                    node={node}
+                  />
+                </group>
+              </group>
+            );
+          })}
+        </group>
+      </group>
+    </ModelTankWrapper>
+  );
+});