feat: getData and loadData

Author: HarshKhandeparkar

dist/gpujs-real-renderer-browser.js

@@ -103,6 +103,31 @@
 	exports.getCloneTextureKernel = getCloneTextureKernel;
 	});
 
+	var loadData = createCommonjsModule(function (module, exports) {
+	Object.defineProperty(exports, "__esModule", { value: true });
+	exports.getLoadDataKernel = void 0;
+	/**
+	 * @param gpu GPU.js Instance
+	 * @param dimensions Dimensions of the Output Graph
+	 */
+	function getLoadDataKernel(gpu, dimensions) {
+	    return gpu.createKernel(function (graphPixels) {
+	        return [
+	            graphPixels[this.thread.y * this.output.x * 3 + this.thread.x * 3 + 0],
+	            graphPixels[this.thread.y * this.output.x * 3 + this.thread.x * 3 + 1],
+	            graphPixels[this.thread.y * this.output.x * 3 + this.thread.x * 3 + 2]
+	        ];
+	    }, {
+	        argumentTypes: {
+	            graphPixels: 'Array'
+	        },
+	        output: dimensions,
+	        pipeline: true
+	    });
+	}
+	exports.getLoadDataKernel = getLoadDataKernel;
+	});
+
 	var RealRendererTypes = createCommonjsModule(function (module, exports) {
 	Object.defineProperty(exports, "__esModule", { value: true });
 	});
@@ -153,6 +178,7 @@
 
 
 
+
 	exports.RealRendererTypes = RealRendererTypes;
 
 	__exportStar(RealRendererDefaults, exports);
@@ -186,6 +212,7 @@
 	        this._blankGraph = blankGraph.getBlankGraphKernel(this.gpu, this.dimensions, this.xOffset, this.yOffset, this.bgColor, this.axesColor, this.drawAxes);
 	        this._cloneTexture = cloneTexture.getCloneTextureKernel(this.gpu, this.dimensions);
 	        this.graphPixels = this._blankGraph();
+	        this._loadData = loadData.getLoadDataKernel(this.gpu, this.dimensions);
 	        this._display = display.getDisplayKernel(this.gpu, this.dimensions);
 	        this._doRender = false;
 	    }
@@ -235,6 +262,22 @@
 	        this.time = 0;
 	        return this;
 	    };
+	    RealRenderer.prototype.getData = function () {
+	        var returnedArray = this.graphPixels.toArray();
+	        var outArr = [];
+	        for (var i = 0; i < returnedArray.length; i++) {
+	            for (var j = 0; j < returnedArray[0].length; j++) {
+	                for (var k = 0; k < returnedArray[0][0].length; k++) {
+	                    outArr[i * returnedArray[0].length * returnedArray[0][0].length + j * returnedArray[0][0].length + k] = returnedArray[i][j][k];
+	                }
+	            }
+	        }
+	        return outArr;
+	    };
+	    RealRenderer.prototype.loadData = function (pixels) {
+	        this.graphPixels = this._loadData(pixels);
+	        this._display(this.graphPixels);
+	    };
 	    RealRenderer.prototype.reset = function () {
 	        this.graphPixels = this._blankGraph();
 	        this.resetTime();

dist/gpujs-real-renderer-browser.min.js

@@ -16,4 +16,4 @@ export function getCloneTextureKernel(
     output: dimensions,
     pipeline: true
   })
-}
+}

src/kernels/cloneTexture.ts

@@ -0,0 +1,28 @@
+import { GPU } from 'gpu.js';
+import { GraphDimensions } from '../types/RealRendererTypes';
+
+/**
+ * @param gpu GPU.js Instance
+ * @param dimensions Dimensions of the Output Graph
+ */
+export function getLoadDataKernel(
+  gpu: GPU,
+  dimensions: GraphDimensions
+) {
+  return gpu.createKernel(
+    function(graphPixels) {
+      return [
+        graphPixels[this.thread.y * this.output.x * 3 + this.thread.x * 3 + 0],
+        graphPixels[this.thread.y * this.output.x * 3 + this.thread.x * 3 + 1],
+        graphPixels[this.thread.y * this.output.x * 3 + this.thread.x * 3 + 2]
+      ]
+    },
+    {
+      argumentTypes: {
+        graphPixels: 'Array'
+      },
+      output: dimensions,
+      pipeline: true
+    }
+  )
+}

src/kernels/loadData.ts

@@ -1,6 +1,7 @@
 import { getDisplayKernel } from '../kernels/display';
 import { getBlankGraphKernel } from '../kernels/blankGraph';
 import { getCloneTextureKernel } from '../kernels/cloneTexture';
+import { getLoadDataKernel } from '../kernels/loadData';
 
 import { GraphDimensions, Color, RealRendererOptions } from '../types/RealRendererTypes';
 export * as RealRendererTypes from '../types/RealRendererTypes';
@@ -28,6 +29,7 @@ export class RealRenderer {
   graphPixels: Texture;
   _blankGraph: IKernelRunShortcut;
   _cloneTexture: IKernelRunShortcut;
+  _loadData: IKernelRunShortcut;
   _display: IKernelRunShortcut;
   _doRender: boolean;
 
@@ -81,6 +83,8 @@ export class RealRenderer {
 
     this.graphPixels = this._blankGraph() as Texture;
 
+    this._loadData = getLoadDataKernel(this.gpu, this.dimensions);
+
     this._display = getDisplayKernel(this.gpu, this.dimensions);
 
     this._doRender = false;
@@ -141,6 +145,26 @@ export class RealRenderer {
     return this;
   }
 
+  getData() {
+    let returnedArray: number[][][] = <number[][][]>this.graphPixels.toArray();
+    let outArr: number[] = [];
+
+    for (let i = 0; i < returnedArray.length; i++) {
+      for (let j = 0; j < returnedArray[0].length; j++) {
+        for (let k = 0; k < returnedArray[0][0].length; k++) {
+          outArr[i * returnedArray[0].length * returnedArray[0][0].length + j * returnedArray[0][0].length + k] = returnedArray[i][j][k];
+        }
+      }
+    }
+
+    return outArr;
+  }
+
+  loadData(pixels: number[]) {
+    this.graphPixels = <Texture>this._loadData(pixels);
+    this._display(this.graphPixels);
+  }
+
   reset() {
     this.graphPixels = this._blankGraph() as Texture;
     this.resetTime();