src/DependencyTree/DependencyTree_Arrow.cxx

   1 // Copyright (C) 2014-2016  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License, or (at your option) any later version.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19
  20 // internal includes
  21 #include "DependencyTree_Arrow.h"
  22 #include "DependencyTree_Object.h"
  23
  24 // GUI includes
  25 #include <SUIT_Session.h>
  26 #include <SUIT_ResourceMgr.h>
  27
  28 // Qt includes
  29 #include <QPainter>
  30
  31 #ifdef _MSC_VER
  32 #define _USE_MATH_DEFINES
  33 #endif
  34
  35 #include <math.h>
  36
  37 const qreal arrowSize = 20;
  38
  39 DependencyTree_Arrow::DependencyTree_Arrow( DependencyTree_Object* theStartItem,
  40                                             DependencyTree_Object* theEndItem,
  41                                             QGraphicsItem* parent, QGraphicsScene* scene )
  42 #if QT_VERSION < QT_VERSION_CHECK(5, 0, 0)
  43 :QGraphicsLineItem( parent, scene ),
  44 #else
  45 :QGraphicsLineItem( parent ),
  46 #endif
  47 myIsBiLink( false ),
  48 myStartItem( theStartItem ),
  49 myEndItem( theEndItem )
  50 {
  51   SUIT_ResourceMgr* resMgr = SUIT_Session::session()->resourceMgr();
  52
  53   myColor = resMgr->colorValue( "Geometry", "dependency_tree_arrow_color", QColor( 0, 0, 130 ) );
  54   myHighlightColor = resMgr->colorValue( "Geometry", "dependency_tree_highlight_arrow_color", QColor( 0, 0, 255 ) );
  55   mySelectColor = resMgr->colorValue( "Geometry", "dependency_tree_select_arrow_color", QColor( 255, 0, 0 ) );
  56
  57   myStartItem = theStartItem;
  58   myEndItem = theEndItem;
  59
  60   myLine = QLineF( myStartItem->pos(), myEndItem->pos() );
  61   myArrowHead  = createArrowHead( myStartItem->pos(), myEndItem->pos() );
  62   myReverseArrowHead = createArrowHead( myEndItem->pos(), myStartItem->pos() );
  63
  64   mySelfDependencyArrow = QRectF();
  65
  66   setPen( QPen( myColor, 2, Qt::SolidLine, Qt::RoundCap, Qt::RoundJoin ) );
  67   setColor( myColor );
  68
  69   setFlag( QGraphicsItem::ItemIsSelectable, true );
  70   setZValue( -1000.0 );
  71 }
  72
  73 DependencyTree_Arrow::~DependencyTree_Arrow()
  74 {
  75 }
  76
  77 //=================================================================================
  78 // function : boundingRect()
  79 // purpose  : return the outer bounds of the item as a rectangle.
  80 //            QGraphicsView uses this to determine whether the item requires redrawing
  81 //=================================================================================
  82 QRectF DependencyTree_Arrow::boundingRect() const
  83 {
  84   qreal extra;
  85   QRectF boundingRect;
  86   if( myStartItem == myEndItem ) {
  87     extra = arrowSize / 2.0 + 2.0;
  88     boundingRect = mySelfDependencyArrow;
  89   }
  90   else {
  91     extra = ( pen().width() + 20 ) / 2.0;
  92     boundingRect = QRectF( myLine.p1(), QSizeF( myLine.p2().x() - myLine.p1().x(),
  93                                                 myLine.p2().y() - myLine.p1().y() ) );
  94   }
  95   return boundingRect.normalized().adjusted( -extra, -extra, extra, extra );
  96 }
  97
  98 //=================================================================================
  99 // function : shape()
 100 // purpose  : return the shape of this item to define an area of preselection
 101 //=================================================================================
 102 QPainterPath DependencyTree_Arrow::shape() const
 103 {
 104   QPainterPath path;
 105
 106   if( myStartItem == myEndItem ) {
 107     qreal extra = 5;
 108     QPolygonF aPolygonBigger( mySelfDependencyArrow.normalized()
 109               .adjusted( -extra, -extra, extra, extra ) );
 110     QPolygonF aPolygonSmaller( mySelfDependencyArrow.normalized()
 111               .adjusted( extra, extra, -extra, -extra ) );
 112     path.addPolygon( aPolygonBigger.subtracted( aPolygonSmaller ) );
 113     path.addPolygon(myArrowHead);
 114   }
 115   else {
 116     QPolygonF aShapePolygon;
 117     QPolygon anArrowHead = myArrowHead.toPolygon();
 118     QPolygon anReverseArrowHead = myReverseArrowHead.toPolygon();
 119     aShapePolygon << anArrowHead.point(1) << anArrowHead.point(0) << anArrowHead.point(2) <<
 120       anReverseArrowHead.point(1) << anReverseArrowHead.point(0) << anReverseArrowHead.point(2);
 121     path.addPolygon( aShapePolygon );
 122   }
 123   return path;
 124  }
 125
 126 //=================================================================================
 127 // function : setColor()
 128 // purpose  : set default color for arrow
 129 //=================================================================================
 130 void DependencyTree_Arrow::setColor( const QColor& theColor )
 131 {
 132   myColor = theColor;
 133 }
 134
 135 //=================================================================================
 136 // function : setHighlightColor()
 137 // purpose  : set color for highlighted arrow
 138 //=================================================================================
 139 void DependencyTree_Arrow::setHighlightColor( const QColor& theColor )
 140 {
 141   myHighlightColor = theColor;
 142 }
 143
 144 //=================================================================================
 145 // function : setSelectColor()
 146 // purpose  : set color for selected arrow
 147 //=================================================================================
 148 void DependencyTree_Arrow::setSelectColor( const QColor& theColor )
 149 {
 150   mySelectColor = theColor;
 151 }
 152
 153 //=================================================================================
 154 // function : getStartItem()
 155 // purpose  : get start item of arrow
 156 //=================================================================================
 157 DependencyTree_Object* DependencyTree_Arrow::getStartItem() const
 158 {
 159   return myStartItem;
 160 }
 161
 162 //=================================================================================
 163 // function : getEndItem()
 164 // purpose  : get end item of arrow
 165 //=================================================================================
 166 DependencyTree_Object* DependencyTree_Arrow::getEndItem() const
 167 {
 168   return myEndItem;
 169 }
 170
 171 //=================================================================================
 172 // function : setIsBiLink()
 173 // purpose  : set true if current arrow is bi-directional link, else set false
 174 //=================================================================================
 175 void DependencyTree_Arrow::setIsBiLink( bool theIsBiLink )
 176 {
 177   myIsBiLink = theIsBiLink;
 178 }
 179
 180 //=================================================================================
 181 // function : paint()
 182 // purpose  : paint the contents of an item in local coordinates (called by QGraphicsView)
 183 //=================================================================================
 184 void DependencyTree_Arrow::paint( QPainter* painter, const QStyleOptionGraphicsItem*, QWidget* )
 185 {
 186   if( isSelected() ) {
 187     painter->setBrush( mySelectColor );
 188     painter->setPen( QPen( mySelectColor, 2, Qt::SolidLine, Qt::RoundCap, Qt::RoundJoin ) );
 189   }
 190   else if( isUnderMouse() ) {
 191     painter->setBrush( myHighlightColor );
 192     painter->setPen( QPen( myHighlightColor, 3, Qt::SolidLine, Qt::RoundCap, Qt::RoundJoin ) );
 193   }
 194   else {
 195     painter->setBrush( myColor );
 196     painter->setPen( QPen( myColor, 2, Qt::SolidLine, Qt::RoundCap, Qt::RoundJoin ) );
 197   }
 198
 199   if( myStartItem == myEndItem ) {
 200     int lineSize = 60;
 201     QPointF p1( myStartItem->pos().x() - myStartItem->boundingRect().width()/2,
 202                 myStartItem->pos().y() );
 203     QPointF p2( p1.x() - lineSize + myStartItem->boundingRect().height()/2, p1.y() );
 204     QPointF p3( p2.x(), p2.y() - lineSize );
 205     QPointF p4( p3.x() + lineSize, p3.y() );
 206     QPointF p5( p4.x(), p4.y() + lineSize - myStartItem->boundingRect().height()/2 );
 207     mySelfDependencyArrow = QRectF( p3.x(), p3.y(), lineSize, lineSize );
 208     myArrowHead = createArrowHead( p4, p5, false );
 209     QVector<QPointF> pointVector;
 210     pointVector << p1 << p2 << p2 << p3 << p3 << p4 << p4 << p5;
 211
 212     painter->drawLines( pointVector);
 213     painter->drawPolygon(myArrowHead);
 214   }
 215   else {
 216     if (myStartItem->collidesWithItem(myEndItem))
 217       return;
 218
 219     myArrowHead  = createArrowHead( myStartItem->pos(), myEndItem->pos() );
 220     myReverseArrowHead = createArrowHead( myEndItem->pos(), myStartItem->pos() );
 221
 222     painter->drawLine( myLine );
 223     painter->drawPolygon( myArrowHead );
 224     if( myIsBiLink )
 225       painter->drawPolygon( myReverseArrowHead );
 226   }
 227 }
 228
 229 //=================================================================================
 230 // function : createArrowHead()
 231 // purpose  : create a head of arrow from start point to end point
 232 //=================================================================================
 233 QPolygonF DependencyTree_Arrow::createArrowHead( QPointF theStartPoint, QPointF theEndPoint,
 234                                                          bool theIsDynamic )
 235 {
 236   if( theIsDynamic ) {
 237     QLineF centerLine( theStartPoint, theEndPoint );
 238     QPolygonF endPolygon = QPolygonF( myEndItem->boundingRect() );
 239     QPointF p1 = endPolygon.first() + theEndPoint;
 240     QPointF p2;
 241     QPointF intersectPoint;
 242     QLineF polyLine;
 243     for( int i = 1; i < endPolygon.count(); ++i ) {
 244       p2 = endPolygon.at(i) + theEndPoint;
 245       polyLine = QLineF( p1, p2 );
 246       QLineF::IntersectType intersectType = polyLine.intersect( centerLine, &intersectPoint );
 247       if( intersectType == QLineF::BoundedIntersection )
 248         break;
 249       p1 = p2;
 250     }
 251     myLine = QLineF( intersectPoint, theStartPoint );
 252   }
 253   else
 254     myLine = QLineF( theEndPoint, theStartPoint );
 255
 256   double angle = acos(myLine.dx() / myLine.length());
 257   if( myLine.dy() >= 0 )
 258     angle = ( M_PI * 2 ) - angle;
 259
 260   QPointF arrowP1 = myLine.p1() + QPointF( sin( angle + M_PI / 3 ) * arrowSize,
 261                                            cos( angle + M_PI / 3 ) * arrowSize );
 262   QPointF arrowP2 = myLine.p1() + QPointF( sin( angle + M_PI - M_PI / 3 ) * arrowSize,
 263                                            cos( angle + M_PI - M_PI / 3 ) * arrowSize );
 264
 265   QPolygonF anArrowHead;
 266   anArrowHead << myLine.p1() << arrowP1 << arrowP2;
 267   return anArrowHead;
 268 }