@lightningjs/renderer 0.7.1 → 0.7.3

package/src/core/renderers/webgl/WebGlCoreShader.ts

@@ -1,337 +1,337 @@
-/*
- * If not stated otherwise in this file or this component's LICENSE file the
- * following copyright and licenses apply:
- *
- * Copyright 2023 Comcast Cable Communications Management, LLC.
- *
- * Licensed under the Apache License, Version 2.0 (the License);
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-import type { Dimensions } from '../../../common/CommonTypes.js';
-import { assertTruthy, hasOwn } from '../../../utils.js';
-import type { WebGlContextWrapper } from '../../lib/WebGlContextWrapper.js';
-import { CoreShader } from '../CoreShader.js';
-import type { WebGlCoreCtxTexture } from './WebGlCoreCtxTexture.js';
-import type { WebGlCoreRenderOp } from './WebGlCoreRenderOp.js';
-import type { WebGlCoreRenderer } from './WebGlCoreRenderer.js';
-import type { BufferCollection } from './internal/BufferCollection.js';
-import {
-  createProgram,
-  createShader,
-  type AttributeInfo,
-  type ShaderOptions,
-  type UniformInfo,
-  type UniformMethodMap,
-  type ShaderProgramSources,
-} from './internal/ShaderUtils.js';
-import { isWebGl2 } from './internal/WebGlUtils.js';
-
-/**
- * Automatic shader prop for the dimensions of the Node being rendered
- *
- * @remarks
- * Shader's who's rendering depends on the dimensions of the Node being rendered
- * should extend this interface from their Prop interface type.
- */
-export interface DimensionsShaderProp {
-  /**
-   * Dimensions of the Node being rendered (Auto-set by the renderer)
-   *
-   * @remarks
-   * DO NOT SET THIS. It is set automatically by the renderer.
-   * Any values set here will be ignored.
-   */
-  $dimensions?: Dimensions;
-}
-
-export interface AlphaShaderProp {
-  /**
-   * Alpha of the Node being rendered (Auto-set by the renderer)
-   *
-   * @remarks
-   * DO NOT SET THIS. It is set automatically by the renderer.
-   * Any values set here will be ignored.
-   */
-  $alpha?: number;
-}
-
-export abstract class WebGlCoreShader extends CoreShader {
-  protected boundBufferCollection: BufferCollection | null = null;
-  protected buffersBound = false;
-  protected program: WebGLProgram;
-  /**
-   * Vertex Array Object
-   *
-   * @remarks
-   * Used by WebGL2 Only
-   */
-  protected vao: WebGLVertexArrayObject | undefined;
-  protected renderer: WebGlCoreRenderer;
-  protected glw: WebGlContextWrapper;
-  protected attributeBuffers: Record<string, WebGLBuffer>;
-  protected attributeLocations: Record<string, number>;
-  protected attributeNames: string[];
-  protected uniformLocations: Record<string, WebGLUniformLocation>;
-  protected uniformTypes: Record<string, keyof UniformMethodMap>;
-  readonly supportsIndexedTextures: boolean;
-
-  constructor(options: ShaderOptions) {
-    super();
-    const renderer = (this.renderer = options.renderer);
-    const glw = (this.glw = this.renderer.glw);
-    this.supportsIndexedTextures = options.supportsIndexedTextures || false;
-
-    // Check that extensions are supported
-    const webGl2 = glw.isWebGl2();
-    const requiredExtensions =
-      (webGl2 && options.webgl2Extensions) ||
-      (!webGl2 && options.webgl1Extensions) ||
-      [];
-    const glVersion = webGl2 ? '2.0' : '1.0';
-    requiredExtensions.forEach((extensionName) => {
-      if (!glw.getExtension(extensionName)) {
-        throw new Error(
-          `Shader "${this.constructor.name}" requires extension "${extensionName}" for WebGL ${glVersion} but wasn't found`,
-        );
-      }
-    });
-
-    // Gather shader sources
-    // - If WebGL 2 and special WebGL 2 sources are provided, we copy those sources and delete
-    // the extra copy of them to save memory.
-    // TODO: This could be further made optimal by just caching the compiled shaders and completely deleting
-    // the source code
-    const shaderSources =
-      options.shaderSources ||
-      (this.constructor as typeof WebGlCoreShader).shaderSources;
-    if (!shaderSources) {
-      throw new Error(
-        `Shader "${this.constructor.name}" is missing shaderSources.`,
-      );
-    } else if (webGl2 && shaderSources?.webGl2) {
-      shaderSources.fragment = shaderSources.webGl2.fragment;
-      shaderSources.vertex = shaderSources.webGl2.vertex;
-      delete shaderSources.webGl2;
-    }
-
-    const textureUnits =
-      renderer.system.parameters.MAX_VERTEX_TEXTURE_IMAGE_UNITS;
-
-    const vertexSource =
-      shaderSources.vertex instanceof Function
-        ? shaderSources.vertex(textureUnits)
-        : shaderSources.vertex;
-
-    const fragmentSource =
-      shaderSources.fragment instanceof Function
-        ? shaderSources.fragment(textureUnits)
-        : shaderSources.fragment;
-
-    const vertexShader = createShader(glw, glw.VERTEX_SHADER, vertexSource);
-    const fragmentShader = createShader(
-      glw,
-      glw.FRAGMENT_SHADER,
-      fragmentSource,
-    );
-    if (!vertexShader || !fragmentShader) {
-      throw new Error();
-    }
-
-    const program = createProgram(glw, vertexShader, fragmentShader);
-    if (!program) {
-      throw new Error();
-    }
-    this.program = program;
-
-    if (webGl2) {
-      const vao = glw.createVertexArray();
-      if (!vao) {
-        throw new Error();
-      }
-      this.vao = vao;
-
-      glw.bindVertexArray(this.vao);
-    }
-
-    this.attributeLocations = {} as Record<string, number>;
-    this.attributeBuffers = {} as Record<string, number>;
-    this.attributeNames = [];
-
-    [...options.attributes].forEach((attributeName) => {
-      const location = glw.getAttribLocation(this.program, attributeName);
-      if (location < 0) {
-        throw new Error(
-          `${this.constructor.name}: Vertex shader must have an attribute "${attributeName}"!`,
-        );
-      }
-      const buffer = glw.createBuffer();
-      if (!buffer) {
-        throw new Error(
-          `${this.constructor.name}: Could not create buffer for attribute "${attributeName}"`,
-        );
-      }
-
-      this.attributeLocations[attributeName] = location;
-      this.attributeBuffers[attributeName] = buffer;
-      this.attributeNames.push(attributeName);
-    });
-
-    this.uniformLocations = {} as Record<string, WebGLRenderingContext>;
-    this.uniformTypes = {} as Record<string, keyof UniformMethodMap>;
-    options.uniforms.forEach((uniform: UniformInfo) => {
-      const location = glw.getUniformLocation(this.program, uniform.name);
-      this.uniformTypes[uniform.name] = uniform.uniform;
-      if (!location) {
-        console.warn(
-          `Shader "${this.constructor.name}" could not get uniform location for "${uniform.name}"`,
-        );
-        return;
-      }
-      this.uniformLocations[uniform.name] = location;
-    });
-  }
-
-  private bindBufferAttribute(
-    location: number,
-    buffer: WebGLBuffer,
-    attribute: AttributeInfo,
-  ) {
-    const { glw } = this;
-    glw.enableVertexAttribArray(location);
-
-    glw.vertexAttribPointer(
-      buffer,
-      location,
-      attribute.size,
-      attribute.type,
-      attribute.normalized,
-      attribute.stride,
-      attribute.offset,
-    );
-  }
-
-  disableAttribute(location: number) {
-    this.glw.disableVertexAttribArray(location);
-  }
-
-  disableAttributes() {
-    for (const loc in this.attributeLocations) {
-      this.disableAttribute(this.attributeLocations[loc] as number);
-    }
-    this.boundBufferCollection = null;
-  }
-
-  /**
-   * Given two sets of Shader props destined for this Shader, determine if they can be batched together
-   * to reduce the number of draw calls.
-   *
-   * @remarks
-   * This is used by the {@link WebGlCoreRenderer} to determine if it can batch multiple consecutive draw
-   * calls into a single draw call.
-   *
-   * By default, this returns false (meaning no batching is allowed), but can be
-   * overridden by child classes to provide more efficient batching.
-   *
-   * @param propsA
-   * @param propsB
-   * @returns
-   */
-  canBatchShaderProps(
-    propsA: Record<string, unknown>,
-    propsB: Record<string, unknown>,
-  ): boolean {
-    return false;
-  }
-
-  bindRenderOp(
-    renderOp: WebGlCoreRenderOp,
-    props: Record<string, unknown> | null,
-  ) {
-    this.bindBufferCollection(renderOp.buffers);
-    if (renderOp.textures.length > 0) {
-      this.bindTextures(renderOp.textures);
-    }
-    const { glw } = renderOp;
-    // Bind standard automatic uniforms
-    this.setUniform('u_resolution', [glw.canvas.width, glw.canvas.height]);
-    this.setUniform('u_pixelRatio', renderOp.options.pixelRatio);
-    if (props) {
-      // Bind optional automatic uniforms
-      // These are only bound if their keys are present in the props.
-      if (hasOwn(props, '$dimensions')) {
-        let dimensions = props.$dimensions as Dimensions | null;
-        if (!dimensions) {
-          dimensions = renderOp.dimensions;
-        }
-        this.setUniform('u_dimensions', [dimensions.width, dimensions.height]);
-      }
-      if (hasOwn(props, '$alpha')) {
-        let alpha = props.$alpha as number | null;
-        if (!alpha) {
-          alpha = renderOp.alpha;
-        }
-        this.setUniform('u_alpha', alpha);
-      }
-      this.bindProps(props);
-    }
-  }
-
-  setUniform(name: string, ...value: any[]): void {
-    // eslint-disable-next-line @typescript-eslint/no-non-null-assertion, @typescript-eslint/no-unsafe-argument
-    this.glw.setUniform(
-      this.uniformTypes[name]!,
-      this.uniformLocations[name]!,
-      ...(value as any),
-    );
-  }
-
-  bindBufferCollection(buffer: BufferCollection) {
-    if (this.boundBufferCollection === buffer) {
-      return;
-    }
-    for (const attributeName in this.attributeLocations) {
-      const resolvedBuffer = buffer.getBuffer(attributeName);
-      const resolvedInfo = buffer.getAttributeInfo(attributeName);
-      assertTruthy(resolvedBuffer, `Buffer for "${attributeName}" not found`);
-      assertTruthy(resolvedInfo);
-      this.bindBufferAttribute(
-        this.attributeLocations[attributeName]!,
-        resolvedBuffer,
-        resolvedInfo,
-      );
-    }
-    this.boundBufferCollection = buffer;
-  }
-
-  protected override bindProps(props: Record<string, unknown>) {
-    // Implement in child class
-  }
-
-  bindTextures(textures: WebGlCoreCtxTexture[]) {
-    // no defaults
-  }
-
-  override attach(): void {
-    this.glw.useProgram(this.program);
-    this.glw.useProgram(this.program);
-    if (this.glw.isWebGl2() && this.vao) {
-      this.glw.bindVertexArray(this.vao);
-    }
-  }
-
-  override detach(): void {
-    this.disableAttributes();
-  }
-
-  protected static shaderSources?: ShaderProgramSources;
-}
+/*
+ * If not stated otherwise in this file or this component's LICENSE file the
+ * following copyright and licenses apply:
+ *
+ * Copyright 2023 Comcast Cable Communications Management, LLC.
+ *
+ * Licensed under the Apache License, Version 2.0 (the License);
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import type { Dimensions } from '../../../common/CommonTypes.js';
+import { assertTruthy, hasOwn } from '../../../utils.js';
+import type { WebGlContextWrapper } from '../../lib/WebGlContextWrapper.js';
+import { CoreShader } from '../CoreShader.js';
+import type { WebGlCoreCtxTexture } from './WebGlCoreCtxTexture.js';
+import type { WebGlCoreRenderOp } from './WebGlCoreRenderOp.js';
+import type { WebGlCoreRenderer } from './WebGlCoreRenderer.js';
+import type { BufferCollection } from './internal/BufferCollection.js';
+import {
+  createProgram,
+  createShader,
+  type AttributeInfo,
+  type ShaderOptions,
+  type UniformInfo,
+  type UniformMethodMap,
+  type ShaderProgramSources,
+} from './internal/ShaderUtils.js';
+import { isWebGl2 } from './internal/WebGlUtils.js';
+
+/**
+ * Automatic shader prop for the dimensions of the Node being rendered
+ *
+ * @remarks
+ * Shader's who's rendering depends on the dimensions of the Node being rendered
+ * should extend this interface from their Prop interface type.
+ */
+export interface DimensionsShaderProp {
+  /**
+   * Dimensions of the Node being rendered (Auto-set by the renderer)
+   *
+   * @remarks
+   * DO NOT SET THIS. It is set automatically by the renderer.
+   * Any values set here will be ignored.
+   */
+  $dimensions?: Dimensions;
+}
+
+export interface AlphaShaderProp {
+  /**
+   * Alpha of the Node being rendered (Auto-set by the renderer)
+   *
+   * @remarks
+   * DO NOT SET THIS. It is set automatically by the renderer.
+   * Any values set here will be ignored.
+   */
+  $alpha?: number;
+}
+
+export abstract class WebGlCoreShader extends CoreShader {
+  protected boundBufferCollection: BufferCollection | null = null;
+  protected buffersBound = false;
+  protected program: WebGLProgram;
+  /**
+   * Vertex Array Object
+   *
+   * @remarks
+   * Used by WebGL2 Only
+   */
+  protected vao: WebGLVertexArrayObject | undefined;
+  protected renderer: WebGlCoreRenderer;
+  protected glw: WebGlContextWrapper;
+  protected attributeBuffers: Record<string, WebGLBuffer>;
+  protected attributeLocations: Record<string, number>;
+  protected attributeNames: string[];
+  protected uniformLocations: Record<string, WebGLUniformLocation>;
+  protected uniformTypes: Record<string, keyof UniformMethodMap>;
+  readonly supportsIndexedTextures: boolean;
+
+  constructor(options: ShaderOptions) {
+    super();
+    const renderer = (this.renderer = options.renderer);
+    const glw = (this.glw = this.renderer.glw);
+    this.supportsIndexedTextures = options.supportsIndexedTextures || false;
+
+    // Check that extensions are supported
+    const webGl2 = glw.isWebGl2();
+    const requiredExtensions =
+      (webGl2 && options.webgl2Extensions) ||
+      (!webGl2 && options.webgl1Extensions) ||
+      [];
+    const glVersion = webGl2 ? '2.0' : '1.0';
+    requiredExtensions.forEach((extensionName) => {
+      if (!glw.getExtension(extensionName)) {
+        throw new Error(
+          `Shader "${this.constructor.name}" requires extension "${extensionName}" for WebGL ${glVersion} but wasn't found`,
+        );
+      }
+    });
+
+    // Gather shader sources
+    // - If WebGL 2 and special WebGL 2 sources are provided, we copy those sources and delete
+    // the extra copy of them to save memory.
+    // TODO: This could be further made optimal by just caching the compiled shaders and completely deleting
+    // the source code
+    const shaderSources =
+      options.shaderSources ||
+      (this.constructor as typeof WebGlCoreShader).shaderSources;
+    if (!shaderSources) {
+      throw new Error(
+        `Shader "${this.constructor.name}" is missing shaderSources.`,
+      );
+    } else if (webGl2 && shaderSources?.webGl2) {
+      shaderSources.fragment = shaderSources.webGl2.fragment;
+      shaderSources.vertex = shaderSources.webGl2.vertex;
+      delete shaderSources.webGl2;
+    }
+
+    const textureUnits =
+      renderer.system.parameters.MAX_VERTEX_TEXTURE_IMAGE_UNITS;
+
+    const vertexSource =
+      shaderSources.vertex instanceof Function
+        ? shaderSources.vertex(textureUnits)
+        : shaderSources.vertex;
+
+    const fragmentSource =
+      shaderSources.fragment instanceof Function
+        ? shaderSources.fragment(textureUnits)
+        : shaderSources.fragment;
+
+    const vertexShader = createShader(glw, glw.VERTEX_SHADER, vertexSource);
+    const fragmentShader = createShader(
+      glw,
+      glw.FRAGMENT_SHADER,
+      fragmentSource,
+    );
+    if (!vertexShader || !fragmentShader) {
+      throw new Error();
+    }
+
+    const program = createProgram(glw, vertexShader, fragmentShader);
+    if (!program) {
+      throw new Error();
+    }
+    this.program = program;
+
+    if (webGl2) {
+      const vao = glw.createVertexArray();
+      if (!vao) {
+        throw new Error();
+      }
+      this.vao = vao;
+
+      glw.bindVertexArray(this.vao);
+    }
+
+    this.attributeLocations = {} as Record<string, number>;
+    this.attributeBuffers = {} as Record<string, number>;
+    this.attributeNames = [];
+
+    [...options.attributes].forEach((attributeName) => {
+      const location = glw.getAttribLocation(this.program, attributeName);
+      if (location < 0) {
+        throw new Error(
+          `${this.constructor.name}: Vertex shader must have an attribute "${attributeName}"!`,
+        );
+      }
+      const buffer = glw.createBuffer();
+      if (!buffer) {
+        throw new Error(
+          `${this.constructor.name}: Could not create buffer for attribute "${attributeName}"`,
+        );
+      }
+
+      this.attributeLocations[attributeName] = location;
+      this.attributeBuffers[attributeName] = buffer;
+      this.attributeNames.push(attributeName);
+    });
+
+    this.uniformLocations = {} as Record<string, WebGLRenderingContext>;
+    this.uniformTypes = {} as Record<string, keyof UniformMethodMap>;
+    options.uniforms.forEach((uniform: UniformInfo) => {
+      const location = glw.getUniformLocation(this.program, uniform.name);
+      this.uniformTypes[uniform.name] = uniform.uniform;
+      if (!location) {
+        console.warn(
+          `Shader "${this.constructor.name}" could not get uniform location for "${uniform.name}"`,
+        );
+        return;
+      }
+      this.uniformLocations[uniform.name] = location;
+    });
+  }
+
+  private bindBufferAttribute(
+    location: number,
+    buffer: WebGLBuffer,
+    attribute: AttributeInfo,
+  ) {
+    const { glw } = this;
+    glw.enableVertexAttribArray(location);
+
+    glw.vertexAttribPointer(
+      buffer,
+      location,
+      attribute.size,
+      attribute.type,
+      attribute.normalized,
+      attribute.stride,
+      attribute.offset,
+    );
+  }
+
+  disableAttribute(location: number) {
+    this.glw.disableVertexAttribArray(location);
+  }
+
+  disableAttributes() {
+    for (const loc in this.attributeLocations) {
+      this.disableAttribute(this.attributeLocations[loc] as number);
+    }
+    this.boundBufferCollection = null;
+  }
+
+  /**
+   * Given two sets of Shader props destined for this Shader, determine if they can be batched together
+   * to reduce the number of draw calls.
+   *
+   * @remarks
+   * This is used by the {@link WebGlCoreRenderer} to determine if it can batch multiple consecutive draw
+   * calls into a single draw call.
+   *
+   * By default, this returns false (meaning no batching is allowed), but can be
+   * overridden by child classes to provide more efficient batching.
+   *
+   * @param propsA
+   * @param propsB
+   * @returns
+   */
+  canBatchShaderProps(
+    propsA: Record<string, unknown>,
+    propsB: Record<string, unknown>,
+  ): boolean {
+    return false;
+  }
+
+  bindRenderOp(
+    renderOp: WebGlCoreRenderOp,
+    props: Record<string, unknown> | null,
+  ) {
+    this.bindBufferCollection(renderOp.buffers);
+    if (renderOp.textures.length > 0) {
+      this.bindTextures(renderOp.textures);
+    }
+    const { glw } = renderOp;
+    // Bind standard automatic uniforms
+    this.setUniform('u_resolution', [glw.canvas.width, glw.canvas.height]);
+    this.setUniform('u_pixelRatio', renderOp.options.pixelRatio);
+    if (props) {
+      // Bind optional automatic uniforms
+      // These are only bound if their keys are present in the props.
+      if (hasOwn(props, '$dimensions')) {
+        let dimensions = props.$dimensions as Dimensions | null;
+        if (!dimensions) {
+          dimensions = renderOp.dimensions;
+        }
+        this.setUniform('u_dimensions', [dimensions.width, dimensions.height]);
+      }
+      if (hasOwn(props, '$alpha')) {
+        let alpha = props.$alpha as number | null;
+        if (!alpha) {
+          alpha = renderOp.alpha;
+        }
+        this.setUniform('u_alpha', alpha);
+      }
+      this.bindProps(props);
+    }
+  }
+
+  setUniform(name: string, ...value: any[]): void {
+    // eslint-disable-next-line @typescript-eslint/no-non-null-assertion, @typescript-eslint/no-unsafe-argument
+    this.glw.setUniform(
+      this.uniformTypes[name]!,
+      this.uniformLocations[name]!,
+      ...(value as any),
+    );
+  }
+
+  bindBufferCollection(buffer: BufferCollection) {
+    if (this.boundBufferCollection === buffer) {
+      return;
+    }
+    for (const attributeName in this.attributeLocations) {
+      const resolvedBuffer = buffer.getBuffer(attributeName);
+      const resolvedInfo = buffer.getAttributeInfo(attributeName);
+      assertTruthy(resolvedBuffer, `Buffer for "${attributeName}" not found`);
+      assertTruthy(resolvedInfo);
+      this.bindBufferAttribute(
+        this.attributeLocations[attributeName]!,
+        resolvedBuffer,
+        resolvedInfo,
+      );
+    }
+    this.boundBufferCollection = buffer;
+  }
+
+  protected override bindProps(props: Record<string, unknown>) {
+    // Implement in child class
+  }
+
+  bindTextures(textures: WebGlCoreCtxTexture[]) {
+    // no defaults
+  }
+
+  override attach(): void {
+    this.glw.useProgram(this.program);
+    this.glw.useProgram(this.program);
+    if (this.glw.isWebGl2() && this.vao) {
+      this.glw.bindVertexArray(this.vao);
+    }
+  }
+
+  override detach(): void {
+    this.disableAttributes();
+  }
+
+  protected static shaderSources?: ShaderProgramSources;
+}