wesl-tooling 0.6.2 → 0.6.4

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-import { VirtualLibraryFn, WeslBundle, WeslDevice } from "wesl";
+import { VirtualLibraryFn, WeslBundle } from "wesl";
 import { GPUElementFormat } from "thimbleberry";
 
 //#region src/ParseDependencies.d.ts
@@ -42,6 +42,18 @@ declare function dependencyBundles(weslSrc: Record<string, string>, projectDir:
 
 //#endregion
 //#region src/SimpleComputeShader.d.ts
+interface CompileShaderParams {
+  /** The project directory, used for resolving dependencies.  */
+  projectDir: string;
+  /** The GPUDevice to use for shader compilation. */
+  device: GPUDevice;
+  /** The WGSL/WESL shader source code. */
+  src: string;
+  /** Optional conditions for shader compilation. */
+  conditions?: Record<string, boolean>;
+  /** Optional virtual libraries to include in the shader. */
+  virtualLibs?: Record<string, VirtualLibraryFn>;
+}
 /**
  * Compiles a single WESL shader source string into a GPUShaderModule for testing
  * with automatic package detection.
@@ -51,11 +63,11 @@ declare function dependencyBundles(weslSrc: Record<string, string>, projectDir:
  * bundle to include in the link.
  *
  * @param projectDir - The project directory, used for resolving dependencies.
- * @param device - The WeslDevice to use for shader compilation.
+ * @param device - The GPUDevice to use for shader compilation.
  * @param src - The WESL shader source code.
  * @returns A Promise that resolves to the compiled GPUShaderModule.
  */
-declare function compileShader(projectDir: string, device: WeslDevice, src: string, virtualLibs?: Record<string, VirtualLibraryFn>): Promise<GPUShaderModule>;
+declare function compileShader(params: CompileShaderParams): Promise<GPUShaderModule>;
 /**
  * Transpiles and runs a simple compute shader on the GPU for testing.
  *
@@ -72,7 +84,7 @@ declare function compileShader(projectDir: string, device: WeslDevice, src: stri
  * @param resultFormat - format for interpreting the result buffer data. (default u32)
  * @returns storage result array (typically four numbers if the buffer format is u32 or f32)
  */
-declare function testComputeShader(projectDir: string, gpu: GPU, src: string, resultFormat?: GPUElementFormat): Promise<number[]>;
+declare function testComputeShader(projectDir: string, gpu: GPU, src: string, resultFormat?: GPUElementFormat, conditions?: Record<string, boolean>): Promise<number[]>;
 /**
  * Transpiles and runs a simple compute shader on the GPU for testing.
  *
@@ -92,4 +104,4 @@ declare function testComputeShader(projectDir: string, gpu: GPU, src: string, re
 declare function runSimpleComputePipeline(device: GPUDevice, module: GPUShaderModule, resultFormat?: GPUElementFormat): Promise<number[]>;
 
 //#endregion
-export { compileShader, dependencyBundles, parseDependencies, runSimpleComputePipeline, testComputeShader };
+export { CompileShaderParams, compileShader, dependencyBundles, parseDependencies, runSimpleComputePipeline, testComputeShader };

package/dist/index.js

@@ -95,17 +95,19 @@ const resultBufferSize = 16;
 * bundle to include in the link.
 *
 * @param projectDir - The project directory, used for resolving dependencies.
-* @param device - The WeslDevice to use for shader compilation.
+* @param device - The GPUDevice to use for shader compilation.
 * @param src - The WESL shader source code.
 * @returns A Promise that resolves to the compiled GPUShaderModule.
 */
-async function compileShader(projectDir, device, src, virtualLibs) {
+async function compileShader(params) {
+	const { projectDir, device, src, conditions, virtualLibs } = params;
 	const weslSrc = { main: src };
 	const libs = await dependencyBundles(weslSrc, projectDir);
 	const linked = await link({
 		weslSrc,
 		libs,
-		virtualLibs
+		virtualLibs,
+		conditions
 	});
 	return device.createShaderModule({ code: linked.dest });
 }
@@ -125,14 +127,21 @@ async function compileShader(projectDir, device, src, virtualLibs) {
 * @param resultFormat - format for interpreting the result buffer data. (default u32)
 * @returns storage result array (typically four numbers if the buffer format is u32 or f32)
 */
-async function testComputeShader(projectDir, gpu, src, resultFormat = "f32") {
+async function testComputeShader(projectDir, gpu, src, resultFormat = "f32", conditions = {}) {
 	const adapter = await gpu.requestAdapter();
 	const device = await requestWeslDevice(adapter);
 	try {
 		const arraySize = resultBufferSize / elementByteSize(resultFormat);
 		const arrayType = `array<${resultFormat}, ${arraySize}>`;
 		const virtualLibs = { test: () => `@group(0) @binding(0) var <storage, read_write> results: ${arrayType};` };
-		const module = await compileShader(projectDir, device, src, virtualLibs);
+		const params = {
+			projectDir,
+			device,
+			src,
+			conditions,
+			virtualLibs
+		};
+		const module = await compileShader(params);
 		const result = await runSimpleComputePipeline(device, module, resultFormat);
 		return result;
 	} finally {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wesl-tooling",
-  "version": "0.6.2",
+  "version": "0.6.4",
   "type": "module",
   "files": [
     "dist"
@@ -17,6 +17,9 @@
     "tsdown": "^0.11.3",
     "wesl": "0.6.2"
   },
+  "peerDependencies": {
+    "wesl": "^0.6.2"
+  },
   "scripts": {
     "echo": "echo",
     "build": "tsdown",