![\"Logo\"](\"http://gpu.rocks/static/media/jelly.3587de60.png\")
](http://gpu.rocks/)\n# GPU.js\nGPU.js is a JavaScript Acceleration library for GPGPU (General purpose computing on GPUs) in JavaScript for Web and Node.\nGPU.js automatically transpiles simple JavaScript functions into shader language and compiles them so they run on your GPU.\nIn case a GPU is not available, the functions will still run in regular JavaScript.\nFor some more quick concepts, see [Quick Concepts](https://github.com/gpujs/gpu.js/wiki/Quick-Concepts) on the wiki.\n\n\n[](https://gitter.im/gpujs/gpu.js?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge&utm_content=badge)\n[](https://slack.bri.im)\n\n# What is this sorcery?\n\nCreates a GPU accelerated kernel transpiled from a javascript function that computes a single element in the 512 x 512 matrix (2D array).\nThe kernel functions are ran in tandem on the GPU often resulting in very fast computations!\nYou can run a benchmark of this [here](http://gpu.rocks). Typically, it will run 1-15x faster depending on your hardware.\nMatrix multiplication (perform matrix multiplication on 2 matrices of size 512 x 512) written in GPU.js:\n\n## Browser\n```html\n\n\n```\n\n## CDN\n``` \nhttps://unpkg.com/gpu.js@latest/dist/gpu-browser.min.js\nhttps://cdn.jsdelivr.net/npm/gpu.js@latest/dist/gpu-browser.min.js\n```\n\n## Node\n```js\nconst { GPU } = require('gpu.js');\nconst gpu = new GPU();\nconst multiplyMatrix = gpu.createKernel(function(a, b) {\n let sum = 0;\n for (let i = 0; i < 512; i++) {\n sum += a[this.thread.y][i] * b[i][this.thread.x];\n }\n return sum;\n}).setOutput([512, 512]);\n\nconst c = multiplyMatrix(a, b);\n```\n\n## Typescript\n```typescript\nimport { GPU } from 'gpu.js';\nconst gpu = new GPU();\nconst multiplyMatrix = gpu.createKernel(function(a: number[][], b: number[][]) {\n let sum = 0;\n for (let i = 0; i < 512; i++) {\n sum += a[this.thread.y][i] * b[i][this.thread.x];\n }\n return sum;\n}).setOutput([512, 512]);\n\nconst c = multiplyMatrix(a, b) as number[][];\n```\n\n[Click here](/examples) for more typescript examples.\n\n# Table of Contents\n\nNotice documentation is off? We do try our hardest, but if you find something,\n [please bring it to our attention](https://github.com/gpujs/gpu.js/issues), or _[become a contributor](#contributors)_!\n\n* [Demos](#demos)\n* [Installation](#installation)\n* [`GPU` Settings](#gpu-settings)\n* [`gpu.createKernel` Settings](#gpucreatekernel-settings)\n * [Declaring variables/functions within kernels](#declaring-variablesfunctions-within-kernels)\n* [Creating and Running Functions](#creating-and-running-functions)\n* [Debugging](#debugging)\n* [Accepting Input](#accepting-input)\n* [Graphical Output](#graphical-output)\n* [Combining Kernels](#combining-kernels)\n* [Create Kernel Map](#create-kernel-map)\n* [Adding Custom Functions](#adding-custom-functions)\n* [Adding Custom Functions Directly to Kernel](#adding-custom-functions-directly-to-kernel)\n* [Types](#types)\n* [Loops](#loops)\n* [Pipelining](#pipelining)\n * [Cloning Textures](#cloning-textures-new-in-v2)\n * [Cleanup pipeline texture memory](#cleanup-pipeline-texture-memory-new-in-v24)\n* [Offscreen Canvas](#offscreen-canvas)\n* [Cleanup](#cleanup)\n* [Flattened typed array support](#flattened-typed-array-support)\n* [Precompiled and Lighter Weight Kernels](#precompiled-and-lighter-weight-kernels)\n * [using JSON](#using-json)\n * [Exporting kernel](#exporting-kernel)\n* [Supported Math functions](#supported-math-functions)\n* [How to check what is supported](#how-to-check-what-is-supported)\n* [Typescript Typings](#typescript-typings)\n* [Destructured Assignments](#destructured-assignments-new-in-v2)\n* [Dealing With Transpilation](#dealing-with-transpilation)\n* [Full API reference](#full-api-reference)\n* [How possible in node](#how-possible-in-node)\n* [Testing](#testing)\n* [Building](#building)\n* [Contributors](#contributors)\n* [Contributing](#contributing)\n* [Terms Explained](#terms-explained)\n* [License](#license)\n\n## Demos\nGPU.js in the wild, all around the net. Add yours here!\n* [Temperature interpolation using GPU.js](https://observablehq.com/@rveciana/temperature-interpolation-using-gpu-js)\n* [Julia Set Fractal using GPU.js](https://observablehq.com/@ukabuer/julia-set-fractal-using-gpu-js)\n* [Hello, gpu.js v2](https://observablehq.com/@fil/hello-gpu-js-v2)\n* [Basic gpu.js canvas example](https://observablehq.com/@rveciana/basic-gpu-js-canvas-example)\n* [Raster projection with GPU.js](https://observablehq.com/@fil/raster-projection-with-gpu-js)\n* [GPU.js Example: Slow Fade](https://observablehq.com/@robertleeplummerjr/gpu-js-example-slow-fade)\n* [GPU.JS CA Proof of Concept](https://observablehq.com/@alexlamb/gpu-js-ca-proof-of-concept)\n* [Image Convolution using GPU.js](https://observablehq.com/@ukabuer/image-convolution-using-gpu-js)\n* [Leaflet + gpu.js canvas](https://observablehq.com/@rveciana/leaflet-gpu-js-canvas)\n* [Image to GPU.js](https://observablehq.com/@fil/image-to-gpu)\n* [GPU Accelerated Heatmap using gpu.js](https://observablehq.com/@tracyhenry/gpu-accelerated-heatmap-using-gpu-js)\n* [Dijkstra’s algorithm in gpu.js](https://observablehq.com/@fil/dijkstras-algorithm-in-gpu-js)\n* [Voronoi with gpu.js](https://observablehq.com/@fil/voronoi-with-gpu-js)\n* [The gpu.js loop](https://observablehq.com/@fil/the-gpu-js-loop)\n* [GPU.js Example: Mandelbrot Set](https://observablehq.com/@robertleeplummerjr/gpu-js-example-mandelbrot-set)\n* [GPU.js Example: Mandelbulb](https://observablehq.com/@robertleeplummerjr/gpu-js-example-mandelbulb)\n* [Inverse of the distance with gpu.js](https://observablehq.com/@rveciana/inverse-of-the-distance-with-gpu-js)\n* [gpu.js laser detection v2](https://observablehq.com/@robertleeplummerjr/gpu-js-laser-detection-v2)\n* [GPU.js Canvas](https://observablehq.com/@hubgit/gpu-js-canvas)\n* [Video Convolution using GPU.js](https://observablehq.com/@robertleeplummerjr/video-convolution-using-gpu-js)\n* [GPU Rock Paper Scissors](https://observablehq.com/@alexlamb/gpu-rock-paper-scissors)\n* [Shaded relief with gpujs and d3js](https://observablehq.com/@rveciana/shaded-relief-with-gpujs-and-d3js/2)\n* [Caesar Cipher GPU.js Example](https://observablehq.com/@robertleeplummerjr/caesar-cipher-gpu-js-example)\n* [Matrix Multiplication GPU.js + Angular Example](https://ng-gpu.surge.sh/)\n* [Conway's game of life](https://observablehq.com/@brakdag/conway-game-of-life-gpu-js)\n\n## Installation\nOn Linux, ensure you have the correct header files installed: `sudo apt install mesa-common-dev libxi-dev` (adjust for your distribution)\n\n### npm\n\n```bash\nnpm install gpu.js --save\n```\n\n### yarn\n\n```bash\nyarn add gpu.js\n```\n\n[npm package](https://www.npmjs.com/package/gpu.js)\n### Node\n```js\nconst { GPU } = require('gpu.js');\nconst gpu = new GPU();\n```\n\n### Node Typescript **New in V2!**\n```js\nimport { GPU } from 'gpu.js';\nconst gpu = new GPU();\n```\n\n### Browser\n\nDownload the latest version of GPU.js and include the files in your HTML page using the following tags:\n\n```html\n\n\n```\n\n## `GPU` Settings\nSettings are an object used to create an instance of `GPU`. Example: `new GPU(settings)`\n* `canvas`: `HTMLCanvasElement`. Optional. For sharing canvas. Example: use THREE.js and GPU.js on same canvas.\n* `context`: `WebGL2RenderingContext` or `WebGLRenderingContext`. For sharing rendering context. Example: use THREE.js and GPU.js on same rendering context.\n* `mode`: Defaults to 'gpu', other values generally for debugging:\n * 'dev' **New in V2!**: VERY IMPORTANT! Use this so you can breakpoint and debug your kernel! This wraps your javascript in loops but DOES NOT transpile your code, so debugging is much easier.\n * 'webgl': Use the `WebGLKernel` for transpiling a kernel\n * 'webgl2': Use the `WebGL2Kernel` for transpiling a kernel\n * 'headlessgl' **New in V2!**: Use the `HeadlessGLKernel` for transpiling a kernel\n * 'cpu': Use the `CPUKernel` for transpiling a kernel\n* `onIstanbulCoverageVariable`: Removed in v2.11.0, use v8 coverage\n* `removeIstanbulCoverage`: Removed in v2.11.0, use v8 coverage\n\n## `gpu.createKernel` Settings\nSettings are an object used to create a `kernel` or `kernelMap`. Example: `gpu.createKernel(settings)`\n* `output` or `kernel.setOutput(output)`: `array` or `object` that describes the output of kernel. When using `kernel.setOutput()` you _can_ call it after the kernel has compiled if `kernel.dynamicOutput` is `true`, to resize your output. Example:\n * as array: `[width]`, `[width, height]`, or `[width, height, depth]`\n * as object: `{ x: width, y: height, z: depth }`\n* `pipeline` or `kernel.setPipeline(true)` **New in V2!**: boolean, default = `false`\n * Causes `kernel()` calls to output a `Texture`. To get array's from a `Texture`, use:\n ```js\n const result = kernel();\n result.toArray();\n ```\n * Can be passed _directly_ into kernels, and is preferred:\n ```js\n kernel(texture);\n ```\n* `graphical` or `kernel.setGraphical(boolean)`: boolean, default = `false`\n* `loopMaxIterations` or `kernel.setLoopMaxIterations(number)`: number, default = 1000\n* `constants` or `kernel.setConstants(object)`: object, default = null\n* `dynamicOutput` or `kernel.setDynamicOutput(boolean)`: boolean, default = false - turns dynamic output on or off\n* `dynamicArguments` or `kernel.setDynamicArguments(boolean)`: boolean, default = false - turns dynamic arguments (use different size arrays and textures) on or off\n* `optimizeFloatMemory` or `kernel.setOptimizeFloatMemory(boolean)` **New in V2!**: boolean - causes a float32 texture to use all 4 channels rather than 1, using less memory, but consuming more GPU.\n* `precision` or `kernel.setPrecision('unsigned' | 'single')` **New in V2!**: 'single' or 'unsigned' - if 'single' output texture uses float32 for each colour channel rather than 8\n* `fixIntegerDivisionAccuracy` or `kernel.setFixIntegerDivisionAccuracy(boolean)` : boolean - some cards have accuracy issues dividing by factors of three and some other primes (most apple kit?). Default on for affected cards, disable if accuracy not required.\n* `functions` or `kernel.setFunctions(array)`: array, array of functions to be used inside kernel. If undefined, inherits from `GPU` instance. Can also be an array of `{ source: function, argumentTypes: object, returnType: string }`.\n* `nativeFunctions` or `kernel.setNativeFunctions(array)`: object, defined as: `{ name: string, source: string, settings: object }`. This is generally set via using GPU.addNativeFunction()\n * VERY IMPORTANT! - Use this to add special native functions to your environment when you need specific functionality is needed.\n* `injectedNative` or `kernel.setInjectedNative(string)` **New in V2!**: string, defined as: `{ functionName: functionSource }`. This is for injecting native code before translated kernel functions.\n* `subKernels` or `kernel.setSubKernels(array)`: array, generally inherited from `GPU` instance.\n* `immutable` or `kernel.setImmutable(boolean)`: boolean, default = `false`\n * VERY IMPORTANT! - This was removed in v2.4.0 - v2.7.0, and brought back in v2.8.0 [by popular demand](https://github.com/gpujs/gpu.js/issues/572), please upgrade to get the feature\n* `strictIntegers` or `kernel.setStrictIntegers(boolean)`: boolean, default = `false` - allows undefined argumentTypes and function return values to use strict integer declarations.\n* `useLegacyEncoder` or `kernel.setUseLegacyEncoder(boolean)`: boolean, default `false` - more info [here](https://github.com/gpujs/gpu.js/wiki/Encoder-details).\n* `tactic` or `kernel.setTactic('speed' | 'balanced' | 'precision')` **New in V2!**: Set the kernel's tactic for compilation. Allows for compilation to better fit how GPU.js is being used (internally uses `lowp` for 'speed', `mediump` for 'balanced', and `highp` for 'precision'). Default is lowest resolution supported for output.\n\n\n## Creating and Running Functions\nDepending on your output type, specify the intended size of your output.\nYou cannot have an accelerated function that does not specify any output size.\n\nOutput size | How to specify output size | How to reference in kernel\n--------------|-------------------------------|--------------------------------\n 1D | `[length]` | `value[this.thread.x]`\n 2D | `[width, height]` | `value[this.thread.y][this.thread.x]`\n 3D | `[width, height, depth]` | `value[this.thread.z][this.thread.y][this.thread.x]`\n\n```js\nconst settings = {\n output: [100]\n};\n```\n\nor\n\n```js\n// You can also use x, y, and z\nconst settings = {\n output: { x: 100 }\n};\n```\n\nCreate the function you want to run on the GPU. The first input parameter to `createKernel` is a kernel function which will compute a single number in the output. The thread identifiers, `this.thread.x`, `this.thread.y` or `this.thread.z` will allow you to specify the appropriate behavior of the kernel function at specific positions of the output.\n\n```js\nconst kernel = gpu.createKernel(function() {\n return this.thread.x;\n}, settings);\n```\n\nThe created function is a regular JavaScript function, and you can use it like one.\n\n```js\nkernel();\n// Result: Float32Array[0, 1, 2, 3, ... 99]\n```\n\nNote: Instead of creating an object, you can use the chainable shortcut methods as a neater way of specifying settings.\n\n```js\nconst kernel = gpu.createKernel(function() {\n return this.thread.x;\n}).setOutput([100]);\n\nkernel();\n// Result: Float32Array[0, 1, 2, 3, ... 99]\n```\n\n### Declaring variables/functions within kernels\n\nGPU.js makes variable declaration inside kernel functions easy. Variable types supported are:\n* `Number` (Integer or Number), example: `let value = 1` or `let value = 1.1` \n* `Boolean`, example: `let value = true`\n* `Array(2)`, example: `let value = [1, 1]`\n* `Array(3)`, example: `let value = [1, 1, 1]`\n* `Array(4)`, example: `let value = [1, 1, 1, 1]`\n* `private Function`, example: `function myFunction(value) { return value + 1; }`\n\n`Number` kernel example:\n```js\nconst kernel = gpu.createKernel(function() {\n const i = 1;\n const j = 0.89;\n return i + j;\n}).setOutput([100]);\n```\n\n`Boolean` kernel example:\n```js\nconst kernel = gpu.createKernel(function() {\n const i = true;\n if (i) return 1;\n return 0;\n}).setOutput([100]);\n```\n\n`Array(2)` kernel examples:\nUsing declaration\n```js\nconst kernel = gpu.createKernel(function() {\n const array2 = [0.08, 2];\n return array2;\n}).setOutput([100]);\n```\n\nDirectly returned\n```js\nconst kernel = gpu.createKernel(function() {\n return [0.08, 2];\n}).setOutput([100]);\n```\n\n`Array(3)` kernel example:\nUsing declaration\n```js\nconst kernel = gpu.createKernel(function() {\n const array2 = [0.08, 2, 0.1];\n return array2;\n}).setOutput([100]);\n```\n\nDirectly returned\n```js\nconst kernel = gpu.createKernel(function() {\n return [0.08, 2, 0.1];\n}).setOutput([100]);\n```\n\n`Array(4)` kernel example:\nUsing declaration\n```js\nconst kernel = gpu.createKernel(function() {\n const array2 = [0.08, 2, 0.1, 3];\n return array2;\n}).setOutput([100]);\n```\n\nDirectly returned\n```js\nconst kernel = gpu.createKernel(function() {\n return [0.08, 2, 0.1, 3];\n}).setOutput([100]);\n```\n\n`private Function` kernel example:\n```js\nconst kernel = gpu.createKernel(function() {\n function myPrivateFunction() {\n return [0.08, 2, 0.1, 3];\n }\n \n return myPrivateFunction(); // <-- type inherited here\n}).setOutput([100]);\n```\n\n## Debugging\nDebugging can be done in a variety of ways, and there are different levels of debugging.\n* Debugging kernels with breakpoints can be done with `new GPU({ mode: 'dev' })`\n * This puts `GPU.js` into development mode. Here you can insert breakpoints, and be somewhat liberal in how your kernel is developed.\n * This mode _does not_ actually \"compile\" (parse, and eval) a kernel, it simply iterates on your code.\n * You can break a lot of rules here, because your kernel's function still has context of the state it came from.\n * PLEASE NOTE: Mapped kernels are not supported in this mode. They simply cannot work because of context.\n * Example:\n ```js\n const gpu = new GPU({ mode: 'dev' });\n const kernel = gpu.createKernel(function(arg1, time) {\n // put a breakpoint on the next line, and watch it get hit\n const v = arg1[this.thread.y][this.thread.x * time];\n return v;\n }, { output: [100, 100] });\n ```\n* Debugging actual kernels on CPU with `debugger`:\n * This will cause \"breakpoint\" like behaviour, but in an actual CPU kernel. You'll peer into the compiled kernel here, for a CPU.\n * Example:\n ```js\n const gpu = new GPU({ mode: 'cpu' });\n const kernel = gpu.createKernel(function(arg1, time) {\n debugger; // <--NOTICE THIS, IMPORTANT!\n const v = arg1[this.thread.y][this.thread.x * time];\n return v;\n }, { output: [100, 100] });\n ```\n* Debugging an actual GPU kernel:\n * There are no breakpoints available on the GPU, period. By providing the same level of abstraction and logic, the above methods should give you enough insight to debug, but sometimes we just need to see what is on the GPU.\n * Be VERY specific and deliberate, and use the kernel to your advantage, rather than just getting frustrated or giving up.\n * Example:\n ```js\n const gpu = new GPU({ mode: 'cpu' });\n const kernel = gpu.createKernel(function(arg1, time) {\n const x = this.thread.x * time;\n return x; // <--NOTICE THIS, IMPORTANT!\n const v = arg1[this.thread.y][x];\n return v;\n }, { output: [100, 100] });\n ```\n In this example, we return early the value of x, to see exactly what it is. The rest of the logic is ignored, but now you can see the value that is calculated from `x`, and debug it.\n This is an overly simplified problem.\n * Sometimes you need to solve graphical problems, that can be done similarly.\n * Example:\n ```js\n const gpu = new GPU({ mode: 'cpu' });\n const kernel = gpu.createKernel(function(arg1, time) {\n const x = this.thread.x * time;\n if (x < 4 || x > 2) {\n // RED\n this.color(1, 0, 0); // <--NOTICE THIS, IMPORTANT!\n return;\n }\n if (x > 6 && x < 12) {\n // GREEN\n this.color(0, 1, 0); // <--NOTICE THIS, IMPORTANT!\n return;\n }\n const v = arg1[this.thread.y][x];\n return v;\n }, { output: [100, 100], graphical: true });\n ```\n Here we are making the canvas red or green depending on the value of `x`.\n\n## Accepting Input\n### Supported Input Types\n* Numbers\n* 1d,2d, or 3d Array of numbers\n * Arrays of `Array`, `Float32Array`, `Int16Array`, `Int8Array`, `Uint16Array`, `uInt8Array`\n* Pre-flattened 2d or 3d Arrays using 'Input', for faster upload of arrays\n * Example:\n ```js\n const { input } = require('gpu.js');\n const value = input(flattenedArray, [width, height, depth]);\n ```\n* HTML Image\n* Array of HTML Images\n* Video Element **New in V2!**\nTo define an argument, simply add it to the kernel function like regular JavaScript.\n\n### Input Examples\n```js\nconst kernel = gpu.createKernel(function(x) {\n return x;\n}).setOutput([100]);\n\nkernel(42);\n// Result: Float32Array[42, 42, 42, 42, ... 42]\n```\n\nSimilarly, with array inputs:\n\n```js\nconst kernel = gpu.createKernel(function(x) {\n return x[this.thread.x % 3];\n}).setOutput([100]);\n\nkernel([1, 2, 3]);\n// Result: Float32Array[1, 2, 3, 1, ... 1 ]\n```\n\nAn HTML Image:\n\n```js\nconst kernel = gpu.createKernel(function(image) {\n const pixel = image[this.thread.y][this.thread.x];\n this.color(pixel[0], pixel[1], pixel[2], pixel[3]);\n})\n .setGraphical(true)\n .setOutput([100, 100]);\n\nconst image = document.createElement('img');\nimage.src = 'my/image/source.png';\nimage.onload = () => {\n kernel(image);\n // Result: colorful image\n \n document.getElementsByTagName('body')[0].appendChild(kernel.canvas);\n};\n```\n\nAn Array of HTML Images:\n\n```js\nconst kernel = gpu.createKernel(function(image) {\n const pixel = image[this.thread.z][this.thread.y][this.thread.x];\n this.color(pixel[0], pixel[1], pixel[2], pixel[3]);\n})\n .setGraphical(true)\n .setOutput([100, 100]);\n\nconst image1 = document.createElement('img');\nimage1.src = 'my/image/source1.png';\nimage1.onload = onload;\nconst image2 = document.createElement('img');\nimage2.src = 'my/image/source2.png';\nimage2.onload = onload;\nconst image3 = document.createElement('img');\nimage3.src = 'my/image/source3.png';\nimage3.onload = onload;\nconst totalImages = 3;\nlet loadedImages = 0;\nfunction onload() {\n loadedImages++;\n if (loadedImages === totalImages) {\n kernel([image1, image2, image3]);\n // Result: colorful image composed of many images\n\n document.getElementsByTagName('body')[0].appendChild(kernel.canvas);\n }\n};\n```\n\nAn HTML Video: **New in V2!**\n\n```js\nconst kernel = gpu.createKernel(function(videoFrame) {\n const pixel = videoFrame[this.thread.y][this.thread.x];\n this.color(pixel[0], pixel[1], pixel[2], pixel[3]);\n})\n .setGraphical(true)\n .setOutput([100, 100]);\n\nconst video = new document.createElement('video');\nvideo.src = 'my/video/source.webm';\nkernel(image); //note, try and use requestAnimationFrame, and the video should be ready or playing\n// Result: video frame\n```\n\n## Graphical Output\n\nSometimes, you want to produce a `canvas` image instead of doing numeric computations. To achieve this, set the `graphical` flag to `true` and the output dimensions to `[width, height]`. The thread identifiers will now refer to the `x` and `y` coordinate of the pixel you are producing. Inside your kernel function, use `this.color(r,g,b)` or `this.color(r,g,b,a)` to specify the color of the pixel.\n\nFor performance reasons, the return value of your function will no longer be anything useful. Instead, to display the image, retrieve the `canvas` DOM node and insert it into your page.\n\n```js\nconst render = gpu.createKernel(function() {\n this.color(0, 0, 0, 1);\n})\n .setOutput([20, 20])\n .setGraphical(true);\n\nrender();\n\nconst canvas = render.canvas;\ndocument.getElementsByTagName('body')[0].appendChild(canvas);\n```\n\nNote: To animate the rendering, use `requestAnimationFrame` instead of `setTimeout` for optimal performance. For more information, see [this](https://developer.mozilla.org/en-US/docs/Web/API/window/requestAnimationFrame).\n\n\n### .getPixels() **New in V2!**\nTo make it easier to get pixels from a context, use `kernel.getPixels()`, which returns a flat array similar to what you get from WebGL's `readPixels` method.\nA note on why: webgl's `readPixels` returns an array ordered differently from javascript's `getImageData`.\nThis makes them behave similarly.\nWhile the values may be somewhat different, because of graphical precision available in the kernel, and alpha, this allows us to easily get pixel data in unified way.\n\nExample:\n```js\nconst render = gpu.createKernel(function() {\n this.color(0, 0, 0, 1);\n})\n .setOutput([20, 20])\n .setGraphical(true);\n\nrender();\nconst pixels = render.getPixels();\n// [r,g,b,a, r,g,b,a...\n```\n\n### Alpha\n\nCurrently, if you need alpha do something like enabling `premultipliedAlpha` with your own gl context:\n```js\nconst canvas = DOM.canvas(500, 500);\nconst gl = canvas.getContext('webgl2', { premultipliedAlpha: false });\n\nconst gpu = new GPU({\n canvas,\n context: gl\n});\nconst krender = gpu.createKernel(function(x) {\n this.color(this.thread.x / 500, this.thread.y / 500, x[0], x[1]);\n})\n .setOutput([500, 500])\n .setGraphical(true);\n ```\n\n## Combining kernels\n\nSometimes you want to do multiple math operations on the gpu without the round trip penalty of data transfer from cpu to gpu to cpu to gpu, etc. To aid this there is the `combineKernels` method.\n_**Note:**_ Kernels can have different output sizes.\n```js\nconst add = gpu.createKernel(function(a, b) {\n return a[this.thread.x] + b[this.thread.x];\n}).setOutput([20]);\n\nconst multiply = gpu.createKernel(function(a, b) {\n return a[this.thread.x] * b[this.thread.x];\n}).setOutput([20]);\n\nconst superKernel = gpu.combineKernels(add, multiply, function(a, b, c) {\n return multiply(add(a, b), c);\n});\n\nsuperKernel(a, b, c);\n```\nThis gives you the flexibility of using multiple transformations but without the performance penalty, resulting in a much much MUCH faster operation.\n\n## Create Kernel Map\n\nSometimes you want to do multiple math operations in one kernel, and save the output of each of those operations. An example is **Machine Learning** where the previous output is required for back propagation. To aid this there is the `createKernelMap` method.\n\n### object outputs\n```js\nconst megaKernel = gpu.createKernelMap({\n addResult: function add(a, b) {\n return a + b;\n },\n multiplyResult: function multiply(a, b) {\n return a * b;\n },\n}, function(a, b, c) {\n return multiply(add(a[this.thread.x], b[this.thread.x]), c[this.thread.x]);\n}, { output: [10] });\n\nmegaKernel(a, b, c);\n// Result: { addResult: Float32Array, multiplyResult: Float32Array, result: Float32Array }\n```\n### array outputs\n```js\nconst megaKernel = gpu.createKernelMap([\n function add(a, b) {\n return a + b;\n },\n function multiply(a, b) {\n return a * b;\n }\n], function(a, b, c) {\n return multiply(add(a[this.thread.x], b[this.thread.x]), c[this.thread.x]);\n}, { output: [10] });\n\nmegaKernel(a, b, c);\n// Result: { 0: Float32Array, 1: Float32Array, result: Float32Array }\n```\nThis gives you the flexibility of using parts of a single transformation without the performance penalty, resulting in much much _MUCH_ faster operation.\n\n## Adding custom functions\n### To `GPU` instance\nuse `gpu.addFunction(function() {}, settings)` for adding custom functions to all kernels. Needs to be called BEFORE `gpu.createKernel`. Example:\n\n\n```js\ngpu.addFunction(function mySuperFunction(a, b) {\n return a - b;\n});\nfunction anotherFunction(value) {\n return value + 1;\n}\ngpu.addFunction(anotherFunction);\nconst kernel = gpu.createKernel(function(a, b) {\n return anotherFunction(mySuperFunction(a[this.thread.x], b[this.thread.x]));\n}).setOutput([20]);\n```\n\n### To `Kernel` instance\nuse `kernel.addFunction(function() {}, settings)` for adding custom functions to all kernels. Example:\n\n\n```js\nkernel.addFunction(function mySuperFunction(a, b) {\n return a - b;\n});\nfunction anotherFunction(value) {\n return value + 1;\n}\nkernel.addFunction(anotherFunction);\nconst kernel = gpu.createKernel(function(a, b) {\n return anotherFunction(mySuperFunction(a[this.thread.x], b[this.thread.x]));\n}).setOutput([20]);\n```\n\n### Adding strongly typed functions\n\nTo manually strongly type a function you may use settings.\nBy setting this value, it makes the build step of the kernel less resource intensive.\nSettings take an optional hash values:\n* `returnType`: optional, defaults to inference from `FunctionBuilder`, the value you'd like to return from the function.\n* `argumentTypes`: optional, defaults to inference from `FunctionBuilder` for each param, a hash of param names with values of the return types.\n\nExample on `GPU` instance:\n```js\ngpu.addFunction(function mySuperFunction(a, b) {\n return [a - b[1], b[0] - a];\n}, { argumentTypes: { a: 'Number', b: 'Array(2)'}, returnType: 'Array(2)' });\n```\n\nExample on `Kernel` instance:\n```js\nkernel.addFunction(function mySuperFunction(a, b) {\n return [a - b[1], b[0] - a];\n}, { argumentTypes: { a: 'Number', b: 'Array(2)'}, returnType: 'Array(2)' });\n```\n\nNOTE: GPU.js infers types if they are not defined and is generally able to detect the types you need, however\n'Array(2)', 'Array(3)', and 'Array(4)' are exceptions, at least on the kernel level. Also, it is nice to have power\nover the automatic type inference system.\n\n## Adding custom functions directly to kernel\n```js\nfunction mySuperFunction(a, b) {\n return a - b;\n}\nconst kernel = gpu.createKernel(function(a, b) {\n return mySuperFunction(a[this.thread.x], b[this.thread.x]);\n})\n .setOutput([20])\n .setFunctions([mySuperFunction]);\n\n```\n\n\n## Types\nGPU.js does type inference when types are not defined, so even if you code weak type, you are typing strongly typed.\nThis is needed because c++, which glsl is a subset of, is, of course, strongly typed.\nTypes that can be used with GPU.js are as follows:\n\n### Argument Types\n* 'Array'\n* 'Array(2)' **New in V2!**\n* 'Array(3)' **New in V2!**\n* 'Array(4)' **New in V2!**\n* 'Array1D(2)' **New in V2!**\n* 'Array1D(3)' **New in V2!**\n* 'Array1D(4)' **New in V2!**\n* 'Array2D(2)' **New in V2!**\n* 'Array2D(3)' **New in V2!**\n* 'Array2D(4)' **New in V2!**\n* 'Array3D(2)' **New in V2!**\n* 'Array3D(3)' **New in V2!**\n* 'Array3D(4)' **New in V2!**\n* 'HTMLCanvas' **New in V2.6**\n* 'OffscreenCanvas' **New in V2.13**\n* 'HTMLImage'\n* 'ImageBitmap' **New in V2.14**\n* 'ImageData' **New in V2.15**\n* 'HTMLImageArray'\n* 'HTMLVideo' **New in V2!**\n* 'Number'\n* 'Float'\n* 'Integer'\n* 'Boolean' **New in V2!**\n\n### Return Types\nNOTE: These refer the the return type of the kernel function, the actual result will always be a collection in the size of the defined `output`\n* 'Array(2)'\n* 'Array(3)'\n* 'Array(4)'\n* 'Number'\n* 'Float'\n* 'Integer'\n\n### Internal Types\nTypes generally used in the `Texture` class, for #pipelining or for advanced usage.\n* 'ArrayTexture(1)' **New in V2!**\n* 'ArrayTexture(2)' **New in V2!**\n* 'ArrayTexture(3)' **New in V2!**\n* 'ArrayTexture(4)' **New in V2!**\n* 'NumberTexture'\n* 'MemoryOptimizedNumberTexture' **New in V2!**\n\n## Loops\n* Any loops defined inside the kernel must have a maximum iteration count defined by the loopMaxIterations setting.\n* Other than defining the iterations by a constant or fixed value as shown [Dynamic sized via constants](dynamic-sized-via-constants), you can also simply pass the number of iterations as a variable to the kernel\n\n### Dynamic sized via constants\n```js\nconst matMult = gpu.createKernel(function(a, b) {\n var sum = 0;\n for (var i = 0; i < this.constants.size; i++) {\n sum += a[this.thread.y][i] * b[i][this.thread.x];\n }\n return sum;\n}, {\n constants: { size: 512 },\n output: [512, 512],\n});\n```\n\n### Fixed sized\n```js\nconst matMult = gpu.createKernel(function(a, b) {\n var sum = 0;\n for (var i = 0; i < 512; i++) {\n sum += a[this.thread.y][i] * b[i][this.thread.x];\n }\n return sum;\n}).setOutput([512, 512]);\n```\n\n## Pipelining\n[Pipeline](https://en.wikipedia.org/wiki/Pipeline_(computing)) is a feature where values are sent directly from kernel to kernel via a texture.\nThis results in extremely fast computing. This is achieved with the kernel setting `pipeline: boolean` or by calling `kernel.setPipeline(true)`\nIn an effort to make the CPU and GPU work similarly, pipeline on CPU and GPU modes causes the kernel result to be reused when `immutable: false` (which is default).\nIf you'd like to keep kernel results around, use `immutable: true` and ensure you cleanup memory:\n* In gpu mode using `texture.delete()` when appropriate.\n* In cpu mode allowing values to go out of context\n\n### Cloning Textures **New in V2!**\nWhen using pipeline mode the outputs from kernels can be cloned using `texture.clone()`.\n\n```js\nconst kernel1 = gpu.createKernel(function(v) {\n return v[this.thread.x];\n})\n .setPipeline(true)\n .setOutput([100]);\n\nconst kernel2 = gpu.createKernel(function(v) {\n return v[this.thread.x];\n})\n .setOutput([100]);\n\nconst result1 = kernel1(array);\n// Result: Texture\nconsole.log(result1.toArray());\n// Result: Float32Array[0, 1, 2, 3, ... 99]\n\nconst result2 = kernel2(result1);\n// Result: Float32Array[0, 1, 2, 3, ... 99]\n```\n\n### Cleanup pipeline texture memory **New in V2.4!**\nWhen using `kernel.immutable = true` recycling GPU memory is handled internally, but a good practice is to clean up memory you no longer need it.\nCleanup kernel outputs by using `texture.delete()` to keep GPU memory as small as possible.\n\nNOTE: Internally textures will only release from memory if there are no references to them.\nWhen using pipeline mode on a kernel `K` the output for each call will be a newly allocated texture `T`.\nIf, after getting texture `T` as an output, `T.delete()` is called, the next call to K will reuse `T` as its output texture.\n\nAlternatively, if you'd like to clear out a `texture` and yet keep it in memory, you may use `texture.clear()`, which\nwill cause the `texture` to persist in memory, but its internal values to become all zeros.\n\n## Offscreen Canvas\nGPU.js supports offscreen canvas where available. Here is an example of how to use it with two files, `gpu-worker.js`, and `index.js`:\n\nfile: `gpu-worker.js`\n```js\nimportScripts('path/to/gpu.js');\nonmessage = function() {\n // define gpu instance\n const gpu = new GPU();\n\n // input values\n const a = [1,2,3];\n const b = [3,2,1];\n\n // setup kernel\n const kernel = gpu.createKernel(function(a, b) {\n return a[this.thread.x] - b[this.thread.x];\n })\n .setOutput([3]);\n\n // output some results!\n postMessage(kernel(a, b));\n};\n```\n\nfile: `index.js`\n```js\nvar worker = new Worker('gpu-worker.js');\nworker.onmessage = function(e) {\n var result = e.data;\n console.log(result);\n};\n```\n\n## Cleanup\n* for instances of `GPU` use the `destroy` method. Example: `gpu.destroy()`\n* for instances of `Kernel` use the `destroy` method. Example: `kernel.destroy()`\n* for instances of `Texture` use the `delete` method. Example: `texture.delete()`\n* for instances of `Texture` that you might want to reuse/reset to zeros, use the `clear` method. Example: `texture.clear()`\n\n## Flattened typed array support\nTo use the useful `x`, `y`, `z` `thread` lookup api inside of GPU.js, and yet use flattened arrays, there is the `Input` type.\nThis is generally much faster for when sending values to the gpu, especially with larger data sets. Usage example:\n```js\nconst { GPU, input, Input } = require('gpu.js');\nconst gpu = new GPU();\nconst kernel = gpu.createKernel(function(a, b) {\n return a[this.thread.y][this.thread.x] + b[this.thread.y][this.thread.x];\n}).setOutput([3,3]);\n\n\nkernel(\n input(\n new Float32Array([1,2,3,4,5,6,7,8,9]),\n [3, 3]\n ),\n input(\n new Float32Array([1,2,3,4,5,6,7,8,9]),\n [3, 3]\n )\n);\n```\n\nNote: `input(value, size)` is a simple pointer for `new Input(value, size)`\n\n## Precompiled and Lighter Weight Kernels\n\n### using JSON\nGPU.js packs a lot of functionality into a single file, such as a complete javascript parse, which may not be needed in some cases.\nTo aid in keeping your kernels lightweight, the `kernel.toJSON()` method was added.\nThis allows you to reuse a previously built kernel, without the need to re-parse the javascript.\nHere is an example:\n\n```js\nconst gpu = new GPU();\nconst kernel = gpu.createKernel(function() {\n return [1,2,3,4];\n}, { output: [1] });\nconsole.log(kernel()); // [Float32Array([1,2,3,4])];\nconst json = kernel.toJSON();\nconst newKernelFromJson = gpu.createKernel(json);\nconsole.log(newKernelFromJSON()); // [Float32Array([1,2,3,4])];\n```\n\nNOTE: There is lighter weight, pre-built, version of GPU.js to assist with serializing from to and from json in the dist folder of the project, which include:\n* [dist/gpu-browser-core.js](dist/gpu-browser-core.js)\n* [dist/gpu-browser-core.min.js](dist/gpu-browser-core.min.js)\n\n### Exporting kernel\nGPU.js supports seeing exactly how it is interacting with the graphics processor by means of the `kernel.toString(...)` method.\nThis method, when called, creates a kernel that executes _exactly the instruction set given to the GPU (or CPU)_ *as a\nvery tiny reusable function* that instantiates a kernel.\n\nNOTE: When exporting a kernel and using `constants` the following constants are *not changeable*:\n* `Array(2)`\n* `Array(3)`\n* `Array(4)`\n* `Integer`\n* `Number`\n* `Float`\n* `Boolean`\n\nHere is an example used to/from file:\n```js\nimport { GPU } from 'gpu.js';\nimport * as fs from 'fs';\nconst gpu = new GPU();\nconst kernel = gpu.createKernel(function(v) {\n return this.thread.x + v + this.constants.v1;\n}, { output: [10], constants: { v1: 100 } });\nconst result = kernel(1);\nconst kernelString = kernel.toString(1);\nfs.writeFileSync('./my-exported-kernel.js', 'module.exports = ' + kernelString);\nimport * as MyExportedKernel from './my-exported-kernel';\nimport gl from 'gl';\nconst myExportedKernel = MyExportedKernel({ context: gl(1,1), constants: { v1: 100 } });\n```\n\n\nHere is an example for just-in-time function creation:\n\n```js\nconst gpu = new GPU();\nconst kernel = gpu.createKernel(function(a) {\n let sum = 0;\n for (let i = 0; i < 6; i++) {\n sum += a[this.thread.x][i];\n }\n return sum;\n }, { output: [6] });\nkernel(input(a, [6, 6]));\nconst kernelString = kernel.toString(input(a, [6, 6]));\nconst newKernel = new Function('return ' + kernelString)()({ context });\nnewKernel(input(a, [6, 6]));\n```\n\n#### using constants with `kernel.toString(...args)`\nYou can assign _some_ new constants when using the function output from `.toString()`,\n\n## Supported Math functions\n\nSince the code running in the kernel is actually compiled to GLSL code, not all functions from the JavaScript Math module are supported.\n\nThis is a list of the supported ones:\n\n* `Math.abs()`\n* `Math.acos()`\n* `Math.acosh()`\n* `Math.asin()`\n* `Math.asinh()`\n* `Math.atan()`\n* `Math.atanh()`\n* `Math.atan2()`\n* `Math.cbrt()`\n* `Math.ceil()`\n* `Math.cos()`\n* `Math.cosh()`\n* `Math.exp()`\n* `Math.expm1()`\n* `Math.floor()`\n* `Math.fround()`\n* `Math.imul()`\n* `Math.log()`\n* `Math.log10()`\n* `Math.log1p()`\n* `Math.log2()`\n* `Math.max()`\n* `Math.min()`\n* `Math.pow()`\n* `Math.random()`\n * A note on random. We use [a plugin](src/plugins/math-random-uniformly-distributed.js) to generate random.\n Random seeded _and_ generated, _both from the GPU_, is not as good as random from the CPU as there are more things that the CPU can seed random from.\n However, we seed random on the GPU, _from a random value in the CPU_.\n We then seed the subsequent randoms from the previous random value.\n So we seed from CPU, and generate from GPU.\n Which is still not as good as CPU, but closer.\n While this isn't perfect, it should suffice in most scenarios.\n In any case, we must give thanks to [RandomPower](https://www.randompower.eu/), and this [issue](https://github.com/gpujs/gpu.js/issues/498), for assisting in improving our implementation of random.\n* `Math.round()`\n* `Math.sign()`\n* `Math.sin()`\n* `Math.sinh()`\n* `Math.sqrt()`\n* `Math.tan()`\n* `Math.tanh()`\n* `Math.trunc()`\n\nThis is a list and reasons of unsupported ones:\n* `Math.clz32` - bits directly are hard\n* `Math.hypot` - dynamically sized\n\n## How to check what is supported\n\nTo assist with mostly unit tests, but perhaps in scenarios outside of GPU.js, there are the following logical checks to determine what support level the system executing a GPU.js kernel may have:\n* `GPU.disableValidation()` - turn off all kernel validation\n* `GPU.enableValidation()` - turn on all kernel validation\n* `GPU.isGPUSupported`: `boolean` - checks if GPU is in-fact supported\n* `GPU.isKernelMapSupported`: `boolean` - checks if kernel maps are supported\n* `GPU.isOffscreenCanvasSupported`: `boolean` - checks if offscreen canvas is supported\n* `GPU.isWebGLSupported`: `boolean` - checks if WebGL v1 is supported\n* `GPU.isWebGL2Supported`: `boolean` - checks if WebGL v2 is supported\n* `GPU.isHeadlessGLSupported`: `boolean` - checks if headlessgl is supported\n* `GPU.isCanvasSupported`: `boolean` - checks if canvas is supported\n* `GPU.isGPUHTMLImageArraySupported`: `boolean` - checks if the platform supports HTMLImageArray's\n* `GPU.isSinglePrecisionSupported`: `boolean` - checks if the system supports single precision float 32 values\n\n## Typescript Typings\nTypescript is supported! Typings can be found [here](src/index.d.ts)!\nFor strongly typed kernels:\n```typescript\nimport { GPU, IKernelFunctionThis } from 'gpu.js';\nconst gpu = new GPU();\n\nfunction kernelFunction(this: IKernelFunctionThis): number {\n return 1 + this.thread.x;\n}\n\nconst kernelMap = gpu.createKernel![](\"https://opencollective.com/gpujs/contributors.svg?width=890&button=false\")
\n\n\n### Backers\n\nThank you to all our backers! 🙏 [[Become a backer](https://opencollective.com/gpujs#backer)]\n\n![](\"https://opencollective.com/gpujs/backers.svg?width=890\")
\n\n\n### Sponsors\n\nSupport this project by becoming a sponsor. Your logo will show up here with a link to your website. [[Become a sponsor](https://opencollective.com/gpujs#sponsor)]\n\n\nSponsored NodeJS GPU environment from [LeaderGPU](https://www.leadergpu.com) - These guys rock!\n\n\nSponsored Browser GPU environment's from [BrowserStack](https://browserstack.com) - Second to none!\n\n![](\"https://opencollective.com/gpujs/sponsor/0/avatar.svg\"){kind=avatar}
\n![](\"https://opencollective.com/gpujs/sponsor/1/avatar.svg\"){kind=avatar}
\n![](\"https://opencollective.com/gpujs/sponsor/2/avatar.svg\"){kind=avatar}
\n![](\"https://opencollective.com/gpujs/sponsor/3/avatar.svg\"){kind=avatar}
\n![](\"https://opencollective.com/gpujs/sponsor/4/avatar.svg\"){kind=avatar}
\n![](\"https://opencollective.com/gpujs/sponsor/5/avatar.svg\"){kind=avatar}
\n![](\"https://opencollective.com/gpujs/sponsor/6/avatar.svg\"){kind=avatar}
\n![](\"https://opencollective.com/gpujs/sponsor/7/avatar.svg\"){kind=avatar}
\n![](\"https://opencollective.com/gpujs/sponsor/8/avatar.svg\"){kind=avatar}
\n![](\"https://opencollective.com/gpujs/sponsor/9/avatar.svg\"){kind=avatar}
\n\n## [License](LICENSE)\n"
},
{
"path": "dist/gpu-browser-core.js",
"content": "/**\n * gpu.js\n * http://gpu.rocks/\n *\n * GPU Accelerated JavaScript\n *\n * @version 2.16.0\n * @date Thu Feb 13 2025 11:46:48 GMT-0800 (Pacific Standard Time)\n *\n * @license MIT\n * The MIT License\n *\n * Copyright (c) 2025 gpu.js Team\n */(function(f){if(typeof exports===\"object\"&&typeof module!==\"undefined\"){module.exports=f()}else if(typeof define===\"function\"&&define.amd){define([],f)}else{var g;if(typeof window!==\"undefined\"){g=window}else if(typeof global!==\"undefined\"){g=global}else if(typeof self!==\"undefined\"){g=self}else{g=this}g.GPU = f()}})(function(){var define,module,exports;return (function(){function r(e,n,t){function o(i,f){if(!n[i]){if(!e[i]){var c=\"function\"==typeof require&&require;if(!f&&c)return c(i,!0);if(u)return u(i,!0);var a=new Error(\"Cannot find module '\"+i+\"'\");throw a.code=\"MODULE_NOT_FOUND\",a}var p=n[i]={exports:{}};e[i][0].call(p.exports,function(r){var n=e[i][1][r];return o(n||r)},p,p.exports,r,e,n,t)}return n[i].exports}for(var u=\"function\"==typeof require&&require,i=0;iImage to GPU.js from https://observablehq.com/@fil/image-to-gpu
\n\n\n\n\n" }, { "path": "examples/fluid.html", "content": "\n\n\n \nGPU.js Internal variable precision:
\n\n\n\n\n" }, { "path": "examples/json-saving.js", "content": "const { GPU } = require('../src');\nconst gpu1 = new GPU();\n\nconst kernel1 = gpu1.createKernel(function(value) {\n return value * 100;\n}, { output: [1] });\n\nconst resultFromRegularKernel = kernel1(42);\nconst json = kernel1.toJSON();\nconsole.log(resultFromRegularKernel);\n\n// Use bin/gpu-browser-core.js to get \"CORE\" mode, which works only with JSON\nconst gpu2 = new GPU();\nconst kernel2 = gpu2.createKernel(json);\nconst resultFromJsonKernel = kernel2(42);\nconsole.log(resultFromJsonKernel);\n\n" }, { "path": "examples/mandelbrot-set.html", "content": "\n\n\n\n\nLeft click to zoom in, right click to zoom out.
\n\n\n\n"
},
{
"path": "examples/mandelbulb.html",
"content": "\n\n\n \n \n \n \n \n
\n\n\n\n\n"
},
{
"path": "examples/parallel-raytracer.html",
"content": "\n\n\n \n \n \n Mandelbulb - Real-time 3d fractal
\n Kamil Kiełczewski Airavana\n
\nMandelbulb - Real-time 3d fractal
in Javascript
\n Kamil Kiełczewski Airavana\n \n
\n Instruction
\n Click on picture and move mouse around
\n A S D W E C keys for change view poin position
\n Push ESC to stop move and unlock mouse\n
\n \n A S D W E C keys for change view poin position
\n Push ESC to stop move and unlock mouse\n
\n
\n Calculate Light
\n \n \n
\n \n \n Level of Details (and camera speed)
\n Mouse whell or +/- keys\n\n
\n \n Mouse whell or +/- keys\n\n
\n \n
\n \n
\n\n\n\n\n\n\n"
},
{
"path": "examples/random.html",
"content": "\n\n\n \n \n
\n \n \n \n
\n \nA parallel raytracer built with TypeScript and GPU.js (WebGL/GLSL).
\nGitHub: http://github.com/jin/raytracer
\n\nPress W, A, S, D to move the camera around.
\n\n\n
Current mode
\n GPU\n \n \n\n
FPS
Loading..\n \n \n\n
Number of spheres
\n 4 \n \n\n\n
Grid dimension
\n 2 \n \n\n \n \n
\n The canvas is 640px by 640px. Each canvas object is controlled by a single GPU.js kernel and a single thread is spawned for each pixel to compute the color of the pixel.\n
\n\n\n Increase the dimensions of the grid to break the canvas up into tiles, so that there are multiple kernels controlling multiple tiles. With this approach, the kernels will run sequentially, computing one canvas after another.
\n
\n
Benchmark
\n\n \n
\n Benchmark the performance of the parallelized GPU and sequential CPU kernels.\n This will render 30 frames each and compute min, max, median frame rendering durations, and speedups. \n \n \n \n
CPU Random
\n \nWebGL1 Random
\n \nWebGL2 Random
\n \n\n\n\n\n" }, { "path": "examples/raster-globe/index.html", "content": "\n\n\n \nRaster projection with GPU.js from https://observablehq.com/d/b70d084526a1a764
\n\n\n\n\n" }, { "path": "examples/raytracer.html", "content": "\n\n\n \n \n \n\n
\n\n\n\n\n"
},
{
"path": "examples/simple-javascript.js",
"content": "const { GPU } = require('../src');\n\nconst gpu = new GPU({ mode: 'gpu' });\n\n// Look ma! I can javascript on my GPU!\nfunction kernelFunction(anInt, anArray, aNestedArray) {\n const x = .25 + anInt + anArray[this.thread.x] + aNestedArray[this.thread.x][this.thread.y];\n return x;\n}\n\nconst kernel = gpu.createKernel(kernelFunction, {\n output: [1]\n});\n\nconst result = kernel(1, [.25], [[1.5]]);\n\nconsole.log(result[0]); // 3\n"
},
{
"path": "examples/simple-typescript.ts",
"content": "import { GPU, KernelFunction, IKernelRunShortcut } from '../src';\n\nconst gpu = new GPU({ mode: 'gpu' });\n\n// Look ma! I can typescript on my GPU!\nconst kernelFunction: KernelFunction = function(anInt: number, anArray: number[], aNestedArray: number[][]) {\n const x = .25 + anInt + anArray[this.thread.x] + aNestedArray[this.thread.x][this.thread.y];\n return x;\n};\n\nconst kernel: IKernelRunShortcut = gpu.createKernel(kernelFunction)\n .setOutput([1]);\n\nconst result = kernel(1, [.25], [[1.5]]);\n\nconsole.log(result[0]); // 3\n"
},
{
"path": "examples/slow-fade.html",
"content": "\n\n\n \n \n
\n \n From https://staceytay.com/raytracer/. A simple ray tracer with Lambertian and specular reflection,\n built with GPU.js. Read more\n about ray tracing and GPU.js in\n my blog\n post. Code available\n on GitHub.\n
\n\n \n \n
\n \n
\n \n
\n \n \n
\n \n
\n \n fps\n
\n \nSlow Fade
\n\n \n
\n\n\n\n\n"
},
{
"path": "examples/video/index.html",
"content": "\n\n\n \n Video input (and output) with GPU.js
\n\n\n
\n
\n
\n
\n
\n\n\n
\n
\n
\n
\n
\n\n\n\n\n"
},
{
"path": "gulpfile.js",
"content": "const fs = require('fs');\nconst gulp = require('gulp');\nconst rename = require('gulp-rename');\nconst header = require('gulp-header');\nconst browserSync = require('browser-sync');\nconst browserify = require('browserify');\nconst replace = require('gulp-replace');\nconst source = require('vinyl-source-stream');\nconst buffer = require('vinyl-buffer');\nconst path = require('path');\nconst uglify = require('gulp-uglify-es').default;\nconst pkg = require('./package.json');\nconst jsprettify = require('gulp-jsbeautifier');\nconst stripComments = require('gulp-strip-comments');\nconst merge = require('ordered-read-streams');\nconst { readDirDeepSync } = require('read-dir-deep');\n\ngulp.task('build', () => {\n const gpu = browserify('./src/browser.js', { standalone: 'GPU', browserField: false })\n .ignore('gl')\n .bundle()\n .pipe(source('gpu-browser.js'))\n .pipe(buffer())\n .pipe(stripComments())\n .pipe(header(fs.readFileSync('./src/browser-header.txt', 'utf8'), { pkg : pkg }))\n .pipe(gulp.dest('./dist'))\n .on('error', console.error);\n\n const gpuCore = browserify('./src/browser.js', { standalone: 'GPU', browserField: false })\n .ignore('gl')\n .ignore('acorn')\n .bundle()\n .pipe(source('gpu-browser-core.js'))\n .pipe(buffer())\n .pipe(stripComments())\n .pipe(header(fs.readFileSync('./src/browser-header.txt', 'utf8'), { pkg : pkg }))\n .pipe(gulp.dest('./dist'))\n .on('error', console.error);\n\n return merge(gpu, gpuCore);\n});\n\n/// Minify the build script, after building it\ngulp.task('minify', () => {\n const gpu = gulp.src('dist/gpu-browser.js')\n .pipe(uglify())\n .pipe(rename('gpu-browser.min.js'))\n .pipe(header(fs.readFileSync('./src/browser-header.txt', 'utf8'), { pkg : pkg }))\n .pipe(gulp.dest('./dist'))\n .on('error', console.error);\n\n const gpuCore = gulp.src('dist/gpu-browser-core.js')\n .pipe(uglify())\n .pipe(rename('gpu-browser-core.min.js'))\n .pipe(header(fs.readFileSync('./src/browser-header.txt', 'utf8'), { pkg : pkg }))\n .pipe(gulp.dest('./dist'))\n .on('error', console.error);\n\n return merge(gpu, gpuCore);\n});\n\n/// The browser sync prototyping\ngulp.task('bsync', () => {\n // Syncs browser\n browserSync.init({\n server: {\n baseDir: './'\n },\n open: true,\n startPath: \"./test/html/test-all.html\",\n // Makes it easier to test on external mobile devices\n host: \"0.0.0.0\",\n tunnel: true\n });\n\n // Detect change -> rebuild TS\n gulp.watch(['src/**.js'], gulp.series('minify'));\n});\n\n/// Auto rebuild and host\ngulp.task('default', gulp.series('minify','bsync'));\n\n/// Beautify source code\n/// Use before merge request\ngulp.task('beautify', () => {\n return gulp.src(['src/**/*.js'])\n .pipe(jsprettify({\n indent_size: 2,\n indent_char: ' ',\n indent_with_tabs: false,\n eol: '\\n',\n brace_style: 'preserve-inline'\n }))\n .pipe(gulp.dest('src'));\n});\n\ngulp.task('build-tests', () => {\n const folder = 'test';\n const testFile = 'all.html';\n try {\n fs.unlinkSync(`${folder}/${testFile}`);\n } catch (e) {}\n const rootPath = path.resolve(process.cwd(), folder);\n const warning = '\\n';\n const files = readDirDeepSync(rootPath, {\n patterns: [\n '**/*.js'\n ],\n ignore: [\n '*.js'\n ]\n })\n .map(file => file.replace(/^test\\//, ''));\n return gulp.src(`${folder}/all-template.html`)\n .pipe(replace('{{test-files}}', warning + files.map(file => ``).join('\\n')))\n .pipe(rename(testFile))\n .pipe(gulp.dest(folder));\n});\n\ngulp.task('make', gulp.series('build', 'beautify', 'minify', 'build-tests'));\n\n"
},
{
"path": "package.json",
"content": "{\n \"name\": \"gpu.js\",\n \"version\": \"2.16.0\",\n \"description\": \"GPU Accelerated JavaScript\",\n \"engines\": {\n \"node\": \">=8.0.0\"\n },\n \"main\": \"./src/index.js\",\n \"files\": [\n \"src\",\n \"dist\"\n ],\n \"unpkg\": \"./dist/gpu-browser.js\",\n \"jsdelivr\": \"./dist/gpu-browser.js\",\n \"browser\": \"./dist/gpu-browser.js\",\n \"directories\": {\n \"doc\": \"doc\",\n \"test\": \"test\"\n },\n \"dependencies\": {\n \"acorn\": \"8.14.0\",\n \"gl\": \"8.1.6\",\n \"gl-wiretap\": \"0.6.2\",\n \"gpu-mock.js\": \"github:weagle08/gpu-mock.js\",\n \"ordered-read-streams\": \"^2.0.0\",\n \"webgpu\": \"0.2.9\"\n },\n \"devDependencies\": {\n \"benchmark\": \"2.1.4\",\n \"browser-sync\": \"3.0.2\",\n \"browserify\": \"17.0.0\",\n \"c8\": \"10.1.2\",\n \"gulp\": \"5.0.0\",\n \"gulp-concat\": \"2.6.1\",\n \"gulp-header\": \"2.0.9\",\n \"gulp-jsbeautifier\": \"3.0.1\",\n \"gulp-rename\": \"2.0.0\",\n \"gulp-replace\": \"1.1.4\",\n \"gulp-strip-comments\": \"2.6.0\",\n \"gulp-uglify-es\": \"3.0.0\",\n \"merge-stream\": \"2.0.0\",\n \"qunit\": \"2.22.0\",\n \"read-dir-deep\": \"8.0.0\",\n \"sinon\": \"18.0.0\",\n \"vinyl-buffer\": \"1.0.1\",\n \"vinyl-source-stream\": \"2.0.0\"\n },\n \"scripts\": {\n \"test\": \"qunit test/issues test/internal test/features\",\n \"coverage\": \"c8 qunit test/issues test/internal test/features\",\n \"setup\": \"npm i -g gulp-cli\",\n \"make\": \"gulp make\",\n \"build\": \"gulp build\",\n \"docs\": \"doxdox ./src --layout bootstrap --output docs.html\"\n },\n \"repository\": {\n \"type\": \"git\",\n \"url\": \"git+https://github.com/gpujs/gpu.js.git\"\n },\n \"keywords\": [\n \"gpgpu\",\n \"webgl\"\n ],\n \"author\": \"The gpu.js Team\",\n \"license\": \"MIT\",\n \"bugs\": {\n \"url\": \"https://github.com/gpujs/gpu.js/issues\"\n },\n \"homepage\": \"http://gpu.rocks/\",\n \"typings\": \"./src/index.d.ts\"\n}\n"
},
{
"path": "src/alias.js",
"content": "const { utils } = require('./utils');\n\n/**\n *\n * @param name\n * @param source\n * @returns {Function}\n */\nfunction alias(name, source) {\n const fnString = source.toString();\n return new Function(`return function ${ name } (${ utils.getArgumentNamesFromString(fnString).join(', ') }) {\n ${ utils.getFunctionBodyFromString(fnString) }\n}`)();\n}\n\nmodule.exports = {\n alias\n};"
},
{
"path": "src/backend/cpu/function-node.js",
"content": "const { FunctionNode } = require('../function-node');\n\n/**\n * @desc [INTERNAL] Represents a single function, inside JS\n *\n * \n
\nVideo
\n\n\n
\nGPU.js Graphical Output 0 fps
\nThis handles all the raw state, converted state, etc. Of a single function.
\n */\nclass CPUFunctionNode extends FunctionNode {\n /**\n * @desc Parses the abstract syntax tree for to its *named function*\n * @param {Object} ast - the AST object to parse\n * @param {Array} retArr - return array string\n * @returns {Array} the append retArr\n */\n astFunction(ast, retArr) {\n\n // Setup function return type and name\n if (!this.isRootKernel) {\n retArr.push('function');\n retArr.push(' ');\n retArr.push(this.name);\n retArr.push('(');\n\n // Arguments handling\n for (let i = 0; i < this.argumentNames.length; ++i) {\n const argumentName = this.argumentNames[i];\n\n if (i > 0) {\n retArr.push(', ');\n }\n retArr.push('user_');\n retArr.push(argumentName);\n }\n\n // Function opening\n retArr.push(') {\\n');\n }\n\n // Body statement iteration\n for (let i = 0; i < ast.body.body.length; ++i) {\n this.astGeneric(ast.body.body[i], retArr);\n retArr.push('\\n');\n }\n\n if (!this.isRootKernel) {\n // Function closing\n retArr.push('}\\n');\n }\n return retArr;\n }\n\n /**\n * @desc Parses the abstract syntax tree for to *return* statement\n * @param {Object} ast - the AST object to parse\n * @param {Array} retArr - return array string\n * @returns {Array} the append retArr\n */\n astReturnStatement(ast, retArr) {\n const type = this.returnType || this.getType(ast.argument);\n\n if (!this.returnType) {\n this.returnType = type;\n }\n\n if (this.isRootKernel) {\n retArr.push(this.leadingReturnStatement);\n this.astGeneric(ast.argument, retArr);\n retArr.push(';\\n');\n retArr.push(this.followingReturnStatement);\n retArr.push('continue;\\n');\n } else if (this.isSubKernel) {\n retArr.push(`subKernelResult_${ this.name } = `);\n this.astGeneric(ast.argument, retArr);\n retArr.push(';');\n retArr.push(`return subKernelResult_${ this.name };`);\n } else {\n retArr.push('return ');\n this.astGeneric(ast.argument, retArr);\n retArr.push(';');\n }\n return retArr;\n }\n\n /**\n * @desc Parses the abstract syntax tree for *literal value*\n * @param {Object} ast - the AST object to parse\n * @param {Array} retArr - return array string\n * @returns {Array} the append retArr\n */\n astLiteral(ast, retArr) {\n\n // Reject non numeric literals\n if (isNaN(ast.value)) {\n throw this.astErrorOutput(\n 'Non-numeric literal not supported : ' + ast.value,\n ast\n );\n }\n\n retArr.push(ast.value);\n\n return retArr;\n }\n\n /**\n * @desc Parses the abstract syntax tree for *binary* expression\n * @param {Object} ast - the AST object to parse\n * @param {Array} retArr - return array string\n * @returns {Array} the append retArr\n */\n astBinaryExpression(ast, retArr) {\n retArr.push('(');\n this.astGeneric(ast.left, retArr);\n retArr.push(ast.operator);\n this.astGeneric(ast.right, retArr);\n retArr.push(')');\n return retArr;\n }\n\n /**\n * @desc Parses the abstract syntax tree for *identifier* expression\n * @param {Object} idtNode - An ast Node\n * @param {Array} retArr - return array string\n * @returns {Array} the append retArr\n */\n astIdentifierExpression(idtNode, retArr) {\n if (idtNode.type !== 'Identifier') {\n throw this.astErrorOutput(\n 'IdentifierExpression - not an Identifier',\n idtNode\n );\n }\n\n switch (idtNode.name) {\n case 'Infinity':\n retArr.push('Infinity');\n break;\n default:\n if (this.constants && this.constants.hasOwnProperty(idtNode.name)) {\n retArr.push('constants_' + idtNode.name);\n } else {\n retArr.push('user_' + idtNode.name);\n }\n }\n\n return retArr;\n }\n\n /**\n * @desc Parses the abstract syntax tree for *for-loop* expression\n * @param {Object} forNode - An ast Node\n * @param {Array} retArr - return array string\n * @returns {Array} the parsed webgl string\n */\n astForStatement(forNode, retArr) {\n if (forNode.type !== 'ForStatement') {\n throw this.astErrorOutput('Invalid for statement', forNode);\n }\n\n const initArr = [];\n const testArr = [];\n const updateArr = [];\n const bodyArr = [];\n let isSafe = null;\n\n if (forNode.init) {\n this.pushState('in-for-loop-init');\n this.astGeneric(forNode.init, initArr);\n for (let i = 0; i < initArr.length; i++) {\n if (initArr[i].includes && initArr[i].includes(',')) {\n isSafe = false;\n }\n }\n this.popState('in-for-loop-init');\n } else {\n isSafe = false;\n }\n\n if (forNode.test) {\n this.astGeneric(forNode.test, testArr);\n } else {\n isSafe = false;\n }\n\n if (forNode.update) {\n this.astGeneric(forNode.update, updateArr);\n } else {\n isSafe = false;\n }\n\n if (forNode.body) {\n this.pushState('loop-body');\n this.astGeneric(forNode.body, bodyArr);\n this.popState('loop-body');\n }\n\n // have all parts, now make them safe\n if (isSafe === null) {\n isSafe = this.isSafe(forNode.init) && this.isSafe(forNode.test);\n }\n\n if (isSafe) {\n retArr.push(`for (${initArr.join('')};${testArr.join('')};${updateArr.join('')}){\\n`);\n retArr.push(bodyArr.join(''));\n retArr.push('}\\n');\n } else {\n const iVariableName = this.getInternalVariableName('safeI');\n if (initArr.length > 0) {\n retArr.push(initArr.join(''), ';\\n');\n }\n retArr.push(`for (let ${iVariableName}=0;${iVariableName}Instantiates properties to the CPU Kernel.
\n */\nclass CPUKernel extends Kernel {\n static getFeatures() {\n return this.features;\n }\n static get features() {\n return Object.freeze({\n kernelMap: true,\n isIntegerDivisionAccurate: true\n });\n }\n static get isSupported() {\n return true;\n }\n static isContextMatch(context) {\n return false;\n }\n /**\n * @desc The current mode in which gpu.js is executing.\n */\n static get mode() {\n return 'cpu';\n }\n\n static nativeFunctionArguments() {\n return null;\n }\n\n static nativeFunctionReturnType() {\n throw new Error(`Looking up native function return type not supported on ${this.name}`);\n }\n\n static combineKernels(combinedKernel) {\n return combinedKernel;\n }\n\n static getSignature(kernel, argumentTypes) {\n return 'cpu' + (argumentTypes.length > 0 ? ':' + argumentTypes.join(',') : '');\n }\n\n constructor(source, settings) {\n super(source, settings);\n this.mergeSettings(source.settings || settings);\n\n this._imageData = null;\n this._colorData = null;\n this._kernelString = null;\n this._prependedString = [];\n this.thread = {\n x: 0,\n y: 0,\n z: 0\n };\n this.translatedSources = null;\n }\n\n initCanvas() {\n if (typeof document !== 'undefined') {\n return document.createElement('canvas');\n } else if (typeof OffscreenCanvas !== 'undefined') {\n return new OffscreenCanvas(0, 0);\n }\n }\n\n initContext() {\n if (!this.canvas) return null;\n return this.canvas.getContext('2d');\n }\n\n initPlugins(settings) {\n return [];\n }\n\n /**\n * @desc Validate settings related to Kernel, such as dimensions size, and auto output support.\n * @param {IArguments} args\n */\n validateSettings(args) {\n if (!this.output || this.output.length === 0) {\n if (args.length !== 1) {\n throw new Error('Auto output only supported for kernels with only one input');\n }\n\n const argType = utils.getVariableType(args[0], this.strictIntegers);\n if (argType === 'Array') {\n this.output = utils.getDimensions(argType);\n } else if (argType === 'NumberTexture' || argType === 'ArrayTexture(4)') {\n this.output = args[0].output;\n } else {\n throw new Error('Auto output not supported for input type: ' + argType);\n }\n }\n\n if (this.graphical) {\n if (this.output.length !== 2) {\n throw new Error('Output must have 2 dimensions on graphical mode');\n }\n }\n\n this.checkOutput();\n }\n\n translateSource() {\n this.leadingReturnStatement = this.output.length > 1 ? 'resultX[x] = ' : 'result[x] = ';\n if (this.subKernels) {\n const followingReturnStatement = [];\n for (let i = 0; i < this.subKernels.length; i++) {\n const {\n name\n } = this.subKernels[i];\n followingReturnStatement.push(this.output.length > 1 ? `resultX_${ name }[x] = subKernelResult_${ name };\\n` : `result_${ name }[x] = subKernelResult_${ name };\\n`);\n }\n this.followingReturnStatement = followingReturnStatement.join('');\n }\n const functionBuilder = FunctionBuilder.fromKernel(this, CPUFunctionNode);\n this.translatedSources = functionBuilder.getPrototypes('kernel');\n if (!this.graphical && !this.returnType) {\n this.returnType = functionBuilder.getKernelResultType();\n }\n }\n\n /**\n * @desc Builds the Kernel, by generating the kernel\n * string using thread dimensions, and arguments\n * supplied to the kernel.\n *\n *If the graphical flag is enabled, canvas is used.
\n */\n build() {\n if (this.built) return;\n this.setupConstants();\n this.setupArguments(arguments);\n this.validateSettings(arguments);\n this.translateSource();\n\n if (this.graphical) {\n const {\n canvas,\n output\n } = this;\n if (!canvas) {\n throw new Error('no canvas available for using graphical output');\n }\n const width = output[0];\n const height = output[1] || 1;\n canvas.width = width;\n canvas.height = height;\n this._imageData = this.context.createImageData(width, height);\n this._colorData = new Uint8ClampedArray(width * height * 4);\n }\n\n const kernelString = this.getKernelString();\n this.kernelString = kernelString;\n\n if (this.debug) {\n console.log('Function output:');\n console.log(kernelString);\n }\n\n try {\n this.run = new Function([], kernelString).bind(this)();\n } catch (e) {\n console.error('An error occurred compiling the javascript: ', e);\n }\n this.buildSignature(arguments);\n this.built = true;\n }\n\n color(r, g, b, a) {\n if (typeof a === 'undefined') {\n a = 1;\n }\n\n r = Math.floor(r * 255);\n g = Math.floor(g * 255);\n b = Math.floor(b * 255);\n a = Math.floor(a * 255);\n\n const width = this.output[0];\n const height = this.output[1];\n\n const x = this.thread.x;\n const y = height - this.thread.y - 1;\n\n const index = x + y * width;\n\n this._colorData[index * 4 + 0] = r;\n this._colorData[index * 4 + 1] = g;\n this._colorData[index * 4 + 2] = b;\n this._colorData[index * 4 + 3] = a;\n }\n\n /**\n * @desc Generates kernel string for this kernel program.\n *\n *If sub-kernels are supplied, they are also factored in.\n * This string can be saved by calling the `toString` method\n * and then can be reused later.
\n *\n * @returns {String} result\n *\n */\n getKernelString() {\n if (this._kernelString !== null) return this._kernelString;\n\n let kernelThreadString = null;\n let {\n translatedSources\n } = this;\n if (translatedSources.length > 1) {\n translatedSources = translatedSources.filter(fn => {\n if (/^function/.test(fn)) return fn;\n kernelThreadString = fn;\n return false;\n });\n } else {\n kernelThreadString = translatedSources.shift();\n }\n return this._kernelString = ` const LOOP_MAX = ${ this._getLoopMaxString() };\n ${ this.injectedNative || '' }\n const _this = this;\n ${ this._resultKernelHeader() }\n ${ this._processConstants() }\n return (${ this.argumentNames.map(argumentName => 'user_' + argumentName).join(', ') }) => {\n ${ this._prependedString.join('') }\n ${ this._earlyThrows() }\n ${ this._processArguments() }\n ${ this.graphical ? this._graphicalKernelBody(kernelThreadString) : this._resultKernelBody(kernelThreadString) }\n ${ translatedSources.length > 0 ? translatedSources.join('\\n') : '' }\n };`;\n }\n\n /**\n * @desc Returns the *pre-compiled* Kernel as a JS Object String, that can be reused.\n */\n toString() {\n return cpuKernelString(this);\n }\n\n /**\n * @desc Get the maximum loop size String.\n * @returns {String} result\n */\n _getLoopMaxString() {\n return (\n this.loopMaxIterations ?\n ` ${ parseInt(this.loopMaxIterations) };` :\n ' 1000;'\n );\n }\n\n _processConstants() {\n if (!this.constants) return '';\n\n const result = [];\n for (let p in this.constants) {\n const type = this.constantTypes[p];\n switch (type) {\n case 'HTMLCanvas':\n case 'OffscreenCanvas':\n case 'HTMLImage':\n case 'ImageBitmap':\n case 'ImageData':\n case 'HTMLVideo':\n result.push(` const constants_${p} = this._mediaTo2DArray(this.constants.${p});\\n`);\n break;\n case 'HTMLImageArray':\n result.push(` const constants_${p} = this._imageTo3DArray(this.constants.${p});\\n`);\n break;\n case 'Input':\n result.push(` const constants_${p} = this.constants.${p}.value;\\n`);\n break;\n default:\n result.push(` const constants_${p} = this.constants.${p};\\n`);\n }\n }\n return result.join('');\n }\n\n _earlyThrows() {\n if (this.graphical) return '';\n if (this.immutable) return '';\n if (!this.pipeline) return '';\n const arrayArguments = [];\n for (let i = 0; i < this.argumentTypes.length; i++) {\n if (this.argumentTypes[i] === 'Array') {\n arrayArguments.push(this.argumentNames[i]);\n }\n }\n if (arrayArguments.length === 0) return '';\n const checks = [];\n for (let i = 0; i < arrayArguments.length; i++) {\n const argumentName = arrayArguments[i];\n const checkSubKernels = this._mapSubKernels(subKernel => `user_${argumentName} === result_${subKernel.name}`).join(' || ');\n checks.push(`user_${argumentName} === result${checkSubKernels ? ` || ${checkSubKernels}` : ''}`);\n }\n return `if (${checks.join(' || ')}) throw new Error('Source and destination arrays are the same. Use immutable = true');`;\n }\n\n _processArguments() {\n const result = [];\n for (let i = 0; i < this.argumentTypes.length; i++) {\n const variableName = `user_${this.argumentNames[i]}`;\n switch (this.argumentTypes[i]) {\n case 'HTMLCanvas':\n case 'OffscreenCanvas':\n case 'HTMLImage':\n case 'ImageBitmap':\n case 'ImageData':\n case 'HTMLVideo':\n result.push(` ${variableName} = this._mediaTo2DArray(${variableName});\\n`);\n break;\n case 'HTMLImageArray':\n result.push(` ${variableName} = this._imageTo3DArray(${variableName});\\n`);\n break;\n case 'Input':\n result.push(` ${variableName} = ${variableName}.value;\\n`);\n break;\n case 'ArrayTexture(1)':\n case 'ArrayTexture(2)':\n case 'ArrayTexture(3)':\n case 'ArrayTexture(4)':\n case 'NumberTexture':\n case 'MemoryOptimizedNumberTexture':\n result.push(`\n if (${variableName}.toArray) {\n if (!_this.textureCache) {\n _this.textureCache = [];\n _this.arrayCache = [];\n }\n const textureIndex = _this.textureCache.indexOf(${variableName});\n if (textureIndex !== -1) {\n ${variableName} = _this.arrayCache[textureIndex];\n } else {\n _this.textureCache.push(${variableName});\n ${variableName} = ${variableName}.toArray();\n _this.arrayCache.push(${variableName});\n }\n }`);\n break;\n }\n }\n return result.join('');\n }\n\n _mediaTo2DArray(media) {\n const canvas = this.canvas;\n const width = media.width > 0 ? media.width : media.videoWidth;\n const height = media.height > 0 ? media.height : media.videoHeight;\n if (canvas.width < width) {\n canvas.width = width;\n }\n if (canvas.height < height) {\n canvas.height = height;\n }\n const ctx = this.context;\n let pixelsData;\n if (media.constructor === ImageData) {\n pixelsData = media.data;\n } else {\n ctx.drawImage(media, 0, 0, width, height);\n pixelsData = ctx.getImageData(0, 0, width, height).data;\n }\n const imageArray = new Array(height);\n let index = 0;\n for (let y = height - 1; y >= 0; y--) {\n const row = imageArray[y] = new Array(width);\n for (let x = 0; x < width; x++) {\n const pixel = new Float32Array(4);\n pixel[0] = pixelsData[index++] / 255; // r\n pixel[1] = pixelsData[index++] / 255; // g\n pixel[2] = pixelsData[index++] / 255; // b\n pixel[3] = pixelsData[index++] / 255; // a\n row[x] = pixel;\n }\n }\n return imageArray;\n }\n\n /**\n *\n * @param flip\n * @return {Uint8ClampedArray}\n */\n getPixels(flip) {\n const [width, height] = this.output;\n // cpu is not flipped by default\n return flip ? utils.flipPixels(this._imageData.data, width, height) : this._imageData.data.slice(0);\n }\n\n _imageTo3DArray(images) {\n const imagesArray = new Array(images.length);\n for (let i = 0; i < images.length; i++) {\n imagesArray[i] = this._mediaTo2DArray(images[i]);\n }\n return imagesArray;\n }\n\n _resultKernelHeader() {\n if (this.graphical) return '';\n if (this.immutable) return '';\n if (!this.pipeline) return '';\n switch (this.output.length) {\n case 1:\n return this._mutableKernel1DResults();\n case 2:\n return this._mutableKernel2DResults();\n case 3:\n return this._mutableKernel3DResults();\n }\n }\n\n _resultKernelBody(kernelString) {\n switch (this.output.length) {\n case 1:\n return (!this.immutable && this.pipeline ? this._resultMutableKernel1DLoop(kernelString) : this._resultImmutableKernel1DLoop(kernelString)) + this._kernelOutput();\n case 2:\n return (!this.immutable && this.pipeline ? this._resultMutableKernel2DLoop(kernelString) : this._resultImmutableKernel2DLoop(kernelString)) + this._kernelOutput();\n case 3:\n return (!this.immutable && this.pipeline ? this._resultMutableKernel3DLoop(kernelString) : this._resultImmutableKernel3DLoop(kernelString)) + this._kernelOutput();\n default:\n throw new Error('unsupported size kernel');\n }\n }\n\n _graphicalKernelBody(kernelThreadString) {\n switch (this.output.length) {\n case 2:\n return this._graphicalKernel2DLoop(kernelThreadString) + this._graphicalOutput();\n default:\n throw new Error('unsupported size kernel');\n }\n }\n\n _graphicalOutput() {\n return `\n this._imageData.data.set(this._colorData);\n this.context.putImageData(this._imageData, 0, 0);\n return;`\n }\n\n _getKernelResultTypeConstructorString() {\n switch (this.returnType) {\n case 'LiteralInteger':\n case 'Number':\n case 'Integer':\n case 'Float':\n return 'Float32Array';\n case 'Array(2)':\n case 'Array(3)':\n case 'Array(4)':\n return 'Array';\n default:\n if (this.graphical) {\n return 'Float32Array';\n }\n throw new Error(`unhandled returnType ${ this.returnType }`);\n }\n }\n\n _resultImmutableKernel1DLoop(kernelString) {\n const constructorString = this._getKernelResultTypeConstructorString();\n return ` const outputX = _this.output[0];\n const result = new ${constructorString}(outputX);\n ${ this._mapSubKernels(subKernel => `const result_${ subKernel.name } = new ${constructorString}(outputX);\\n`).join(' ') }\n ${ this._mapSubKernels(subKernel => `let subKernelResult_${ subKernel.name };\\n`).join(' ') }\n for (let x = 0; x < outputX; x++) {\n this.thread.x = x;\n this.thread.y = 0;\n this.thread.z = 0;\n ${ kernelString }\n }`;\n }\n\n _mutableKernel1DResults() {\n const constructorString = this._getKernelResultTypeConstructorString();\n return ` const outputX = _this.output[0];\n const result = new ${constructorString}(outputX);\n ${ this._mapSubKernels(subKernel => `const result_${ subKernel.name } = new ${constructorString}(outputX);\\n`).join(' ') }\n ${ this._mapSubKernels(subKernel => `let subKernelResult_${ subKernel.name };\\n`).join(' ') }`;\n }\n\n _resultMutableKernel1DLoop(kernelString) {\n return ` const outputX = _this.output[0];\n for (let x = 0; x < outputX; x++) {\n this.thread.x = x;\n this.thread.y = 0;\n this.thread.z = 0;\n ${ kernelString }\n }`;\n }\n\n _resultImmutableKernel2DLoop(kernelString) {\n const constructorString = this._getKernelResultTypeConstructorString();\n return ` const outputX = _this.output[0];\n const outputY = _this.output[1];\n const result = new Array(outputY);\n ${ this._mapSubKernels(subKernel => `const result_${ subKernel.name } = new Array(outputY);\\n`).join(' ') }\n ${ this._mapSubKernels(subKernel => `let subKernelResult_${ subKernel.name };\\n`).join(' ') }\n for (let y = 0; y < outputY; y++) {\n this.thread.z = 0;\n this.thread.y = y;\n const resultX = result[y] = new ${constructorString}(outputX);\n ${ this._mapSubKernels(subKernel => `const resultX_${ subKernel.name } = result_${subKernel.name}[y] = new ${constructorString}(outputX);\\n`).join('') }\n for (let x = 0; x < outputX; x++) {\n this.thread.x = x;\n ${ kernelString }\n }\n }`;\n }\n\n _mutableKernel2DResults() {\n const constructorString = this._getKernelResultTypeConstructorString();\n return ` const outputX = _this.output[0];\n const outputY = _this.output[1];\n const result = new Array(outputY);\n ${ this._mapSubKernels(subKernel => `const result_${ subKernel.name } = new Array(outputY);\\n`).join(' ') }\n ${ this._mapSubKernels(subKernel => `let subKernelResult_${ subKernel.name };\\n`).join(' ') }\n for (let y = 0; y < outputY; y++) {\n const resultX = result[y] = new ${constructorString}(outputX);\n ${ this._mapSubKernels(subKernel => `const resultX_${ subKernel.name } = result_${subKernel.name}[y] = new ${constructorString}(outputX);\\n`).join('') }\n }`;\n }\n\n _resultMutableKernel2DLoop(kernelString) {\n const constructorString = this._getKernelResultTypeConstructorString();\n return ` const outputX = _this.output[0];\n const outputY = _this.output[1];\n for (let y = 0; y < outputY; y++) {\n this.thread.z = 0;\n this.thread.y = y;\n const resultX = result[y];\n ${ this._mapSubKernels(subKernel => `const resultX_${ subKernel.name } = result_${subKernel.name}[y] = new ${constructorString}(outputX);\\n`).join('') }\n for (let x = 0; x < outputX; x++) {\n this.thread.x = x;\n ${ kernelString }\n }\n }`;\n }\n\n _graphicalKernel2DLoop(kernelString) {\n return ` const outputX = _this.output[0];\n const outputY = _this.output[1];\n for (let y = 0; y < outputY; y++) {\n this.thread.z = 0;\n this.thread.y = y;\n for (let x = 0; x < outputX; x++) {\n this.thread.x = x;\n ${ kernelString }\n }\n }`;\n }\n\n _resultImmutableKernel3DLoop(kernelString) {\n const constructorString = this._getKernelResultTypeConstructorString();\n return ` const outputX = _this.output[0];\n const outputY = _this.output[1];\n const outputZ = _this.output[2];\n const result = new Array(outputZ);\n ${ this._mapSubKernels(subKernel => `const result_${ subKernel.name } = new Array(outputZ);\\n`).join(' ') }\n ${ this._mapSubKernels(subKernel => `let subKernelResult_${ subKernel.name };\\n`).join(' ') }\n for (let z = 0; z < outputZ; z++) {\n this.thread.z = z;\n const resultY = result[z] = new Array(outputY);\n ${ this._mapSubKernels(subKernel => `const resultY_${ subKernel.name } = result_${subKernel.name}[z] = new Array(outputY);\\n`).join(' ') }\n for (let y = 0; y < outputY; y++) {\n this.thread.y = y;\n const resultX = resultY[y] = new ${constructorString}(outputX);\n ${ this._mapSubKernels(subKernel => `const resultX_${ subKernel.name } = resultY_${subKernel.name}[y] = new ${constructorString}(outputX);\\n`).join(' ') }\n for (let x = 0; x < outputX; x++) {\n this.thread.x = x;\n ${ kernelString }\n }\n }\n }`;\n }\n\n _mutableKernel3DResults() {\n const constructorString = this._getKernelResultTypeConstructorString();\n return ` const outputX = _this.output[0];\n const outputY = _this.output[1];\n const outputZ = _this.output[2];\n const result = new Array(outputZ);\n ${ this._mapSubKernels(subKernel => `const result_${ subKernel.name } = new Array(outputZ);\\n`).join(' ') }\n ${ this._mapSubKernels(subKernel => `let subKernelResult_${ subKernel.name };\\n`).join(' ') }\n for (let z = 0; z < outputZ; z++) {\n const resultY = result[z] = new Array(outputY);\n ${ this._mapSubKernels(subKernel => `const resultY_${ subKernel.name } = result_${subKernel.name}[z] = new Array(outputY);\\n`).join(' ') }\n for (let y = 0; y < outputY; y++) {\n const resultX = resultY[y] = new ${constructorString}(outputX);\n ${ this._mapSubKernels(subKernel => `const resultX_${ subKernel.name } = resultY_${subKernel.name}[y] = new ${constructorString}(outputX);\\n`).join(' ') }\n }\n }`;\n }\n\n _resultMutableKernel3DLoop(kernelString) {\n return ` const outputX = _this.output[0];\n const outputY = _this.output[1];\n const outputZ = _this.output[2];\n for (let z = 0; z < outputZ; z++) {\n this.thread.z = z;\n const resultY = result[z];\n for (let y = 0; y < outputY; y++) {\n this.thread.y = y;\n const resultX = resultY[y];\n for (let x = 0; x < outputX; x++) {\n this.thread.x = x;\n ${ kernelString }\n }\n }\n }`;\n }\n\n _kernelOutput() {\n if (!this.subKernels) {\n return '\\n return result;';\n }\n return `\\n return {\n result: result,\n ${ this.subKernels.map(subKernel => `${ subKernel.property }: result_${ subKernel.name }`).join(',\\n ') }\n };`;\n }\n\n _mapSubKernels(fn) {\n return this.subKernels === null ? [''] :\n this.subKernels.map(fn);\n }\n\n destroy(removeCanvasReference) {\n if (removeCanvasReference) {\n delete this.canvas;\n }\n }\n\n static destroyContext(context) {}\n\n toJSON() {\n const json = super.toJSON();\n json.functionNodes = FunctionBuilder.fromKernel(this, CPUFunctionNode).toJSON();\n return json;\n }\n\n setOutput(output) {\n super.setOutput(output);\n const [width, height] = this.output;\n if (this.graphical) {\n this._imageData = this.context.createImageData(width, height);\n this._colorData = new Uint8ClampedArray(width * height * 4);\n }\n }\n\n prependString(value) {\n if (this._kernelString) throw new Error('Kernel already built');\n this._prependedString.push(value);\n }\n\n hasPrependString(value) {\n return this._prependedString.indexOf(value) > -1;\n }\n}\n\nmodule.exports = {\n CPUKernel\n};" }, { "path": "src/backend/function-builder.js", "content": "/**\n * @desc This handles all the raw state, converted state, etc. of a single function.\n * [INTERNAL] A collection of functionNodes.\n * @class\n */\nclass FunctionBuilder {\n /**\n *\n * @param {Kernel} kernel\n * @param {FunctionNode} FunctionNode\n * @param {object} [extraNodeOptions]\n * @returns {FunctionBuilder}\n * @static\n */\n static fromKernel(kernel, FunctionNode, extraNodeOptions) {\n const {\n kernelArguments,\n kernelConstants,\n argumentNames,\n argumentSizes,\n argumentBitRatios,\n constants,\n constantBitRatios,\n debug,\n loopMaxIterations,\n nativeFunctions,\n output,\n optimizeFloatMemory,\n precision,\n plugins,\n source,\n subKernels,\n functions,\n leadingReturnStatement,\n followingReturnStatement,\n dynamicArguments,\n dynamicOutput,\n } = kernel;\n\n const argumentTypes = new Array(kernelArguments.length);\n const constantTypes = {};\n\n for (let i = 0; i < kernelArguments.length; i++) {\n argumentTypes[i] = kernelArguments[i].type;\n }\n\n for (let i = 0; i < kernelConstants.length; i++) {\n const kernelConstant = kernelConstants[i];\n constantTypes[kernelConstant.name] = kernelConstant.type;\n }\n\n const needsArgumentType = (functionName, index) => {\n return functionBuilder.needsArgumentType(functionName, index);\n };\n\n const assignArgumentType = (functionName, index, type) => {\n functionBuilder.assignArgumentType(functionName, index, type);\n };\n\n const lookupReturnType = (functionName, ast, requestingNode) => {\n return functionBuilder.lookupReturnType(functionName, ast, requestingNode);\n };\n\n const lookupFunctionArgumentTypes = (functionName) => {\n return functionBuilder.lookupFunctionArgumentTypes(functionName);\n };\n\n const lookupFunctionArgumentName = (functionName, argumentIndex) => {\n return functionBuilder.lookupFunctionArgumentName(functionName, argumentIndex);\n };\n\n const lookupFunctionArgumentBitRatio = (functionName, argumentName) => {\n return functionBuilder.lookupFunctionArgumentBitRatio(functionName, argumentName);\n };\n\n const triggerImplyArgumentType = (functionName, i, argumentType, requestingNode) => {\n functionBuilder.assignArgumentType(functionName, i, argumentType, requestingNode);\n };\n\n const triggerImplyArgumentBitRatio = (functionName, argumentName, calleeFunctionName, argumentIndex) => {\n functionBuilder.assignArgumentBitRatio(functionName, argumentName, calleeFunctionName, argumentIndex);\n };\n\n const onFunctionCall = (functionName, calleeFunctionName, args) => {\n functionBuilder.trackFunctionCall(functionName, calleeFunctionName, args);\n };\n\n const onNestedFunction = (ast, source) => {\n const argumentNames = [];\n for (let i = 0; i < ast.params.length; i++) {\n argumentNames.push(ast.params[i].name);\n }\n const nestedFunction = new FunctionNode(source, Object.assign({}, nodeOptions, {\n returnType: null,\n ast,\n name: ast.id.name,\n argumentNames,\n lookupReturnType,\n lookupFunctionArgumentTypes,\n lookupFunctionArgumentName,\n lookupFunctionArgumentBitRatio,\n needsArgumentType,\n assignArgumentType,\n triggerImplyArgumentType,\n triggerImplyArgumentBitRatio,\n onFunctionCall,\n }));\n nestedFunction.traceFunctionAST(ast);\n functionBuilder.addFunctionNode(nestedFunction);\n };\n\n const nodeOptions = Object.assign({\n isRootKernel: false,\n onNestedFunction,\n lookupReturnType,\n lookupFunctionArgumentTypes,\n lookupFunctionArgumentName,\n lookupFunctionArgumentBitRatio,\n needsArgumentType,\n assignArgumentType,\n triggerImplyArgumentType,\n triggerImplyArgumentBitRatio,\n onFunctionCall,\n optimizeFloatMemory,\n precision,\n constants,\n constantTypes,\n constantBitRatios,\n debug,\n loopMaxIterations,\n output,\n plugins,\n dynamicArguments,\n dynamicOutput,\n }, extraNodeOptions || {});\n\n const rootNodeOptions = Object.assign({}, nodeOptions, {\n isRootKernel: true,\n name: 'kernel',\n argumentNames,\n argumentTypes,\n argumentSizes,\n argumentBitRatios,\n leadingReturnStatement,\n followingReturnStatement,\n });\n\n if (typeof source === 'object' && source.functionNodes) {\n return new FunctionBuilder().fromJSON(source.functionNodes, FunctionNode);\n }\n\n const rootNode = new FunctionNode(source, rootNodeOptions);\n\n let functionNodes = null;\n if (functions) {\n functionNodes = functions.map((fn) => new FunctionNode(fn.source, {\n returnType: fn.returnType,\n argumentTypes: fn.argumentTypes,\n output,\n plugins,\n constants,\n constantTypes,\n constantBitRatios,\n optimizeFloatMemory,\n precision,\n lookupReturnType,\n lookupFunctionArgumentTypes,\n lookupFunctionArgumentName,\n lookupFunctionArgumentBitRatio,\n needsArgumentType,\n assignArgumentType,\n triggerImplyArgumentType,\n triggerImplyArgumentBitRatio,\n onFunctionCall,\n onNestedFunction,\n }));\n }\n\n let subKernelNodes = null;\n if (subKernels) {\n subKernelNodes = subKernels.map((subKernel) => {\n const { name, source } = subKernel;\n return new FunctionNode(source, Object.assign({}, nodeOptions, {\n name,\n isSubKernel: true,\n isRootKernel: false,\n }));\n });\n }\n\n const functionBuilder = new FunctionBuilder({\n kernel,\n rootNode,\n functionNodes,\n nativeFunctions,\n subKernelNodes\n });\n\n return functionBuilder;\n }\n\n /**\n *\n * @param {IFunctionBuilderSettings} [settings]\n */\n constructor(settings) {\n settings = settings || {};\n this.kernel = settings.kernel;\n this.rootNode = settings.rootNode;\n this.functionNodes = settings.functionNodes || [];\n this.subKernelNodes = settings.subKernelNodes || [];\n this.nativeFunctions = settings.nativeFunctions || [];\n this.functionMap = {};\n this.nativeFunctionNames = [];\n this.lookupChain = [];\n this.functionNodeDependencies = {};\n this.functionCalls = {};\n\n if (this.rootNode) {\n this.functionMap['kernel'] = this.rootNode;\n }\n\n if (this.functionNodes) {\n for (let i = 0; i < this.functionNodes.length; i++) {\n this.functionMap[this.functionNodes[i].name] = this.functionNodes[i];\n }\n }\n\n if (this.subKernelNodes) {\n for (let i = 0; i < this.subKernelNodes.length; i++) {\n this.functionMap[this.subKernelNodes[i].name] = this.subKernelNodes[i];\n }\n }\n\n if (this.nativeFunctions) {\n for (let i = 0; i < this.nativeFunctions.length; i++) {\n const nativeFunction = this.nativeFunctions[i];\n this.nativeFunctionNames.push(nativeFunction.name);\n }\n }\n }\n\n /**\n * @desc Add the function node directly\n *\n * @param {FunctionNode} functionNode - functionNode to add\n *\n */\n addFunctionNode(functionNode) {\n if (!functionNode.name) throw new Error('functionNode.name needs set');\n this.functionMap[functionNode.name] = functionNode;\n if (functionNode.isRootKernel) {\n this.rootNode = functionNode;\n }\n }\n\n /**\n * @desc Trace all the depending functions being called, from a single function\n *\n * This allow for 'unneeded' functions to be automatically optimized out.\n * Note that the 0-index, is the starting function trace.\n *\n * @param {String} functionName - Function name to trace from, default to 'kernel'\n * @param {String[]} [retList] - Returning list of function names that is traced. Including itself.\n *\n * @returns {String[]} Returning list of function names that is traced. Including itself.\n */\n traceFunctionCalls(functionName, retList) {\n functionName = functionName || 'kernel';\n retList = retList || [];\n\n if (this.nativeFunctionNames.indexOf(functionName) > -1) {\n const nativeFunctionIndex = retList.indexOf(functionName);\n if (nativeFunctionIndex === -1) {\n retList.push(functionName);\n } else {\n /**\n * https://github.com/gpujs/gpu.js/issues/207\n * if dependent function is already in the list, because a function depends on it, and because it has\n * already been traced, we know that we must move the dependent function to the end of the the retList.\n * */\n const dependantNativeFunctionName = retList.splice(nativeFunctionIndex, 1)[0];\n retList.push(dependantNativeFunctionName);\n }\n return retList;\n }\n\n const functionNode = this.functionMap[functionName];\n if (functionNode) {\n // Check if function already exists\n const functionIndex = retList.indexOf(functionName);\n if (functionIndex === -1) {\n retList.push(functionName);\n functionNode.toString(); //ensure JS trace is done\n for (let i = 0; i < functionNode.calledFunctions.length; ++i) {\n this.traceFunctionCalls(functionNode.calledFunctions[i], retList);\n }\n } else {\n /**\n * https://github.com/gpujs/gpu.js/issues/207\n * if dependent function is already in the list, because a function depends on it, and because it has\n * already been traced, we know that we must move the dependent function to the end of the the retList.\n * */\n const dependantFunctionName = retList.splice(functionIndex, 1)[0];\n retList.push(dependantFunctionName);\n }\n }\n\n return retList;\n }\n\n /**\n * @desc Return the string for a function\n * @param {String} functionName - Function name to trace from. If null, it returns the WHOLE builder stack\n * @returns {String} The full string, of all the various functions. Trace optimized if functionName given\n */\n getPrototypeString(functionName) {\n return this.getPrototypes(functionName).join('\\n');\n }\n\n /**\n * @desc Return the string for a function\n * @param {String} [functionName] - Function name to trace from. If null, it returns the WHOLE builder stack\n * @returns {Array} The full string, of all the various functions. Trace optimized if functionName given\n */\n getPrototypes(functionName) {\n if (this.rootNode) {\n this.rootNode.toString();\n }\n if (functionName) {\n return this.getPrototypesFromFunctionNames(this.traceFunctionCalls(functionName, []).reverse());\n }\n return this.getPrototypesFromFunctionNames(Object.keys(this.functionMap));\n }\n\n /**\n * @desc Get string from function names\n * @param {String[]} functionList - List of function to build string\n * @returns {String} The string, of all the various functions. Trace optimized if functionName given\n */\n getStringFromFunctionNames(functionList) {\n const ret = [];\n for (let i = 0; i < functionList.length; ++i) {\n const node = this.functionMap[functionList[i]];\n if (node) {\n ret.push(this.functionMap[functionList[i]].toString());\n }\n }\n return ret.join('\\n');\n }\n\n /**\n * @desc Return string of all functions converted\n * @param {String[]} functionList - List of function names to build the string.\n * @returns {Array} Prototypes of all functions converted\n */\n getPrototypesFromFunctionNames(functionList) {\n const ret = [];\n for (let i = 0; i < functionList.length; ++i) {\n const functionName = functionList[i];\n const functionIndex = this.nativeFunctionNames.indexOf(functionName);\n if (functionIndex > -1) {\n ret.push(this.nativeFunctions[functionIndex].source);\n continue;\n }\n const node = this.functionMap[functionName];\n if (node) {\n ret.push(node.toString());\n }\n }\n return ret;\n }\n\n toJSON() {\n return this.traceFunctionCalls(this.rootNode.name).reverse().map(name => {\n const nativeIndex = this.nativeFunctions.indexOf(name);\n if (nativeIndex > -1) {\n return {\n name,\n source: this.nativeFunctions[nativeIndex].source\n };\n } else if (this.functionMap[name]) {\n return this.functionMap[name].toJSON();\n } else {\n throw new Error(`function ${ name } not found`);\n }\n });\n }\n\n fromJSON(jsonFunctionNodes, FunctionNode) {\n this.functionMap = {};\n for (let i = 0; i < jsonFunctionNodes.length; i++) {\n const jsonFunctionNode = jsonFunctionNodes[i];\n this.functionMap[jsonFunctionNode.settings.name] = new FunctionNode(jsonFunctionNode.ast, jsonFunctionNode.settings);\n }\n return this;\n }\n\n /**\n * @desc Get string for a particular function name\n * @param {String} functionName - Function name to trace from. If null, it returns the WHOLE builder stack\n * @returns {String} settings - The string, of all the various functions. Trace optimized if functionName given\n */\n getString(functionName) {\n if (functionName) {\n return this.getStringFromFunctionNames(this.traceFunctionCalls(functionName).reverse());\n }\n return this.getStringFromFunctionNames(Object.keys(this.functionMap));\n }\n\n lookupReturnType(functionName, ast, requestingNode) {\n if (ast.type !== 'CallExpression') {\n throw new Error(`expected ast type of \"CallExpression\", but is ${ ast.type }`);\n }\n if (this._isNativeFunction(functionName)) {\n return this._lookupNativeFunctionReturnType(functionName);\n } else if (this._isFunction(functionName)) {\n const node = this._getFunction(functionName);\n if (node.returnType) {\n return node.returnType;\n } else {\n for (let i = 0; i < this.lookupChain.length; i++) {\n // detect circlical logic\n if (this.lookupChain[i].ast === ast) {\n // detect if arguments have not resolved, preventing a return type\n // if so, go ahead and resolve them, so we can resolve the return type\n if (node.argumentTypes.length === 0 && ast.arguments.length > 0) {\n const args = ast.arguments;\n for (let j = 0; j < args.length; j++) {\n this.lookupChain.push({\n name: requestingNode.name,\n ast: args[i],\n requestingNode\n });\n node.argumentTypes[j] = requestingNode.getType(args[j]);\n this.lookupChain.pop();\n }\n return node.returnType = node.getType(node.getJsAST());\n }\n\n throw new Error('circlical logic detected!');\n }\n }\n // get ready for a ride!\n this.lookupChain.push({\n name: requestingNode.name,\n ast,\n requestingNode\n });\n const type = node.getType(node.getJsAST());\n this.lookupChain.pop();\n return node.returnType = type;\n }\n }\n\n return null;\n }\n\n /**\n *\n * @param {String} functionName\n * @return {FunctionNode}\n * @private\n */\n _getFunction(functionName) {\n if (!this._isFunction(functionName)) {\n new Error(`Function ${functionName} not found`);\n }\n return this.functionMap[functionName];\n }\n\n _isFunction(functionName) {\n return Boolean(this.functionMap[functionName]);\n }\n\n _getNativeFunction(functionName) {\n for (let i = 0; i < this.nativeFunctions.length; i++) {\n if (this.nativeFunctions[i].name === functionName) return this.nativeFunctions[i];\n }\n return null;\n }\n\n _isNativeFunction(functionName) {\n return Boolean(this._getNativeFunction(functionName));\n }\n\n _lookupNativeFunctionReturnType(functionName) {\n let nativeFunction = this._getNativeFunction(functionName);\n if (nativeFunction) {\n return nativeFunction.returnType;\n }\n throw new Error(`Native function ${ functionName } not found`);\n }\n\n lookupFunctionArgumentTypes(functionName) {\n if (this._isNativeFunction(functionName)) {\n return this._getNativeFunction(functionName).argumentTypes;\n } else if (this._isFunction(functionName)) {\n return this._getFunction(functionName).argumentTypes;\n }\n return null;\n }\n\n lookupFunctionArgumentName(functionName, argumentIndex) {\n return this._getFunction(functionName).argumentNames[argumentIndex];\n }\n\n /**\n *\n * @param {string} functionName\n * @param {string} argumentName\n * @return {number}\n */\n lookupFunctionArgumentBitRatio(functionName, argumentName) {\n if (!this._isFunction(functionName)) {\n throw new Error('function not found');\n }\n if (this.rootNode.name === functionName) {\n const i = this.rootNode.argumentNames.indexOf(argumentName);\n if (i !== -1) {\n return this.rootNode.argumentBitRatios[i];\n }\n }\n const node = this._getFunction(functionName);\n const i = node.argumentNames.indexOf(argumentName);\n if (i === -1) {\n throw new Error('argument not found');\n }\n const bitRatio = node.argumentBitRatios[i];\n if (typeof bitRatio !== 'number') {\n throw new Error('argument bit ratio not found');\n }\n return bitRatio;\n }\n\n needsArgumentType(functionName, i) {\n if (!this._isFunction(functionName)) return false;\n const fnNode = this._getFunction(functionName);\n return !fnNode.argumentTypes[i];\n }\n\n assignArgumentType(functionName, i, argumentType, requestingNode) {\n if (!this._isFunction(functionName)) return;\n const fnNode = this._getFunction(functionName);\n if (!fnNode.argumentTypes[i]) {\n fnNode.argumentTypes[i] = argumentType;\n }\n }\n\n /**\n * @param {string} functionName\n * @param {string} argumentName\n * @param {string} calleeFunctionName\n * @param {number} argumentIndex\n * @return {number|null}\n */\n assignArgumentBitRatio(functionName, argumentName, calleeFunctionName, argumentIndex) {\n const node = this._getFunction(functionName);\n if (this._isNativeFunction(calleeFunctionName)) return null;\n const calleeNode = this._getFunction(calleeFunctionName);\n const i = node.argumentNames.indexOf(argumentName);\n if (i === -1) {\n throw new Error(`Argument ${argumentName} not found in arguments from function ${functionName}`);\n }\n const bitRatio = node.argumentBitRatios[i];\n if (typeof bitRatio !== 'number') {\n throw new Error(`Bit ratio for argument ${argumentName} not found in function ${functionName}`);\n }\n if (!calleeNode.argumentBitRatios) {\n calleeNode.argumentBitRatios = new Array(calleeNode.argumentNames.length);\n }\n const calleeBitRatio = calleeNode.argumentBitRatios[i];\n if (typeof calleeBitRatio === 'number') {\n if (calleeBitRatio !== bitRatio) {\n throw new Error(`Incompatible bit ratio found at function ${functionName} at argument ${argumentName}`);\n }\n return calleeBitRatio;\n }\n calleeNode.argumentBitRatios[i] = bitRatio;\n return bitRatio;\n }\n\n trackFunctionCall(functionName, calleeFunctionName, args) {\n if (!this.functionNodeDependencies[functionName]) {\n this.functionNodeDependencies[functionName] = new Set();\n this.functionCalls[functionName] = [];\n }\n this.functionNodeDependencies[functionName].add(calleeFunctionName);\n this.functionCalls[functionName].push(args);\n }\n\n getKernelResultType() {\n return this.rootNode.returnType || this.rootNode.getType(this.rootNode.ast);\n }\n\n getSubKernelResultType(index) {\n const subKernelNode = this.subKernelNodes[index];\n let called = false;\n for (let functionCallIndex = 0; functionCallIndex < this.rootNode.functionCalls.length; functionCallIndex++) {\n const functionCall = this.rootNode.functionCalls[functionCallIndex];\n if (functionCall.ast.callee.name === subKernelNode.name) {\n called = true;\n }\n }\n if (!called) {\n throw new Error(`SubKernel ${ subKernelNode.name } never called by kernel`);\n }\n return subKernelNode.returnType || subKernelNode.getType(subKernelNode.getJsAST());\n }\n\n getReturnTypes() {\n const result = {\n [this.rootNode.name]: this.rootNode.getType(this.rootNode.ast),\n };\n const list = this.traceFunctionCalls(this.rootNode.name);\n for (let i = 0; i < list.length; i++) {\n const functionName = list[i];\n const functionNode = this.functionMap[functionName];\n result[functionName] = functionNode.getType(functionNode.ast);\n }\n return result;\n }\n}\n\nmodule.exports = {\n FunctionBuilder\n};" }, { "path": "src/backend/function-node.js", "content": "const acorn = require('acorn');\nconst { utils } = require('../utils');\nconst { FunctionTracer } = require('./function-tracer');\n\n/**\n *\n * @desc Represents a single function, inside JS, webGL, or openGL.\n *This handles all the raw state, converted state, etc. Of a single function.
\n */\nclass FunctionNode {\n /**\n *\n * @param {string|object} source\n * @param {IFunctionSettings} [settings]\n */\n constructor(source, settings) {\n if (!source && !settings.ast) {\n throw new Error('source parameter is missing');\n }\n settings = settings || {};\n this.source = source;\n this.ast = null;\n this.name = typeof source === 'string' ? settings.isRootKernel ?\n 'kernel' :\n (settings.name || utils.getFunctionNameFromString(source)) : null;\n this.calledFunctions = [];\n this.constants = {};\n this.constantTypes = {};\n this.constantBitRatios = {};\n this.isRootKernel = false;\n this.isSubKernel = false;\n this.debug = null;\n this.functions = null;\n this.identifiers = null;\n this.contexts = null;\n this.functionCalls = null;\n this.states = [];\n this.needsArgumentType = null;\n this.assignArgumentType = null;\n this.lookupReturnType = null;\n this.lookupFunctionArgumentTypes = null;\n this.lookupFunctionArgumentBitRatio = null;\n this.triggerImplyArgumentType = null;\n this.triggerImplyArgumentBitRatio = null;\n this.onNestedFunction = null;\n this.onFunctionCall = null;\n this.optimizeFloatMemory = null;\n this.precision = null;\n this.loopMaxIterations = null;\n this.argumentNames = (typeof this.source === 'string' ? utils.getArgumentNamesFromString(this.source) : null);\n this.argumentTypes = [];\n this.argumentSizes = [];\n this.argumentBitRatios = null;\n this.returnType = null;\n this.output = [];\n this.plugins = null;\n this.leadingReturnStatement = null;\n this.followingReturnStatement = null;\n this.dynamicOutput = null;\n this.dynamicArguments = null;\n this.strictTypingChecking = false;\n this.fixIntegerDivisionAccuracy = null;\n\n if (settings) {\n for (const p in settings) {\n if (!settings.hasOwnProperty(p)) continue;\n if (!this.hasOwnProperty(p)) continue;\n this[p] = settings[p];\n }\n }\n\n this.literalTypes = {};\n\n this.validate();\n this._string = null;\n this._internalVariableNames = {};\n }\n\n validate() {\n if (typeof this.source !== 'string' && !this.ast) {\n throw new Error('this.source not a string');\n }\n\n if (!this.ast && !utils.isFunctionString(this.source)) {\n throw new Error('this.source not a function string');\n }\n\n if (!this.name) {\n throw new Error('this.name could not be set');\n }\n\n if (this.argumentTypes.length > 0 && this.argumentTypes.length !== this.argumentNames.length) {\n throw new Error(`argumentTypes count of ${ this.argumentTypes.length } exceeds ${ this.argumentNames.length }`);\n }\n\n if (this.output.length < 1) {\n throw new Error('this.output is not big enough');\n }\n }\n\n /**\n * @param {String} name\n * @returns {boolean}\n */\n isIdentifierConstant(name) {\n if (!this.constants) return false;\n return this.constants.hasOwnProperty(name);\n }\n\n isInput(argumentName) {\n return this.argumentTypes[this.argumentNames.indexOf(argumentName)] === 'Input';\n }\n\n pushState(state) {\n this.states.push(state);\n }\n\n popState(state) {\n if (this.state !== state) {\n throw new Error(`Cannot popState ${ state } when in ${ this.state }`);\n }\n this.states.pop();\n }\n\n isState(state) {\n return this.state === state;\n }\n\n get state() {\n return this.states[this.states.length - 1];\n }\n\n /**\n * @function\n * @name astMemberExpressionUnroll\n * @desc Parses the abstract syntax tree for binary expression.\n *\n *Utility function for astCallExpression.
\n *\n * @param {Object} ast - the AST object to parse\n *\n * @returns {String} the function namespace call, unrolled\n */\n astMemberExpressionUnroll(ast) {\n if (ast.type === 'Identifier') {\n return ast.name;\n } else if (ast.type === 'ThisExpression') {\n return 'this';\n }\n\n if (ast.type === 'MemberExpression') {\n if (ast.object && ast.property) {\n //babel sniffing\n if (ast.object.hasOwnProperty('name') && ast.object.name !== 'Math') {\n return this.astMemberExpressionUnroll(ast.property);\n }\n\n return (\n this.astMemberExpressionUnroll(ast.object) +\n '.' +\n this.astMemberExpressionUnroll(ast.property)\n );\n }\n }\n\n //babel sniffing\n if (ast.hasOwnProperty('expressions')) {\n const firstExpression = ast.expressions[0];\n if (firstExpression.type === 'Literal' && firstExpression.value === 0 && ast.expressions.length === 2) {\n return this.astMemberExpressionUnroll(ast.expressions[1]);\n }\n }\n\n // Failure, unknown expression\n throw this.astErrorOutput('Unknown astMemberExpressionUnroll', ast);\n }\n\n /**\n * @desc Parses the class function JS, and returns its Abstract Syntax Tree object.\n * This is used internally to convert to shader code\n *\n * @param {Object} [inParser] - Parser to use, assumes in scope 'parser' if null or undefined\n *\n * @returns {Object} The function AST Object, note that result is cached under this.ast;\n */\n getJsAST(inParser) {\n if (this.ast) {\n return this.ast;\n }\n if (typeof this.source === 'object') {\n this.traceFunctionAST(this.source);\n return this.ast = this.source;\n }\n\n inParser = inParser || acorn;\n if (inParser === null) {\n throw new Error('Missing JS to AST parser');\n }\n\n const ast = Object.freeze(inParser.parse(`const parser_${ this.name } = ${ this.source };`, {\n locations: true\n }));\n // take out the function object, outside the var declarations\n const functionAST = ast.body[0].declarations[0].init;\n this.traceFunctionAST(functionAST);\n\n if (!ast) {\n throw new Error('Failed to parse JS code');\n }\n\n return this.ast = functionAST;\n }\n\n traceFunctionAST(ast) {\n const { contexts, declarations, functions, identifiers, functionCalls } = new FunctionTracer(ast);\n this.contexts = contexts;\n this.identifiers = identifiers;\n this.functionCalls = functionCalls;\n this.functions = functions;\n for (let i = 0; i < declarations.length; i++) {\n const declaration = declarations[i];\n const { ast, inForLoopInit, inForLoopTest } = declaration;\n const { init } = ast;\n const dependencies = this.getDependencies(init);\n let valueType = null;\n\n if (inForLoopInit && inForLoopTest) {\n valueType = 'Integer';\n } else {\n if (init) {\n const realType = this.getType(init);\n switch (realType) {\n case 'Integer':\n case 'Float':\n case 'Number':\n if (init.type === 'MemberExpression') {\n valueType = realType;\n } else {\n valueType = 'Number';\n }\n break;\n case 'LiteralInteger':\n valueType = 'Number';\n break;\n default:\n valueType = realType;\n }\n }\n }\n declaration.valueType = valueType;\n declaration.dependencies = dependencies;\n declaration.isSafe = this.isSafeDependencies(dependencies);\n }\n\n for (let i = 0; i < functions.length; i++) {\n this.onNestedFunction(functions[i], this.source);\n }\n }\n\n getDeclaration(ast) {\n for (let i = 0; i < this.identifiers.length; i++) {\n const identifier = this.identifiers[i];\n if (ast === identifier.ast) {\n return identifier.declaration;\n }\n }\n return null;\n }\n\n /**\n * @desc Return the type of parameter sent to subKernel/Kernel.\n * @param {Object} ast - Identifier\n * @returns {String} Type of the parameter\n */\n getVariableType(ast) {\n if (ast.type !== 'Identifier') {\n throw new Error(`ast of ${ast.type} not \"Identifier\"`);\n }\n let type = null;\n const argumentIndex = this.argumentNames.indexOf(ast.name);\n if (argumentIndex === -1) {\n const declaration = this.getDeclaration(ast);\n if (declaration) {\n return declaration.valueType;\n }\n } else {\n const argumentType = this.argumentTypes[argumentIndex];\n if (argumentType) {\n type = argumentType;\n }\n }\n if (!type && this.strictTypingChecking) {\n throw new Error(`Declaration of ${name} not found`);\n }\n return type;\n }\n\n /**\n * Generally used to lookup the value type returned from a member expressions\n * @param {String} type\n * @return {String}\n */\n getLookupType(type) {\n if (!typeLookupMap.hasOwnProperty(type)) {\n throw new Error(`unknown typeLookupMap ${ type }`);\n }\n return typeLookupMap[type];\n }\n\n getConstantType(constantName) {\n if (this.constantTypes[constantName]) {\n const type = this.constantTypes[constantName];\n if (type === 'Float') {\n return 'Number';\n } else {\n return type;\n }\n }\n throw new Error(`Type for constant \"${ constantName }\" not declared`);\n }\n\n toString() {\n if (this._string) return this._string;\n return this._string = this.astGeneric(this.getJsAST(), []).join('').trim();\n }\n\n toJSON() {\n const settings = {\n source: this.source,\n name: this.name,\n constants: this.constants,\n constantTypes: this.constantTypes,\n isRootKernel: this.isRootKernel,\n isSubKernel: this.isSubKernel,\n debug: this.debug,\n output: this.output,\n loopMaxIterations: this.loopMaxIterations,\n argumentNames: this.argumentNames,\n argumentTypes: this.argumentTypes,\n argumentSizes: this.argumentSizes,\n returnType: this.returnType,\n leadingReturnStatement: this.leadingReturnStatement,\n followingReturnStatement: this.followingReturnStatement,\n };\n\n return {\n ast: this.ast,\n settings\n };\n }\n\n /**\n * Recursively looks up type for ast expression until it's found\n * @param ast\n * @returns {String|null}\n */\n getType(ast) {\n if (Array.isArray(ast)) {\n return this.getType(ast[ast.length - 1]);\n }\n switch (ast.type) {\n case 'BlockStatement':\n return this.getType(ast.body);\n case 'ArrayExpression':\n const childType = this.getType(ast.elements[0]);\n switch (childType) {\n case 'Array(2)':\n case 'Array(3)':\n case 'Array(4)':\n return `Matrix(${ast.elements.length})`;\n }\n return `Array(${ ast.elements.length })`;\n case 'Literal':\n const literalKey = this.astKey(ast);\n if (this.literalTypes[literalKey]) {\n return this.literalTypes[literalKey];\n }\n if (Number.isInteger(ast.value)) {\n return 'LiteralInteger';\n } else if (ast.value === true || ast.value === false) {\n return 'Boolean';\n } else {\n return 'Number';\n }\n case 'AssignmentExpression':\n return this.getType(ast.left);\n case 'CallExpression':\n if (this.isAstMathFunction(ast)) {\n return 'Number';\n }\n if (!ast.callee || !ast.callee.name) {\n if (ast.callee.type === 'SequenceExpression' && ast.callee.expressions[ast.callee.expressions.length - 1].property.name) {\n const functionName = ast.callee.expressions[ast.callee.expressions.length - 1].property.name;\n this.inferArgumentTypesIfNeeded(functionName, ast.arguments);\n return this.lookupReturnType(functionName, ast, this);\n }\n if (this.getVariableSignature(ast.callee, true) === 'this.color') {\n return null;\n }\n if (ast.callee.type === 'MemberExpression' && ast.callee.object && ast.callee.property && ast.callee.property.name && ast.arguments) {\n const functionName = ast.callee.property.name;\n this.inferArgumentTypesIfNeeded(functionName, ast.arguments);\n return this.lookupReturnType(functionName, ast, this);\n }\n throw this.astErrorOutput('Unknown call expression', ast);\n }\n if (ast.callee && ast.callee.name) {\n const functionName = ast.callee.name;\n this.inferArgumentTypesIfNeeded(functionName, ast.arguments);\n return this.lookupReturnType(functionName, ast, this);\n }\n throw this.astErrorOutput(`Unhandled getType Type \"${ ast.type }\"`, ast);\n case 'LogicalExpression':\n return 'Boolean';\n case 'BinaryExpression':\n // modulos is Number\n switch (ast.operator) {\n case '%':\n case '/':\n if (this.fixIntegerDivisionAccuracy) {\n return 'Number';\n } else {\n break;\n }\n case '>':\n case '<':\n return 'Boolean';\n case '&':\n case '|':\n case '^':\n case '<<':\n case '>>':\n case '>>>':\n return 'Integer';\n }\n const type = this.getType(ast.left);\n if (this.isState('skip-literal-correction')) return type;\n if (type === 'LiteralInteger') {\n const rightType = this.getType(ast.right);\n if (rightType === 'LiteralInteger') {\n if (ast.left.value % 1 === 0) {\n return 'Integer';\n } else {\n return 'Float';\n }\n }\n return rightType;\n }\n return typeLookupMap[type] || type;\n case 'UpdateExpression':\n return this.getType(ast.argument);\n case 'UnaryExpression':\n if (ast.operator === '~') {\n return 'Integer';\n }\n return this.getType(ast.argument);\n case 'VariableDeclaration': {\n const declarations = ast.declarations;\n let lastType;\n for (let i = 0; i < declarations.length; i++) {\n const declaration = declarations[i];\n lastType = this.getType(declaration);\n }\n if (!lastType) {\n throw this.astErrorOutput(`Unable to find type for declaration`, ast);\n }\n return lastType;\n }\n case 'VariableDeclarator':\n const declaration = this.getDeclaration(ast.id);\n if (!declaration) {\n throw this.astErrorOutput(`Unable to find declarator`, ast);\n }\n\n if (!declaration.valueType) {\n throw this.astErrorOutput(`Unable to find declarator valueType`, ast);\n }\n\n return declaration.valueType;\n case 'Identifier':\n if (ast.name === 'Infinity') {\n return 'Number';\n }\n if (this.isAstVariable(ast)) {\n const signature = this.getVariableSignature(ast);\n if (signature === 'value') {\n return this.getCheckVariableType(ast);\n }\n }\n const origin = this.findIdentifierOrigin(ast);\n if (origin && origin.init) {\n return this.getType(origin.init);\n }\n return null;\n case 'ReturnStatement':\n return this.getType(ast.argument);\n case 'MemberExpression':\n if (this.isAstMathFunction(ast)) {\n switch (ast.property.name) {\n case 'ceil':\n return 'Integer';\n case 'floor':\n return 'Integer';\n case 'round':\n return 'Integer';\n }\n return 'Number';\n }\n if (this.isAstVariable(ast)) {\n const variableSignature = this.getVariableSignature(ast);\n switch (variableSignature) {\n case 'value[]':\n return this.getLookupType(this.getCheckVariableType(ast.object));\n case 'value[][]':\n return this.getLookupType(this.getCheckVariableType(ast.object.object));\n case 'value[][][]':\n return this.getLookupType(this.getCheckVariableType(ast.object.object.object));\n case 'value[][][][]':\n return this.getLookupType(this.getCheckVariableType(ast.object.object.object.object));\n case 'value.thread.value':\n case 'this.thread.value':\n return 'Integer';\n case 'this.output.value':\n return this.dynamicOutput ? 'Integer' : 'LiteralInteger';\n case 'this.constants.value':\n return this.getConstantType(ast.property.name);\n case 'this.constants.value[]':\n return this.getLookupType(this.getConstantType(ast.object.property.name));\n case 'this.constants.value[][]':\n return this.getLookupType(this.getConstantType(ast.object.object.property.name));\n case 'this.constants.value[][][]':\n return this.getLookupType(this.getConstantType(ast.object.object.object.property.name));\n case 'this.constants.value[][][][]':\n return this.getLookupType(this.getConstantType(ast.object.object.object.object.property.name));\n case 'fn()[]':\n case 'fn()[][]':\n case 'fn()[][][]':\n return this.getLookupType(this.getType(ast.object));\n case 'value.value':\n if (this.isAstMathVariable(ast)) {\n return 'Number';\n }\n switch (ast.property.name) {\n case 'r':\n case 'g':\n case 'b':\n case 'a':\n return this.getLookupType(this.getCheckVariableType(ast.object));\n }\n case '[][]':\n return 'Number';\n }\n throw this.astErrorOutput('Unhandled getType MemberExpression', ast);\n }\n throw this.astErrorOutput('Unhandled getType MemberExpression', ast);\n case 'ConditionalExpression':\n return this.getType(ast.consequent);\n case 'FunctionDeclaration':\n case 'FunctionExpression':\n const lastReturn = this.findLastReturn(ast.body);\n if (lastReturn) {\n return this.getType(lastReturn);\n }\n return null;\n case 'IfStatement':\n return this.getType(ast.consequent);\n case 'SequenceExpression':\n return this.getType(ast.expressions[ast.expressions.length - 1]);\n default:\n throw this.astErrorOutput(`Unhandled getType Type \"${ ast.type }\"`, ast);\n }\n }\n\n getCheckVariableType(ast) {\n const type = this.getVariableType(ast);\n if (!type) {\n throw this.astErrorOutput(`${ast.type} is not defined`, ast);\n }\n return type;\n }\n\n inferArgumentTypesIfNeeded(functionName, args) {\n // ensure arguments are filled in, so when we lookup return type, we already can infer it\n for (let i = 0; i < args.length; i++) {\n if (!this.needsArgumentType(functionName, i)) continue;\n const type = this.getType(args[i]);\n if (!type) {\n throw this.astErrorOutput(`Unable to infer argument ${i}`, args[i]);\n }\n this.assignArgumentType(functionName, i, type);\n }\n }\n\n isAstMathVariable(ast) {\n const mathProperties = [\n 'E',\n 'PI',\n 'SQRT2',\n 'SQRT1_2',\n 'LN2',\n 'LN10',\n 'LOG2E',\n 'LOG10E',\n ];\n return ast.type === 'MemberExpression' &&\n ast.object && ast.object.type === 'Identifier' &&\n ast.object.name === 'Math' &&\n ast.property &&\n ast.property.type === 'Identifier' &&\n mathProperties.indexOf(ast.property.name) > -1;\n }\n\n isAstMathFunction(ast) {\n const mathFunctions = [\n 'abs',\n 'acos',\n 'acosh',\n 'asin',\n 'asinh',\n 'atan',\n 'atan2',\n 'atanh',\n 'cbrt',\n 'ceil',\n 'clz32',\n 'cos',\n 'cosh',\n 'expm1',\n 'exp',\n 'floor',\n 'fround',\n 'imul',\n 'log',\n 'log2',\n 'log10',\n 'log1p',\n 'max',\n 'min',\n 'pow',\n 'random',\n 'round',\n 'sign',\n 'sin',\n 'sinh',\n 'sqrt',\n 'tan',\n 'tanh',\n 'trunc',\n ];\n return ast.type === 'CallExpression' &&\n ast.callee &&\n ast.callee.type === 'MemberExpression' &&\n ast.callee.object &&\n ast.callee.object.type === 'Identifier' &&\n ast.callee.object.name === 'Math' &&\n ast.callee.property &&\n ast.callee.property.type === 'Identifier' &&\n mathFunctions.indexOf(ast.callee.property.name) > -1;\n }\n\n isAstVariable(ast) {\n return ast.type === 'Identifier' || ast.type === 'MemberExpression';\n }\n\n isSafe(ast) {\n return this.isSafeDependencies(this.getDependencies(ast));\n }\n\n isSafeDependencies(dependencies) {\n return dependencies && dependencies.every ? dependencies.every(dependency => dependency.isSafe) : true;\n }\n\n /**\n *\n * @param ast\n * @param dependencies\n * @param isNotSafe\n * @return {Array}\n */\n getDependencies(ast, dependencies, isNotSafe) {\n if (!dependencies) {\n dependencies = [];\n }\n if (!ast) return null;\n if (Array.isArray(ast)) {\n for (let i = 0; i < ast.length; i++) {\n this.getDependencies(ast[i], dependencies, isNotSafe);\n }\n return dependencies;\n }\n switch (ast.type) {\n case 'AssignmentExpression':\n this.getDependencies(ast.left, dependencies, isNotSafe);\n this.getDependencies(ast.right, dependencies, isNotSafe);\n return dependencies;\n case 'ConditionalExpression':\n this.getDependencies(ast.test, dependencies, isNotSafe);\n this.getDependencies(ast.alternate, dependencies, isNotSafe);\n this.getDependencies(ast.consequent, dependencies, isNotSafe);\n return dependencies;\n case 'Literal':\n dependencies.push({\n origin: 'literal',\n value: ast.value,\n isSafe: isNotSafe === true ? false : ast.value > -Infinity && ast.value < Infinity && !isNaN(ast.value)\n });\n break;\n case 'VariableDeclarator':\n return this.getDependencies(ast.init, dependencies, isNotSafe);\n case 'Identifier':\n const declaration = this.getDeclaration(ast);\n if (declaration) {\n dependencies.push({\n name: ast.name,\n origin: 'declaration',\n isSafe: isNotSafe ? false : this.isSafeDependencies(declaration.dependencies),\n });\n } else if (this.argumentNames.indexOf(ast.name) > -1) {\n dependencies.push({\n name: ast.name,\n origin: 'argument',\n isSafe: false,\n });\n } else if (this.strictTypingChecking) {\n throw new Error(`Cannot find identifier origin \"${ast.name}\"`);\n }\n break;\n case 'FunctionDeclaration':\n return this.getDependencies(ast.body.body[ast.body.body.length - 1], dependencies, isNotSafe);\n case 'ReturnStatement':\n return this.getDependencies(ast.argument, dependencies);\n case 'BinaryExpression':\n case 'LogicalExpression':\n isNotSafe = (ast.operator === '/' || ast.operator === '*');\n this.getDependencies(ast.left, dependencies, isNotSafe);\n this.getDependencies(ast.right, dependencies, isNotSafe);\n return dependencies;\n case 'UnaryExpression':\n case 'UpdateExpression':\n return this.getDependencies(ast.argument, dependencies, isNotSafe);\n case 'VariableDeclaration':\n return this.getDependencies(ast.declarations, dependencies, isNotSafe);\n case 'ArrayExpression':\n dependencies.push({\n origin: 'declaration',\n isSafe: true,\n });\n return dependencies;\n case 'CallExpression':\n dependencies.push({\n origin: 'function',\n isSafe: true,\n });\n return dependencies;\n case 'MemberExpression':\n const details = this.getMemberExpressionDetails(ast);\n switch (details.signature) {\n case 'value[]':\n this.getDependencies(ast.object, dependencies, isNotSafe);\n break;\n case 'value[][]':\n this.getDependencies(ast.object.object, dependencies, isNotSafe);\n break;\n case 'value[][][]':\n this.getDependencies(ast.object.object.object, dependencies, isNotSafe);\n break;\n case 'this.output.value':\n if (this.dynamicOutput) {\n dependencies.push({\n name: details.name,\n origin: 'output',\n isSafe: false,\n });\n }\n break;\n }\n if (details) {\n if (details.property) {\n this.getDependencies(details.property, dependencies, isNotSafe);\n }\n if (details.xProperty) {\n this.getDependencies(details.xProperty, dependencies, isNotSafe);\n }\n if (details.yProperty) {\n this.getDependencies(details.yProperty, dependencies, isNotSafe);\n }\n if (details.zProperty) {\n this.getDependencies(details.zProperty, dependencies, isNotSafe);\n }\n return dependencies;\n }\n case 'SequenceExpression':\n return this.getDependencies(ast.expressions, dependencies, isNotSafe);\n default:\n throw this.astErrorOutput(`Unhandled type ${ ast.type } in getDependencies`, ast);\n }\n return dependencies;\n }\n\n getVariableSignature(ast, returnRawValue) {\n if (!this.isAstVariable(ast)) {\n throw new Error(`ast of type \"${ ast.type }\" is not a variable signature`);\n }\n if (ast.type === 'Identifier') {\n return 'value';\n }\n const signature = [];\n while (true) {\n if (!ast) break;\n if (ast.computed) {\n signature.push('[]');\n } else if (ast.type === 'ThisExpression') {\n signature.unshift('this');\n } else if (ast.property && ast.property.name) {\n if (\n ast.property.name === 'x' ||\n ast.property.name === 'y' ||\n ast.property.name === 'z'\n ) {\n signature.unshift(returnRawValue ? '.' + ast.property.name : '.value');\n } else if (\n ast.property.name === 'constants' ||\n ast.property.name === 'thread' ||\n ast.property.name === 'output'\n ) {\n signature.unshift('.' + ast.property.name);\n } else {\n signature.unshift(returnRawValue ? '.' + ast.property.name : '.value');\n }\n } else if (ast.name) {\n signature.unshift(returnRawValue ? ast.name : 'value');\n } else if (ast.callee && ast.callee.name) {\n signature.unshift(returnRawValue ? ast.callee.name + '()' : 'fn()');\n } else if (ast.elements) {\n signature.unshift('[]');\n } else {\n signature.unshift('unknown');\n }\n ast = ast.object;\n }\n\n const signatureString = signature.join('');\n if (returnRawValue) {\n return signatureString;\n }\n\n const allowedExpressions = [\n 'value',\n 'value[]',\n 'value[][]',\n 'value[][][]',\n 'value[][][][]',\n 'value.value',\n 'value.thread.value',\n 'this.thread.value',\n 'this.output.value',\n 'this.constants.value',\n 'this.constants.value[]',\n 'this.constants.value[][]',\n 'this.constants.value[][][]',\n 'this.constants.value[][][][]',\n 'fn()[]',\n 'fn()[][]',\n 'fn()[][][]',\n '[][]',\n ];\n if (allowedExpressions.indexOf(signatureString) > -1) {\n return signatureString;\n }\n return null;\n }\n\n build() {\n return this.toString().length > 0;\n }\n\n /**\n * @desc Parses the abstract syntax tree for generically to its respective function\n * @param {Object} ast - the AST object to parse\n * @param {Array} retArr - return array string\n * @returns {Array} the parsed string array\n */\n astGeneric(ast, retArr) {\n if (ast === null) {\n throw this.astErrorOutput('NULL ast', ast);\n } else {\n if (Array.isArray(ast)) {\n for (let i = 0; i < ast.length; i++) {\n this.astGeneric(ast[i], retArr);\n }\n return retArr;\n }\n\n switch (ast.type) {\n case 'FunctionDeclaration':\n return this.astFunctionDeclaration(ast, retArr);\n case 'FunctionExpression':\n return this.astFunctionExpression(ast, retArr);\n case 'ReturnStatement':\n return this.astReturnStatement(ast, retArr);\n case 'Literal':\n return this.astLiteral(ast, retArr);\n case 'BinaryExpression':\n return this.astBinaryExpression(ast, retArr);\n case 'Identifier':\n return this.astIdentifierExpression(ast, retArr);\n case 'AssignmentExpression':\n return this.astAssignmentExpression(ast, retArr);\n case 'ExpressionStatement':\n return this.astExpressionStatement(ast, retArr);\n case 'EmptyStatement':\n return this.astEmptyStatement(ast, retArr);\n case 'BlockStatement':\n return this.astBlockStatement(ast, retArr);\n case 'IfStatement':\n return this.astIfStatement(ast, retArr);\n case 'SwitchStatement':\n return this.astSwitchStatement(ast, retArr);\n case 'BreakStatement':\n return this.astBreakStatement(ast, retArr);\n case 'ContinueStatement':\n return this.astContinueStatement(ast, retArr);\n case 'ForStatement':\n return this.astForStatement(ast, retArr);\n case 'WhileStatement':\n return this.astWhileStatement(ast, retArr);\n case 'DoWhileStatement':\n return this.astDoWhileStatement(ast, retArr);\n case 'VariableDeclaration':\n return this.astVariableDeclaration(ast, retArr);\n case 'VariableDeclarator':\n return this.astVariableDeclarator(ast, retArr);\n case 'ThisExpression':\n return this.astThisExpression(ast, retArr);\n case 'SequenceExpression':\n return this.astSequenceExpression(ast, retArr);\n case 'UnaryExpression':\n return this.astUnaryExpression(ast, retArr);\n case 'UpdateExpression':\n return this.astUpdateExpression(ast, retArr);\n case 'LogicalExpression':\n return this.astLogicalExpression(ast, retArr);\n case 'MemberExpression':\n return this.astMemberExpression(ast, retArr);\n case 'CallExpression':\n return this.astCallExpression(ast, retArr);\n case 'ArrayExpression':\n return this.astArrayExpression(ast, retArr);\n case 'DebuggerStatement':\n return this.astDebuggerStatement(ast, retArr);\n case 'ConditionalExpression':\n return this.astConditionalExpression(ast, retArr);\n }\n\n throw this.astErrorOutput('Unknown ast type : ' + ast.type, ast);\n }\n }\n /**\n * @desc To throw the AST error, with its location.\n * @param {string} error - the error message output\n * @param {Object} ast - the AST object where the error is\n */\n astErrorOutput(error, ast) {\n if (typeof this.source !== 'string') {\n return new Error(error);\n }\n\n const debugString = utils.getAstString(this.source, ast);\n const leadingSource = this.source.substr(ast.start);\n const splitLines = leadingSource.split(/\\n/);\n const lineBefore = splitLines.length > 0 ? splitLines[splitLines.length - 1] : 0;\n return new Error(`${error} on line ${ splitLines.length }, position ${ lineBefore.length }:\\n ${ debugString }`);\n }\n\n astDebuggerStatement(arrNode, retArr) {\n return retArr;\n }\n\n astConditionalExpression(ast, retArr) {\n if (ast.type !== 'ConditionalExpression') {\n throw this.astErrorOutput('Not a conditional expression', ast);\n }\n retArr.push('(');\n this.astGeneric(ast.test, retArr);\n retArr.push('?');\n this.astGeneric(ast.consequent, retArr);\n retArr.push(':');\n this.astGeneric(ast.alternate, retArr);\n retArr.push(')');\n return retArr;\n }\n\n /**\n * @abstract\n * @param {Object} ast\n * @param {String[]} retArr\n * @returns {String[]}\n */\n astFunction(ast, retArr) {\n throw new Error(`\"astFunction\" not defined on ${ this.constructor.name }`);\n }\n\n /**\n * @desc Parses the abstract syntax tree for to its *named function declaration*\n * @param {Object} ast - the AST object to parse\n * @param {Array} retArr - return array string\n * @returns {Array} the append retArr\n */\n astFunctionDeclaration(ast, retArr) {\n if (this.isChildFunction(ast)) {\n return retArr;\n }\n return this.astFunction(ast, retArr);\n }\n astFunctionExpression(ast, retArr) {\n if (this.isChildFunction(ast)) {\n return retArr;\n }\n return this.astFunction(ast, retArr);\n }\n isChildFunction(ast) {\n for (let i = 0; i < this.functions.length; i++) {\n if (this.functions[i] === ast) {\n return true;\n }\n }\n return false;\n }\n astReturnStatement(ast, retArr) {\n return retArr;\n }\n astLiteral(ast, retArr) {\n this.literalTypes[this.astKey(ast)] = 'Number';\n return retArr;\n }\n astBinaryExpression(ast, retArr) {\n return retArr;\n }\n astIdentifierExpression(ast, retArr) {\n return retArr;\n }\n astAssignmentExpression(ast, retArr) {\n return retArr;\n }\n /**\n * @desc Parses the abstract syntax tree for *generic expression* statement\n * @param {Object} esNode - An ast Node\n * @param {Array} retArr - return array string\n * @returns {Array} the append retArr\n */\n astExpressionStatement(esNode, retArr) {\n this.astGeneric(esNode.expression, retArr);\n retArr.push(';');\n return retArr;\n }\n /**\n * @desc Parses the abstract syntax tree for an *Empty* Statement\n * @param {Object} eNode - An ast Node\n * @param {Array} retArr - return array string\n * @returns {Array} the append retArr\n */\n astEmptyStatement(eNode, retArr) {\n return retArr;\n }\n astBlockStatement(ast, retArr) {\n return retArr;\n }\n astIfStatement(ast, retArr) {\n return retArr;\n }\n astSwitchStatement(ast, retArr) {\n return retArr;\n }\n /**\n * @desc Parses the abstract syntax tree for *Break* Statement\n * @param {Object} brNode - An ast Node\n * @param {Array} retArr - return array string\n * @returns {Array} the append retArr\n */\n astBreakStatement(brNode, retArr) {\n retArr.push('break;');\n return retArr;\n }\n /**\n * @desc Parses the abstract syntax tree for *Continue* Statement\n * @param {Object} crNode - An ast Node\n * @param {Array} retArr - return array string\n * @returns {Array} the append retArr\n */\n astContinueStatement(crNode, retArr) {\n retArr.push('continue;\\n');\n return retArr;\n }\n astForStatement(ast, retArr) {\n return retArr;\n }\n astWhileStatement(ast, retArr) {\n return retArr;\n }\n astDoWhileStatement(ast, retArr) {\n return retArr;\n }\n /**\n * @desc Parses the abstract syntax tree for *Variable Declarator*\n * @param {Object} iVarDecNode - An ast Node\n * @param {Array} retArr - return array string\n * @returns {Array} the append retArr\n */\n astVariableDeclarator(iVarDecNode, retArr) {\n this.astGeneric(iVarDecNode.id, retArr);\n if (iVarDecNode.init !== null) {\n retArr.push('=');\n this.astGeneric(iVarDecNode.init, retArr);\n }\n return retArr;\n }\n astThisExpression(ast, retArr) {\n return retArr;\n }\n astSequenceExpression(sNode, retArr) {\n const { expressions } = sNode;\n const sequenceResult = [];\n for (let i = 0; i < expressions.length; i++) {\n const expression = expressions[i];\n const expressionResult = [];\n this.astGeneric(expression, expressionResult);\n sequenceResult.push(expressionResult.join(''));\n }\n if (sequenceResult.length > 1) {\n retArr.push('(', sequenceResult.join(','), ')');\n } else {\n retArr.push(sequenceResult[0]);\n }\n return retArr;\n }\n /**\n * @desc Parses the abstract syntax tree for *Unary* Expression\n * @param {Object} uNode - An ast Node\n * @param {Array} retArr - return array string\n * @returns {Array} the append retArr\n */\n astUnaryExpression(uNode, retArr) {\n const unaryResult = this.checkAndUpconvertBitwiseUnary(uNode, retArr);\n if (unaryResult) {\n return retArr;\n }\n\n if (uNode.prefix) {\n retArr.push(uNode.operator);\n this.astGeneric(uNode.argument, retArr);\n } else {\n this.astGeneric(uNode.argument, retArr);\n retArr.push(uNode.operator);\n }\n\n return retArr;\n }\n\n checkAndUpconvertBitwiseUnary(uNode, retArr) {}\n\n /**\n * @desc Parses the abstract syntax tree for *Update* Expression\n * @param {Object} uNode - An ast Node\n * @param {Array} retArr - return array string\n * @returns {Array} the append retArr\n */\n astUpdateExpression(uNode, retArr) {\n if (uNode.prefix) {\n retArr.push(uNode.operator);\n this.astGeneric(uNode.argument, retArr);\n } else {\n this.astGeneric(uNode.argument, retArr);\n retArr.push(uNode.operator);\n }\n\n return retArr;\n }\n /**\n * @desc Parses the abstract syntax tree for *Logical* Expression\n * @param {Object} logNode - An ast Node\n * @param {Array} retArr - return array string\n * @returns {Array} the append retArr\n */\n astLogicalExpression(logNode, retArr) {\n retArr.push('(');\n this.astGeneric(logNode.left, retArr);\n retArr.push(logNode.operator);\n this.astGeneric(logNode.right, retArr);\n retArr.push(')');\n return retArr;\n }\n astMemberExpression(ast, retArr) {\n return retArr;\n }\n astCallExpression(ast, retArr) {\n return retArr;\n }\n astArrayExpression(ast, retArr) {\n return retArr;\n }\n\n /**\n *\n * @param ast\n * @return {IFunctionNodeMemberExpressionDetails}\n */\n getMemberExpressionDetails(ast) {\n if (ast.type !== 'MemberExpression') {\n throw this.astErrorOutput(`Expression ${ ast.type } not a MemberExpression`, ast);\n }\n let name = null;\n let type = null;\n const variableSignature = this.getVariableSignature(ast);\n switch (variableSignature) {\n case 'value':\n return null;\n case 'value.thread.value':\n case 'this.thread.value':\n case 'this.output.value':\n return {\n signature: variableSignature,\n type: 'Integer',\n name: ast.property.name\n };\n case 'value[]':\n if (typeof ast.object.name !== 'string') {\n throw this.astErrorOutput('Unexpected expression', ast);\n }\n name = ast.object.name;\n return {\n name,\n origin: 'user',\n signature: variableSignature,\n type: this.getVariableType(ast.object),\n xProperty: ast.property\n };\n case 'value[][]':\n if (typeof ast.object.object.name !== 'string') {\n throw this.astErrorOutput('Unexpected expression', ast);\n }\n name = ast.object.object.name;\n return {\n name,\n origin: 'user',\n signature: variableSignature,\n type: this.getVariableType(ast.object.object),\n yProperty: ast.object.property,\n xProperty: ast.property,\n };\n case 'value[][][]':\n if (typeof ast.object.object.object.name !== 'string') {\n throw this.astErrorOutput('Unexpected expression', ast);\n }\n name = ast.object.object.object.name;\n return {\n name,\n origin: 'user',\n signature: variableSignature,\n type: this.getVariableType(ast.object.object.object),\n zProperty: ast.object.object.property,\n yProperty: ast.object.property,\n xProperty: ast.property,\n };\n case 'value[][][][]':\n if (typeof ast.object.object.object.object.name !== 'string') {\n throw this.astErrorOutput('Unexpected expression', ast);\n }\n name = ast.object.object.object.object.name;\n return {\n name,\n origin: 'user',\n signature: variableSignature,\n type: this.getVariableType(ast.object.object.object.object),\n zProperty: ast.object.object.property,\n yProperty: ast.object.property,\n xProperty: ast.property,\n };\n case 'value.value':\n if (typeof ast.property.name !== 'string') {\n throw this.astErrorOutput('Unexpected expression', ast);\n }\n if (this.isAstMathVariable(ast)) {\n name = ast.property.name;\n return {\n name,\n origin: 'Math',\n type: 'Number',\n signature: variableSignature,\n };\n }\n switch (ast.property.name) {\n case 'r':\n case 'g':\n case 'b':\n case 'a':\n name = ast.object.name;\n return {\n name,\n property: ast.property.name,\n origin: 'user',\n signature: variableSignature,\n type: 'Number'\n };\n default:\n throw this.astErrorOutput('Unexpected expression', ast);\n }\n case 'this.constants.value':\n if (typeof ast.property.name !== 'string') {\n throw this.astErrorOutput('Unexpected expression', ast);\n }\n name = ast.property.name;\n type = this.getConstantType(name);\n if (!type) {\n throw this.astErrorOutput('Constant has no type', ast);\n }\n return {\n name,\n type,\n origin: 'constants',\n signature: variableSignature,\n };\n case 'this.constants.value[]':\n if (typeof ast.object.property.name !== 'string') {\n throw this.astErrorOutput('Unexpected expression', ast);\n }\n name = ast.object.property.name;\n type = this.getConstantType(name);\n if (!type) {\n throw this.astErrorOutput('Constant has no type', ast);\n }\n return {\n name,\n type,\n origin: 'constants',\n signature: variableSignature,\n xProperty: ast.property,\n };\n case 'this.constants.value[][]': {\n if (typeof ast.object.object.property.name !== 'string') {\n throw this.astErrorOutput('Unexpected expression', ast);\n }\n name = ast.object.object.property.name;\n type = this.getConstantType(name);\n if (!type) {\n throw this.astErrorOutput('Constant has no type', ast);\n }\n return {\n name,\n type,\n origin: 'constants',\n signature: variableSignature,\n yProperty: ast.object.property,\n xProperty: ast.property,\n };\n }\n case 'this.constants.value[][][]': {\n if (typeof ast.object.object.object.property.name !== 'string') {\n throw this.astErrorOutput('Unexpected expression', ast);\n }\n name = ast.object.object.object.property.name;\n type = this.getConstantType(name);\n if (!type) {\n throw this.astErrorOutput('Constant has no type', ast);\n }\n return {\n name,\n type,\n origin: 'constants',\n signature: variableSignature,\n zProperty: ast.object.object.property,\n yProperty: ast.object.property,\n xProperty: ast.property,\n };\n }\n case 'fn()[]':\n case 'fn()[][]':\n case '[][]':\n return {\n signature: variableSignature,\n property: ast.property,\n };\n default:\n throw this.astErrorOutput('Unexpected expression', ast);\n }\n }\n\n findIdentifierOrigin(astToFind) {\n const stack = [this.ast];\n\n while (stack.length > 0) {\n const atNode = stack[0];\n if (atNode.type === 'VariableDeclarator' && atNode.id && atNode.id.name && atNode.id.name === astToFind.name) {\n return atNode;\n }\n stack.shift();\n if (atNode.argument) {\n stack.push(atNode.argument);\n } else if (atNode.body) {\n stack.push(atNode.body);\n } else if (atNode.declarations) {\n stack.push(atNode.declarations);\n } else if (Array.isArray(atNode)) {\n for (let i = 0; i < atNode.length; i++) {\n stack.push(atNode[i]);\n }\n }\n }\n return null;\n }\n\n findLastReturn(ast) {\n const stack = [ast || this.ast];\n\n while (stack.length > 0) {\n const atNode = stack.pop();\n if (atNode.type === 'ReturnStatement') {\n return atNode;\n }\n if (atNode.type === 'FunctionDeclaration') {\n continue;\n }\n if (atNode.argument) {\n stack.push(atNode.argument);\n } else if (atNode.body) {\n stack.push(atNode.body);\n } else if (atNode.declarations) {\n stack.push(atNode.declarations);\n } else if (Array.isArray(atNode)) {\n for (let i = 0; i < atNode.length; i++) {\n stack.push(atNode[i]);\n }\n } else if (atNode.consequent) {\n stack.push(atNode.consequent);\n } else if (atNode.cases) {\n stack.push(atNode.cases);\n }\n }\n return null;\n }\n\n getInternalVariableName(name) {\n if (!this._internalVariableNames.hasOwnProperty(name)) {\n this._internalVariableNames[name] = 0;\n }\n this._internalVariableNames[name]++;\n if (this._internalVariableNames[name] === 1) {\n return name;\n }\n return name + this._internalVariableNames[name];\n }\n\n astKey(ast, separator = ',') {\n if (!ast.start || !ast.end) throw new Error('AST start and end needed');\n return `${ast.start}${separator}${ast.end}`;\n }\n}\n\nconst typeLookupMap = {\n 'Number': 'Number',\n 'Float': 'Float',\n 'Integer': 'Integer',\n 'Array': 'Number',\n 'Array(2)': 'Number',\n 'Array(3)': 'Number',\n 'Array(4)': 'Number',\n 'Matrix(2)': 'Number',\n 'Matrix(3)': 'Number',\n 'Matrix(4)': 'Number',\n 'Array2D': 'Number',\n 'Array3D': 'Number',\n 'Input': 'Number',\n 'HTMLCanvas': 'Array(4)',\n 'OffscreenCanvas': 'Array(4)',\n 'HTMLImage': 'Array(4)',\n 'ImageBitmap': 'Array(4)',\n 'ImageData': 'Array(4)',\n 'HTMLVideo': 'Array(4)',\n 'HTMLImageArray': 'Array(4)',\n 'NumberTexture': 'Number',\n 'MemoryOptimizedNumberTexture': 'Number',\n 'Array1D(2)': 'Array(2)',\n 'Array1D(3)': 'Array(3)',\n 'Array1D(4)': 'Array(4)',\n 'Array2D(2)': 'Array(2)',\n 'Array2D(3)': 'Array(3)',\n 'Array2D(4)': 'Array(4)',\n 'Array3D(2)': 'Array(2)',\n 'Array3D(3)': 'Array(3)',\n 'Array3D(4)': 'Array(4)',\n 'ArrayTexture(1)': 'Number',\n 'ArrayTexture(2)': 'Array(2)',\n 'ArrayTexture(3)': 'Array(3)',\n 'ArrayTexture(4)': 'Array(4)',\n};\n\nmodule.exports = {\n FunctionNode\n};" }, { "path": "src/backend/function-tracer.js", "content": "const { utils } = require('../utils');\n\nfunction last(array) {\n return array.length > 0 ? array[array.length - 1] : null;\n}\n\nconst states = {\n trackIdentifiers: 'trackIdentifiers',\n memberExpression: 'memberExpression',\n inForLoopInit: 'inForLoopInit'\n};\n\nclass FunctionTracer {\n constructor(ast) {\n this.runningContexts = [];\n this.functionContexts = [];\n this.contexts = [];\n this.functionCalls = [];\n /**\n *\n * @type {IDeclaration[]}\n */\n this.declarations = [];\n this.identifiers = [];\n this.functions = [];\n this.returnStatements = [];\n this.trackedIdentifiers = null;\n this.states = [];\n this.newFunctionContext();\n this.scan(ast);\n }\n\n isState(state) {\n return this.states[this.states.length - 1] === state;\n }\n\n hasState(state) {\n return this.states.indexOf(state) > -1;\n }\n\n pushState(state) {\n this.states.push(state);\n }\n\n popState(state) {\n if (this.isState(state)) {\n this.states.pop();\n } else {\n throw new Error(`Cannot pop the non-active state \"${state}\"`);\n }\n }\n\n get currentFunctionContext() {\n return last(this.functionContexts);\n }\n\n get currentContext() {\n return last(this.runningContexts);\n }\n\n newFunctionContext() {\n const newContext = { '@contextType': 'function' };\n this.contexts.push(newContext);\n this.functionContexts.push(newContext);\n }\n\n newContext(run) {\n const newContext = Object.assign({ '@contextType': 'const/let' }, this.currentContext);\n this.contexts.push(newContext);\n this.runningContexts.push(newContext);\n run();\n const { currentFunctionContext } = this;\n for (const p in currentFunctionContext) {\n if (!currentFunctionContext.hasOwnProperty(p) || newContext.hasOwnProperty(p)) continue;\n newContext[p] = currentFunctionContext[p];\n }\n this.runningContexts.pop();\n return newContext;\n }\n\n useFunctionContext(run) {\n const functionContext = last(this.functionContexts);\n this.runningContexts.push(functionContext);\n run();\n this.runningContexts.pop();\n }\n\n getIdentifiers(run) {\n const trackedIdentifiers = this.trackedIdentifiers = [];\n this.pushState(states.trackIdentifiers);\n run();\n this.trackedIdentifiers = null;\n this.popState(states.trackIdentifiers);\n return trackedIdentifiers;\n }\n\n /**\n * @param {string} name\n * @returns {IDeclaration}\n */\n getDeclaration(name) {\n const { currentContext, currentFunctionContext, runningContexts } = this;\n const declaration = currentContext[name] || currentFunctionContext[name] || null;\n\n if (\n !declaration &&\n currentContext === currentFunctionContext &&\n runningContexts.length > 0\n ) {\n const previousRunningContext = runningContexts[runningContexts.length - 2];\n if (previousRunningContext[name]) {\n return previousRunningContext[name];\n }\n }\n\n return declaration;\n }\n\n /**\n * Recursively scans AST for declarations and functions, and add them to their respective context\n * @param ast\n */\n scan(ast) {\n if (!ast) return;\n if (Array.isArray(ast)) {\n for (let i = 0; i < ast.length; i++) {\n this.scan(ast[i]);\n }\n return;\n }\n switch (ast.type) {\n case 'Program':\n this.useFunctionContext(() => {\n this.scan(ast.body);\n });\n break;\n case 'BlockStatement':\n this.newContext(() => {\n this.scan(ast.body);\n });\n break;\n case 'AssignmentExpression':\n case 'LogicalExpression':\n this.scan(ast.left);\n this.scan(ast.right);\n break;\n case 'BinaryExpression':\n this.scan(ast.left);\n this.scan(ast.right);\n break;\n case 'UpdateExpression':\n if (ast.operator === '++') {\n const declaration = this.getDeclaration(ast.argument.name);\n if (declaration) {\n declaration.suggestedType = 'Integer';\n }\n }\n this.scan(ast.argument);\n break;\n case 'UnaryExpression':\n this.scan(ast.argument);\n break;\n case 'VariableDeclaration':\n if (ast.kind === 'var') {\n this.useFunctionContext(() => {\n ast.declarations = utils.normalizeDeclarations(ast);\n this.scan(ast.declarations);\n });\n } else {\n ast.declarations = utils.normalizeDeclarations(ast);\n this.scan(ast.declarations);\n }\n break;\n case 'VariableDeclarator': {\n const { currentContext } = this;\n const inForLoopInit = this.hasState(states.inForLoopInit);\n const declaration = {\n ast: ast,\n context: currentContext,\n name: ast.id.name,\n origin: 'declaration',\n inForLoopInit,\n inForLoopTest: null,\n assignable: currentContext === this.currentFunctionContext || (!inForLoopInit && !currentContext.hasOwnProperty(ast.id.name)),\n suggestedType: null,\n valueType: null,\n dependencies: null,\n isSafe: null,\n };\n if (!currentContext[ast.id.name]) {\n currentContext[ast.id.name] = declaration;\n }\n this.declarations.push(declaration);\n this.scan(ast.id);\n this.scan(ast.init);\n break;\n }\n case 'FunctionExpression':\n case 'FunctionDeclaration':\n if (this.runningContexts.length === 0) {\n this.scan(ast.body);\n } else {\n this.functions.push(ast);\n }\n break;\n case 'IfStatement':\n this.scan(ast.test);\n this.scan(ast.consequent);\n if (ast.alternate) this.scan(ast.alternate);\n break;\n case 'ForStatement': {\n let testIdentifiers;\n const context = this.newContext(() => {\n this.pushState(states.inForLoopInit);\n this.scan(ast.init);\n this.popState(states.inForLoopInit);\n\n testIdentifiers = this.getIdentifiers(() => {\n this.scan(ast.test);\n });\n\n this.scan(ast.update);\n this.newContext(() => {\n this.scan(ast.body);\n });\n });\n\n if (testIdentifiers) {\n for (const p in context) {\n if (p === '@contextType') continue;\n if (testIdentifiers.indexOf(p) > -1) {\n context[p].inForLoopTest = true;\n }\n }\n }\n break;\n }\n case 'DoWhileStatement':\n case 'WhileStatement':\n this.newContext(() => {\n this.scan(ast.body);\n this.scan(ast.test);\n });\n break;\n case 'Identifier': {\n if (this.isState(states.trackIdentifiers)) {\n this.trackedIdentifiers.push(ast.name);\n }\n this.identifiers.push({\n context: this.currentContext,\n declaration: this.getDeclaration(ast.name),\n ast,\n });\n break;\n }\n case 'ReturnStatement':\n this.returnStatements.push(ast);\n this.scan(ast.argument);\n break;\n case 'MemberExpression':\n this.pushState(states.memberExpression);\n this.scan(ast.object);\n this.scan(ast.property);\n this.popState(states.memberExpression);\n break;\n case 'ExpressionStatement':\n this.scan(ast.expression);\n break;\n case 'SequenceExpression':\n this.scan(ast.expressions);\n break;\n case 'CallExpression':\n this.functionCalls.push({\n context: this.currentContext,\n ast,\n });\n this.scan(ast.arguments);\n break;\n case 'ArrayExpression':\n this.scan(ast.elements);\n break;\n case 'ConditionalExpression':\n this.scan(ast.test);\n this.scan(ast.alternate);\n this.scan(ast.consequent);\n break;\n case 'SwitchStatement':\n this.scan(ast.discriminant);\n this.scan(ast.cases);\n break;\n case 'SwitchCase':\n this.scan(ast.test);\n this.scan(ast.consequent);\n break;\n\n case 'ThisExpression':\n case 'Literal':\n case 'DebuggerStatement':\n case 'EmptyStatement':\n case 'BreakStatement':\n case 'ContinueStatement':\n break;\n default:\n throw new Error(`unhandled type \"${ast.type}\"`);\n }\n }\n}\n\nmodule.exports = {\n FunctionTracer,\n};" }, { "path": "src/backend/gl/kernel-string.js", "content": "const { glWiretap } = require('gl-wiretap');\nconst { utils } = require('../../utils');\n\nfunction toStringWithoutUtils(fn) {\n return fn.toString()\n .replace('=>', '')\n .replace(/^function /, '')\n .replace(/utils[.]/g, '/*utils.*/');\n}\n\n/**\n *\n * @param {GLKernel} Kernel\n * @param {KernelVariable[]} args\n * @param {Kernel} originKernel\n * @param {string} [setupContextString]\n * @param {string} [destroyContextString]\n * @returns {string}\n */\nfunction glKernelString(Kernel, args, originKernel, setupContextString, destroyContextString) {\n if (!originKernel.built) {\n originKernel.build.apply(originKernel, args);\n }\n args = args ? Array.from(args).map(arg => {\n switch (typeof arg) {\n case 'boolean':\n return new Boolean(arg);\n case 'number':\n return new Number(arg);\n default:\n return arg;\n }\n }) : null;\n const uploadedValues = [];\n const postResult = [];\n const context = glWiretap(originKernel.context, {\n useTrackablePrimitives: true,\n onReadPixels: (targetName) => {\n if (kernel.subKernels) {\n if (!subKernelsResultVariableSetup) {\n postResult.push(` const result = { result: ${getRenderString(targetName, kernel)} };`);\n subKernelsResultVariableSetup = true;\n } else {\n const property = kernel.subKernels[subKernelsResultIndex++].property;\n postResult.push(` result${isNaN(property) ? '.' + property : `[${property}]`} = ${getRenderString(targetName, kernel)};`);\n }\n if (subKernelsResultIndex === kernel.subKernels.length) {\n postResult.push(' return result;');\n }\n return;\n }\n if (targetName) {\n postResult.push(` return ${getRenderString(targetName, kernel)};`);\n } else {\n postResult.push(` return null;`);\n }\n },\n onUnrecognizedArgumentLookup: (argument) => {\n const argumentName = findKernelValue(argument, kernel.kernelArguments, [], context, uploadedValues);\n if (argumentName) {\n return argumentName;\n }\n const constantName = findKernelValue(argument, kernel.kernelConstants, constants ? Object.keys(constants).map(key => constants[key]) : [], context, uploadedValues);\n if (constantName) {\n return constantName;\n }\n return null;\n }\n });\n let subKernelsResultVariableSetup = false;\n let subKernelsResultIndex = 0;\n const {\n source,\n canvas,\n output,\n pipeline,\n graphical,\n loopMaxIterations,\n constants,\n optimizeFloatMemory,\n precision,\n fixIntegerDivisionAccuracy,\n functions,\n nativeFunctions,\n subKernels,\n immutable,\n argumentTypes,\n constantTypes,\n kernelArguments,\n kernelConstants,\n tactic,\n } = originKernel;\n const kernel = new Kernel(source, {\n canvas,\n context,\n checkContext: false,\n output,\n pipeline,\n graphical,\n loopMaxIterations,\n constants,\n optimizeFloatMemory,\n precision,\n fixIntegerDivisionAccuracy,\n functions,\n nativeFunctions,\n subKernels,\n immutable,\n argumentTypes,\n constantTypes,\n tactic,\n });\n let result = [];\n context.setIndent(2);\n kernel.build.apply(kernel, args);\n result.push(context.toString());\n context.reset();\n\n kernel.kernelArguments.forEach((kernelArgument, i) => {\n switch (kernelArgument.type) {\n // primitives\n case 'Integer':\n case 'Boolean':\n case 'Number':\n case 'Float':\n // non-primitives\n case 'Array':\n case 'Array(2)':\n case 'Array(3)':\n case 'Array(4)':\n case 'HTMLCanvas':\n case 'HTMLImage':\n case 'HTMLVideo':\n context.insertVariable(`uploadValue_${kernelArgument.name}`, kernelArgument.uploadValue);\n break;\n case 'HTMLImageArray':\n for (let imageIndex = 0; imageIndex < args[i].length; imageIndex++) {\n const arg = args[i];\n context.insertVariable(`uploadValue_${kernelArgument.name}[${imageIndex}]`, arg[imageIndex]);\n }\n break;\n case 'Input':\n context.insertVariable(`uploadValue_${kernelArgument.name}`, kernelArgument.uploadValue);\n break;\n case 'MemoryOptimizedNumberTexture':\n case 'NumberTexture':\n case 'Array1D(2)':\n case 'Array1D(3)':\n case 'Array1D(4)':\n case 'Array2D(2)':\n case 'Array2D(3)':\n case 'Array2D(4)':\n case 'Array3D(2)':\n case 'Array3D(3)':\n case 'Array3D(4)':\n case 'ArrayTexture(1)':\n case 'ArrayTexture(2)':\n case 'ArrayTexture(3)':\n case 'ArrayTexture(4)':\n context.insertVariable(`uploadValue_${kernelArgument.name}`, args[i].texture);\n break;\n default:\n throw new Error(`unhandled kernelArgumentType insertion for glWiretap of type ${kernelArgument.type}`);\n }\n });\n result.push('/** start of injected functions **/');\n result.push(`function ${toStringWithoutUtils(utils.flattenTo)}`);\n result.push(`function ${toStringWithoutUtils(utils.flatten2dArrayTo)}`);\n result.push(`function ${toStringWithoutUtils(utils.flatten3dArrayTo)}`);\n result.push(`function ${toStringWithoutUtils(utils.flatten4dArrayTo)}`);\n result.push(`function ${toStringWithoutUtils(utils.isArray)}`);\n if (kernel.renderOutput !== kernel.renderTexture && kernel.formatValues) {\n result.push(\n ` const renderOutput = function ${toStringWithoutUtils(kernel.formatValues)};`\n );\n }\n result.push('/** end of injected functions **/');\n result.push(` const innerKernel = function (${kernel.kernelArguments.map(kernelArgument => kernelArgument.varName).join(', ')}) {`);\n context.setIndent(4);\n kernel.run.apply(kernel, args);\n if (kernel.renderKernels) {\n kernel.renderKernels();\n } else if (kernel.renderOutput) {\n kernel.renderOutput();\n }\n result.push(' /** start setup uploads for kernel values **/');\n kernel.kernelArguments.forEach(kernelArgument => {\n result.push(' ' + kernelArgument.getStringValueHandler().split('\\n').join('\\n '));\n });\n result.push(' /** end setup uploads for kernel values **/');\n result.push(context.toString());\n if (kernel.renderOutput === kernel.renderTexture) {\n context.reset();\n const framebufferName = context.getContextVariableName(kernel.framebuffer);\n if (kernel.renderKernels) {\n const results = kernel.renderKernels();\n const textureName = context.getContextVariableName(kernel.texture.texture);\n result.push(` return {\n result: {\n texture: ${ textureName },\n type: '${ results.result.type }',\n toArray: ${ getToArrayString(results.result, textureName, framebufferName) }\n },`);\n const { subKernels, mappedTextures } = kernel;\n for (let i = 0; i < subKernels.length; i++) {\n const texture = mappedTextures[i];\n const subKernel = subKernels[i];\n const subKernelResult = results[subKernel.property];\n const subKernelTextureName = context.getContextVariableName(texture.texture);\n result.push(`\n ${subKernel.property}: {\n texture: ${ subKernelTextureName },\n type: '${ subKernelResult.type }',\n toArray: ${ getToArrayString(subKernelResult, subKernelTextureName, framebufferName) }\n },`);\n }\n result.push(` };`);\n } else {\n const rendered = kernel.renderOutput();\n const textureName = context.getContextVariableName(kernel.texture.texture);\n result.push(` return {\n texture: ${ textureName },\n type: '${ rendered.type }',\n toArray: ${ getToArrayString(rendered, textureName, framebufferName) }\n };`);\n }\n }\n result.push(` ${destroyContextString ? '\\n' + destroyContextString + ' ': ''}`);\n result.push(postResult.join('\\n'));\n result.push(' };');\n if (kernel.graphical) {\n result.push(getGetPixelsString(kernel));\n result.push(` innerKernel.getPixels = getPixels;`);\n }\n result.push(' return innerKernel;');\n\n let constantsUpload = [];\n kernelConstants.forEach((kernelConstant) => {\n constantsUpload.push(`${kernelConstant.getStringValueHandler()}`);\n });\n return `function kernel(settings) {\n const { context, constants } = settings;\n ${constantsUpload.join('')}\n ${setupContextString ? setupContextString : ''}\n${result.join('\\n')}\n}`;\n}\n\nfunction getRenderString(targetName, kernel) {\n const readBackValue = kernel.precision === 'single' ? targetName : `new Float32Array(${targetName}.buffer)`;\n if (kernel.output[2]) {\n return `renderOutput(${readBackValue}, ${kernel.output[0]}, ${kernel.output[1]}, ${kernel.output[2]})`;\n }\n if (kernel.output[1]) {\n return `renderOutput(${readBackValue}, ${kernel.output[0]}, ${kernel.output[1]})`;\n }\n\n return `renderOutput(${readBackValue}, ${kernel.output[0]})`;\n}\n\nfunction getGetPixelsString(kernel) {\n const getPixels = kernel.getPixels.toString();\n const useFunctionKeyword = !/^function/.test(getPixels);\n return utils.flattenFunctionToString(`${useFunctionKeyword ? 'function ' : ''}${ getPixels }`, {\n findDependency: (object, name) => {\n if (object === 'utils') {\n return `const ${name} = ${utils[name].toString()};`;\n }\n return null;\n },\n thisLookup: (property) => {\n if (property === 'context') {\n return null;\n }\n if (kernel.hasOwnProperty(property)) {\n return JSON.stringify(kernel[property]);\n }\n throw new Error(`unhandled thisLookup ${ property }`);\n }\n });\n}\n\nfunction getToArrayString(kernelResult, textureName, framebufferName) {\n const toArray = kernelResult.toArray.toString();\n const useFunctionKeyword = !/^function/.test(toArray);\n const flattenedFunctions = utils.flattenFunctionToString(`${useFunctionKeyword ? 'function ' : ''}${ toArray }`, {\n findDependency: (object, name) => {\n if (object === 'utils') {\n return `const ${name} = ${utils[name].toString()};`;\n } else if (object === 'this') {\n if (name === 'framebuffer') {\n return '';\n }\n return `${useFunctionKeyword ? 'function ' : ''}${kernelResult[name].toString()}`;\n } else {\n throw new Error('unhandled fromObject');\n }\n },\n thisLookup: (property, isDeclaration) => {\n if (property === 'texture') {\n return textureName;\n }\n if (property === 'context') {\n if (isDeclaration) return null;\n return 'gl';\n }\n if (kernelResult.hasOwnProperty(property)) {\n return JSON.stringify(kernelResult[property]);\n }\n throw new Error(`unhandled thisLookup ${ property }`);\n }\n });\n return `() => {\n function framebuffer() { return ${framebufferName}; };\n ${flattenedFunctions}\n return toArray();\n }`;\n}\n\n/**\n *\n * @param {KernelVariable} argument\n * @param {KernelValue[]} kernelValues\n * @param {KernelVariable[]} values\n * @param context\n * @param {KernelVariable[]} uploadedValues\n * @return {string|null}\n */\nfunction findKernelValue(argument, kernelValues, values, context, uploadedValues) {\n if (argument === null) return null;\n if (kernelValues === null) return null;\n switch (typeof argument) {\n case 'boolean':\n case 'number':\n return null;\n }\n if (\n typeof HTMLImageElement !== 'undefined' &&\n argument instanceof HTMLImageElement\n ) {\n for (let i = 0; i < kernelValues.length; i++) {\n const kernelValue = kernelValues[i];\n if (kernelValue.type !== 'HTMLImageArray' && kernelValue) continue;\n if (kernelValue.uploadValue !== argument) continue;\n // TODO: if we send two of the same image, the parser could get confused, and short circuit to the first, handle that here\n const variableIndex = values[i].indexOf(argument);\n if (variableIndex === -1) continue;\n const variableName = `uploadValue_${kernelValue.name}[${variableIndex}]`;\n context.insertVariable(variableName, argument);\n return variableName;\n }\n }\n\n for (let i = 0; i < kernelValues.length; i++) {\n const kernelValue = kernelValues[i];\n if (argument !== kernelValue.uploadValue) continue;\n const variable = `uploadValue_${kernelValue.name}`;\n context.insertVariable(variable, kernelValue);\n return variable;\n }\n return null;\n}\n\nmodule.exports = {\n glKernelString\n};" }, { "path": "src/backend/gl/kernel.js", "content": "const { Kernel } = require('../kernel');\nconst { utils } = require('../../utils');\nconst { GLTextureArray2Float } = require('./texture/array-2-float');\nconst { GLTextureArray2Float2D } = require('./texture/array-2-float-2d');\nconst { GLTextureArray2Float3D } = require('./texture/array-2-float-3d');\nconst { GLTextureArray3Float } = require('./texture/array-3-float');\nconst { GLTextureArray3Float2D } = require('./texture/array-3-float-2d');\nconst { GLTextureArray3Float3D } = require('./texture/array-3-float-3d');\nconst { GLTextureArray4Float } = require('./texture/array-4-float');\nconst { GLTextureArray4Float2D } = require('./texture/array-4-float-2d');\nconst { GLTextureArray4Float3D } = require('./texture/array-4-float-3d');\nconst { GLTextureFloat } = require('./texture/float');\nconst { GLTextureFloat2D } = require('./texture/float-2d');\nconst { GLTextureFloat3D } = require('./texture/float-3d');\nconst { GLTextureMemoryOptimized } = require('./texture/memory-optimized');\nconst { GLTextureMemoryOptimized2D } = require('./texture/memory-optimized-2d');\nconst { GLTextureMemoryOptimized3D } = require('./texture/memory-optimized-3d');\nconst { GLTextureUnsigned } = require('./texture/unsigned');\nconst { GLTextureUnsigned2D } = require('./texture/unsigned-2d');\nconst { GLTextureUnsigned3D } = require('./texture/unsigned-3d');\nconst { GLTextureGraphical } = require('./texture/graphical');\n\n/**\n * @abstract\n * @extends Kernel\n */\nclass GLKernel extends Kernel {\n static get mode() {\n return 'gpu';\n }\n\n static getIsFloatRead() {\n const kernelString = `function kernelFunction() {\n return 1;\n }`;\n const kernel = new this(kernelString, {\n context: this.testContext,\n canvas: this.testCanvas,\n validate: false,\n output: [1],\n precision: 'single',\n returnType: 'Number',\n tactic: 'speed',\n });\n kernel.build();\n kernel.run();\n const result = kernel.renderOutput();\n kernel.destroy(true);\n return result[0] === 1;\n }\n\n static getIsIntegerDivisionAccurate() {\n function kernelFunction(v1, v2) {\n return v1[this.thread.x] / v2[this.thread.x];\n }\n const kernel = new this(kernelFunction.toString(), {\n context: this.testContext,\n canvas: this.testCanvas,\n validate: false,\n output: [2],\n returnType: 'Number',\n precision: 'unsigned',\n tactic: 'speed',\n });\n const args = [\n [6, 6030401],\n [3, 3991]\n ];\n kernel.build.apply(kernel, args);\n kernel.run.apply(kernel, args);\n const result = kernel.renderOutput();\n kernel.destroy(true);\n // have we not got whole numbers for 6/3 or 6030401/3991\n // add more here if others see this problem\n return result[0] === 2 && result[1] === 1511;\n }\n\n static getIsSpeedTacticSupported() {\n function kernelFunction(value) {\n return value[this.thread.x];\n }\n const kernel = new this(kernelFunction.toString(), {\n context: this.testContext,\n canvas: this.testCanvas,\n validate: false,\n output: [4],\n returnType: 'Number',\n precision: 'unsigned',\n tactic: 'speed',\n });\n const args = [\n [0, 1, 2, 3]\n ];\n kernel.build.apply(kernel, args);\n kernel.run.apply(kernel, args);\n const result = kernel.renderOutput();\n kernel.destroy(true);\n return Math.round(result[0]) === 0 && Math.round(result[1]) === 1 && Math.round(result[2]) === 2 && Math.round(result[3]) === 3;\n }\n\n /**\n * @abstract\n */\n static get testCanvas() {\n throw new Error(`\"testCanvas\" not defined on ${ this.name }`);\n }\n\n /**\n * @abstract\n */\n static get testContext() {\n throw new Error(`\"testContext\" not defined on ${ this.name }`);\n }\n\n static getFeatures() {\n const gl = this.testContext;\n const isDrawBuffers = this.getIsDrawBuffers();\n return Object.freeze({\n isFloatRead: this.getIsFloatRead(),\n isIntegerDivisionAccurate: this.getIsIntegerDivisionAccurate(),\n isSpeedTacticSupported: this.getIsSpeedTacticSupported(),\n isTextureFloat: this.getIsTextureFloat(),\n isDrawBuffers,\n kernelMap: isDrawBuffers,\n channelCount: this.getChannelCount(),\n maxTextureSize: this.getMaxTextureSize(),\n lowIntPrecision: gl.getShaderPrecisionFormat(gl.FRAGMENT_SHADER, gl.LOW_INT),\n lowFloatPrecision: gl.getShaderPrecisionFormat(gl.FRAGMENT_SHADER, gl.LOW_FLOAT),\n mediumIntPrecision: gl.getShaderPrecisionFormat(gl.FRAGMENT_SHADER, gl.MEDIUM_INT),\n mediumFloatPrecision: gl.getShaderPrecisionFormat(gl.FRAGMENT_SHADER, gl.MEDIUM_FLOAT),\n highIntPrecision: gl.getShaderPrecisionFormat(gl.FRAGMENT_SHADER, gl.HIGH_INT),\n highFloatPrecision: gl.getShaderPrecisionFormat(gl.FRAGMENT_SHADER, gl.HIGH_FLOAT),\n });\n }\n\n /**\n * @abstract\n */\n static setupFeatureChecks() {\n throw new Error(`\"setupFeatureChecks\" not defined on ${ this.name }`);\n }\n\n static getSignature(kernel, argumentTypes) {\n return kernel.getVariablePrecisionString() + (argumentTypes.length > 0 ? ':' + argumentTypes.join(',') : '');\n }\n\n /**\n * @desc Fix division by factor of 3 FP accuracy bug\n * @param {Boolean} fix - should fix\n */\n setFixIntegerDivisionAccuracy(fix) {\n this.fixIntegerDivisionAccuracy = fix;\n return this;\n }\n\n /**\n * @desc Toggle output mode\n * @param {String} flag - 'single' or 'unsigned'\n */\n setPrecision(flag) {\n this.precision = flag;\n return this;\n }\n\n /**\n * @desc Toggle texture output mode\n * @param {Boolean} flag - true to enable floatTextures\n * @deprecated\n */\n setFloatTextures(flag) {\n utils.warnDeprecated('method', 'setFloatTextures', 'setOptimizeFloatMemory');\n this.floatTextures = flag;\n return this;\n }\n\n /**\n * A highly readable very forgiving micro-parser for a glsl function that gets argument types\n * @param {String} source\n * @returns {{argumentTypes: String[], argumentNames: String[]}}\n */\n static nativeFunctionArguments(source) {\n const argumentTypes = [];\n const argumentNames = [];\n const states = [];\n const isStartingVariableName = /^[a-zA-Z_]/;\n const isVariableChar = /[a-zA-Z_0-9]/;\n let i = 0;\n let argumentName = null;\n let argumentType = null;\n while (i < source.length) {\n const char = source[i];\n const nextChar = source[i + 1];\n const state = states.length > 0 ? states[states.length - 1] : null;\n\n // begin MULTI_LINE_COMMENT handling\n if (state === 'FUNCTION_ARGUMENTS' && char === '/' && nextChar === '*') {\n states.push('MULTI_LINE_COMMENT');\n i += 2;\n continue;\n } else if (state === 'MULTI_LINE_COMMENT' && char === '*' && nextChar === '/') {\n states.pop();\n i += 2;\n continue;\n }\n // end MULTI_LINE_COMMENT handling\n\n // begin COMMENT handling\n else if (state === 'FUNCTION_ARGUMENTS' && char === '/' && nextChar === '/') {\n states.push('COMMENT');\n i += 2;\n continue;\n } else if (state === 'COMMENT' && char === '\\n') {\n states.pop();\n i++;\n continue;\n }\n // end COMMENT handling\n\n // being FUNCTION_ARGUMENTS handling\n else if (state === null && char === '(') {\n states.push('FUNCTION_ARGUMENTS');\n i++;\n continue;\n } else if (state === 'FUNCTION_ARGUMENTS') {\n if (char === ')') {\n states.pop();\n break;\n }\n if (char === 'f' && nextChar === 'l' && source[i + 2] === 'o' && source[i + 3] === 'a' && source[i + 4] === 't' && source[i + 5] === ' ') {\n states.push('DECLARE_VARIABLE');\n argumentType = 'float';\n argumentName = '';\n i += 6;\n continue;\n } else if (char === 'i' && nextChar === 'n' && source[i + 2] === 't' && source[i + 3] === ' ') {\n states.push('DECLARE_VARIABLE');\n argumentType = 'int';\n argumentName = '';\n i += 4;\n continue;\n } else if (char === 'v' && nextChar === 'e' && source[i + 2] === 'c' && source[i + 3] === '2' && source[i + 4] === ' ') {\n states.push('DECLARE_VARIABLE');\n argumentType = 'vec2';\n argumentName = '';\n i += 5;\n continue;\n } else if (char === 'v' && nextChar === 'e' && source[i + 2] === 'c' && source[i + 3] === '3' && source[i + 4] === ' ') {\n states.push('DECLARE_VARIABLE');\n argumentType = 'vec3';\n argumentName = '';\n i += 5;\n continue;\n } else if (char === 'v' && nextChar === 'e' && source[i + 2] === 'c' && source[i + 3] === '4' && source[i + 4] === ' ') {\n states.push('DECLARE_VARIABLE');\n argumentType = 'vec4';\n argumentName = '';\n i += 5;\n continue;\n }\n }\n // end FUNCTION_ARGUMENTS handling\n\n // begin DECLARE_VARIABLE handling\n else if (state === 'DECLARE_VARIABLE') {\n if (argumentName === '') {\n if (char === ' ') {\n i++;\n continue;\n }\n if (!isStartingVariableName.test(char)) {\n throw new Error('variable name is not expected string');\n }\n }\n argumentName += char;\n if (!isVariableChar.test(nextChar)) {\n states.pop();\n argumentNames.push(argumentName);\n argumentTypes.push(typeMap[argumentType]);\n }\n }\n // end DECLARE_VARIABLE handling\n\n // Progress to next character\n i++;\n }\n if (states.length > 0) {\n throw new Error('GLSL function was not parsable');\n }\n return {\n argumentNames,\n argumentTypes,\n };\n }\n\n static nativeFunctionReturnType(source) {\n return typeMap[source.match(/int|float|vec[2-4]/)[0]];\n }\n\n static combineKernels(combinedKernel, lastKernel) {\n combinedKernel.apply(null, arguments);\n const {\n texSize,\n context,\n threadDim\n } = lastKernel.texSize;\n let result;\n if (lastKernel.precision === 'single') {\n const w = texSize[0];\n const h = Math.ceil(texSize[1] / 4);\n result = new Float32Array(w * h * 4 * 4);\n context.readPixels(0, 0, w, h * 4, context.RGBA, context.FLOAT, result);\n } else {\n const bytes = new Uint8Array(texSize[0] * texSize[1] * 4);\n context.readPixels(0, 0, texSize[0], texSize[1], context.RGBA, context.UNSIGNED_BYTE, bytes);\n result = new Float32Array(bytes.buffer);\n }\n\n result = result.subarray(0, threadDim[0] * threadDim[1] * threadDim[2]);\n\n if (lastKernel.output.length === 1) {\n return result;\n } else if (lastKernel.output.length === 2) {\n return utils.splitArray(result, lastKernel.output[0]);\n } else if (lastKernel.output.length === 3) {\n const cube = utils.splitArray(result, lastKernel.output[0] * lastKernel.output[1]);\n return cube.map(function(x) {\n return utils.splitArray(x, lastKernel.output[0]);\n });\n }\n }\n\n constructor(source, settings) {\n super(source, settings);\n this.transferValues = null;\n this.formatValues = null;\n /**\n *\n * @type {Texture}\n */\n this.TextureConstructor = null;\n this.renderOutput = null;\n this.renderRawOutput = null;\n this.texSize = null;\n this.translatedSource = null;\n this.compiledFragmentShader = null;\n this.compiledVertexShader = null;\n this.switchingKernels = null;\n this._textureSwitched = null;\n this._mappedTextureSwitched = null;\n }\n\n checkTextureSize() {\n const { features } = this.constructor;\n if (this.texSize[0] > features.maxTextureSize || this.texSize[1] > features.maxTextureSize) {\n throw new Error(`Texture size [${this.texSize[0]},${this.texSize[1]}] generated by kernel is larger than supported size [${features.maxTextureSize},${features.maxTextureSize}]`);\n }\n }\n\n translateSource() {\n throw new Error(`\"translateSource\" not defined on ${this.constructor.name}`);\n }\n\n /**\n * Picks a render strategy for the now finally parsed kernel\n * @param args\n * @return {null|KernelOutput}\n */\n pickRenderStrategy(args) {\n if (this.graphical) {\n this.renderRawOutput = this.readPackedPixelsToUint8Array;\n this.transferValues = (pixels) => pixels;\n this.TextureConstructor = GLTextureGraphical;\n return null;\n }\n if (this.precision === 'unsigned') {\n this.renderRawOutput = this.readPackedPixelsToUint8Array;\n this.transferValues = this.readPackedPixelsToFloat32Array;\n if (this.pipeline) {\n this.renderOutput = this.renderTexture;\n if (this.subKernels !== null) {\n this.renderKernels = this.renderKernelsToTextures;\n }\n switch (this.returnType) {\n case 'LiteralInteger':\n case 'Float':\n case 'Number':\n case 'Integer':\n if (this.output[2] > 0) {\n this.TextureConstructor = GLTextureUnsigned3D;\n return null;\n } else if (this.output[1] > 0) {\n this.TextureConstructor = GLTextureUnsigned2D;\n return null;\n } else {\n this.TextureConstructor = GLTextureUnsigned;\n return null;\n }\n case 'Array(2)':\n case 'Array(3)':\n case 'Array(4)':\n return this.requestFallback(args);\n }\n } else {\n if (this.subKernels !== null) {\n this.renderKernels = this.renderKernelsToArrays;\n }\n switch (this.returnType) {\n case 'LiteralInteger':\n case 'Float':\n case 'Number':\n case 'Integer':\n this.renderOutput = this.renderValues;\n if (this.output[2] > 0) {\n this.TextureConstructor = GLTextureUnsigned3D;\n this.formatValues = utils.erect3DPackedFloat;\n return null;\n } else if (this.output[1] > 0) {\n this.TextureConstructor = GLTextureUnsigned2D;\n this.formatValues = utils.erect2DPackedFloat;\n return null;\n } else {\n this.TextureConstructor = GLTextureUnsigned;\n this.formatValues = utils.erectPackedFloat;\n return null;\n }\n case 'Array(2)':\n case 'Array(3)':\n case 'Array(4)':\n return this.requestFallback(args);\n }\n }\n } else if (this.precision === 'single') {\n this.renderRawOutput = this.readFloatPixelsToFloat32Array;\n this.transferValues = this.readFloatPixelsToFloat32Array;\n if (this.pipeline) {\n this.renderOutput = this.renderTexture;\n if (this.subKernels !== null) {\n this.renderKernels = this.renderKernelsToTextures;\n }\n switch (this.returnType) {\n case 'LiteralInteger':\n case 'Float':\n case 'Number':\n case 'Integer': {\n if (this.optimizeFloatMemory) {\n if (this.output[2] > 0) {\n this.TextureConstructor = GLTextureMemoryOptimized3D;\n return null;\n } else if (this.output[1] > 0) {\n this.TextureConstructor = GLTextureMemoryOptimized2D;\n return null;\n } else {\n this.TextureConstructor = GLTextureMemoryOptimized;\n return null;\n }\n } else {\n if (this.output[2] > 0) {\n this.TextureConstructor = GLTextureFloat3D;\n return null;\n } else if (this.output[1] > 0) {\n this.TextureConstructor = GLTextureFloat2D;\n return null;\n } else {\n this.TextureConstructor = GLTextureFloat;\n return null;\n }\n }\n }\n case 'Array(2)': {\n if (this.output[2] > 0) {\n this.TextureConstructor = GLTextureArray2Float3D;\n return null;\n } else if (this.output[1] > 0) {\n this.TextureConstructor = GLTextureArray2Float2D;\n return null;\n } else {\n this.TextureConstructor = GLTextureArray2Float;\n return null;\n }\n }\n case 'Array(3)': {\n if (this.output[2] > 0) {\n this.TextureConstructor = GLTextureArray3Float3D;\n return null;\n } else if (this.output[1] > 0) {\n this.TextureConstructor = GLTextureArray3Float2D;\n return null;\n } else {\n this.TextureConstructor = GLTextureArray3Float;\n return null;\n }\n }\n case 'Array(4)': {\n if (this.output[2] > 0) {\n this.TextureConstructor = GLTextureArray4Float3D;\n return null;\n } else if (this.output[1] > 0) {\n this.TextureConstructor = GLTextureArray4Float2D;\n return null;\n } else {\n this.TextureConstructor = GLTextureArray4Float;\n return null;\n }\n }\n }\n }\n this.renderOutput = this.renderValues;\n if (this.subKernels !== null) {\n this.renderKernels = this.renderKernelsToArrays;\n }\n if (this.optimizeFloatMemory) {\n switch (this.returnType) {\n case 'LiteralInteger':\n case 'Float':\n case 'Number':\n case 'Integer': {\n if (this.output[2] > 0) {\n this.TextureConstructor = GLTextureMemoryOptimized3D;\n this.formatValues = utils.erectMemoryOptimized3DFloat;\n return null;\n } else if (this.output[1] > 0) {\n this.TextureConstructor = GLTextureMemoryOptimized2D;\n this.formatValues = utils.erectMemoryOptimized2DFloat;\n return null;\n } else {\n this.TextureConstructor = GLTextureMemoryOptimized;\n this.formatValues = utils.erectMemoryOptimizedFloat;\n return null;\n }\n }\n case 'Array(2)': {\n if (this.output[2] > 0) {\n this.TextureConstructor = GLTextureArray2Float3D;\n this.formatValues = utils.erect3DArray2;\n return null;\n } else if (this.output[1] > 0) {\n this.TextureConstructor = GLTextureArray2Float2D;\n this.formatValues = utils.erect2DArray2;\n return null;\n } else {\n this.TextureConstructor = GLTextureArray2Float;\n this.formatValues = utils.erectArray2;\n return null;\n }\n }\n case 'Array(3)': {\n if (this.output[2] > 0) {\n this.TextureConstructor = GLTextureArray3Float3D;\n this.formatValues = utils.erect3DArray3;\n return null;\n } else if (this.output[1] > 0) {\n this.TextureConstructor = GLTextureArray3Float2D;\n this.formatValues = utils.erect2DArray3;\n return null;\n } else {\n this.TextureConstructor = GLTextureArray3Float;\n this.formatValues = utils.erectArray3;\n return null;\n }\n }\n case 'Array(4)': {\n if (this.output[2] > 0) {\n this.TextureConstructor = GLTextureArray4Float3D;\n this.formatValues = utils.erect3DArray4;\n return null;\n } else if (this.output[1] > 0) {\n this.TextureConstructor = GLTextureArray4Float2D;\n this.formatValues = utils.erect2DArray4;\n return null;\n } else {\n this.TextureConstructor = GLTextureArray4Float;\n this.formatValues = utils.erectArray4;\n return null;\n }\n }\n }\n } else {\n switch (this.returnType) {\n case 'LiteralInteger':\n case 'Float':\n case 'Number':\n case 'Integer': {\n if (this.output[2] > 0) {\n this.TextureConstructor = GLTextureFloat3D;\n this.formatValues = utils.erect3DFloat;\n return null;\n } else if (this.output[1] > 0) {\n this.TextureConstructor = GLTextureFloat2D;\n this.formatValues = utils.erect2DFloat;\n return null;\n } else {\n this.TextureConstructor = GLTextureFloat;\n this.formatValues = utils.erectFloat;\n return null;\n }\n }\n case 'Array(2)': {\n if (this.output[2] > 0) {\n this.TextureConstructor = GLTextureArray2Float3D;\n this.formatValues = utils.erect3DArray2;\n return null;\n } else if (this.output[1] > 0) {\n this.TextureConstructor = GLTextureArray2Float2D;\n this.formatValues = utils.erect2DArray2;\n return null;\n } else {\n this.TextureConstructor = GLTextureArray2Float;\n this.formatValues = utils.erectArray2;\n return null;\n }\n }\n case 'Array(3)': {\n if (this.output[2] > 0) {\n this.TextureConstructor = GLTextureArray3Float3D;\n this.formatValues = utils.erect3DArray3;\n return null;\n } else if (this.output[1] > 0) {\n this.TextureConstructor = GLTextureArray3Float2D;\n this.formatValues = utils.erect2DArray3;\n return null;\n } else {\n this.TextureConstructor = GLTextureArray3Float;\n this.formatValues = utils.erectArray3;\n return null;\n }\n }\n case 'Array(4)': {\n if (this.output[2] > 0) {\n this.TextureConstructor = GLTextureArray4Float3D;\n this.formatValues = utils.erect3DArray4;\n return null;\n } else if (this.output[1] > 0) {\n this.TextureConstructor = GLTextureArray4Float2D;\n this.formatValues = utils.erect2DArray4;\n return null;\n } else {\n this.TextureConstructor = GLTextureArray4Float;\n this.formatValues = utils.erectArray4;\n return null;\n }\n }\n }\n }\n } else {\n throw new Error(`unhandled precision of \"${this.precision}\"`);\n }\n\n throw new Error(`unhandled return type \"${this.returnType}\"`);\n }\n\n /**\n * @abstract\n * @returns String\n */\n getKernelString() {\n throw new Error(`abstract method call`);\n }\n\n getMainResultTexture() {\n switch (this.returnType) {\n case 'LiteralInteger':\n case 'Float':\n case 'Integer':\n case 'Number':\n return this.getMainResultNumberTexture();\n case 'Array(2)':\n return this.getMainResultArray2Texture();\n case 'Array(3)':\n return this.getMainResultArray3Texture();\n case 'Array(4)':\n return this.getMainResultArray4Texture();\n default:\n throw new Error(`unhandled returnType type ${ this.returnType }`);\n }\n }\n\n /**\n * @abstract\n * @returns String[]\n */\n getMainResultKernelNumberTexture() {\n throw new Error(`abstract method call`);\n }\n /**\n * @abstract\n * @returns String[]\n */\n getMainResultSubKernelNumberTexture() {\n throw new Error(`abstract method call`);\n }\n /**\n * @abstract\n * @returns String[]\n */\n getMainResultKernelArray2Texture() {\n throw new Error(`abstract method call`);\n }\n /**\n * @abstract\n * @returns String[]\n */\n getMainResultSubKernelArray2Texture() {\n throw new Error(`abstract method call`);\n }\n /**\n * @abstract\n * @returns String[]\n */\n getMainResultKernelArray3Texture() {\n throw new Error(`abstract method call`);\n }\n /**\n * @abstract\n * @returns String[]\n */\n getMainResultSubKernelArray3Texture() {\n throw new Error(`abstract method call`);\n }\n /**\n * @abstract\n * @returns String[]\n */\n getMainResultKernelArray4Texture() {\n throw new Error(`abstract method call`);\n }\n /**\n * @abstract\n * @returns String[]\n */\n getMainResultSubKernelArray4Texture() {\n throw new Error(`abstract method call`);\n }\n /**\n * @abstract\n * @returns String[]\n */\n getMainResultGraphical() {\n throw new Error(`abstract method call`);\n }\n /**\n * @abstract\n * @returns String[]\n */\n getMainResultMemoryOptimizedFloats() {\n throw new Error(`abstract method call`);\n }\n /**\n * @abstract\n * @returns String[]\n */\n getMainResultPackedPixels() {\n throw new Error(`abstract method call`);\n }\n\n getMainResultString() {\n if (this.graphical) {\n return this.getMainResultGraphical();\n } else if (this.precision === 'single') {\n if (this.optimizeFloatMemory) {\n return this.getMainResultMemoryOptimizedFloats();\n }\n return this.getMainResultTexture();\n } else {\n return this.getMainResultPackedPixels();\n }\n }\n\n getMainResultNumberTexture() {\n return utils.linesToString(this.getMainResultKernelNumberTexture()) +\n utils.linesToString(this.getMainResultSubKernelNumberTexture());\n }\n\n getMainResultArray2Texture() {\n return utils.linesToString(this.getMainResultKernelArray2Texture()) +\n utils.linesToString(this.getMainResultSubKernelArray2Texture());\n }\n\n getMainResultArray3Texture() {\n return utils.linesToString(this.getMainResultKernelArray3Texture()) +\n utils.linesToString(this.getMainResultSubKernelArray3Texture());\n }\n\n getMainResultArray4Texture() {\n return utils.linesToString(this.getMainResultKernelArray4Texture()) +\n utils.linesToString(this.getMainResultSubKernelArray4Texture());\n }\n\n /**\n *\n * @return {string}\n */\n getFloatTacticDeclaration() {\n const variablePrecision = this.getVariablePrecisionString(this.texSize, this.tactic);\n return `precision ${variablePrecision} float;\\n`;\n }\n\n /**\n *\n * @return {string}\n */\n getIntTacticDeclaration() {\n return `precision ${this.getVariablePrecisionString(this.texSize, this.tactic, true)} int;\\n`;\n }\n\n /**\n *\n * @return {string}\n */\n getSampler2DTacticDeclaration() {\n return `precision ${this.getVariablePrecisionString(this.texSize, this.tactic)} sampler2D;\\n`;\n }\n\n getSampler2DArrayTacticDeclaration() {\n return `precision ${this.getVariablePrecisionString(this.texSize, this.tactic)} sampler2DArray;\\n`;\n }\n\n renderTexture() {\n return this.immutable ? this.texture.clone() : this.texture;\n }\n readPackedPixelsToUint8Array() {\n if (this.precision !== 'unsigned') throw new Error('Requires this.precision to be \"unsigned\"');\n const {\n texSize,\n context: gl\n } = this;\n const result = new Uint8Array(texSize[0] * texSize[1] * 4);\n gl.readPixels(0, 0, texSize[0], texSize[1], gl.RGBA, gl.UNSIGNED_BYTE, result);\n return result;\n }\n\n readPackedPixelsToFloat32Array() {\n return new Float32Array(this.readPackedPixelsToUint8Array().buffer);\n }\n\n readFloatPixelsToFloat32Array() {\n if (this.precision !== 'single') throw new Error('Requires this.precision to be \"single\"');\n const {\n texSize,\n context: gl\n } = this;\n const w = texSize[0];\n const h = texSize[1];\n const result = new Float32Array(w * h * 4);\n gl.readPixels(0, 0, w, h, gl.RGBA, gl.FLOAT, result);\n return result;\n }\n\n /**\n *\n * @param {Boolean} [flip]\n * @return {Uint8ClampedArray}\n */\n getPixels(flip) {\n const {\n context: gl,\n output\n } = this;\n const [width, height] = output;\n const pixels = new Uint8Array(width * height * 4);\n gl.readPixels(0, 0, width, height, gl.RGBA, gl.UNSIGNED_BYTE, pixels);\n // flipped by default, so invert\n return new Uint8ClampedArray((flip ? pixels : utils.flipPixels(pixels, width, height)).buffer);\n }\n\n renderKernelsToArrays() {\n const result = {\n result: this.renderOutput(),\n };\n for (let i = 0; i < this.subKernels.length; i++) {\n result[this.subKernels[i].property] = this.mappedTextures[i].toArray();\n }\n return result;\n }\n\n renderKernelsToTextures() {\n const result = {\n result: this.renderOutput(),\n };\n if (this.immutable) {\n for (let i = 0; i < this.subKernels.length; i++) {\n result[this.subKernels[i].property] = this.mappedTextures[i].clone();\n }\n } else {\n for (let i = 0; i < this.subKernels.length; i++) {\n result[this.subKernels[i].property] = this.mappedTextures[i];\n }\n }\n return result;\n }\n\n resetSwitchingKernels() {\n const existingValue = this.switchingKernels;\n this.switchingKernels = null;\n return existingValue;\n }\n\n setOutput(output) {\n const newOutput = this.toKernelOutput(output);\n if (this.program) {\n if (!this.dynamicOutput) {\n throw new Error('Resizing a kernel with dynamicOutput: false is not possible');\n }\n const newThreadDim = [newOutput[0], newOutput[1] || 1, newOutput[2] || 1];\n const newTexSize = utils.getKernelTextureSize({\n optimizeFloatMemory: this.optimizeFloatMemory,\n precision: this.precision,\n }, newThreadDim);\n const oldTexSize = this.texSize;\n if (oldTexSize) {\n const oldPrecision = this.getVariablePrecisionString(oldTexSize, this.tactic);\n const newPrecision = this.getVariablePrecisionString(newTexSize, this.tactic);\n if (oldPrecision !== newPrecision) {\n if (this.debug) {\n console.warn('Precision requirement changed, asking GPU instance to recompile');\n }\n this.switchKernels({\n type: 'outputPrecisionMismatch',\n precision: newPrecision,\n needed: output\n });\n return;\n }\n }\n this.output = newOutput;\n this.threadDim = newThreadDim;\n this.texSize = newTexSize;\n const { context: gl } = this;\n gl.bindFramebuffer(gl.FRAMEBUFFER, this.framebuffer);\n this.updateMaxTexSize();\n this.framebuffer.width = this.texSize[0];\n this.framebuffer.height = this.texSize[1];\n gl.viewport(0, 0, this.maxTexSize[0], this.maxTexSize[1]);\n this.canvas.width = this.maxTexSize[0];\n this.canvas.height = this.maxTexSize[1];\n if (this.texture) {\n this.texture.delete();\n }\n this.texture = null;\n this._setupOutputTexture();\n if (this.mappedTextures && this.mappedTextures.length > 0) {\n for (let i = 0; i < this.mappedTextures.length; i++) {\n this.mappedTextures[i].delete();\n }\n this.mappedTextures = null;\n this._setupSubOutputTextures();\n }\n } else {\n this.output = newOutput;\n }\n return this;\n }\n renderValues() {\n return this.formatValues(\n this.transferValues(),\n this.output[0],\n this.output[1],\n this.output[2]\n );\n }\n switchKernels(reason) {\n if (this.switchingKernels) {\n this.switchingKernels.push(reason);\n } else {\n this.switchingKernels = [reason];\n }\n }\n getVariablePrecisionString(textureSize = this.texSize, tactic = this.tactic, isInt = false) {\n if (!tactic) {\n if (!this.constructor.features.isSpeedTacticSupported) return 'highp';\n const low = this.constructor.features[isInt ? 'lowIntPrecision' : 'lowFloatPrecision'];\n const medium = this.constructor.features[isInt ? 'mediumIntPrecision' : 'mediumFloatPrecision'];\n const high = this.constructor.features[isInt ? 'highIntPrecision' : 'highFloatPrecision'];\n const requiredSize = Math.log2(textureSize[0] * textureSize[1]);\n if (requiredSize <= low.rangeMax) {\n return 'lowp';\n } else if (requiredSize <= medium.rangeMax) {\n return 'mediump';\n } else if (requiredSize <= high.rangeMax) {\n return 'highp';\n } else {\n throw new Error(`The required size exceeds that of the ability of your system`);\n }\n }\n switch (tactic) {\n case 'speed':\n return 'lowp';\n case 'balanced':\n return 'mediump';\n case 'precision':\n return 'highp';\n default:\n throw new Error(`Unknown tactic \"${tactic}\" use \"speed\", \"balanced\", \"precision\", or empty for auto`);\n }\n }\n\n /**\n *\n * @param {WebGLKernelValue} kernelValue\n * @param {GLTexture} arg\n */\n updateTextureArgumentRefs(kernelValue, arg) {\n if (!this.immutable) return;\n if (this.texture.texture === arg.texture) {\n const { prevArg } = kernelValue;\n if (prevArg) {\n if (prevArg.texture._refs === 1) {\n this.texture.delete();\n this.texture = prevArg.clone();\n this._textureSwitched = true;\n }\n prevArg.delete();\n }\n kernelValue.prevArg = arg.clone();\n } else if (this.mappedTextures && this.mappedTextures.length > 0) {\n const { mappedTextures } = this;\n for (let i = 0; i < mappedTextures.length; i++) {\n const mappedTexture = mappedTextures[i];\n if (mappedTexture.texture === arg.texture) {\n const { prevArg } = kernelValue;\n if (prevArg) {\n if (prevArg.texture._refs === 1) {\n mappedTexture.delete();\n mappedTextures[i] = prevArg.clone();\n this._mappedTextureSwitched[i] = true;\n }\n prevArg.delete();\n }\n kernelValue.prevArg = arg.clone();\n return;\n }\n }\n }\n }\n\n onActivate(previousKernel) {\n this._textureSwitched = true;\n this.texture = previousKernel.texture;\n if (this.mappedTextures) {\n for (let i = 0; i < this.mappedTextures.length; i++) {\n this._mappedTextureSwitched[i] = true;\n }\n this.mappedTextures = previousKernel.mappedTextures;\n }\n }\n\n initCanvas() {}\n}\n\nconst typeMap = {\n int: 'Integer',\n float: 'Number',\n vec2: 'Array(2)',\n vec3: 'Array(3)',\n vec4: 'Array(4)',\n};\n\nmodule.exports = {\n GLKernel\n};" }, { "path": "src/backend/gl/texture/array-2-float-2d.js", "content": "const { utils } = require('../../../utils');\nconst { GLTextureFloat } = require('./float');\n\nclass GLTextureArray2Float2D extends GLTextureFloat {\n constructor(settings) {\n super(settings);\n this.type = 'ArrayTexture(2)';\n }\n toArray() {\n return utils.erect2DArray2(this.renderValues(), this.output[0], this.output[1]);\n }\n}\n\nmodule.exports = {\n GLTextureArray2Float2D\n};" }, { "path": "src/backend/gl/texture/array-2-float-3d.js", "content": "const { utils } = require('../../../utils');\nconst { GLTextureFloat } = require('./float');\n\nclass GLTextureArray2Float3D extends GLTextureFloat {\n constructor(settings) {\n super(settings);\n this.type = 'ArrayTexture(2)';\n }\n toArray() {\n return utils.erect3DArray2(this.renderValues(), this.output[0], this.output[1], this.output[2]);\n }\n}\n\nmodule.exports = {\n GLTextureArray2Float3D\n};" }, { "path": "src/backend/gl/texture/array-2-float.js", "content": "const { utils } = require('../../../utils');\nconst { GLTextureFloat } = require('./float');\n\nclass GLTextureArray2Float extends GLTextureFloat {\n constructor(settings) {\n super(settings);\n this.type = 'ArrayTexture(2)';\n }\n toArray() {\n return utils.erectArray2(this.renderValues(), this.output[0], this.output[1]);\n }\n}\n\nmodule.exports = {\n GLTextureArray2Float\n};" }, { "path": "src/backend/gl/texture/array-3-float-2d.js", "content": "const { utils } = require('../../../utils');\nconst { GLTextureFloat } = require('./float');\n\nclass GLTextureArray3Float2D extends GLTextureFloat {\n constructor(settings) {\n super(settings);\n this.type = 'ArrayTexture(3)';\n }\n toArray() {\n return utils.erect2DArray3(this.renderValues(), this.output[0], this.output[1]);\n }\n}\n\nmodule.exports = {\n GLTextureArray3Float2D\n};" }, { "path": "src/backend/gl/texture/array-3-float-3d.js", "content": "const { utils } = require('../../../utils');\nconst { GLTextureFloat } = require('./float');\n\nclass GLTextureArray3Float3D extends GLTextureFloat {\n constructor(settings) {\n super(settings);\n this.type = 'ArrayTexture(3)';\n }\n toArray() {\n return utils.erect3DArray3(this.renderValues(), this.output[0], this.output[1], this.output[2]);\n }\n}\n\nmodule.exports = {\n GLTextureArray3Float3D\n};" }, { "path": "src/backend/gl/texture/array-3-float.js", "content": "const { utils } = require('../../../utils');\nconst { GLTextureFloat } = require('./float');\n\nclass GLTextureArray3Float extends GLTextureFloat {\n constructor(settings) {\n super(settings);\n this.type = 'ArrayTexture(3)';\n }\n toArray() {\n return utils.erectArray3(this.renderValues(), this.output[0]);\n }\n}\n\nmodule.exports = {\n GLTextureArray3Float\n};" }, { "path": "src/backend/gl/texture/array-4-float-2d.js", "content": "const { utils } = require('../../../utils');\nconst { GLTextureFloat } = require('./float');\n\nclass GLTextureArray4Float2D extends GLTextureFloat {\n constructor(settings) {\n super(settings);\n this.type = 'ArrayTexture(4)';\n }\n toArray() {\n return utils.erect2DArray4(this.renderValues(), this.output[0], this.output[1]);\n }\n}\n\nmodule.exports = {\n GLTextureArray4Float2D\n};" }, { "path": "src/backend/gl/texture/array-4-float-3d.js", "content": "const { utils } = require('../../../utils');\nconst { GLTextureFloat } = require('./float');\n\nclass GLTextureArray4Float3D extends GLTextureFloat {\n constructor(settings) {\n super(settings);\n this.type = 'ArrayTexture(4)';\n }\n toArray() {\n return utils.erect3DArray4(this.renderValues(), this.output[0], this.output[1], this.output[2]);\n }\n}\n\nmodule.exports = {\n GLTextureArray4Float3D\n};" }, { "path": "src/backend/gl/texture/array-4-float.js", "content": "const { utils } = require('../../../utils');\nconst { GLTextureFloat } = require('./float');\n\nclass GLTextureArray4Float extends GLTextureFloat {\n constructor(settings) {\n super(settings);\n this.type = 'ArrayTexture(4)';\n }\n toArray() {\n return utils.erectArray4(this.renderValues(), this.output[0]);\n }\n}\n\nmodule.exports = {\n GLTextureArray4Float\n};" }, { "path": "src/backend/gl/texture/float-2d.js", "content": "const { utils } = require('../../../utils');\nconst { GLTextureFloat } = require('./float');\n\nclass GLTextureFloat2D extends GLTextureFloat {\n constructor(settings) {\n super(settings);\n this.type = 'ArrayTexture(1)';\n }\n toArray() {\n return utils.erect2DFloat(this.renderValues(), this.output[0], this.output[1]);\n }\n}\n\nmodule.exports = {\n GLTextureFloat2D\n};" }, { "path": "src/backend/gl/texture/float-3d.js", "content": "const { utils } = require('../../../utils');\nconst { GLTextureFloat } = require('./float');\n\nclass GLTextureFloat3D extends GLTextureFloat {\n constructor(settings) {\n super(settings);\n this.type = 'ArrayTexture(1)';\n }\n toArray() {\n return utils.erect3DFloat(this.renderValues(), this.output[0], this.output[1], this.output[2]);\n }\n}\n\nmodule.exports = {\n GLTextureFloat3D\n};" }, { "path": "src/backend/gl/texture/float.js", "content": "const { utils } = require('../../../utils');\nconst { GLTexture } = require('./index');\n\nclass GLTextureFloat extends GLTexture {\n get textureType() {\n return this.context.FLOAT;\n }\n constructor(settings) {\n super(settings);\n this.type = 'ArrayTexture(1)';\n }\n renderRawOutput() {\n const gl = this.context;\n const size = this.size;\n gl.bindFramebuffer(gl.FRAMEBUFFER, this.framebuffer());\n gl.framebufferTexture2D(\n gl.FRAMEBUFFER,\n gl.COLOR_ATTACHMENT0,\n gl.TEXTURE_2D,\n this.texture,\n 0\n );\n const result = new Float32Array(size[0] * size[1] * 4);\n gl.readPixels(0, 0, size[0], size[1], gl.RGBA, gl.FLOAT, result);\n return result;\n }\n renderValues() {\n if (this._deleted) return null;\n return this.renderRawOutput();\n }\n toArray() {\n return utils.erectFloat(this.renderValues(), this.output[0]);\n }\n}\n\nmodule.exports = {\n GLTextureFloat\n};" }, { "path": "src/backend/gl/texture/graphical.js", "content": "const { GLTextureUnsigned } = require('./unsigned');\n\nclass GLTextureGraphical extends GLTextureUnsigned {\n constructor(settings) {\n super(settings);\n this.type = 'ArrayTexture(4)';\n }\n toArray() {\n return this.renderValues();\n }\n}\n\nmodule.exports = {\n GLTextureGraphical\n};" }, { "path": "src/backend/gl/texture/index.js", "content": "const { Texture } = require('../../../texture');\n\n/**\n * @class\n * @property framebuffer\n * @extends Texture\n */\nclass GLTexture extends Texture {\n /**\n * @returns {Number}\n * @abstract\n */\n get textureType() {\n throw new Error(`\"textureType\" not implemented on ${ this.name }`);\n }\n\n clone() {\n return new this.constructor(this);\n }\n\n /**\n * @returns {Boolean}\n */\n beforeMutate() {\n if (this.texture._refs > 1) {\n this.newTexture();\n return true;\n }\n return false;\n }\n\n /**\n * @private\n */\n cloneTexture() {\n this.texture._refs--;\n const { context: gl, size, texture, kernel } = this;\n if (kernel.debug) {\n console.warn('cloning internal texture');\n }\n gl.bindFramebuffer(gl.FRAMEBUFFER, this.framebuffer());\n selectTexture(gl, texture);\n gl.framebufferTexture2D(gl.FRAMEBUFFER, gl.COLOR_ATTACHMENT0, gl.TEXTURE_2D, texture, 0);\n const target = gl.createTexture();\n selectTexture(gl, target);\n gl.texImage2D(gl.TEXTURE_2D, 0, this.internalFormat, size[0], size[1], 0, this.textureFormat, this.textureType, null);\n gl.copyTexSubImage2D(gl.TEXTURE_2D, 0, 0, 0, 0, 0, size[0], size[1]);\n target._refs = 1;\n this.texture = target;\n }\n\n /**\n * @private\n */\n newTexture() {\n this.texture._refs--;\n const gl = this.context;\n const size = this.size;\n const kernel = this.kernel;\n if (kernel.debug) {\n console.warn('new internal texture');\n }\n const target = gl.createTexture();\n selectTexture(gl, target);\n gl.texImage2D(gl.TEXTURE_2D, 0, this.internalFormat, size[0], size[1], 0, this.textureFormat, this.textureType, null);\n target._refs = 1;\n this.texture = target;\n }\n\n clear() {\n if (this.texture._refs) {\n this.texture._refs--;\n const gl = this.context;\n const target = this.texture = gl.createTexture();\n selectTexture(gl, target);\n const size = this.size;\n target._refs = 1;\n gl.texImage2D(gl.TEXTURE_2D, 0, this.internalFormat, size[0], size[1], 0, this.textureFormat, this.textureType, null);\n }\n const { context: gl, texture } = this;\n gl.bindFramebuffer(gl.FRAMEBUFFER, this.framebuffer());\n gl.bindTexture(gl.TEXTURE_2D, texture);\n selectTexture(gl, texture);\n gl.framebufferTexture2D(gl.FRAMEBUFFER, gl.COLOR_ATTACHMENT0, gl.TEXTURE_2D, texture, 0);\n gl.clearColor(0, 0, 0, 0);\n gl.clear(gl.COLOR_BUFFER_BIT | gl.DEPTH_BUFFER_BIT);\n }\n\n delete() {\n if (this._deleted) return;\n this._deleted = true;\n if (this.texture._refs) {\n this.texture._refs--;\n if (this.texture._refs) return;\n }\n this.context.deleteTexture(this.texture);\n // TODO: Remove me\n // if (this.texture._refs === 0 && this._framebuffer) {\n // this.context.deleteFramebuffer(this._framebuffer);\n // this._framebuffer = null;\n // }\n }\n\n framebuffer() {\n if (!this._framebuffer) {\n this._framebuffer = this.kernel.getRawValueFramebuffer(this.size[0], this.size[1]);\n }\n return this._framebuffer;\n }\n}\n\nfunction selectTexture(gl, texture) {\n /* Maximum a texture can be, so that collision is highly unlikely\n * basically gl.TEXTURE15 + gl.getParameter(gl.MAX_TEXTURE_IMAGE_UNITS);\n * Was gl.TEXTURE31, but safari didn't like it\n * */\n gl.activeTexture(gl.TEXTURE15);\n gl.bindTexture(gl.TEXTURE_2D, texture);\n gl.texParameteri(gl.TEXTURE_2D, gl.TEXTURE_WRAP_S, gl.CLAMP_TO_EDGE);\n gl.texParameteri(gl.TEXTURE_2D, gl.TEXTURE_WRAP_T, gl.CLAMP_TO_EDGE);\n gl.texParameteri(gl.TEXTURE_2D, gl.TEXTURE_MIN_FILTER, gl.NEAREST);\n gl.texParameteri(gl.TEXTURE_2D, gl.TEXTURE_MAG_FILTER, gl.NEAREST);\n}\n\nmodule.exports = { GLTexture };" }, { "path": "src/backend/gl/texture/memory-optimized-2d.js", "content": "const { utils } = require('../../../utils');\nconst { GLTextureFloat } = require('./float');\n\nclass GLTextureMemoryOptimized2D extends GLTextureFloat {\n constructor(settings) {\n super(settings);\n this.type = 'MemoryOptimizedNumberTexture';\n }\n toArray() {\n return utils.erectMemoryOptimized2DFloat(this.renderValues(), this.output[0], this.output[1]);\n }\n}\n\nmodule.exports = {\n GLTextureMemoryOptimized2D\n};" }, { "path": "src/backend/gl/texture/memory-optimized-3d.js", "content": "const { utils } = require('../../../utils');\nconst { GLTextureFloat } = require('./float');\n\nclass GLTextureMemoryOptimized3D extends GLTextureFloat {\n constructor(settings) {\n super(settings);\n this.type = 'MemoryOptimizedNumberTexture';\n }\n toArray() {\n return utils.erectMemoryOptimized3DFloat(this.renderValues(), this.output[0], this.output[1], this.output[2]);\n }\n}\n\nmodule.exports = {\n GLTextureMemoryOptimized3D\n};" }, { "path": "src/backend/gl/texture/memory-optimized.js", "content": "const { utils } = require('../../../utils');\nconst { GLTextureFloat } = require('./float');\n\nclass GLTextureMemoryOptimized extends GLTextureFloat {\n constructor(settings) {\n super(settings);\n this.type = 'MemoryOptimizedNumberTexture';\n }\n toArray() {\n return utils.erectMemoryOptimizedFloat(this.renderValues(), this.output[0]);\n }\n}\n\nmodule.exports = {\n GLTextureMemoryOptimized\n};" }, { "path": "src/backend/gl/texture/unsigned-2d.js", "content": "const { utils } = require('../../../utils');\nconst { GLTextureUnsigned } = require('./unsigned');\n\nclass GLTextureUnsigned2D extends GLTextureUnsigned {\n constructor(settings) {\n super(settings);\n this.type = 'NumberTexture';\n }\n toArray() {\n return utils.erect2DPackedFloat(this.renderValues(), this.output[0], this.output[1]);\n }\n}\n\nmodule.exports = {\n GLTextureUnsigned2D\n};" }, { "path": "src/backend/gl/texture/unsigned-3d.js", "content": "const { utils } = require('../../../utils');\nconst { GLTextureUnsigned } = require('./unsigned');\n\nclass GLTextureUnsigned3D extends GLTextureUnsigned {\n constructor(settings) {\n super(settings);\n this.type = 'NumberTexture';\n }\n toArray() {\n return utils.erect3DPackedFloat(this.renderValues(), this.output[0], this.output[1], this.output[2]);\n }\n}\n\nmodule.exports = {\n GLTextureUnsigned3D\n};" }, { "path": "src/backend/gl/texture/unsigned.js", "content": "const { utils } = require('../../../utils');\nconst { GLTexture } = require('./index');\n\nclass GLTextureUnsigned extends GLTexture {\n get textureType() {\n return this.context.UNSIGNED_BYTE;\n }\n constructor(settings) {\n super(settings);\n this.type = 'NumberTexture';\n }\n renderRawOutput() {\n const { context: gl } = this;\n gl.bindFramebuffer(gl.FRAMEBUFFER, this.framebuffer());\n gl.framebufferTexture2D(\n gl.FRAMEBUFFER,\n gl.COLOR_ATTACHMENT0,\n gl.TEXTURE_2D,\n this.texture,\n 0\n );\n const result = new Uint8Array(this.size[0] * this.size[1] * 4);\n gl.readPixels(0, 0, this.size[0], this.size[1], gl.RGBA, gl.UNSIGNED_BYTE, result);\n return result;\n }\n renderValues() {\n if (this._deleted) return null;\n return new Float32Array(this.renderRawOutput().buffer);\n }\n toArray() {\n return utils.erectPackedFloat(this.renderValues(), this.output[0]);\n }\n}\n\nmodule.exports = {\n GLTextureUnsigned\n};" }, { "path": "src/backend/headless-gl/kernel.js", "content": "const getContext = require('gl');\nconst { WebGLKernel } = require('../web-gl/kernel');\nconst { glKernelString } = require('../gl/kernel-string');\n\nlet isSupported = null;\nlet testCanvas = null;\nlet testContext = null;\nlet testExtensions = null;\nlet features = null;\n\nclass HeadlessGLKernel extends WebGLKernel {\n static get isSupported() {\n if (isSupported !== null) return isSupported;\n this.setupFeatureChecks();\n isSupported = testContext !== null;\n return isSupported;\n }\n\n static setupFeatureChecks() {\n testCanvas = null;\n testExtensions = null;\n if (typeof getContext !== 'function') return;\n try { // just in case, edge cases\n testContext = getContext(2, 2, {\n preserveDrawingBuffer: true\n });\n if (!testContext || !testContext.getExtension) return;\n testExtensions = {\n STACKGL_resize_drawingbuffer: testContext.getExtension('STACKGL_resize_drawingbuffer'),\n STACKGL_destroy_context: testContext.getExtension('STACKGL_destroy_context'),\n OES_texture_float: testContext.getExtension('OES_texture_float'),\n OES_texture_float_linear: testContext.getExtension('OES_texture_float_linear'),\n OES_element_index_uint: testContext.getExtension('OES_element_index_uint'),\n WEBGL_draw_buffers: testContext.getExtension('WEBGL_draw_buffers'),\n WEBGL_color_buffer_float: testContext.getExtension('WEBGL_color_buffer_float'),\n };\n features = this.getFeatures();\n } catch (e) {\n console.warn(e);\n }\n }\n\n static isContextMatch(context) {\n try {\n return context.getParameter(context.RENDERER) === 'ANGLE';\n } catch (e) {\n return false;\n }\n }\n\n static getIsTextureFloat() {\n return Boolean(testExtensions.OES_texture_float);\n }\n\n static getIsDrawBuffers() {\n return Boolean(testExtensions.WEBGL_draw_buffers);\n }\n\n static getChannelCount() {\n return testExtensions.WEBGL_draw_buffers ?\n testContext.getParameter(testExtensions.WEBGL_draw_buffers.MAX_DRAW_BUFFERS_WEBGL) :\n 1;\n }\n\n static getMaxTextureSize() {\n return testContext.getParameter(testContext.MAX_TEXTURE_SIZE);\n }\n\n static get testCanvas() {\n return testCanvas;\n }\n\n static get testContext() {\n return testContext;\n }\n\n static get features() {\n return features;\n }\n\n initCanvas() {\n return {};\n }\n\n initContext() {\n return getContext(2, 2, {\n preserveDrawingBuffer: true\n });\n }\n\n initExtensions() {\n this.extensions = {\n STACKGL_resize_drawingbuffer: this.context.getExtension('STACKGL_resize_drawingbuffer'),\n STACKGL_destroy_context: this.context.getExtension('STACKGL_destroy_context'),\n OES_texture_float: this.context.getExtension('OES_texture_float'),\n OES_texture_float_linear: this.context.getExtension('OES_texture_float_linear'),\n OES_element_index_uint: this.context.getExtension('OES_element_index_uint'),\n WEBGL_draw_buffers: this.context.getExtension('WEBGL_draw_buffers'),\n };\n }\n\n build() {\n super.build.apply(this, arguments);\n if (!this.fallbackRequested) {\n this.extensions.STACKGL_resize_drawingbuffer.resize(this.maxTexSize[0], this.maxTexSize[1]);\n }\n }\n\n destroyExtensions() {\n this.extensions.STACKGL_resize_drawingbuffer = null;\n this.extensions.STACKGL_destroy_context = null;\n this.extensions.OES_texture_float = null;\n this.extensions.OES_texture_float_linear = null;\n this.extensions.OES_element_index_uint = null;\n this.extensions.WEBGL_draw_buffers = null;\n }\n\n static destroyContext(context) {\n const extension = context.getExtension('STACKGL_destroy_context');\n if (extension && extension.destroy) {\n extension.destroy();\n }\n }\n\n /**\n * @desc Returns the *pre-compiled* Kernel as a JS Object String, that can be reused.\n */\n toString() {\n const setupContextString = `const gl = context || require('gl')(1, 1);\\n`;\n const destroyContextString = ` if (!context) { gl.getExtension('STACKGL_destroy_context').destroy(); }\\n`;\n return glKernelString(this.constructor, arguments, this, setupContextString, destroyContextString);\n }\n\n setOutput(output) {\n super.setOutput(output);\n if (this.graphical && this.extensions.STACKGL_resize_drawingbuffer) {\n this.extensions.STACKGL_resize_drawingbuffer.resize(this.maxTexSize[0], this.maxTexSize[1]);\n }\n return this;\n }\n}\n\nmodule.exports = {\n HeadlessGLKernel\n};" }, { "path": "src/backend/kernel-value.js", "content": "/**\n * @class KernelValue\n */\nclass KernelValue {\n /**\n * @param {KernelVariable} value\n * @param {IKernelValueSettings} settings\n */\n constructor(value, settings) {\n const {\n name,\n kernel,\n context,\n checkContext,\n onRequestContextHandle,\n onUpdateValueMismatch,\n origin,\n strictIntegers,\n type,\n tactic,\n } = settings;\n if (!name) {\n throw new Error('name not set');\n }\n if (!type) {\n throw new Error('type not set');\n }\n if (!origin) {\n throw new Error('origin not set');\n }\n if (origin !== 'user' && origin !== 'constants') {\n throw new Error(`origin must be \"user\" or \"constants\" value is \"${ origin }\"`);\n }\n if (!onRequestContextHandle) {\n throw new Error('onRequestContextHandle is not set');\n }\n this.name = name;\n this.origin = origin;\n this.tactic = tactic;\n this.varName = origin === 'constants' ? `constants.${name}` : name;\n this.kernel = kernel;\n this.strictIntegers = strictIntegers;\n // handle textures\n this.type = value.type || type;\n this.size = value.size || null;\n this.index = null;\n this.context = context;\n this.checkContext = checkContext !== null && checkContext !== undefined ? checkContext : true;\n this.contextHandle = null;\n this.onRequestContextHandle = onRequestContextHandle;\n this.onUpdateValueMismatch = onUpdateValueMismatch;\n this.forceUploadEachRun = null;\n }\n\n get id() {\n return `${this.origin}_${name}`;\n }\n\n getSource() {\n throw new Error(`\"getSource\" not defined on ${ this.constructor.name }`);\n }\n\n updateValue(value) {\n throw new Error(`\"updateValue\" not defined on ${ this.constructor.name }`);\n }\n}\n\nmodule.exports = {\n KernelValue\n};" }, { "path": "src/backend/kernel.js", "content": "const { utils } = require('../utils');\nconst { Input } = require('../input');\n\nclass Kernel {\n /**\n * @type {Boolean}\n */\n static get isSupported() {\n throw new Error(`\"isSupported\" not implemented on ${ this.name }`);\n }\n\n /**\n * @abstract\n * @returns {Boolean}\n */\n static isContextMatch(context) {\n throw new Error(`\"isContextMatch\" not implemented on ${ this.name }`);\n }\n\n /**\n * @type {IKernelFeatures}\n * Used internally to populate the kernel.feature, which is a getter for the output of this value\n */\n static getFeatures() {\n throw new Error(`\"getFeatures\" not implemented on ${ this.name }`);\n }\n\n static destroyContext(context) {\n throw new Error(`\"destroyContext\" called on ${ this.name }`);\n }\n\n static nativeFunctionArguments() {\n throw new Error(`\"nativeFunctionArguments\" called on ${ this.name }`);\n }\n\n static nativeFunctionReturnType() {\n throw new Error(`\"nativeFunctionReturnType\" called on ${ this.name }`);\n }\n\n static combineKernels() {\n throw new Error(`\"combineKernels\" called on ${ this.name }`);\n }\n\n /**\n *\n * @param {string|IKernelJSON} source\n * @param [settings]\n */\n constructor(source, settings) {\n if (typeof source !== 'object') {\n if (typeof source !== 'string') {\n throw new Error('source not a string');\n }\n if (!utils.isFunctionString(source)) {\n throw new Error('source not a function string');\n }\n }\n this.useLegacyEncoder = false;\n this.fallbackRequested = false;\n this.onRequestFallback = null;\n\n /**\n * Name of the arguments found from parsing source argument\n * @type {String[]}\n */\n this.argumentNames = typeof source === 'string' ? utils.getArgumentNamesFromString(source) : null;\n this.argumentTypes = null;\n this.argumentSizes = null;\n this.argumentBitRatios = null;\n this.kernelArguments = null;\n this.kernelConstants = null;\n this.forceUploadKernelConstants = null;\n\n\n /**\n * The function source\n * @type {String|IKernelJSON}\n */\n this.source = source;\n\n /**\n * The size of the kernel's output\n * @type {Number[]}\n */\n this.output = null;\n\n /**\n * Debug mode\n * @type {Boolean}\n */\n this.debug = false;\n\n /**\n * Graphical mode\n * @type {Boolean}\n */\n this.graphical = false;\n\n /**\n * Maximum loops when using argument values to prevent infinity\n * @type {Number}\n */\n this.loopMaxIterations = 0;\n\n /**\n * Constants used in kernel via `this.constants`\n * @type {Object}\n */\n this.constants = null;\n\n /**\n *\n * @type {Object.This method calls a helper method *renderOutput* to return the result.
\n * @returns {Float32Array|Float32Array[]|Float32Array[][]|void} Result The final output of the program, as float, and as Textures for reuse.\n * @abstract\n */\n run() {\n throw new Error(`\"run\" not defined on ${ this.constructor.name }`)\n }\n\n /**\n * @abstract\n * @return {Object}\n */\n initCanvas() {\n throw new Error(`\"initCanvas\" not defined on ${ this.constructor.name }`);\n }\n\n /**\n * @abstract\n * @return {Object}\n */\n initContext() {\n throw new Error(`\"initContext\" not defined on ${ this.constructor.name }`);\n }\n\n /**\n * @param {IDirectKernelSettings} settings\n * @return {string[]};\n * @abstract\n */\n initPlugins(settings) {\n throw new Error(`\"initPlugins\" not defined on ${ this.constructor.name }`);\n }\n\n /**\n *\n * @param {KernelFunction|string|IGPUFunction} source\n * @param {IFunctionSettings} [settings]\n * @return {Kernel}\n */\n addFunction(source, settings = {}) {\n if (source.name && source.source && source.argumentTypes && 'returnType' in source) {\n this.functions.push(source);\n } else if ('settings' in source && 'source' in source) {\n this.functions.push(this.functionToIGPUFunction(source.source, source.settings));\n } else if (typeof source === 'string' || typeof source === 'function') {\n this.functions.push(this.functionToIGPUFunction(source, settings));\n } else {\n throw new Error(`function not properly defined`);\n }\n return this;\n }\n\n /**\n *\n * @param {string} name\n * @param {string} source\n * @param {IGPUFunctionSettings} [settings]\n */\n addNativeFunction(name, source, settings = {}) {\n const { argumentTypes, argumentNames } = settings.argumentTypes ?\n splitArgumentTypes(settings.argumentTypes) :\n this.constructor.nativeFunctionArguments(source) || {};\n this.nativeFunctions.push({\n name,\n source,\n settings,\n argumentTypes,\n argumentNames,\n returnType: settings.returnType || this.constructor.nativeFunctionReturnType(source)\n });\n return this;\n }\n\n /**\n * @desc Setup the parameter types for the parameters\n * supplied to the Kernel function\n *\n * @param {IArguments} args - The actual parameters sent to the Kernel\n */\n setupArguments(args) {\n this.kernelArguments = [];\n if (!this.argumentTypes) {\n if (!this.argumentTypes) {\n this.argumentTypes = [];\n for (let i = 0; i < args.length; i++) {\n const argType = utils.getVariableType(args[i], this.strictIntegers);\n const type = argType === 'Integer' ? 'Number' : argType;\n this.argumentTypes.push(type);\n this.kernelArguments.push({\n type\n });\n }\n }\n } else {\n for (let i = 0; i < this.argumentTypes.length; i++) {\n this.kernelArguments.push({\n type: this.argumentTypes[i]\n });\n }\n }\n\n // setup sizes\n this.argumentSizes = new Array(args.length);\n this.argumentBitRatios = new Int32Array(args.length);\n\n for (let i = 0; i < args.length; i++) {\n const arg = args[i];\n this.argumentSizes[i] = arg.constructor === Input ? arg.size : null;\n this.argumentBitRatios[i] = this.getBitRatio(arg);\n }\n\n if (this.argumentNames.length !== args.length) {\n throw new Error(`arguments are miss-aligned`);\n }\n }\n\n /**\n * Setup constants\n */\n setupConstants() {\n this.kernelConstants = [];\n let needsConstantTypes = this.constantTypes === null;\n if (needsConstantTypes) {\n this.constantTypes = {};\n }\n this.constantBitRatios = {};\n if (this.constants) {\n for (let name in this.constants) {\n if (needsConstantTypes) {\n const type = utils.getVariableType(this.constants[name], this.strictIntegers);\n this.constantTypes[name] = type;\n this.kernelConstants.push({\n name,\n type\n });\n } else {\n this.kernelConstants.push({\n name,\n type: this.constantTypes[name]\n });\n }\n this.constantBitRatios[name] = this.getBitRatio(this.constants[name]);\n }\n }\n }\n\n /**\n *\n * @param flag\n * @return {this}\n */\n setOptimizeFloatMemory(flag) {\n this.optimizeFloatMemory = flag;\n return this;\n }\n\n /**\n *\n * @param {Array|Object} output\n * @return {number[]}\n */\n toKernelOutput(output) {\n if (output.hasOwnProperty('x')) {\n if (output.hasOwnProperty('y')) {\n if (output.hasOwnProperty('z')) {\n return [output.x, output.y, output.z];\n } else {\n return [output.x, output.y];\n }\n } else {\n return [output.x];\n }\n } else {\n return output;\n }\n }\n\n /**\n * @desc Set output dimensions of the kernel function\n * @param {Array|Object} output - The output array to set the kernel output size to\n * @return {this}\n */\n setOutput(output) {\n this.output = this.toKernelOutput(output);\n return this;\n }\n\n /**\n * @desc Toggle debug mode\n * @param {Boolean} flag - true to enable debug\n * @return {this}\n */\n setDebug(flag) {\n this.debug = flag;\n return this;\n }\n\n /**\n * @desc Toggle graphical output mode\n * @param {Boolean} flag - true to enable graphical output\n * @return {this}\n */\n setGraphical(flag) {\n this.graphical = flag;\n this.precision = 'unsigned';\n return this;\n }\n\n /**\n * @desc Set the maximum number of loop iterations\n * @param {number} max - iterations count\n * @return {this}\n */\n setLoopMaxIterations(max) {\n this.loopMaxIterations = max;\n return this;\n }\n\n /**\n * @desc Set Constants\n * @return {this}\n */\n setConstants(constants) {\n this.constants = constants;\n return this;\n }\n\n /**\n *\n * @param {IKernelValueTypes} constantTypes\n * @return {this}\n */\n setConstantTypes(constantTypes) {\n this.constantTypes = constantTypes;\n return this;\n }\n\n /**\n *\n * @param {IFunction[]|KernelFunction[]} functions\n * @return {this}\n */\n setFunctions(functions) {\n for (let i = 0; i < functions.length; i++) {\n this.addFunction(functions[i]);\n }\n return this;\n }\n\n /**\n *\n * @param {IGPUNativeFunction[]} nativeFunctions\n * @return {this}\n */\n setNativeFunctions(nativeFunctions) {\n for (let i = 0; i < nativeFunctions.length; i++) {\n const settings = nativeFunctions[i];\n const { name, source } = settings;\n this.addNativeFunction(name, source, settings);\n }\n return this;\n }\n\n /**\n *\n * @param {String} injectedNative\n * @return {this}\n */\n setInjectedNative(injectedNative) {\n this.injectedNative = injectedNative;\n return this;\n }\n\n /**\n * Set writing to texture on/off\n * @param flag\n * @return {this}\n */\n setPipeline(flag) {\n this.pipeline = flag;\n return this;\n }\n\n /**\n * Set precision to 'unsigned' or 'single'\n * @param {String} flag 'unsigned' or 'single'\n * @return {this}\n */\n setPrecision(flag) {\n this.precision = flag;\n return this;\n }\n\n /**\n * @param flag\n * @return {Kernel}\n * @deprecated\n */\n setDimensions(flag) {\n utils.warnDeprecated('method', 'setDimensions', 'setOutput');\n this.output = flag;\n return this;\n }\n\n /**\n * @param flag\n * @return {this}\n * @deprecated\n */\n setOutputToTexture(flag) {\n utils.warnDeprecated('method', 'setOutputToTexture', 'setPipeline');\n this.pipeline = flag;\n return this;\n }\n\n /**\n * Set to immutable\n * @param flag\n * @return {this}\n */\n setImmutable(flag) {\n this.immutable = flag;\n return this;\n }\n\n /**\n * @desc Bind the canvas to kernel\n * @param {Object} canvas\n * @return {this}\n */\n setCanvas(canvas) {\n this.canvas = canvas;\n return this;\n }\n\n /**\n * @param {Boolean} flag\n * @return {this}\n */\n setStrictIntegers(flag) {\n this.strictIntegers = flag;\n return this;\n }\n\n /**\n *\n * @param flag\n * @return {this}\n */\n setDynamicOutput(flag) {\n this.dynamicOutput = flag;\n return this;\n }\n\n /**\n * @deprecated\n * @param flag\n * @return {this}\n */\n setHardcodeConstants(flag) {\n utils.warnDeprecated('method', 'setHardcodeConstants');\n this.setDynamicOutput(flag);\n this.setDynamicArguments(flag);\n return this;\n }\n\n /**\n *\n * @param flag\n * @return {this}\n */\n setDynamicArguments(flag) {\n this.dynamicArguments = flag;\n return this;\n }\n\n /**\n * @param {Boolean} flag\n * @return {this}\n */\n setUseLegacyEncoder(flag) {\n this.useLegacyEncoder = flag;\n return this;\n }\n\n /**\n *\n * @param {Boolean} flag\n * @return {this}\n */\n setWarnVarUsage(flag) {\n utils.warnDeprecated('method', 'setWarnVarUsage');\n return this;\n }\n\n /**\n * @deprecated\n * @returns {Object}\n */\n getCanvas() {\n utils.warnDeprecated('method', 'getCanvas');\n return this.canvas;\n }\n\n /**\n * @deprecated\n * @returns {Object}\n */\n getWebGl() {\n utils.warnDeprecated('method', 'getWebGl');\n return this.context;\n }\n\n /**\n * @desc Bind the webGL instance to kernel\n * @param {WebGLRenderingContext} context - webGl instance to bind\n */\n setContext(context) {\n this.context = context;\n return this;\n }\n\n /**\n *\n * @param {IKernelValueTypes|GPUVariableType[]} argumentTypes\n * @return {this}\n */\n setArgumentTypes(argumentTypes) {\n if (Array.isArray(argumentTypes)) {\n this.argumentTypes = argumentTypes;\n } else {\n this.argumentTypes = [];\n for (const p in argumentTypes) {\n if (!argumentTypes.hasOwnProperty(p)) continue;\n const argumentIndex = this.argumentNames.indexOf(p);\n if (argumentIndex === -1) throw new Error(`unable to find argument ${ p }`);\n this.argumentTypes[argumentIndex] = argumentTypes[p];\n }\n }\n return this;\n }\n\n /**\n *\n * @param {Tactic} tactic\n * @return {this}\n */\n setTactic(tactic) {\n this.tactic = tactic;\n return this;\n }\n\n requestFallback(args) {\n if (!this.onRequestFallback) {\n throw new Error(`\"onRequestFallback\" not defined on ${ this.constructor.name }`);\n }\n this.fallbackRequested = true;\n return this.onRequestFallback(args);\n }\n\n /**\n * @desc Validate settings\n * @abstract\n */\n validateSettings() {\n throw new Error(`\"validateSettings\" not defined on ${ this.constructor.name }`);\n }\n\n /**\n * @desc Add a sub kernel to the root kernel instance.\n * This is what `createKernelMap` uses.\n *\n * @param {ISubKernel} subKernel - function (as a String) of the subKernel to add\n */\n addSubKernel(subKernel) {\n if (this.subKernels === null) {\n this.subKernels = [];\n }\n if (!subKernel.source) throw new Error('subKernel missing \"source\" property');\n if (!subKernel.property && isNaN(subKernel.property)) throw new Error('subKernel missing \"property\" property');\n if (!subKernel.name) throw new Error('subKernel missing \"name\" property');\n this.subKernels.push(subKernel);\n return this;\n }\n\n /**\n * @desc Destroys all memory associated with this kernel\n * @param {Boolean} [removeCanvasReferences] remove any associated canvas references\n */\n destroy(removeCanvasReferences) {\n throw new Error(`\"destroy\" called on ${ this.constructor.name }`);\n }\n\n /**\n * bit storage ratio of source to target 'buffer', i.e. if 8bit array -> 32bit tex = 4\n * @param value\n * @returns {number}\n */\n getBitRatio(value) {\n if (this.precision === 'single') {\n // 8 and 16 are up-converted to float32\n return 4;\n } else if (Array.isArray(value[0])) {\n return this.getBitRatio(value[0]);\n } else if (value.constructor === Input) {\n return this.getBitRatio(value.value);\n }\n switch (value.constructor) {\n case Uint8ClampedArray:\n case Uint8Array:\n case Int8Array:\n return 1;\n case Uint16Array:\n case Int16Array:\n return 2;\n case Float32Array:\n case Int32Array:\n default:\n return 4;\n }\n }\n\n /**\n * @param {Boolean} [flip]\n * @returns {Uint8ClampedArray}\n */\n getPixels(flip) {\n throw new Error(`\"getPixels\" called on ${ this.constructor.name }`);\n }\n\n checkOutput() {\n if (!this.output || !utils.isArray(this.output)) throw new Error('kernel.output not an array');\n if (this.output.length < 1) throw new Error('kernel.output is empty, needs at least 1 value');\n for (let i = 0; i < this.output.length; i++) {\n if (isNaN(this.output[i]) || this.output[i] < 1) {\n throw new Error(`${ this.constructor.name }.output[${ i }] incorrectly defined as \\`${ this.output[i] }\\`, needs to be numeric, and greater than 0`);\n }\n }\n }\n\n /**\n *\n * @param {String} value\n */\n prependString(value) {\n throw new Error(`\"prependString\" called on ${ this.constructor.name }`);\n }\n\n /**\n *\n * @param {String} value\n * @return Boolean\n */\n hasPrependString(value) {\n throw new Error(`\"hasPrependString\" called on ${ this.constructor.name }`);\n }\n\n /**\n * @return {IKernelJSON}\n */\n toJSON() {\n return {\n settings: {\n output: this.output,\n pipeline: this.pipeline,\n argumentNames: this.argumentNames,\n argumentsTypes: this.argumentTypes,\n constants: this.constants,\n pluginNames: this.plugins ? this.plugins.map(plugin => plugin.name) : null,\n returnType: this.returnType,\n }\n };\n }\n\n /**\n * @param {IArguments} args\n */\n buildSignature(args) {\n const Constructor = this.constructor;\n this.signature = Constructor.getSignature(this, Constructor.getArgumentTypes(this, args));\n }\n\n /**\n * @param {Kernel} kernel\n * @param {IArguments} args\n * @returns GPUVariableType[]\n */\n static getArgumentTypes(kernel, args) {\n const argumentTypes = new Array(args.length);\n for (let i = 0; i < args.length; i++) {\n const arg = args[i];\n const type = kernel.argumentTypes[i];\n if (arg.type) {\n argumentTypes[i] = arg.type;\n } else {\n switch (type) {\n case 'Number':\n case 'Integer':\n case 'Float':\n case 'ArrayTexture(1)':\n argumentTypes[i] = utils.getVariableType(arg);\n break;\n default:\n argumentTypes[i] = type;\n }\n }\n }\n return argumentTypes;\n }\n\n /**\n *\n * @param {Kernel} kernel\n * @param {GPUVariableType[]} argumentTypes\n * @abstract\n */\n static getSignature(kernel, argumentTypes) {\n throw new Error(`\"getSignature\" not implemented on ${ this.name }`);\n }\n\n /**\n *\n * @param {String|Function} source\n * @param {IFunctionSettings} [settings]\n * @returns {IGPUFunction}\n */\n functionToIGPUFunction(source, settings = {}) {\n if (typeof source !== 'string' && typeof source !== 'function') throw new Error('source not a string or function');\n const sourceString = typeof source === 'string' ? source : source.toString();\n let argumentTypes = [];\n\n if (Array.isArray(settings.argumentTypes)) {\n argumentTypes = settings.argumentTypes;\n } else if (typeof settings.argumentTypes === 'object') {\n argumentTypes = utils.getArgumentNamesFromString(sourceString)\n .map(name => settings.argumentTypes[name]) || [];\n } else {\n argumentTypes = settings.argumentTypes || [];\n }\n\n return {\n name: utils.getFunctionNameFromString(sourceString) || null,\n source: sourceString,\n argumentTypes,\n returnType: settings.returnType || null,\n };\n }\n\n /**\n *\n * @param {Kernel} previousKernel\n * @abstract\n */\n onActivate(previousKernel) {}\n}\n\nfunction splitArgumentTypes(argumentTypesObject) {\n const argumentNames = Object.keys(argumentTypesObject);\n const argumentTypes = [];\n for (let i = 0; i < argumentNames.length; i++) {\n const argumentName = argumentNames[i];\n argumentTypes.push(argumentTypesObject[argumentName]);\n }\n return { argumentTypes, argumentNames };\n}\n\nmodule.exports = {\n Kernel\n};" }, { "path": "src/backend/web-gl/fragment-shader.js", "content": "// language=GLSL\nconst fragmentShader = `__HEADER__;\n__FLOAT_TACTIC_DECLARATION__;\n__INT_TACTIC_DECLARATION__;\n__SAMPLER_2D_TACTIC_DECLARATION__;\n\nconst int LOOP_MAX = __LOOP_MAX__;\n\n__PLUGINS__;\n__CONSTANTS__;\n\nvarying vec2 vTexCoord;\n\nfloat acosh(float x) {\n return log(x + sqrt(x * x - 1.0));\n}\n\nfloat sinh(float x) {\n return (pow(${Math.E}, x) - pow(${Math.E}, -x)) / 2.0;\n}\n\nfloat asinh(float x) {\n return log(x + sqrt(x * x + 1.0));\n}\n\nfloat atan2(float v1, float v2) {\n if (v1 == 0.0 || v2 == 0.0) return 0.0;\n return atan(v1 / v2);\n}\n\nfloat atanh(float x) {\n x = (x + 1.0) / (x - 1.0);\n if (x < 0.0) {\n return 0.5 * log(-x);\n }\n return 0.5 * log(x);\n}\n\nfloat cbrt(float x) {\n if (x >= 0.0) {\n return pow(x, 1.0 / 3.0);\n } else {\n return -pow(x, 1.0 / 3.0);\n }\n}\n\nfloat cosh(float x) {\n return (pow(${Math.E}, x) + pow(${Math.E}, -x)) / 2.0; \n}\n\nfloat expm1(float x) {\n return pow(${Math.E}, x) - 1.0; \n}\n\nfloat fround(highp float x) {\n return x;\n}\n\nfloat imul(float v1, float v2) {\n return float(int(v1) * int(v2));\n}\n\nfloat log10(float x) {\n return log2(x) * (1.0 / log2(10.0));\n}\n\nfloat log1p(float x) {\n return log(1.0 + x);\n}\n\nfloat _pow(float v1, float v2) {\n if (v2 == 0.0) return 1.0;\n return pow(v1, v2);\n}\n\nfloat tanh(float x) {\n float e = exp(2.0 * x);\n return (e - 1.0) / (e + 1.0);\n}\n\nfloat trunc(float x) {\n if (x >= 0.0) {\n return floor(x); \n } else {\n return ceil(x);\n }\n}\n\nvec4 _round(vec4 x) {\n return floor(x + 0.5);\n}\n\nfloat _round(float x) {\n return floor(x + 0.5);\n}\n\nconst int BIT_COUNT = 32;\nint modi(int x, int y) {\n return x - y * (x / y);\n}\n\nint bitwiseOr(int a, int b) {\n int result = 0;\n int n = 1;\n \n for (int i = 0; i < BIT_COUNT; i++) {\n if ((modi(a, 2) == 1) || (modi(b, 2) == 1)) {\n result += n;\n }\n a = a / 2;\n b = b / 2;\n n = n * 2;\n if(!(a > 0 || b > 0)) {\n break;\n }\n }\n return result;\n}\nint bitwiseXOR(int a, int b) {\n int result = 0;\n int n = 1;\n \n for (int i = 0; i < BIT_COUNT; i++) {\n if ((modi(a, 2) == 1) != (modi(b, 2) == 1)) {\n result += n;\n }\n a = a / 2;\n b = b / 2;\n n = n * 2;\n if(!(a > 0 || b > 0)) {\n break;\n }\n }\n return result;\n}\nint bitwiseAnd(int a, int b) {\n int result = 0;\n int n = 1;\n for (int i = 0; i < BIT_COUNT; i++) {\n if ((modi(a, 2) == 1) && (modi(b, 2) == 1)) {\n result += n;\n }\n a = a / 2;\n b = b / 2;\n n = n * 2;\n if(!(a > 0 && b > 0)) {\n break;\n }\n }\n return result;\n}\nint bitwiseNot(int a) {\n int result = 0;\n int n = 1;\n \n for (int i = 0; i < BIT_COUNT; i++) {\n if (modi(a, 2) == 0) {\n result += n; \n }\n a = a / 2;\n n = n * 2;\n }\n return result;\n}\nint bitwiseZeroFillLeftShift(int n, int shift) {\n int maxBytes = BIT_COUNT;\n for (int i = 0; i < BIT_COUNT; i++) {\n if (maxBytes >= n) {\n break;\n }\n maxBytes *= 2;\n }\n for (int i = 0; i < BIT_COUNT; i++) {\n if (i >= shift) {\n break;\n }\n n *= 2;\n }\n\n int result = 0;\n int byteVal = 1;\n for (int i = 0; i < BIT_COUNT; i++) {\n if (i >= maxBytes) break;\n if (modi(n, 2) > 0) { result += byteVal; }\n n = int(n / 2);\n byteVal *= 2;\n }\n return result;\n}\n\nint bitwiseSignedRightShift(int num, int shifts) {\n return int(floor(float(num) / pow(2.0, float(shifts))));\n}\n\nint bitwiseZeroFillRightShift(int n, int shift) {\n int maxBytes = BIT_COUNT;\n for (int i = 0; i < BIT_COUNT; i++) {\n if (maxBytes >= n) {\n break;\n }\n maxBytes *= 2;\n }\n for (int i = 0; i < BIT_COUNT; i++) {\n if (i >= shift) {\n break;\n }\n n /= 2;\n }\n int result = 0;\n int byteVal = 1;\n for (int i = 0; i < BIT_COUNT; i++) {\n if (i >= maxBytes) break;\n if (modi(n, 2) > 0) { result += byteVal; }\n n = int(n / 2);\n byteVal *= 2;\n }\n return result;\n}\n\nvec2 integerMod(vec2 x, float y) {\n vec2 res = floor(mod(x, y));\n return res * step(1.0 - floor(y), -res);\n}\n\nvec3 integerMod(vec3 x, float y) {\n vec3 res = floor(mod(x, y));\n return res * step(1.0 - floor(y), -res);\n}\n\nvec4 integerMod(vec4 x, vec4 y) {\n vec4 res = floor(mod(x, y));\n return res * step(1.0 - floor(y), -res);\n}\n\nfloat integerMod(float x, float y) {\n float res = floor(mod(x, y));\n return res * (res > floor(y) - 1.0 ? 0.0 : 1.0);\n}\n\nint integerMod(int x, int y) {\n return x - (y * int(x / y));\n}\n\n__DIVIDE_WITH_INTEGER_CHECK__;\n\n// Here be dragons!\n// DO NOT OPTIMIZE THIS CODE\n// YOU WILL BREAK SOMETHING ON SOMEBODY\\'S MACHINE\n// LEAVE IT AS IT IS, LEST YOU WASTE YOUR OWN TIME\nconst vec2 MAGIC_VEC = vec2(1.0, -256.0);\nconst vec4 SCALE_FACTOR = vec4(1.0, 256.0, 65536.0, 0.0);\nconst vec4 SCALE_FACTOR_INV = vec4(1.0, 0.00390625, 0.0000152587890625, 0.0); // 1, 1/256, 1/65536\nfloat decode32(vec4 texel) {\n __DECODE32_ENDIANNESS__;\n texel *= 255.0;\n vec2 gte128;\n gte128.x = texel.b >= 128.0 ? 1.0 : 0.0;\n gte128.y = texel.a >= 128.0 ? 1.0 : 0.0;\n float exponent = 2.0 * texel.a - 127.0 + dot(gte128, MAGIC_VEC);\n float res = exp2(_round(exponent));\n texel.b = texel.b - 128.0 * gte128.x;\n res = dot(texel, SCALE_FACTOR) * exp2(_round(exponent-23.0)) + res;\n res *= gte128.y * -2.0 + 1.0;\n return res;\n}\n\nfloat decode16(vec4 texel, int index) {\n int channel = integerMod(index, 2);\n if (channel == 0) return texel.r * 255.0 + texel.g * 65280.0;\n if (channel == 1) return texel.b * 255.0 + texel.a * 65280.0;\n return 0.0;\n}\n\nfloat decode8(vec4 texel, int index) {\n int channel = integerMod(index, 4);\n if (channel == 0) return texel.r * 255.0;\n if (channel == 1) return texel.g * 255.0;\n if (channel == 2) return texel.b * 255.0;\n if (channel == 3) return texel.a * 255.0;\n return 0.0;\n}\n\nvec4 legacyEncode32(float f) {\n float F = abs(f);\n float sign = f < 0.0 ? 1.0 : 0.0;\n float exponent = floor(log2(F));\n float mantissa = (exp2(-exponent) * F);\n // exponent += floor(log2(mantissa));\n vec4 texel = vec4(F * exp2(23.0-exponent)) * SCALE_FACTOR_INV;\n texel.rg = integerMod(texel.rg, 256.0);\n texel.b = integerMod(texel.b, 128.0);\n texel.a = exponent*0.5 + 63.5;\n texel.ba += vec2(integerMod(exponent+127.0, 2.0), sign) * 128.0;\n texel = floor(texel);\n texel *= 0.003921569; // 1/255\n __ENCODE32_ENDIANNESS__;\n return texel;\n}\n\n// https://github.com/gpujs/gpu.js/wiki/Encoder-details\nvec4 encode32(float value) {\n if (value == 0.0) return vec4(0, 0, 0, 0);\n\n float exponent;\n float mantissa;\n vec4 result;\n float sgn;\n\n sgn = step(0.0, -value);\n value = abs(value);\n\n exponent = floor(log2(value));\n\n mantissa = value*pow(2.0, -exponent)-1.0;\n exponent = exponent+127.0;\n result = vec4(0,0,0,0);\n\n result.a = floor(exponent/2.0);\n exponent = exponent - result.a*2.0;\n result.a = result.a + 128.0*sgn;\n\n result.b = floor(mantissa * 128.0);\n mantissa = mantissa - result.b / 128.0;\n result.b = result.b + exponent*128.0;\n\n result.g = floor(mantissa*32768.0);\n mantissa = mantissa - result.g/32768.0;\n\n result.r = floor(mantissa*8388608.0);\n return result/255.0;\n}\n// Dragons end here\n\nint index;\nivec3 threadId;\n\nivec3 indexTo3D(int idx, ivec3 texDim) {\n int z = int(idx / (texDim.x * texDim.y));\n idx -= z * int(texDim.x * texDim.y);\n int y = int(idx / texDim.x);\n int x = int(integerMod(idx, texDim.x));\n return ivec3(x, y, z);\n}\n\nfloat get32(sampler2D tex, ivec2 texSize, ivec3 texDim, int z, int y, int x) {\n int index = x + texDim.x * (y + texDim.y * z);\n int w = texSize.x;\n vec2 st = vec2(float(integerMod(index, w)), float(index / w)) + 0.5;\n vec4 texel = texture2D(tex, st / vec2(texSize));\n return decode32(texel);\n}\n\nfloat get16(sampler2D tex, ivec2 texSize, ivec3 texDim, int z, int y, int x) {\n int index = x + texDim.x * (y + texDim.y * z);\n int w = texSize.x * 2;\n vec2 st = vec2(float(integerMod(index, w)), float(index / w)) + 0.5;\n vec4 texel = texture2D(tex, st / vec2(texSize.x * 2, texSize.y));\n return decode16(texel, index);\n}\n\nfloat get8(sampler2D tex, ivec2 texSize, ivec3 texDim, int z, int y, int x) {\n int index = x + texDim.x * (y + texDim.y * z);\n int w = texSize.x * 4;\n vec2 st = vec2(float(integerMod(index, w)), float(index / w)) + 0.5;\n vec4 texel = texture2D(tex, st / vec2(texSize.x * 4, texSize.y));\n return decode8(texel, index);\n}\n\nfloat getMemoryOptimized32(sampler2D tex, ivec2 texSize, ivec3 texDim, int z, int y, int x) {\n int index = x + texDim.x * (y + texDim.y * z);\n int channel = integerMod(index, 4);\n index = index / 4;\n int w = texSize.x;\n vec2 st = vec2(float(integerMod(index, w)), float(index / w)) + 0.5;\n vec4 texel = texture2D(tex, st / vec2(texSize));\n if (channel == 0) return texel.r;\n if (channel == 1) return texel.g;\n if (channel == 2) return texel.b;\n if (channel == 3) return texel.a;\n return 0.0;\n}\n\nvec4 getImage2D(sampler2D tex, ivec2 texSize, ivec3 texDim, int z, int y, int x) {\n int index = x + texDim.x * (y + texDim.y * z);\n int w = texSize.x;\n vec2 st = vec2(float(integerMod(index, w)), float(index / w)) + 0.5;\n return texture2D(tex, st / vec2(texSize));\n}\n\nfloat getFloatFromSampler2D(sampler2D tex, ivec2 texSize, ivec3 texDim, int z, int y, int x) {\n vec4 result = getImage2D(tex, texSize, texDim, z, y, x);\n return result[0];\n}\n\nvec2 getVec2FromSampler2D(sampler2D tex, ivec2 texSize, ivec3 texDim, int z, int y, int x) {\n vec4 result = getImage2D(tex, texSize, texDim, z, y, x);\n return vec2(result[0], result[1]);\n}\n\nvec2 getMemoryOptimizedVec2(sampler2D tex, ivec2 texSize, ivec3 texDim, int z, int y, int x) {\n int index = x + (texDim.x * (y + (texDim.y * z)));\n int channel = integerMod(index, 2);\n index = index / 2;\n int w = texSize.x;\n vec2 st = vec2(float(integerMod(index, w)), float(index / w)) + 0.5;\n vec4 texel = texture2D(tex, st / vec2(texSize));\n if (channel == 0) return vec2(texel.r, texel.g);\n if (channel == 1) return vec2(texel.b, texel.a);\n return vec2(0.0, 0.0);\n}\n\nvec3 getVec3FromSampler2D(sampler2D tex, ivec2 texSize, ivec3 texDim, int z, int y, int x) {\n vec4 result = getImage2D(tex, texSize, texDim, z, y, x);\n return vec3(result[0], result[1], result[2]);\n}\n\nvec3 getMemoryOptimizedVec3(sampler2D tex, ivec2 texSize, ivec3 texDim, int z, int y, int x) {\n int fieldIndex = 3 * (x + texDim.x * (y + texDim.y * z));\n int vectorIndex = fieldIndex / 4;\n int vectorOffset = fieldIndex - vectorIndex * 4;\n int readY = vectorIndex / texSize.x;\n int readX = vectorIndex - readY * texSize.x;\n vec4 tex1 = texture2D(tex, (vec2(readX, readY) + 0.5) / vec2(texSize));\n \n if (vectorOffset == 0) {\n return tex1.xyz;\n } else if (vectorOffset == 1) {\n return tex1.yzw;\n } else {\n readX++;\n if (readX >= texSize.x) {\n readX = 0;\n readY++;\n }\n vec4 tex2 = texture2D(tex, vec2(readX, readY) / vec2(texSize));\n if (vectorOffset == 2) {\n return vec3(tex1.z, tex1.w, tex2.x);\n } else {\n return vec3(tex1.w, tex2.x, tex2.y);\n }\n }\n}\n\nvec4 getVec4FromSampler2D(sampler2D tex, ivec2 texSize, ivec3 texDim, int z, int y, int x) {\n return getImage2D(tex, texSize, texDim, z, y, x);\n}\n\nvec4 getMemoryOptimizedVec4(sampler2D tex, ivec2 texSize, ivec3 texDim, int z, int y, int x) {\n int index = x + texDim.x * (y + texDim.y * z);\n int channel = integerMod(index, 2);\n int w = texSize.x;\n vec2 st = vec2(float(integerMod(index, w)), float(index / w)) + 0.5;\n vec4 texel = texture2D(tex, st / vec2(texSize));\n return vec4(texel.r, texel.g, texel.b, texel.a);\n}\n\nvec4 actualColor;\nvoid color(float r, float g, float b, float a) {\n actualColor = vec4(r,g,b,a);\n}\n\nvoid color(float r, float g, float b) {\n color(r,g,b,1.0);\n}\n\nvoid color(sampler2D image) {\n actualColor = texture2D(image, vTexCoord);\n}\n\nfloat modulo(float number, float divisor) {\n if (number < 0.0) {\n number = abs(number);\n if (divisor < 0.0) {\n divisor = abs(divisor);\n }\n return -mod(number, divisor);\n }\n if (divisor < 0.0) {\n divisor = abs(divisor);\n }\n return mod(number, divisor);\n}\n\n__INJECTED_NATIVE__;\n__MAIN_CONSTANTS__;\n__MAIN_ARGUMENTS__;\n__KERNEL__;\n\nvoid main(void) {\n index = int(vTexCoord.s * float(uTexSize.x)) + int(vTexCoord.t * float(uTexSize.y)) * uTexSize.x;\n __MAIN_RESULT__;\n}`;\n\nmodule.exports = {\n fragmentShader\n};" }, { "path": "src/backend/web-gl/function-node.js", "content": "const { utils } = require('../../utils');\nconst { FunctionNode } = require('../function-node');\n\n/**\n * @desc [INTERNAL] Takes in a function node, and does all the AST voodoo required to toString its respective WebGL code\n */\nclass WebGLFunctionNode extends FunctionNode {\n constructor(source, settings) {\n super(source, settings);\n if (settings && settings.hasOwnProperty('fixIntegerDivisionAccuracy')) {\n this.fixIntegerDivisionAccuracy = settings.fixIntegerDivisionAccuracy;\n }\n }\n\n astConditionalExpression(ast, retArr) {\n if (ast.type !== 'ConditionalExpression') {\n throw this.astErrorOutput('Not a conditional expression', ast);\n }\n const consequentType = this.getType(ast.consequent);\n const alternateType = this.getType(ast.alternate);\n // minification handling if void\n if (consequentType === null && alternateType === null) {\n retArr.push('if (');\n this.astGeneric(ast.test, retArr);\n retArr.push(') {');\n this.astGeneric(ast.consequent, retArr);\n retArr.push(';');\n retArr.push('} else {');\n this.astGeneric(ast.alternate, retArr);\n retArr.push(';');\n retArr.push('}');\n return retArr;\n }\n retArr.push('(');\n this.astGeneric(ast.test, retArr);\n retArr.push('?');\n this.astGeneric(ast.consequent, retArr);\n retArr.push(':');\n this.astGeneric(ast.alternate, retArr);\n retArr.push(')');\n return retArr;\n }\n\n /**\n * @desc Parses the abstract syntax tree for to its *named function*\n * @param {Object} ast - the AST object to parse\n * @param {Array} retArr - return array string\n * @returns {Array} the append retArr\n */\n astFunction(ast, retArr) {\n // Setup function return type and name\n if (this.isRootKernel) {\n retArr.push('void');\n } else {\n // looking up return type, this is a little expensive, and can be avoided if returnType is set\n if (!this.returnType) {\n const lastReturn = this.findLastReturn();\n if (lastReturn) {\n this.returnType = this.getType(ast.body);\n if (this.returnType === 'LiteralInteger') {\n this.returnType = 'Number';\n }\n }\n }\n\n const { returnType } = this;\n if (!returnType) {\n retArr.push('void');\n } else {\n const type = typeMap[returnType];\n if (!type) {\n throw new Error(`unknown type ${returnType}`);\n }\n retArr.push(type);\n }\n }\n retArr.push(' ');\n retArr.push(this.name);\n retArr.push('(');\n\n if (!this.isRootKernel) {\n // Arguments handling\n for (let i = 0; i < this.argumentNames.length; ++i) {\n const argumentName = this.argumentNames[i];\n\n if (i > 0) {\n retArr.push(', ');\n }\n let argumentType = this.argumentTypes[this.argumentNames.indexOf(argumentName)];\n // The type is too loose ended, here we decide to solidify a type, lets go with float\n if (!argumentType) {\n throw this.astErrorOutput(`Unknown argument ${argumentName} type`, ast);\n }\n if (argumentType === 'LiteralInteger') {\n this.argumentTypes[i] = argumentType = 'Number';\n }\n const type = typeMap[argumentType];\n if (!type) {\n throw this.astErrorOutput('Unexpected expression', ast);\n }\n const name = utils.sanitizeName(argumentName);\n if (type === 'sampler2D' || type === 'sampler2DArray') {\n // mash needed arguments together, since now we have end to end inference\n retArr.push(`${type} user_${name},ivec2 user_${name}Size,ivec3 user_${name}Dim`);\n } else {\n retArr.push(`${type} user_${name}`);\n }\n }\n }\n\n // Function opening\n retArr.push(') {\\n');\n\n // Body statement iteration\n for (let i = 0; i < ast.body.body.length; ++i) {\n this.astGeneric(ast.body.body[i], retArr);\n retArr.push('\\n');\n }\n\n // Function closing\n retArr.push('}\\n');\n return retArr;\n }\n\n /**\n * @desc Parses the abstract syntax tree for to *return* statement\n * @param {Object} ast - the AST object to parse\n * @param {Array} retArr - return array string\n * @returns {Array} the append retArr\n */\n astReturnStatement(ast, retArr) {\n if (!ast.argument) throw this.astErrorOutput('Unexpected return statement', ast);\n this.pushState('skip-literal-correction');\n const type = this.getType(ast.argument);\n this.popState('skip-literal-correction');\n\n const result = [];\n\n if (!this.returnType) {\n if (type === 'LiteralInteger' || type === 'Integer') {\n this.returnType = 'Number';\n } else {\n this.returnType = type;\n }\n }\n\n switch (this.returnType) {\n case 'LiteralInteger':\n case 'Number':\n case 'Float':\n switch (type) {\n case 'Integer':\n result.push('float(');\n this.astGeneric(ast.argument, result);\n result.push(')');\n break;\n case 'LiteralInteger':\n this.castLiteralToFloat(ast.argument, result);\n\n // Running astGeneric forces the LiteralInteger to pick a type, and here, if we are returning a float, yet\n // the LiteralInteger has picked to be an integer because of constraints on it we cast it to float.\n if (this.getType(ast) === 'Integer') {\n result.unshift('float(');\n result.push(')');\n }\n break;\n default:\n this.astGeneric(ast.argument, result);\n }\n break;\n case 'Integer':\n switch (type) {\n case 'Float':\n case 'Number':\n this.castValueToInteger(ast.argument, result);\n break;\n case 'LiteralInteger':\n this.castLiteralToInteger(ast.argument, result);\n break;\n default:\n this.astGeneric(ast.argument, result);\n }\n break;\n case 'Array(4)':\n case 'Array(3)':\n case 'Array(2)':\n case 'Matrix(2)':\n case 'Matrix(3)':\n case 'Matrix(4)':\n case 'Input':\n this.astGeneric(ast.argument, result);\n break;\n default:\n throw this.astErrorOutput(`unhandled return type ${this.returnType}`, ast);\n }\n\n if (this.isRootKernel) {\n retArr.push(`kernelResult = ${ result.join('') };`);\n retArr.push('return;');\n } else if (this.isSubKernel) {\n retArr.push(`subKernelResult_${ this.name } = ${ result.join('') };`);\n retArr.push(`return subKernelResult_${ this.name };`);\n } else {\n retArr.push(`return ${ result.join('') };`);\n }\n return retArr;\n }\n\n /**\n * @desc Parses the abstract syntax tree for *literal value*\n *\n * @param {Object} ast - the AST object to parse\n * @param {Array} retArr - return array string\n *\n * @returns {Array} the append retArr\n */\n astLiteral(ast, retArr) {\n // Reject non numeric literals\n if (isNaN(ast.value)) {\n throw this.astErrorOutput(\n 'Non-numeric literal not supported : ' + ast.value,\n ast\n );\n }\n\n const key = this.astKey(ast);\n if (Number.isInteger(ast.value)) {\n if (this.isState('casting-to-integer') || this.isState('building-integer')) {\n this.literalTypes[key] = 'Integer';\n retArr.push(`${ast.value}`);\n } else if (this.isState('casting-to-float') || this.isState('building-float')) {\n this.literalTypes[key] = 'Number';\n retArr.push(`${ast.value}.0`);\n } else {\n this.literalTypes[key] = 'Number';\n retArr.push(`${ast.value}.0`);\n }\n } else if (this.isState('casting-to-integer') || this.isState('building-integer')) {\n this.literalTypes[key] = 'Integer';\n retArr.push(Math.round(ast.value));\n } else {\n this.literalTypes[key] = 'Number';\n retArr.push(`${ast.value}`);\n }\n return retArr;\n }\n\n /**\n * @desc Parses the abstract syntax tree for *binary* expression\n * @param {Object} ast - the AST object to parse\n * @param {Array} retArr - return array string\n * @returns {Array} the append retArr\n */\n astBinaryExpression(ast, retArr) {\n if (this.checkAndUpconvertOperator(ast, retArr)) {\n return retArr;\n }\n\n if (this.fixIntegerDivisionAccuracy && ast.operator === '/') {\n retArr.push('divWithIntCheck(');\n this.pushState('building-float');\n switch (this.getType(ast.left)) {\n case 'Integer':\n this.castValueToFloat(ast.left, retArr);\n break;\n case 'LiteralInteger':\n this.castLiteralToFloat(ast.left, retArr);\n break;\n default:\n this.astGeneric(ast.left, retArr);\n }\n retArr.push(', ');\n switch (this.getType(ast.right)) {\n case 'Integer':\n this.castValueToFloat(ast.right, retArr);\n break;\n case 'LiteralInteger':\n this.castLiteralToFloat(ast.right, retArr);\n break;\n default:\n this.astGeneric(ast.right, retArr);\n }\n this.popState('building-float');\n retArr.push(')');\n return retArr;\n }\n\n retArr.push('(');\n const leftType = this.getType(ast.left) || 'Number';\n const rightType = this.getType(ast.right) || 'Number';\n if (!leftType || !rightType) {\n throw this.astErrorOutput(`Unhandled binary expression`, ast);\n }\n const key = leftType + ' & ' + rightType;\n switch (key) {\n case 'Integer & Integer':\n this.pushState('building-integer');\n this.astGeneric(ast.left, retArr);\n retArr.push(operatorMap[ast.operator] || ast.operator);\n this.astGeneric(ast.right, retArr);\n this.popState('building-integer');\n break;\n case 'Number & Float':\n case 'Float & Number':\n case 'Float & Float':\n case 'Number & Number':\n this.pushState('building-float');\n this.astGeneric(ast.left, retArr);\n retArr.push(operatorMap[ast.operator] || ast.operator);\n this.astGeneric(ast.right, retArr);\n this.popState('building-float');\n break;\n case 'LiteralInteger & LiteralInteger':\n if (this.isState('casting-to-integer') || this.isState('building-integer')) {\n this.pushState('building-integer');\n this.astGeneric(ast.left, retArr);\n retArr.push(operatorMap[ast.operator] || ast.operator);\n this.astGeneric(ast.right, retArr);\n this.popState('building-integer');\n } else {\n this.pushState('building-float');\n this.castLiteralToFloat(ast.left, retArr);\n retArr.push(operatorMap[ast.operator] || ast.operator);\n this.castLiteralToFloat(ast.right, retArr);\n this.popState('building-float');\n }\n break;\n\n case 'Integer & Float':\n case 'Integer & Number':\n if (ast.operator === '>' || ast.operator === '<' && ast.right.type === 'Literal') {\n // if right value is actually a float, don't loose that information, cast left to right rather than the usual right to left\n if (!Number.isInteger(ast.right.value)) {\n this.pushState('building-float');\n this.castValueToFloat(ast.left, retArr);\n retArr.push(operatorMap[ast.operator] || ast.operator);\n this.astGeneric(ast.right, retArr);\n this.popState('building-float');\n break;\n }\n }\n this.pushState('building-integer');\n this.astGeneric(ast.left, retArr);\n retArr.push(operatorMap[ast.operator] || ast.operator);\n this.pushState('casting-to-integer');\n if (ast.right.type === 'Literal') {\n const literalResult = [];\n this.astGeneric(ast.right, literalResult);\n const literalType = this.getType(ast.right);\n if (literalType === 'Integer') {\n retArr.push(literalResult.join(''));\n } else {\n throw this.astErrorOutput(`Unhandled binary expression with literal`, ast);\n }\n } else {\n retArr.push('int(');\n this.astGeneric(ast.right, retArr);\n retArr.push(')');\n }\n this.popState('casting-to-integer');\n this.popState('building-integer');\n break;\n case 'Integer & LiteralInteger':\n this.pushState('building-integer');\n this.astGeneric(ast.left, retArr);\n retArr.push(operatorMap[ast.operator] || ast.operator);\n this.castLiteralToInteger(ast.right, retArr);\n this.popState('building-integer');\n break;\n\n case 'Number & Integer':\n this.pushState('building-float');\n this.astGeneric(ast.left, retArr);\n retArr.push(operatorMap[ast.operator] || ast.operator);\n this.castValueToFloat(ast.right, retArr);\n this.popState('building-float');\n break;\n case 'Float & LiteralInteger':\n case 'Number & LiteralInteger':\n this.pushState('building-float');\n this.astGeneric(ast.left, retArr);\n retArr.push(operatorMap[ast.operator] || ast.operator);\n this.castLiteralToFloat(ast.right, retArr);\n this.popState('building-float');\n break;\n case 'LiteralInteger & Float':\n case 'LiteralInteger & Number':\n if (this.isState('casting-to-integer')) {\n this.pushState('building-integer');\n this.castLiteralToInteger(ast.left, retArr);\n retArr.push(operatorMap[ast.operator] || ast.operator);\n this.castValueToInteger(ast.right, retArr);\n this.popState('building-integer');\n } else {\n this.pushState('building-float');\n this.astGeneric(ast.left, retArr);\n retArr.push(operatorMap[ast.operator] || ast.operator);\n this.pushState('casting-to-float');\n this.astGeneric(ast.right, retArr);\n this.popState('casting-to-float');\n this.popState('building-float');\n }\n break;\n case 'LiteralInteger & Integer':\n this.pushState('building-integer');\n this.castLiteralToInteger(ast.left, retArr);\n retArr.push(operatorMap[ast.operator] || ast.operator);\n this.astGeneric(ast.right, retArr);\n this.popState('building-integer');\n break;\n\n case 'Boolean & Boolean':\n this.pushState('building-boolean');\n this.astGeneric(ast.left, retArr);\n retArr.push(operatorMap[ast.operator] || ast.operator);\n this.astGeneric(ast.right, retArr);\n this.popState('building-boolean');\n break;\n\n case 'Float & Integer':\n this.pushState('building-float');\n this.astGeneric(ast.left, retArr);\n retArr.push(operatorMap[ast.operator] || ast.operator);\n this.castValueToFloat(ast.right, retArr);\n this.popState('building-float');\n break;\n\n default:\n throw this.astErrorOutput(`Unhandled binary expression between ${key}`, ast);\n }\n retArr.push(')');\n\n return retArr;\n }\n\n checkAndUpconvertOperator(ast, retArr) {\n const bitwiseResult = this.checkAndUpconvertBitwiseOperators(ast, retArr);\n if (bitwiseResult) {\n return bitwiseResult;\n }\n const upconvertableOperators = {\n '%': this.fixIntegerDivisionAccuracy ? 'integerCorrectionModulo' : 'modulo',\n '**': 'pow',\n };\n const foundOperator = upconvertableOperators[ast.operator];\n if (!foundOperator) return null;\n retArr.push(foundOperator);\n retArr.push('(');\n switch (this.getType(ast.left)) {\n case 'Integer':\n this.castValueToFloat(ast.left, retArr);\n break;\n case 'LiteralInteger':\n this.castLiteralToFloat(ast.left, retArr);\n break;\n default:\n this.astGeneric(ast.left, retArr);\n }\n retArr.push(',');\n switch (this.getType(ast.right)) {\n case 'Integer':\n this.castValueToFloat(ast.right, retArr);\n break;\n case 'LiteralInteger':\n this.castLiteralToFloat(ast.right, retArr);\n break;\n default:\n this.astGeneric(ast.right, retArr);\n }\n retArr.push(')');\n return retArr;\n }\n\n checkAndUpconvertBitwiseOperators(ast, retArr) {\n const upconvertableOperators = {\n '&': 'bitwiseAnd',\n '|': 'bitwiseOr',\n '^': 'bitwiseXOR',\n '<<': 'bitwiseZeroFillLeftShift',\n '>>': 'bitwiseSignedRightShift',\n '>>>': 'bitwiseZeroFillRightShift',\n };\n const foundOperator = upconvertableOperators[ast.operator];\n if (!foundOperator) return null;\n retArr.push(foundOperator);\n retArr.push('(');\n const leftType = this.getType(ast.left);\n switch (leftType) {\n case 'Number':\n case 'Float':\n this.castValueToInteger(ast.left, retArr);\n break;\n case 'LiteralInteger':\n this.castLiteralToInteger(ast.left, retArr);\n break;\n default:\n this.astGeneric(ast.left, retArr);\n }\n retArr.push(',');\n const rightType = this.getType(ast.right);\n switch (rightType) {\n case 'Number':\n case 'Float':\n this.castValueToInteger(ast.right, retArr);\n break;\n case 'LiteralInteger':\n this.castLiteralToInteger(ast.right, retArr);\n break;\n default:\n this.astGeneric(ast.right, retArr);\n }\n retArr.push(')');\n return retArr;\n }\n\n checkAndUpconvertBitwiseUnary(ast, retArr) {\n const upconvertableOperators = {\n '~': 'bitwiseNot',\n };\n const foundOperator = upconvertableOperators[ast.operator];\n if (!foundOperator) return null;\n retArr.push(foundOperator);\n retArr.push('(');\n switch (this.getType(ast.argument)) {\n case 'Number':\n case 'Float':\n this.castValueToInteger(ast.argument, retArr);\n break;\n case 'LiteralInteger':\n this.castLiteralToInteger(ast.argument, retArr);\n break;\n default:\n this.astGeneric(ast.argument, retArr);\n }\n retArr.push(')');\n return retArr;\n }\n\n /**\n *\n * @param {Object} ast\n * @param {Array} retArr\n * @return {String[]}\n */\n castLiteralToInteger(ast, retArr) {\n this.pushState('casting-to-integer');\n this.astGeneric(ast, retArr);\n this.popState('casting-to-integer');\n return retArr;\n }\n\n /**\n *\n * @param {Object} ast\n * @param {Array} retArr\n * @return {String[]}\n */\n castLiteralToFloat(ast, retArr) {\n this.pushState('casting-to-float');\n this.astGeneric(ast, retArr);\n this.popState('casting-to-float');\n return retArr;\n }\n\n /**\n *\n * @param {Object} ast\n * @param {Array} retArr\n * @return {String[]}\n */\n castValueToInteger(ast, retArr) {\n this.pushState('casting-to-integer');\n retArr.push('int(');\n this.astGeneric(ast, retArr);\n retArr.push(')');\n this.popState('casting-to-integer');\n return retArr;\n }\n\n /**\n *\n * @param {Object} ast\n * @param {Array} retArr\n * @return {String[]}\n */\n castValueToFloat(ast, retArr) {\n this.pushState('casting-to-float');\n retArr.push('float(');\n this.astGeneric(ast, retArr);\n retArr.push(')');\n this.popState('casting-to-float');\n return retArr;\n }\n\n /**\n * @desc Parses the abstract syntax tree for *identifier* expression\n * @param {Object} idtNode - An ast Node\n * @param {Array} retArr - return array string\n * @returns {Array} the append retArr\n */\n astIdentifierExpression(idtNode, retArr) {\n if (idtNode.type !== 'Identifier') {\n throw this.astErrorOutput('IdentifierExpression - not an Identifier', idtNode);\n }\n\n const type = this.getType(idtNode);\n\n const name = utils.sanitizeName(idtNode.name);\n if (idtNode.name === 'Infinity') {\n // https://stackoverflow.com/a/47543127/1324039\n retArr.push('3.402823466e+38');\n } else if (type === 'Boolean') {\n if (this.argumentNames.indexOf(name) > -1) {\n retArr.push(`bool(user_${name})`);\n } else {\n retArr.push(`user_${name}`);\n }\n } else {\n retArr.push(`user_${name}`);\n }\n\n return retArr;\n }\n\n /**\n * @desc Parses the abstract syntax tree for *for-loop* expression\n * @param {Object} forNode - An ast Node\n * @param {Array} retArr - return array string\n * @returns {Array} the parsed webgl string\n */\n astForStatement(forNode, retArr) {\n if (forNode.type !== 'ForStatement') {\n throw this.astErrorOutput('Invalid for statement', forNode);\n }\n\n const initArr = [];\n const testArr = [];\n const updateArr = [];\n const bodyArr = [];\n let isSafe = null;\n\n if (forNode.init) {\n const { declarations } = forNode.init;\n if (declarations.length > 1) {\n isSafe = false;\n }\n this.astGeneric(forNode.init, initArr);\n for (let i = 0; i < declarations.length; i++) {\n if (declarations[i].init && declarations[i].init.type !== 'Literal') {\n isSafe = false;\n }\n }\n } else {\n isSafe = false;\n }\n\n if (forNode.test) {\n this.astGeneric(forNode.test, testArr);\n } else {\n isSafe = false;\n }\n\n if (forNode.update) {\n this.astGeneric(forNode.update, updateArr);\n } else {\n isSafe = false;\n }\n\n if (forNode.body) {\n this.pushState('loop-body');\n this.astGeneric(forNode.body, bodyArr);\n this.popState('loop-body');\n }\n\n // have all parts, now make them safe\n if (isSafe === null) {\n isSafe = this.isSafe(forNode.init) && this.isSafe(forNode.test);\n }\n\n if (isSafe) {\n const initString = initArr.join('');\n const initNeedsSemiColon = initString[initString.length - 1] !== ';';\n retArr.push(`for (${initString}${initNeedsSemiColon ? ';' : ''}${testArr.join('')};${updateArr.join('')}){\\n`);\n retArr.push(bodyArr.join(''));\n retArr.push('}\\n');\n } else {\n const iVariableName = this.getInternalVariableName('safeI');\n if (initArr.length > 0) {\n retArr.push(initArr.join(''), '\\n');\n }\n retArr.push(`for (int ${iVariableName}=0;${iVariableName}This builds the shaders and runs them on the GPU,\n * the outputs the result back as float(enabled by default) and Texture.
\n *\n * @property {WebGLTexture[]} textureCache - webGl Texture cache\n * @property {Object.