fix: Add nested custom function support

Author: robertleeplummerjr

dist/gpu-browser-core.js

@@ -4,8 +4,8 @@
  *
  * GPU Accelerated JavaScript
  *
- * @version 2.0.4
- * @date Tue Oct 08 2019 10:27:37 GMT-0400 (Eastern Daylight Time)
+ * @version 2.0.5
+ * @date Fri Oct 11 2019 07:23:33 GMT-0400 (Eastern Daylight Time)
  *
  * @license MIT
  * The MIT License
@@ -2023,6 +2023,7 @@ class FunctionBuilder {
         triggerImplyArgumentType,
         triggerImplyArgumentBitRatio,
         onFunctionCall,
+        onNestedFunction,
       }));
     }
 
@@ -2392,6 +2393,7 @@ class FunctionBuilder {
 module.exports = {
   FunctionBuilder
 };
+
 },{}],9:[function(require,module,exports){
 const acorn = require('acorn');
 const { utils } = require('../utils');

dist/gpu-browser-core.min.js

@@ -4,8 +4,8 @@
  *
  * GPU Accelerated JavaScript
  *
- * @version 2.0.4
- * @date Tue Oct 08 2019 10:27:37 GMT-0400 (Eastern Daylight Time)
+ * @version 2.0.5
+ * @date Fri Oct 11 2019 07:23:33 GMT-0400 (Eastern Daylight Time)
  *
  * @license MIT
  * The MIT License
@@ -6787,6 +6787,7 @@ class FunctionBuilder {
         triggerImplyArgumentType,
         triggerImplyArgumentBitRatio,
         onFunctionCall,
+        onNestedFunction,
       }));
     }
 
@@ -7156,6 +7157,7 @@ class FunctionBuilder {
 module.exports = {
   FunctionBuilder
 };
+
 },{}],10:[function(require,module,exports){
 const acorn = require('acorn');
 const { utils } = require('../utils');

dist/gpu-browser.js

@@ -1,6 +1,6 @@
 {
   "name": "gpu.js",
-  "version": "2.0.4",
+  "version": "2.0.5",
   "description": "GPU Accelerated JavaScript",
   "engines": {
     "node": ">=8.0.0"

dist/gpu-browser.min.js

@@ -174,6 +174,7 @@ class FunctionBuilder {
         triggerImplyArgumentType,
         triggerImplyArgumentBitRatio,
         onFunctionCall,
+        onNestedFunction,
       }));
     }
 

package.json

@@ -50,3 +50,44 @@ test('nested_sum gpu', () => {
 test('nested_sum cpu', () => {
   nestedSumABTest('cpu');
 });
+
+function testNestedInCustomFunction(mode) {
+  function custom1() {
+    function nested1() {
+      return 1;
+    }
+
+    return nested1();
+  }
+  const gpu = new GPU({ mode });
+  gpu.addFunction(custom1);
+  const kernel = gpu.createKernel(function() {
+    return custom1();
+  }, { output: [1] });
+  assert.deepEqual(kernel(), new Float32Array([1]));
+  gpu.destroy();
+}
+
+test('nested in custom auto', () => {
+  testNestedInCustomFunction();
+});
+
+test('nested in custom gpu', () => {
+  testNestedInCustomFunction('gpu');
+});
+
+(GPU.isWebGLSupported ? test : skip)('nested in custom webgl', () => {
+  testNestedInCustomFunction('webgl');
+});
+
+(GPU.isWebGL2Supported ? test : skip)('nested in custom webgl2', () => {
+  testNestedInCustomFunction('webgl2');
+});
+
+(GPU.isHeadlessGLSupported ? test : skip)('nested in custom headlessgl', () => {
+  testNestedInCustomFunction('headlessgl');
+});
+
+test('nested in custom cpu', () => {
+  testNestedInCustomFunction('cpu');
+});