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