qgsmodelarrowitem.h

Go to the documentation of this file.

/***************************************************************************
                             qgsmodelarrowitem.h
                             ----------------------------------
    Date                 : March 2020
    Copyright            : (C) 2020 Nyall Dawson
    Email                : nyall dot dawson at gmail dot com
 ***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/
 
#ifndef QGSMODELARROWITEM_H
#define QGSMODELARROWITEM_H
 
#include "qgis.h"
#include "qgis_gui.h"
 
#include <QGraphicsPathItem>
#include <QGraphicsRectItem>
#include <QObject>
 
class QgsModelComponentGraphicItem;
class QgsModelArrowItem;

 

class GUI_EXPORT QgsModelDesignerArrowBadgeItem : public QGraphicsRectItem SIP_SKIP
{
  public:
    QgsModelDesignerArrowBadgeItem( QgsModelArrowItem *link SIP_TRANSFERTHIS );

    void setCenter( const QPointF &center );

    QgsModelArrowItem *arrow();
 
    void paint( QPainter *painter, const QStyleOptionGraphicsItem *option, QWidget *widget = nullptr ) override;

    void setValue( const QVariant &value );

    QVariant value() const;

    static QString textForValue( const QVariant &value );
 
  private:
    void resizeToContents();
 
    static constexpr int FONT_SIZE = 10; // Font size for the feature count text
    static constexpr double BORDER_RADIUS = 5;
    static constexpr double CONTENTS_MARGIN = 0.2;
    QVariant mValue;
};
 

class GUI_EXPORT QgsModelArrowItem : public QObject, public QGraphicsPathItem
{
    Q_OBJECT
  public:
    enum Marker
    {
      Circle,
      ArrowHead,
      NoMarker
    };

    QgsModelArrowItem(
      QgsModelComponentGraphicItem *startItem,
      Qt::Edge startEdge,
      int startIndex,
      bool startIsOutgoing,
      Marker startMarker,
      QgsModelComponentGraphicItem *endItem,
      Qt::Edge endEdge,
      int endIndex,
      bool endIsIncoming,
      Marker endMarker
    );

    QgsModelArrowItem( QgsModelComponentGraphicItem *startItem, Qt::Edge startEdge, int startIndex, Marker startMarker, QgsModelComponentGraphicItem *endItem, Marker endMarker );

    QgsModelArrowItem( QgsModelComponentGraphicItem *startItem, Marker startMarker, QgsModelComponentGraphicItem *endItem, Qt::Edge endEdge, int endIndex, Marker endMarker );

    QgsModelArrowItem( QgsModelComponentGraphicItem *startItem, Marker startMarker, QgsModelComponentGraphicItem *endItem, Marker endMarker );
 
    void paint( QPainter *painter, const QStyleOptionGraphicsItem *option, QWidget *widget = nullptr ) override;

    void setPenStyle( Qt::PenStyle style );

    QgsModelComponentGraphicItem *startItem();

    Qt::Edge startEdge() { return mStartEdge; }

    int startIndex() { return mStartIndex; }

    QgsModelComponentGraphicItem *endItem();

    Qt::Edge endEdge() { return mEndEdge; }

    int endIndex() { return mEndIndex; }

    SIP_SKIP QgsModelDesignerArrowBadgeItem *badgeItem();

    SIP_SKIP void setShowBadge( bool visible );
 
  public slots:

    void updatePath();
 
  private:
    QPointF bezierPointForCurve( const QPointF &point, Qt::Edge edge, bool incoming, bool hasSpecificDirectionalFlow ) const;
 
    void drawArrowHead( QPainter *painter, const QPointF &point, const QPointF &vector );
 
    QgsModelComponentGraphicItem *mStartItem = nullptr;
    Qt::Edge mStartEdge = Qt::LeftEdge;
    int mStartIndex = -1;
    bool mStartIsOutgoing = true;
    Marker mStartMarker;
 
    QgsModelComponentGraphicItem *mEndItem = nullptr;
    Qt::Edge mEndEdge = Qt::LeftEdge;
    int mEndIndex = -1;
    bool mEndIsIncoming = false;
    Marker mEndMarker;
 
    // Optional child badge item. Not created by default.
    QgsModelDesignerArrowBadgeItem *mBadgeItem = nullptr;
 
    QPointF mStartPoint;
    QPointF mEndPoint;
 
    QColor mColor;
};

 
#endif // QGSMODELARROWITEM_H