fix: Improvement to argument usage tracking

Author: iwoplaza

apps/typegpu-docs/tests/individual-example-tests/3d-fish.test.ts

@@ -305,15 +305,6 @@ describe('3d fish example', () => {
         return vertexShader_Output(worldPosition.xyz, worldNormal, canvasPosition, (*currentModelData).variant, _arg_textureUV, (*currentModelData).applySeaFog, (*currentModelData).applySeaDesaturation);
       }
 
-      struct fragmentShader_Input {
-        @location(0) worldPosition: vec3f,
-        @location(1) worldNormal: vec3f,
-        @location(2) variant: f32,
-        @location(3) textureUV: vec2f,
-        @location(4) @interpolate(flat) applySeaFog: u32,
-        @location(5) @interpolate(flat) applySeaDesaturation: u32,
-      }
-
       @group(0) @binding(1) var modelTexture: texture_2d<f32>;
 
       @group(0) @binding(3) var sampler_1: sampler;
@@ -418,6 +409,15 @@ describe('3d fish example', () => {
         return vec3f(r, g, b);
       }
 
+      struct fragmentShader_Input {
+        @location(0) worldPosition: vec3f,
+        @location(1) worldNormal: vec3f,
+        @location(2) variant: f32,
+        @location(3) textureUV: vec2f,
+        @location(4) @interpolate(flat) applySeaFog: u32,
+        @location(5) @interpolate(flat) applySeaDesaturation: u32,
+      }
+
       @fragment fn fragmentShader(_arg_0: fragmentShader_Input) -> @location(0) vec4f {
         var textureColorWithAlpha = textureSample(modelTexture, sampler_1, _arg_0.textureUV);
         var textureColor = textureColorWithAlpha.rgb;

apps/typegpu-docs/tests/individual-example-tests/bitonic-sort.test.ts

@@ -118,12 +118,12 @@ describe('bitonic sort example', () => {
         return fullScreenTriangle_Output(vec4f(pos[vertexIndex], 0, 1), uv[vertexIndex]);
       }
 
+      @group(0) @binding(0) var<storage, read> data_1: array<u32>;
+
       struct fragmentFn_Input {
         @location(0) uv: vec2f,
       }
 
-      @group(0) @binding(0) var<storage, read> data_1: array<u32>;
-
       @fragment fn fragmentFn(_arg_0: fragmentFn_Input) -> @location(0) vec4f {
         let data = (&data_1);
         let arrayLength_1 = arrayLength(&(*data));

apps/typegpu-docs/tests/individual-example-tests/blur.test.ts

@@ -470,14 +470,14 @@ describe('blur example', () => {
         return fullScreenTriangle_Output(vec4f(pos[vertexIndex], 0, 1), uv[vertexIndex]);
       }
 
-      struct renderFragment_Input {
-        @location(0) uv: vec2f,
-      }
-
       @group(0) @binding(0) var renderView: texture_2d<f32>;
 
       @group(0) @binding(1) var sampler_1: sampler;
 
+      struct renderFragment_Input {
+        @location(0) uv: vec2f,
+      }
+
       @fragment fn renderFragment(_arg_0: renderFragment_Input) -> @location(0) vec4f {
         return textureSample(renderView, sampler_1, _arg_0.uv);
       }"

apps/typegpu-docs/tests/individual-example-tests/camera-thresholding.test.ts

@@ -32,10 +32,6 @@ describe('camera thresholding example', () => {
         return fullScreenTriangle_Output(vec4f(pos[vertexIndex], 0, 1), uv[vertexIndex]);
       }
 
-      struct mainFrag_Input {
-        @location(0) uv: vec2f,
-      }
-
       @group(0) @binding(0) var<uniform> uvTransformUniform: mat2x2f;
 
       @group(1) @binding(0) var inputTexture: texture_external;
@@ -48,6 +44,10 @@ describe('camera thresholding example', () => {
 
       @group(0) @binding(3) var<uniform> thresholdBuffer: f32;
 
+      struct mainFrag_Input {
+        @location(0) uv: vec2f,
+      }
+
       @fragment fn mainFrag(_arg_0: mainFrag_Input) -> @location(0) vec4f {
         var uv2 = ((uvTransformUniform * (_arg_0.uv - 0.5f)) + 0.5f);
         var col = textureSampleBaseClampToEdge(inputTexture, sampler_1, uv2);

apps/typegpu-docs/tests/individual-example-tests/caustics.test.ts

@@ -31,10 +31,6 @@ describe('caustics example', () => {
         return mainVertex_Output(vec4f(pos[vertexIndex], 0f, 1f), uv[vertexIndex]);
       }
 
-      struct mainFragment_Input {
-        @location(0) uv: vec2f,
-      }
-
       @group(0) @binding(0) var<uniform> tileDensity: f32;
 
       fn tilePattern(uv: vec2f) -> f32 {
@@ -120,6 +116,10 @@ describe('caustics example', () => {
         return mat2x2f(vec2f(cos(angle), sin(angle)), vec2f(-(sin(angle)), cos(angle)));
       }
 
+      struct mainFragment_Input {
+        @location(0) uv: vec2f,
+      }
+
       @fragment fn mainFragment(_arg_0: mainFragment_Input) -> @location(0) vec4f {
         var skewMat = mat2x2f(vec2f(0.9800665974617004, 0.19866932928562164), vec2f((-1.9866933079506122f + (_arg_0.uv.x * 3f)), 4.900332889206208f));
         var skewedUv = (skewMat * _arg_0.uv);

apps/typegpu-docs/tests/individual-example-tests/chroma-keying.test.ts

@@ -32,10 +32,6 @@ describe('chroma keying example', () => {
         return fullScreenTriangle_Output(vec4f(pos[vertexIndex], 0, 1), uv[vertexIndex]);
       }
 
-      struct fragment_Input {
-        @location(0) uv: vec2f,
-      }
-
       @group(0) @binding(0) var<uniform> uvTransform: mat2x2f;
 
       @group(1) @binding(0) var inputTexture: texture_external;
@@ -48,6 +44,10 @@ describe('chroma keying example', () => {
 
       @group(0) @binding(3) var<uniform> threshold: f32;
 
+      struct fragment_Input {
+        @location(0) uv: vec2f,
+      }
+
       @fragment fn fragment(_arg_0: fragment_Input) -> @location(0) vec4f {
         var uv2 = ((uvTransform * (_arg_0.uv - 0.5f)) + 0.5f);
         var col = textureSampleBaseClampToEdge(inputTexture, sampler_1, uv2);

apps/typegpu-docs/tests/individual-example-tests/confetti.test.ts

@@ -59,10 +59,10 @@ describe('confetti example', () => {
 
       struct VertexIn {
         @location(0) tilt: f32,
-        @location(1) angle: f32,
-        @location(2) color: vec4f,
-        @location(3) center: vec2f,
         @builtin(vertex_index) vertexIndex: u32,
+        @location(1) angle: f32,
+        @location(2) center: vec2f,
+        @location(3) color: vec4f,
       }
 
       @vertex fn vertex(_arg_0: VertexIn) -> VertexOut {

apps/typegpu-docs/tests/individual-example-tests/cubemap-reflection.test.ts

@@ -257,14 +257,14 @@ describe('cubemap reflection example', () => {
         return cubeVertexFn_Output((camera.projection * vec4f(viewPos, 1f)), _arg_position.xyz);
       }
 
-      struct cubeFragmentFn_Input {
-        @location(0) texCoord: vec3f,
-      }
-
       @group(1) @binding(0) var cubemap: texture_cube<f32>;
 
       @group(1) @binding(1) var texSampler: sampler;
 
+      struct cubeFragmentFn_Input {
+        @location(0) texCoord: vec3f,
+      }
+
       @fragment fn cubeFragmentFn(_arg_0: cubeFragmentFn_Input) -> @location(0) vec4f {
         return textureSample(cubemap, texSampler, normalize(_arg_0.texCoord));
       }
@@ -287,11 +287,6 @@ describe('cubemap reflection example', () => {
         return vertexFn_Output((camera.projection * (camera.view * _arg_position)), _arg_normal, _arg_position);
       }
 
-      struct fragmentFn_Input {
-        @location(0) normal: vec4f,
-        @location(1) worldPos: vec4f,
-      }
-
       struct DirectionalLight {
         direction: vec3f,
         color: vec3f,
@@ -314,6 +309,11 @@ describe('cubemap reflection example', () => {
 
       @group(1) @binding(1) var texSampler: sampler;
 
+      struct fragmentFn_Input {
+        @location(0) normal: vec4f,
+        @location(1) worldPos: vec4f,
+      }
+
       @fragment fn fragmentFn(_arg_0: fragmentFn_Input) -> @location(0) vec4f {
         var normalizedNormal = normalize(_arg_0.normal.xyz);
         var normalizedLightDir = normalize(light.direction);

apps/typegpu-docs/tests/individual-example-tests/disco.test.ts

@@ -32,10 +32,6 @@ describe('disco example', () => {
         return mainVertex_Output(vec4f(pos[vertexIndex], 0f, 1f), uv[vertexIndex]);
       }
 
-      struct mainFragment2_Input {
-        @location(0) uv: vec2f,
-      }
-
       @group(0) @binding(0) var<uniform> resolutionUniform: vec2f;
 
       fn aspectCorrected(uv: vec2f) -> vec2f {
@@ -65,6 +61,10 @@ describe('disco example', () => {
         return (acc + (col * weight));
       }
 
+      struct mainFragment2_Input {
+        @location(0) uv: vec2f,
+      }
+
       @fragment fn mainFragment2(_arg_0: mainFragment2_Input) -> @location(0) vec4f {
         var originalUv = aspectCorrected(_arg_0.uv);
         var aspectUv = originalUv;
@@ -243,10 +243,6 @@ describe('disco example', () => {
         return mainVertex_Output(vec4f(pos[vertexIndex], 0f, 1f), uv[vertexIndex]);
       }
 
-      struct mainFragment1_Input {
-        @location(0) uv: vec2f,
-      }
-
       @group(0) @binding(0) var<uniform> resolutionUniform: vec2f;
 
       fn aspectCorrected(uv: vec2f) -> vec2f {
@@ -276,6 +272,10 @@ describe('disco example', () => {
         return (acc + (col * weight));
       }
 
+      struct mainFragment1_Input {
+        @location(0) uv: vec2f,
+      }
+
       @fragment fn mainFragment1(_arg_0: mainFragment1_Input) -> @location(0) vec4f {
         var originalUv = aspectCorrected(_arg_0.uv);
         var aspectUv = originalUv;

apps/typegpu-docs/tests/individual-example-tests/fluid-double-buffering.test.ts

@@ -558,10 +558,6 @@ describe('fluid double buffering example', () => {
         return vertexMain_Output(vec4f(pos[_arg_idx].x, pos[_arg_idx].y, 0f, 1f), uv[_arg_idx]);
       }
 
-      struct fragmentMain_Input {
-        @location(0) uv: vec2f,
-      }
-
       fn coordsToIndex(x: i32, y: i32) -> i32 {
         return (x + (y * 256i));
       }
@@ -595,6 +591,10 @@ describe('fluid double buffering example', () => {
         return false;
       }
 
+      struct fragmentMain_Input {
+        @location(0) uv: vec2f,
+      }
+
       @fragment fn fragmentMain(_arg_0: fragmentMain_Input) -> @location(0) vec4f {
         let x = i32((_arg_0.uv.x * 256f));
         let y = i32((_arg_0.uv.y * 256f));