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