rev |
line source |
nuclear@0
|
1 /*
|
nuclear@0
|
2 A simple interactive mandelbrot fractal explorer
|
nuclear@0
|
3 Copyright (C) John Tsiombikas <nuclear@member.fsf.org>
|
nuclear@0
|
4
|
nuclear@0
|
5 This program is free software: you can redistribute it and/or modify
|
nuclear@0
|
6 it under the terms of the GNU General Public License as published by
|
nuclear@0
|
7 the Free Software Foundation, either version 3 of the License, or
|
nuclear@0
|
8 (at your option) any later version.
|
nuclear@0
|
9
|
nuclear@0
|
10 This program is distributed in the hope that it will be useful,
|
nuclear@0
|
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
|
nuclear@0
|
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
nuclear@0
|
13 GNU General Public License for more details.
|
nuclear@0
|
14
|
nuclear@0
|
15 You should have received a copy of the GNU General Public License
|
nuclear@0
|
16 along with this program. If not, see <http://www.gnu.org/licenses/>.
|
nuclear@0
|
17 */
|
nuclear@0
|
18 #include "mbrot.h"
|
nuclear@0
|
19 #include "palette.h"
|
nuclear@0
|
20
|
nuclear@0
|
21 static int mandelbrot(double cx, double cy, int iter);
|
nuclear@0
|
22
|
nuclear@0
|
23 void draw_mandelbrot(unsigned char *pix, int xsz, int ysz, struct area *area, int iter)
|
nuclear@0
|
24 {
|
nuclear@0
|
25 int i, j;
|
nuclear@0
|
26 double x, y, dx, dy;
|
nuclear@0
|
27
|
nuclear@0
|
28 dx = area->width / (double)xsz;
|
nuclear@0
|
29 dy = area->height / (double)ysz;
|
nuclear@0
|
30
|
nuclear@0
|
31 y = area->y;
|
nuclear@0
|
32 for(i=0; i<ysz; i++) {
|
nuclear@0
|
33 x = area->x;
|
nuclear@0
|
34 for(j=0; j<xsz; j++) {
|
nuclear@0
|
35 int res = mandelbrot(x, y, iter);
|
nuclear@0
|
36 /* visualize according to the number of iterations
|
nuclear@0
|
37 * required to determine membership in the mandelbrot set.
|
nuclear@0
|
38 * Members of the set will have res = iter and so col = PAL_SIZE-1
|
nuclear@0
|
39 */
|
nuclear@0
|
40 int col = (PAL_SIZE - 1) * res / iter;
|
nuclear@0
|
41 *pix++ = palette[col].r;
|
nuclear@0
|
42 *pix++ = palette[col].g;
|
nuclear@0
|
43 *pix++ = palette[col].b;
|
nuclear@0
|
44 *pix++ = 0;
|
nuclear@0
|
45
|
nuclear@0
|
46 x += dx;
|
nuclear@0
|
47 }
|
nuclear@0
|
48 y += dy;
|
nuclear@0
|
49 }
|
nuclear@0
|
50 }
|
nuclear@0
|
51
|
nuclear@0
|
52 static int mandelbrot(double cx, double cy, int iter)
|
nuclear@0
|
53 {
|
nuclear@0
|
54 int i;
|
nuclear@0
|
55 double zx, zy; /* complex number Z */
|
nuclear@0
|
56
|
nuclear@0
|
57 /* start with Z0 = C */
|
nuclear@0
|
58 zx = cx;
|
nuclear@0
|
59 zy = cy;
|
nuclear@0
|
60
|
nuclear@0
|
61 for(i=0; i<iter; i++) {
|
nuclear@0
|
62 /* calculate Zn+1 = Zn * Zn + C */
|
nuclear@0
|
63 double x = (zx * zx - zy * zy) + cx;
|
nuclear@0
|
64 double y = (zy * zx + zx * zy) + cy;
|
nuclear@0
|
65 zx = x;
|
nuclear@0
|
66 zy = y;
|
nuclear@0
|
67
|
nuclear@0
|
68 /* escape if ||Z|| > 2 */
|
nuclear@0
|
69 if((x * x + y * y) > 4.0) {
|
nuclear@0
|
70 break;
|
nuclear@0
|
71 }
|
nuclear@0
|
72 }
|
nuclear@0
|
73
|
nuclear@0
|
74 return i;
|
nuclear@0
|
75 }
|