gpu.js/test/issues/390-thread-assignment.js

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

describe('issue #390');

test('Issue #390 - thread assignment webgl', function(assert) {
  const node = new WebGLFunctionNode(function assignThreadToVar() {
    const x = this.thread.x;
    const y = this.thread.y;
    const sum = x + y;
    return sum;
  }.toString(), { output: [1], returnType: 'Number' });
  assert.equal(node.toString(), 'float assignThreadToVar() {'
    + '\nfloat user_x=float(threadId.x);'
    + '\nfloat user_y=float(threadId.y);'
    + '\nfloat user_sum=(user_x+user_y);'
    + '\nreturn user_sum;'
    + '\n}');
  assert.equal(node.declarations.x.type, 'Number');
  assert.equal(node.declarations.y.type, 'Number');
  assert.equal(node.declarations.sum.type, 'Number');
});

test('Issue #390 - thread assignment webgl2', function(assert) {
  const node = new WebGL2FunctionNode(function assignThreadToVar() {
    const x = this.thread.x;
    const y = this.thread.y;
    const sum = x + y;
    return sum;
  }.toString(), { output: [1], returnType: 'Number' });
  assert.equal(node.toString(), 'float assignThreadToVar() {'
    + '\nfloat user_x=float(threadId.x);'
    + '\nfloat user_y=float(threadId.y);'
    + '\nfloat user_sum=(user_x+user_y);'
    + '\nreturn user_sum;'
    + '\n}');
  assert.equal(node.declarations.x.type, 'Number');
  assert.equal(node.declarations.y.type, 'Number');
  assert.equal(node.declarations.sum.type, 'Number');
});

test('Issue #390 - thread assignment cpu', function(assert) {
  const node = new CPUFunctionNode(function assignThreadToVar() {
    const x = this.thread.x;
    const y = this.thread.y;
    const sum = x + y;
    return sum;
  }.toString(), { output: [1] });
  assert.equal(node.toString(), 'function assignThreadToVar() {'
    + '\nconst user_x=_this.thread.x;'
    + '\nconst user_y=_this.thread.y;'
    + '\nconst user_sum=(user_x+user_y);'
    + '\nreturn user_sum;'
    + '\n}');
  assert.equal(node.declarations.x.type, 'Integer');
  assert.equal(node.declarations.y.type, 'Integer');
  assert.equal(node.declarations.sum.type, 'Integer');
});


test('Issue #390 (related) - output assignment webgl', function(assert) {
  const node = new WebGLFunctionNode(function assignThreadToVar() {
    const x = this.output.x;
    const y = this.output.y;
    const z = this.output.z;
    const sum = x + y + z;
    return sum;
  }.toString(), {
    output: [1,2,3]
  });
  assert.equal(node.toString(), 'float assignThreadToVar() {'
    + '\nfloat user_x=1.0;'
    + '\nfloat user_y=2.0;'
    + '\nfloat user_z=3.0;'
    + '\nfloat user_sum=((user_x+user_y)+user_z);'
    + '\nreturn user_sum;'
    + '\n}');
  assert.equal(node.declarations.x.type, 'Number');
  assert.equal(node.declarations.y.type, 'Number');
  assert.equal(node.declarations.z.type, 'Number');
  assert.equal(node.declarations.sum.type, 'Number');
});

test('Issue #390 (related) - output assignment webgl2', function(assert) {
  const node = new WebGL2FunctionNode(function assignThreadToVar() {
    const x = this.output.x;
    const y = this.output.y;
    const z = this.output.z;
    const sum = x + y + z;
    return sum;
  }.toString(), {
    output: [1,2,3]
  });
  assert.equal(node.toString(), 'float assignThreadToVar() {'
    + '\nfloat user_x=1.0;'
    + '\nfloat user_y=2.0;'
    + '\nfloat user_z=3.0;'
    + '\nfloat user_sum=((user_x+user_y)+user_z);'
    + '\nreturn user_sum;'
    + '\n}');
  assert.equal(node.declarations.x.type, 'Number');
  assert.equal(node.declarations.y.type, 'Number');
  assert.equal(node.declarations.z.type, 'Number');
  assert.equal(node.declarations.sum.type, 'Number');
});

test('Issue #390 (related) - output assignment cpu', function(assert) {
  const node = new CPUFunctionNode(function assignThreadToVar() {
    const x = this.output.x;
    const y = this.output.y;
    const z = this.output.z;
    const sum = x + y + z;
    return sum;
  }.toString(), {
    output: [1,2,3]
  });
  assert.equal(node.toString(), 'function assignThreadToVar() {'
    + '\nconst user_x=outputX;'
    + '\nconst user_y=outputY;'
    + '\nconst user_z=outputZ;'
    + '\nconst user_sum=((user_x+user_y)+user_z);'
    + '\nreturn user_sum;'
    + '\n}');
  assert.equal(node.declarations.x.type, 'Number');
  assert.equal(node.declarations.y.type, 'Number');
  assert.equal(node.declarations.z.type, 'Number');
  assert.equal(node.declarations.sum.type, 'Number');
});