49 lines
1.9 KiB
TypeScript
49 lines
1.9 KiB
TypeScript
import Size from './utils/Size';
|
|
import Canvas from './Canvas';
|
|
|
|
export default class YUVCanvas extends Canvas {
|
|
private canvasCtx: CanvasRenderingContext2D;
|
|
private canvasBuffer: ImageData;
|
|
|
|
constructor(readonly canvas: HTMLCanvasElement, readonly size: Size) {
|
|
super(canvas, size);
|
|
this.canvasCtx = this.canvas.getContext('2d') as CanvasRenderingContext2D;
|
|
this.canvasBuffer = this.canvasCtx.createImageData(size.w, size.h);
|
|
}
|
|
|
|
public decode(buffer: Uint8Array, width: number, height: number): void {
|
|
if (!buffer) {
|
|
return;
|
|
}
|
|
|
|
const lumaSize = width * height;
|
|
const chromaSize = lumaSize >> 2;
|
|
|
|
const ybuf = buffer.subarray(0, lumaSize);
|
|
const ubuf = buffer.subarray(lumaSize, lumaSize + chromaSize);
|
|
const vbuf = buffer.subarray(lumaSize + chromaSize, lumaSize + 2 * chromaSize);
|
|
|
|
for (let y = 0; y < height; y++) {
|
|
for (let x = 0; x < width; x++) {
|
|
const yIndex = x + y * width;
|
|
const uIndex = ~~(y / 2) * ~~(width / 2) + ~~(x / 2);
|
|
const vIndex = ~~(y / 2) * ~~(width / 2) + ~~(x / 2);
|
|
const R = 1.164 * (ybuf[yIndex] - 16) + 1.596 * (vbuf[vIndex] - 128);
|
|
const G = 1.164 * (ybuf[yIndex] - 16) - 0.813 * (vbuf[vIndex] - 128) - 0.391 * (ubuf[uIndex] - 128);
|
|
const B = 1.164 * (ybuf[yIndex] - 16) + 2.018 * (ubuf[uIndex] - 128);
|
|
|
|
const rgbIndex = yIndex * 4;
|
|
this.canvasBuffer.data[rgbIndex + 0] = R;
|
|
this.canvasBuffer.data[rgbIndex + 1] = G;
|
|
this.canvasBuffer.data[rgbIndex + 2] = B;
|
|
this.canvasBuffer.data[rgbIndex + 3] = 0xff;
|
|
}
|
|
}
|
|
|
|
this.canvasCtx.putImageData(this.canvasBuffer, 0, 0);
|
|
|
|
// const date = new Date();
|
|
// console.log('WSAvcPlayer: Decode time: ' + (date.getTime() - this.rcvtime) + ' ms');
|
|
}
|
|
}
|