DiagramDesigner/source/diagramView.cpp

#include "diagramView.h"
#include "ui_diagramView.h"
#include "tools.h"
#include <QStandardItemModel>
#include <QMenu>
#include <QUuid>


DiagramView::DiagramView(QWidget *parent)
    : QDialog(parent)
    , ui(new Ui::diagramView)
    ,_pModel(nullptr)
{
    ui->setupUi(this);
    _pModel = new QStandardItemModel(this);
    ui->treeView->setContextMenuPolicy(Qt::CustomContextMenu);
}

DiagramView::~DiagramView()
{
    delete ui;
}

void DiagramView::initial()
{
    connect(ui->treeView, &QTreeView::customContextMenuRequested, this, &DiagramView::onIndexRbtnClicked);
    connect(ui->treeView, &QTreeView::clicked, this, &DiagramView::onItemClicked);
    connect(_pModel, &QStandardItemModel::itemChanged, this, &DiagramView::onItemChanged);
    ui->treeView->setHeaderHidden(true);
    // 设置模型的列数
    _pModel->setColumnCount(1);

    // 创建树视图
    ui->treeView->setModel(_pModel);

    // 展开所有节点
    ui->treeView->expandAll();
    // 显示树视图
    ui->treeView->setWindowTitle(QObject::tr("组态图结构"));
}

void DiagramView::loadTopologyFromDB()
{

}

void DiagramView::onIndexRbtnClicked(const QPoint &pos)
{
    // 获取当前点击的位置对应的索引
    QMenu menu;
    QModelIndex index = ui->treeView->indexAt(pos);
    if (!index.isValid()) {
        // 如果点击的是空白区域，创建新组态图菜单
        int nCount = _pModel->invisibleRootItem()->rowCount();
        QAction *addAction = new QAction("添加组态图", this);

        connect(addAction,&QAction::triggered,this,[&](){
            QVariant id = QUuid::createUuid();
            QString sName = QString::fromWCharArray(L"组态图_")+QString::number(nCount);
            QStandardItem *gridItem = new QStandardItem(sName);
            gridItem->setData(id,Qt::UserRole+1);
            _pModel->appendRow(gridItem);

            DiagramInfo info;
            //todo:具体信息
            info.id = id;          //新建的组态图随机赋予临时id
            info.sName = sName;
            info.sTag = sName;
            emit diagramCreate(info);
        });

        menu.addAction(addAction);
    }
    else{       //目标组态图下添加子组态图
        QStandardItem* item = _pModel->itemFromIndex(index);
        QVariant parentId = item->data(Qt::UserRole+1);

        if(item)
        {
            int nCount = item->rowCount();
            QAction *addAction = new QAction("添加子组态图", this);

            connect(addAction,&QAction::triggered,this,[&](){
                QVariant id = QUuid::createUuid();
                QString sName = item->text()+"_"+QString::number(nCount);
                QStandardItem *gridItem = new QStandardItem(sName);
                gridItem->setData(id,Qt::UserRole+1);
                item->appendRow(gridItem);

                DiagramInfo info;
                //todo:具体信息
                info.id = id;          //新建的随机赋予临时id
                info.sName = sName;
                info.sTag = sName;
                info.parentId = parentId;
                emit diagramCreate(info);
            });

            menu.addAction(addAction);


            QAction *delAction = new QAction("删除", this);
            connect(delAction,&QAction::triggered,this,[&](){
                QModelIndex currentIndex = ui->treeView->currentIndex();
                if(currentIndex.isValid())
                {
                    QStandardItem* pItem = _pModel->itemFromIndex(currentIndex);
                    QVariant id = pItem->data(Qt::UserRole+1);
                    QString sName = pItem->text();
                    if(pItem)
                    {
                        QString str = item->text();
                        DiagramInfo info;
                        //todo:具体信息
                        info.id = id;          //新建的随机赋予临时id
                        info.sName = sName;
                        info.sTag = sName;
                        info.parentId = parentId;
                        emit diagramDelete(info);

                        QStandardItem* pParent = item->parent();
                        if(pParent)
                        {
                            pParent->removeRow(currentIndex.row());
                        }
                    }
                }
            });
            menu.addAction(delAction);
        }
    }

    // 在点击位置显示菜单
    menu.exec(ui->treeView->mapToGlobal(pos));
}

void DiagramView::onItemChanged(QStandardItem *item)
{
    int nLevel = getLevel(item);
    QString str = item->text();
    DiagramInfo info;
    //todo:具体信息
    emit diagramChange(info);
}

void DiagramView::onItemClicked(const QModelIndex &index)
{
    QStandardItem* item = _pModel->itemFromIndex(index);
    if(item)
    {
        DiagramInfo info;
        //todo:具体信息
        emit diagramSelected(info);
    }
}