@livekit/track-processors 0.5.3 → 0.5.4

package/src/webgl/shader-programs/boxBlurShader.ts

@@ -0,0 +1,98 @@
+import { glsl } from '../utils';
+import { vertexShaderSource } from './vertexShader';
+
+export const boxBlurFragmentShader = glsl`#version 300 es
+precision mediump float;
+
+in vec2 texCoords;
+
+uniform sampler2D u_texture;
+uniform vec2 u_texelSize;    // 1.0 / texture size
+uniform vec2 u_direction;    // (1.0, 0.0) for horizontal, (0.0, 1.0) for vertical
+uniform float u_radius;      // blur radius in texels
+
+out vec4 fragColor;
+
+void main() {
+    vec3 sum = vec3(0.0);
+    float count = 0.0;
+
+    // Limit radius to avoid excessive loop cost
+    const int MAX_RADIUS = 16;
+    int radius = int(min(float(MAX_RADIUS), u_radius));
+
+    for (int i = -MAX_RADIUS; i <= MAX_RADIUS; ++i) {
+        if (abs(i) > radius) continue;
+
+        vec2 offset = u_direction * u_texelSize * float(i);
+        sum += texture(u_texture, texCoords + offset).rgb;
+        count += 1.0;
+  }
+
+  fragColor = vec4(sum / count, 1.0);
+}
+`;
+
+/**
+ * Create the box blur shader program
+ */
+export function createBoxBlurProgram(gl: WebGL2RenderingContext) {
+  // Create vertex shader
+  const vertexShader = gl.createShader(gl.VERTEX_SHADER);
+  if (!vertexShader) {
+    throw Error('cannot create vertex shader');
+  }
+  gl.shaderSource(vertexShader, vertexShaderSource());
+  gl.compileShader(vertexShader);
+
+  // Check vertex shader compilation
+  if (!gl.getShaderParameter(vertexShader, gl.COMPILE_STATUS)) {
+    const info = gl.getShaderInfoLog(vertexShader);
+    throw Error(`Failed to compile vertex shader: ${info}`);
+  }
+
+  // Create fragment shader
+  const fragmentShader = gl.createShader(gl.FRAGMENT_SHADER);
+  if (!fragmentShader) {
+    throw Error('cannot create fragment shader');
+  }
+  gl.shaderSource(fragmentShader, boxBlurFragmentShader);
+  gl.compileShader(fragmentShader);
+
+  // Check fragment shader compilation
+  if (!gl.getShaderParameter(fragmentShader, gl.COMPILE_STATUS)) {
+    const info = gl.getShaderInfoLog(fragmentShader);
+    throw Error(`Failed to compile box blur shader: ${info}`);
+  }
+
+  // Create the program
+  const program = gl.createProgram();
+  if (!program) {
+    throw Error('cannot create box blur program');
+  }
+  gl.attachShader(program, vertexShader);
+  gl.attachShader(program, fragmentShader);
+  gl.linkProgram(program);
+
+  // Check program link status
+  if (!gl.getProgramParameter(program, gl.LINK_STATUS)) {
+    const info = gl.getProgramInfoLog(program);
+    throw Error(`Failed to link box blur program: ${info}`);
+  }
+
+  // Get attribute and uniform locations
+  const uniforms = {
+    position: gl.getAttribLocation(program, 'position'),
+    texture: gl.getUniformLocation(program, 'u_texture'),
+    texelSize: gl.getUniformLocation(program, 'u_texelSize'),
+    direction: gl.getUniformLocation(program, 'u_direction'),
+    radius: gl.getUniformLocation(program, 'u_radius'),
+  };
+
+  return {
+    program,
+    vertexShader,
+    fragmentShader,
+    uniforms,
+  };
+}

package/src/webgl/shader-programs/compositeShader.ts

