node_widget.h

#ifndef GUI_NODE_WIDGET_H_
#define GUI_NODE_WIDGET_H_

#include <QGraphicsEllipseItem>
#include <QPainter>
#include <QDebug>
#include <QWidget>
#include <QGraphicsItem>
#include <iostream>
#include <QGraphicsScene>
#include <QObject>
#include <QStyleOptionGraphicsItem>
#include <set>
#include "edge_widget.h"

class EdgeWidget;

struct CompareEdgeByNodes : public std::binary_function<EdgeWidget*, EdgeWidget*, bool> {
    bool operator()(EdgeWidget* a, EdgeWidget* b) const;
};

class NodeWidget : public QGraphicsItem  {

    public:

        NodeWidget(const QString& node_text, QGraphicsItem * parent = 0);

        ~NodeWidget() {};

        void set_marked(bool marked);

        bool is_marked();

    void set_color(QColor);
  
    void set_opacity(int);
  
    QColor color();

    int opacity();

        QPainterPath shape() const;

        QRectF boundingRect() const;

        void paint ( QPainter * painter, const QStyleOptionGraphicsItem * option, QWidget * widget = 0 );

        void add_edge(EdgeWidget* w);

        QString node_text();

        qreal height();

        qreal width();  
    
        void unmark_path();

        void mark_path();

        QModelIndex get_index_in_model();

        void set_index_in_model(QModelIndex);

    protected:
        QVariant itemChange(GraphicsItemChange change, const QVariant &value);  

        void mouseDoubleClickEvent ( QGraphicsSceneMouseEvent * event );

        void mousePressEvent ( QGraphicsSceneMouseEvent * event );

    private:

        void force_update(NodeWidget*, bool);

        void force_update(EdgeWidget*, bool);
    
        QString node_text_; 
    QColor color_;
    int opacity_;
        qreal height_;
        qreal width_;
        qreal pen_width_;
        bool is_marked_;
        QModelIndex index_;

        typedef std::set<EdgeWidget*, CompareEdgeByNodes> edge_list_;
        edge_list_ edges;
};

#endif