feat: Sampler schemas (#1819)

Author: reczkok

apps/typegpu-docs/src/examples/image-processing/ascii-filter/index.ts

@@ -15,7 +15,7 @@ const glyphSize = root.createUniform(d.u32, 8);
 const uvTransformBuffer = root
   .createUniform(d.mat2x2f, d.mat2x2f.identity());
 
-const shaderSampler = tgpu['~unstable'].sampler({
+const shaderSampler = root['~unstable'].createSampler({
   magFilter: 'linear',
   minFilter: 'linear',
 });
@@ -78,7 +78,7 @@ const fragmentFn = tgpu['~unstable'].fragmentFn({
 
   const color = std.textureSampleBaseClampToEdge(
     layout.$.externalTexture,
-    shaderSampler,
+    shaderSampler.$,
     blockCoord,
   );
 

apps/typegpu-docs/src/examples/image-processing/blur/index.ts

@@ -54,7 +54,7 @@ const textures = [0, 1].map(() => {
 });
 const renderView = textures[1].createView(d.texture2d(d.f32));
 
-const sampler = tgpu['~unstable'].sampler({
+const sampler = root['~unstable'].createSampler({
   magFilter: 'linear',
   minFilter: 'linear',
 });
@@ -93,7 +93,7 @@ const computeFn = tgpu['~unstable'].computeFn({
 
       tileData.$[r][lid.x * 4 + d.u32(c)] = std.textureSampleLevel(
         ioLayout.$.inTexture,
-        sampler,
+        sampler.$,
         d.vec2f(d.vec2f(loadIndex).add(d.vec2f(0.5)).div(d.vec2f(dims))),
         0,
       ).xyz;
@@ -146,7 +146,7 @@ const renderFragment = tgpu['~unstable'].fragmentFn({
 })((input) =>
   std.textureSample(
     renderView.$,
-    sampler,
+    sampler.$,
     input.uv,
   )
 );

apps/typegpu-docs/src/examples/rendering/cubemap-reflection/index.ts

@@ -109,7 +109,7 @@ const texture = root['~unstable']
 await loadCubemap(texture, chosenCubemap);
 
 const cubemap = texture.createView(d.textureCube(d.f32));
-const sampler = tgpu['~unstable'].sampler({
+const sampler = root['~unstable'].createSampler({
   magFilter: 'linear',
   minFilter: 'linear',
 });
@@ -218,7 +218,7 @@ const fragmentFn = tgpu['~unstable'].fragmentFn({
   );
   const environmentColor = std.textureSample(
     cubemapBinding.$,
-    texSampler,
+    texSampler.$,
     reflectionVector,
   );
 
@@ -266,7 +266,7 @@ const cubeFragmentFn = tgpu['~unstable'].fragmentFn({
 })((input) => {
   return std.textureSample(
     cubemapBinding.$,
-    texSampler,
+    texSampler.$,
     std.normalize(input.texCoord),
   );
 });

apps/typegpu-docs/src/examples/rendering/simple-shadow/index.ts

@@ -65,7 +65,7 @@ function createShadowTextures(
     format: 'depth32float',
   }).$usage('render', 'sampled');
 
-  const comparisonSampler = tgpu['~unstable'].comparisonSampler({
+  const comparisonSampler = root['~unstable'].createComparisonSampler({
     compare: sampleCompare,
     magFilter: pcf ? 'linear' : 'nearest',
     minFilter: pcf ? 'linear' : 'nearest',

apps/typegpu-docs/src/examples/simple/liquid-glass/index.ts

@@ -24,7 +24,7 @@ imageTexture.write(imageBitmap);
 imageTexture.generateMipmaps();
 
 const sampledView = imageTexture.createView();
-const sampler = tgpu['~unstable'].sampler({
+const sampler = root['~unstable'].createSampler({
   magFilter: 'linear',
   minFilter: 'linear',
   mipmapFilter: 'linear',
@@ -97,6 +97,7 @@ const applyTint = (color: d.v3f, tint: d.Infer<typeof TintParams>) => {
 
 const sampleWithChromaticAberration = (
   tex: d.texture2d<d.F32>,
+  sampler: d.sampler,
   uv: d.v2f,
   offset: number,
   dir: d.v2f,
@@ -137,18 +138,19 @@ const fragmentShader = tgpu['~unstable'].fragmentFn({
 
   const blurSample = std.textureSampleBias(
     sampledView.$,
-    sampler,
+    sampler.$,
     uv,
     paramsUniform.$.blur,
   );
   const refractedSample = sampleWithChromaticAberration(
     sampledView.$,
+    sampler.$,
     uv.add(dir.mul(paramsUniform.$.refractionStrength * normalizedDist)),
     paramsUniform.$.chromaticStrength * normalizedDist,
     dir,
     paramsUniform.$.blur * paramsUniform.$.edgeBlurMultiplier,
   );
-  const normalSample = std.textureSampleLevel(sampledView.$, sampler, uv, 0);
+  const normalSample = std.textureSampleLevel(sampledView.$, sampler.$, uv, 0);
 
   const tint = TintParams({
     color: paramsUniform.$.tintColor,

apps/typegpu-docs/src/examples/simulation/gravity/index.ts

@@ -53,7 +53,7 @@ context.configure({
 
 // static resources (created on the example load)
 
-const sampler = tgpu['~unstable'].sampler({
+const sampler = root['~unstable'].createSampler({
   magFilter: 'linear',
   minFilter: 'linear',
 });

apps/typegpu-docs/src/examples/simulation/slime-mold-3d/index.ts

@@ -359,7 +359,7 @@ const fullScreenTriangle = tgpu['~unstable'].vertexFn({
   };
 });
 
-const sampler = tgpu['~unstable'].sampler({
+const sampler = root['~unstable'].createSampler({
   magFilter: canFilter ? 'linear' : 'nearest',
   minFilter: canFilter ? 'linear' : 'nearest',
 });
@@ -433,7 +433,7 @@ const fragmentShader = tgpu['~unstable'].fragmentFn({
     const texCoord = pos.div(resolution);
 
     const sampleValue = std
-      .textureSampleLevel(renderLayout.$.state, sampler, texCoord, 0)
+      .textureSampleLevel(renderLayout.$.state, sampler.$, texCoord, 0)
       .x;
 
     const d0 = std.smoothstep(thresholdLo, thresholdHi, sampleValue);

apps/typegpu-docs/src/examples/simulation/stable-fluid/index.ts

@@ -125,7 +125,7 @@ const newInkTex = createField('addedInk');
 const forceTex = createField('force');
 const divergenceTex = createField('divergence');
 
-const linSampler = tgpu['~unstable'].sampler({
+const linSampler = root['~unstable'].createSampler({
   magFilter: 'linear',
   minFilter: 'linear',
 });

apps/typegpu-docs/src/examples/tests/texture-test/index.ts

@@ -35,7 +35,7 @@ let texture = root['~unstable'].createTexture({
 texture.copyFrom(sourceTex);
 texture.generateMipmaps();
 
-const filteringSampler = tgpu['~unstable'].sampler({
+const filteringSampler = root['~unstable'].createSampler({
   magFilter: 'linear',
   minFilter: 'linear',
 });

packages/typegpu/src/core/resolve/resolveData.ts

@@ -5,6 +5,7 @@ import {
   isLooseData,
   type Unstruct,
 } from '../../data/dataTypes.ts';
+import { isWgslComparisonSampler, isWgslSampler } from '../../data/sampler.ts';
 import {
   accessModeMap,
   isWgslStorageTexture,
@@ -302,5 +303,9 @@ export function resolveData(ctx: ResolutionCtx, data: AnyData): string {
       : `${data.type}<${data.sampleType.type}>`;
   }
 
+  if (isWgslComparisonSampler(data) || isWgslSampler(data)) {
+    return data.type;
+  }
+
   assertExhaustive(data, 'resolveData');
 }