QFCTopologyInput.rar

#include "graphwidget.h" #include "edge.h" #include "node.h" #include <math.h> #include <QKeyEvent> #include <QRandomGenerator> GraphWidget::GraphWidget(QWidget *parent,int node_number, QSize size) : QGraphicsView(parent) { myscene = new QGraphicsScene(this); myscene->setItemIndexMethod(QGraphicsScene::NoIndex); myscene->setSceneRect(-200, -200, 400, 400); setScene(myscene); setCacheMode(CacheBackground); setViewportUpdateMode(BoundingRectViewportUpdate); setRenderHint(QPainter::Antialiasing); setTransformationAnchor(AnchorUnderMouse); scale(qreal(1), qreal(1)); setMinimumSize(400, 400); setWindowTitle(tr("Elastic Nodes")); m_node_number = node_number>10?10:node_number; //不能超过10个节点 m_size = size; is_waitlink = false; for(int i = 0;i<m_node_number;i++) { node[i] = new Node(this,i,m_size); myscene->addItem(node[i]); } } void GraphWidget::setNodeNumber(int node_number) { m_node_number = node_number>10?10:node_number; //不能超过10个节点 myscene->clear(); for(int i = 0;i<m_node_number;i++) { node[i] = new Node(this,i,m_size); myscene->addItem(node[i]); } //删除节点间连线 for(int i = 0;i<10;i++) { for(int j = 0;j<10;j++) { edge[i][j] = NULL; } } } void GraphWidget::itemMoved() { if (!timerId) timerId = startTimer(1000 / 25); } void GraphWidget::keyPressEvent(QKeyEvent *event) { switch (event->key()) { case Qt::Key_Up: centerNode->moveBy(0, -20); break; case Qt::Key_Down: centerNode->moveBy(0, 20); break; case Qt::Key_Left: centerNode->moveBy(-20, 0); break; case Qt::Key_Right: centerNode->moveBy(20, 0); break; case Qt::Key_Plus: zoomIn(); break; case Qt::Key_Minus: zoomOut(); break; case Qt::Key_Space: case Qt::Key_Enter: shuffle(); break; default: QGraphicsView::keyPressEvent(event); } } void GraphWidget::timerEvent(QTimerEvent *event) { Q_UNUSED(event); QVector<Node *> nodes; const QList<QGraphicsItem *> items = scene()->items(); for (QGraphicsItem *item : items) { if (Node *node = qgraphicsitem_cast<Node *>(item)) nodes << node; } for (Node *node : qAsConst(nodes)) node->calculateForces(); bool itemsMoved = false; for (Node *node : qAsConst(nodes)) { if (node->advancePosition()) itemsMoved = true; } if (!itemsMoved) { killTimer(timerId); timerId = 0; } } void GraphWidget::choosenode(int node_id) { if(is_waitlink == false) //记录第一个点 { firstnode = node_id; node[firstnode]->setchoosed(true); node[firstnode]->update(); is_waitlink = true; } else //判断和第一个点的关系 { node[firstnode]->setchoosed(false); //不起作用 node[firstnode]->update(); //需要加上这一句 if(node_id != firstnode) { //int maxid = qMax(node_id,firstnode); //int minid = qMin(node_id,firstnode); //int edge_id = (18-minid)*(minid+1)/2+maxid-10; if(!edge[firstnode][node_id]) //指针需要初始化，野指针和空指针的区别 { edge[firstnode][node_id] = new Edge(node[firstnode], node[node_id]); myscene->addItem(edge[firstnode][node_id]); } else { myscene->removeItem(edge[firstnode][node_id]); edge[firstnode][node_id] = NULL; node[firstnode]->setchoosed(false); } } is_waitlink = false; } } void GraphWidget::addEdge(int firstnode,int node_id) { edge[firstnode][node_id] = new Edge(node[firstnode], node[node_id]); myscene->addItem(edge[firstnode][node_id]); } void GraphWidget::wheelEvent(QWheelEvent *event) { scaleView(pow(2., -event->angleDelta().y() / 240.0)); } void GraphWidget::drawBackground(QPainter *painter, const QRectF &rect) { Q_UNUSED(rect); // // Shadow // QRectF sceneRect = this->sceneRect(); // QRectF rightShadow(sceneRect.right(), sceneRect.top() + 5, 5, sceneRect.height()); // QRectF bottomShadow(sceneRect.left() + 5, sceneRect.bottom(), sceneRect.width(), 5); // if (rightShadow.intersects(rect) || rightShadow.contains(rect)) // painter->fillRect(rightShadow, Qt::darkGray); // if (bottomShadow.intersects(rect) || bottomShadow.contains(rect)) // painter->fillRect(bottomShadow, Qt::darkGray); // // Fill // QLinearGradient gradient(sceneRect.topLeft(), sceneRect.bottomRight()); // gradient.setColorAt(0, Qt::white); // gradient.setColorAt(1, Qt::lightGray); // painter->fillRect(rect.intersected(sceneRect), gradient); // painter->setBrush(Qt::NoBrush); // painter->drawRect(sceneRect); // // Text // QRectF textRect(sceneRect.left() + 4, sceneRect.top() + 4, // sceneRect.width() - 4, sceneRect.height() - 4); // QString message(tr("Please select two nodes to connect")); // QFont font = painter->font(); // font.setBold(true); // font.setPointSize(14); // painter->setFont(font); // painter->setPen(Qt::lightGray); // painter->drawText(textRect.translated(2, 2), message); // painter->setPen(Qt::black); // painter->drawText(textRect, message); } void GraphWidget::scaleView(qreal scaleFactor) { qreal factor = transform().scale(scaleFactor, scaleFactor).mapRect(QRectF(0, 0, 1, 1)).width(); if (factor < 0.07 || factor > 100) return; scale(scaleFactor, scaleFactor); } void GraphWidget::shuffle() { const QList<QGraphicsItem *> items = scene()->items(); for (QGraphicsItem *item : items) { if (qgraphicsitem_cast<Node *>(item)) item->setPos(-150 + QRandomGenerator::global()->bounded(300), -150 + QRandomGenerator::global()->bounded(300)); } } void GraphWidget::zoomIn() { scaleView(qreal(1.2)); } void GraphWidget::zoomOut() { scaleView(1 / qreal(1.2)); }