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
|
onrtctransform = event => {
const transformer = event.transformer;
let keyFrameCount = 0;
let gotFrame;
transformer.options.port.onmessage = event => {
const {method, rid} = event.data;
// Maybe refactor to have transaction ids?
if (method == 'generateKeyFrame') {
generateKeyFrame(rid);
} else if (method == 'waitForFrame') {
waitForFrame();
}
}
async function rejectInMs(timeout) {
return new Promise((_, reject) => {
const rejectWithTimeout = () => {
reject(new DOMException(`Timed out after waiting for ${timeout} ms`,
'TimeoutError'));
};
setTimeout(rejectWithTimeout, timeout);
});
}
async function generateKeyFrame(rid) {
try {
const timestamp = await Promise.race([transformer.generateKeyFrame(rid), rejectInMs(8000)]);
transformer.options.port.postMessage({result: 'success', value: timestamp, count: keyFrameCount});
} catch (e) {
// TODO: This does not work if we send e.name, why?
transformer.options.port.postMessage({result: 'failure', value: `${e.name}`, message: `${e.message}`});
}
}
async function waitForFrame() {
try {
await Promise.race([new Promise(r => gotFrameCallback = r), rejectInMs(8000)]);
transformer.options.port.postMessage('got frame');
} catch (e) {
// TODO: This does not work if we send e.name, why?
transformer.options.port.postMessage({result: 'failure', value: `${e.name}`, message: `${e.message}`});
}
}
transformer.options.port.postMessage('started');
transformer.reader = transformer.readable.getReader();
transformer.writer = transformer.writable.getWriter();
function process(transformer)
{
transformer.reader.read().then(chunk => {
if (chunk.done)
return;
if (chunk.value instanceof RTCEncodedVideoFrame) {
if (chunk.value.type == 'key') {
keyFrameCount++;
}
}
if (gotFrameCallback) {
gotFrameCallback();
}
transformer.writer.write(chunk.value);
process(transformer);
});
}
process(transformer);
};
self.postMessage('registered');
|