Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
.
[graphlib.git] / DrawingWindow.cpp
index 6c49e12..4e34d9e 100644 (file)
 #include "DrawingWindow.h"
-#include <QPainter>
+#include <QPaintEvent>
+#include <QThread>
+#include <QTimerEvent>
 
 #include <iostream>
 
-DrawingWindow::DrawingWindow(const DrawingArea &a)
+class DrawingWindow::DrawingThread: public QThread {
+public:
+    DrawingThread(DrawingWindow &w, ThreadFunction f)
+        : drawingWindow(w)
+        , threadFunction(f)
+    {
+    }
+
+    void run()
+    {
+        threadFunction(drawingWindow);
+    }
+
+private:
+    DrawingWindow &drawingWindow;
+    ThreadFunction threadFunction;
+
+};
+
+DrawingWindow::DrawingWindow(ThreadFunction fun, int width, int height)
     : QWidget()
-    , drawingArea(a)
 {
-    initialize();
+    initialize(fun, width, height);
 }
 
-DrawingWindow::DrawingWindow(QWidget *parent, const DrawingArea &a)
+DrawingWindow::DrawingWindow(QWidget *parent,
+                             ThreadFunction fun, int width, int height)
     : QWidget(parent)
-    , drawingArea(a)
 {
-    initialize();
+    initialize(fun, width, height);
 }
 
 DrawingWindow::DrawingWindow(QWidget *parent, Qt::WindowFlags flags,
-                             const DrawingArea &a)
+                             ThreadFunction fun, int width, int height)
     : QWidget(parent, flags)
-    , drawingArea(a)
 {
-    initialize();
+    initialize(fun, width, height);
 }
 
-void DrawingWindow::paintEvent(QPaintEvent *)
+void DrawingWindow::initialize(ThreadFunction fun, int width, int height)
 {
-    std::cerr << "paint!\n";
-    QPainter painter(this);
-    painter.drawImage(0, 0, drawingArea.getImage());
-}
+    image = new QImage(width, height, QImage::Format_RGB32);
+    image->fill(QColor(Qt::white).rgb());
+
+    painter = new QPainter(image);
+
+    dirtyFlag = false;
 
-void DrawingWindow::initialize()
-{
     setFocusPolicy(Qt::StrongFocus);
-    setFixedSize(drawingArea.getImage().size());
+    setFixedSize(image->size());
     setAttribute(Qt::WA_OpaquePaintEvent);
-    connect(&drawingArea, SIGNAL(update()), this, SLOT(update()));
+    setFocus();
+    timer.start(paintInterval, this);
+
+    thread = new DrawingThread(*this, fun);
+    thread_started = false;
+
+    mutex_enabled = true;
+}
+
+DrawingWindow::~DrawingWindow()
+{
+    mutex.lock();
+    mutex_enabled = false;
+    mutex.unlock();
+    std::cerr << "A\n";
+    thread->terminate();    
+    std::cerr << "B\n";
+    thread->wait();
+    std::cerr << "C\n";
+    delete thread;
+    delete painter;
+    delete image;
+    std::cerr << "D\n";
+}
+
+void DrawingWindow::setColor(const QColor &color)
+{
+    QPen pen(painter->pen());
+    pen.setColor(color);
+    painter->setPen(pen);
+}
+
+void DrawingWindow::setColor(float red, float green, float blue)
+{
+    QColor color;
+    color.setRgbF(red, green, blue);
+    this->setColor(color);
+}
+
+void DrawingWindow::drawPoint(int x, int y)
+{
+    lock();
+    painter->drawPoint(x, y);
+    setDirtyRect(x, y);
+    unlock();
+}
+
+void DrawingWindow::drawLine(int x1, int y1, int x2, int y2)
+{
+    lock();
+    painter->drawLine(x1, y1, x2, y2);
+    setDirtyRect(x1, y1, x2, y2);
+    unlock();
+}
+
+void DrawingWindow::paintEvent(QPaintEvent *ev)
+{
+    QPainter widgetPainter(this);
+    QRect rect = ev->rect();
+    lock();
+    QImage imageCopy(*image);
+    unlock();
+    widgetPainter.drawImage(rect, imageCopy, rect);
+}
+
+void DrawingWindow::showEvent(QShowEvent *ev)
+{
+    if (!thread_started) {
+        thread->start();
+        thread_started = true;
+    }
+    QWidget::showEvent(ev);
+}
 
+void DrawingWindow::timerEvent(QTimerEvent *ev)
+{
+    if (ev->timerId() == timer.timerId()) {
+        lock();
+        if (dirtyFlag) {
+            update(dirtyRect);
+            dirtyFlag = false;
+        }
+        unlock();
+        timer.start(paintInterval, this);
+    } else {
+        QWidget::timerEvent(ev);
+    }
+}
+
+void DrawingWindow::setDirtyRect(const QRect &rect)
+{
+    if (dirtyFlag) {
+        dirtyRect |= rect;
+    } else {
+        dirtyFlag = true;
+        dirtyRect = rect;
+    }
 }