1 | /****************************************************************************
|
---|
2 | **
|
---|
3 | ** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies).
|
---|
4 | ** Contact: Qt Software Information ([email protected])
|
---|
5 | **
|
---|
6 | ** This file is part of the examples of the Qt Toolkit.
|
---|
7 | **
|
---|
8 | ** $QT_BEGIN_LICENSE:LGPL$
|
---|
9 | ** Commercial Usage
|
---|
10 | ** Licensees holding valid Qt Commercial licenses may use this file in
|
---|
11 | ** accordance with the Qt Commercial License Agreement provided with the
|
---|
12 | ** Software or, alternatively, in accordance with the terms contained in
|
---|
13 | ** a written agreement between you and Nokia.
|
---|
14 | **
|
---|
15 | ** GNU Lesser General Public License Usage
|
---|
16 | ** Alternatively, this file may be used under the terms of the GNU Lesser
|
---|
17 | ** General Public License version 2.1 as published by the Free Software
|
---|
18 | ** Foundation and appearing in the file LICENSE.LGPL included in the
|
---|
19 | ** packaging of this file. Please review the following information to
|
---|
20 | ** ensure the GNU Lesser General Public License version 2.1 requirements
|
---|
21 | ** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
|
---|
22 | **
|
---|
23 | ** In addition, as a special exception, Nokia gives you certain
|
---|
24 | ** additional rights. These rights are described in the Nokia Qt LGPL
|
---|
25 | ** Exception version 1.0, included in the file LGPL_EXCEPTION.txt in this
|
---|
26 | ** package.
|
---|
27 | **
|
---|
28 | ** GNU General Public License Usage
|
---|
29 | ** Alternatively, this file may be used under the terms of the GNU
|
---|
30 | ** General Public License version 3.0 as published by the Free Software
|
---|
31 | ** Foundation and appearing in the file LICENSE.GPL included in the
|
---|
32 | ** packaging of this file. Please review the following information to
|
---|
33 | ** ensure the GNU General Public License version 3.0 requirements will be
|
---|
34 | ** met: http://www.gnu.org/copyleft/gpl.html.
|
---|
35 | **
|
---|
36 | ** If you are unsure which license is appropriate for your use, please
|
---|
37 | ** contact the sales department at [email protected].
|
---|
38 | ** $QT_END_LICENSE$
|
---|
39 | **
|
---|
40 | ****************************************************************************/
|
---|
41 |
|
---|
42 | #include "graphwidget.h"
|
---|
43 | #include "edge.h"
|
---|
44 | #include "node.h"
|
---|
45 |
|
---|
46 | #include <QDebug>
|
---|
47 | #include <QGraphicsScene>
|
---|
48 | #include <QWheelEvent>
|
---|
49 |
|
---|
50 | #include <math.h>
|
---|
51 |
|
---|
52 | GraphWidget::GraphWidget()
|
---|
53 | : timerId(0)
|
---|
54 | {
|
---|
55 | QGraphicsScene *scene = new QGraphicsScene(this);
|
---|
56 | scene->setItemIndexMethod(QGraphicsScene::NoIndex);
|
---|
57 | scene->setSceneRect(-200, -200, 400, 400);
|
---|
58 | setScene(scene);
|
---|
59 | setCacheMode(CacheBackground);
|
---|
60 | setViewportUpdateMode(BoundingRectViewportUpdate);
|
---|
61 | setRenderHint(QPainter::Antialiasing);
|
---|
62 | setTransformationAnchor(AnchorUnderMouse);
|
---|
63 | setResizeAnchor(AnchorViewCenter);
|
---|
64 |
|
---|
65 | Node *node1 = new Node(this);
|
---|
66 | Node *node2 = new Node(this);
|
---|
67 | Node *node3 = new Node(this);
|
---|
68 | Node *node4 = new Node(this);
|
---|
69 | centerNode = new Node(this);
|
---|
70 | Node *node6 = new Node(this);
|
---|
71 | Node *node7 = new Node(this);
|
---|
72 | Node *node8 = new Node(this);
|
---|
73 | Node *node9 = new Node(this);
|
---|
74 | scene->addItem(node1);
|
---|
75 | scene->addItem(node2);
|
---|
76 | scene->addItem(node3);
|
---|
77 | scene->addItem(node4);
|
---|
78 | scene->addItem(centerNode);
|
---|
79 | scene->addItem(node6);
|
---|
80 | scene->addItem(node7);
|
---|
81 | scene->addItem(node8);
|
---|
82 | scene->addItem(node9);
|
---|
83 | scene->addItem(new Edge(node1, node2));
|
---|
84 | scene->addItem(new Edge(node2, node3));
|
---|
85 | scene->addItem(new Edge(node2, centerNode));
|
---|
86 | scene->addItem(new Edge(node3, node6));
|
---|
87 | scene->addItem(new Edge(node4, node1));
|
---|
88 | scene->addItem(new Edge(node4, centerNode));
|
---|
89 | scene->addItem(new Edge(centerNode, node6));
|
---|
90 | scene->addItem(new Edge(centerNode, node8));
|
---|
91 | scene->addItem(new Edge(node6, node9));
|
---|
92 | scene->addItem(new Edge(node7, node4));
|
---|
93 | scene->addItem(new Edge(node8, node7));
|
---|
94 | scene->addItem(new Edge(node9, node8));
|
---|
95 |
|
---|
96 | node1->setPos(-50, -50);
|
---|
97 | node2->setPos(0, -50);
|
---|
98 | node3->setPos(50, -50);
|
---|
99 | node4->setPos(-50, 0);
|
---|
100 | centerNode->setPos(0, 0);
|
---|
101 | node6->setPos(50, 0);
|
---|
102 | node7->setPos(-50, 50);
|
---|
103 | node8->setPos(0, 50);
|
---|
104 | node9->setPos(50, 50);
|
---|
105 |
|
---|
106 | scale(qreal(0.8), qreal(0.8));
|
---|
107 | setMinimumSize(400, 400);
|
---|
108 | setWindowTitle(tr("Elastic Nodes"));
|
---|
109 | }
|
---|
110 |
|
---|
111 | void GraphWidget::itemMoved()
|
---|
112 | {
|
---|
113 | if (!timerId)
|
---|
114 | timerId = startTimer(1000 / 25);
|
---|
115 | }
|
---|
116 |
|
---|
117 | void GraphWidget::keyPressEvent(QKeyEvent *event)
|
---|
118 | {
|
---|
119 | switch (event->key()) {
|
---|
120 | case Qt::Key_Up:
|
---|
121 | centerNode->moveBy(0, -20);
|
---|
122 | break;
|
---|
123 | case Qt::Key_Down:
|
---|
124 | centerNode->moveBy(0, 20);
|
---|
125 | break;
|
---|
126 | case Qt::Key_Left:
|
---|
127 | centerNode->moveBy(-20, 0);
|
---|
128 | break;
|
---|
129 | case Qt::Key_Right:
|
---|
130 | centerNode->moveBy(20, 0);
|
---|
131 | break;
|
---|
132 | case Qt::Key_Plus:
|
---|
133 | scaleView(qreal(1.2));
|
---|
134 | break;
|
---|
135 | case Qt::Key_Minus:
|
---|
136 | scaleView(1 / qreal(1.2));
|
---|
137 | break;
|
---|
138 | case Qt::Key_Space:
|
---|
139 | case Qt::Key_Enter:
|
---|
140 | foreach (QGraphicsItem *item, scene()->items()) {
|
---|
141 | if (qgraphicsitem_cast<Node *>(item))
|
---|
142 | item->setPos(-150 + qrand() % 300, -150 + qrand() % 300);
|
---|
143 | }
|
---|
144 | break;
|
---|
145 | default:
|
---|
146 | QGraphicsView::keyPressEvent(event);
|
---|
147 | }
|
---|
148 | }
|
---|
149 |
|
---|
150 | void GraphWidget::timerEvent(QTimerEvent *event)
|
---|
151 | {
|
---|
152 | Q_UNUSED(event);
|
---|
153 |
|
---|
154 | QList<Node *> nodes;
|
---|
155 | foreach (QGraphicsItem *item, scene()->items()) {
|
---|
156 | if (Node *node = qgraphicsitem_cast<Node *>(item))
|
---|
157 | nodes << node;
|
---|
158 | }
|
---|
159 |
|
---|
160 | foreach (Node *node, nodes)
|
---|
161 | node->calculateForces();
|
---|
162 |
|
---|
163 | bool itemsMoved = false;
|
---|
164 | foreach (Node *node, nodes) {
|
---|
165 | if (node->advance())
|
---|
166 | itemsMoved = true;
|
---|
167 | }
|
---|
168 |
|
---|
169 | if (!itemsMoved) {
|
---|
170 | killTimer(timerId);
|
---|
171 | timerId = 0;
|
---|
172 | }
|
---|
173 | }
|
---|
174 |
|
---|
175 | void GraphWidget::wheelEvent(QWheelEvent *event)
|
---|
176 | {
|
---|
177 | scaleView(pow((double)2, -event->delta() / 240.0));
|
---|
178 | }
|
---|
179 |
|
---|
180 | void GraphWidget::drawBackground(QPainter *painter, const QRectF &rect)
|
---|
181 | {
|
---|
182 | Q_UNUSED(rect);
|
---|
183 |
|
---|
184 | // Shadow
|
---|
185 | QRectF sceneRect = this->sceneRect();
|
---|
186 | QRectF rightShadow(sceneRect.right(), sceneRect.top() + 5, 5, sceneRect.height());
|
---|
187 | QRectF bottomShadow(sceneRect.left() + 5, sceneRect.bottom(), sceneRect.width(), 5);
|
---|
188 | if (rightShadow.intersects(rect) || rightShadow.contains(rect))
|
---|
189 | painter->fillRect(rightShadow, Qt::darkGray);
|
---|
190 | if (bottomShadow.intersects(rect) || bottomShadow.contains(rect))
|
---|
191 | painter->fillRect(bottomShadow, Qt::darkGray);
|
---|
192 |
|
---|
193 | // Fill
|
---|
194 | QLinearGradient gradient(sceneRect.topLeft(), sceneRect.bottomRight());
|
---|
195 | gradient.setColorAt(0, Qt::white);
|
---|
196 | gradient.setColorAt(1, Qt::lightGray);
|
---|
197 | painter->fillRect(rect.intersect(sceneRect), gradient);
|
---|
198 | painter->setBrush(Qt::NoBrush);
|
---|
199 | painter->drawRect(sceneRect);
|
---|
200 |
|
---|
201 | // Text
|
---|
202 | QRectF textRect(sceneRect.left() + 4, sceneRect.top() + 4,
|
---|
203 | sceneRect.width() - 4, sceneRect.height() - 4);
|
---|
204 | QString message(tr("Click and drag the nodes around, and zoom with the mouse "
|
---|
205 | "wheel or the '+' and '-' keys"));
|
---|
206 |
|
---|
207 | QFont font = painter->font();
|
---|
208 | font.setBold(true);
|
---|
209 | font.setPointSize(14);
|
---|
210 | painter->setFont(font);
|
---|
211 | painter->setPen(Qt::lightGray);
|
---|
212 | painter->drawText(textRect.translated(2, 2), message);
|
---|
213 | painter->setPen(Qt::black);
|
---|
214 | painter->drawText(textRect, message);
|
---|
215 | }
|
---|
216 |
|
---|
217 | void GraphWidget::scaleView(qreal scaleFactor)
|
---|
218 | {
|
---|
219 | qreal factor = matrix().scale(scaleFactor, scaleFactor).mapRect(QRectF(0, 0, 1, 1)).width();
|
---|
220 | if (factor < 0.07 || factor > 100)
|
---|
221 | return;
|
---|
222 |
|
---|
223 | scale(scaleFactor, scaleFactor);
|
---|
224 | }
|
---|