feat: all armor rules... I think?

src/app/tools/tankopedia/[id]/components/Model/components/TankArmor.tsx

@@ -94,15 +94,15 @@ export function TankArmor() {
               key={node.uuid}
               geometry={(node as Mesh).geometry}
               thickness={thickness}
-              penetration={100}
-              ricochet={antagonist.shell.ricochet * (Math.PI / 180)}
+              penetration={225}
+              ricochet={antagonist.shell.ricochet}
               caliber={antagonist.shell.caliber}
               normalization={antagonist.shell.normalization}
             />
           );
         })}
 
-        {/* <group position={turretPosition} rotation={turretRotation}>
+        <group position={turretPosition} rotation={turretRotation}>
           {nodes.map((node) => {
             const isCurrentTurret = node.name.startsWith(
               `turret_${turretModelDefinition.model
@@ -110,6 +110,9 @@ export function TankArmor() {
                 .padStart(2, '0')}`,
             );
             const isVisible = isCurrentTurret;
+            const armorId = parseInt(node.name.match(/.+_armor_(\d+)/)![1]);
+            const thickness =
+              turretModelDefinition.armor.thickness[armorId] ?? 0;
 
             if (!isVisible) return null;
             return (
@@ -117,6 +120,11 @@ export function TankArmor() {
                 key={node.uuid}
                 position={turretOrigin}
                 geometry={(node as Mesh).geometry}
+                thickness={thickness}
+                penetration={225}
+                ricochet={antagonist.shell.ricochet}
+                caliber={antagonist.shell.caliber}
+                normalization={antagonist.shell.normalization}
               />
             );
           })}
@@ -135,6 +143,9 @@ export function TankArmor() {
                 `gun_${gunModelDefinition.model.toString().padStart(2, '0')}`,
               );
               const isVisible = isCurrentGun;
+              const armorId = parseInt(node.name.match(/.+_armor_(\d+)/)![1]);
+              const thickness =
+                gunModelDefinition.armor.thickness[armorId] ?? 0;
 
               if (!isVisible) return null;
 
@@ -143,11 +154,16 @@ export function TankArmor() {
                   key={node.uuid}
                   position={turretOrigin.clone().add(gunOrigin)}
                   geometry={(node as Mesh).geometry}
+                  thickness={thickness}
+                  penetration={225}
+                  ricochet={antagonist.shell.ricochet}
+                  caliber={antagonist.shell.caliber}
+                  normalization={antagonist.shell.normalization}
                 />
               );
             })}
           </group>
-        </group> */}
+        </group>
       </group>
     </HeadsUpDisplay>
   );

src/components/ArmorMesh.tsx

@@ -1,130 +0,0 @@
-// // Vertex shader
-// varying vec3 vNormal; // declare a varying variable to pass the normal vector
-// void main() {
-//   vNormal = normal; // assign the normal vector to the varying variable
-//   gl_Position = projectionMatrix * modelViewMatrix * vec4(position, 1.0); // transform the vertex position
-// }
-
-// // Fragment shader
-// varying vec3 vNormal; // declare the same varying variable to receive the normal vector
-// void main() {
-//   vec3 cameraDir = -viewPosition; // get the camera direction vector by negating the view position
-//   float angle = acos(dot(vNormal, cameraDir)); // calculate the angle of the fragment using the dot product and the acos function
-//   vec3 color = mix(vec3(1.0, 0.0, 0.0), vec3(0.0, 0.0, 1.0), angle / PI); // create a gradient color from red to blue based on the angle
-//   gl_FragColor = vec4(color, 1.0); // set the fragment color
-// }
-
-import { MeshProps } from '@react-three/fiber';
-import { MeshBasicMaterial } from 'three';
-import CustomShaderMaterial from 'three-custom-shader-material';
-
-interface ArmorMeshProps extends MeshProps {
-  thickness: number;
-  penetration: number;
-  ricochet: number;
-  caliber: number;
-  normalization: number;
-}
-
-export function ArmorMesh({
-  thickness,
-  penetration,
-  caliber,
-  ricochet,
-  normalization,
-  ...props
-}: ArmorMeshProps) {
-  return (
-    <>
-      <mesh {...props}>
-        <meshBasicMaterial colorWrite={false} />
-      </mesh>
-
-      <mesh {...props} renderOrder={1}>
-        <CustomShaderMaterial
-          baseMaterial={MeshBasicMaterial}
-          transparent
-          silent
-          depthWrite={false}
-          uniforms={{
-            thickness: { value: thickness },
-            penetration: { value: penetration },
-            ricochet: { value: ricochet },
-            caliber: { value: caliber },
-            normalization: { value: normalization },
-          }}
-          vertexShader={`
-            varying vec3 vNormal;
-            varying vec3 vViewPosition;
-
-            void main() {
-              vec4 vViewPosition4 = modelViewMatrix * vec4(position, 1.0);
-              vViewPosition = vViewPosition4.xyz;
-              vNormal = normalMatrix * normal;
-            }
-          `}
-          fragmentShader={`
-            varying vec3 vViewPosition;
-            uniform float thickness;
-            uniform float penetration;
-            uniform float ricochet;
-            uniform float caliber;
-            uniform float normalization;
-
-            void main() {
-              vec3 normalizedNormal = normalize(vNormal);
-              vec3 normalizedViewPosition = normalize(vViewPosition);
-              float angle = acos(dot(normalizedNormal, -normalizedViewPosition));
-              float sinAngle = sin(angle);
-
-              float penetrationChance = 0.0;
-              float effectiveArmor = thickness * sinAngle;
-              bool twoCaliberRule = caliber > 2.0 * thickness;
-              bool threeCaliberRule = caliber > 3.0 * thickness;
-              float finalNormalization = normalization;
-
-              if (twoCaliberRule) {
-                finalNormalization = (normalization * 1.4 * caliber) / (2.0 * thickness);
-              }
-
-              if (
-                angle >= ricochet
-                && !threeCaliberRule
-              ) {
-                penetrationChance = 1.0;
-              }
-
-              csm_FragColor *= vec4(1.0, 0.0, 0.0, penetrationChance * 0.5);
-            }
-          `}
-        />
-
-        {/* <meshBasicMaterial
-          transparent
-          opacity={0.5}
-          color="red"
-          depthWrite={false}
-        /> */}
-        {/* <shaderMaterial
-          transparent
-          depthWrite={false}
-          args={[
-            {
-              vertexShader: `
-                void main() {
-                  gl_Position = projectionMatrix * modelViewMatrix * vec4(position, 1.0);
-                }
-              `,
-
-              fragmentShader: `
-                void main() {
-                  gl_FragColor = vec4(1.0, 0.0, 0.0, 0.5);
-                }
-              `,
-            },
-          ]}
-        /> */}
-      </mesh>
-    </>
-  );
-}

