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
|
import { kUnitCaseParamsBuilder } from '../../../common/framework/params_builder.js';
import { Float16Array } from '../../../external/petamoriken/float16/float16.js';
import { kCanvasAlphaModes, kCanvasColorSpaces } from '../../capability_info.js';
import { runRefTest } from './gpu_ref_test.js';
function bgra8UnormFromRgba8Unorm(rgba8Unorm: Uint8Array) {
const bgra8Unorm = rgba8Unorm.slice();
for (let i = 0; i < bgra8Unorm.length; i += 4) {
[bgra8Unorm[i], bgra8Unorm[i + 2]] = [bgra8Unorm[i + 2], bgra8Unorm[i]];
}
return bgra8Unorm;
}
function rgba16floatFromRgba8unorm(rgba8Unorm: Uint8Array) {
const rgba16Float = new Float16Array(rgba8Unorm.length);
for (let i = 0; i < rgba8Unorm.length; ++i) {
rgba16Float[i] = rgba8Unorm[i] / 255;
}
return rgba16Float;
}
export function runColorSpaceTest(format: GPUTextureFormat) {
runRefTest(async t => {
const device = t.device;
// prettier-ignore
const kRGBA8UnormData = new Uint8Array([
0, 255, 0, 255,
117, 251, 7, 255,
170, 35, 209, 255,
80, 150, 200, 255,
]);
const kBGRA8UnormData = bgra8UnormFromRgba8Unorm(kRGBA8UnormData);
const kRGBA16FloatData = rgba16floatFromRgba8unorm(kRGBA8UnormData);
const width = kRGBA8UnormData.length / 4;
const testData: { [id: string]: Uint8Array | Float16Array } = {
rgba8unorm: kRGBA8UnormData,
bgra8unorm: kBGRA8UnormData,
rgba16float: kRGBA16FloatData,
};
function createCanvas(
alphaMode: GPUCanvasAlphaMode,
format: GPUTextureFormat,
colorSpace: PredefinedColorSpace
) {
const canvas = document.createElement('canvas');
canvas.width = width;
canvas.height = 1;
const context = canvas.getContext('webgpu') as GPUCanvasContext;
context.configure({
device,
format,
usage: GPUTextureUsage.COPY_DST | GPUTextureUsage.RENDER_ATTACHMENT,
alphaMode,
colorSpace,
});
const textureData = testData[format];
const texture = context.getCurrentTexture();
device.queue.writeTexture(
{ texture },
textureData.buffer,
{
bytesPerRow: textureData.byteLength,
},
{ width: 4, height: 1 }
);
document.body.appendChild(canvas);
}
const u = kUnitCaseParamsBuilder
.combine('alphaMode', kCanvasAlphaModes)
.combine('colorSpace', kCanvasColorSpaces);
for (const { alphaMode, colorSpace } of u) {
createCanvas(alphaMode, format, colorSpace);
}
});
}
|