gpu.js/test/features/input.js

const { assert, skip, test, module: describe, only } = require('qunit');
const { GPU, input } = require('../../src');

describe('input');

function inputX(mode) {
  const gpu = new GPU({ mode });
  const kernel = gpu.createKernel(function(a) {
    return a[this.thread.x];
  })
    .setOutput([9]);

  const a = new Float32Array(9);
  a.set([1,2,3,4,5,6,7,8,9]);

  const result = kernel(input(a, [3, 3]));
  assert.deepEqual(Array.from(result), [1,2,3,4,5,6,7,8,9]);
  gpu.destroy();
}

test("inputX auto", () => {
  inputX();
});

test("inputX gpu", () => {
  inputX('gpu');
});

(GPU.isWebGLSupported ? test : skip)("inputX webgl", () => {
  inputX('webgl');
});

(GPU.isWebGL2Supported ? test : skip)("inputX webgl2", () => {
  inputX('webgl2');
});

test("inputX cpu", () => {
  inputX('cpu');
});


function inputXY(mode) {
  const gpu = new GPU({ mode });
  const kernel = gpu.createKernel(function(a) {
    return a[this.thread.y][this.thread.x];
  })
    .setOutput([9]);

  const a = new Float32Array(9);
  a.set([1,2,3,4,5,6,7,8,9]);

  const b = new Float32Array(9);
  b.set([1,2,3,4,5,6,7,8,9]);

  const result = kernel(input(a, [3, 3]));
  assert.deepEqual(Array.from(result), [1,2,3,4,5,6,7,8,9]);
  gpu.destroy();
}

test("inputXY auto", () => {
  inputXY();
});

test("inputXY gpu", () => {
  inputXY('gpu');
});

(GPU.isWebGLSupported ? test : skip)("inputXY webgl", () => {
  inputXY('webgl');
});

(GPU.isWebGL2Supported ? test : skip)("inputXY webgl2", () => {
  inputXY('webgl2');
});

(GPU.isHeadlessGLSupported ? test : skip)("inputXY headlessgl", () => {
  inputXY('headlessgl');
});

test("inputXY cpu", () => {
  inputXY('cpu');
});

function inputYX(mode) {
  const gpu = new GPU({ mode });
  const kernel = gpu.createKernel(function(a) {
    return a[this.thread.y][this.thread.x];
  })
    .setOutput([3, 3]);

  const a = new Float32Array(9);
  a.set([1,2,3,4,5,6,7,8,9]);

  const result = kernel(input(a, [3, 3]));
  assert.deepEqual(result.map(function(v) { return Array.from(v); }), [[1,2,3],[4,5,6],[7,8,9]]);
  gpu.destroy();
}

test("inputYX auto", () => {
  inputYX();
});

test("inputYX gpu", () => {
  inputYX('gpu');
});

(GPU.isWebGLSupported ? test : skip)("inputYX webgl", () => {
  inputYX('webgl');
});

(GPU.isWebGL2Supported ? test : skip)("inputYX webgl2", () => {
  inputYX('webgl2');
});

(GPU.isHeadlessGLSupported ? test : skip)("inputYX headlessgl", () => {
  inputYX('headlessgl');
});

test("inputYX cpu", () => {
  inputYX('cpu');
});

function inputYXOffset(mode) {
  const gpu = new GPU({ mode });
  const kernel = gpu.createKernel(function(a) {
    return a[this.thread.x][this.thread.y];
  })
    .setOutput([8, 2]);

  const a = new Float32Array(16);
  a.set([1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16]);

  const result = kernel(input(a, [2, 8]));
  assert.deepEqual(result.map(function(v) { return Array.from(v); }), [[1,3,5,7,9,11,13,15],[2,4,6,8,10,12,14,16]]);
  gpu.destroy();
}

test("inputYXOffset auto", () => {
  inputYXOffset();
});

test("inputYXOffset gpu", () => {
  inputYXOffset('gpu');
});

(GPU.isWebGLSupported ? test : skip)("inputYXOffset webgl", () => {
  inputYXOffset('webgl');
});

(GPU.isWebGL2Supported ? test : skip)("inputYXOffset webgl2", () => {
  inputYXOffset('webgl2');
});

