typegpu 0.11.1 → 0.11.3

package/std/boolean.d.ts

@@ -2,6 +2,7 @@ import { DualFn } from "../types.js";
 import { AnyBooleanVecInstance, AnyFloatVecInstance, AnyNumericVecInstance, AnyVec2Instance, AnyVec3Instance, AnyVecInstance, v2b, v3b, v4b } from "../data/wgslTypes.js";
 
 //#region src/std/boolean.d.ts
+
 /**
  * Checks whether `lhs == rhs` on all components.
  * Equivalent to `all(eq(lhs, rhs))`.
@@ -9,7 +10,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;
+declare const allEq: DualFn<(<T extends AnyVecInstance>(lhs: T, rhs: T) => boolean)>;
 /**
  * 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 +20,7 @@ declare const allEq: DualFn;
  * 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;
+declare const eq: DualFn<(<T extends AnyVecInstance>(lhs: T, rhs: T) => T extends AnyVec2Instance ? v2b : T extends AnyVec3Instance ? v3b : v4b)>;
 /**
  * Checks **component-wise** whether `lhs != rhs`.
  * This function does **not** return `bool`, for that use-case, wrap the result in `any`.
@@ -28,7 +29,7 @@ declare const eq: DualFn;
  * 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;
+declare const ne: DualFn<(<T extends AnyVecInstance>(lhs: T, rhs: T) => VecInstanceToBooleanVecInstance<T extends AnyVec2Instance ? v2b : T extends AnyVec3Instance ? v3b : v4b>)>;
 /**
  * Checks **component-wise** whether `lhs < rhs`.
  * This function does **not** return `bool`, for that use-case, wrap the result in `all`.
@@ -37,7 +38,7 @@ declare const ne: DualFn;
  * 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;
+declare const lt: DualFn<(<T extends AnyNumericVecInstance>(lhs: T, rhs: T) => T extends AnyVec2Instance ? v2b : T extends AnyVec3Instance ? v3b : v4b)>;
 /**
  * Checks **component-wise** whether `lhs <= rhs`.
  * This function does **not** return `bool`, for that use-case, wrap the result in `all`.
@@ -46,7 +47,7 @@ declare const lt: DualFn;
  * 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;
+declare const le: DualFn<(<T extends AnyNumericVecInstance>(lhs: T, rhs: T) => T extends AnyVec2Instance ? v2b : T extends AnyVec3Instance ? v3b : v4b)>;
 /**
  * Checks **component-wise** whether `lhs > rhs`.
  * This function does **not** return `bool`, for that use-case, wrap the result in `all`.
@@ -55,7 +56,7 @@ declare const le: DualFn;
  * 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;
+declare const gt: DualFn<(<T extends AnyNumericVecInstance>(lhs: T, rhs: T) => VecInstanceToBooleanVecInstance<VecInstanceToBooleanVecInstance<VecInstanceToBooleanVecInstance<T extends AnyVec2Instance ? v2b : T extends AnyVec3Instance ? v3b : v4b>>>)>;
 /**
  * Checks **component-wise** whether `lhs >= rhs`.
  * This function does **not** return `bool`, for that use-case, wrap the result in `all`.
@@ -64,7 +65,7 @@ declare const gt: DualFn;
  * 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;
+declare const ge: DualFn<(<T extends AnyNumericVecInstance>(lhs: T, rhs: T) => VecInstanceToBooleanVecInstance<T extends AnyVec2Instance ? v2b : T extends AnyVec3Instance ? v3b : v4b>)>;
 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;
@@ -85,35 +86,35 @@ declare function cpuNot(value: unknown): boolean;
  * not({a: 1882}) // returns false
  * not(NaN) // returns false **as in WGSL**
  */