src/components/ArmorMesh/index.tsx

@@ -0,0 +1,48 @@
+import { MeshProps } from '@react-three/fiber';
+import { MeshBasicMaterial } from 'three';
+import CustomShaderMaterial from 'three-custom-shader-material';
+import fragment from './shaders/fragment.glsl';
+import vertex from './shaders/vertex.glsl';
+
+interface ArmorMeshProps extends MeshProps {
+  thickness: number;
+  penetration: number;
+  ricochet: number;
+  caliber: number;
+  normalization: number;
+}
+
+export function ArmorMesh({
+  thickness,
+  penetration,
+  caliber,
+  ricochet,
+  normalization,
+  ...props
+}: ArmorMeshProps) {
+  return (
+    <>
+      <mesh {...props}>
+        <meshBasicMaterial colorWrite={false} />
+      </mesh>
+
+      <mesh {...props} renderOrder={1}>
+        <CustomShaderMaterial
+          baseMaterial={MeshBasicMaterial}
+          transparent
+          silent
+          depthWrite={false}
+          uniforms={{
+            thickness: { value: thickness },
+            penetration: { value: penetration },
+            ricochet: { value: ricochet * (Math.PI / 180) },
+            caliber: { value: caliber },
+            normalization: { value: normalization * (Math.PI / 180) },
+          }}
+          vertexShader={vertex}
+          fragmentShader={fragment}
+        />
+      </mesh>
+    </>
+  );
+}

src/components/ArmorMesh/shaders/fragment.glsl

@@ -0,0 +1,29 @@
+varying vec3 vViewPosition;
+uniform float thickness;
+uniform float penetration;
+uniform float ricochet;
+uniform float caliber;
+uniform float normalization;
+
+void main() {
+  vec3 normalizedNormal = normalize(vNormal);
+  vec3 normalizedViewPosition = normalize(vViewPosition);
+  float cosAngle = dot(normalizedNormal, -normalizedViewPosition);
+  float angle = acos(cosAngle);
+
+  float penetrationChance = 0.0;
+  bool twoCaliberRule = caliber > 2.0 * thickness;
+  bool threeCaliberRule = caliber > 3.0 * thickness;
+  float finalNormalization = twoCaliberRule ? (normalization * 1.4 * caliber) / (2.0 * thickness) : normalization;
+
+  float effectiveArmor = thickness / cos(angle - finalNormalization);
+
+  if(angle < ricochet) {
+    // no auto-ricochet, resume checks
+    if(effectiveArmor <= penetration || threeCaliberRule) {
+      penetrationChance = 1.0;
+    }
+  }
+
+  csm_FragColor *= vec4(1.0, 0.0, 0.0, (1.0 - penetrationChance) * 0.5);
+}

src/components/ArmorMesh/shaders/vertex.glsl

@@ -0,0 +1,8 @@
+varying vec3 vNormal;
+varying vec3 vViewPosition;
+
+void main() {
+  vec4 vViewPosition4 = modelViewMatrix * vec4(position, 1.0);
+  vViewPosition = vViewPosition4.xyz;
+  vNormal = normalMatrix * normal;
+}

src/types/glsl.d.ts

@@ -0,0 +1,4 @@
+declare module '*.glsl' {
+  const value: string;
+  export default value;
+}