@luma.gl/shaders 0.0.0-stage → 10.0.0-alpha.10

package/dist/lib/shader-assembly/shader-plugin-varyings.cjs

@@ -0,0 +1,434 @@
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var shader_plugin_varyings_exports = {};
+__export(shader_plugin_varyings_exports, {
+  assembleShaderPluginInterStageVariablesGLSL: () => assembleShaderPluginInterStageVariablesGLSL,
+  assembleShaderPluginInterStageVariablesWGSL: () => assembleShaderPluginInterStageVariablesWGSL
+});
+module.exports = __toCommonJS(shader_plugin_varyings_exports);
+var import_types = require("../../types.cjs");
+var import_shader_plugin_vertex_inputs = require("./shader-plugin-vertex-inputs.cjs");
+var import_wgsl_binding_scan = require("./wgsl-binding-scan.cjs");
+function assembleShaderPluginInterStageVariablesGLSL(source, stage, interStageVariables) {
+  const declarations = [];
+  const initialization = [];
+  for (const [name, varying] of Object.entries(interStageVariables)) {
+    assertGLSLVaryingIsNew(source, name);
+    const qualifier = varying.interpolation === "flat" ? "flat " : "";
+    const direction = stage === "vertex" ? "out" : "in";
+    declarations.push(`${qualifier}${direction} ${(0, import_shader_plugin_vertex_inputs.getGLSLAttributeType)(varying.type)} ${name};`);
+    if (stage === "vertex") {
+      initialization.push(`${name} = ${getGLSLZeroValue(varying.type)};`);
+    }
+  }
+  return { declarations: declarations.join("\n"), initialization: initialization.join("\n") };
+}
+function assembleShaderPluginInterStageVariablesWGSL(source, vertexEntryPoint, fragmentEntryPoint, interStageVariables) {
+  const entries = Object.entries(interStageVariables);
+  if (entries.length === 0) {
+    return {
+      source,
+      declarations: "",
+      vertexInitialization: "",
+      fragmentInitialization: ""
+    };
+  }
+  let transformedSource = source;
+  let vertexFunction = getWGSLFunctionRange(transformedSource, vertexEntryPoint, "vertex");
+  const vertexOutputType = getWGSLNamedReturnType(transformedSource, vertexFunction);
+  let fragmentFunction = getWGSLFunctionRange(transformedSource, fragmentEntryPoint, "fragment");
+  const fragmentInput = getWGSLFragmentStructInput(transformedSource, fragmentFunction);
+  const vertexOutputStruct = getWGSLStructRange(transformedSource, vertexOutputType);
+  const fragmentInputStruct = fragmentInput ? getWGSLStructRange(transformedSource, fragmentInput.type) : null;
+  const interfaceNames = /* @__PURE__ */ new Set([
+    ...getWGSLDeclaredNames(vertexFunction.parameters),
+    ...getWGSLDeclaredNames(vertexOutputStruct.body),
+    ...getWGSLDeclaredNames(fragmentFunction.parameters),
+    ...getWGSLDeclaredNames((fragmentInputStruct == null ? void 0 : fragmentInputStruct.body) || "")
+  ]);
+  const usedLocations = /* @__PURE__ */ new Set([
+    ...getWGSLLocations(vertexOutputStruct.body),
+    ...getWGSLLocations((fragmentInputStruct == null ? void 0 : fragmentInputStruct.body) || ""),
+    ...getWGSLLocations(fragmentFunction.parameters)
+  ]);
+  const fields = [];
+  const fragmentParameters = [];
+  const declarations = [];
+  const vertexInitialization = [];
+  const fragmentInitialization = [];
+  for (const [name, varying] of entries) {
+    if (interfaceNames.has(name) || hasWGSLModuleVariable(transformedSource, name)) {
+      throw new Error(`ShaderPlugin varying "${name}" conflicts with existing WGSL stage I/O or a module variable`);
+    }
+    const location = getFirstUnusedLocation(usedLocations);
+    usedLocations.add(location);
+    const interpolation = varying.interpolation === "flat" ? " @interpolate(flat)" : "";
+    fields.push(`  @location(${location})${interpolation} ${name}: ${varying.type},`);
+    declarations.push(`var<private> ${name}: ${varying.type};`);
+    vertexInitialization.push(`${name} = ${getWGSLZeroValue(varying.type)};`);
+    if (fragmentInput) {
+      fragmentInitialization.push(`${name} = ${fragmentInput.name}.${name};`);
+    } else {
+      const parameterName = `_luma_${name}`;
+      if (interfaceNames.has(parameterName)) {
+        throw new Error(`ShaderPlugin varying "${name}" conflicts with a generated fragment parameter`);
+      }
+      fragmentParameters.push(`@location(${location})${interpolation} ${parameterName}: ${varying.type}`);
+      fragmentInitialization.push(`${name} = ${parameterName};`);
+    }
+  }
+  assertWGSLStructConstructorsAreLocal(transformedSource, vertexOutputType, vertexFunction.openBrace, vertexFunction.closeBrace);
+  transformedSource = appendWGSLConstructorArgumentsInFunction(transformedSource, vertexOutputType, vertexFunction, entries.map(([name]) => name));
+  vertexFunction = getWGSLFunctionRange(transformedSource, vertexEntryPoint, "vertex");
+  transformedSource = rewriteWGSLVertexReturns(transformedSource, vertexFunction, entries.map(([name]) => name));
+  const structNames = !fragmentInput || vertexOutputType === fragmentInput.type ? [vertexOutputType] : [vertexOutputType, fragmentInput.type];
+  const structInsertions = structNames.map((structName) => getWGSLStructRange(transformedSource, structName).closeBrace).sort((left, right) => right - left);
+  for (const insertionIndex of structInsertions) {
+    transformedSource = transformedSource.slice(0, insertionIndex) + `${fields.join("\n")}
+` + transformedSource.slice(insertionIndex);
+  }
+  fragmentFunction = getWGSLFunctionRange(transformedSource, fragmentEntryPoint, "fragment");
+  if (fragmentParameters.length) {
+    let parameters = fragmentFunction.parameters;
+    const parameterTokens = (0, import_wgsl_binding_scan.maskWGSLComments)(parameters).trimEnd();
+    if (parameterTokens.trim() && !parameterTokens.endsWith(",")) {
+      const insertionIndex = parameterTokens.length;
+      parameters = `${parameters.slice(0, insertionIndex)},${parameters.slice(insertionIndex)}`;
+    }
+    transformedSource = transformedSource.slice(0, fragmentFunction.openParenthesis + 1) + parameters + "\n  " + fragmentParameters.join(",\n  ") + "\n" + transformedSource.slice(fragmentFunction.closeParenthesis);
+  }
+  return {
+    source: transformedSource,
+    declarations: declarations.join("\n"),
+    vertexInitialization: vertexInitialization.join("\n"),
+    fragmentInitialization: fragmentInitialization.join("\n")
+  };
+}
+function getWGSLFunctionRange(source, entryPoint, stage) {
+  const functionPattern = new RegExp(`\\bfn\\s+${escapeRegExp(entryPoint)}\\s*\\(`, "g");
+  const match = functionPattern.exec(source);
+  if (!match) {
+    throw new Error(`ShaderPlugin interStageVariables require WGSL ${stage} entry point "${entryPoint}"`);
+  }
+  const openParenthesis = source.indexOf("(", match.index);
+  const closeParenthesis = findMatchingDelimiter(source, openParenthesis, "(", ")");
+  const openBrace = source.indexOf("{", closeParenthesis);
+  const closeBrace = findMatchingDelimiter(source, openBrace, "{", "}");
+  if (closeParenthesis < 0 || openBrace < 0 || closeBrace < 0) {
+    throw new Error(`Unable to parse WGSL ${stage} entry point "${entryPoint}"`);
+  }
+  return {
+    openParenthesis,
+    closeParenthesis,
+    openBrace,
+    closeBrace,
+    parameters: source.slice(openParenthesis + 1, closeParenthesis)
+  };
+}
+function getWGSLNamedReturnType(source, functionRange) {
+  const resultSource = source.slice(functionRange.closeParenthesis + 1, functionRange.openBrace);
+  const match = /->\s*([A-Za-z_][\w]*)\s*$/.exec(resultSource.trim());
+  if (!match || getWGSLStructRangeOrNull(source, match[1]) === null) {
+    throw new Error("ShaderPlugin interStageVariables require the WGSL vertex entry point to return a named struct");
+  }
+  return match[1];
+}
+function getWGSLFragmentStructInput(source, functionRange) {
+  const candidates = [];
+  for (const parameter of splitTopLevel((0, import_wgsl_binding_scan.maskWGSLComments)(functionRange.parameters), ",")) {
+    const match = /(?:@[A-Za-z_][\w]*(?:\([^)]*\))?\s*)*([A-Za-z_][\w]*)\s*:\s*([A-Za-z_][\w]*)\s*$/.exec(parameter.trim());
+    if (match && getWGSLStructRangeOrNull(source, match[2])) {
+      candidates.push({ name: match[1], type: match[2] });
+    }
+  }
+  if (candidates.length > 1) {
+    throw new Error(`ShaderPlugin interStageVariables require exactly one named WGSL fragment input struct; found ${candidates.length}`);
+  }
+  return candidates[0] || null;
+}
+function getWGSLStructRange(source, structName) {
+  const range = getWGSLStructRangeOrNull(source, structName);
+  if (!range) {
+    throw new Error(`Unable to find WGSL stage I/O struct "${structName}"`);
+  }
+  return range;
+}
+function getWGSLStructRangeOrNull(source, structName) {
+  const structPattern = new RegExp(`\\bstruct\\s+${escapeRegExp(structName)}\\s*\\{`, "g");
+  const match = structPattern.exec(source);
+  if (!match) {
+    return null;
+  }
+  const openBrace = source.indexOf("{", match.index);
+  const closeBrace = findMatchingDelimiter(source, openBrace, "{", "}");
+  return closeBrace < 0 ? null : { openBrace, closeBrace, body: source.slice(openBrace + 1, closeBrace) };
+}
+function assertWGSLStructConstructorsAreLocal(source, structName, functionStart, functionEnd) {
+  const constructorPattern = new RegExp(`\\b${escapeRegExp(structName)}\\s*\\(`, "g");
+  let match = constructorPattern.exec(source);
+  while (match) {
+    if (match.index < functionStart || match.index > functionEnd) {
+      throw new Error(`ShaderPlugin varying output struct "${structName}" is constructed outside the selected vertex entry point`);
+    }
+    match = constructorPattern.exec(source);
+  }
+}
+function appendWGSLConstructorArgumentsInFunction(source, structName, functionRange, varyingNames) {
+  const constructorPattern = new RegExp(`\\b${escapeRegExp(structName)}\\s*\\(`, "g");
+  const insertions = [];
+  let match = constructorPattern.exec(source);
+  while (match) {
+    if (match.index > functionRange.openBrace && match.index < functionRange.closeBrace) {
+      const openParenthesis = source.indexOf("(", match.index);
+      const closeParenthesis = findMatchingDelimiter(source, openParenthesis, "(", ")");
+      if (closeParenthesis < 0 || closeParenthesis > functionRange.closeBrace) {
+        throw new Error(`Unable to parse WGSL output constructor "${structName}"`);
+      }
+      insertions.push({ openParenthesis, closeParenthesis });
+    }
+    match = constructorPattern.exec(source);
+  }
+  for (const insertion of insertions.sort((left, right) => right.closeParenthesis - left.closeParenthesis)) {
+    const existingArguments = source.slice(insertion.openParenthesis + 1, insertion.closeParenthesis).trim();
+    const separator = existingArguments ? ", " : "";
+    source = source.slice(0, insertion.closeParenthesis) + separator + varyingNames.join(", ") + source.slice(insertion.closeParenthesis);
+  }
+  return source;
+}
+function rewriteWGSLVertexReturns(source, functionRange, varyingNames) {
+  const returns = findWGSLReturnStatements(source, functionRange.openBrace + 1, functionRange.closeBrace);
+  for (let returnIndex = returns.length - 1; returnIndex >= 0; returnIndex--) {
+    const statement = returns[returnIndex];
+    const expression = source.slice(statement.expressionStart, statement.semicolon).trim();
+    if (!expression) {
+      throw new Error("ShaderPlugin varying vertex entry point cannot use an empty return");
+    }
+    const outputName = `_luma_vertexOutput${returnIndex}`;
+    const assignments = varyingNames.map((name) => `${outputName}.${name} = ${name};`).join("\n");
+    const replacement = `{
+var ${outputName} = ${expression};
+${assignments}
+return ${outputName};
+}`;
+    source = source.slice(0, statement.start) + replacement + source.slice(statement.semicolon + 1);
+  }
+  return source;
+}
+function findWGSLReturnStatements(source, startIndex, endIndex) {
+  const statements = [];
+  let index = startIndex;
+  while (index < endIndex) {
+    index = skipWGSLTrivia(source, index, endIndex);
+    if (source.slice(index, index + 6) === "return" && !/[A-Za-z0-9_]/.test(source[index + 6] || "")) {
+      const expressionStart = index + 6;
+      const semicolon = findStatementSemicolon(source, expressionStart, endIndex);
+      if (semicolon < 0) {
+        throw new Error("Unable to parse WGSL return statement in selected vertex entry point");
+      }
+      statements.push({ start: index, expressionStart, semicolon });
+      index = semicolon + 1;
+    } else {
+      index++;
+    }
+  }
+  return statements;
+}
+function findStatementSemicolon(source, startIndex, endIndex) {
+  let parenthesisDepth = 0;
+  let bracketDepth = 0;
+  for (let index = startIndex; index < endIndex; index++) {
+    const skippedIndex = skipWGSLTrivia(source, index, endIndex);
+    if (skippedIndex !== index) {
+      index = skippedIndex - 1;
+      continue;
+    }
+    const character = source[index];
+    if (character === "(")
+      parenthesisDepth++;
+    if (character === ")")
+      parenthesisDepth--;
+    if (character === "[")
+      bracketDepth++;
+    if (character === "]")
+      bracketDepth--;
+    if (character === ";" && parenthesisDepth === 0 && bracketDepth === 0)
+      return index;
+  }
+  return -1;
+}
+function skipWGSLTrivia(source, startIndex, endIndex) {
+  let index = startIndex;
+  if (source[index] === "/" && source[index + 1] === "/") {
+    const newline = source.indexOf("\n", index + 2);
+    return newline < 0 || newline > endIndex ? endIndex : newline + 1;
+  }
+  if (source[index] === "/" && source[index + 1] === "*") {
+    let depth = 1;
+    index += 2;
+    while (index < endIndex && depth > 0) {
+      if (source[index] === "/" && source[index + 1] === "*") {
+        depth++;
+        index += 2;
+      } else if (source[index] === "*" && source[index + 1] === "/") {
+        depth--;
+        index += 2;
+      } else {
+        index++;
+      }
+    }
+  }
+  return index;
+}
+function splitTopLevel(source, delimiter) {
+  const parts = [];
+  let startIndex = 0;
+  let parenthesisDepth = 0;
+  let angleDepth = 0;
+  for (let index = 0; index < source.length; index++) {
+    const character = source[index];
+    if (character === "(")
+      parenthesisDepth++;
+    if (character === ")")
+      parenthesisDepth--;
+    if (character === "<")
+      angleDepth++;
+    if (character === ">")
+      angleDepth--;
+    if (character === delimiter && parenthesisDepth === 0 && angleDepth === 0) {
+      parts.push(source.slice(startIndex, index));
+      startIndex = index + 1;
+    }
+  }
+  parts.push(source.slice(startIndex));
+  return parts;
+}
+function getWGSLLocations(source) {
+  const locations = [];
+  const locationPattern = /@location\s*\(\s*(\d+)\s*\)/g;
+  let match = locationPattern.exec(source);
+  while (match) {
+    locations.push(Number(match[1]));
+    match = locationPattern.exec(source);
+  }
+  return locations;
+}
+function getWGSLDeclaredNames(source) {
+  const names = [];
+  const namePattern = /(?:^|,)\s*(?:@[A-Za-z_][\w]*(?:\([^)]*\))?\s*)*([A-Za-z_][\w]*)\s*:/gm;
+  let match = namePattern.exec(source);
+  while (match) {
+    names.push(match[1]);
+    match = namePattern.exec(source);
+  }
+  return names;
+}
+function hasWGSLModuleVariable(source, name) {
+  const variablePattern = new RegExp(`\\b(?:var(?:<[^>]+>)?|let|const)\\s+${escapeRegExp(name)}\\b`, "g");
+  let match = variablePattern.exec(source);
+  while (match) {
+    if (getBraceDepthAt(source, match.index) === 0) {
+      return true;
+    }
+    match = variablePattern.exec(source);
+  }
+  return false;
+}
+function getFirstUnusedLocation(usedLocations) {
+  let location = 0;
+  while (usedLocations.has(location))
+    location++;
+  return location;
+}
+function getGLSLZeroValue(type) {
+  const { primitiveType, components } = import_types.shaderTypeDecoder.getAttributeShaderTypeInfo(type);
+  const scalar = primitiveType === "u32" ? "0u" : primitiveType === "i32" ? "0" : "0.0";
+  return components === 1 ? scalar : `${(0, import_shader_plugin_vertex_inputs.getGLSLAttributeType)(type)}(${scalar})`;
+}
+function getWGSLZeroValue(type) {
+  const { primitiveType, components } = import_types.shaderTypeDecoder.getAttributeShaderTypeInfo(type);
+  const scalar = `${primitiveType}(0)`;
+  return components === 1 ? scalar : `${type}(${scalar})`;
+}
+function assertGLSLVaryingIsNew(source, name) {
+  const declaration = new RegExp(`\\b(?:flat\\s+|smooth\\s+)?(?:in|out|varying)\\s+(?:(?:lowp|mediump|highp)\\s+)?[A-Za-z_][A-Za-z0-9_]*\\s+${escapeRegExp(name)}\\s*(?:\\[|;)`);
+  if (declaration.test(source)) {
+    throw new Error(`ShaderPlugin varying "${name}" conflicts with existing GLSL stage I/O`);
+  }
+}
+function findMatchingDelimiter(source, openIndex, openCharacter, closeCharacter) {
+  let depth = 0;
+  let blockCommentDepth = 0;
+  let lineComment = false;
+  for (let index = openIndex; index < source.length; index++) {
+    const character = source[index];
+    const nextCharacter = source[index + 1];
+    if (lineComment) {
+      if (character === "\n")
+        lineComment = false;
+      continue;
+    }
+    if (blockCommentDepth > 0) {
+      if (character === "/" && nextCharacter === "*") {
+        blockCommentDepth++;
+        index++;
+      } else if (character === "*" && nextCharacter === "/") {
+        blockCommentDepth--;
+        index++;
+      }
+      continue;
+    }
+    if (character === "/" && nextCharacter === "/") {
+      lineComment = true;
+      index++;
+      continue;
+    }
+    if (character === "/" && nextCharacter === "*") {
+      blockCommentDepth = 1;
+      index++;
+      continue;
+    }
+    if (character === openCharacter)
+      depth++;
+    if (character === closeCharacter && --depth === 0)
+      return index;
+  }
+  return -1;
+}
+function getBraceDepthAt(source, endIndex) {
+  let depth = 0;
+  for (let index = 0; index < endIndex; index++) {
+    const skippedIndex = skipWGSLTrivia(source, index, endIndex);
+    if (skippedIndex !== index) {
+      index = skippedIndex - 1;
+      continue;
+    }
+    if (source[index] === "{")
+      depth++;
+    if (source[index] === "}")
+      depth--;
+  }
+  return depth;
+}
+function escapeRegExp(value) {
+  return value.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  assembleShaderPluginInterStageVariablesGLSL,
+  assembleShaderPluginInterStageVariablesWGSL
+});

