fragmentShaderTest(3, tester => {
const shaderManager = new SegmentColorShaderManager('getColor');
let {gl, builder} = tester;
shaderManager.defineShader(builder);
builder.addUniform('highp vec4', 'inputValue', 2);
const colorHash = SegmentColorHash.getDefault();
builder.addOutputBuffer('vec4', 'v4f_fragData0', 0);
builder.addOutputBuffer('vec4', 'v4f_fragData1', 1);
builder.addOutputBuffer('vec4', 'v4f_fragData2', 2);
builder.setFragmentMain(`
uint64_t x;
x.low = inputValue[0];
x.high = inputValue[1];
highp vec3 color = getColor(x);
v4f_fragData0 = packFloatIntoVec4(color.x);
v4f_fragData1 = packFloatIntoVec4(color.y);
v4f_fragData2 = packFloatIntoVec4(color.z);
`);
tester.build();
let {shader} = tester;
shader.bind();
shaderManager.enable(gl, shader, colorHash);
function testValue(x: Uint64) {
let temp = new Uint32Array(2);
temp[0] = x.low;
temp[1] = x.high;
let inputValue = encodeBytesToFloat32(temp);
gl.uniform4fv(shader.uniform('inputValue'), inputValue);
tester.execute();
let actual = new Float32Array(3);
for (let i = 0; i < 3; ++i) {
actual[i] = tester.readFloat(i);
}
let expected = new Float32Array(3);
colorHash.compute(expected, x);
expect(actual).toEqual(expected, `x = ${x}`);
}
testValue(Uint64.parseString('0'));
testValue(Uint64.parseString('8'));
const COUNT = 100;
for (let iter = 0; iter < COUNT; ++iter) {
let x = Uint64.random();
testValue(x);
}
});
fragmentShaderTest({outR: 'float', outG: 'float', outB: 'float'}, tester => {
const shaderManager = new SegmentColorShaderManager('getColor');
let {gl, builder} = tester;
shaderManager.defineShader(builder);
builder.addUniform('highp uvec2', 'inputValue');
const colorHash = SegmentColorHash.getDefault();
builder.addFragmentCode(glsl_unpackUint64leFromUint32);
builder.setFragmentMain(`
uint64_t x = unpackUint64leFromUint32(inputValue);
highp vec3 color = getColor(x);
outR = color.r;
outG = color.g;
outB = color.b;
`);
tester.build();
let {shader} = tester;
shader.bind();
shaderManager.enable(gl, shader, colorHash);
function testValue(x: Uint64) {
gl.uniform2ui(shader.uniform('inputValue'), x.low, x.high);
tester.execute();
let actual = new Float32Array(3);
for (let i = 0; i < 3; ++i) {
actual[i] = tester.readFloat(i);
}
let expected = new Float32Array(3);
colorHash.compute(expected, x);
const {values} = tester;
expect(values.outR).toBeCloseTo(expected[0]);
expect(values.outG).toBeCloseTo(expected[1]);
expect(values.outB).toBeCloseTo(expected[2]);
}
testValue(Uint64.parseString('0'));
testValue(Uint64.parseString('8'));
const COUNT = 100;
for (let iter = 0; iter < COUNT; ++iter) {
let x = Uint64.random();
testValue(x);
}
});