glKernel is an ultra light-weight WebGL2 compute library. Define your compute shader, pass in you data, and get the output without writing any boilterplate code.
import { glKernel } from 'gl-kernel';
// A simple elementwise addition
const elementwise = `
float add(in sampler2D data0, in sampler2D data1, int idx) {
return read(data0, idx) + read(data1, idx);
}
`
const a = new Float32Array([0.5, 0.5]);
const b = new Float32Array([1.6, 1.6]);
const out = new Float32Array(2);
glKernel.compute({entry: "add", code: elementwise}, [a, b], out);
console.log(out); // [2.1, 2.1]
// A simple sum
const reduce = `
float sum(in sampler2D data0, int idx) {
float acc = 0.0;
for (int i = 0; i < 4; i++) {
acc += read(data0, idx+i);
}
return acc;
}
`
const test = new Float32Array([0.5, 1.0, 3.0, 4.0]);
const sumOut = new Float32Array(1);
glKernel.compute({ entry: "sum", code: reduce }, [test], sumOut);
console.log(sumOut); // 8.5MIT