move more variables to moz.build
[mandelbrot.git] / content / mandelbrot.js
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-2011
19  * the Initial Developer. All Rights Reserved.
20  *
21  * Contributor(s):
22  *   Robert Kaiser <kairo@kairo.at>
23  *   prefiks (patch for some speedups)
24  *   Boris Zbarsky <bzbarsky@mit.edu> (use imageData for canvas interaction)
25  *
26  * Alternatively, the contents of this file may be used under the terms of
27  * either the GNU General Public License Version 2 or later (the "GPL"), or
28  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
29  * in which case the provisions of the GPL or the LGPL are applicable instead
30  * of those above. If you wish to allow use of your version of this file only
31  * under the terms of either the GPL or the LGPL, and not to allow others to
32  * use your version of this file under the terms of the MPL, indicate your
33  * decision by deleting the provisions above and replace them with the notice
34  * and other provisions required by the GPL or the LGPL. If you do not delete
35  * the provisions above, a recipient may use your version of this file under
36  * the terms of any one of the MPL, the GPL or the LGPL.
37  *
38  * ***** END LICENSE BLOCK ***** */
39
40 Components.utils.import("resource://gre/modules/Services.jsm");
41
42 var gColorPalette = [];
43 var gStartTime = 0;
44 var gMbrotBundle;
45 var gCurrentImageData;
46 var gLastImageData;
47
48 function Startup() {
49   gMbrotBundle = document.getElementById("mbrotBundle");
50   document.getElementById("statusLabel").value = gMbrotBundle.getString("statusEmpty");
51
52   let img = document.getElementById("mbrotImage");
53   img.addEventListener("mouseup", imgEvHandler, false);
54   img.addEventListener("mousedown", imgEvHandler, false);
55   img.addEventListener("mousemove", imgEvHandler, false);
56   img.addEventListener("touchstart", imgEvHandler, false);
57   img.addEventListener("touchend", imgEvHandler, false);
58   img.addEventListener("touchcancel", imgEvHandler, false);
59   img.addEventListener("touchleave", imgEvHandler, false);
60   img.addEventListener("touchmove", imgEvHandler, false);
61
62   Services.obs.notifyObservers(window, "mandelbrot-loaded", null);
63 }
64
65 function getAdjustPref(prefname) {
66   let value;
67   switch (prefname) {
68     case "image.width":
69     case "image.height":
70       value = 0;
71       try {
72         value = Services.prefs.getIntPref("mandelbrot." + prefname);
73       }
74       catch (e) { }
75       if ((value < 10) || (value > 5000)) {
76         value = 300;
77         Services.prefs.setIntPref("mandelbrot." + prefname, value);
78       }
79       return value;
80     case "last_image.Cr_*":
81       let Cr_min = -2.0;
82       let Cr_max = 1.0;
83       try {
84         Cr_min = parseFloat(Services.prefs.getCharPref("mandelbrot.last_image.Cr_min"));
85         Cr_max = parseFloat(Services.prefs.getCharPref("mandelbrot.last_image.Cr_max"));
86       }
87       catch (e) { }
88       if ((Cr_min < -3) || (Cr_min > 2) ||
89           (Cr_max < -3) || (Cr_max > 2) || (Cr_min >= Cr_max)) {
90         Cr_min = -2.0; Cr_max = 1.0;
91       }
92       Services.prefs.setCharPref("mandelbrot.last_image.Cr_min", Cr_min);
93       Services.prefs.setCharPref("mandelbrot.last_image.Cr_max", Cr_max);
94       return {Cr_min: Cr_min, Cr_max: Cr_max};
95     case "last_image.Ci_*":
96       let Ci_min = -1.5;
97       let Ci_max = 1.5;
98       try {
99         Ci_min = parseFloat(Services.prefs.getCharPref("mandelbrot.last_image.Ci_min"));
100         Ci_max = parseFloat(Services.prefs.getCharPref("mandelbrot.last_image.Ci_max"));
101       }
102       catch (e) { }
103       if ((Ci_min < -2.5) || (Ci_min > 2.5) ||
104           (Ci_max < -2.5) || (Ci_max > 2.5) || (Ci_min >= Ci_max)) {
105         Ci_min = -1.5; Ci_max = 1.5;
106       }
107       Services.prefs.setCharPref("mandelbrot.last_image.Ci_min", Ci_min);
108       Services.prefs.setCharPref("mandelbrot.last_image.Ci_max", Ci_max);
109       return {Ci_min: Ci_min, Ci_max: Ci_max};
110     case "iteration_max":
111       value = 500;
112       try {
113         value = Services.prefs.getIntPref("mandelbrot." + prefname);
114       }
115       catch (e) {
116         setIter(value);
117       }
118       if (value < 10 || value > 10000) {
119         value = 500;
120         setIter(value);
121       }
122       return value;
123     case "use_algorithm":
124       value = "numeric";
125       try {
126         value = Services.prefs.getCharPref("mandelbrot." + prefname);
127       }
128       catch (e) {
129         setAlgorithm(value);
130       }
131       return value;
132    case "color_palette":
133       value = "kairo";
134       try {
135         value = Services.prefs.getCharPref("mandelbrot." + prefname);
136       }
137       catch(e) {
138         setPalette(value);
139       }
140       return value;
141    case "syncProportions":
142       value = true;
143       try {
144         value = Services.prefs.getBoolPref("mandelbrot." + prefname);
145       }
146       catch(e) {
147         Services.prefs.setBoolPref("mandelbrot." + prefname, value);
148       }
149       return value;
150     default:
151       return false;
152   }
153 }
154
155 function adjustCoordsAndDraw(aC_min, aC_max) {
156   let iWidth = getAdjustPref("image.width");
157   let iHeight = getAdjustPref("image.height");
158
159   // correct coordinates
160   if (aC_min.r < -2)
161     aC_min.r = -2;
162   if (aC_max.r > 2)
163     aC_max.r = 2;
164   if ((aC_min.r > 2) || (aC_max.r < -2) || (aC_min.r >= aC_max.r)) {
165     aC_min.r = -2.0; aC_max.r = 1.0;
166   }
167   if (aC_min.i < -2)
168     aC_min.i = -2;
169   if (aC_max.i > 2)
170     aC_max.i = 2;
171   if ((aC_min.i > 2) || (aC_max.i < -2) || (aC_min.i >= aC_max.i)) {
172     aC_min.i = -1.3; aC_max.i = 1.3;
173   }
174
175   let CWidth = aC_max.r - aC_min.r;
176   let CHeight = aC_max.i - aC_min.i;
177   let C_mid = new complex(aC_min.r + CWidth / 2, aC_min.i + CHeight / 2);
178
179   let CRatio = Math.max(CWidth / iWidth, CHeight / iHeight);
180
181   Services.prefs.setCharPref("mandelbrot.last_image.Cr_min", C_mid.r - iWidth * CRatio / 2);
182   Services.prefs.setCharPref("mandelbrot.last_image.Cr_max", C_mid.r + iWidth * CRatio / 2);
183   Services.prefs.setCharPref("mandelbrot.last_image.Ci_min", C_mid.i - iHeight * CRatio / 2);
184   Services.prefs.setCharPref("mandelbrot.last_image.Ci_max", C_mid.i + iHeight * CRatio / 2);
185
186   drawImage();
187 }
188
189 function drawImage() {
190   let canvas = document.getElementById("mbrotImage");
191   let context = canvas.getContext("2d");
192
193   document.getElementById("drawButton").hidden = true;
194
195   document.getElementById("statusLabel").value = gMbrotBundle.getString("statusDrawing");
196
197   if (gCurrentImageData) {
198     gLastImageData = gCurrentImageData;
199   }
200
201   let Cr_vals = getAdjustPref("last_image.Cr_*");
202   let Cr_min = Cr_vals.Cr_min;
203   let Cr_max = Cr_vals.Cr_max;
204
205   let Ci_vals = getAdjustPref("last_image.Ci_*");
206   let Ci_min = Ci_vals.Ci_min;
207   let Ci_max = Ci_vals.Ci_max;
208
209   let iterMax = getAdjustPref("iteration_max");
210   let algorithm = getAdjustPref("use_algorithm");
211
212   let currentPalette = getAdjustPref("color_palette");
213   if (!gColorPalette || !gColorPalette.length)
214     gColorPalette = getColorPalette(currentPalette);
215
216   let iWidth = getAdjustPref("image.width");
217   let iHeight = getAdjustPref("image.height");
218
219   gCurrentImageData = {
220     C_min: new complex(Cr_min, Ci_min),
221     C_max: new complex(Cr_max, Ci_max),
222     iWidth: iWidth,
223     iHeight: iHeight,
224     iterMax: iterMax
225   };
226
227   canvas.width = iWidth;
228   canvas.height = iHeight;
229
230   context.fillStyle = "rgba(255, 255, 255, 127)";
231   context.fillRect(0, 0, canvas.width, canvas.height);
232
233   gStartTime = new Date();
234
235   drawLine(0, [Cr_min, Cr_max, Ci_min, Ci_max],
236               canvas, context, iterMax, algorithm);
237 }
238
239 function drawLine(line, dimensions, canvas, context, iterMax, algorithm) {
240   let Cr_min = dimensions[0];
241   let Cr_max = dimensions[1];
242   let Cr_scale = Cr_max - Cr_min;
243
244   let Ci_min = dimensions[2];
245   let Ci_max = dimensions[3];
246   let Ci_scale = Ci_max - Ci_min;
247
248   let lines = Math.min(canvas.height - line, 8);
249   let imageData = context.createImageData(canvas.width, lines);
250   let pixels = imageData.data;
251   let idx = 0;
252   for (var img_y = line; img_y < canvas.height && img_y < line+8; img_y++)
253     for (let img_x = 0; img_x < canvas.width; img_x++) {
254       let C = new complex(Cr_min + (img_x / canvas.width) * Cr_scale,
255                           Ci_min + (img_y / canvas.height) * Ci_scale);
256       let colors = drawPoint(context, img_x, img_y, C, iterMax, algorithm);
257       pixels[idx++] = colors[0];
258       pixels[idx++] = colors[1];
259       pixels[idx++] = colors[2];
260       pixels[idx++] = colors[3];
261     }
262   context.putImageData(imageData, 0, line);
263
264   if (img_y < canvas.height)
265     setTimeout(drawLine, 0, img_y, dimensions, canvas, context, iterMax, algorithm);
266   else if (gStartTime)
267     EndCalc();
268 }
269
270 function EndCalc() {
271   let endTime = new Date();
272   let timeUsed = (endTime.getTime() - gStartTime.getTime()) / 1000;
273   document.getElementById("statusLabel").value =
274       gMbrotBundle.getFormattedString("statusTime", [timeUsed.toFixed(3)]);
275   gStartTime = 0;
276 }
277
278 function complex(aReal, aImag) {
279   this.r = aReal;
280   this.i = aImag;
281 }
282 complex.prototype = {
283   square: function() {
284     return new complex(this.r * this.r - this.i * this.i,
285                        2 * this.r * this.i);
286   },
287   dist: function() {
288     return Math.sqrt(this.r * this.r + this.i * this.i);
289   },
290   add: function(aComplex) {
291     return new complex(this.r + aComplex.r, this.i + aComplex.i);
292   }
293 }
294
295 function mandelbrotValueOO (aC, aIterMax) {
296   // this would be nice code in general but it looks like JS objects are too heavy for normal use.
297   let Z = new complex(0.0, 0.0);
298   for (var iter = 0; iter < aIterMax; iter++) {
299     Z = Z.square().add(aC);
300     if (Z.r * Z.r + Z.i * Z.i > 256) { break; }
301   }
302   return iter;
303 }
304
305 function mandelbrotValueNumeric (aC, aIterMax) {
306   // optimized numeric code for fast calculation
307   let Cr = aC.r, Ci = aC.i;
308   let Zr = 0.0, Zi = 0.0;
309   let Zr2 = Zr * Zr, Zi2 = Zi * Zi;
310   for (var iter = 0; iter < aIterMax; iter++) {
311     Zi = 2 * Zr * Zi + Ci;
312     Zr = Zr2 - Zi2 + Cr;
313
314     Zr2 = Zr * Zr; Zi2 = Zi * Zi;
315     if (Zr2 + Zi2 > 256) { break; }
316   }
317   return iter;
318 }
319
320 function getColor(aIterValue, aIterMax) {
321   let standardizedValue = Math.round(aIterValue * 1024 / aIterMax);
322   if (gColorPalette && gColorPalette.length)
323     return gColorPalette[standardizedValue];
324
325   // fallback to simple b/w if for some reason we don't have a palette
326   if (aIterValue == aIterMax)
327     return [0, 0, 0, 255];
328   else
329     return [255, 255, 255, 255];
330 }
331
332 function getColorPalette(palName) {
333   var palette = [];
334   switch (palName) {
335     case 'bw':
336       for (let i = 0; i < 1024; i++) {
337         palette[i] = [255, 255, 255, 255];
338       }
339       palette[1024] = [0, 0, 0, 255];
340       break;
341     case 'kairo':
342       // outer areas
343       for (let i = 0; i < 32; i++) {
344         let cc1 = Math.floor(i * 127 / 31);
345         let cc2 = 170 - Math.floor(i * 43 / 31);
346         palette[i] = [cc1, cc2, cc1, 255];
347       }
348       // inner areas
349       for (let i = 0; i < 51; i++) {
350         let cc = Math.floor(i * 170 / 50);
351         palette[32 + i] = [cc, 0, (170-cc), 255];
352       }
353       // corona
354       for (let i = 0; i < 101; i++) {
355         let cc = Math.floor(i * 200 / 100);
356         palette[83 + i] = [255, cc, 0, 255];
357       }
358       // inner corona
359       for (let i = 0; i < 201; i++) {
360         let cc1 = 255 - Math.floor(i * 85 / 200);
361         let cc2 = 200 - Math.floor(i * 30 / 200);
362         let cc3 = Math.floor(i * 170 / 200);
363         palette[184 + i] = [cc1, cc2, cc3, 255];
364       }
365       for (let i = 0; i < 301; i++) {
366         let cc1 = 170 - Math.floor(i * 43 / 300);
367         let cc2 = 170 + Math.floor(i * 85 / 300);
368         palette[385 + i] = [cc1, cc1, cc2, 255];
369       }
370       for (let i = 0; i < 338; i++) {
371         let cc = 127 + Math.floor(i * 128 / 337);
372         palette[686 + i] = [cc, cc, 255, 255];
373       }
374       palette[1024] = [0, 0, 0, 255];
375       break;
376     case 'rainbow-linear1':
377       for (let i = 0; i < 256; i++) {
378         palette[i] = [i, 0, 0, 255];
379         palette[256 + i] = [255, i, 0, 255];
380         palette[512 + i] = [255 - i, 255, i, 255];
381         palette[768 + i] = [i, 255-i, 255, 255];
382       }
383       palette[1024] = [0, 0, 0, 255];
384       break;
385     case 'rainbow-squared1':
386       for (let i = 0; i < 34; i++) {
387         let cc = Math.floor(i * 255 / 33);
388         palette[i] = [cc, 0, 0, 255];
389       }
390       for (let i = 0; i < 137; i++) {
391         let cc = Math.floor(i * 255 / 136);
392         palette[34 + i] = [255, cc, 0, 255];
393       }
394       for (let i = 0; i < 307; i++) {
395         let cc = Math.floor(i * 255 / 306);
396         palette[171 + i] = [255 - cc, 255, cc, 255];
397       }
398       for (let i = 0; i < 546; i++) {
399         let cc = Math.floor(i * 255 / 545);
400         palette[478 + i] = [cc, 255 - cc, 255, 255];
401       }
402       palette[1024] = [0, 0, 0, 255];
403       break;
404     case 'rainbow-linear2':
405       for (let i = 0; i < 205; i++) {
406         let cc = Math.floor(i * 255 / 204);
407         palette[i] = [255, cc, 0, 255];
408         palette[204 + i] = [255 - cc, 255, 0, 255];
409         palette[409 + i] = [0, 255, cc, 255];
410         palette[614 + i] = [0, 255 - cc, 255, 255];
411         palette[819 + i] = [cc, 0, 255, 255];
412       }
413       palette[1024] = [0, 0, 0, 255];
414       break;
415     case 'rainbow-squared2':
416       for (let i = 0; i < 19; i++) {
417         let cc = Math.floor(i * 255 / 18);
418         palette[i] = [255, cc, 0, 255];
419       }
420       for (let i = 0; i < 74; i++) {
421         let cc = Math.floor(i * 255 / 73);
422         palette[19 + i] = [255 - cc, 255, 0, 255];
423       }
424       for (let i = 0; i < 168; i++) {
425         let cc = Math.floor(i * 255 / 167);
426         palette[93 + i] = [0, 255, cc, 255];
427       }
428       for (let i = 0; i < 298; i++) {
429         let cc = Math.floor(i * 255 / 297);
430         palette[261 + i] = [0, 255 - cc, 255, 255];
431       }
432       for (let i = 0; i < 465; i++) {
433         let cc = Math.floor(i * 255 / 464);
434         palette[559 + i] = [cc, 0, 255, 255];
435       }
436       palette[1024] = [0, 0, 0, 255];
437       break;
438   }
439   /*
440      'Standard-Palette (QB-Colors)
441      For i = 0 To 1024
442          xx = CInt(i * 500 / 1024 + 2)
443          If xx <= 15 Then clr = xx
444          If xx > 15 Then clr = CInt(Sqr((xx - 15 + 1) * 15 ^ 2 / 485))
445          If xx >= 500 Then clr = 0
446          palette(i) = QBColor(clr)
447      Next
448   */
449   return palette;
450 }
451
452 function drawPoint(context, img_x, img_y, C, iterMax, algorithm) {
453   var itVal;
454   switch (algorithm) {
455     case 'oo':
456       itVal = mandelbrotValueOO(C, iterMax);
457       break;
458     case 'numeric':
459     default:
460       itVal = mandelbrotValueNumeric(C, iterMax);
461       break;
462   }
463   return getColor(itVal, iterMax);
464 }
465
466 /***** pure UI functions *****/
467
468 var zoomstart;
469 var imgBackup;
470 var zoomTouchID;
471
472 let imgEvHandler = {
473   handleEvent: function(aEvent) {
474     let canvas = document.getElementById("mbrotImage");
475     let context = canvas.getContext("2d");
476     let touchEvent = aEvent.type.indexOf('touch') != -1;
477
478     // Bail out if this is neither a touch nor left-click.
479     if (!touchEvent && aEvent.button != 0)
480       return;
481
482     // Bail out if the started touch can't be found.
483     if (touchEvent && zoomstart &&
484         !aEvent.changedTouches.identifiedTouch(zoomTouchID))
485       return;
486
487     let coordObj = touchEvent ?
488                    aEvent.changedTouches.identifiedTouch(zoomTouchID) :
489                    aEvent;
490
491     switch (aEvent.type) {
492       case 'mousedown':
493       case 'touchstart':
494         if (touchEvent) {
495           zoomTouchID = aEvent.changedTouches.item(0).identifier;
496           coordObj = aEvent.changedTouches.identifiedTouch(zoomTouchID);
497         }
498         // left button - start dragzoom
499         zoomstart = {x: coordObj.clientX - canvas.offsetLeft,
500                      y: coordObj.clientY - canvas.offsetTop};
501         imgBackup = context.getImageData(0, 0, canvas.width, canvas.height);
502         break;
503       case 'mouseup':
504       case 'touchend':
505         if (zoomstart) {
506           context.putImageData(imgBackup, 0, 0);
507           let zoomend = {x: coordObj.clientX - canvas.offsetLeft,
508                          y: coordObj.clientY - canvas.offsetTop};
509
510           // make sure zoomend is bigger than zoomstart
511           if ((zoomend.x == zoomstart.x) || (zoomend.y == zoomstart.y)) {
512             // cannot zoom what has no area, discard it
513             zoomstart = undefined;
514             return;
515           }
516           if (zoomend.x < zoomstart.x)
517             [zoomend.x, zoomstart.x] = [zoomstart.x, zoomend.x];
518           if (zoomend.y < zoomstart.y)
519             [zoomend.y, zoomstart.y] = [zoomstart.y, zoomend.y];
520
521           // determine new "coordinates"
522           let CWidth = gCurrentImageData.C_max.r - gCurrentImageData.C_min.r;
523           let CHeight = gCurrentImageData.C_max.i - gCurrentImageData.C_min.i;
524           let newC_min = new complex(
525               gCurrentImageData.C_min.r + zoomstart.x / gCurrentImageData.iWidth * CWidth,
526               gCurrentImageData.C_min.i + zoomstart.y / gCurrentImageData.iHeight * CHeight);
527           let newC_max = new complex(
528               gCurrentImageData.C_min.r + zoomend.x / gCurrentImageData.iWidth * CWidth,
529               gCurrentImageData.C_min.i + zoomend.y / gCurrentImageData.iHeight * CHeight);
530
531           adjustCoordsAndDraw(newC_min, newC_max);
532         }
533         zoomstart = undefined;
534         break;
535       case 'mousemove':
536       case 'touchmove':
537         if (zoomstart) {
538           context.putImageData(imgBackup, 0, 0);
539           context.strokeStyle = "rgb(255,255,31)";
540           context.strokeRect(zoomstart.x, zoomstart.y,
541                              coordObj.clientX - canvas.offsetLeft - zoomstart.x,
542                              coordObj.clientY - canvas.offsetTop - zoomstart.y);
543         }
544       break;
545     }
546   }
547 };
548
549 function saveImage() {
550   const nsIFilePicker = Components.interfaces.nsIFilePicker;
551   let fp = null;
552   try {
553     fp = Components.classes["@mozilla.org/filepicker;1"]
554                    .createInstance(nsIFilePicker);
555   } catch (e) {}
556   if (!fp) return;
557   let promptString = gMbrotBundle.getString("savePrompt");
558   fp.init(window, promptString, nsIFilePicker.modeSave);
559   fp.appendFilter(gMbrotBundle.getString("pngFilterName"), "*.png");
560   fp.defaultString = "mandelbrot.png";
561
562   let fpResult = fp.show();
563   if (fpResult != nsIFilePicker.returnCancel) {
564     saveCanvas(document.getElementById("mbrotImage"), fp.file);
565   }
566 }
567
568 function updateBookmarkMenu(aParent) {
569   document.getElementById("bookmarkSave").disabled =
570       (!document.getElementById("drawButton").hidden || (gStartTime > 0));
571
572   document.getElementById("bookmarkBack").disabled = !gLastImageData;
573
574   while (aParent.hasChildNodes() &&
575          aParent.lastChild.id != "bookmarkSeparator")
576     aParent.removeChild(aParent.lastChild);
577
578   let file = Components.classes["@mozilla.org/file/directory_service;1"]
579                        .getService(Components.interfaces.nsIProperties)
580                        .get("ProfD", Components.interfaces.nsIFile);
581   file.append("mandelbookmarks.sqlite");
582   if (file.exists()) {
583     let connection = Components.classes["@mozilla.org/storage/service;1"]
584                                .getService(Components.interfaces.mozIStorageService)
585                                .openDatabase(file);
586     try {
587       if (connection.tableExists("bookmarks")) {
588         let statement = connection.createStatement(
589             "SELECT name,ROWID FROM bookmarks ORDER BY ROWID ASC");
590         while (statement.executeStep()) {
591           let newItem = aParent.appendChild(document.createElement("menuitem"));
592           newItem.setAttribute("label", statement.getString(0));
593           newItem.setAttribute("bmRowID", statement.getString(1));
594         }
595         statement.reset();
596         statement.finalize();
597         return;
598       }
599     } finally {
600       connection.close();
601     }
602   }
603   // Create the "Nothing Available" Menu item and disable it.
604   let na = aParent.appendChild(document.createElement("menuitem"));
605   na.setAttribute("label", gMbrotBundle.getString("noBookmarks"));
606   na.setAttribute("disabled", "true");
607 }
608
609 function goBack() {
610   if (gLastImageData) {
611     Services.prefs.setIntPref("mandelbrot.iteration_max",
612                               gLastImageData.iterMax);
613     // use gLastImageData.iWidth, gLastImageData.iHeight ???
614     adjustCoordsAndDraw(gLastImageData.C_min, gLastImageData.C_max);
615     gLastImageData = undefined;
616   }
617 }
618
619 function callBookmark(evtarget) {
620   if (evtarget.id == "bookmarkSave" || evtarget.id == "bookmarkSeparator")
621     return;
622   if (evtarget.id == "bookmarkOverview") {
623     adjustCoordsAndDraw(new complex(0,0), new complex(0,0));
624     return;
625   }
626
627   if (evtarget.getAttribute('bmRowID')) {
628     let iterMax = 0;
629     let C_min = null;
630     let C_max = null;
631
632     let file = Components.classes["@mozilla.org/file/directory_service;1"]
633                          .getService(Components.interfaces.nsIProperties)
634                          .get("ProfD", Components.interfaces.nsIFile);
635     file.append("mandelbookmarks.sqlite");
636     let connection = Components.classes["@mozilla.org/storage/service;1"]
637                                .getService(Components.interfaces.mozIStorageService)
638                                .openDatabase(file);
639     let statement = connection.createStatement(
640         "SELECT iteration_max,Cr_min,Cr_max,Ci_min,Ci_max FROM bookmarks WHERE ROWID=?1");
641     statement.bindStringParameter(0, evtarget.getAttribute('bmRowID'));
642     while (statement.executeStep()) {
643       iterMax = statement.getInt32(0);
644       C_min = new complex(statement.getDouble(1), statement.getDouble(3));
645       C_max = new complex(statement.getDouble(2), statement.getDouble(4));
646     }
647     statement.finalize();
648     connection.close();
649
650     if (iterMax && C_min && C_max) {
651       Services.prefs.setIntPref("mandelbrot.iteration_max", iterMax);
652       adjustCoordsAndDraw(C_min, C_max);
653     }
654   }
655 }
656
657 function saveBookmark() {
658   // retrieve wanted bookmark name with a prompt
659   let prompts = Components.classes["@mozilla.org/embedcomp/prompt-service;1"]
660                           .getService(Components.interfaces.nsIPromptService);
661   let input = {value: ""}; // empty default value
662   let ok = prompts.prompt(null, gMbrotBundle.getString("saveBookmarkTitle"), gMbrotBundle.getString("saveBookmarkLabel"), input, null, {});
663   // ok is true if OK is pressed, false if Cancel. input.value holds the value of the edit field if "OK" was pressed.
664   if (!ok || !input.value)
665     return
666
667   let bmName = input.value;
668
669   // Open or create the bookmarks database.
670   let file = Components.classes["@mozilla.org/file/directory_service;1"]
671                        .getService(Components.interfaces.nsIProperties)
672                        .get("ProfD", Components.interfaces.nsIFile);
673   file.append("mandelbookmarks.sqlite");
674   let connection = Components.classes["@mozilla.org/storage/service;1"]
675                              .getService(Components.interfaces.mozIStorageService)
676                              .openDatabase(file);
677   connection.beginTransaction();
678   if (!connection.tableExists("bookmarks"))
679     connection.createTable("bookmarks", "name TEXT, iteration_max INTEGER, Cr_min REAL, Cr_max REAL, Ci_min REAL, Ci_max REAL");
680   // NULL. The value is a NULL value.
681   // INTEGER. The value is a signed integer, stored in 1, 2, 3, 4, 6, or 8 bytes depending on the magnitude of the value.
682   // REAL. The value is a floating point value, stored as an 8-byte IEEE floating point number.
683   // TEXT. The value is a text string, stored using the database encoding (UTF-8, UTF-16BE or UTF-16-LE).
684
685   // Put value of the current image into the bookmarks table
686   let statement = connection.createStatement(
687       "INSERT INTO bookmarks (name,iteration_max,Cr_min,Cr_max,Ci_min,Ci_max) VALUES (?1,?2,?3,?4,?5,?6)");
688   statement.bindStringParameter(0, bmName);
689   statement.bindStringParameter(1, gCurrentImageData.iterMax);
690   statement.bindStringParameter(2, gCurrentImageData.C_min.r);
691   statement.bindStringParameter(3, gCurrentImageData.C_max.r);
692   statement.bindStringParameter(4, gCurrentImageData.C_min.i);
693   statement.bindStringParameter(5, gCurrentImageData.C_max.i);
694   statement.execute();
695   statement.finalize();
696   connection.commitTransaction();
697   connection.close();
698 }
699
700 function imgSettings() {
701   let anchor = null;
702   let position = "before_start";
703   if (document.getElementById("mandelbrot-page").nodeName == "page") {
704     anchor = document.getElementById("mandelbrotToolbar");
705   }
706   else {
707     anchor = document.getElementById("mandelbrotMenubar");
708     position = "after_start";
709   }
710   document.getElementById("imgSettingsPanel").showPopup(anchor, position);
711 }
712
713 function initImgSettings() {
714   // Get values from prefs.
715   for each (let coord in ["Cr", "Ci"]) {
716     let coord_vals = getAdjustPref("last_image." + coord + "_*");
717     document.getElementById("is_" + coord + "_min").value = coord_vals[coord + "_min"];
718     document.getElementById("is_" + coord + "_max").value = coord_vals[coord + "_max"];
719   }
720   for each (let dim in ["width", "height"]) {
721     document.getElementById("is_img_" + dim).value = getAdjustPref("image." + dim);
722   }
723   document.getElementById("is_syncProp").checked = getAdjustPref("syncProportions");
724
725   // Calculate scales.
726   recalcCoord("Cr", "scale");
727   recalcCoord("Ci", "scale");
728
729   // Clear the preview.
730   let canvas = document.getElementById("is_mbrotPreview");
731   let context = canvas.getContext("2d");
732   context.fillStyle = "rgba(255, 255, 255, 127)";
733   context.fillRect(0, 0, canvas.width, canvas.height);
734
735   // Set lists to correct values.
736   updateIterList();
737   updatePaletteList();
738   updateAlgoList();
739 }
740
741 function closeImgSettings() {
742   // Hide popup, which will automatically make a call to save values.
743   document.getElementById("imgSettingsPanel").hidePopup();
744 }
745
746 function saveImgSettings() {
747   // Get values to prefs.
748   for each (let coord in ["Cr_min", "Cr_max", "Ci_min", "Ci_max"]) {
749     Services.prefs.setCharPref("mandelbrot.last_image." + coord,
750                                document.getElementById("is_" + coord).value);
751   }
752   for each (let dim in ["width", "height"]) {
753     Services.prefs.setIntPref("mandelbrot.image." + dim,
754                               document.getElementById("is_img_" + dim).value);
755   }
756   Services.prefs.setBoolPref("mandelbrot.syncProportions",
757                              document.getElementById("is_syncProp").checked);
758 }
759
760 function checkISValue(textbox, type) {
761   if (type == "coord") {
762     textbox.value = roundCoord(parseFloat(textbox.value));
763   }
764   else if (type == "dim") {
765     textbox.value = parseInt(textbox.value);
766   }
767 }
768
769 function drawPreview() {
770   let canvas = document.getElementById("is_mbrotPreview");
771   let context = canvas.getContext("2d");
772
773   if (document.getElementById("is_img_width").value /
774       document.getElementById("is_img_height").value
775         < 80 / 50) {
776     canvas.height = 50;
777     canvas.width = canvas.height *
778       document.getElementById("is_img_width").value /
779       document.getElementById("is_img_height").value;
780   }
781   else {
782     canvas.width = 80;
783     canvas.height = canvas.width *
784       document.getElementById("is_imgHeight").value /
785       document.getElementById("is_imgWidth").value;
786   }
787
788   let Cr_min = parseFloat(document.getElementById("is_Cr_min").value);
789   let Cr_max = parseFloat(document.getElementById("is_Cr_max").value);
790   if ((Cr_min < -2) || (Cr_min > 2) ||
791       (Cr_max < -2) || (Cr_max > 2) || (Cr_min >= Cr_max)) {
792     Cr_min = -2.0; Cr_max = 1.0;
793   }
794
795   let Ci_min = parseFloat(document.getElementById("is_Ci_min").value);
796   let Ci_max = parseFloat(document.getElementById("is_Ci_max").value);
797   if ((Ci_min < -2) || (Ci_min > 2) ||
798       (Ci_max < -2) || (Ci_max > 2) || (Ci_min >= Ci_max)) {
799     Ci_min = -2.0; Ci_max = 1.0;
800   }
801
802   let iterMax = getAdjustPref("iteration_max");
803   let algorithm = getAdjustPref("use_algorithm");
804
805   context.fillStyle = "rgba(255, 255, 255, 127)";
806   context.fillRect(0, 0, canvas.width, canvas.height);
807
808   let currentPalette = getAdjustPref("color_palette");
809   gColorPalette = getColorPalette(currentPalette);
810
811   drawLine(0, [Cr_min, Cr_max, Ci_min, Ci_max],
812               canvas, context, iterMax, algorithm);
813 }
814
815 function recalcCoord(coord, target) {
816   let othercoord = (coord == "Ci") ? "Cr" : "Ci";
817   let owndim = (coord == "Ci") ? "height" : "width";
818   let otherdim = (coord == "Ci") ? "width" : "height";
819   let myscale;
820   if (target == "scale") {
821     myscale =
822       parseFloat(document.getElementById("is_" + coord + "_max").value) -
823       parseFloat(document.getElementById("is_" + coord + "_min").value);
824     document.getElementById("is_" + coord + "_scale").value = roundCoord(myscale);
825   }
826   else if (target == 'max') {
827     let mymax =
828       parseFloat(document.getElementById("is_" + coord + "_min").value) +
829       parseFloat(document.getElementById("is_" + coord + "_scale").value);
830     document.getElementById("is_" + coord + "_max").value = roundCoord(mymax);
831     myscale = document.getElementById("is_" + coord + "_scale").value;
832   }
833   if (document.getElementById("is_syncProp").checked) {
834     let otherscale = myscale *
835       document.getElementById("is_img_" + otherdim).value /
836       document.getElementById("is_img_" + owndim).value;
837     document.getElementById("is_" + othercoord + "_scale").value = roundCoord(otherscale);
838     let othermax =
839       parseFloat(document.getElementById("is_" + othercoord + "_min").value) +
840       parseFloat(document.getElementById("is_" + othercoord + "_scale").value);
841     document.getElementById("is_" + othercoord + "_max").value = roundCoord(othermax);
842   }
843 }
844
845 function checkProportions() {
846   if (!document.getElementById("is_syncProp").checked) {
847     recalcCoord("Cr", "scale");
848   }
849 }
850
851 function roundCoord(floatval) {
852   // We should round to 10 decimals here or so
853   return parseFloat(floatval.toFixed(10));
854 }
855
856 function updateIterList() {
857   let currentIter = getAdjustPref("iteration_max");
858   document.getElementById("iterList").value = currentIter;
859 }
860
861 function updatePaletteList() {
862   let currentPalette = getAdjustPref("color_palette");
863   if (!gColorPalette || !gColorPalette.length)
864     gColorPalette = getColorPalette(currentPalette);
865   document.getElementById("colorList").value = currentPalette;
866 }
867
868 function updateAlgoList() {
869   let currentAlgo = getAdjustPref("use_algorithm");
870   document.getElementById("algoList").value = currentAlgo;
871 }
872
873 function setIter(aIter) {
874   Services.prefs.setIntPref("mandelbrot.iteration_max", aIter);
875 }
876
877 function setPalette(aPaletteID) {
878   Services.prefs.setCharPref("mandelbrot.color_palette", aPaletteID);
879   gColorPalette = getColorPalette(aPaletteID);
880 }
881
882 function setAlgorithm(algoID) {
883   Services.prefs.setCharPref("mandelbrot.use_algorithm", algoID);
884 }
885
886
887 /***** helper functions from external sources *****/
888
889 // function below is based on http://developer.mozilla.org/en/docs/Code_snippets:Canvas
890 // custom modifications:
891 //   - use "a"-prefix on function arguments
892 //   - take an nsILocalFile as aDestFile argument
893 //   - always do silent download
894 function saveCanvas(aCanvas, aDestFile) {
895   // create a data url from the canvas and then create URIs of the source and targets  
896   var io = Components.classes["@mozilla.org/network/io-service;1"]
897                      .getService(Components.interfaces.nsIIOService);
898   var source = io.newURI(aCanvas.toDataURL("image/png", ""), "UTF8", null);
899
900   // prepare to save the canvas data
901   var persist = Components.classes["@mozilla.org/embedding/browser/nsWebBrowserPersist;1"]
902                           .createInstance(Components.interfaces.nsIWebBrowserPersist);
903
904   persist.persistFlags = Components.interfaces.nsIWebBrowserPersist.PERSIST_FLAGS_REPLACE_EXISTING_FILES;
905   persist.persistFlags |= Components.interfaces.nsIWebBrowserPersist.PERSIST_FLAGS_AUTODETECT_APPLY_CONVERSION;
906
907   // save the canvas data to the file
908   persist.saveURI(source, null, null, null, null, aDestFile);
909 }