view2D.h

/*
 * This file is part of Copiste.
 *
 * Copiste 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.
 *
 *  Copiste 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 Copiste.  If not, see <http://www.gnu.org/licenses/>.
 */


#ifndef INCLUDED_VIEW2DH
#define INCLUDED_VIEW2DH

class View2D;

#include "algo/neuralnetwork.h"
#include "algo/corpus.h"
#include "gui/viewport.h"

#include <QWidget>
#include <QPainter>
#include <QPaintEvent>
#include <QTime>

#include <iostream>
#include <stack>

const int VIEW_OUTPUT_WIDTH = 700;
const int VIEW_OUTPUT_HEIGHT = 500;
const char POINT_0_PATH[16] = "img/point-0.png";
const char POINT_1_PATH[16] = "img/point-1.png";
const float CORPUS_EPSILON = 0.0001;
const int MAX_WIDTH_HISTORY = 2000;
const int TIME_BETWEEN_TWO_LABEL_UPDATES = 500; // in milliseconds

void plotHistory(double* history, int size, int corpusSize);

class View2D : public QWidget
{
    Q_OBJECT
    public:
        View2D(QWidget *parent = 0);

        void renderToImage(std::string fileName, std::string format = "PNG",
                int w = VIEW_OUTPUT_WIDTH, int h = VIEW_OUTPUT_HEIGHT);

        void setCorpus(Corpus *corpus);
        void setNet(NeuralNetwork *net);
        void updateQuadtree();

        void renderScene();

        Corpus* corpus();
        NeuralNetwork* net();

        void handleKeyReleaseEvent(QKeyEvent *event);

        void setCurrentPoint(bool point);

    signals:
        void rendering();
        void rendered();
        void hoveringElement(std::string);

    protected:
        void paintEvent(QPaintEvent *event);

        void mouseMoveEvent(QMouseEvent *event);
        void mousePressEvent(QMouseEvent *event);
        void mouseReleaseEvent(QMouseEvent *event);
        void keyReleaseEvent(QKeyEvent *event);

        void emitHoveringElement(std::string name);

    private:
        Corpus *mCorpus;
        NeuralNetwork *mNet;

        QPixmap mPoint0;
        QPixmap mPoint1;

        // Viewport managing
        Viewport mViewport;

        QTime mLastUpdate;

        // Dynamic zoom
        bool mZooming;
        int mStartX;
        int mStartY;
        int mCurrentX;
        int mCurrentY;

        // Corpus editing
        bool mCurrentPoint;

        // Zoom history
        std::stack<Viewport> mZoomHistory;
};

#endif