@@ -0,0 +1,102 @@
+import { glsl } from '../utils';
+import { vertexShaderSource } from './vertexShader';
+
+// Fragment shader source for compositing
+export const compositeFragmentShader = glsl`#version 300 es
+  precision highp float;
+  in vec2 texCoords;
+  uniform sampler2D background;
+  uniform sampler2D frame;
+  uniform sampler2D mask;
+  out vec4 fragColor;
+  
+  void main() {
+      
+    vec4 frameTex = texture(frame, texCoords);
+    vec4 bgTex = texture(background, texCoords);
+
+    float maskVal = texture(mask, texCoords).r;
+
+    // Compute screen-space gradient to detect edge sharpness
+    float grad = length(vec2(dFdx(maskVal), dFdy(maskVal)));
+
+    float edgeSoftness = 2.0; // higher = softer
+    
+    // Create a smooth edge around binary transition
+    float smoothAlpha = smoothstep(0.5 - grad * edgeSoftness, 0.5 + grad * edgeSoftness, maskVal);
+
+    // Optional: preserve frame alpha, or override as fully opaque
+    vec4 blended = mix(bgTex, vec4(frameTex.rgb, 1.0), 1.0 - smoothAlpha);
+    
+    fragColor = blended;
+  
+  }
+`;
+
+/**
+ * Create the composite shader program
+ */
+export function createCompositeProgram(gl: WebGL2RenderingContext) {
+  // Create vertex shader
+  const vertexShader = gl.createShader(gl.VERTEX_SHADER);
+  if (!vertexShader) {
+    throw Error('cannot create vertex shader');
+  }
+  gl.shaderSource(vertexShader, vertexShaderSource());
+  gl.compileShader(vertexShader);
+
+  // Check vertex shader compilation
+  if (!gl.getShaderParameter(vertexShader, gl.COMPILE_STATUS)) {
+    const info = gl.getShaderInfoLog(vertexShader);
+    throw Error(`Failed to compile vertex shader: ${info}`);
+  }
+
+  // Create fragment shader
+  const compositeShader = gl.createShader(gl.FRAGMENT_SHADER);
+  if (!compositeShader) {
+    throw Error('cannot create fragment shader');
+  }
+  gl.shaderSource(compositeShader, compositeFragmentShader);
+  gl.compileShader(compositeShader);
+
+  // Check fragment shader compilation
+  if (!gl.getShaderParameter(compositeShader, gl.COMPILE_STATUS)) {
+    const info = gl.getShaderInfoLog(compositeShader);
+    throw Error(`Failed to compile composite shader: ${info}`);
+  }
+
+  // Create the program
+  const compositeProgram = gl.createProgram();
+  if (!compositeProgram) {
+    throw Error('cannot create composite program');
+  }
+  gl.attachShader(compositeProgram, vertexShader);
+  gl.attachShader(compositeProgram, compositeShader);
+  gl.linkProgram(compositeProgram);
+
+  // Check program link status
+  if (!gl.getProgramParameter(compositeProgram, gl.LINK_STATUS)) {
+    const info = gl.getProgramInfoLog(compositeProgram);
+    throw Error(`Failed to link composite program: ${info}`);
+  }
+
+  // Get attribute and uniform locations
+  const attribLocations = {
+    position: gl.getAttribLocation(compositeProgram, 'position'),
+  };
+
+  const uniformLocations = {
+    mask: gl.getUniformLocation(compositeProgram, 'mask')!,
+    frame: gl.getUniformLocation(compositeProgram, 'frame')!,
+    background: gl.getUniformLocation(compositeProgram, 'background')!,
+    stepWidth: gl.getUniformLocation(compositeProgram, 'u_stepWidth')!,
+  };
+
+  return {
+    program: compositeProgram,
+    vertexShader,
+    fragmentShader: compositeShader,
+    attribLocations,
+    uniformLocations,
+  };
+}

package/src/webgl/shader-programs/vertexShader.ts

@@ -0,0 +1,11 @@
+// Vertex shader source
+export const vertexShaderSource = (flipY: boolean = true) => `#version 300 es
+  in vec2 position;
+  out vec2 texCoords;
+
+  void main() {
+    texCoords = (position + 1.0) / 2.0;
+    texCoords.y = ${flipY ? '1.0 - texCoords.y' : 'texCoords.y'};
+    gl_Position = vec4(position, 0, 1.0);
+  }
+`;

package/src/webgl/utils.ts

