qsmartgraphicsview.h

Go to the documentation of this file.

/*M///////////////////////////////////////////////////////////////////////////////////////
// Written by Qt 5.3
// Original by Shih-Jhong Yu, Revised by KKyang.
// Protected under LPGLv3 license.
//M*/

#ifndef QSMARTGRAPHICSVIEW_H
#define QSMARTGRAPHICSVIEW_H

#include <QApplication>
#include <QClipboard>
#include <QFileInfo>
#include <QGraphicsItem>
#include <QGraphicsPixmapItem>
#include <QGraphicsView>
#include <QList>
#include <QMimeData>
#include <QPushButton>
#include <QWheelEvent>
#include <QMessageBox>
#include <QVector>
#include <vector>

#include <QDebug>
#ifdef HAVE_OPENCV
#include "opencv2/opencv.hpp"
#endif

#ifdef QT_OPENGL_LIB
#include <QGL>
#endif

#ifndef NO_SIDEMENU
#include <QButtonGroup>
#include <QGraphicsOpacityEffect>
#include <QPropertyAnimation>
#include <QStyleOption>
#ifdef QT_SVG_LIB
#include <QtSvg>
#endif
class QSideButtonBar;
class customQGraphicsItem;
#endif

class QSmartGraphicsView : public QGraphicsView
{
    Q_OBJECT
public:
    explicit QSmartGraphicsView(QWidget *parent = 0);
    ~QSmartGraphicsView();
    void initialize(const int _img_num, const int width, const int height, int changeRow = 4); 
    //Set images.
#ifdef HAVE_OPENCV
    void setImage(const cv::Mat &img); 
    void setImage(const std::vector<cv::Mat> &imgs); 
#endif
    void setImagefromQImage(const QImage &qimg); 
    void setImagefromQImage(const std::vector<QImage> &qimgs); 

    bool isInitial() const{return _initial;} 
    int getImgNum(){return img_num;} 
protected:
    void wheelEvent(QWheelEvent *event);
    void mousePressEvent(QMouseEvent *event);
    void mouseDoubleClickEvent(QMouseEvent *event);
    void mouseMoveEvent(QMouseEvent *event);
    void mouseReleaseEvent(QMouseEvent *event);
#ifndef NO_SIDEMENU
    void keyPressEvent(QKeyEvent *event);
#endif
signals:
    void sendItemMouXY(const double x, const double y); 
    void sendMousePress(); 
#ifndef NO_SIDEMENU
    void sendSelectedRegion(const QImage &img); 
#endif
public slots:
    void updateImg();
private slots:
    void on_saveAction_triggered(); 
    void on_copyToClipboardAction_triggered(); 
#ifndef NO_SIDEMENU
    void on_normal_triggered(); 
    void on_selectRegion_triggered(); 
    void on_hideSideBar_triggered(); 
    void on_copySelectedRegionAction_triggered(); 
    void onAnimationFinished(); 
#endif
private:
    QAction *hideSideBar, *saveAction, *copyToClipBoardAction, *copySelectedRegion; 
    QClipboard *clipboard; 
    QGraphicsScene *scene; 
    QVector<QGraphicsPixmapItem*> pix_item_vec; 

    QPoint mou_press; 
    int mou_x; 
    int mou_y; 

    bool _initial = false; 
    int img_num; 

#ifndef NO_SIDEMENU
    QSideButtonBar *sbtn; 
    int mouse_status = 0; 
    QPointF select_start, select_item_bound, select_item_start; 
    bool is_item = false; 
    QRubberBand *rubberBand = 0; 

    void fadeOut(); 
    QGraphicsOpacityEffect* mEffect; 
#endif
};

#ifndef NO_SIDEMENU

struct btnID{
    QPushButton* btn;
    QString ID;
};

class QSideButtonBar : public QWidget
{
    Q_OBJECT
public:
    QSideButtonBar(QWidget *parent = 0) : QWidget(parent){
        this->setStyleSheet("QSideButtonBar{border: 2px solid grey;}");
    }
    QSideButtonBar(int button_num, QWidget *parent = 0) : QWidget(parent){
        this->setStyleSheet("QSideButtonBar{border: 2px solid grey;}");
        btn.resize(button_num);
        for(int i = 0; i < btn.size(); i++)
        {
            btn[i].btn = new QPushButton(this);
            btn[i].btn->setCheckable(true);
            btn[i].btn->setGeometry(0, i * 31 + 0, 30, 30);
        }
    }

    ~QSideButtonBar()
    {
        btn.clear();
    }

    void addButton()
    {
        btnID newb;
        newb.ID = "";
        newb.btn = new QPushButton(this);
        btn.push_back(newb);
    }

    void renameButtonText(QString text, int btn_num)
    {
        if(btn_num >= 0 && btn_num < btn.size())
            btn[btn_num].btn->setText(text);
    }

    void renameButtonID(QString ID, int btn_num)
    {
        if(btn_num >= 0 && btn_num < btn.size())
            btn[btn_num].ID = ID;
    }

    int findButtonID(QString name)
    {
        int i = 0;
        for(i = 0; i < btn.size(); i++)
        {
            if(!btn[i].ID.compare(name))
                return i;
        }

        return -1;
    }

    QPushButton* returnButtonByName(QString name)
    {
        int i = 0;
        for(i = 0; i < btn.size(); i++)
        {
            if(!btn[i].ID.compare(name))
                return btn[i].btn;
        }

        return NULL;
    }

    void setButtonIcon(QString name, const QIcon &icon)
    {
        int i = 0;
        for(i = 0; i < btn.size(); i++)
        {
            if(!btn[i].ID.compare(name))
            {
                btn[i].btn->setIcon(icon);
                return;
            }
        }

        return;
    }

    void setButtonCheckable(QString name, const bool c)
    {
        int i = 0;
        for(i = 0; i < btn.size(); i++)
        {
            if(!btn[i].ID.compare(name))
            {
                btn[i].btn->setCheckable(c);
                return;
            }
        }

        return;
    }

    std::vector<btnID> btn;
private:
    void paintEvent(QPaintEvent *)
    {
        QStyleOption opt;
        opt.init(this);
        QPainter p(this);
        style()->drawPrimitive(QStyle::PE_Widget, &opt, &p, this);
    }
};
#endif


#endif // SMARTGRAPHICSVIEW_H