typegpu 0.10.2 → 0.11.0

package/std/array.d.ts

@@ -2,6 +2,6 @@ import { DualFn } from "../types.js";
 import { _ref } from "../data/ref.js";
 
 //#region src/std/array.d.ts
-declare const arrayLength: DualFn<(a: unknown[] | _ref<unknown[]>) => number>;
+declare const arrayLength: DualFn;
 //#endregion
 export { arrayLength };

package/std/array.js

@@ -4,7 +4,6 @@ import { ptrFn } from "../data/ptr.js";
 import { isRef } from "../data/ref.js";
 import { abstractInt, u32 } from "../data/numeric.js";
 import { dualImpl } from "../core/function/dualImpl.js";
-
 //#region src/std/array.ts
 const sizeOfPointedToArray = (dataType) => isPtr(dataType) && isWgslArray(dataType.inner) ? dataType.inner.elementCount : 0;
 const arrayLength = dualImpl({
@@ -22,6 +21,5 @@ const arrayLength = dualImpl({
 		return length > 0 ? `${length}` : stitch`arrayLength(${a})`;
 	}
 });
-
 //#endregion
-export { arrayLength };
+export { arrayLength };

package/std/atomic.d.ts

@@ -3,17 +3,17 @@ import { atomicI32, atomicU32 } from "../data/wgslTypes.js";
 
 //#region src/std/atomic.d.ts
 type AnyAtomic = atomicI32 | atomicU32;
-declare const workgroupBarrier: DualFn<(...args: never[]) => unknown>;
-declare const storageBarrier: DualFn<(...args: never[]) => unknown>;
-declare const textureBarrier: DualFn<(...args: never[]) => unknown>;
-declare const atomicLoad: DualFn<(<T extends AnyAtomic>(a: T) => number)>;
-declare const atomicStore: DualFn<(<T extends AnyAtomic>(a: T, value: number) => void)>;
-declare const atomicAdd: DualFn<(<T extends AnyAtomic>(a: T, value: number) => number)>;
-declare const atomicSub: DualFn<(<T extends AnyAtomic>(a: T, value: number) => number)>;
-declare const atomicMax: DualFn<(<T extends AnyAtomic>(a: T, value: number) => number)>;
-declare const atomicMin: DualFn<(<T extends AnyAtomic>(a: T, value: number) => number)>;
-declare const atomicAnd: DualFn<(<T extends AnyAtomic>(a: T, value: number) => number)>;
-declare const atomicOr: DualFn<(<T extends AnyAtomic>(a: T, value: number) => number)>;
-declare const atomicXor: DualFn<(<T extends AnyAtomic>(a: T, value: number) => number)>;
+declare const workgroupBarrier: DualFn;
+declare const storageBarrier: DualFn;
+declare const textureBarrier: DualFn;
+declare const atomicLoad: DualFn;
+declare const atomicStore: DualFn;
+declare const atomicAdd: DualFn;
+declare const atomicSub: DualFn;
+declare const atomicMax: DualFn;
+declare const atomicMin: DualFn;
+declare const atomicAnd: DualFn;
+declare const atomicOr: DualFn;
+declare const atomicXor: DualFn;
 //#endregion
 export { atomicAdd, atomicAnd, atomicLoad, atomicMax, atomicMin, atomicOr, atomicStore, atomicSub, atomicXor, storageBarrier, textureBarrier, workgroupBarrier };

package/std/atomic.js

@@ -3,7 +3,6 @@ import { stitch } from "../core/resolve/stitch.js";
 import { safeStringify } from "../shared/stringify.js";
 import { i32, u32 } from "../data/numeric.js";
 import { dualImpl } from "../core/function/dualImpl.js";
-
 //#region src/std/atomic.ts
 const workgroupBarrier = dualImpl({
 	name: "workgroupBarrier",
@@ -108,6 +107,5 @@ const atomicXor = dualImpl({
 	signature: atomicOpSignature,
 	codegenImpl: (_ctx, [a, value]) => stitch`atomicXor(&${a}, ${value})`
 });
-
 //#endregion
-export { atomicAdd, atomicAnd, atomicLoad, atomicMax, atomicMin, atomicOr, atomicStore, atomicSub, atomicXor, storageBarrier, textureBarrier, workgroupBarrier };
+export { atomicAdd, atomicAnd, atomicLoad, atomicMax, atomicMin, atomicOr, atomicStore, atomicSub, atomicXor, storageBarrier, textureBarrier, workgroupBarrier };

package/std/bitcast.d.ts

@@ -3,8 +3,8 @@ import { v2f, v2i, v2u, v3f, v3i, v3u, v4f, v4i, v4u } from "../data/wgslTypes.j
 
 //#region src/std/bitcast.d.ts
 type BitcastU32toF32Overload = ((value: number) => number) & ((value: v2u) => v2f) & ((value: v3u) => v3f) & ((value: v4u) => v4f);
-declare const bitcastU32toF32: DualFn<BitcastU32toF32Overload>;
+declare const bitcastU32toF32: DualFn;
 type BitcastU32toI32Overload = ((value: number) => number) & ((value: v2u) => v2i) & ((value: v3u) => v3i) & ((value: v4u) => v4i);
-declare const bitcastU32toI32: DualFn<BitcastU32toI32Overload>;
+declare const bitcastU32toI32: DualFn;
 //#endregion
 export { bitcastU32toF32, bitcastU32toI32 };

package/std/bitcast.js

@@ -6,7 +6,6 @@ import { vec2f, vec2i, vec3f, vec3i, vec4f, vec4i } from "../data/vector.js";
 import { dualImpl } from "../core/function/dualImpl.js";
 import { bitcastU32toF32Impl, bitcastU32toI32Impl } from "../data/numberOps.js";
 import { VectorOps } from "../data/vectorOps.js";
-
 //#region src/std/bitcast.ts
 const bitcastU32toF32 = dualImpl({
 	name: "bitcastU32toF32",
@@ -38,6 +37,5 @@ const bitcastU32toI32 = dualImpl({
 		};
 	}
 });
-
 //#endregion
-export { bitcastU32toF32, bitcastU32toI32 };
+export { bitcastU32toF32, bitcastU32toI32 };

package/std/boolean.d.ts

@@ -9,7 +9,7 @@ import { AnyBooleanVecInstance, AnyFloatVecInstance, AnyNumericVecInstance, AnyV
  * allEq(vec2f(0.0, 1.0), vec2f(0.0, 2.0)) // returns false
  * allEq(vec3u(0, 1, 2), vec3u(0, 1, 2)) // returns true
  */
-declare const allEq: DualFn<(<T extends AnyVecInstance>(lhs: T, rhs: T) => boolean)>;
+declare const allEq: DualFn;
 /**
  * Checks **component-wise** whether `lhs == rhs`.
  * This function does **not** return `bool`, for that use-case, wrap the result in `all`, or use `allEq`.
@@ -19,7 +19,7 @@ declare const allEq: DualFn<(<T extends AnyVecInstance>(lhs: T, rhs: T) => boole
  * all(eq(vec4i(4, 3, 2, 1), vec4i(4, 3, 2, 1))) // returns true
  * allEq(vec4i(4, 3, 2, 1), vec4i(4, 3, 2, 1)) // returns true
  */
-declare const eq: DualFn<(<T extends AnyVecInstance>(lhs: T, rhs: T) => T extends AnyVec2Instance ? v2b : T extends AnyVec3Instance ? v3b : v4b)>;
+declare const eq: DualFn;
 /**
  * Checks **component-wise** whether `lhs != rhs`.
  * This function does **not** return `bool`, for that use-case, wrap the result in `any`.
@@ -28,7 +28,7 @@ declare const eq: DualFn<(<T extends AnyVecInstance>(lhs: T, rhs: T) => T extend
  * ne(vec3u(0, 1, 2), vec3u(2, 1, 0)) // returns vec3b(true, false, true)
  * any(ne(vec4i(4, 3, 2, 1), vec4i(4, 2, 2, 1))) // returns true
  */
-declare const ne: DualFn<(<T extends AnyVecInstance>(lhs: T, rhs: T) => T extends AnyVec2Instance ? v2b : T extends AnyVec3Instance ? v3b : v4b)>;
+declare const ne: DualFn;
 /**
  * Checks **component-wise** whether `lhs < rhs`.
  * This function does **not** return `bool`, for that use-case, wrap the result in `all`.
@@ -37,7 +37,7 @@ declare const ne: DualFn<(<T extends AnyVecInstance>(lhs: T, rhs: T) => T extend
  * lt(vec3u(0, 1, 2), vec3u(2, 1, 0)) // returns vec3b(true, false, false)
  * all(lt(vec4i(1, 2, 3, 4), vec4i(2, 3, 4, 5))) // returns true
  */
-declare const lt: DualFn<(<T extends AnyNumericVecInstance>(lhs: T, rhs: T) => T extends AnyVec2Instance ? v2b : T extends AnyVec3Instance ? v3b : v4b)>;
+declare const lt: DualFn;
 /**
  * Checks **component-wise** whether `lhs <= rhs`.
  * This function does **not** return `bool`, for that use-case, wrap the result in `all`.
@@ -46,7 +46,7 @@ declare const lt: DualFn<(<T extends AnyNumericVecInstance>(lhs: T, rhs: T) => T
  * le(vec3u(0, 1, 2), vec3u(2, 1, 0)) // returns vec3b(true, true, false)
  * all(le(vec4i(1, 2, 3, 4), vec4i(2, 3, 3, 5))) // returns true
  */
-declare const le: DualFn<(<T extends AnyNumericVecInstance>(lhs: T, rhs: T) => T extends AnyVec2Instance ? v2b : T extends AnyVec3Instance ? v3b : v4b)>;
+declare const le: DualFn;
 /**
  * Checks **component-wise** whether `lhs > rhs`.
  * This function does **not** return `bool`, for that use-case, wrap the result in `all`.
@@ -55,7 +55,7 @@ declare const le: DualFn<(<T extends AnyNumericVecInstance>(lhs: T, rhs: T) => T
  * gt(vec3u(0, 1, 2), vec3u(2, 1, 0)) // returns vec3b(false, false, true)
  * all(gt(vec4i(2, 3, 4, 5), vec4i(1, 2, 3, 4))) // returns true
  */
-declare const gt: DualFn<(<T extends AnyNumericVecInstance>(lhs: T, rhs: T) => T extends AnyVec2Instance ? v2b : T extends AnyVec3Instance ? v3b : v4b)>;
+declare const gt: DualFn;
 /**
  * Checks **component-wise** whether `lhs >= rhs`.
  * This function does **not** return `bool`, for that use-case, wrap the result in `all`.
@@ -64,42 +64,56 @@ declare const gt: DualFn<(<T extends AnyNumericVecInstance>(lhs: T, rhs: T) => T
  * ge(vec3u(0, 1, 2), vec3u(2, 1, 0)) // returns vec3b(false, true, true)
  * all(ge(vec4i(2, 2, 4, 5), vec4i(1, 2, 3, 4))) // returns true
  */
-declare const ge: DualFn<(<T extends AnyNumericVecInstance>(lhs: T, rhs: T) => T extends AnyVec2Instance ? v2b : T extends AnyVec3Instance ? v3b : v4b)>;
+declare const ge: DualFn;
+type VecInstanceToBooleanVecInstance<T extends AnyVecInstance> = T extends AnyVec2Instance ? v2b : T extends AnyVec3Instance ? v3b : v4b;
+declare function cpuNot(value: boolean): boolean;
+declare function cpuNot(value: number): boolean;
+declare function cpuNot<T extends AnyVecInstance>(value: T): VecInstanceToBooleanVecInstance<T>;
+declare function cpuNot(value: unknown): boolean;
 /**
- * Returns **component-wise** `!value`.
+ * Returns the logical negation of the given value.
+ * For scalars (bool, number), returns `!value`.
+ * For boolean vectors, returns **component-wise** `!value`.
+ * For numeric vectors, returns a boolean vector with component-wise truthiness negation.
+ * For all other types, returns the truthiness negation (in WGSL, this applies only if the value is known at compile-time).
  * @example
- * not(vec2b(false, true)) // returns vec2b(true, false)
+ * not(true) // returns false
+ * not(-1) // returns false
+ * not(0) // returns true
  * not(vec3b(true, true, false)) // returns vec3b(false, false, true)
+ * not(vec3f(1.0, 0.0, -1.0)) // returns vec3b(false, true, false)
+ * not({a: 1882}) // returns false
+ * not(NaN) // returns false **as in WGSL**
  */
-declare const not: DualFn<(<T extends AnyBooleanVecInstance>(value: T) => T)>;
+declare const not: DualFn;
 /**
  * Returns **component-wise** logical `or` result.
  * @example
  * or(vec2b(false, true), vec2b(false, false)) // returns vec2b(false, true)
  * or(vec3b(true, true, false), vec3b(false, true, false)) // returns vec3b(true, true, false)
  */
-declare const or: DualFn<(<T extends AnyBooleanVecInstance>(lhs: T, rhs: T) => T)>;
+declare const or: DualFn;
 /**
  * Returns **component-wise** logical `and` result.
  * @example
  * and(vec2b(false, true), vec2b(true, true)) // returns vec2b(false, true)
  * and(vec3b(true, true, false), vec3b(false, true, false)) // returns vec3b(false, true, false)
  */
-declare const and: DualFn<(<T extends AnyBooleanVecInstance>(lhs: T, rhs: T) => T)>;
+declare const and: DualFn;
 /**
  * Returns `true` if each component of `value` is true.
  * @example
  * all(vec2b(false, true)) // returns false
  * all(vec3b(true, true, true)) // returns true
  */
-declare const all: DualFn<(value: AnyBooleanVecInstance) => boolean>;
+declare const all: DualFn;
 /**
  * Returns `true` if any component of `value` is true.
  * @example
  * any(vec2b(false, true)) // returns true
  * any(vec3b(false, false, false)) // returns false
  */
-declare const any: DualFn<(value: AnyBooleanVecInstance) => boolean>;
+declare const any: DualFn;
 /**
  * Checks whether the given elements differ by at most the `precision` value.
  * Checks all elements of `lhs` and `rhs` if arguments are vectors.
@@ -109,7 +123,7 @@ declare const any: DualFn<(value: AnyBooleanVecInstance) => boolean>;
  *
  * @param {number} precision argument that specifies the maximum allowed difference, 0.01 by default.
  */
-declare const isCloseTo: DualFn<(<T extends AnyFloatVecInstance | number>(lhs: T, rhs: T, precision?: number) => boolean)>;
+declare const isCloseTo: DualFn;
 declare function cpuSelect(f: boolean, t: boolean, cond: boolean): boolean;
 declare function cpuSelect(f: number, t: number, cond: boolean): number;
 declare function cpuSelect<T extends AnyVecInstance>(f: T, t: T, cond: boolean | (T extends AnyVec2Instance ? v2b : T extends AnyVec3Instance ? v3b : v4b)): T;
@@ -122,6 +136,6 @@ declare function cpuSelect<T extends AnyVecInstance>(f: T, t: T, cond: boolean |
  * select(vec2i(1, 2), vec2i(3, 4), true) // returns vec2i(3, 4)
  * select(vec2i(1, 2), vec2i(3, 4), vec2b(false, true)) // returns vec2i(1, 4)
  */
-declare const select: DualFn<typeof cpuSelect>;
+declare const select: DualFn;
 //#endregion
 export { all, allEq, and, any, eq, ge, gt, isCloseTo, le, lt, ne, not, or, select };

package/std/boolean.js

@@ -1,4 +1,4 @@
-import { isVecInstance } from "../data/wgslTypes.js";
+import { isBool, isNumericSchema, isVec, isVecBool, isVecInstance } from "../data/wgslTypes.js";
 import { isSnippetNumeric, snip } from "../data/snippet.js";
 import { stitch } from "../core/resolve/stitch.js";
 import { unify } from "../tgsl/conversion.js";
@@ -7,7 +7,6 @@ import { vec2b, vec3b, vec4b } from "../data/vector.js";
 import { dualImpl } from "../core/function/dualImpl.js";
 import { VectorOps } from "../data/vectorOps.js";
 import { sub } from "./operators.js";
-
 //#region src/std/boolean.ts
 function correspondingBooleanVectorSchema(dataType) {
 	if (dataType.type.includes("2")) return vec2b;
@@ -135,21 +134,48 @@ const ge = dualImpl({
 	normalImpl: (lhs, rhs) => cpuNot(cpuLt(lhs, rhs)),
 	codegenImpl: (_ctx, [lhs, rhs]) => stitch`(${lhs} >= ${rhs})`
 });
-const cpuNot = (value) => VectorOps.neg[value.kind](value);
+function cpuNot(value) {
+	if (typeof value === "number" && isNaN(value)) return false;
+	if (isVecInstance(value)) {
+		if (value.length === 2) return vec2b(cpuNot(value.x), cpuNot(value.y));
+		if (value.length === 3) return vec3b(cpuNot(value.x), cpuNot(value.y), cpuNot(value.z));
+		if (value.length === 4) return vec4b(cpuNot(value.x), cpuNot(value.y), cpuNot(value.z), cpuNot(value.w));
+	}
+	return !value;
+}
 /**
-* Returns **component-wise** `!value`.
+* Returns the logical negation of the given value.
+* For scalars (bool, number), returns `!value`.
+* For boolean vectors, returns **component-wise** `!value`.
+* For numeric vectors, returns a boolean vector with component-wise truthiness negation.
+* For all other types, returns the truthiness negation (in WGSL, this applies only if the value is known at compile-time).
 * @example
-* not(vec2b(false, true)) // returns vec2b(true, false)
+* not(true) // returns false
+* not(-1) // returns false
+* not(0) // returns true
 * not(vec3b(true, true, false)) // returns vec3b(false, false, true)
+* not(vec3f(1.0, 0.0, -1.0)) // returns vec3b(false, true, false)
+* not({a: 1882}) // returns false
+* not(NaN) // returns false **as in WGSL**
 */
 const not = dualImpl({
 	name: "not",
-	signature: (...argTypes) => ({
-		argTypes,
-		returnType: argTypes[0]
-	}),
+	signature: (arg) => {
+		const returnType = isVec(arg) ? correspondingBooleanVectorSchema(arg) : bool;
+		return {
+			argTypes: [arg],
+			returnType
+		};
+	},
 	normalImpl: cpuNot,
-	codegenImpl: (_ctx, [arg]) => stitch`!(${arg})`
+	codegenImpl: (_ctx, [arg]) => {
+		const { dataType } = arg;
+		if (isBool(dataType)) return stitch`!${arg}`;
+		if (isNumericSchema(dataType)) return stitch`!bool(${arg})`;
+		if (isVecBool(dataType)) return stitch`!(${arg})`;
+		if (isVec(dataType)) return stitch`!(${`vec${dataType.componentCount}<bool>`}(${arg}))`;
+		return "false";
+	}
 });
 const cpuOr = (lhs, rhs) => VectorOps.or[lhs.kind](lhs, rhs);
 /**
@@ -269,6 +295,5 @@ const select = dualImpl({
 	normalImpl: cpuSelect,
 	codegenImpl: (_ctx, [f, t, cond]) => stitch`select(${f}, ${t}, ${cond})`
 });
-
 //#endregion
-export { all, allEq, and, any, eq, ge, gt, isCloseTo, le, lt, ne, not, or, select };
+export { all, allEq, and, any, eq, ge, gt, isCloseTo, le, lt, ne, not, or, select };

package/std/derivative.d.ts

@@ -3,14 +3,14 @@ import { AnyFloat32VecInstance } from "../data/wgslTypes.js";
 
 //#region src/std/derivative.d.ts
 type DerivativeSignature = ((value: number) => number) & (<T extends AnyFloat32VecInstance>(value: T) => T);
-declare const dpdx: DualFn<DerivativeSignature>;
-declare const dpdxCoarse: DualFn<DerivativeSignature>;
-declare const dpdxFine: DualFn<DerivativeSignature>;
-declare const dpdy: DualFn<DerivativeSignature>;
-declare const dpdyCoarse: DualFn<DerivativeSignature>;
-declare const dpdyFine: DualFn<DerivativeSignature>;
-declare const fwidth: DualFn<DerivativeSignature>;
-declare const fwidthCoarse: DualFn<DerivativeSignature>;
-declare const fwidthFine: DualFn<DerivativeSignature>;
+declare const dpdx: DualFn;
+declare const dpdxCoarse: DualFn;
+declare const dpdxFine: DualFn;
+declare const dpdy: DualFn;
+declare const dpdyCoarse: DualFn;
+declare const dpdyFine: DualFn;
+declare const fwidth: DualFn;
+declare const fwidthCoarse: DualFn;
+declare const fwidthFine: DualFn;
 //#endregion
 export { dpdx, dpdxCoarse, dpdxFine, dpdy, dpdyCoarse, dpdyFine, fwidth, fwidthCoarse, fwidthFine };

package/std/derivative.js

@@ -1,6 +1,5 @@
 import { stitch } from "../core/resolve/stitch.js";
 import { dualImpl } from "../core/function/dualImpl.js";
-
 //#region src/std/derivative.ts
 const derivativeNormalError = "Derivative builtins are not allowed on the CPU";
 const dpdx = dualImpl({
@@ -84,6 +83,5 @@ const fwidthFine = dualImpl({
 	}),
 	codegenImpl: (_ctx, [value]) => stitch`fwidthFine(${value})`
 });
-
 //#endregion
-export { dpdx, dpdxCoarse, dpdxFine, dpdy, dpdyCoarse, dpdyFine, fwidth, fwidthCoarse, fwidthFine };
+export { dpdx, dpdxCoarse, dpdxFine, dpdy, dpdyCoarse, dpdyFine, fwidth, fwidthCoarse, fwidthFine };

package/std/discard.d.ts

@@ -1,6 +1,6 @@
 import { DualFn } from "../types.js";
 
 //#region src/std/discard.d.ts
-declare const discard: DualFn<() => never>;
+declare const discard: DualFn;
 //#endregion
 export { discard };

package/std/discard.js

@@ -1,6 +1,5 @@
 import { Void } from "../data/wgslTypes.js";
 import { dualImpl } from "../core/function/dualImpl.js";
-
 //#region src/std/discard.ts
 const discard = dualImpl({
 	name: "discard",
@@ -11,6 +10,5 @@ const discard = dualImpl({
 	},
 	codegenImpl: () => "discard;"
 });
-
 //#endregion
-export { discard };
+export { discard };

package/std/extensions.d.ts

@@ -1,8 +1,6 @@
 import { WgslExtension } from "../wgslExtensions.js";
 import { TgpuComptime } from "../core/function/comptime.js";
-import "../indexNamedExports.js";
-
 //#region src/std/extensions.d.ts
-declare const extensionEnabled: TgpuComptime<(extensionName: WgslExtension) => boolean>;
+declare const extensionEnabled: TgpuComptime;
 //#endregion
 export { extensionEnabled };

package/std/extensions.js

@@ -1,7 +1,6 @@
 import { getResolutionCtx } from "../execMode.js";
 import { comptime } from "../core/function/comptime.js";
 import { wgslExtensions } from "../wgslExtensions.js";
-
 //#region src/std/extensions.ts
 const extensionEnabled = comptime((extensionName) => {
 	const resolutionCtx = getResolutionCtx();
@@ -9,6 +8,5 @@ const extensionEnabled = comptime((extensionName) => {
 	if (typeof extensionName !== "string" || !wgslExtensions.includes(extensionName)) throw new Error(`extensionEnabled has to be called with a string literal representing a valid WGSL extension name. Got: '${extensionName}'`);
 	return (resolutionCtx.enableExtensions ?? []).includes(extensionName);
 });
-
 //#endregion
-export { extensionEnabled };
+export { extensionEnabled };

package/std/index.d.ts

@@ -1,21 +1,22 @@
 import { identity2, identity3, identity4, rotationX4, rotationY4, rotationZ4, scaling4, translation4 } from "../data/matrix.js";
 import { discard } from "./discard.js";
 import { abs, acos, acosh, asin, asinh, atan, atan2, atanh, ceil, clamp, cos, cosh, countLeadingZeros, countOneBits, countTrailingZeros, cross, degrees, determinant, distance, dot, dot4I8Packed, dot4U8Packed, exp, exp2, extractBits, faceForward, firstLeadingBit, firstTrailingBit, floor, fma, fract, frexp, insertBits, inverseSqrt, ldexp, length, log, log2, max, min, mix, modf, normalize, pow, quantizeToF16, radians, reflect, refract, reverseBits, round, saturate, sign, sin, sinh, smoothstep, sqrt, step, tan, tanh, transpose, trunc } from "./numeric.js";
-import { add, div, mod, mul, neg, sub } from "./operators.js";
+import { add, bitShiftLeft, bitShiftRight, div, mod, mul, neg, sub } from "./operators.js";
 import { rotateX4, rotateY4, rotateZ4, scale4, translate4 } from "./matrix.js";
 import { all, allEq, and, any, eq, ge, gt, isCloseTo, le, lt, ne, not, or, select } from "./boolean.js";
 import { atomicAdd, atomicAnd, atomicLoad, atomicMax, atomicMin, atomicOr, atomicStore, atomicSub, atomicXor, storageBarrier, textureBarrier, workgroupBarrier } from "./atomic.js";
 import { dpdx, dpdxCoarse, dpdxFine, dpdy, dpdyCoarse, dpdyFine, fwidth, fwidthCoarse, fwidthFine } from "./derivative.js";
 import { arrayLength } from "./array.js";
 import { pack2x16float, pack4x8unorm, unpack2x16float, unpack4x8unorm } from "./packing.js";
-import { textureDimensions, textureGather, textureLoad, textureSample, textureSampleBaseClampToEdge, textureSampleBias, textureSampleCompare, textureSampleCompareLevel, textureSampleLevel, textureStore } from "./texture.js";
+import { textureDimensions, textureGather, textureLoad, textureSample, textureSampleBaseClampToEdge, textureSampleBias, textureSampleCompare, textureSampleCompareLevel, textureSampleGrad, textureSampleLevel, textureStore } from "./texture.js";
 import { subgroupAdd, subgroupAll, subgroupAnd, subgroupAny, subgroupBallot, subgroupBroadcast, subgroupBroadcastFirst, subgroupElect, subgroupExclusiveAdd, subgroupExclusiveMul, subgroupInclusiveAdd, subgroupInclusiveMul, subgroupMax, subgroupMin, subgroupMul, subgroupOr, subgroupShuffle, subgroupShuffleDown, subgroupShuffleUp, subgroupShuffleXor, subgroupXor } from "./subgroup.js";
 import { extensionEnabled } from "./extensions.js";
 import { bitcastU32toF32, bitcastU32toI32 } from "./bitcast.js";
+import { range } from "./range.js";
 
 //#region src/std/index.d.ts
 declare namespace index_d_exports {
-  export { abs, acos, acosh, add, all, allEq, and, any, arrayLength, asin, asinh, atan, atan2, atanh, atomicAdd, atomicAnd, atomicLoad, atomicMax, atomicMin, atomicOr, atomicStore, atomicSub, atomicXor, bitcastU32toF32, bitcastU32toI32, ceil, clamp, cos, cosh, countLeadingZeros, countOneBits, countTrailingZeros, cross, degrees, determinant, discard, distance, div, dot, dot4I8Packed, dot4U8Packed, dpdx, dpdxCoarse, dpdxFine, dpdy, dpdyCoarse, dpdyFine, eq, exp, exp2, extensionEnabled, extractBits, faceForward, firstLeadingBit, firstTrailingBit, floor, fma, fract, frexp, fwidth, fwidthCoarse, fwidthFine, ge, gt, identity2, identity3, identity4, insertBits, inverseSqrt, isCloseTo, ldexp, le, length, log, log2, lt, max, min, mix, mod, modf, mul, ne, neg, normalize, not, or, pack2x16float, pack4x8unorm, pow, quantizeToF16, radians, reflect, refract, reverseBits, rotateX4, rotateY4, rotateZ4, rotationX4, rotationY4, rotationZ4, round, saturate, scale4, scaling4, select, sign, sin, sinh, smoothstep, sqrt, step, storageBarrier, sub, subgroupAdd, subgroupAll, subgroupAnd, subgroupAny, subgroupBallot, subgroupBroadcast, subgroupBroadcastFirst, subgroupElect, subgroupExclusiveAdd, subgroupExclusiveMul, subgroupInclusiveAdd, subgroupInclusiveMul, subgroupMax, subgroupMin, subgroupMul, subgroupOr, subgroupShuffle, subgroupShuffleDown, subgroupShuffleUp, subgroupShuffleXor, subgroupXor, tan, tanh, textureBarrier, textureDimensions, textureGather, textureLoad, textureSample, textureSampleBaseClampToEdge, textureSampleBias, textureSampleCompare, textureSampleCompareLevel, textureSampleLevel, textureStore, translate4, translation4, transpose, trunc, unpack2x16float, unpack4x8unorm, workgroupBarrier };
+  export { abs, acos, acosh, add, all, allEq, and, any, arrayLength, asin, asinh, atan, atan2, atanh, atomicAdd, atomicAnd, atomicLoad, atomicMax, atomicMin, atomicOr, atomicStore, atomicSub, atomicXor, bitShiftLeft, bitShiftRight, bitcastU32toF32, bitcastU32toI32, ceil, clamp, cos, cosh, countLeadingZeros, countOneBits, countTrailingZeros, cross, degrees, determinant, discard, distance, div, dot, dot4I8Packed, dot4U8Packed, dpdx, dpdxCoarse, dpdxFine, dpdy, dpdyCoarse, dpdyFine, eq, exp, exp2, extensionEnabled, extractBits, faceForward, firstLeadingBit, firstTrailingBit, floor, fma, fract, frexp, fwidth, fwidthCoarse, fwidthFine, ge, gt, identity2, identity3, identity4, insertBits, inverseSqrt, isCloseTo, ldexp, le, length, log, log2, lt, max, min, mix, mod, modf, mul, ne, neg, normalize, not, or, pack2x16float, pack4x8unorm, pow, quantizeToF16, radians, range, reflect, refract, reverseBits, rotateX4, rotateY4, rotateZ4, rotationX4, rotationY4, rotationZ4, round, saturate, scale4, scaling4, select, sign, sin, sinh, smoothstep, sqrt, step, storageBarrier, sub, subgroupAdd, subgroupAll, subgroupAnd, subgroupAny, subgroupBallot, subgroupBroadcast, subgroupBroadcastFirst, subgroupElect, subgroupExclusiveAdd, subgroupExclusiveMul, subgroupInclusiveAdd, subgroupInclusiveMul, subgroupMax, subgroupMin, subgroupMul, subgroupOr, subgroupShuffle, subgroupShuffleDown, subgroupShuffleUp, subgroupShuffleXor, subgroupXor, tan, tanh, textureBarrier, textureDimensions, textureGather, textureLoad, textureSample, textureSampleBaseClampToEdge, textureSampleBias, textureSampleCompare, textureSampleCompareLevel, textureSampleGrad, textureSampleLevel, textureStore, translate4, translation4, transpose, trunc, unpack2x16float, unpack4x8unorm, workgroupBarrier };
 }
 //#endregion
-export { abs, acos, acosh, add, all, allEq, and, any, arrayLength, asin, asinh, atan, atan2, atanh, atomicAdd, atomicAnd, atomicLoad, atomicMax, atomicMin, atomicOr, atomicStore, atomicSub, atomicXor, bitcastU32toF32, bitcastU32toI32, ceil, clamp, cos, cosh, countLeadingZeros, countOneBits, countTrailingZeros, cross, degrees, determinant, discard, distance, div, dot, dot4I8Packed, dot4U8Packed, dpdx, dpdxCoarse, dpdxFine, dpdy, dpdyCoarse, dpdyFine, eq, exp, exp2, extensionEnabled, extractBits, faceForward, firstLeadingBit, firstTrailingBit, floor, fma, fract, frexp, fwidth, fwidthCoarse, fwidthFine, ge, gt, identity2, identity3, identity4, index_d_exports, insertBits, inverseSqrt, isCloseTo, ldexp, le, length, log, log2, lt, max, min, mix, mod, modf, mul, ne, neg, normalize, not, or, pack2x16float, pack4x8unorm, pow, quantizeToF16, radians, reflect, refract, reverseBits, rotateX4, rotateY4, rotateZ4, rotationX4, rotationY4, rotationZ4, round, saturate, scale4, scaling4, select, sign, sin, sinh, smoothstep, sqrt, step, storageBarrier, sub, subgroupAdd, subgroupAll, subgroupAnd, subgroupAny, subgroupBallot, subgroupBroadcast, subgroupBroadcastFirst, subgroupElect, subgroupExclusiveAdd, subgroupExclusiveMul, subgroupInclusiveAdd, subgroupInclusiveMul, subgroupMax, subgroupMin, subgroupMul, subgroupOr, subgroupShuffle, subgroupShuffleDown, subgroupShuffleUp, subgroupShuffleXor, subgroupXor, tan, tanh, textureBarrier, textureDimensions, textureGather, textureLoad, textureSample, textureSampleBaseClampToEdge, textureSampleBias, textureSampleCompare, textureSampleCompareLevel, textureSampleLevel, textureStore, translate4, translation4, transpose, trunc, unpack2x16float, unpack4x8unorm, workgroupBarrier };
+export { abs, acos, acosh, add, all, allEq, and, any, arrayLength, asin, asinh, atan, atan2, atanh, atomicAdd, atomicAnd, atomicLoad, atomicMax, atomicMin, atomicOr, atomicStore, atomicSub, atomicXor, bitShiftLeft, bitShiftRight, bitcastU32toF32, bitcastU32toI32, ceil, clamp, cos, cosh, countLeadingZeros, countOneBits, countTrailingZeros, cross, degrees, determinant, discard, distance, div, dot, dot4I8Packed, dot4U8Packed, dpdx, dpdxCoarse, dpdxFine, dpdy, dpdyCoarse, dpdyFine, eq, exp, exp2, extensionEnabled, extractBits, faceForward, firstLeadingBit, firstTrailingBit, floor, fma, fract, frexp, fwidth, fwidthCoarse, fwidthFine, ge, gt, identity2, identity3, identity4, index_d_exports, insertBits, inverseSqrt, isCloseTo, ldexp, le, length, log, log2, lt, max, min, mix, mod, modf, mul, ne, neg, normalize, not, or, pack2x16float, pack4x8unorm, pow, quantizeToF16, radians, range, reflect, refract, reverseBits, rotateX4, rotateY4, rotateZ4, rotationX4, rotationY4, rotationZ4, round, saturate, scale4, scaling4, select, sign, sin, sinh, smoothstep, sqrt, step, storageBarrier, sub, subgroupAdd, subgroupAll, subgroupAnd, subgroupAny, subgroupBallot, subgroupBroadcast, subgroupBroadcastFirst, subgroupElect, subgroupExclusiveAdd, subgroupExclusiveMul, subgroupInclusiveAdd, subgroupInclusiveMul, subgroupMax, subgroupMin, subgroupMul, subgroupOr, subgroupShuffle, subgroupShuffleDown, subgroupShuffleUp, subgroupShuffleXor, subgroupXor, tan, tanh, textureBarrier, textureDimensions, textureGather, textureLoad, textureSample, textureSampleBaseClampToEdge, textureSampleBias, textureSampleCompare, textureSampleCompareLevel, textureSampleGrad, textureSampleLevel, textureStore, translate4, translation4, transpose, trunc, unpack2x16float, unpack4x8unorm, workgroupBarrier };

package/std/index.js

@@ -1,8 +1,9 @@
 import { __exportAll } from "../_virtual/_rolldown/runtime.js";
 import { identity2, identity3, identity4, rotationX4, rotationY4, rotationZ4, scaling4, translation4 } from "../data/matrix.js";
-import { add, div, mod, mul, neg, sub } from "./operators.js";
+import { add, bitShiftLeft, bitShiftRight, div, mod, mul, neg, sub } from "./operators.js";
 import { abs, acos, acosh, asin, asinh, atan, atan2, atanh, ceil, clamp, cos, cosh, countLeadingZeros, countOneBits, countTrailingZeros, cross, degrees, determinant, distance, dot, dot4I8Packed, dot4U8Packed, exp, exp2, extractBits, faceForward, firstLeadingBit, firstTrailingBit, floor, fma, fract, frexp, insertBits, inverseSqrt, ldexp, length, log, log2, max, min, mix, modf, normalize, pow, quantizeToF16, radians, reflect, refract, reverseBits, round, saturate, sign, sin, sinh, smoothstep, sqrt, step, tan, tanh, transpose, trunc } from "./numeric.js";
 import { arrayLength } from "./array.js";
+import { range } from "./range.js";
 import { bitcastU32toF32, bitcastU32toI32 } from "./bitcast.js";
 import { pack2x16float, pack4x8unorm, unpack2x16float, unpack4x8unorm } from "./packing.js";
 import { all, allEq, and, any, eq, ge, gt, isCloseTo, le, lt, ne, not, or, select } from "./boolean.js";
@@ -10,10 +11,9 @@ import { discard } from "./discard.js";
 import { rotateX4, rotateY4, rotateZ4, scale4, translate4 } from "./matrix.js";
 import { atomicAdd, atomicAnd, atomicLoad, atomicMax, atomicMin, atomicOr, atomicStore, atomicSub, atomicXor, storageBarrier, textureBarrier, workgroupBarrier } from "./atomic.js";
 import { dpdx, dpdxCoarse, dpdxFine, dpdy, dpdyCoarse, dpdyFine, fwidth, fwidthCoarse, fwidthFine } from "./derivative.js";
-import { textureDimensions, textureGather, textureLoad, textureSample, textureSampleBaseClampToEdge, textureSampleBias, textureSampleCompare, textureSampleCompareLevel, textureSampleLevel, textureStore } from "./texture.js";
+import { textureDimensions, textureGather, textureLoad, textureSample, textureSampleBaseClampToEdge, textureSampleBias, textureSampleCompare, textureSampleCompareLevel, textureSampleGrad, textureSampleLevel, textureStore } from "./texture.js";
 import { subgroupAdd, subgroupAll, subgroupAnd, subgroupAny, subgroupBallot, subgroupBroadcast, subgroupBroadcastFirst, subgroupElect, subgroupExclusiveAdd, subgroupExclusiveMul, subgroupInclusiveAdd, subgroupInclusiveMul, subgroupMax, subgroupMin, subgroupMul, subgroupOr, subgroupShuffle, subgroupShuffleDown, subgroupShuffleUp, subgroupShuffleXor, subgroupXor } from "./subgroup.js";
 import { extensionEnabled } from "./extensions.js";
-
 //#region src/std/index.ts
 var std_exports = /* @__PURE__ */ __exportAll({
 	abs: () => abs,
@@ -39,6 +39,8 @@ var std_exports = /* @__PURE__ */ __exportAll({
 	atomicStore: () => atomicStore,
 	atomicSub: () => atomicSub,
 	atomicXor: () => atomicXor,
+	bitShiftLeft: () => bitShiftLeft,
+	bitShiftRight: () => bitShiftRight,
 	bitcastU32toF32: () => bitcastU32toF32,
 	bitcastU32toI32: () => bitcastU32toI32,
 	ceil: () => ceil,
@@ -108,6 +110,7 @@ var std_exports = /* @__PURE__ */ __exportAll({
 	pow: () => pow,
 	quantizeToF16: () => quantizeToF16,
 	radians: () => radians,
+	range: () => range,
 	reflect: () => reflect,
 	refract: () => refract,
 	reverseBits: () => reverseBits,
@@ -162,6 +165,7 @@ var std_exports = /* @__PURE__ */ __exportAll({
 	textureSampleBias: () => textureSampleBias,
 	textureSampleCompare: () => textureSampleCompare,
 	textureSampleCompareLevel: () => textureSampleCompareLevel,
+	textureSampleGrad: () => textureSampleGrad,
 	textureSampleLevel: () => textureSampleLevel,
 	textureStore: () => textureStore,
 	translate4: () => translate4,
@@ -172,6 +176,5 @@ var std_exports = /* @__PURE__ */ __exportAll({
 	unpack4x8unorm: () => unpack4x8unorm,
 	workgroupBarrier: () => workgroupBarrier
 });
-
 //#endregion
-export { abs, acos, acosh, add, all, allEq, and, any, arrayLength, asin, asinh, atan, atan2, atanh, atomicAdd, atomicAnd, atomicLoad, atomicMax, atomicMin, atomicOr, atomicStore, atomicSub, atomicXor, bitcastU32toF32, bitcastU32toI32, ceil, clamp, cos, cosh, countLeadingZeros, countOneBits, countTrailingZeros, cross, degrees, determinant, discard, distance, div, dot, dot4I8Packed, dot4U8Packed, dpdx, dpdxCoarse, dpdxFine, dpdy, dpdyCoarse, dpdyFine, eq, exp, exp2, extensionEnabled, extractBits, faceForward, firstLeadingBit, firstTrailingBit, floor, fma, fract, frexp, fwidth, fwidthCoarse, fwidthFine, ge, gt, identity2, identity3, identity4, insertBits, inverseSqrt, isCloseTo, ldexp, le, length, log, log2, lt, max, min, mix, mod, modf, mul, ne, neg, normalize, not, or, pack2x16float, pack4x8unorm, pow, quantizeToF16, radians, reflect, refract, reverseBits, rotateX4, rotateY4, rotateZ4, rotationX4, rotationY4, rotationZ4, round, saturate, scale4, scaling4, select, sign, sin, sinh, smoothstep, sqrt, std_exports, step, storageBarrier, sub, subgroupAdd, subgroupAll, subgroupAnd, subgroupAny, subgroupBallot, subgroupBroadcast, subgroupBroadcastFirst, subgroupElect, subgroupExclusiveAdd, subgroupExclusiveMul, subgroupInclusiveAdd, subgroupInclusiveMul, subgroupMax, subgroupMin, subgroupMul, subgroupOr, subgroupShuffle, subgroupShuffleDown, subgroupShuffleUp, subgroupShuffleXor, subgroupXor, tan, tanh, textureBarrier, textureDimensions, textureGather, textureLoad, textureSample, textureSampleBaseClampToEdge, textureSampleBias, textureSampleCompare, textureSampleCompareLevel, textureSampleLevel, textureStore, translate4, translation4, transpose, trunc, unpack2x16float, unpack4x8unorm, workgroupBarrier };
+export { abs, acos, acosh, add, all, allEq, and, any, arrayLength, asin, asinh, atan, atan2, atanh, atomicAdd, atomicAnd, atomicLoad, atomicMax, atomicMin, atomicOr, atomicStore, atomicSub, atomicXor, bitShiftLeft, bitShiftRight, bitcastU32toF32, bitcastU32toI32, ceil, clamp, cos, cosh, countLeadingZeros, countOneBits, countTrailingZeros, cross, degrees, determinant, discard, distance, div, dot, dot4I8Packed, dot4U8Packed, dpdx, dpdxCoarse, dpdxFine, dpdy, dpdyCoarse, dpdyFine, eq, exp, exp2, extensionEnabled, extractBits, faceForward, firstLeadingBit, firstTrailingBit, floor, fma, fract, frexp, fwidth, fwidthCoarse, fwidthFine, ge, gt, identity2, identity3, identity4, insertBits, inverseSqrt, isCloseTo, ldexp, le, length, log, log2, lt, max, min, mix, mod, modf, mul, ne, neg, normalize, not, or, pack2x16float, pack4x8unorm, pow, quantizeToF16, radians, range, reflect, refract, reverseBits, rotateX4, rotateY4, rotateZ4, rotationX4, rotationY4, rotationZ4, round, saturate, scale4, scaling4, select, sign, sin, sinh, smoothstep, sqrt, std_exports, step, storageBarrier, sub, subgroupAdd, subgroupAll, subgroupAnd, subgroupAny, subgroupBallot, subgroupBroadcast, subgroupBroadcastFirst, subgroupElect, subgroupExclusiveAdd, subgroupExclusiveMul, subgroupInclusiveAdd, subgroupInclusiveMul, subgroupMax, subgroupMin, subgroupMul, subgroupOr, subgroupShuffle, subgroupShuffleDown, subgroupShuffleUp, subgroupShuffleXor, subgroupXor, tan, tanh, textureBarrier, textureDimensions, textureGather, textureLoad, textureSample, textureSampleBaseClampToEdge, textureSampleBias, textureSampleCompare, textureSampleCompareLevel, textureSampleGrad, textureSampleLevel, textureStore, translate4, translation4, transpose, trunc, unpack2x16float, unpack4x8unorm, workgroupBarrier };

package/std/matrix.d.ts

@@ -8,34 +8,34 @@ import { m4x4f, v3f } from "../data/wgslTypes.js";
  * @param {v3f} vector - The vector by which to translate the matrix.
  * @returns {m4x4f} The translated matrix.
  */
-declare const translate4: DualFn<(matrix: m4x4f, vector: v3f) => m4x4f>;
+declare const translate4: DualFn;
 /**
  * Scales the given 4-by-4 matrix in each dimension by an amount given by the corresponding entry in the given vector.
  * @param {m4x4f} matrix - The matrix to be modified.
  * @param {v3f} vector - A vector of three entries specifying the factor by which to scale in each dimension.
  * @returns {m4x4f} The scaled matrix.
  */
-declare const scale4: DualFn<(matrix: m4x4f, vector: v3f) => m4x4f>;
+declare const scale4: DualFn;
 /**
  * Rotates the given 4-by-4 matrix around the x-axis by the given angle.
  * @param {m4x4f} matrix - The matrix to be modified.
  * @param {number} angle - The angle by which to rotate (in radians).
  * @returns {m4x4f} The rotated matrix.
  */
-declare const rotateX4: DualFn<(matrix: m4x4f, angle: number) => m4x4f>;
+declare const rotateX4: DualFn;
 /**
  * Rotates the given 4-by-4 matrix around the y-axis by the given angle.
  * @param {m4x4f} matrix - The matrix to be modified.
  * @param {number} angle - The angle by which to rotate (in radians).
  * @returns {m4x4f} The rotated matrix.
  */
-declare const rotateY4: DualFn<(matrix: m4x4f, angle: number) => m4x4f>;
+declare const rotateY4: DualFn;
 /**
  * Rotates the given 4-by-4 matrix around the z-axis by the given angle.
  * @param {m4x4f} matrix - The matrix to be modified.
  * @param {number} angle - The angle by which to rotate (in radians).
  * @returns {m4x4f} The rotated matrix.
  */
-declare const rotateZ4: DualFn<(matrix: m4x4f, angle: number) => m4x4f>;
+declare const rotateZ4: DualFn;
 //#endregion
 export { rotateX4, rotateY4, rotateZ4, scale4, translate4 };

package/std/matrix.js

@@ -5,7 +5,6 @@ import { vec3f } from "../data/vector.js";
 import { dualImpl } from "../core/function/dualImpl.js";
 import { mat4x4f, rotationX4, rotationY4, rotationZ4, scaling4, translation4 } from "../data/matrix.js";
 import { mul } from "./operators.js";
-
 //#region src/std/matrix.ts
 const gpuTranslation4 = translation4[$gpuCallable].call.bind(translation4);
 const gpuScaling4 = scaling4[$gpuCallable].call.bind(scaling4);
@@ -82,6 +81,5 @@ const rotateZ4 = dualImpl({
 	signature: rotateSignature,
 	codegenImpl: (ctx, [matrix, angle]) => stitch`(${gpuRotationZ4(ctx, [angle])} * ${matrix})`
 });
-
 //#endregion
-export { rotateX4, rotateY4, rotateZ4, scale4, translate4 };
+export { rotateX4, rotateY4, rotateZ4, scale4, translate4 };