function testInsert(k: Uint64, v: Uint64) {
let s = '' + k;
map.set(s, v);
expect(ht.has(k)).toBe(false, `Unexpected positive has result for ${s}`);
ht.set(k, v);
compare();
}
fragmentShaderTest({isPresent: 'uint', outLow: 'uint', outHigh: 'uint'}, tester => {
let {gl, builder} = tester;
let shaderManager = new HashMapShaderManager('h');
shaderManager.defineShader(builder);
builder.addUniform('highp uvec2', 'inputValue');
builder.setFragmentMain(`
uint64_t key = unpackUint64leFromUint32(inputValue);
uint64_t value;
isPresent = uint(h_get(key, value));
outLow = value.value[0];
outHigh = value.value[1];
`);
tester.build();
let {shader} = tester;
shader.bind();
let hashTable = new HashMapUint64();
let gpuHashTable = tester.registerDisposer(GPUHashTable.get(gl, hashTable));
let testValues = new Array<Uint64>();
while (testValues.length < COUNT) {
let x = Uint64.random();
if (hashTable.has(x)) {
continue;
}
testValues.push(x);
hashTable.set(x, Uint64.random());
}
let notPresentValues = new Array<Uint64>();
notPresentValues.push(new Uint64(hashTable.emptyLow, hashTable.emptyHigh));
while (notPresentValues.length < COUNT) {
let x = Uint64.random();
if (hashTable.has(x)) {
continue;
}
notPresentValues.push(x);
}
function checkPresent(x: Uint64) {
gl.uniform2ui(shader.uniform('inputValue'), x.low, x.high);
shaderManager.enable(gl, shader, gpuHashTable);
tester.execute();
const {values} = tester;
let expectedValue = new Uint64();
let expectedHas = hashTable.get(x, expectedValue);
const has = values.isPresent === 1;
expect(has).toBe(expectedHas, `x=${x}`);
if (has) {
expect(values.outLow).toBe(expectedValue.low, `x=${x}, low`);
expect(values.outHigh).toBe(expectedValue.high, `x=${x}, high`);
}
}
testValues.forEach((x, i) => {
expect(hashTable.has(x)).toBe(true, `cpu: i = ${i}, x = ${x}`);
checkPresent(x);
});
notPresentValues.forEach(x => {
checkPresent(x);
});
});
function compareViaGet() {
let value = new Uint64();
for (let [s, expectedValue] of map) {
let key = Uint64.parseString(s);
let has = ht.get(key, value);
expect(has && Uint64.equal(value, expectedValue))
.toBe(
true,
`Hash table maps ${key} -> ${has ? value : undefined} rather than -> ${expectedValue}`);
}
}
function checkPresent(x: Uint64) {
gl.uniform2ui(shader.uniform('inputValue'), x.low, x.high);
shaderManager.enable(gl, shader, gpuHashTable);
tester.execute();
const {values} = tester;
let expectedValue = new Uint64();
let expectedHas = hashTable.get(x, expectedValue);
const has = values.isPresent === 1;
expect(has).toBe(expectedHas, `x=${x}`);
if (has) {
expect(values.outLow).toBe(expectedValue.low, `x=${x}, low`);
expect(values.outHigh).toBe(expectedValue.high, `x=${x}, high`);
}
}
function checkPresent(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);
shaderManager.enable(gl, shader, gpuHashTable);
tester.execute();
let resultBytes = tester.readBytes();
let has = resultBytes[0] === 255;
let expectedValue = new Uint64();
let expectedHas = hashTable.get(x, expectedValue);
expect(has).toEqual(expectedHas);
if (has) {
expect(tester.readUint32(1)).toEqual(expectedValue.low);
expect(tester.readUint32(2)).toEqual(expectedValue.high);
}
}
it('HashMapUint64', () => {
let ht = new HashMapUint64();
let map = new Map<string, Uint64>();
function compareViaIterate() {
let htValues = new Map<string, Uint64>();
for (let [key, value] of ht) {
let s = key.toString();
expect(htValues.has(s)).toBe(false, `Duplicate key in hash table: ${s}`);
expect(map.has(s)).toBe(true, `Unexpected key ${s} in hash table`);
htValues.set(s, value.clone());
}
for (let [s, value] of map) {
let v = htValues.get(s);
expect(v !== undefined && Uint64.equal(v, value))
.toBe(true, `Hash table maps ${s} -> ${v} rather than -> ${value}`);
}
}
function compareViaGet() {
let value = new Uint64();
for (let [s, expectedValue] of map) {
let key = Uint64.parseString(s);
let has = ht.get(key, value);
expect(has && Uint64.equal(value, expectedValue))
.toBe(
true,
`Hash table maps ${key} -> ${has ? value : undefined} ` +
`rather than -> ${expectedValue}`);
}
}
function compare() {
compareViaIterate();
compareViaGet();
}
let numInsertions = 100;
function testInsert(k: Uint64, v: Uint64) {
let s = '' + k;
map.set(s, v);
expect(ht.has(k)).toBe(false, `Unexpected positive has result for ${s}`);
ht.set(k, v);
compare();
}
let empty0 = new Uint64(ht.emptyLow, ht.emptyHigh);
testInsert(empty0, Uint64.random());
for (let i = 0; i < numInsertions; ++i) {
let k: Uint64;
let s: string;
while (true) {
k = Uint64.random();
s = k.toString();
if (!map.has(s)) {
break;
}
}
testInsert(k, Uint64.random());
}
let empty1 = new Uint64(ht.emptyLow, ht.emptyHigh);
testInsert(empty1, Uint64.random());
});
testValues.forEach((x, i) => {
expect(hashTable.has(x)).toBe(true, `cpu: i = ${i}, x = ${x}`);
checkPresent(x);
});
fragmentShaderTest(3, tester => {
let {gl, builder} = tester;
let shaderManager = new HashMapShaderManager('h');
shaderManager.defineShader(builder);
builder.addUniform('vec4', 'inputValue', 2);
builder.setFragmentMain(`
uint64_t x;
x.low = inputValue[0];
x.high = inputValue[1];
uint64_t y;
gl_FragData[0] = h_get(x, y) ? vec4(1.0, 1.0, 1.0, 1.0) : vec4(0.0, 0.0, 0.0, 0.0);
gl_FragData[1] = y.low;
gl_FragData[2] = y.high;
`);
tester.build();
let {shader} = tester;
shader.bind();
let hashTable = new HashMapUint64();
let gpuHashTable = tester.registerDisposer(GPUHashTable.get(gl, hashTable));
let testValues = new Array<Uint64>();
while (testValues.length < COUNT) {
let x = Uint64.random();
if (hashTable.has(x)) {
continue;
}
testValues.push(x);
hashTable.set(x, Uint64.random());
}
let notPresentValues = new Array<Uint64>();
notPresentValues.push(new Uint64(hashTable.emptyLow, hashTable.emptyHigh));
while (notPresentValues.length < COUNT) {
let x = Uint64.random();
if (hashTable.has(x)) {
continue;
}
notPresentValues.push(x);
}
let mungedTable: Uint32Array;
hashTable.tableWithMungedEmptyKey(table => { mungedTable = new Uint32Array(table); });
function checkPresent(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);
shaderManager.enable(gl, shader, gpuHashTable);
tester.execute();
let resultBytes = tester.readBytes();
let has = resultBytes[0] === 255;
let expectedValue = new Uint64();
let expectedHas = hashTable.get(x, expectedValue);
expect(has).toEqual(expectedHas);
if (has) {
expect(tester.readUint32(1)).toEqual(expectedValue.low);
expect(tester.readUint32(2)).toEqual(expectedValue.high);
}
}
testValues.forEach((x, i) => {
expect(hashTable.has(x)).toBe(true, `cpu: i = ${i}, x = ${x}`);
checkPresent(x);
});
notPresentValues.forEach((x, ) => { checkPresent(x); });
});