mandelbrot

annotate src/mbrot.c @ 0:4d85805eb875

mandelbrot initial import
author John Tsiombikas <nuclear@mutantstargoat.com>
date Tue, 19 Jun 2012 06:48:38 +0300
parents
children
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 }