You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
94 lines
2.0 KiB
C
94 lines
2.0 KiB
C
#include "SDL.h"
|
|
#include <complex.h>
|
|
#include <stdbool.h>
|
|
|
|
/* The maximum iteration count. */
|
|
const int max_it = 170;
|
|
|
|
/* Compute the out-coloring based on the iteration counter. */
|
|
Uint32 outcolor(int it) {
|
|
return 0x00010001 * ((it * 0xff) / max_it);
|
|
}
|
|
|
|
/* "Compute" the in-coloring. */
|
|
Uint32 incolor() {
|
|
return 0x00000000; /* black */
|
|
}
|
|
|
|
void drawmandelbrot(SDL_Surface * surface) {
|
|
if (SDL_MUSTLOCK(surface)) {
|
|
SDL_LockSurface(surface);
|
|
}
|
|
Uint32 *pixels = (Uint32 *) surface->pixels;
|
|
|
|
#pragma omp parallel for schedule(dynamic, 50000)
|
|
for (int i = 0; i < surface->w * surface->h; i++) {
|
|
int y = i / surface->w;
|
|
int x = i % surface->w;
|
|
|
|
float complex c = ((3.0f * x / surface->w) - 2.0f)
|
|
+ I * ((2.0f * y / surface->h) - 1.0f);
|
|
|
|
bool diverges = false;
|
|
float complex z = 0;
|
|
int it;
|
|
for (it = 1; it <= max_it; it++) {
|
|
/* z = z² + c */
|
|
z = cpowf(z, 2) + c;
|
|
|
|
/* If |z| ever gets greater than 2, it diverges. */
|
|
if (cabsf(z) > 2) {
|
|
diverges = true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
Uint32 color;
|
|
if (diverges) {
|
|
color = outcolor(it);
|
|
} else {
|
|
color = incolor();
|
|
}
|
|
|
|
pixels[i] = color;
|
|
|
|
/* Update the screen every 10 lines. */
|
|
if (y % 10 == 0 && x == 0) {
|
|
#pragma omp critical
|
|
{
|
|
SDL_Flip(surface);
|
|
}
|
|
}
|
|
}
|
|
|
|
/* Update the screen a final time. */
|
|
SDL_Flip(surface);
|
|
if (SDL_MUSTLOCK(surface)) {
|
|
SDL_UnlockSurface(surface);
|
|
}
|
|
}
|
|
|
|
int main(void) {
|
|
/* Set up SDL. */
|
|
if (SDL_Init(SDL_INIT_EVERYTHING) < 0) {
|
|
fprintf(stderr, "Error: Could not initialize SDL: %s.\n",
|
|
SDL_GetError());
|
|
exit(1);
|
|
}
|
|
SDL_Surface *screen = SDL_SetVideoMode(1024, 768, 32, SDL_SWSURFACE);
|
|
|
|
/* Do the mandelbrot. */
|
|
drawmandelbrot(screen);
|
|
|
|
/* Save BMP */
|
|
char *file = "mandelbrot.bmp";
|
|
if (SDL_SaveBMP(screen, file) != 0) {
|
|
fprintf(stderr, "Could not write %s!\n", file);
|
|
}
|
|
|
|
/* Quit. */
|
|
/* FIXME: wait for key press here */
|
|
SDL_Delay(0*20000);
|
|
SDL_Quit();
|
|
}
|