-declare const not: DualFn;
+declare const not: DualFn<typeof cpuNot>;
 /**
  * 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;
+declare const or: DualFn<(<T extends AnyBooleanVecInstance>(lhs: T, rhs: T) => T)>;
 /**
  * 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;
+declare const and: DualFn<(<T extends AnyBooleanVecInstance>(lhs: T, rhs: T) => VecInstanceToBooleanVecInstance<VecInstanceToBooleanVecInstance<T>>)>;
 /**
  * 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;
+declare const all: DualFn<(value: AnyBooleanVecInstance) => boolean>;
 /**
  * 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;
+declare const any: DualFn<(value: AnyBooleanVecInstance) => boolean>;
 /**
  * Checks whether the given elements differ by at most the `precision` value.
  * Checks all elements of `lhs` and `rhs` if arguments are vectors.
@@ -123,7 +124,7 @@ declare const any: DualFn;
  *
  * @param {number} precision argument that specifies the maximum allowed difference, 0.01 by default.
  */
-declare const isCloseTo: DualFn;
+declare const isCloseTo: DualFn<(<T extends AnyFloatVecInstance | number>(lhs: T, rhs: T, precision?: number) => boolean)>;
 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;
@@ -136,6 +137,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;
+declare const select: DualFn<typeof cpuSelect>;
 //#endregion
 export { all, allEq, and, any, eq, ge, gt, isCloseTo, le, lt, ne, not, or, select };

package/std/boolean.js

@@ -7,6 +7,7 @@ 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;
@@ -295,5 +296,6 @@ 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;
-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;
+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>;
 //#endregion
 export { dpdx, dpdxCoarse, dpdxFine, dpdy, dpdyCoarse, dpdyFine, fwidth, fwidthCoarse, fwidthFine };

package/std/derivative.js

@@ -1,5 +1,6 @@
 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({
@@ -83,5 +84,6 @@ 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;
+declare const discard: DualFn<() => never>;
 //#endregion
 export { discard };

package/std/discard.js

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

package/std/extensions.d.ts

@@ -1,6 +1,8 @@
 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;
+declare const extensionEnabled: TgpuComptime<(extensionName: WgslExtension) => boolean>;
 //#endregion
 export { extensionEnabled };

package/std/extensions.js

@@ -1,6 +1,7 @@
 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();
@@ -8,5 +9,6 @@ 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.js

@@ -1,4 +1,4 @@
-import { __exportAll } from "../_virtual/_rolldown/runtime.js";
+import { __export } from "../_virtual/rolldown_runtime.js";
 import { identity2, identity3, identity4, rotationX4, rotationY4, rotationZ4, scaling4, translation4 } from "../data/matrix.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";
@@ -14,8 +14,9 @@ import { dpdx, dpdxCoarse, dpdxFine, dpdy, dpdyCoarse, dpdyFine, fwidth, fwidthC
 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({
+var std_exports = /* @__PURE__ */ __export({
 	abs: () => abs,
 	acos: () => acos,
 	acosh: () => acosh,
@@ -176,5 +177,6 @@ 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, 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 };
+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

@@ -2,40 +2,41 @@ import { DualFn } from "../types.js";
 import { m4x4f, v3f } from "../data/wgslTypes.js";
 
 //#region src/std/matrix.d.ts
+
 /**
  * Translates the given 4-by-4 matrix by the given vector.
  * @param {m4x4f} matrix - The matrix to be modified.
  * @param {v3f} vector - The vector by which to translate the matrix.
  * @returns {m4x4f} The translated matrix.
  */
-declare const translate4: DualFn;
+declare const translate4: DualFn<(matrix: m4x4f, vector: v3f) => m4x4f>;
 /**
  * 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;
+declare const scale4: DualFn<(matrix: m4x4f, vector: v3f) => m4x4f>;
 /**
  * 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;
+declare const rotateX4: DualFn<(matrix: m4x4f, angle: number) => m4x4f>;
 /**
  * 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;
+declare const rotateY4: DualFn<(matrix: m4x4f, angle: number) => m4x4f>;
 /**
  * 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;
+declare const rotateZ4: DualFn<(matrix: m4x4f, angle: number) => m4x4f>;
 //#endregion
 export { rotateX4, rotateY4, rotateZ4, scale4, translate4 };

package/std/matrix.js

@@ -5,6 +5,7 @@ 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);
@@ -81,5 +82,6 @@ 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 };