1 // Copyright (C) 2014-2016 CEA/DEN, EDF R&D, OPEN CASCADE
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.
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.
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
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 #include "DependencyTree_Arrow.h"
22 #include "DependencyTree_Object.h"
25 #include <SUIT_Session.h>
26 #include <SUIT_ResourceMgr.h>
32 #define _USE_MATH_DEFINES
37 const qreal arrowSize = 20;
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 ),
45 :QGraphicsLineItem( parent ),
48 myStartItem( theStartItem ),
49 myEndItem( theEndItem )
51 SUIT_ResourceMgr* resMgr = SUIT_Session::session()->resourceMgr();
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 ) );
57 myStartItem = theStartItem;
58 myEndItem = theEndItem;
60 myLine = QLineF( myStartItem->pos(), myEndItem->pos() );
61 myArrowHead = createArrowHead( myStartItem->pos(), myEndItem->pos() );
62 myReverseArrowHead = createArrowHead( myEndItem->pos(), myStartItem->pos() );
64 mySelfDependencyArrow = QRectF();
66 setPen( QPen( myColor, 2, Qt::SolidLine, Qt::RoundCap, Qt::RoundJoin ) );
69 setFlag( QGraphicsItem::ItemIsSelectable, true );
73 DependencyTree_Arrow::~DependencyTree_Arrow()
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
86 if( myStartItem == myEndItem ) {
87 extra = arrowSize / 2.0 + 2.0;
88 boundingRect = mySelfDependencyArrow;
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() ) );
95 return boundingRect.normalized().adjusted( -extra, -extra, extra, extra );
98 //=================================================================================
100 // purpose : return the shape of this item to define an area of preselection
101 //=================================================================================
102 QPainterPath DependencyTree_Arrow::shape() const
106 if( myStartItem == myEndItem ) {
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);
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 );
126 //=================================================================================
127 // function : setColor()
128 // purpose : set default color for arrow
129 //=================================================================================
130 void DependencyTree_Arrow::setColor( const QColor& theColor )
135 //=================================================================================
136 // function : setHighlightColor()
137 // purpose : set color for highlighted arrow
138 //=================================================================================
139 void DependencyTree_Arrow::setHighlightColor( const QColor& theColor )
141 myHighlightColor = theColor;
144 //=================================================================================
145 // function : setSelectColor()
146 // purpose : set color for selected arrow
147 //=================================================================================
148 void DependencyTree_Arrow::setSelectColor( const QColor& theColor )
150 mySelectColor = theColor;
153 //=================================================================================
154 // function : getStartItem()
155 // purpose : get start item of arrow
156 //=================================================================================
157 DependencyTree_Object* DependencyTree_Arrow::getStartItem() const
162 //=================================================================================
163 // function : getEndItem()
164 // purpose : get end item of arrow
165 //=================================================================================
166 DependencyTree_Object* DependencyTree_Arrow::getEndItem() const
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 )
177 myIsBiLink = theIsBiLink;
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* )
187 painter->setBrush( mySelectColor );
188 painter->setPen( QPen( mySelectColor, 2, Qt::SolidLine, Qt::RoundCap, Qt::RoundJoin ) );
190 else if( isUnderMouse() ) {
191 painter->setBrush( myHighlightColor );
192 painter->setPen( QPen( myHighlightColor, 3, Qt::SolidLine, Qt::RoundCap, Qt::RoundJoin ) );
195 painter->setBrush( myColor );
196 painter->setPen( QPen( myColor, 2, Qt::SolidLine, Qt::RoundCap, Qt::RoundJoin ) );
199 if( myStartItem == myEndItem ) {
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;
212 painter->drawLines( pointVector);
213 painter->drawPolygon(myArrowHead);
216 if (myStartItem->collidesWithItem(myEndItem))
219 myArrowHead = createArrowHead( myStartItem->pos(), myEndItem->pos() );
220 myReverseArrowHead = createArrowHead( myEndItem->pos(), myStartItem->pos() );
222 painter->drawLine( myLine );
223 painter->drawPolygon( myArrowHead );
225 painter->drawPolygon( myReverseArrowHead );
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,
237 QLineF centerLine( theStartPoint, theEndPoint );
238 QPolygonF endPolygon = QPolygonF( myEndItem->boundingRect() );
239 QPointF p1 = endPolygon.first() + theEndPoint;
241 QPointF intersectPoint;
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 )
251 myLine = QLineF( intersectPoint, theStartPoint );
254 myLine = QLineF( theEndPoint, theStartPoint );
256 double angle = acos(myLine.dx() / myLine.length());
257 if( myLine.dy() >= 0 )
258 angle = ( M_PI * 2 ) - angle;
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 );
265 QPolygonF anArrowHead;
266 anArrowHead << myLine.p1() << arrowP1 << arrowP2;