fltk/test/mandelbrot.cxx

228 lines
5.7 KiB
C++

//
// "$Id$"
//
// Mandelbrot set demo for the Fast Light Tool Kit (FLTK).
//
// Copyright 1998-2010 by Bill Spitzak and others.
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Library General Public
// License as published by the Free Software Foundation; either
// version 2 of the License, or (at your option) any later version.
//
// This library 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
// Library General Public License for more details.
//
// You should have received a copy of the GNU Library General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
// USA.
//
// Please report all bugs and problems on the following page:
//
// http://www.fltk.org/str.php
//
#include "mandelbrot_ui.h"
#include <FL/fl_draw.H>
#include <FL/Fl_Button.H>
#include <FL/Fl_Printer.H>
#include <stdio.h>
#include <stdlib.h>
Drawing_Window mbrot;
Drawing_Window jbrot;
void idle(void*) {
if (!mbrot.d->idle() && !(jbrot.d && jbrot.d->idle())) Fl::remove_idle(idle);
}
void set_idle() {
Fl::add_idle(idle);
}
static void window_callback(Fl_Widget*, void*) {exit(0);}
static void print(Fl_Widget *o, void *data)
{
Fl_Printer printer;
Fl_Window *win = o->window();
if(!win->visible()) return;
win->make_current();
uchar *image_data = fl_read_image(NULL, 0, 0, win->w(), win->h(), 0);
if( printer.start_job(1) ) return;
if( printer.start_page() ) return;
printer.scale(.7,.7);
fl_draw_image(image_data, 0,0, win->w(), win->h());
printer.end_page();
delete image_data;
printer.end_job();
}
int main(int argc, char **argv) {
mbrot.make_window();
mbrot.window->begin();
Fl_Button* o = new Fl_Button(0, 0, 0, 0, NULL);
o->callback(print,NULL);
o->shortcut(FL_CTRL+'p');
mbrot.window->end();
mbrot.d->X = -.75;
mbrot.d->scale = 2.5;
mbrot.update_label();
int i = 0;
if (Fl::args(argc,argv,i) < argc) Fl::fatal(Fl::help);
Fl::visual(FL_RGB);
mbrot.window->callback(window_callback);
mbrot.window->show(argc,argv);
Fl::run();
return 0;
}
void Drawing_Window::update_label() {
char buffer[128];
sprintf(buffer, "%+.10f", d->X); x_input->value(buffer);
sprintf(buffer, "%+.10f", d->Y); y_input->value(buffer);
sprintf(buffer, "%.2g", d->scale); w_input->value(buffer);
}
void Drawing_Area::draw() {
draw_box();
drawn = 0;
set_idle();
}
int Drawing_Area::idle() {
if (!window()->visible()) return 0;
if (drawn < nextline) {
window()->make_current();
int yy = drawn+y()+4;
if (yy >= sy && yy <= sy+sh) erase_box();
fl_draw_image_mono(buffer+drawn*W,x()+3,yy,W,1,1,W);
drawn++;
return 1;
}
if (nextline < H) {
if (!buffer) buffer = new uchar[W*H];
double yy = Y+(H/2-nextline)*scale/W;
double yi = yy; if (julia) yy = jY;
uchar *p = buffer+nextline*W;
for (int xi = 0; xi < W; xi++) {
double xx = X+(xi-W/2)*scale/W;
double wx = xx; double wy = yi;
if (julia) xx = jX;
for (int i=0; ; i++) {
if (i >= iterations) {*p = 0; break;}
double t = wx*wx - wy*wy + xx;
wy = 2*wx*wy + yy;
wx = t;
if (wx*wx + wy*wy > 4) {
wx = t = 1-double(i)/(1<<10);
if (t <= 0) t = 0; else for (i=brightness; i--;) t*=wx;
*p = 255-int(254*t);
break;
}
}
p++;
}
nextline++;
return nextline < H;
}
return 0;
}
void Drawing_Area::erase_box() {
window()->make_current();
fl_overlay_clear();
}
int Drawing_Area::handle(int event) {
static int ix, iy;
static int dragged;
static int button;
int x2,y2;
switch (event) {
case FL_PUSH:
erase_box();
ix = Fl::event_x(); if (ix<x()) ix=x(); if (ix>=x()+w()) ix=x()+w()-1;
iy = Fl::event_y(); if (iy<y()) iy=y(); if (iy>=y()+h()) iy=y()+h()-1;
dragged = 0;
button = Fl::event_button();
return 1;
case FL_DRAG:
dragged = 1;
erase_box();
x2 = Fl::event_x(); if (x2<x()) x2=x(); if (x2>=x()+w()) x2=x()+w()-1;
y2 = Fl::event_y(); if (y2<y()) y2=y(); if (y2>=y()+h()) y2=y()+h()-1;
if (button != 1) {ix = x2; iy = y2; return 1;}
if (ix < x2) {sx = ix; sw = x2-ix;} else {sx = x2; sw = ix-x2;}
if (iy < y2) {sy = iy; sh = y2-iy;} else {sy = y2; sh = iy-y2;}
window()->make_current();
fl_overlay_rect(sx,sy,sw,sh);
return 1;
case FL_RELEASE:
if (button == 1) {
erase_box();
if (dragged && sw > 3 && sh > 3) {
X = X + (sx+sw/2-x()-W/2)*scale/W;
Y = Y + (-sy-sh/2+y()+H/2)*scale/W;
scale = sw*scale/W;
} else if (!dragged) {
scale = 2*scale;
if (julia) {
if (scale >= 4) {
scale = 4;
X = Y = 0;
}
} else {
if (scale >= 2.5) {
scale = 2.5;
X = -.75;
Y = 0;
}
}
} else return 1;
((Drawing_Window*)(user_data()))->update_label();
new_display();
} else if (!julia) {
if (!jbrot.d) {
jbrot.make_window();
jbrot.d->julia = 1;
jbrot.d->X = 0;
jbrot.d->Y = 0;
jbrot.d->scale = 4;
jbrot.update_label();
}
jbrot.d->jX = X + (ix-x()-W/2)*scale/W;
jbrot.d->jY = Y + (H/2-iy+y())*scale/W;
static char s[128];
sprintf(s, "Julia %.7f %.7f",jbrot.d->jX,jbrot.d->jY);
jbrot.window->label(s);
jbrot.window->show();
jbrot.d->new_display();
}
return 1;
}
return 0;
}
void Drawing_Area::new_display() {
drawn = nextline = 0;
set_idle();
}
void Drawing_Area::resize(int XX,int YY,int WW,int HH) {
if (WW != w() || HH != h()) {
W = WW-6;
H = HH-8;
if (buffer) {delete[] buffer; buffer = 0; new_display();}
}
Fl_Box::resize(XX,YY,WW,HH);
}
//
// End of "$Id$".
//