greenhouse/src/ObjWidget.h

/*
 * ObjWidget.h
 *
 *  Created on: 2011-10-21
 *      Author: matlab
 */

#ifndef OBJWIDGET_H_
#define OBJWIDGET_H_

#include <opencv2/features2d/features2d.hpp>
#include <QtGui/QWidget>

class KeypointItem;
class ImageKptsView;
class QAction;
class QMenu;
class QGraphicsView;
class QGraphicsScene;
class QGraphicsRectItem;
class QGraphicsItem;

class ObjWidget : public QWidget
{
	Q_OBJECT

public:
	ObjWidget(QWidget * parent = 0);
	ObjWidget(int id,
			const std::vector<cv::KeyPoint> & keypoints,
			const cv::Mat & descriptors,
			const IplImage * image,
			const QString & detectorType = "NA",
			const QString & descriptorType = "NA",
			QWidget * parent = 0);
	virtual ~ObjWidget();

	void setId(int id);
	void setData(const std::vector<cv::KeyPoint> & keypoints,
			const cv::Mat & descriptors,
			const IplImage * image);
	void resetKptsColor();
	void setKptColor(unsigned int index, const QColor & color);
	void setGraphicsViewMode(bool on);
	void setAutoScale(bool autoScale);
	void setDeletable(bool deletable);
	void addRect(QGraphicsRectItem * rect);

	const std::vector<cv::KeyPoint> & keypoints() const {return keypoints_;}
	const cv::Mat & descriptors() const {return descriptors_;}
	const QPixmap & image() const {return image_;}
	const IplImage * iplImage() const {return iplImage_;}
	int id() const {return id_;}
	QColor defaultColor() const;
	bool isImageShown() const;
	bool isFeaturesShown() const;
	bool isMirrorView() const;
	//QGraphicsScene * scene() const;
	std::vector<cv::KeyPoint> selectedKeypoints() const;
	const QString & detectorType() const {return detectorType_;}
	const QString & descriptorType() const {return descriptorType_;}
	QList<QGraphicsItem*> selectedItems() const;

	QPixmap getSceneAsPixmap();

	void save(QDataStream & streamPtr) const;
	void load(QDataStream & streamPtr);

protected:
	virtual void paintEvent(QPaintEvent *event);
	virtual void contextMenuEvent(QContextMenuEvent * event);
	virtual void resizeEvent(QResizeEvent* event);

signals:
	void removalTriggered(ObjWidget *);
	void selectionChanged();

private:
	void setupGraphicsView();
	void drawKeypoints(QPainter * painter = 0);
	void setupUi();
	void updateItemsShown();

private:
	std::vector<cv::KeyPoint> keypoints_;
	cv::Mat descriptors_;
	QPixmap image_;
	IplImage * iplImage_;
	QList<KeypointItem*> keypointItems_;
	QGraphicsView * graphicsView_;
	int id_;
	QVector<QColor> kptColors_;
	QString detectorType_;
	QString descriptorType_;
	QList<QGraphicsRectItem*> rectItems_;
	bool graphicsViewInitialized_;

	// menu stuff
	QString _savedFileName;
	QMenu * _menu;
	QAction * _showImage;
	QAction * _showFeatures;
	QAction * _saveImage;
	QAction * _mirrorView;
	QAction * _delete;
	QAction * _graphicsViewMode;
	QAction * _autoScale;
};


#endif /* OBJWIDGET_H_ */