gpu.js/test/features/function-return.js

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

describe('features: function return');

function functionReturnFloat( mode ) {
  const gpu = new GPU({ mode });
  const f = gpu.createKernel(function() {
    return 42;
  }, {
    output : [1]
  });
  assert.equal(f()[0], 42);
  gpu.destroy();
}

test('float auto', () => {
  functionReturnFloat(null);
});

test('float gpu', () => {
  functionReturnFloat('gpu');
});

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

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

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

test('float cpu', () => {
  functionReturnFloat('cpu');
});


function functionReturnArray2( mode ) {
  const gpu = new GPU({ mode });
  const f = gpu.createKernel(function() {
    return [42, 43];
  }, {
    output : [1]
  });
  const result = f();
  assert.equal(result[0].constructor, Float32Array);
  assert.equal(result[0][0], 42);
  assert.equal(result[0][1], 43);
  gpu.destroy();
}

(GPU.isSinglePrecisionSupported ? test : skip)('Array(2) auto', () => {
  functionReturnArray2(null);
});

(GPU.isSinglePrecisionSupported ? test : skip)('Array(2) gpu', () => {
  functionReturnArray2('gpu');
});

(GPU.isWebGLSupported && GPU.isSinglePrecisionSupported ? test : skip)('Array(2) webgl', () => {
  functionReturnArray2('webgl');
});

(GPU.isWebGL2Supported && GPU.isSinglePrecisionSupported ? test : skip)('Array(2) webgl2', () => {
  functionReturnArray2('webgl2');
});

(GPU.isHeadlessGLSupported && GPU.isSinglePrecisionSupported ? test : skip)('Array(2) headlessgl', () => {
  functionReturnArray2('headlessgl');
});

test('Array(2) cpu', () => {
  functionReturnArray2('cpu');
});

function functionReturnArray3( mode ) {
  const gpu = new GPU({ mode });
  const f = gpu.createKernel(function() {
    return [42, 43, 44];
  }, {
    output : [1]
  });
  const result = f();
  assert.equal(result[0].constructor, Float32Array);
  assert.equal(result[0][0], 42);
  assert.equal(result[0][1], 43);
  assert.equal(result[0][2], 44);
  gpu.destroy();
}

(GPU.isSinglePrecisionSupported ? test : skip)('Array(3) auto', () => {
  functionReturnArray3(null);
});

(GPU.isSinglePrecisionSupported ? test : skip)('Array(3) gpu', () => {
  functionReturnArray3('gpu');
});

(GPU.isWebGLSupported && GPU.isSinglePrecisionSupported ? test : skip)('Array(3) webgl', () => {
  functionReturnArray3('webgl');
});

(GPU.isWebGL2Supported && GPU.isSinglePrecisionSupported ? test : skip)('Array(3) webgl2', () => {
  functionReturnArray3('webgl2');
});

(GPU.isHeadlessGLSupported && GPU.isSinglePrecisionSupported ? test : skip)('Array(3) headlessgl', () => {
  functionReturnArray3('headlessgl');
});

test('Array(3) cpu', () => {
  functionReturnArray3('cpu');
});


function functionReturnArray4( mode ) {
  const gpu = new GPU({ mode });
  const f = gpu.createKernel(function() {
    return [42, 43, 44, 45];
  }, {
    output : [1]
  });

  const result = f();
  assert.equal(result[0].constructor, Float32Array);
  assert.equal(result[0][0], 42);
  assert.equal(result[0][1], 43);
  assert.equal(result[0][2], 44);
  assert.equal(result[0][3], 45);
  gpu.destroy();
}

(GPU.isSinglePrecisionSupported ? test : skip)('Array(4) auto', () => {
  functionReturnArray4(null);
});

(GPU.isSinglePrecisionSupported ? test : skip)('Array(4) gpu', () => {
  functionReturnArray4('gpu');
});

(GPU.isWebGLSupported && GPU.isSinglePrecisionSupported ? test : skip)('Array(4) webgl', () => {
  functionReturnArray4('webgl');
});

(GPU.isWebGL2Supported && GPU.isSinglePrecisionSupported ? test : skip)('Array(4) webgl2', () => {
  functionReturnArray4('webgl2');
});

(GPU.isHeadlessGLSupported && GPU.isSinglePrecisionSupported ? test : skip)('Array(4) headlessgl', () => {
  functionReturnArray4('headlessgl');
});

test('Array(4) cpu', () => {
  functionReturnArray4('cpu');
});