ImageView.h

Go to the documentation of this file.

#ifndef IMAGE_VIEW_H
#define IMAGE_VIEW_H

#include <QtGui>
#include <QtCore>

#include <base/samples/Frame.hpp>
#include <base/samples/DistanceImage.hpp>
#include <frame_helper/FrameQImageConverter.h>
#include "rock_widget_collection/progress_indicator/ProgressIndicator.h"

#include "GraphicsPointsItem.h"

//TODO NOT ALL OBJECTS GET DELETED AFTER DECONSTRUCTION

class ImageView : public QWidget
{
    Q_OBJECT
    Q_CLASSINFO("Author", "Allan Conquest") 
    
    
    Q_PROPERTY(QColor backgroundColor READ getBackgroundColor WRITE setBackgroundColor)
    
    Q_PROPERTY(bool use_progress_indicator READ useProgressIndicator WRITE setUseProgressIndicator)
    
    Q_PROPERTY(int progress_indicator_timeout READ getProgressIndicatorTimeout WRITE setProgressIndicatorTimeout)
    
    Q_PROPERTY(bool use_smooth_transformation READ useSmoothTransformation WRITE setSmoothTransformation)

    Q_PROPERTY(bool rgb_swapped READ getRgbSwapped WRITE setRgbSwapped)
    Q_PROPERTY(bool invert_color READ getInvertColor WRITE setInvertColor)
    
#ifdef USE_GST
    /* Properties for use with GStreamer. That means in particluar that any of 
     * the following properties is unimportant if useGst is set to false. */

    Q_PROPERTY(QString pipelineDescription READ getPipelineDescription WRITE setPipelineDescription)
    
    
    Q_PROPERTY(bool useGl READ getUseGl WRITE setUseGl)
#endif
    
public:
    ImageView(QWidget *parent = 0);
    virtual ~ImageView();
    
    enum TextLocation
    {
        TOPLEFT,
        TOPRIGHT,
        BOTTOMLEFT,
        BOTTOMRIGHT
    };
    Q_ENUMS(TextLocation)
    
    QSize sizeHint() const;
    
public slots:
    const QColor& getBackgroundColor() const;
    void setBackgroundColor(const QColor & color);
    bool useProgressIndicator();
    void setUseProgressIndicator(bool use);
    bool useSmoothTransformation();
    void setSmoothTransformation(bool smooth);
    const int getProgressIndicatorTimeout() const;
    void setProgressIndicatorTimeout(int timeout);

    void setInvertColor(bool invert);
    bool getInvertColor(){return invert_color;};
    void setRgbSwapped(bool swapped);
    bool getRgbSwapped(){return rgb_swapped;};
    
#ifdef USE_GST
    const QString getPipelineDescription() const;
    void setPipelineDescription(QString descr);
    bool getUseGl();
    void setUseGl(bool use_gl);
#endif
    
    /* Overlays */
    
    void addCircle(QPointF &center, double radius, QColor &color, int width, bool persistent = 0);
    
    void addLine(QLineF &line, QColor &color, int width, bool persistent = 0);
    
    void addPolygon(QPolygonF &polygon, QColor &color, int width, bool persistent = 0);
    
    void addPoints(const QList<int> points_x,QList<int> points_y, QColor &color, int width, bool persistent=0);

    void addText(QString text, /*TextLocation*/ int location, QColor color = QColor(Qt::black), bool persistent = 0);
    
    void clearOverlays(bool clear_persistent_items = 0);
    
    void rotate(int deg);
    
    void saveImage(QString path = QString(), bool overlay = 0);
    
    /* Other slots */
    
    void setFrame(const base::samples::DistanceImage &frame);
    void setFrame(const base::samples::frame::Frame &frame);
    void setImage(const QImage &image);
    void setRawImage(const QString &mode, int pixel_size, int width, int height,const char* pbuffer, const int size);
    void refresh();
    void processImage();
    
    void update2();

    int getHeight() const;

    int getWidth() const;


    
signals:
    
    void clickedImage(const QPoint &point);

protected:
    void resizeEvent(QResizeEvent *event);
    void contextMenuEvent ( QContextMenuEvent * event );
    bool eventFilter(QObject *obj, QEvent *event);

private slots:
    void displayContextMenu(QPoint screenPos);
    void rotate_clockwise();
    void save_image();
    void save_image_overlay();
    void startProgressIndicator();
    
private:
    void addDrawItem(QGraphicsScene *scene, QGraphicsItem *item, bool persistent = 0);
    void setItemPositions();
    void setupContextMenu();
    
    QColor bgColor;
    bool use_progress_indicator;
    bool use_smooth_transformation;
    int progress_indicator_timeout;
    bool rgb_swapped;
    bool invert_color;
    QString last_path;
#ifdef USE_GST
    QString pipelineDescription;
    bool use_gl;
#endif
    
    QGraphicsView *imageView;
    QGraphicsProxyWidget *imageViewProxy;
    QGraphicsScene *imageScene;
    QGraphicsView *fixedOverlayView;
    QGraphicsScene *fixedOverlayScene;
    QGraphicsWidget *overlayWidget;
    QGraphicsGridLayout *overlay_grid;
    QMap<int, QLabel*> overlayMap;
    QGraphicsPixmapItem *imageItem;
    QImage image;

    QList<QGraphicsItem*> persistentDrawItems;
    QList<QGraphicsItem*> volatileDrawItems;

    QSize imageSize;

    frame_helper::FrameQImageConverter frame_converter;
    
    ProgressIndicator progress_indicator;
    QTimer progress_indicator_timer;
    
    /* Menus and actions */
    QMenu context_menu;
    QAction rotate_image_clockwise_act;
    QAction save_image_act;
    QAction save_image_overlay_act;
    QAction activate_progress_indicator_act;
    QAction rgb_swapped_act;
    QAction invert_color_act;
};

#endif /* IMAGE_VIEW_H */