Function convertNv12ToRgbaMainThread

apps/desktop/src/utils/socket.ts:109–196

(
	nv12Data: Uint8ClampedArray,
	width: number,
	height: number,
	yStride: number,
	fullRange = false,
)

}

function convertNv12ToRgbaMainThread(
	nv12Data: Uint8ClampedArray,
	width: number,
	height: number,
	yStride: number,
	fullRange = false,
): Uint8ClampedArray {
	const rgbaSize = width * height * 4;
	if (!mainThreadNv12Buffer || mainThreadNv12BufferSize < rgbaSize) {
		mainThreadNv12Buffer = new Uint8ClampedArray(rgbaSize);
		mainThreadNv12BufferSize = rgbaSize;
	}
	const rgba = mainThreadNv12Buffer;

	const ySize = yStride * height;
	const yPlane = nv12Data;
	const uvPlane = nv12Data.subarray(ySize);
	const uvStride = yStride;

	for (let row = 0; row < height; row++) {
		const yRowOffset = row * yStride;
		const uvRowOffset = (row >> 1) * uvStride;
		const rgbaRowOffset = row * width * 4;

		for (let col = 0; col < width; col += 2) {
			const uvCol = (col >> 1) * 2;
			const u = uvPlane[uvRowOffset + uvCol] - 128;
			const v = uvPlane[uvRowOffset + uvCol + 1] - 128;
			const d = u;
			const e = v;

			const y0 = yPlane[yRowOffset + col];
			let r: number;
			let g: number;
			let b: number;
			if (fullRange) {
				r = y0 + ((359 * e + 128) >> 8);
				g = y0 - ((88 * d + 183 * e + 128) >> 8);
				b = y0 + ((454 * d + 128) >> 8);
			} else {
				const c0 = 298 * (y0 - 16);
				r = (c0 + 409 * e + 128) >> 8;
				g = (c0 - 100 * d - 208 * e + 128) >> 8;
				b = (c0 + 516 * d + 128) >> 8;
			}

			r = r < 0 ? 0 : r > 255 ? 255 : r;
			g = g < 0 ? 0 : g > 255 ? 255 : g;
			b = b < 0 ? 0 : b > 255 ? 255 : b;

			const rgbaOffset = rgbaRowOffset + col * 4;
			rgba[rgbaOffset] = r;
			rgba[rgbaOffset + 1] = g;
			rgba[rgbaOffset + 2] = b;
			rgba[rgbaOffset + 3] = 255;

			const nextCol = col + 1;
			if (nextCol < width) {