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 | |