edge_widget.h

#ifndef GUI_EDGE_WIDGET_H_
#define GUI_EDGE_WIDGET_H_
#include <QGraphicsItem>
#include <QGraphicsSceneHoverEvent>
#include <QWidget>
#include <QObject>
#include <QDebug>
#include <QToolTip>
#include <iostream>
#include <qcolor.h>


#include "math.h"
#include "QtCore/qmath.h"
#include "node_widget.h"

class NodeWidget;

class EdgeWidget : public QGraphicsItem {
  public:
    EdgeWidget(NodeWidget* source_node, 
               NodeWidget* destination_node, 
               QColor color = Qt::black, 
               QColor alt_color = Qt::black, 
               Qt::PenStyle pen_style = Qt::SolidLine,
               int opacity = 255, 
               QGraphicsItem* parent = 0);
    
    void set_source_node(NodeWidget* source_node);
    void set_destination_node(NodeWidget* destination_node);
    NodeWidget* source_node() const;
    NodeWidget* destination_node() const;
    
    void set_intensity(bool intensity);
    bool high_intensity() const;

    // Main color of the node
    void set_color(QColor);
    QColor color() const;

    // Alternate color of the node, can be used to specify a trigger on the node
    void set_alternate_color(QColor);
    QColor alternate_color() const;

    // Opacity, between 0..255 
    void set_opacity(int);
    int opacity() const;

    // @see Qt::PenStyle
    void set_pen_style(Qt::PenStyle pen);
    Qt::PenStyle pen_style();
    
    QPainterPath shape() const;
    
    void adjust();
    
    void hoverEnterEvent(QGraphicsSceneHoverEvent* event);
    void hoverLeaveEvent(QGraphicsSceneHoverEvent* event);
  
    void use_alternate_color(bool condition);

  protected:
    void paint ( QPainter * painter, const QStyleOptionGraphicsItem * option, QWidget * widget = 0 );
    
    QRectF boundingRect() const;
    
    
  private:
    NodeWidget *source_node_, *destination_node_;
    
    bool high_intensity_;
    
    QPointF source_point_;
    QPointF destination_point_;
    
    qreal arrowSize;

    bool use_alternate_color_;

    QColor color_;

    QColor alternate_color_;

    int opacity_;
  
    Qt::PenStyle pen_style_;
};

#endif