Added Drag'n'Drop support for ProjectView

Author: serhmarch

changelog.md

@@ -165,3 +165,4 @@ List of changes of new v0.4 version of ModbusTools:
 * Improved MDI window management
 * Improved script text editor
 * Added 'Import Project' functionality which unites current project with the imported one
+* Implemented Drag`n`Drop for ProjectView window

src/client/gui/project/client_projectmodel.cpp

@@ -134,6 +134,30 @@ QString mbClientProjectModel::deviceName(const mbClientDevice *device) const
         return device->name();
 }
 
+bool mbClientProjectModel::dropDevice(Qt::DropAction action, mbCorePort *sourcePort, int srcIndex, mbCorePort *destPort, int dstIndex)
+{
+    mbClientPort *srcPort = static_cast<mbClientPort*>(sourcePort);
+    mbClientPort *dstPort = static_cast<mbClientPort*>(destPort);
+    mbClientDevice *device = srcPort->device(srcIndex);
+    if (dstIndex < 0 || dstIndex > dstPort->deviceCount())
+        dstIndex = dstPort->deviceCount();
+    if (action == Qt::MoveAction && srcPort == dstPort && srcPort->deviceIndex(device) == dstIndex)
+        return false; // No move
+    if (action == Qt::MoveAction)
+    {
+        srcPort->deviceRemove(device);
+        dstPort->deviceInsert(device, dstIndex);
+    }
+    else if (action == Qt::CopyAction)
+    {
+        mbClientDevice *newDevice = new mbClientDevice();
+        newDevice->setSettings(device->settings());
+        dstPort->deviceInsert(newDevice, dstIndex);
+        project()->deviceAdd(newDevice);
+    }
+    return true;
+}
+
 void mbClientProjectModel::portAdd(mbCorePort *port)
 {
     int i = project()->portIndex(static_cast<mbClientPort*>(port));

src/client/gui/project/client_projectmodel.h

@@ -57,13 +57,16 @@ class mbClientProjectModel : public mbCoreProjectModel
 
 protected:
     QString deviceName(const mbClientDevice *device) const;
+    bool dropDevice(Qt::DropAction action, mbCorePort *sourcePort, int srcIndex, mbCorePort *destPort, int dstIndex) override;
 
 protected Q_SLOTS:
     void portAdd(mbCorePort *port) override;
     void portRemove(mbCorePort *port) override;
     void deviceAdd(mbClientDevice *device);
     void deviceRemove(mbClientDevice *device);
     void deviceChanged();
+
+public:
 };
 
 #endif // CLIENT_PROJECTMODEL_H

src/client/gui/project/client_projectui.cpp

@@ -22,12 +22,12 @@
 */
 #include "client_projectui.h"
 
-#include <QTreeView>
-
 #include <project/client_project.h>
 #include <project/client_port.h>
 #include <project/client_device.h>
 
+#include <gui/project/core_projecttreeview.h>
+
 #include "client_projectmodel.h"
 #include "client_projectdelegate.h"
 

src/client/gui/project/client_projectui.h

@@ -27,6 +27,7 @@
 
 class mbClientPort;
 class mbClientDevice;
