updated mandelbrot demo
This commit is contained in:
Joel Bodenmann
parent
6c23777b5b
commit
c15b2b4200
2 changed files with 38 additions and 128 deletions
|
|
@ -22,17 +22,53 @@
|
||||||
#include "hal.h"
|
#include "hal.h"
|
||||||
#include "gdisp.h"
|
#include "gdisp.h"
|
||||||
|
|
||||||
|
void mandelbrot(float x1, float y1, float x2, float y2) {
|
||||||
|
unsigned int i,j;
|
||||||
|
uint16_t iter;
|
||||||
|
color_t color;
|
||||||
|
|
||||||
|
float sy = y2 - y1;
|
||||||
|
float sx = x2 - x1;
|
||||||
|
const int MAX = 512;
|
||||||
|
|
||||||
|
for(i = 0; i < 320; i++) {
|
||||||
|
for(j = 0; j < 240; j++) {
|
||||||
|
float cy = j * sy / 240.0f + y1;
|
||||||
|
float cx = i * sx / 320.0f + x1;
|
||||||
|
float x=0.0f, y=0.0f, xx=0.0f, yy=0.0f;
|
||||||
|
for(iter=0; iter <= MAX && xx+yy<4.0f; iter++) {
|
||||||
|
xx = x*x;
|
||||||
|
yy = y*y;
|
||||||
|
y = 2.0f*x*y + cy;
|
||||||
|
x = xx - yy + cx;
|
||||||
|
}
|
||||||
|
color = ((iter << 8) | (iter&0xFF));
|
||||||
|
//color = RGB565CONVERT(iter*4, iter*13, iter*10);
|
||||||
|
gdispDrawPixel(i, j, color);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
int main(void) {
|
int main(void) {
|
||||||
|
float cx, cy;
|
||||||
|
float zoom = 1.0f;
|
||||||
|
|
||||||
halInit();
|
halInit();
|
||||||
chSysInit();
|
chSysInit();
|
||||||
|
|
||||||
gdispInit();
|
gdispInit();
|
||||||
gdispSetOrientation(GDISP_ROTATE_270);
|
gdispSetOrientation(GDISP_ROTATE_270);
|
||||||
|
|
||||||
mandelbrotInit(128, 128, 512);
|
/* where to zoom in */
|
||||||
|
cx = -0.086f;
|
||||||
|
cy = 0.85f;
|
||||||
|
|
||||||
while(TRUE) {
|
while(TRUE) {
|
||||||
|
mandelbrot(-2.0f*zoom+cx, -1.5f*zoom+cy, 2.0f*zoom+cx, 1.5f*zoom+cy);
|
||||||
|
|
||||||
|
zoom *= 0.7f;
|
||||||
|
if(zoom <= 0.00001f)
|
||||||
|
zoom = 1.0f;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -1,126 +0,0 @@
|
||||||
/*
|
|
||||||
ChibiOS/GFX - Copyright (C) 2012
|
|
||||||
Joel Bodenmann aka Tectu <joel@unormal.org>
|
|
||||||
|
|
||||||
This file is part of ChibiOS/GFX.
|
|
||||||
|
|
||||||
ChibiOS/GFX is free software; you can redistribute it and/or modify
|
|
||||||
it under the terms of the GNU General Public License as published by
|
|
||||||
the Free Software Foundation; either version 3 of the License, or
|
|
||||||
(at your option) any later version.
|
|
||||||
|
|
||||||
ChibiOS/GFX is distributed in the hope that it will be useful,
|
|
||||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
||||||
GNU General Public License for more details.
|
|
||||||
|
|
||||||
You should have received a copy of the GNU General Public License
|
|
||||||
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#include "ch.h"
|
|
||||||
#include "hal.h"
|
|
||||||
#include "gdisp.h"
|
|
||||||
|
|
||||||
int count = 0;
|
|
||||||
float mag = 1;
|
|
||||||
float offsetx = -0.756423894274328;
|
|
||||||
float offsety = 0.064179410646170;
|
|
||||||
static float width, height;
|
|
||||||
static uint16_t iterations, screen_off_x, screen_off_y;
|
|
||||||
static uint16_t *FrameBuffer;
|
|
||||||
|
|
||||||
static int generateColor(int count, int base) {
|
|
||||||
int r,g,b;
|
|
||||||
int d = (count % base) * 256 / base;
|
|
||||||
int m = (int)(d / 42.667);
|
|
||||||
|
|
||||||
switch(m) {
|
|
||||||
case 0: r=0; g=6*d; b=255; break;
|
|
||||||
case 1: r=0; g=255; b=255-6*(d-43); break;
|
|
||||||
case 2: r=6*(d-86); g=255; b=0; break;
|
|
||||||
case 3: r=255; g=255-6*(d-129); b=0; break;
|
|
||||||
case 4: r=255; g=0; b=6*(d-171); break;
|
|
||||||
case 5: r=255-6*(d-214); g=0; b=255; break;
|
|
||||||
default: r=0; g=0; b=0; break;
|
|
||||||
}
|
|
||||||
|
|
||||||
return (((r >> 3) & 0x1f) << 11) | (((g >> 2) & 0x3f) << 5) | ((b >> 3) & 0x1f);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void complex_mult(float ar,float ai,float br,float bi, float *cr,float *ci) {
|
|
||||||
// C = A * B
|
|
||||||
*cr = ar * br - ai * bi;
|
|
||||||
*ci = ar * bi + ai * br;
|
|
||||||
}
|
|
||||||
|
|
||||||
static float complex_abs2(float r,float i) {
|
|
||||||
float ar,ai;
|
|
||||||
complex_mult(r,i,r,-i,&ar,&ai);
|
|
||||||
|
|
||||||
return ar;
|
|
||||||
}
|
|
||||||
|
|
||||||
void mandelbrotInit(uint16_t mandel_width, uint16_t mandel_height, uint16_t mandel_iterations) {
|
|
||||||
const char *msg = "ChibiOS/GFX";
|
|
||||||
|
|
||||||
width = (float)mandel_width;
|
|
||||||
height = (float)mandel_height;
|
|
||||||
iterations = mandel_iterations;
|
|
||||||
|
|
||||||
FrameBuffer = chHeapAlloc(NULL, width * height * sizeof(uint16_t));
|
|
||||||
if(FrameBuffer == NULL)
|
|
||||||
return;
|
|
||||||
|
|
||||||
screen_off_x = (gdispGetWidth() - width) / 2;
|
|
||||||
screen_off_y = (gdispGetHeight() - height) / 2;
|
|
||||||
|
|
||||||
gdispClear(Black);
|
|
||||||
gdispDrawString((gdispGetWidth() - gdispGetStringWidth(msg, &fontUI2Double)) / 2, 5, "ChibiOS/GFX", &fontUI2Double, White);
|
|
||||||
|
|
||||||
while(TRUE) {
|
|
||||||
mandelbrotLoop();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void mandelbrotLoop() {
|
|
||||||
int x, y;
|
|
||||||
int pos = 0;
|
|
||||||
|
|
||||||
for(y = 0; y < height; y++) {
|
|
||||||
for(x = 0; x < width; x++) {
|
|
||||||
float cr = (x - width/2.) / (width/2.) / mag + offsetx;
|
|
||||||
float ci = (y - width/2.) / (width/2.) / mag + offsety;
|
|
||||||
float zr = 0;
|
|
||||||
float zi = 0;
|
|
||||||
int t;
|
|
||||||
|
|
||||||
for(t = 0; t < iterations; t++) {
|
|
||||||
if(complex_abs2(zr,zi) > 4)
|
|
||||||
break;
|
|
||||||
complex_mult(zr,zi,zr,zi,&zr,&zi);
|
|
||||||
zr += cr;
|
|
||||||
zi += ci;
|
|
||||||
}
|
|
||||||
|
|
||||||
FrameBuffer[pos++] = generateColor(t, 64);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
gdispBlitArea(screen_off_x, screen_off_y, width, height, FrameBuffer);
|
|
||||||
|
|
||||||
// magnification
|
|
||||||
mag = mag * 1.2;
|
|
||||||
|
|
||||||
// panning
|
|
||||||
//offsety += 0.001;
|
|
||||||
//offsetx += 0.0016;
|
|
||||||
|
|
||||||
if(count++ > 80) {
|
|
||||||
offsetx = -0.756423894274328;
|
|
||||||
offsety = 0.064179410646170;
|
|
||||||
mag = 1;
|
|
||||||
count = 0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Loading…
Add table
Reference in a new issue