package/dist/lib/shader-assembly/shader-plugin-varyings.d.ts

@@ -0,0 +1,15 @@
+import type { ResolvedShaderPluginInterStageVariable } from "../shader-plugin.js";
+export type ShaderPluginInterStageVariableAssembly = {
+    source: string;
+    declarations: string;
+    vertexInitialization: string;
+    fragmentInitialization: string;
+};
+/** Generate one stage of the GLSL interface contributed by shader plugins. */
+export declare function assembleShaderPluginInterStageVariablesGLSL(source: string, stage: 'vertex' | 'fragment', interStageVariables: Record<string, ResolvedShaderPluginInterStageVariable>): {
+    declarations: string;
+    initialization: string;
+};
+/** Add plugin interStageVariables to selected WGSL vertex and fragment entry points. */
+export declare function assembleShaderPluginInterStageVariablesWGSL(source: string, vertexEntryPoint: string, fragmentEntryPoint: string, interStageVariables: Record<string, ResolvedShaderPluginInterStageVariable>): ShaderPluginInterStageVariableAssembly;
+//# sourceMappingURL=shader-plugin-varyings.d.ts.map

package/dist/lib/shader-assembly/shader-plugin-varyings.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"shader-plugin-varyings.d.ts","sourceRoot":"","sources":["../../../src/lib/shader-assembly/shader-plugin-varyings.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAC,sCAAsC,EAAC,4BAAyB;AAI7E,MAAM,MAAM,sCAAsC,GAAG;IACnD,MAAM,EAAE,MAAM,CAAC;IACf,YAAY,EAAE,MAAM,CAAC;IACrB,oBAAoB,EAAE,MAAM,CAAC;IAC7B,sBAAsB,EAAE,MAAM,CAAC;CAChC,CAAC;AAEF,8EAA8E;AAC9E,wBAAgB,2CAA2C,CACzD,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,QAAQ,GAAG,UAAU,EAC5B,mBAAmB,EAAE,MAAM,CAAC,MAAM,EAAE,sCAAsC,CAAC,GAC1E;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,cAAc,EAAE,MAAM,CAAA;CAAC,CAahD;AAED,wFAAwF;AACxF,wBAAgB,2CAA2C,CACzD,MAAM,EAAE,MAAM,EACd,gBAAgB,EAAE,MAAM,EACxB,kBAAkB,EAAE,MAAM,EAC1B,mBAAmB,EAAE,MAAM,CAAC,MAAM,EAAE,sCAAsC,CAAC,GAC1E,sCAAsC,CA0HxC"}