@@ -0,0 +1,110 @@
+/**
+ * Initialize a WebGL texture
+ */
+export function initTexture(gl: WebGL2RenderingContext, texIndex: number) {
+  const texRef = gl.TEXTURE0 + texIndex;
+  gl.activeTexture(texRef);
+  const texture = gl.createTexture();
+  gl.bindTexture(gl.TEXTURE_2D, texture);
+  gl.texParameteri(gl.TEXTURE_2D, gl.TEXTURE_WRAP_S, gl.CLAMP_TO_EDGE);
+  gl.texParameteri(gl.TEXTURE_2D, gl.TEXTURE_WRAP_T, gl.CLAMP_TO_EDGE);
+  gl.texParameteri(gl.TEXTURE_2D, gl.TEXTURE_MIN_FILTER, gl.LINEAR);
+  gl.texParameteri(gl.TEXTURE_2D, gl.TEXTURE_MAG_FILTER, gl.LINEAR);
+  gl.bindTexture(gl.TEXTURE_2D, texture);
+
+  return texture;
+}
+
+/**
+ * Create a WebGL framebuffer with the given texture as color attachment
+ */
+export function createFramebuffer(
+  gl: WebGL2RenderingContext,
+  texture: WebGLTexture,
+  width: number,
+  height: number,
+) {
+  const framebuffer = gl.createFramebuffer();
+  gl.bindFramebuffer(gl.FRAMEBUFFER, framebuffer);
+
+  // Set the texture as the color attachment
+  gl.framebufferTexture2D(gl.FRAMEBUFFER, gl.COLOR_ATTACHMENT0, gl.TEXTURE_2D, texture, 0);
+
+  // Ensure texture dimensions match the provided width and height
+  gl.bindTexture(gl.TEXTURE_2D, texture);
+  gl.texImage2D(gl.TEXTURE_2D, 0, gl.RGBA, width, height, 0, gl.RGBA, gl.UNSIGNED_BYTE, null);
+
+  // Check if framebuffer is complete
+  const status = gl.checkFramebufferStatus(gl.FRAMEBUFFER);
+  if (status !== gl.FRAMEBUFFER_COMPLETE) {
+    throw new Error('Framebuffer not complete');
+  }
+
+  gl.bindFramebuffer(gl.FRAMEBUFFER, null);
+  return framebuffer;
+}
+
+/**
+ * Create a vertex buffer for a full-screen quad
+ */
+export function createVertexBuffer(gl: WebGL2RenderingContext): WebGLBuffer | null {
+  const vertexBuffer = gl.createBuffer();
+  gl.bindBuffer(gl.ARRAY_BUFFER, vertexBuffer);
+  gl.bufferData(
+    gl.ARRAY_BUFFER,
+    new Float32Array([-1, -1, -1, 1, 1, 1, -1, -1, 1, 1, 1, -1]),
+    gl.STATIC_DRAW,
+  );
+  return vertexBuffer;
+}
+
+/**
+ * Resizes and crops an image to cover a target canvas while maintaining aspect ratio
+ * @param image The source image
+ * @param targetWidth The target width
+ * @param targetHeight The target height
+ * @returns A cropped and resized ImageBitmap
+ */
+export async function resizeImageToCover(
+  image: ImageBitmap,
+  targetWidth: number,
+  targetHeight: number,
+): Promise<ImageBitmap> {
+  // Calculate dimensions and crop for "cover" mode
+  const imgAspect = image.width / image.height;
+  const targetAspect = targetWidth / targetHeight;
+
+  let sx = 0;
+  let sy = 0;
+  let sWidth = image.width;
+  let sHeight = image.height;
+
+  // For cover mode, we need to crop some parts of the image
+  // to ensure it covers the canvas while maintaining aspect ratio
+  if (imgAspect > targetAspect) {
+    // Image is wider than target - crop the sides
+    sWidth = Math.round(image.height * targetAspect);
+    sx = Math.round((image.width - sWidth) / 2); // Center the crop horizontally
+  } else if (imgAspect < targetAspect) {
+    // Image is taller than target - crop the top/bottom
+    sHeight = Math.round(image.width / targetAspect);
+    sy = Math.round((image.height - sHeight) / 2); // Center the crop vertically
+  }
+
+  // Create a new ImageBitmap with the cropped portion
+  return createImageBitmap(image, sx, sy, sWidth, sHeight, {
+    resizeWidth: targetWidth,
+    resizeHeight: targetHeight,
+    resizeQuality: 'medium',
+  });
+}
+
+const emptyImageData = new ImageData(2, 2);
+emptyImageData.data[0] = 0;
+emptyImageData.data[1] = 0;
+emptyImageData.data[2] = 0;
+emptyImageData.data[3] = 0;
+
+const glsl = (source: any) => source;
+
+export { emptyImageData, glsl };