Commit | Line | Data |
---|---|---|
5b823560 RK |
1 | /* ***** BEGIN LICENSE BLOCK ***** |
2 | * Version: MPL 1.1/GPL 2.0/LGPL 2.1 | |
3 | * | |
4 | * The contents of this file are subject to the Mozilla Public License Version | |
5 | * 1.1 (the "License"); you may not use this file except in compliance with | |
6 | * the License. You may obtain a copy of the License at | |
7 | * http://www.mozilla.org/MPL/ | |
8 | * | |
9 | * Software distributed under the License is distributed on an "AS IS" basis, | |
10 | * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License | |
11 | * for the specific language governing rights and limitations under the | |
12 | * License. | |
13 | * | |
14 | * The Original Code is KaiRo.at Mandelbrot, XULRunner version. | |
15 | * | |
16 | * The Initial Developer of the Original Code is | |
17 | * Robert Kaiser <kairo@kairo.at>. | |
18 | * Portions created by the Initial Developer are Copyright (C) 2008 | |
19 | * the Initial Developer. All Rights Reserved. | |
20 | * | |
21 | * Contributor(s): | |
22 | * Robert Kaiser <kairo@kairo.at> | |
23 | * | |
24 | * Alternatively, the contents of this file may be used under the terms of | |
25 | * either the GNU General Public License Version 2 or later (the "GPL"), or | |
26 | * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"), | |
27 | * in which case the provisions of the GPL or the LGPL are applicable instead | |
28 | * of those above. If you wish to allow use of your version of this file only | |
29 | * under the terms of either the GPL or the LGPL, and not to allow others to | |
30 | * use your version of this file under the terms of the MPL, indicate your | |
31 | * decision by deleting the provisions above and replace them with the notice | |
32 | * and other provisions required by the GPL or the LGPL. If you do not delete | |
33 | * the provisions above, a recipient may use your version of this file under | |
34 | * the terms of any one of the MPL, the GPL or the LGPL. | |
35 | * | |
36 | * ***** END LICENSE BLOCK ***** */ | |
37 | ||
6e98af87 RK |
38 | var gColorPalette = []; |
39 | var gPref = Components.classes["@mozilla.org/preferences-service;1"] | |
40 | .getService(Components.interfaces.nsIPrefService) | |
41 | .getBranch(null); | |
42 | var gStartTime = 0; | |
43 | ||
44 | function Startup() { | |
45 | updateIterMenu(); | |
46 | updatePaletteMenu(); | |
47 | document.getElementById("statusLabel").value = | |
48 | document.getElementById("mbrotBundle").getString("statusEmpty"); | |
49 | } | |
37b05b56 RK |
50 | |
51 | function drawImage() { | |
8a9c8e3f | 52 | let canvas = document.getElementById("mbrotImage"); |
2cb9a6b5 | 53 | let context = canvas.getContext("2d"); |
37b05b56 | 54 | |
2cb9a6b5 RK |
55 | document.getElementById("statusLabel").value = |
56 | document.getElementById("mbrotBundle").getString("statusDrawing"); | |
57 | ||
58 | let iterMax = gPref.getIntPref("mandelbrot.iteration_max"); | |
59 | let algorithm = gPref.getCharPref("mandelbrot.use_algorithm"); | |
6e98af87 | 60 | |
2cb9a6b5 RK |
61 | context.fillStyle = "rgb(255, 255, 255)"; |
62 | context.fillRect(0, 0, canvas.width, canvas.height); | |
37b05b56 | 63 | |
2cb9a6b5 RK |
64 | gStartTime = new Date(); |
65 | ||
66 | drawLine(0, canvas, context, iterMax, algorithm); | |
67 | } | |
68 | ||
69 | function drawLine(line, canvas, context, iterMax, algorithm) { | |
8a9c8e3f RK |
70 | let Cr_min = -2.0; |
71 | let Cr_max = 1.0; | |
6403d662 RK |
72 | try { |
73 | Cr_min = gPref.getCharPref("mandelbrot.last_image.Cr_min"); | |
74 | Cr_max = gPref.getCharPref("mandelbrot.last_image.Cr_max"); | |
75 | } | |
76 | catch (e) { } | |
8a9c8e3f | 77 | let Cr_scale = Cr_max - Cr_min; |
37b05b56 | 78 | |
8a9c8e3f RK |
79 | let Ci_min = -1.5; |
80 | let Ci_max = 1.5; | |
6403d662 RK |
81 | try { |
82 | Cr_min = gPref.getCharPref("mandelbrot.last_image.Ci_min"); | |
83 | Cr_max = gPref.getCharPref("mandelbrot.last_image.Ci_max"); | |
84 | } | |
85 | catch (e) { } | |
8a9c8e3f | 86 | let Ci_scale = Ci_max - Ci_min; |
37b05b56 | 87 | |
2cb9a6b5 RK |
88 | let pixels = []; |
89 | for (var img_y = line; img_y < canvas.height && img_y < line+8; img_y++) | |
90 | for (let img_x = 0; img_x < canvas.width; img_x++) { | |
8a9c8e3f | 91 | let C = new complex(Cr_min + (img_x / canvas.width) * Cr_scale, |
37b05b56 | 92 | Ci_min + (img_y / canvas.height) * Ci_scale); |
2cb9a6b5 | 93 | pixels.push.apply(pixels, drawPoint(context, img_x, img_y, C, iterMax, algorithm)); |
37b05b56 | 94 | } |
2cb9a6b5 RK |
95 | context.putImageData({width: canvas.width, height: pixels.length/4/canvas.width, data: pixels}, 0, line); |
96 | ||
97 | if (img_y < canvas.height) | |
98 | setTimeout(drawLine, 0, img_y, canvas, context, iterMax, algorithm); | |
99 | else | |
100 | EndCalc(); | |
37b05b56 RK |
101 | } |
102 | ||
6e98af87 RK |
103 | function EndCalc() { |
104 | let endTime = new Date(); | |
105 | let timeUsed = (endTime.getTime() - gStartTime.getTime()) / 1000; | |
106 | document.getElementById("statusLabel").value = | |
107 | document.getElementById("mbrotBundle").getFormattedString("statusTime", [timeUsed.toFixed(3)]); | |
108 | } | |
109 | ||
37b05b56 RK |
110 | function complex(aReal, aImag) { |
111 | this.r = aReal; | |
112 | this.i = aImag; | |
2cb9a6b5 RK |
113 | } |
114 | complex.prototype = { | |
115 | square: function() { | |
37b05b56 RK |
116 | return new complex(this.r * this.r - this.i * this.i, |
117 | 2 * this.r * this.i); | |
2cb9a6b5 RK |
118 | }, |
119 | dist: function() { | |
37b05b56 | 120 | return Math.sqrt(this.r * this.r + this.i * this.i); |
2cb9a6b5 RK |
121 | }, |
122 | add: function(aComplex) { | |
37b05b56 RK |
123 | return new complex(this.r + aComplex.r, this.i + aComplex.i); |
124 | } | |
125 | } | |
126 | ||
6e98af87 RK |
127 | function mandelbrotValueOO (aC, aIterMax) { |
128 | // this would be nice code in general but it looks like JS objects are too heavy for normal use. | |
8444612a | 129 | let Z = new complex(0.0, 0.0); |
37b05b56 RK |
130 | for (var iter = 0; iter < aIterMax; iter++) { |
131 | Z = Z.square().add(aC); | |
132 | if (Z.r * Z.r + Z.i * Z.i > 256) { break; } | |
133 | } | |
6e98af87 RK |
134 | return iter; |
135 | } | |
8444612a | 136 | |
6e98af87 RK |
137 | function mandelbrotValueNumeric (aC, aIterMax) { |
138 | // optimized numeric code for fast calculation | |
8444612a RK |
139 | let Cr = aC.r, Ci = aC.i; |
140 | let Zr = 0.0, Zi = 0.0; | |
141 | let Zr2 = Zr * Zr, Zi2 = Zi * Zi; | |
142 | for (var iter = 0; iter < aIterMax; iter++) { | |
143 | Zi = 2 * Zr * Zi + Ci; | |
144 | Zr = Zr2 - Zi2 + Cr; | |
145 | ||
146 | Zr2 = Zr * Zr; Zi2 = Zi * Zi; | |
147 | if (Zr2 + Zi2 > 256) { break; } | |
148 | } | |
37b05b56 RK |
149 | return iter; |
150 | } | |
151 | ||
152 | function getColor(aIterValue, aIterMax) { | |
8a9c8e3f | 153 | let standardizedValue = Math.round(aIterValue * 1024 / aIterMax); |
6e98af87 RK |
154 | if (gColorPalette && gColorPalette.length) |
155 | return gColorPalette[standardizedValue]; | |
156 | ||
157 | // fallback to simple b/w if for some reason we don't have a palette | |
158 | if (aIterValue == aIterMax) | |
2cb9a6b5 | 159 | return [0, 0, 0, 255]; |
6e98af87 | 160 | else |
2cb9a6b5 | 161 | return [255, 255, 255, 255]; |
37b05b56 RK |
162 | } |
163 | ||
164 | function getColorPalette(palName) { | |
165 | var palette = []; | |
166 | switch (palName) { | |
167 | case 'bw': | |
8a9c8e3f | 168 | for (let i = 0; i < 1024; i++) { |
2cb9a6b5 | 169 | palette[i] = [255, 255, 255, 255]; |
37b05b56 | 170 | } |
2cb9a6b5 | 171 | palette[1024] = [0, 0, 0, 255]; |
37b05b56 RK |
172 | break; |
173 | case 'kairo': | |
174 | // outer areas | |
8a9c8e3f RK |
175 | for (let i = 0; i < 32; i++) { |
176 | let cc1 = Math.floor(i * 127 / 31); | |
177 | let cc2 = 170 - Math.floor(i * 43 / 31); | |
2cb9a6b5 | 178 | palette[i] = [cc1, cc2, cc1, 255]; |
37b05b56 RK |
179 | } |
180 | // inner areas | |
8a9c8e3f RK |
181 | for (let i = 0; i < 51; i++) { |
182 | let cc = Math.floor(i * 170 / 50); | |
2cb9a6b5 | 183 | palette[32 + i] = [cc, 0, (170-cc), 255]; |
37b05b56 RK |
184 | } |
185 | // corona | |
8a9c8e3f RK |
186 | for (let i = 0; i < 101; i++) { |
187 | let cc = Math.floor(i * 200 / 100); | |
2cb9a6b5 | 188 | palette[83 + i] = [255, cc, 0, 255]; |
37b05b56 RK |
189 | } |
190 | // inner corona | |
8a9c8e3f RK |
191 | for (let i = 0; i < 201; i++) { |
192 | let cc1 = 255 - Math.floor(i * 85 / 200); | |
193 | let cc2 = 200 - Math.floor(i * 30 / 200); | |
194 | let cc3 = Math.floor(i * 170 / 200); | |
2cb9a6b5 | 195 | palette[184 + i] = [cc1, cc2, cc3, 255]; |
37b05b56 | 196 | } |
8a9c8e3f RK |
197 | for (let i = 0; i < 301; i++) { |
198 | let cc1 = 170 - Math.floor(i * 43 / 300); | |
199 | let cc2 = 170 + Math.floor(i * 85 / 300); | |
2cb9a6b5 | 200 | palette[385 + i] = [cc1, cc1, cc2, 255]; |
37b05b56 | 201 | } |
8a9c8e3f RK |
202 | for (let i = 0; i < 338; i++) { |
203 | let cc = 127 + Math.floor(i * 128 / 337); | |
2cb9a6b5 | 204 | palette[686 + i] = [cc, cc, 255, 255]; |
37b05b56 | 205 | } |
2cb9a6b5 | 206 | palette[1024] = [0, 0, 0, 255]; |
37b05b56 RK |
207 | break; |
208 | case 'rainbow-linear1': | |
8a9c8e3f | 209 | for (let i = 0; i < 256; i++) { |
2cb9a6b5 RK |
210 | palette[i] = [i, 0, 0, 255]; |
211 | palette[256 + i] = [255, i, 0, 255]; | |
212 | palette[512 + i] = [255 - i, 255, i, 255]; | |
213 | palette[768 + i] = [i, 255-i, 255, 255]; | |
37b05b56 | 214 | } |
2cb9a6b5 | 215 | palette[1024] = [0, 0, 0, 255]; |
37b05b56 RK |
216 | break; |
217 | } | |
218 | /* | |
219 | Select Case palnr | |
220 | Case 1 'Standard-Palette (QB-Colors) | |
221 | For i = 0 To 1024 | |
222 | xx = CInt(i * 500 / 1024 + 2) | |
223 | If xx <= 15 Then clr = xx | |
224 | If xx > 15 Then clr = CInt(Sqr((xx - 15 + 1) * 15 ^ 2 / 485)) | |
225 | If xx >= 500 Then clr = 0 | |
226 | palette(i) = QBColor(clr) | |
227 | Next | |
228 | Case 3 'Regenbogen-Palette 1 (qu.) | |
229 | For i = 0 To 33 | |
230 | clr = CInt(i * 255 / 33) | |
231 | palette(i) = RGB(clr, 0, 0) | |
232 | Next | |
233 | For i = 0 To 136 | |
234 | clr = CInt(i * 255 / 136) | |
235 | palette(34 + i) = RGB(255, clr, 0) | |
236 | Next | |
237 | For i = 0 To 306 | |
238 | clr = CInt(i * 255 / 306) | |
239 | palette(171 + i) = RGB(255 - clr, 255, clr) | |
240 | Next | |
241 | For i = 0 To 545 | |
242 | clr = CInt(i * 255 / 545) | |
243 | palette(478 + i) = RGB(clr, 255 - clr, 255) | |
244 | Next | |
245 | Case 4 'Regenbogen-Palette 2 (linear) | |
246 | For i = 0 To 204 | |
247 | clr = CInt(i * 255 / 204) | |
248 | palette(i) = RGB(255, clr, 0) | |
249 | palette(204 + i) = RGB(255 - clr, 255, 0) | |
250 | palette(409 + i) = RGB(0, 255, clr) | |
251 | palette(614 + i) = RGB(0, 255 - clr, 255) | |
252 | palette(819 + i) = RGB(clr, 0, 255) | |
253 | Next | |
254 | Case 5 'Regenbogen-Palette 2 (qu.) | |
255 | For i = 0 To 18 | |
256 | clr = CInt(i * 255 / 18) | |
257 | palette(i) = RGB(255, clr, 0) | |
258 | Next | |
259 | For i = 0 To 73 | |
260 | clr = CInt(i * 255 / 73) | |
261 | palette(20 + i) = RGB(255 - clr, 255, 0) | |
262 | Next | |
263 | For i = 0 To 167 | |
264 | clr = CInt(i * 255 / 167) | |
265 | palette(93 + i) = RGB(0, 255, clr) | |
266 | Next | |
267 | For i = 0 To 297 | |
268 | clr = CInt(i * 255 / 297) | |
269 | palette(261 + i) = RGB(0, 255 - clr, 255) | |
270 | Next | |
271 | For i = 0 To 464 | |
272 | clr = CInt(i * 255 / 464) | |
273 | palette(559 + i) = RGB(clr, 0, 255) | |
274 | Next | |
275 | */ | |
276 | return palette; | |
277 | } | |
278 | ||
6e98af87 RK |
279 | function drawPoint(context, img_x, img_y, C, iterMax, algorithm) { |
280 | var itVal; | |
281 | switch (algorithm) { | |
282 | case 'oo': | |
283 | itVal = mandelbrotValueOO(C, iterMax); | |
284 | break; | |
285 | case 'numeric': | |
286 | default: | |
287 | itVal = mandelbrotValueNumeric(C, iterMax); | |
288 | break; | |
289 | } | |
2cb9a6b5 | 290 | return getColor(itVal, iterMax); |
37b05b56 RK |
291 | } |
292 | ||
6e98af87 RK |
293 | /***** pure UI functions *****/ |
294 | ||
37b05b56 | 295 | function saveImage() { |
740b86d1 RK |
296 | const bundle = document.getElementById("mbrotBundle"); |
297 | const nsIFilePicker = Components.interfaces.nsIFilePicker; | |
298 | var fp = null; | |
299 | try { | |
300 | fp = Components.classes["@mozilla.org/filepicker;1"] | |
301 | .createInstance(nsIFilePicker); | |
302 | } catch (e) {} | |
303 | if (!fp) return; | |
304 | var promptString = bundle.getString("savePrompt"); | |
305 | fp.init(window, promptString, nsIFilePicker.modeSave); | |
306 | fp.appendFilter(bundle.getString("pngFilterName"), "*.png"); | |
307 | fp.defaultString = "mandelbrot.png"; | |
308 | ||
309 | var fpResult = fp.show(); | |
310 | if (fpResult != nsIFilePicker.returnCancel) { | |
311 | saveCanvas(document.getElementById("mbrotImage"), fp.file); | |
312 | } | |
37b05b56 RK |
313 | } |
314 | ||
6e98af87 RK |
315 | function updateIterMenu() { |
316 | try { | |
317 | var currentIter = gPref.getIntPref("mandelbrot.iteration_max"); | |
318 | } | |
319 | catch(e) { | |
320 | var currentIter = 0; | |
321 | } | |
322 | if (currentIter < 10) { | |
323 | currentIter = 500; | |
324 | setIter(currentIter); | |
325 | } | |
326 | ||
327 | var popup = document.getElementById("menu_iterPopup"); | |
328 | var item = popup.firstChild; | |
329 | while (item) { | |
330 | if (item.getAttribute("name") == "iter") { | |
331 | if (item.getAttribute("value") == currentIter) | |
332 | item.setAttribute("checked","true"); | |
333 | else | |
334 | item.removeAttribute("checked"); | |
335 | } | |
336 | item = item.nextSibling; | |
337 | } | |
338 | } | |
339 | ||
340 | function setIter(aIter) { | |
341 | gPref.setIntPref("mandelbrot.iteration_max", aIter); | |
342 | } | |
343 | ||
344 | function updatePaletteMenu() { | |
345 | try { | |
346 | var currentPalette = gPref.getCharPref("mandelbrot.color_palette"); | |
347 | } | |
348 | catch(e) { | |
349 | var currentPalette = ''; | |
350 | } | |
351 | if (!currentPalette.length) { | |
352 | currentPalette = 'kairo'; | |
353 | setPalette(currentPalette); | |
354 | } | |
355 | if (!gColorPalette || !gColorPalette.length) | |
356 | gColorPalette = getColorPalette(currentPalette); | |
357 | ||
358 | var popup = document.getElementById("menu_palettePopup"); | |
359 | var item = popup.firstChild; | |
360 | while (item) { | |
361 | if (item.getAttribute("name") == "palette") { | |
362 | if (item.getAttribute("value") == currentPalette) | |
363 | item.setAttribute("checked", "true"); | |
364 | else | |
365 | item.removeAttribute("checked"); | |
366 | } | |
367 | item = item.nextSibling; | |
368 | } | |
369 | } | |
370 | ||
371 | function setPalette(aPaletteID) { | |
372 | gPref.setCharPref("mandelbrot.color_palette", aPaletteID); | |
373 | gColorPalette = getColorPalette(aPaletteID); | |
374 | } | |
375 | ||
6403d662 RK |
376 | function imgSettings() { |
377 | window.openDialog("chrome://mandelbrot/content/image-settings.xul"); | |
378 | } | |
379 | ||
6e98af87 RK |
380 | function updateDebugMenu() { |
381 | var jitMenuItem = document.getElementById("jitEnabled"); | |
382 | jitMenuItem.setAttribute("checked", gPref.getBoolPref("javascript.options.jit.chrome")); | |
383 | } | |
384 | ||
385 | function toggleJITState(jitMenuItem) { | |
386 | var jitEnabled = !gPref.getBoolPref("javascript.options.jit.chrome"); | |
387 | gPref.setBoolPref("javascript.options.jit.chrome", jitEnabled) | |
388 | jitMenuItem.setAttribute("checked", jitEnabled? "true" : "false"); | |
389 | } | |
390 | ||
391 | function updateAlgoMenu() { | |
392 | try { | |
393 | var currentAlgo = gPref.getCharPref("mandelbrot.use_algorithm"); | |
394 | } | |
395 | catch(e) { | |
396 | var currentAlgo = ''; | |
397 | } | |
398 | if (!currentAlgo.length) { | |
399 | currentAlgo = 'numeric'; | |
400 | setAlgorithm(currentAlgo); | |
401 | } | |
402 | ||
403 | var popup = document.getElementById("menu_algoPopup"); | |
404 | var item = popup.firstChild; | |
405 | while (item) { | |
406 | if (item.getAttribute("name") == "algorithm") { | |
407 | if (item.getAttribute("value") == currentAlgo) | |
408 | item.setAttribute("checked", "true"); | |
409 | else | |
410 | item.removeAttribute("checked"); | |
411 | } | |
412 | item = item.nextSibling; | |
413 | } | |
414 | } | |
415 | ||
416 | function setAlgorithm(algoID) { | |
417 | gPref.setCharPref("mandelbrot.use_algorithm", algoID); | |
418 | } | |
419 | ||
6e98af87 RK |
420 | /***** helper functions from external sources *****/ |
421 | ||
740b86d1 RK |
422 | // function below is based on http://developer.mozilla.org/en/docs/Code_snippets:Canvas |
423 | // custom modifications: | |
424 | // - use "a"-prefix on function arguments | |
425 | // - take an nsILocalFile as aDestFile argument | |
426 | // - always do silent download | |
427 | function saveCanvas(aCanvas, aDestFile) { | |
37b05b56 RK |
428 | // create a data url from the canvas and then create URIs of the source and targets |
429 | var io = Components.classes["@mozilla.org/network/io-service;1"] | |
430 | .getService(Components.interfaces.nsIIOService); | |
740b86d1 | 431 | var source = io.newURI(aCanvas.toDataURL("image/png", ""), "UTF8", null); |
37b05b56 RK |
432 | |
433 | // prepare to save the canvas data | |
434 | var persist = Components.classes["@mozilla.org/embedding/browser/nsWebBrowserPersist;1"] | |
435 | .createInstance(Components.interfaces.nsIWebBrowserPersist); | |
436 | ||
437 | persist.persistFlags = Components.interfaces.nsIWebBrowserPersist.PERSIST_FLAGS_REPLACE_EXISTING_FILES; | |
438 | persist.persistFlags |= Components.interfaces.nsIWebBrowserPersist.PERSIST_FLAGS_AUTODETECT_APPLY_CONVERSION; | |
439 | ||
37b05b56 | 440 | // save the canvas data to the file |
740b86d1 | 441 | persist.saveURI(source, null, null, null, null, aDestFile); |
37b05b56 RK |
442 | } |
443 | ||
444 | // function below is from http://developer.mozilla.org/en/docs/How_to_Quit_a_XUL_Application | |
445 | function quitApp(aForceQuit) { | |
446 | var appStartup = Components.classes['@mozilla.org/toolkit/app-startup;1'] | |
447 | .getService(Components.interfaces.nsIAppStartup); | |
448 | ||
449 | // eAttemptQuit will try to close each XUL window, but the XUL window can cancel the quit | |
450 | // process if there is unsaved data. eForceQuit will quit no matter what. | |
451 | var quitSeverity = aForceQuit ? Components.interfaces.nsIAppStartup.eForceQuit : | |
452 | Components.interfaces.nsIAppStartup.eAttemptQuit; | |
453 | appStartup.quit(quitSeverity); | |
454 | } |