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