|
15 | 15 |
|
16 | 16 | return temp; |
17 | 17 | } |
18 | | - |
| 18 | + |
19 | 19 | function getArgumentType(arg) { |
20 | 20 | if (Array.isArray(arg)) { |
21 | 21 | return 'Array'; |
|
34 | 34 | /// @param opt The parameter object |
35 | 35 | /// |
36 | 36 | /// @returns callable function if converted, else returns null |
37 | | - GPU.prototype._backendFallback = function(kernel, opt) { |
| 37 | + GPU.prototype._mode_cpu = function(kernel, opt) { |
38 | 38 | var gpu = this; |
39 | | - |
| 39 | + |
40 | 40 | function ret() { |
41 | 41 | if (!opt.dimensions || opt.dimensions.length === 0) { |
42 | 42 | if (arguments.length != 1) { |
43 | 43 | throw "Auto dimensions only supported for kernels with only one input"; |
44 | 44 | } |
45 | | - |
| 45 | + |
46 | 46 | var argType = getArgumentType(arguments[0]); |
47 | 47 | if (argType == "Array") { |
48 | 48 | opt.dimensions = getDimensions(argType); |
|
52 | 52 | throw "Auto dimensions not supported for input type: " + argType; |
53 | 53 | } |
54 | 54 | } |
55 | | - |
| 55 | + |
56 | 56 | var kernelArgs = []; |
57 | 57 | for (var i=0; i<arguments.length; i++) { |
58 | 58 | var argType = getArgumentType(arguments[i]); |
|
64 | 64 | throw "Input type not supported: " + arguments[i]; |
65 | 65 | } |
66 | 66 | } |
67 | | - |
| 67 | + |
68 | 68 | var threadDim = clone(opt.dimensions); |
69 | | - |
| 69 | + |
70 | 70 | while (threadDim.length < 3) { |
71 | 71 | threadDim.push(1); |
72 | 72 | } |
73 | | - |
| 73 | + |
74 | 74 | var ret = new Array(threadDim[2]); |
75 | 75 | for (var i=0; i<threadDim[2]; i++) { |
76 | 76 | ret[i] = new Array(threadDim[1]); |
77 | 77 | for (var j=0; j<threadDim[1]; j++) { |
78 | 78 | ret[i][j] = new Array(threadDim[0]); |
79 | 79 | } |
80 | 80 | } |
81 | | - |
| 81 | + |
82 | 82 | var canvas; |
83 | 83 | var canvasCtx; |
84 | 84 | var imageData; |
|
87 | 87 | canvas = gpu.getCanvas('cpu'); |
88 | 88 | canvas.width = threadDim[0]; |
89 | 89 | canvas.height = threadDim[1]; |
90 | | - |
| 90 | + |
91 | 91 | canvasCtx = canvas.getContext("2d"); |
92 | 92 | imageData = canvasCtx.createImageData(threadDim[0], threadDim[1]); |
93 | 93 | data = new Uint8ClampedArray(threadDim[0]*threadDim[1]*4); |
94 | 94 | } |
95 | | - |
| 95 | + |
96 | 96 | var ctx = { |
97 | 97 | thread: { |
98 | 98 | x: 0, |
|
105 | 105 | z: threadDim[2] |
106 | 106 | } |
107 | 107 | }; |
108 | | - |
| 108 | + |
109 | 109 | ctx.color = function(r, g, b, a) { |
110 | 110 | if (a == undefined) { |
111 | 111 | a = 1.0; |
112 | 112 | } |
113 | | - |
| 113 | + |
114 | 114 | r = Math.floor(r * 255); |
115 | 115 | g = Math.floor(g * 255); |
116 | 116 | b = Math.floor(b * 255); |
117 | 117 | a = Math.floor(a * 255); |
118 | | - |
| 118 | + |
119 | 119 | var width = ctx.dimensions.x; |
120 | 120 | var height = ctx.dimensions.y; |
121 | | - |
| 121 | + |
122 | 122 | var x = ctx.thread.x; |
123 | 123 | var y = height - ctx.thread.y - 1; |
124 | | - |
| 124 | + |
125 | 125 | var index = x + y*width; |
126 | | - |
| 126 | + |
127 | 127 | data[index*4+0] = r; |
128 | 128 | data[index*4+1] = g; |
129 | 129 | data[index*4+2] = b; |
130 | 130 | data[index*4+3] = a; |
131 | 131 | }; |
132 | | - |
| 132 | + |
133 | 133 | for (ctx.thread.z=0; ctx.thread.z<threadDim[2]; ctx.thread.z++) { |
134 | 134 | for (ctx.thread.y=0; ctx.thread.y<threadDim[1]; ctx.thread.y++) { |
135 | 135 | for (ctx.thread.x=0; ctx.thread.x<threadDim[0]; ctx.thread.x++) { |
136 | 136 | ret[ctx.thread.z][ctx.thread.y][ctx.thread.x] = kernel.apply(ctx, kernelArgs); |
137 | 137 | } |
138 | 138 | } |
139 | 139 | } |
140 | | - |
| 140 | + |
141 | 141 | if (opt.graphical) { |
142 | 142 | imageData.data.set(data); |
143 | 143 | canvasCtx.putImageData(imageData, 0, 0); |
144 | 144 | } |
145 | | - |
| 145 | + |
146 | 146 | if (opt.dimensions.length == 1) { |
147 | 147 | ret = ret[0][0]; |
148 | 148 | } else if (opt.dimensions.length == 2) { |
149 | 149 | ret = ret[0]; |
150 | 150 | } |
151 | | - |
| 151 | + |
152 | 152 | return ret; |
153 | 153 | } |
154 | | - |
| 154 | + |
155 | 155 | ret.dimensions = function(dim) { |
156 | 156 | opt.dimensions = dim; |
157 | 157 | return ret; |
158 | 158 | }; |
159 | | - |
| 159 | + |
160 | 160 | ret.debug = function(flag) { |
161 | 161 | opt.debug = flag; |
162 | 162 | return ret; |
163 | 163 | }; |
164 | | - |
| 164 | + |
165 | 165 | ret.graphical = function(flag) { |
166 | 166 | opt.graphical = flag; |
167 | 167 | return ret; |
168 | 168 | }; |
169 | | - |
| 169 | + |
170 | 170 | ret.loopMaxIterations = function(max) { |
171 | 171 | opt.loopMaxIterations = max; |
172 | 172 | return ret; |
173 | 173 | }; |
174 | | - |
| 174 | + |
175 | 175 | ret.wraparound = function() { |
176 | 176 | opt.wraparound = false; |
177 | 177 | return ret; |
178 | 178 | }; |
179 | | - |
| 179 | + |
180 | 180 | ret.hardcodeConstants = function() { |
181 | 181 | opt.hardcodeConstants = false; |
182 | 182 | return ret; |
183 | 183 | }; |
184 | | - |
| 184 | + |
185 | 185 | ret.outputToTexture = function() { |
186 | 186 | opt.outputToTexture = false; |
187 | 187 | return ret; |
188 | 188 | }; |
189 | | - |
| 189 | + |
190 | 190 | ret.mode = function(mode) { |
191 | 191 | opt.mode = mode; |
192 | 192 | return gpu.createKernel(kernel, opt); |
193 | 193 | }; |
194 | | - |
| 194 | + |
195 | 195 | ret.getCanvas = function() { |
196 | 196 | return gpu.getCanvas('cpu'); |
197 | 197 | }; |
198 | | - |
| 198 | + |
199 | 199 | return ret; |
200 | 200 | }; |
201 | 201 | })(GPU); |
0 commit comments