editor/hmi.cpp

#include "hmi.h"
#include "ui_hmi.h"
#include <QListWidgetItem>
#include <QMimeData>
#include <QDrag>
#include <QPainter>
#include <QMouseEvent>
#include <QDropEvent>
#include <QGraphicsView>
#include <QMessageBox>
#include "creatitem.h"

HMI::HMI(QWidget *parent) :
    QWidget(parent),
    ui(new Ui::HMI)
{
    ui->setupUi(this);
    /* 1. 场景 */
    hmi_scene_ = new QGraphicsScene(this);
    ui->graphicsView->setScene(hmi_scene_);
    ui->graphicsView->setSceneRect(0, 0, 800, 600);
    ui->graphicsView->setDragMode(QGraphicsView::RubberBandDrag);

    /* 2. 列表 */
    ui->listWidget->setViewMode(QListView::IconMode);
    ui->listWidget->setIconSize(QSize(60, 30));
    ui->listWidget->setDragEnabled(false);
    ui->listWidget->viewport()->installEventFilter(this);

    /* 3. 填充列表 */
    createComponents();
    connect(ui->listWidget,&QListWidget::currentTextChanged,this,&HMI::onListwidgetCurrenttextchanged);
}

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

bool HMI::eventFilter(QObject *obj, QEvent *event)
{
    if (obj == ui->listWidget->viewport()) {
        static QListWidgetItem *dragItem = nullptr;
        static QPoint startPos;

        if (event->type() == QEvent::MouseButtonPress) {
            auto *me = static_cast<QMouseEvent*>(event);
            if (me->button() == Qt::LeftButton) {
                dragItem  = ui->listWidget->itemAt(me->pos());
                startPos  = me->pos();
            }
        } else if (event->type() == QEvent::MouseMove && dragItem) {
            auto *me = static_cast<QMouseEvent*>(event);
            if ((me->pos() - startPos).manhattanLength()
                    >= QApplication::startDragDistance()) {

                QString type = dragItem->data(Qt::UserRole).toString();

                QMimeData *mime = new QMimeData;
                mime->setData("application/x-component", type.toUtf8());

                QDrag *drag = new QDrag(this);
                drag->setMimeData(mime);
                drag->setPixmap(dragItem->icon().pixmap(ui->listWidget->iconSize()));
                drag->setHotSpot(drag->pixmap().rect().center());
                drag->exec(Qt::CopyAction);

                dragItem = nullptr;
            }
        } else if (event->type() == QEvent::MouseButtonRelease) {
            dragItem = nullptr;
        }
    }
    return QWidget::eventFilter(obj, event);
}

void HMI::createComponents()
{
    struct Comp { QString name; QColor color; };
    const QVector<Comp> comps = {
        {"按钮",     Qt::blue},
        {"指示灯",     Qt::red}
    };

    for (const Comp &c : comps) {
        QListWidgetItem *it = new QListWidgetItem(c.name, ui->listWidget);

        QPixmap pix(60, 30);
        pix.fill(Qt::white);
        QPainter p(&pix);
        p.setRenderHint(QPainter::Antialiasing);
        p.setBrush(c.color.lighter(150));
        p.setPen(QPen(c.color, 2));
        p.drawRoundedRect(QRect(5, 5, 50, 20), 3, 3);
        p.setPen(Qt::black);
        p.drawText(QRect(5, 5, 50, 20), Qt::AlignCenter, c.name);

        it->setIcon(QIcon(pix));
        it->setData(Qt::UserRole, c.name);
    }
}

void HMI::clearScene()
{
    hmi_scene_->clear();
}

void HMI::applyProjectToScene(const Project& proj)
{
    clearScene();
    QVector<Item*> itemObjs;
    for (const auto& d : proj.items_) {
        Item* item = creatItem(d.type);
        if (!item) continue;
        item->setPos(d.x, d.y);
        item->setRegisterId(d.registerId);
        connect(item, &Item::requestCopy, ui->graphicsView, &MyGraphicsView::onItemRequestCopy);
        connect(item, &Item::requestDelete, ui->graphicsView, &MyGraphicsView::onItemRequestDelete);
        connect(item, &Item::requestBindRegister, ui->graphicsView, &MyGraphicsView::onItemRequestBindRegister);
        hmi_scene_->addItem(item);
        itemObjs.append(item);
    }
}

void HMI::extractSceneToProject(Project& proj)
{
    proj.clear();
    QList<Item*> items;
    for (QGraphicsItem* gi : hmi_scene_->items()) {
        if (Item* it = dynamic_cast<Item*>(gi)) {
            items.append(it);
        }
    }
    for (Item* it : items) {
        Project::ItemData d;
        d.type = it->itemType();
        d.x = it->pos().x();
        d.y = it->pos().y();
        d.registerId = it->registerId();
        proj.items_.append(d);
    }
    for (QGraphicsItem* gi : hmi_scene_->items()) {
        if (Connection* conn = dynamic_cast<Connection*>(gi)) {
            int fromIdx = items.indexOf(conn->from_);
            int toIdx = items.indexOf(conn->to_);
            if (fromIdx >= 0 && toIdx >= 0) {
                Project::ConnectionData c;
                c.from = fromIdx;
                c.to = toIdx;
                c.fromType = static_cast<int>(conn->fromType_);
                c.toType = static_cast<int>(conn->toType_);
                proj.connections_.append(c);
            }
        }
    }
}

void HMI::onListwidgetCurrenttextchanged(const QString &text)
{
    selectedComponentType = text;
}