flatten armor dir

packages/website/src/components/SpacedArmorSubSpaced/index.tsx

@@ -0,0 +1,180 @@
+import { isExplosive, resolvePenetrationCoefficient } from "@blitzkit/core";
+import { useEffect } from "react";
+import {
+  AdditiveBlending,
+  MeshBasicMaterial,
+  ShaderMaterial,
+  type Object3D,
+} from "three";
+import { degToRad } from "three/src/math/MathUtils.js";
+import type { ArmorUserData } from "../SpacedArmorSceneComponent";
+import { hasEquipment } from "../../core/blitzkit/hasEquipment";
+import { jsxTree } from "../../core/blitzkit/jsxTree";
+import { Duel } from "../../stores/duel";
+import { Tankopedia } from "../../stores/tankopedia";
+import { ArmorType } from "./SpacedArmorScene";
+import fragmentShader from "./shaders/fragment.glsl?raw";
+import vertexShader from "./shaders/vertex.glsl?raw";
+
+interface SpacedArmorSubSpacedProps {
+  node: Object3D;
+  thickness: number;
+}
+
+const depthWriteMaterial = new MeshBasicMaterial({
+  depthWrite: true,
+  colorWrite: false,
+});
+
+export function SpacedArmorSubSpaced({
+  node,
+  thickness,
+}: SpacedArmorSubSpacedProps) {
+  const material = new ShaderMaterial({
+    fragmentShader,
+    vertexShader,
+
+    depthTest: true,
+    depthWrite: false,
+    blending: AdditiveBlending,
+
+    uniforms: {
+      thickness: { value: null },
+      penetration: { value: null },
+      caliber: { value: null },
+      ricochet: { value: null },
+      normalization: { value: null },
+    },
+  });
+
+  useEffect(() => {
+    function handleShellChange() {
+      const tankopediaEphemeral = Tankopedia.state;
+      const shell =
+        tankopediaEphemeral.customShell ?? Duel.state.antagonist.shell;
+
+      material.uniforms.penetration.value = shell.penetration!.near;
+      material.uniforms.caliber.value = shell.caliber;
+      material.uniforms.ricochet.value = degToRad(
+        isExplosive(shell.type) ? 90 : shell.ricochet!,
+      );
+      material.uniforms.normalization.value = degToRad(
+        shell.normalization ?? 0,
+      );
+    }
+    function handleProtagonistEquipmentChange() {
+      const equipment = Duel.state.protagonist.equipmentMatrix;
+      const hasEnhancedArmor = hasEquipment(
+        110,
+        Duel.state.protagonist.tank.equipment_preset,
+        equipment,
+      );
+      const equalizer =
+        (Duel.state.equalize
+          ? Duel.state.protagonist.tank.equalizer?.armor
+          : undefined) ?? 1;
+
+      material.uniforms.thickness.value =
+        thickness * (hasEnhancedArmor ? 1.04 : 1) * equalizer;
+    }
+    function handleAntagonistEquipmentChange() {
+      const equipment = Duel.state.antagonist.equipmentMatrix;
+      const tankopediaEphemeral = Tankopedia.state;
+      const shell =
+        tankopediaEphemeral.customShell ?? Duel.state.antagonist.shell;
+      const penetration = shell.penetration!.near;
+      const hasCalibratedShells = hasEquipment(
+        103,
+        Duel.state.antagonist.tank.equipment_preset,
+        equipment,
+      );
+      const equalize =
+        (Duel.state.equalize
+          ? Duel.state.antagonist.tank.equalizer?.penetration
+          : undefined) ?? 1;
+
+      material.uniforms.penetration.value =
+        penetration *
+        resolvePenetrationCoefficient(
+          hasCalibratedShells,
+          Duel.state.equalize,
+          shell.type,
+          Duel.state.antagonist.tank.equalizer,
+        ) *
+        equalize;
+    }
+
+    handleShellChange();
+    handleProtagonistEquipmentChange();
+    handleAntagonistEquipmentChange();
+
+    const unsubscribes = [
+      Duel.on((state) => state.antagonist.shell, handleShellChange),
+      Tankopedia.on((state) => state.customShell, handleShellChange),
+      Duel.on(
+        (state) => state.protagonist.equipmentMatrix,
+        handleProtagonistEquipmentChange,
+      ),
+      Duel.on(
+        (state) => state.antagonist.equipmentMatrix,
+        handleAntagonistEquipmentChange,
+      ),
+      Duel.on(
+        (state) => state.equalize,
+        () => {
+          handleProtagonistEquipmentChange();
+          handleAntagonistEquipmentChange();
+        },
+      ),
+    ];
+
+    return () => {
+      unsubscribes.forEach((unsubscribe) => unsubscribe());
+    };
+  });
+
+  return (
+    <>
+      {jsxTree(node, {
+        group(_, props, key) {
+          return <group {...props} key={`${key}-spaced-sub-spaced-exclude`} />;
+        },
+
+        mesh(_, props, key) {
+          return (
+            <mesh
+              {...props}
+              key={`${key}-spaced-sub-spaced-exclude`}
+              renderOrder={2}
+              material={material}
+              onClick={() => {}}
+              userData={
+                {
+                  type: ArmorType.Spaced,
+                  thickness,
+                } satisfies ArmorUserData
+              }
+            />
+          );
+        },
+      })}
+
+      {jsxTree(node, {
+        group(_, props, key) {
+          return <group {...props} key={`${key}-spaced-sub-spaced-include`} />;
+        },
+
+        mesh(_, props, key) {
+          return (
+            <mesh
+              {...props}
+              key={`${key}-spaced-sub-spaced-include`}
+              renderOrder={5}
+              material={depthWriteMaterial}
+            />
+          );
+        },
+      })}
+    </>
+  );
+}