QFD2/QFD/CCanvas/CMindView.cpp

#include "CMindView.h"

#include "CRectItem.h"
#include "CLineItem.h"
#include "CTextItem.h"
#include "CMind.h"
#include "CNodeItem.h"

#include <QMenu>
#include <QContextMenuEvent>

#include <QDebug>

CMindView::CMindView(QWidget *parent) : QGraphicsView(new QGraphicsScene(), parent)
{
    setRenderHints(QPainter::Antialiasing);  // 抗锯齿

    m_mind  = new CMind(this);
    m_group = new QGraphicsItemGroup();
    m_group->setFlags(QGraphicsItem::ItemIsMovable);
    m_group->setHandlesChildEvents(false);

    scene()->addItem(m_group);

    setStyleSheet("QGraphicsView {border: 1px solid rgba(0, 0, 0, 0.073);background: rgb(244, 244, "
                  "255);}");
}

CMind *CMindView::mind() const
{
    return m_mind;
}

void CMindView::setNodeList(QList<CNodeData> list)
{
    clear();
    m_mind->setNodeList(list);
    m_root = new CNodeItem(m_mind->root());
    refreshItems();
}

void CMindView::addNode(CNodeData n)
{
    m_mind->addNode(n);
    CNodeItem *item = new CNodeItem(n);
    connect(item, &CNodeItem::sigAddSubItem, this, &CMindView::slotAddSubNode);
    connect(item, &CNodeItem::sigRemoveItem, this, &CMindView::slotRemoveNode);
    connect(item, &CNodeItem::sigTextChanged, this, &CMindView::slotTextChanged);

    if (m_root == nullptr) {
        m_root = item;
    } else {
        m_root->endEditing();
        m_root->addSubNode(item);
    }
    item->textItem()->beginEditing();

    refreshItems();
}

void CMindView::clear()
{
    m_mind->nodeList().clear();
    delete m_root;
    m_root = nullptr;

    refreshItems();
}

void CMindView::refreshItems()
{
    for (QGraphicsItem *item : m_items) {
        scene()->removeItem(item);
    }
    m_items.clear();

    refreshNodeGeometry(m_root);
    collectItems(m_root);
    for (QGraphicsItem *item : m_items) {
        m_group->addToGroup(item);
    }

    QRectF r = m_group->childrenBoundingRect();
    setSceneRect(QRectF(QPointF(), r.size()));
}

void CMindView::collectItems(CNodeItem *node)
{
    if (node == nullptr) {
        return;
    }

    m_items.append(node->rectItem());
    for (QObject *o : node->children()) {
        CNodeItem *n = dynamic_cast<CNodeItem *>(o);
        collectItems(n);
    }
}

qreal CMindView::hNodeSpace() const
{
    return m_hNodeSpace;
}

void CMindView::setHNodeSpace(qreal h)
{
    m_hNodeSpace = h;
}

qreal CMindView::vNodeSpace() const
{
    return m_vNodeSpace;
}

void CMindView::setVNodeSpace(qreal v)
{
    m_vNodeSpace = v;
}

void CMindView::setALignment(CMindView::Alignment align)
{
    m_align = align;
}

CNodeItem *CMindView::root() const
{
    return m_root;
}

void CMindView::refreshNodeGeometry(CNodeItem *node, QPointF topLeft)
{
    /// 边框
    if (node == nullptr) {
        return;
    }

    QRect borderRect = QRect(topLeft.x(), topLeft.y() + (node->treeHeight() - node->borderHeight()) / 2,
                             node->borderWidth(), node->borderHeight());
    node->rectItem()->setRect(borderRect);

    /// 文本
    QPointF textPos = QPointF(borderRect.left() + node->xMargin(),
                              borderRect.top() + (node->borderHeight() - node->textHeight()) / 2);
    node->textItem()->setPos(textPos);
    node->textItem()->setTextWidth(borderRect.width() - node->xMargin() * 2);

    node->rectItem()->setPos(QPoint(0, 0));

    /// 子节点
    int x = borderRect.right() + m_hNodeSpace;
    if (m_align) {
        x = borderRect.left() + m_root->maxBorderWidthOfLevel(node->depth()) + m_hNodeSpace;
    }

    int y = topLeft.y();
    if (node->borderHeight() > node->childrenHeight()) {
        y += (node->borderHeight() - node->childrenHeight()) / 2;
    }

    for (QObject *obj : node->children()) {
        CNodeItem *subNode = dynamic_cast<CNodeItem *>(obj);
        refreshNodeGeometry(subNode, QPointF(x, y));
        y += subNode->treeHeight() + m_vNodeSpace;

        subNode->lineItem()->setStartPos(node->rectItem()->centerRight());
        subNode->lineItem()->setEndPos(subNode->rectItem()->centerLeft());
    }
}

void CMindView::mousePressEvent(QMouseEvent *event)
{
    QGraphicsItem *item = itemAt(event->pos());
    CTextItem *text     = dynamic_cast<CTextItem *>(item);
    if (text == nullptr && m_root != nullptr && m_root->isEditing()) {
        m_root->endEditing();
        refreshItems();
    }
    QGraphicsView::mousePressEvent(event);
}

void CMindView::slotAddSubNode(int pNumber)
{
    if (pNumber < 0) {
        return;
    }
    CNodeData data = root()->data();
    CNodeData n    = CNodeData(data.projectId, data.evalType, m_mind->maxNumber() + 1, pNumber);

    addNode(n);
}

void CMindView::slotRemoveNode(int number)
{
    m_mind->removeNode(number);
    if (number == m_root->data().number) {
        clear();
    } else {
        m_root->removeNode(number);
    }
    refreshItems();
}

void CMindView::slotTextChanged()
{
    refreshItems();
}