1 // This file is part of Eigen, a lightweight C++ template library 2 // for linear algebra. 3 // 4 // Copyright (C) 2008 Benoit Jacob <jacob.benoit.1@gmail.com> 5 // 6 // This Source Code Form is subject to the terms of the Mozilla 7 // Public License v. 2.0. If a copy of the MPL was not distributed 8 // with this file, You can obtain one at http://mozilla.org/MPL/2.0/. 9 10 #ifndef MANDELBROT_H 11 #define MANDELBROT_H 12 13 #include <Eigen/Core> 14 #include <QtGui/QApplication> 15 #include <QtGui/QWidget> 16 #include <QtCore/QThread> 17 18 class MandelbrotWidget; 19 20 class MandelbrotThread : public QThread 21 { 22 friend class MandelbrotWidget; 23 MandelbrotWidget *widget; 24 long long total_iter; 25 int id, max_iter; 26 bool single_precision; 27 28 public: MandelbrotThread(MandelbrotWidget * w,int i)29 MandelbrotThread(MandelbrotWidget *w, int i) : widget(w), id(i) {} 30 void run(); 31 template<typename Real> void render(int img_width, int img_height); 32 }; 33 34 class MandelbrotWidget : public QWidget 35 { 36 Q_OBJECT 37 38 friend class MandelbrotThread; 39 Eigen::Vector2d center; 40 double xradius; 41 int size; 42 unsigned char *buffer; 43 QPoint lastpos; 44 int draft; 45 MandelbrotThread **threads; 46 int threadcount; 47 48 protected: 49 void resizeEvent(QResizeEvent *); 50 void paintEvent(QPaintEvent *); 51 void mousePressEvent(QMouseEvent *event); 52 void mouseMoveEvent(QMouseEvent *event); 53 54 public: MandelbrotWidget()55 MandelbrotWidget() : QWidget(), center(0,0), xradius(2), 56 size(0), buffer(0), draft(16) 57 { 58 setAutoFillBackground(false); 59 threadcount = QThread::idealThreadCount(); 60 threads = new MandelbrotThread*[threadcount]; 61 for(int th = 0; th < threadcount; th++) threads[th] = new MandelbrotThread(this, th); 62 } ~MandelbrotWidget()63 ~MandelbrotWidget() 64 { 65 if(buffer) delete[]buffer; 66 for(int th = 0; th < threadcount; th++) delete threads[th]; 67 delete[] threads; 68 } 69 }; 70 71 #endif // MANDELBROT_H 72