-
Notifications
You must be signed in to change notification settings - Fork 6.6k
/
copyTo.html
158 lines (143 loc) · 5.18 KB
/
copyTo.html
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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
<!DOCTYPE html>
<!--
Take frames coming from various sources and read them using copyTo().
-->
<title>copyTo() test</title>
<script src="webcodecs_common.js"></script>
<script type="text/javascript">
'use strict';
function yuv_to_rgb601(y, u, v) {
let b = 1.164 * (y - 16) + 2.018 * (u - 128);
let g = 1.164 * (y - 16) - 0.813 * (v - 128) - 0.391 * (u - 128);
let r = 1.164 * (y - 16) + 1.596 * (v - 128);
return { r: r, g: g, b: b };
}
function yuv_to_rgb709(y, u, v) {
let b = 1.164 * (y - 16) + 2.113 * (u - 128);
let g = 1.164 * (y - 16) - 0.533 * (v - 128) - 0.213 * (u - 128);
let r = 1.164 * (y - 16) + 1.793 * (v - 128);
return { r: r, g: g, b: b };
}
async function validateFourColorsBytes(frame) {
const m = 4;
const tolerance = 8;
let expected_xy_color = [
// Left-top yellow
{ x: m, y: m, r: 255, g: 255, b: 0 },
// Right-top red
{ x: frame.displayWidth - m, y: m, r: 255, g: 0, b: 0 },
// Left-bottom blue
{ x: m, y: frame.displayHeight - m, r: 0, g: 0, b: 255 },
// Right-bottom green
{ x: frame.displayWidth - m, y: frame.displayHeight - m,
r: 0, g: 255, b: 0 },
]
for (let test of expected_xy_color) {
const options = {
rect: { x: test.x, y: test.y, width: 2, height: 2 }
}
let size = frame.allocationSize(options);
let buffer = new ArrayBuffer(size);
let layout = await frame.copyTo(buffer, options);
let view = new DataView(buffer);
let rgb = null;
let yuv_to_rgb = yuv_to_rgb601;
if (frame.colorSpace.matrix == "bt709")
yuv_to_rgb = yuv_to_rgb709;
switch (frame.format) {
case "I420":
case "I420A":
rgb = yuv_to_rgb(view.getUint8(layout[0].offset), // Y
view.getUint8(layout[1].offset), // U
view.getUint8(layout[2].offset)); // V
break;
case "NV12":
rgb = yuv_to_rgb(view.getUint8(layout[0].offset), // Y
view.getUint8(layout[1].offset), // U
view.getUint8(layout[1].offset + 1)); // V
break;
case "RGBX":
case "RGBA":
rgb = {
r: view.getUint8(layout[0].offset),
g: view.getUint8(layout[0].offset + 1),
b: view.getUint8(layout[0].offset + 2)
};
break;
case "BGRX":
case "BGRA":
rgb = {
r: view.getUint8(layout[0].offset + 2),
g: view.getUint8(layout[0].offset + 1),
b: view.getUint8(layout[0].offset)
};
break;
default:
TEST.reportFailure("Unexpected frame format: " + frame.format);
return;
}
let message =
`Test ${frame.format} ${JSON.stringify(test)} ${JSON.stringify(rgb)}`;
TEST.assert(Math.abs(rgb.r - test.r) < tolerance, message);
TEST.assert(Math.abs(rgb.g - test.g) < tolerance, message);
TEST.assert(Math.abs(rgb.b - test.b) < tolerance, message);
}
}
function readWholeBuffer(buffer) {
let bytes = new Uint8Array(buffer);
return bytes.reduce((acc, byte) => { return byte ? acc + 1 : acc; }, 0);
}
function makeWorker() {
const worker = new Worker('buffer-read-and-reply-worker.js');
let resolve_promise = null;
worker.onmessage = function(e) { resolve_promise(e.data); }
let worker_promise = new Promise((resolve) => {
resolve_promise = resolve;
});
return { worker , worker_promise };
}
async function main(arg) {
let source_type = arg.source_type;
let source = await createFrameSource(source_type, 320, 240);
if (!source) {
TEST.skip('Unsupported source: ' + source_type);
return;
}
let frame = await source.getNextFrame();
let size = frame.allocationSize();
// Readback a whole frame to a regular buffer and send it to a worker
{
let {worker, worker_promise } = makeWorker();
let buf = new ArrayBuffer(size);
TEST.assert(readWholeBuffer(buf) == 0, "Buffer should be zero");
let copy_promise = frame.copyTo(buf);
worker.postMessage(buf, [buf]);
let result_buf = await worker_promise;
let layout = await copy_promise;
TEST.assert(readWholeBuffer(result_buf) > 0, "Buffer shouldn't be zero");
TEST.assert(layout, "layout is empty / ArrayBuffer");
worker.terminate();
}
// Readback a whole frame to a shared buffer and send it to a worker
{
let {worker, worker_promise} = makeWorker();
let buf = new SharedArrayBuffer(size);
TEST.assert(readWholeBuffer(buf) == 0, "Buffer should be zero");
let copy_promise = frame.copyTo(buf);
worker.postMessage(buf);
await worker_promise;
let layout = await copy_promise;
TEST.assert(readWholeBuffer(buf) > 0, "SharedBuffer shouldn't be zero");
TEST.assert(layout, "layout is empty / SharedArrayBuffer");
worker.terminate();
}
// Validate pixels
if (!arg.validate_camera_frames && source_type == 'camera') {
TEST.log("Skip copyTo result validation");
} else {
await validateFourColorsBytes(frame);
}
frame.close();
source.close();
}
</script>