typegpu 0.10.2 → 0.11.0-rc.1

package/core/pipeline/timeable.js

@@ -1,15 +1,12 @@
 import { $internal } from "../../shared/symbols.js";
 import { isQuerySet } from "../querySet/querySet.js";
-
 //#region src/core/pipeline/timeable.ts
-function createWithPerformanceCallback(currentPriors, callback, root) {
-	if (!root.enabledFeatures.has("timestamp-query")) throw new Error("Performance callback requires the \"timestamp-query\" feature to be enabled on GPU device.");
+function createWithPerformanceCallback(currentPriors, callback, querySet) {
 	if (!currentPriors.timestampWrites) return {
 		...currentPriors,
 		performanceCallback: callback,
-		hasAutoQuerySet: true,
 		timestampWrites: {
-			querySet: root.createQuerySet("timestamp", 2),
+			querySet,
 			beginningOfPassWriteIndex: 0,
 			endOfPassWriteIndex: 1
 		}
@@ -21,13 +18,11 @@ function createWithPerformanceCallback(currentPriors, callback, root) {
 }
 function createWithTimestampWrites(currentPriors, options, root) {
 	if (!root.enabledFeatures.has("timestamp-query")) throw new Error("Timestamp writes require the \"timestamp-query\" feature to be enabled on GPU device.");
-	if (currentPriors.hasAutoQuerySet && currentPriors.timestampWrites) currentPriors.timestampWrites.querySet.destroy();
 	const timestampWrites = { querySet: options.querySet };
 	if (options.beginningOfPassWriteIndex !== void 0) timestampWrites.beginningOfPassWriteIndex = options.beginningOfPassWriteIndex;
 	if (options.endOfPassWriteIndex !== void 0) timestampWrites.endOfPassWriteIndex = options.endOfPassWriteIndex;
 	return {
 		...currentPriors,
-		hasAutoQuerySet: false,
 		timestampWrites
 	};
 }
@@ -56,6 +51,5 @@ function triggerPerformanceCallback({ root, priors }) {
 		await callback(start, end);
 	});
 }
-
 //#endregion
-export { createWithPerformanceCallback, createWithTimestampWrites, setupTimestampWrites, triggerPerformanceCallback };
+export { createWithPerformanceCallback, createWithTimestampWrites, setupTimestampWrites, triggerPerformanceCallback };

package/core/pipeline/typeGuards.js

@@ -1,5 +1,4 @@
 import { $internal } from "../../shared/symbols.js";
-
 //#region src/core/pipeline/typeGuards.ts
 function isComputePipeline(value) {
 	const maybe = value;
@@ -24,6 +23,5 @@ function isGPURenderPassEncoder(value) {
 function isGPURenderBundleEncoder(value) {
 	return !!value && typeof value === "object" && "draw" in value && "finish" in value && !("executeBundles" in value) && !("beginRenderPass" in value) && !("dispatchWorkgroups" in value);
 }
-
 //#endregion
-export { isComputePipeline, isGPUCommandEncoder, isGPUComputePassEncoder, isGPURenderBundleEncoder, isGPURenderPassEncoder, isPipeline, isRenderPipeline };
+export { isComputePipeline, isGPUCommandEncoder, isGPUComputePassEncoder, isGPURenderBundleEncoder, isGPURenderPassEncoder, isPipeline, isRenderPipeline };

package/core/querySet/querySet.d.ts

@@ -1,7 +1,5 @@
 import { TgpuNamable } from "../../shared/meta.js";
 import { $internal } from "../../shared/symbols.js";
-import "../root/rootTypes.js";
-
 //#region src/core/querySet/querySet.d.ts
 interface TgpuQuerySet<T extends GPUQueryType> extends TgpuNamable {
   readonly resourceType: 'query-set';

package/core/querySet/querySet.js

@@ -1,6 +1,5 @@
 import { $internal } from "../../shared/symbols.js";
 import { setName } from "../../shared/meta.js";
-
 //#region src/core/querySet/querySet.ts
 function INTERNAL_createQuerySet(group, type, count, rawQuerySet) {
 	return new TgpuQuerySetImpl(group, type, count, rawQuerySet);
@@ -11,71 +10,74 @@ function isQuerySet(value) {
 }
 var TgpuQuerySetImpl = class {
 	resourceType = "query-set";
+	type;
+	count;
+	#rawQuerySet;
 	#device;
-	_querySet = null;
-	_ownQuerySet;
-	_destroyed = false;
-	_available = true;
-	_readBuffer = null;
-	_resolveBuffer = null;
+	#querySet;
+	#ownQuerySet;
+	#destroyed = false;
+	#available = true;
+	#readBuffer = void 0;
+	#resolveBuffer = void 0;
 	constructor(root, type, count, rawQuerySet) {
+		this.#device = root.device;
 		this.type = type;
 		this.count = count;
-		this.rawQuerySet = rawQuerySet;
-		this.#device = root.device;
-		this._ownQuerySet = !rawQuerySet;
-		this._querySet = rawQuerySet || null;
+		this.#rawQuerySet = rawQuerySet;
+		this.#ownQuerySet = !rawQuerySet;
+		this.#querySet = rawQuerySet;
 	}
 	get querySet() {
-		if (this._destroyed) throw new Error("This QuerySet has been destroyed.");
-		if (this.rawQuerySet) return this.rawQuerySet;
-		if (this._querySet) return this._querySet;
-		this._querySet = this.#device.createQuerySet({
+		if (this.#destroyed) throw new Error("This QuerySet has been destroyed.");
+		if (this.#rawQuerySet) return this.#rawQuerySet;
+		if (this.#querySet) return this.#querySet;
+		this.#querySet = this.#device.createQuerySet({
 			type: this.type,
 			count: this.count
 		});
-		return this._querySet;
+		return this.#querySet;
 	}
 	get destroyed() {
-		return this._destroyed;
+		return this.#destroyed;
 	}
 	get available() {
-		return this._available;
+		return this.#available;
 	}
 	get [$internal]() {
 		const self = this;
 		return {
 			get readBuffer() {
-				if (!self._readBuffer) self._readBuffer = self.#device.createBuffer({
+				if (!self.#readBuffer) self.#readBuffer = self.#device.createBuffer({
 					size: self.count * BigUint64Array.BYTES_PER_ELEMENT,
 					usage: GPUBufferUsage.COPY_DST | GPUBufferUsage.MAP_READ
 				});
-				return self._readBuffer;
+				return self.#readBuffer;
 			},
 			get resolveBuffer() {
-				if (!self._resolveBuffer) self._resolveBuffer = self.#device.createBuffer({
+				if (!self.#resolveBuffer) self.#resolveBuffer = self.#device.createBuffer({
 					size: self.count * BigUint64Array.BYTES_PER_ELEMENT,
 					usage: GPUBufferUsage.QUERY_RESOLVE | GPUBufferUsage.COPY_SRC
 				});
-				return self._resolveBuffer;
+				return self.#resolveBuffer;
 			}
 		};
 	}
 	$name(label) {
 		setName(this, label);
-		if (this._querySet) this._querySet.label = label;
+		if (this.#querySet) this.#querySet.label = label;
 		return this;
 	}
 	resolve() {
-		if (this._destroyed) throw new Error("This QuerySet has been destroyed.");
-		if (!this._available) throw new Error("This QuerySet is busy resolving or reading.");
+		if (this.#destroyed) throw new Error("This QuerySet has been destroyed.");
+		if (!this.#available) throw new Error("This QuerySet is busy resolving or reading.");
 		const commandEncoder = this.#device.createCommandEncoder();
 		commandEncoder.resolveQuerySet(this.querySet, 0, this.count, this[$internal].resolveBuffer, 0);
 		this.#device.queue.submit([commandEncoder.finish()]);
 	}
 	async read() {
-		if (!this._resolveBuffer) throw new Error("QuerySet must be resolved before reading.");
-		this._available = false;
+		if (!this.#resolveBuffer) throw new Error("QuerySet must be resolved before reading.");
+		this.#available = false;
 		try {
 			const commandEncoder = this.#device.createCommandEncoder();
 			commandEncoder.copyBufferToBuffer(this[$internal].resolveBuffer, 0, this[$internal].readBuffer, 0, this.count * BigUint64Array.BYTES_PER_ELEMENT);
@@ -86,18 +88,17 @@ var TgpuQuerySetImpl = class {
 			readBuffer.unmap();
 			return Array.from(data);
 		} finally {
-			this._available = true;
+			this.#available = true;
 		}
 	}
 	destroy() {
-		if (this._destroyed) return;
-		this._destroyed = true;
-		if (this._querySet && this._ownQuerySet) this._querySet.destroy();
-		this._readBuffer?.destroy();
-		this._resolveBuffer?.destroy();
-		this._readBuffer = this._resolveBuffer = null;
+		if (this.#destroyed) return;
+		this.#destroyed = true;
+		if (this.#querySet && this.#ownQuerySet) this.#querySet.destroy();
+		this.#readBuffer?.destroy();
+		this.#resolveBuffer?.destroy();
+		this.#readBuffer = this.#resolveBuffer = void 0;
 	}
 };
-
 //#endregion
-export { INTERNAL_createQuerySet, isQuerySet };
+export { INTERNAL_createQuerySet, isQuerySet };

package/core/rawCodeSnippet/tgpuRawCodeSnippet.js

@@ -3,7 +3,6 @@ import { snip } from "../../data/snippet.js";
 import { inCodegenMode } from "../../execMode.js";
 import { valueProxyHandler } from "../valueProxyUtils.js";
 import { applyExternals, replaceExternalsInWgsl } from "../resolve/externals.js";
-
 //#region src/core/rawCodeSnippet/tgpuRawCodeSnippet.ts
 /**
 * An advanced API that creates a typed shader expression which
@@ -91,6 +90,5 @@ var TgpuRawCodeSnippetImpl = class {
 		return this.$;
 	}
 };
-
 //#endregion
-export { rawCodeSnippet };
+export { rawCodeSnippet };

package/core/resolve/externals.d.ts

@@ -1,5 +1,3 @@
-import "../../types.js";
-
 //#region src/core/resolve/externals.d.ts
 /**
  * A key-value mapping where keys represent identifiers within shader code,

package/core/resolve/externals.js

@@ -2,7 +2,6 @@ import { getName, hasTinyestMetadata, setName } from "../../shared/meta.js";
 import { isWgslStruct } from "../../data/wgslTypes.js";
 import { isLooseData } from "../../data/dataTypes.js";
 import { isWgsl } from "../../types.js";
-
 //#region src/core/resolve/externals.ts
 /**
 * Merges two external maps into one. If a key is present in both maps, the value from the new map is used.
@@ -43,7 +42,7 @@ function identifierRegex(name) {
 function replaceExternalsInWgsl(ctx, externalMap, wgsl) {
 	return Object.entries(externalMap).reduce((acc, [externalName, external]) => {
 		const externalRegex = identifierRegex(externalName);
-		if (wgsl && externalName !== "Out" && externalName !== "In" && !externalRegex.test(wgsl)) console.warn(`The external '${externalName}' wasn't used in the resolved template.`);
+		if (wgsl && externalName !== "Out" && externalName !== "in" && !externalRegex.test(wgsl)) console.warn(`The external '${externalName}' wasn't used in the resolved template.`);
 		if (isWgsl(external) || isLooseData(external) || hasTinyestMetadata(external)) return acc.replaceAll(externalRegex, ctx.resolve(external).value);
 		if (external !== null && typeof external === "object") {
 			const foundProperties = [...wgsl.matchAll(new RegExp(`${externalName.replaceAll(".", "\\.").replaceAll("$", "\\$")}\\.(?<prop>.*?)(?![\\w\\$_])`, "g"))].map((found) => found[1]);
@@ -53,6 +52,5 @@ function replaceExternalsInWgsl(ctx, externalMap, wgsl) {
 		return acc;
 	}, wgsl);
 }
-
 //#endregion
-export { addArgTypesToExternals, addReturnTypeToExternals, applyExternals, replaceExternalsInWgsl };
+export { addArgTypesToExternals, addReturnTypeToExternals, applyExternals, replaceExternalsInWgsl };

package/core/resolve/namespace.js

@@ -2,7 +2,6 @@ import { $internal } from "../../shared/symbols.js";
 import { getName } from "../../shared/meta.js";
 import { RandomNameRegistry, StrictNameRegistry } from "../../nameRegistry.js";
 import { ShelllessRepository } from "../../tgsl/shellless.js";
-
 //#region src/core/resolve/namespace.ts
 var NamespaceImpl = class {
 	[$internal];
@@ -36,6 +35,5 @@ function namespace(options) {
 	const { names = "strict" } = options ?? {};
 	return new NamespaceImpl(names === "strict" ? new StrictNameRegistry() : new RandomNameRegistry());
 }
-
 //#endregion
-export { getUniqueName, namespace };
+export { getUniqueName, namespace };

package/core/resolve/resolveData.js

@@ -6,7 +6,6 @@ import { getAttributesString } from "../../data/attributes.js";
 import { isWgslComparisonSampler, isWgslSampler } from "../../data/sampler.js";
 import { accessModeMap, isWgslStorageTexture, isWgslTexture } from "../../data/texture.js";
 import { isAttribute } from "../vertexLayout/connectAttributesToShader.js";
-
 //#region src/core/resolve/resolveData.ts
 /**
 * Schemas for which their `type` property directly
@@ -141,6 +140,5 @@ function resolveData(ctx, data) {
 	if (isWgslComparisonSampler(data) || isWgslSampler(data)) return data.type;
 	assertExhaustive(data, "resolveData");
 }
-
 //#endregion
-export { resolveData };
+export { resolveData };

package/core/resolve/stitch.js

@@ -1,6 +1,5 @@
 import { isSnippet } from "../../data/snippet.js";
 import { getResolutionCtx } from "../../execMode.js";
-
 //#region src/core/resolve/stitch.ts
 /**
 * "The reverse of snipping"
@@ -20,6 +19,5 @@ function stitch(strings, ...snippets) {
 	}
 	return result;
 }
-
 //#endregion
-export { stitch };
+export { stitch };

package/core/resolve/tgpuResolve.d.ts

@@ -29,10 +29,12 @@ interface TgpuResolveOptions {
    */
   enableExtensions?: WgslExtension[] | undefined;
   /**
+   * **NOTE: This is an unstable API and may change in the future.**
+   *
    * A custom shader code generator, used when resolving TypeGPU functions.
    * If not provided, the default WGSL generator will be used.
    */
-  shaderGenerator?: ShaderGenerator | undefined;
+  unstable_shaderGenerator?: ShaderGenerator | undefined;
 }
 interface TgpuExtendedResolveOptions extends TgpuResolveOptions {
   /**

package/core/resolve/tgpuResolve.js

@@ -6,7 +6,6 @@ import { namespace } from "./namespace.js";
 import { isBindGroupLayout } from "../../tgpuBindGroupLayout.js";
 import { resolve as resolve$1 } from "../../resolutionCtx.js";
 import { isPipeline } from "../pipeline/typeGuards.js";
-
 //#region src/core/resolve/tgpuResolve.ts
 function resolveWithContext(arg0, options) {
 	if (Array.isArray(arg0)) return resolveFromArray(arg0, options);
@@ -17,7 +16,7 @@ function resolve(arg, options) {
 	return resolveWithContext(arg).code;
 }
 function resolveFromTemplate(options) {
-	const { template, externals, shaderGenerator, names = "strict", config, enableExtensions } = options;
+	const { template, externals, unstable_shaderGenerator: shaderGenerator, names = "strict", config, enableExtensions } = options;
 	if (!template) console.warn("Calling resolve with an empty template is deprecated and will soon return an empty string. Consider using the 'tgpu.resolve(resolvableArray, options)' API instead.");
 	const dependencies = {};
 	applyExternals(dependencies, externals ?? {});
@@ -36,7 +35,7 @@ function resolveFromTemplate(options) {
 	});
 }
 function resolveFromArray(items, options) {
-	const { shaderGenerator, names = "strict", config, enableExtensions } = options ?? {};
+	const { unstable_shaderGenerator: shaderGenerator, names = "strict", config, enableExtensions } = options ?? {};
 	return resolve$1({
 		[$internal]: true,
 		[$resolve](ctx) {
@@ -63,6 +62,5 @@ function tryFindRoot(items) {
 	if (pipelines.length > 1) throw new Error(`Found ${pipelines.length} pipelines but can only resolve one at a time.`);
 	return pipelines[0]?.[$internal].root;
 }
-
 //#endregion
-export { resolve, resolveWithContext };
+export { resolve, resolveWithContext };

package/core/root/configurableImpl.js

@@ -1,7 +1,7 @@
 import { isAccessor, isMutableAccessor } from "../slot/slotTypes.js";
-
 //#region src/core/root/configurableImpl.ts
 var ConfigurableImpl = class ConfigurableImpl {
+	bindings;
 	constructor(bindings) {
 		this.bindings = bindings;
 	}
@@ -13,6 +13,5 @@ var ConfigurableImpl = class ConfigurableImpl {
 		return new ConfigurableImpl([...this.bindings, ...newCfg.bindings]);
 	}
 };
-
 //#endregion
-export { ConfigurableImpl };
+export { ConfigurableImpl };

package/core/root/init.d.ts

@@ -1,11 +1,6 @@
-import "../buffer/bufferShorthand.js";
 import { LogGeneratorOptions } from "../../tgsl/consoleLog/types.js";
-import "../pipeline/computePipeline.js";
-import "../../tgpuBindGroupLayout.js";
 import { ShaderGenerator } from "../../tgsl/shaderGenerator.js";
 import { TgpuRoot } from "./rootTypes.js";
-import "../../data/wgslTypes.js";
-
 //#region src/core/root/init.d.ts
 /**
  * Options passed into {@link init}.

package/core/root/init.js

@@ -23,7 +23,6 @@ import { INTERNAL_createComputePipeline } from "../pipeline/computePipeline.js";
 import { isVertexLayout } from "../vertexLayout/vertexLayout.js";
 import { INTERNAL_createRenderPipeline } from "../pipeline/renderPipeline.js";
 import { allEq } from "../../std/boolean.js";
-
 //#region src/core/root/init.ts
 /**
 * Changes the given array to a vec of 3 numbers, filling missing values with 1.
@@ -51,8 +50,8 @@ var TgpuGuardedComputePipelineImpl = class TgpuGuardedComputePipelineImpl {
 		this.#workgroupSize = workgroupSize;
 		this.#lastSize = vec3u();
 	}
-	with(bindGroup) {
-		return new TgpuGuardedComputePipelineImpl(this.#root, this.#pipeline.with(bindGroup), this.#sizeUniform, this.#workgroupSize);
+	with(bindGroupOrEncoder) {
+		return new TgpuGuardedComputePipelineImpl(this.#root, this.#pipeline.with(bindGroupOrEncoder), this.#sizeUniform, this.#workgroupSize);
 	}
 	dispatchThreads(...threads) {
 		const sanitizedSize = toVec3(threads);
@@ -79,28 +78,30 @@ var TgpuGuardedComputePipelineImpl = class TgpuGuardedComputePipelineImpl {
 	}
 };
 var WithBindingImpl = class WithBindingImpl {
-	constructor(_getRoot, _slotBindings) {
-		this._getRoot = _getRoot;
-		this._slotBindings = _slotBindings;
+	#getRoot;
+	#slotBindings;
+	constructor(getRoot, slotBindings) {
+		this.#getRoot = getRoot;
+		this.#slotBindings = slotBindings;
 	}
 	with(slot, value) {
-		return new WithBindingImpl(this._getRoot, [...this._slotBindings, [isAccessor(slot) || isMutableAccessor(slot) ? slot.slot : slot, value]]);
+		return new WithBindingImpl(this.#getRoot, [...this.#slotBindings, [isAccessor(slot) || isMutableAccessor(slot) ? slot.slot : slot, value]]);
 	}
 	withCompute(entryFn) {
-		return new WithComputeImpl(this._getRoot(), this._slotBindings, entryFn);
+		return new WithComputeImpl(this.#getRoot(), this.#slotBindings, entryFn);
 	}
 	createComputePipeline(descriptor) {
-		return INTERNAL_createComputePipeline(this._getRoot(), this._slotBindings, descriptor);
+		return INTERNAL_createComputePipeline(this.#getRoot(), this.#slotBindings, descriptor);
 	}
 	createRenderPipeline(descriptor) {
 		return INTERNAL_createRenderPipeline({
-			root: this._getRoot(),
-			slotBindings: this._slotBindings,
+			root: this.#getRoot(),
+			slotBindings: this.#slotBindings,
 			descriptor
 		});
 	}
 	createGuardedComputePipeline(callback) {
-		const root = this._getRoot();
+		const root = this.#getRoot();
 		if (callback.length >= 4) throw new Error("Guarded compute callback only supports up to three dimensions.");
 		const workgroupSize = workgroupSizeConfigs[callback.length];
 		const wrappedCallback = fn([
@@ -124,24 +125,27 @@ var WithBindingImpl = class WithBindingImpl {
 		return new TgpuGuardedComputePipelineImpl(root, this.createComputePipeline({ compute: mainCompute }), sizeUniform, workgroupSize);
 	}
 	withVertex(entryFn, attribs) {
-		return new WithVertexImpl(this._getRoot(), this._slotBindings, {
+		return new WithVertexImpl(this.#getRoot(), this.#slotBindings, {
 			vertex: entryFn,
 			attribs
 		});
 	}
 	pipe(transform) {
 		const newCfg = transform(new ConfigurableImpl([]));
-		return new WithBindingImpl(this._getRoot, [...this._slotBindings, ...newCfg.bindings]);
+		return new WithBindingImpl(this.#getRoot, [...this.#slotBindings, ...newCfg.bindings]);
 	}
 };
 var WithComputeImpl = class {
-	constructor(_root, _slotBindings, _entryFn) {
-		this._root = _root;
-		this._slotBindings = _slotBindings;
-		this._entryFn = _entryFn;
+	#root;
+	#slotBindings;
+	#entryFn;
+	constructor(root, slotBindings, entryFn) {
+		this.#root = root;
+		this.#slotBindings = slotBindings;
+		this.#entryFn = entryFn;
 	}
 	createPipeline() {
-		return INTERNAL_createComputePipeline(this._root, this._slotBindings, { compute: this._entryFn });
+		return INTERNAL_createComputePipeline(this.#root, this.#slotBindings, { compute: this.#entryFn });
 	}
 };
 var WithVertexImpl = class WithVertexImpl {
@@ -229,14 +233,18 @@ var WithFragmentImpl = class WithFragmentImpl {
 */
 var TgpuRootImpl = class extends WithBindingImpl {
 	"~unstable";
-	_unwrappedBindGroupLayouts = new WeakMemo((key) => key.unwrap(this));
-	_unwrappedBindGroups = new WeakMemo((key) => key.unwrap(this));
+	device;
+	nameRegistrySetting;
+	shaderGenerator;
+	#unwrappedBindGroupLayouts = new WeakMemo((key) => key.unwrap(this));
+	#unwrappedBindGroups = new WeakMemo((key) => key.unwrap(this));
+	#ownDevice;
 	[$internal];
-	constructor(device, nameRegistrySetting, _ownDevice, logOptions, shaderGenerator) {
+	constructor(device, nameRegistrySetting, ownDevice, logOptions, shaderGenerator) {
 		super(() => this, []);
 		this.device = device;
 		this.nameRegistrySetting = nameRegistrySetting;
-		this._ownDevice = _ownDevice;
+		this.#ownDevice = ownDevice;
 		this.shaderGenerator = shaderGenerator;
 		this["~unstable"] = this;
 		this[$internal] = { logOptions };
@@ -274,7 +282,7 @@ var TgpuRootImpl = class extends WithBindingImpl {
 	}
 	destroy() {
 		clearTextureUtilsCache(this.device);
-		if (this._ownDevice) this.device.destroy();
+		if (this.#ownDevice) this.device.destroy();
 	}
 	createTexture(props) {
 		return INTERNAL_createTexture(props, this);
@@ -288,8 +296,8 @@ var TgpuRootImpl = class extends WithBindingImpl {
 	unwrap(resource) {
 		if (isComputePipeline(resource)) return resource[$internal].rawPipeline;
 		if (isRenderPipeline(resource)) return resource[$internal].core.unwrap().pipeline;
-		if (isBindGroupLayout(resource)) return this._unwrappedBindGroupLayouts.getOrMake(resource);
-		if (isBindGroup(resource)) return this._unwrappedBindGroups.getOrMake(resource);
+		if (isBindGroupLayout(resource)) return this.#unwrappedBindGroupLayouts.getOrMake(resource);
+		if (isBindGroup(resource)) return this.#unwrappedBindGroups.getOrMake(resource);
 		if (isBuffer(resource)) return resource.buffer;
 		if (isTexture(resource)) return resource[$internal].unwrap();
 		if (isTextureView(resource)) {
@@ -452,6 +460,5 @@ function initFromDevice(options) {
 	const { device, unstable_names: names = "strict", unstable_logOptions } = options ?? {};
 	return new TgpuRootImpl(device, names, false, unstable_logOptions ?? {}, options?.shaderGenerator);
 }
-
 //#endregion
-export { init, initFromDevice };
+export { init, initFromDevice };

package/core/root/rootTypes.d.ts

@@ -24,7 +24,7 @@ import { ShaderGenerator } from "../../tgsl/shaderGenerator.js";
 import { TgpuBuffer, VertexFlag } from "../buffer/buffer.js";
 import { AnyWgslData, BaseData, U16, U32, Vec3u, Void, WgslArray, v4f } from "../../data/wgslTypes.js";
 import { TgpuTexture } from "../texture/texture.js";
-import { ExtractInvalidSchemaError, Infer, InferGPURecord, IsValidBufferSchema, IsValidStorageSchema, IsValidUniformSchema } from "../../shared/repr.js";
+import { ExtractInvalidSchemaError, InferGPURecord, InferInput, IsValidBufferSchema, IsValidStorageSchema, IsValidUniformSchema } from "../../shared/repr.js";
 import { AnyData, Disarray, UndecorateRecord } from "../../data/dataTypes.js";
 
 //#region src/core/root/rootTypes.d.ts
@@ -34,6 +34,12 @@ interface TgpuGuardedComputePipeline<TArgs extends number[] = number[]> extends
    * Analogous to `TgpuComputePipeline.with(bindGroup)`.
    */
   with(bindGroup: TgpuBindGroup): TgpuGuardedComputePipeline<TArgs>;
+  /**
+   * Returns a pipeline wrapper that encodes dispatches into the provided
+   * command encoder instead of submitting them immediately.
+   * Analogous to `TgpuComputePipeline.with(encoder)`.
+   */
+  with(encoder: GPUCommandEncoder): TgpuGuardedComputePipeline<TArgs>;
   /**
    * Dispatches the pipeline.
    * Unlike `TgpuComputePipeline.dispatchWorkgroups()`, this method takes in the
@@ -469,7 +475,9 @@ interface TgpuRoot extends Unwrapper, WithBinding {
    * @param typeSchema The type of data that this buffer will hold.
    * @param initial The initial value of the buffer. (optional)
    */
-  createBuffer<TData extends AnyData>(typeSchema: ValidateBufferSchema<TData>, initial?: Infer<NoInfer<TData>>): TgpuBuffer<TData>;
+  createBuffer<TData extends AnyData>(typeSchema: ValidateBufferSchema<TData>, initializer: (buffer: TgpuBuffer<NoInfer<TData>>) => void): TgpuBuffer<TData>;
+  createBuffer<TData extends AnyData>(typeSchema: ValidateBufferSchema<TData>, initial?: InferInput<NoInfer<TData>>): TgpuBuffer<TData>;
+  createBuffer<TData extends AnyData>(typeSchema: ValidateBufferSchema<TData>, initial?: ((buffer: TgpuBuffer<NoInfer<TData>>) => void) | InferInput<NoInfer<TData>>): TgpuBuffer<TData>;
   /**
    * Allocates memory on the GPU, allows passing data between host and shader.
    *
@@ -488,7 +496,9 @@ interface TgpuRoot extends Unwrapper, WithBinding {
    * @param typeSchema The type of data that this buffer will hold.
    * @param initial The initial value of the buffer. (optional)
    */
-  createUniform<TData extends AnyWgslData>(typeSchema: ValidateUniformSchema<TData>, initial?: Infer<NoInfer<TData>>): TgpuUniform<TData>;
+  createUniform<TData extends AnyWgslData>(typeSchema: ValidateUniformSchema<TData>, initializer: (buffer: TgpuBuffer<NoInfer<TData>>) => void): TgpuUniform<TData>;
+  createUniform<TData extends AnyWgslData>(typeSchema: ValidateUniformSchema<TData>, initial?: InferInput<NoInfer<TData>>): TgpuUniform<TData>;
+  createUniform<TData extends AnyWgslData>(typeSchema: ValidateUniformSchema<TData>, initial?: ((buffer: TgpuBuffer<NoInfer<TData>>) => void) | InferInput<NoInfer<TData>>): TgpuUniform<TData>;
   /**
    * Allocates memory on the GPU, allows passing data between host and shader.
    * Read-only on the GPU, optimized for small data. For a general-purpose buffer,
@@ -506,7 +516,9 @@ interface TgpuRoot extends Unwrapper, WithBinding {
    * @param typeSchema The type of data that this buffer will hold.
    * @param initial The initial value of the buffer. (optional)
    */
-  createMutable<TData extends AnyWgslData>(typeSchema: ValidateStorageSchema<TData>, initial?: Infer<NoInfer<TData>>): TgpuMutable<TData>;
+  createMutable<TData extends AnyWgslData>(typeSchema: ValidateStorageSchema<TData>, initializer: (buffer: TgpuBuffer<NoInfer<TData>>) => void): TgpuMutable<TData>;
+  createMutable<TData extends AnyWgslData>(typeSchema: ValidateStorageSchema<TData>, initial?: InferInput<NoInfer<TData>>): TgpuMutable<TData>;
+  createMutable<TData extends AnyWgslData>(typeSchema: ValidateStorageSchema<TData>, initial?: ((buffer: TgpuBuffer<NoInfer<TData>>) => void) | InferInput<NoInfer<TData>>): TgpuMutable<TData>;
   /**
    * Allocates memory on the GPU, allows passing data between host and shader.
    * Can be mutated in-place on the GPU. For a general-purpose buffer,
@@ -524,7 +536,9 @@ interface TgpuRoot extends Unwrapper, WithBinding {
    * @param typeSchema The type of data that this buffer will hold.
    * @param initial The initial value of the buffer. (optional)
    */
-  createReadonly<TData extends AnyWgslData>(typeSchema: ValidateStorageSchema<TData>, initial?: Infer<NoInfer<TData>>): TgpuReadonly<TData>;
+  createReadonly<TData extends AnyWgslData>(typeSchema: ValidateStorageSchema<TData>, initializer: (buffer: TgpuBuffer<NoInfer<TData>>) => void): TgpuReadonly<TData>;
+  createReadonly<TData extends AnyWgslData>(typeSchema: ValidateStorageSchema<TData>, initial?: InferInput<NoInfer<TData>>): TgpuReadonly<TData>;
+  createReadonly<TData extends AnyWgslData>(typeSchema: ValidateStorageSchema<TData>, initial?: ((buffer: TgpuBuffer<NoInfer<TData>>) => void) | InferInput<NoInfer<TData>>): TgpuReadonly<TData>;
   /**
    * Allocates memory on the GPU, allows passing data between host and shader.
    * Read-only on the GPU, optimized for large data. For a general-purpose buffer,
@@ -534,6 +548,9 @@ interface TgpuRoot extends Unwrapper, WithBinding {
    * @param gpuBuffer A vanilla WebGPU buffer.
    */
   createReadonly<TData extends AnyWgslData>(typeSchema: ValidateStorageSchema<TData>, gpuBuffer: GPUBuffer): TgpuReadonly<TData>;
+  createTexture<TWidth extends number, THeight extends number, TDepth extends number, TSize extends readonly [TWidth] | readonly [TWidth, THeight] | readonly [TWidth, THeight, TDepth], TFormat extends GPUTextureFormat, TMipLevelCount extends number, TSampleCount extends number, TViewFormats extends GPUTextureFormat[], TDimension extends GPUTextureDimension>(props: CreateTextureOptions<TSize, TFormat, TMipLevelCount, TSampleCount, TViewFormats, TDimension>): TgpuTexture<CreateTextureResult<TSize, TFormat, TMipLevelCount, TSampleCount, TViewFormats, TDimension>>;
+  createSampler(props: WgslSamplerProps): TgpuFixedSampler;
+  createComparisonSampler(props: WgslComparisonSamplerProps): TgpuFixedComparisonSampler;
   /**
    * Creates a query set for collecting timestamps or occlusion queries.
    *
@@ -586,6 +603,7 @@ interface TgpuRoot extends Unwrapper, WithBinding {
 interface ExperimentalTgpuRoot extends Omit<TgpuRoot, 'with'>, Withable_Deprecated<WithBinding> {
   readonly nameRegistrySetting: 'strict' | 'random';
   readonly shaderGenerator?: ShaderGenerator | undefined;
+  /** @deprecated Use `root.createTexture` instead. */
   createTexture<TWidth extends number, THeight extends number, TDepth extends number, TSize extends readonly [TWidth] | readonly [TWidth, THeight] | readonly [TWidth, THeight, TDepth], TFormat extends GPUTextureFormat, TMipLevelCount extends number, TSampleCount extends number, TViewFormats extends GPUTextureFormat[], TDimension extends GPUTextureDimension>(props: CreateTextureOptions<TSize, TFormat, TMipLevelCount, TSampleCount, TViewFormats, TDimension>): TgpuTexture<CreateTextureResult<TSize, TFormat, TMipLevelCount, TSampleCount, TViewFormats, TDimension>>;
   beginRenderPass(descriptor: GPURenderPassDescriptor, callback: (pass: RenderPass) => void): void;
   /**
@@ -601,7 +619,9 @@ interface ExperimentalTgpuRoot extends Omit<TgpuRoot, 'with'>, Withable_Deprecat
    * @param callback - A function that records draw commands into the bundle.
    */
   beginRenderBundleEncoder(descriptor: GPURenderBundleEncoderDescriptor, callback: (pass: RenderBundleEncoderPass) => void): GPURenderBundle;
+  /** @deprecated Use `root.createSampler` instead. */
   createSampler(props: WgslSamplerProps): TgpuFixedSampler;
+  /** @deprecated Use `root.createComparisonSampler` instead. */
   createComparisonSampler(props: WgslComparisonSamplerProps): TgpuFixedComparisonSampler;
   /**
    * @deprecated Used to cause all commands enqueued by pipelines to be

package/core/sampler/sampler.d.ts

@@ -1,10 +1,6 @@
 import { TgpuNamable } from "../../shared/meta.js";
 import { $gpuValueOf, $internal } from "../../shared/symbols.js";
 import { WgslComparisonSampler, WgslSampler } from "../../data/sampler.js";
-import "../../data/snippet.js";
-import "../../unwrapper.js";
-import "../../tgpuBindGroupLayout.js";
-import "../../types.js";
 import { Infer } from "../../shared/repr.js";
 
 //#region src/core/sampler/sampler.d.ts