+class mbClientProjectTreeView;
 
 class mbClientProjectUi : public mbCoreProjectUi
 {

src/core/gui/core_rsc.qrc

@@ -22,5 +22,6 @@
         <file>icons/arrow_down.png</file>
         <file>icons/arrow_up.png</file>
         <file>icons/logexport.png</file>
+        <file>icons/device_small.png</file>
     </qresource>
 </RCC>

src/core/gui/icons/device_small.png

@@ -22,11 +22,26 @@
 */
 #include "core_projectmodel.h"
 
+#include <QMimeData>
+
 #include <project/core_project.h>
 #include <project/core_port.h>
+#include <project/core_device.h>
 
 #include <gui/core_ui.h>
 
+mbCoreProjectModel::Strings::Strings() :
+    mimeTypeDevice(QStringLiteral("application/x-modbustools-device"))
+
+{
+}
+
+const mbCoreProjectModel::Strings &mbCoreProjectModel::Strings::instance()
+{
+    static const Strings s;
+    return s;
+}
+
 mbCoreProjectModel::mbCoreProjectModel(QObject* parent) :
     QAbstractItemModel (parent)
 {
@@ -69,6 +84,63 @@ Qt::ItemFlags mbCoreProjectModel::flags(const QModelIndex &index) const
     return f;
 }
 
+Qt::DropActions mbCoreProjectModel::supportedDropActions() const
+{
+    return Qt::MoveAction | Qt::CopyAction;
+}
+
+QStringList mbCoreProjectModel::mimeTypes() const
+{
+    return {mbCoreProjectModel::Strings::instance().mimeTypeDevice};
+}
+
+QMimeData *mbCoreProjectModel::mimeData(const QModelIndexList &indexes) const
+{
+    QMimeData *mimeData = new QMimeData();
+    if (indexes.size() == 1)
+    {
+        QModelIndex index = indexes.first();
+        if (mbCoreDevice *dev = this->getDeviceByIndex(index))
+        {
+            QByteArray encoded;
+            QDataStream stream(&encoded, QIODevice::WriteOnly);
+            mbCorePort *port = getPortByIndex(indexes.first());
+            stream << port->name() << index.row();
+            mimeData->setData(mbCoreProjectModel::Strings::instance().mimeTypeDevice, encoded);
+        }
+    }
+    return mimeData;
+}
+
+bool mbCoreProjectModel::dropMimeData(const QMimeData *data, Qt::DropAction action, int row, int column, const QModelIndex &parent)
+{
+    if (!data->hasFormat(mbCoreProjectModel::Strings::instance().mimeTypeDevice))
+        return false;
+    QByteArray encoded = data->data(mbCoreProjectModel::Strings::instance().mimeTypeDevice);
+    QDataStream stream(&encoded, QIODevice::ReadOnly);
+    QString srcPortName;
+    int srcIndex;
+    stream >> srcPortName >> srcIndex;
+    mbCoreProject *prj = projectCore();
+    if (!prj)
+        return false;
+    mbCorePort *srcPort = prj->portCore(srcPortName);
+    mbCoreDevice *device = prj->deviceCore(srcIndex);
+    if (!srcPort || !device)
+        return false;
+    QModelIndex portIndex = parent;
+    if (!portIndex.isValid())
+        portIndex = index(row, column);
+    mbCorePort *dstPort = portCore(portIndex);
+    if (!dstPort)
+    {
+        dstPort = prj->portCore(prj->portCount()-1);
+        if (!dstPort)
+            return false;
+    }
+    return dropDevice(action, srcPort, srcIndex, dstPort, row);
+}
+
 void mbCoreProjectModel::setUseNameWithSettings(bool use)
 {
     if (m_settings.useNameWithSettings != use)

src/core/gui/project/core_projectmodel.cpp

@@ -36,6 +36,15 @@ class MB_EXPORT mbCoreProjectModel : public QAbstractItemModel
 {
     Q_OBJECT
 
+public:
+    struct MB_EXPORT Strings
+    {
+        const QString mimeTypeDevice;
+
+        Strings();
+        static const Strings &instance();
+    };
+
 public:
     mbCoreProjectModel(QObject* parent = nullptr);
 
@@ -48,6 +57,10 @@ class MB_EXPORT mbCoreProjectModel : public QAbstractItemModel
     int columnCount(const QModelIndex &parent = QModelIndex()) const override;
     bool hasChildren(const QModelIndex &parent = QModelIndex()) const override;
     Qt::ItemFlags flags(const QModelIndex &index) const override;
+    Qt::DropActions supportedDropActions() const override;
+    QStringList mimeTypes() const override;
+    QMimeData *mimeData(const QModelIndexList &indexes) const override;
+    bool dropMimeData(const QMimeData *data, Qt::DropAction action, int row, int column, const QModelIndex &parent) override;
 
 public: // settings
     inline bool useNameWithSettings() const { return m_settings.useNameWithSettings; }
@@ -67,6 +80,7 @@ class MB_EXPORT mbCoreProjectModel : public QAbstractItemModel
 
 protected:
     QString portName(const mbCorePort *port) const;
+    virtual bool dropDevice(Qt::DropAction action, mbCorePort *srcPort, int srcIndex, mbCorePort *dstPort, int dstIndex) = 0;
 
 protected Q_SLOTS:
     virtual void portChanged();

src/core/gui/project/core_projectmodel.h

@@ -0,0 +1,60 @@
+#include "core_projecttreeview.h"
+
+#include <QApplication>
+#include <QTreeView>
+#include <QMouseEvent>
+#include <QDrag>
+
+mbCoreProjectTreeView::mbCoreProjectTreeView(QWidget *parent) : QTreeView(parent)
+{
+
+}
+
+void mbCoreProjectTreeView::mousePressEvent(QMouseEvent *event)
+{
+    if (event->button() == Qt::LeftButton)
+        m_dragStartPosition = event->pos();
+    QTreeView::mousePressEvent(event);
+}
+
+void mbCoreProjectTreeView::mouseMoveEvent(QMouseEvent *event)
+{
+    if (!(event->buttons() & Qt::LeftButton))
+        return;
+    if ((event->pos() - m_dragStartPosition).manhattanLength() < QApplication::startDragDistance())
+        return;
+    QModelIndex index = this->indexAt(m_dragStartPosition);
+    if (!index.isValid())
+        return;
+    if (!index.parent().isValid()) // Note: invalid parent index means that is port
+        return;
+    m_dragInProgress = true;
+    QDrag *drag = new QDrag(this);
+    QMimeData *mimeData = model()->mimeData({index});
+    drag->setMimeData(mimeData);
+    drag->setPixmap(QPixmap(":/core/icons/device_small.png"));
+    drag->exec(Qt::MoveAction | Qt::CopyAction);
+    m_dragInProgress = false;
+}
+
+void mbCoreProjectTreeView::dropEvent(QDropEvent *event)
+{
+    QModelIndex index = indexAt(event->pos());
+    if (index.isValid())
+        model()->dropMimeData(event->mimeData(), event->dropAction(), index.row(), index.column(), index.parent());
+}
+
+void mbCoreProjectTreeView::dragEnterEvent(QDragEnterEvent *event)
+{
+    event->acceptProposedAction();
+}
+
+void mbCoreProjectTreeView::dragMoveEvent(QDragMoveEvent *event)
+{
+    event->acceptProposedAction();
+}
+
+void mbCoreProjectTreeView::dragLeaveEvent(QDragLeaveEvent *event)
+{
+    event->accept();
+}