File size: 2,080 Bytes
52ae39e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 |
importScripts('./webm-writer2.js');
let webmWriter = null;
let fileWritableStream = null;
let frameReader = null;
async function startRecording(
fileHandle,
frameStream,
trackSettings
) {
let frameCounter = 0;
fileWritableStream = await fileHandle.createWritable();
webmWriter = new WebMWriter({
fileWriter: fileWritableStream,
codec: 'VP9',
width: trackSettings.width,
height: trackSettings.height,
});
frameReader = frameStream.getReader();
const init = {
output: (chunk) => {
webmWriter.addFrame(chunk);
},
error: (e) => {
console.log(e.message);
stopRecording();
},
};
const config = {
codec: 'vp09.00.10.08',
width: trackSettings.width,
height: trackSettings.height,
bitrate: 10e6,
};
let encoder = new VideoEncoder(init);
let support = await VideoEncoder.isConfigSupported(config);
console.assert(support.supported);
encoder.configure(config);
frameReader
.read()
.then(async function processFrame({ done, value }) {
let frame = value;
if (done) {
await encoder.flush();
encoder.close();
return;
}
if (encoder.encodeQueueSize <= 30) {
if (++frameCounter % 20 == 0) {
console.log(frameCounter + ' frames processed');
}
const insert_keyframe = frameCounter % 150 == 0;
encoder.encode(frame, { keyFrame: insert_keyframe });
} else {
console.log('dropping frame, encoder falling behind');
}
frame.close();
frameReader.read().then(processFrame);
});
}
async function stopRecording() {
await frameReader.cancel();
await webmWriter.complete();
fileWritableStream.close();
frameReader = null;
webmWriter = null;
fileWritableStream = null;
}
self.addEventListener('message', function (e) {
switch (e.data.type) {
case 'start':
startRecording(
e.data.fileHandle,
e.data.frameStream,
e.data.trackSettings
);
break;
case 'stop':
stopRecording();
break;
}
});
|