| 1 | /**************************************************************************** | 
| 2 | ** | 
| 3 | ** Copyright (C) 2016 The Qt Company Ltd. | 
| 4 | ** Contact: https://www.qt.io/licensing/ | 
| 5 | ** | 
| 6 | ** This file is part of the examples of the Qt Toolkit. | 
| 7 | ** | 
| 8 | ** $QT_BEGIN_LICENSE:BSD$ | 
| 9 | ** Commercial License Usage | 
| 10 | ** Licensees holding valid commercial Qt licenses may use this file in | 
| 11 | ** accordance with the commercial license agreement provided with the | 
| 12 | ** Software or, alternatively, in accordance with the terms contained in | 
| 13 | ** a written agreement between you and The Qt Company. For licensing terms | 
| 14 | ** and conditions see https://www.qt.io/terms-conditions. For further | 
| 15 | ** information use the contact form at https://www.qt.io/contact-us. | 
| 16 | ** | 
| 17 | ** BSD License Usage | 
| 18 | ** Alternatively, you may use this file under the terms of the BSD license | 
| 19 | ** as follows: | 
| 20 | ** | 
| 21 | ** "Redistribution and use in source and binary forms, with or without | 
| 22 | ** modification, are permitted provided that the following conditions are | 
| 23 | ** met: | 
| 24 | **   * Redistributions of source code must retain the above copyright | 
| 25 | **     notice, this list of conditions and the following disclaimer. | 
| 26 | **   * Redistributions in binary form must reproduce the above copyright | 
| 27 | **     notice, this list of conditions and the following disclaimer in | 
| 28 | **     the documentation and/or other materials provided with the | 
| 29 | **     distribution. | 
| 30 | **   * Neither the name of The Qt Company Ltd nor the names of its | 
| 31 | **     contributors may be used to endorse or promote products derived | 
| 32 | **     from this software without specific prior written permission. | 
| 33 | ** | 
| 34 | ** | 
| 35 | ** THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS | 
| 36 | ** "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT | 
| 37 | ** LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR | 
| 38 | ** A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT | 
| 39 | ** OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, | 
| 40 | ** SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT | 
| 41 | ** LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, | 
| 42 | ** DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY | 
| 43 | ** THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | 
| 44 | ** (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | 
| 45 | ** OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE." | 
| 46 | ** | 
| 47 | ** $QT_END_LICENSE$ | 
| 48 | ** | 
| 49 | ****************************************************************************/ | 
| 50 |  | 
| 51 | #include "edge.h" | 
| 52 | #include "node.h" | 
| 53 |  | 
| 54 | #include <QPainter> | 
| 55 | #include <QtMath> | 
| 56 |  | 
| 57 | //! [0] | 
| 58 | Edge::Edge(Node *sourceNode, Node *destNode) | 
| 59 |     : source(sourceNode), dest(destNode) | 
| 60 | { | 
| 61 |     setAcceptedMouseButtons(Qt::NoButton); | 
| 62 |     source->addEdge(edge: this); | 
| 63 |     dest->addEdge(edge: this); | 
| 64 |     adjust(); | 
| 65 | } | 
| 66 | //! [0] | 
| 67 |  | 
| 68 | //! [1] | 
| 69 | Node *Edge::sourceNode() const | 
| 70 | { | 
| 71 |     return source; | 
| 72 | } | 
| 73 |  | 
| 74 | Node *Edge::destNode() const | 
| 75 | { | 
| 76 |     return dest; | 
| 77 | } | 
| 78 | //! [1] | 
| 79 |  | 
| 80 | //! [2] | 
| 81 | void Edge::adjust() | 
| 82 | { | 
| 83 |     if (!source || !dest) | 
| 84 |         return; | 
| 85 |  | 
| 86 |     QLineF line(mapFromItem(item: source, ax: 0, ay: 0), mapFromItem(item: dest, ax: 0, ay: 0)); | 
| 87 |     qreal length = line.length(); | 
| 88 |  | 
| 89 |     prepareGeometryChange(); | 
| 90 |  | 
| 91 |     if (length > qreal(20.)) { | 
| 92 |         QPointF edgeOffset((line.dx() * 10) / length, (line.dy() * 10) / length); | 
| 93 |         sourcePoint = line.p1() + edgeOffset; | 
| 94 |         destPoint = line.p2() - edgeOffset; | 
| 95 |     } else { | 
| 96 |         sourcePoint = destPoint = line.p1(); | 
| 97 |     } | 
| 98 | } | 
| 99 | //! [2] | 
| 100 |  | 
| 101 | //! [3] | 
| 102 | QRectF Edge::boundingRect() const | 
| 103 | { | 
| 104 |     if (!source || !dest) | 
| 105 |         return QRectF(); | 
| 106 |  | 
| 107 |     qreal penWidth = 1; | 
| 108 |     qreal  = (penWidth + arrowSize) / 2.0; | 
| 109 |  | 
| 110 |     return QRectF(sourcePoint, QSizeF(destPoint.x() - sourcePoint.x(), | 
| 111 |                                       destPoint.y() - sourcePoint.y())) | 
| 112 |         .normalized() | 
| 113 |         .adjusted(xp1: -extra, yp1: -extra, xp2: extra, yp2: extra); | 
| 114 | } | 
| 115 | //! [3] | 
| 116 |  | 
| 117 | //! [4] | 
| 118 | void Edge::paint(QPainter *painter, const QStyleOptionGraphicsItem *, QWidget *) | 
| 119 | { | 
| 120 |     if (!source || !dest) | 
| 121 |         return; | 
| 122 |  | 
| 123 |     QLineF line(sourcePoint, destPoint); | 
| 124 |     if (qFuzzyCompare(p1: line.length(), p2: qreal(0.))) | 
| 125 |         return; | 
| 126 | //! [4] | 
| 127 |  | 
| 128 | //! [5] | 
| 129 |     // Draw the line itself | 
| 130 |     painter->setPen(QPen(Qt::black, 1, Qt::SolidLine, Qt::RoundCap, Qt::RoundJoin)); | 
| 131 |     painter->drawLine(l: line); | 
| 132 | //! [5] | 
| 133 |  | 
| 134 | //! [6] | 
| 135 |     // Draw the arrows | 
| 136 |     double angle = std::atan2(y: -line.dy(), x: line.dx()); | 
| 137 |  | 
| 138 |     QPointF sourceArrowP1 = sourcePoint + QPointF(sin(x: angle + M_PI / 3) * arrowSize, | 
| 139 |                                                   cos(x: angle + M_PI / 3) * arrowSize); | 
| 140 |     QPointF sourceArrowP2 = sourcePoint + QPointF(sin(x: angle + M_PI - M_PI / 3) * arrowSize, | 
| 141 |                                                   cos(x: angle + M_PI - M_PI / 3) * arrowSize); | 
| 142 |     QPointF destArrowP1 = destPoint + QPointF(sin(x: angle - M_PI / 3) * arrowSize, | 
| 143 |                                               cos(x: angle - M_PI / 3) * arrowSize); | 
| 144 |     QPointF destArrowP2 = destPoint + QPointF(sin(x: angle - M_PI + M_PI / 3) * arrowSize, | 
| 145 |                                               cos(x: angle - M_PI + M_PI / 3) * arrowSize); | 
| 146 |  | 
| 147 |     painter->setBrush(Qt::black); | 
| 148 |     painter->drawPolygon(polygon: QPolygonF() << line.p1() << sourceArrowP1 << sourceArrowP2); | 
| 149 |     painter->drawPolygon(polygon: QPolygonF() << line.p2() << destArrowP1 << destArrowP2); | 
| 150 | } | 
| 151 | //! [6] | 
| 152 |  |