(GPU.isHeadlessGLSupported ? test : skip)("inputYXOffset headlessgl", () => {
  inputYXOffset('headlessgl');
});

test("inputYXOffset cpu", () => {
  inputYXOffset('cpu');
});

function inputYXOffsetPlus1(mode) {
  const gpu = new GPU({ mode });
  const kernel = gpu.createKernel(function(a) {
    return a[this.thread.x][this.thread.y];
  })
    .setOutput([2, 8]);

  const a = new Float32Array(16);
  a.set([1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16]);

  const result = kernel(input(a, [8, 2]));
  assert.deepEqual(result.map(function(v) { return Array.from(v); }), [[1,9],[2,10],[3,11],[4,12],[5,13],[6,14],[7,15],[8,16]]);
  gpu.destroy();
}

test("inputYXOffsetPlus1 auto", () => {
  inputYXOffsetPlus1();
});

test("inputYXOffsetPlus1 gpu", () => {
  inputYXOffsetPlus1('gpu');
});

(GPU.isWebGLSupported ? test : skip)("inputYXOffsetPlus1 webgl", () => {
  inputYXOffsetPlus1('webgl');
});

(GPU.isWebGL2Supported ? test : skip)("inputYXOffsetPlus1 webgl2", () => {
  inputYXOffsetPlus1('webgl2');
});

(GPU.isHeadlessGLSupported ? test : skip)("inputYXOffsetPlus1 headlessgl", () => {
  inputYXOffsetPlus1('headlessgl');
});

test("inputYXOffsetPlus1 cpu", () => {
  inputYXOffsetPlus1('cpu');
});

function inputZYX(mode) {
  const gpu = new GPU({ mode });
  const kernel = gpu.createKernel(function(a) {
    return a[this.thread.z][this.thread.y][this.thread.x];
  })
    .setOutput([2, 4, 4]);

  const a = new Float32Array(64);
  a.set([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]);

  const result = kernel(input(a, [2, 4, 4]));
  assert.deepEqual(result.map(function(v) { return v.map(function(v) { return Array.from(v); }); }), [[[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]]]);
  gpu.destroy();
}

test("inputZYX auto", () => {
  inputZYX();
});

test("inputZYX gpu", () => {
  inputZYX('gpu');
});

(GPU.isWebGLSupported ? test : skip)("inputZYX webgl", () => {
  inputZYX('webgl');
});

(GPU.isWebGL2Supported ? test : skip)("inputZYX webgl2", () => {
  inputZYX('webgl2');
});

(GPU.isHeadlessGLSupported ? test : skip)("inputZYX headlessgl", () => {
  inputZYX('headlessgl');
});

test("inputZYX cpu", () => {
  inputZYX('cpu');
});


function inputZYXVariables(mode) {
  const gpu = new GPU({ mode });
  const kernel = gpu.createKernel(function(a, x, y, z) {
    return a[z][y][x];
  })
    .setOutput([1]);

  const a = new Float32Array(64);
  a.set([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]);
  const aInput = input(a, [2, 4, 4]);
  assert.deepEqual(Array.from(kernel(aInput, 1, 2, 3)), [30]);
  assert.deepEqual(Array.from(kernel(aInput, 0, 2, 3)), [29]);
  assert.deepEqual(Array.from(kernel(aInput, 0, 2, 1)), [13]);
  assert.deepEqual(Array.from(kernel(aInput, 1, 2, 2)), [22]);
  assert.deepEqual(Array.from(kernel(aInput, 0, 2, 2)), [21]);
  gpu.destroy();
}

test("inputZYXVariables auto", () => {
  inputZYXVariables();
});

test("inputZYXVariables gpu", () => {
  inputZYXVariables('gpu');
});

(GPU.isWebGLSupported ? test : skip)("inputZYXVariables webgl", () => {
  inputZYXVariables('webgl');
});

(GPU.isWebGL2Supported ? test : skip)("inputZYXVariables webgl2", () => {
  inputZYXVariables('webgl2');
});

(GPU.isHeadlessGLSupported ? test : skip)("inputZYXVariables headlessgl", () => {
  inputZYXVariables('headlessgl');
});

test("inputZYXVariables cpu", () => {
  inputZYXVariables('cpu');
});