From 73092e81030786aa48df0b24fc50c10a013b320c 2017-10-26 15:34:10
From: Thibaud Rabillard <trabillard@itlink.fr>
Date: 2017-10-26 15:34:10
Subject: [PATCH] Drag of variables

---

diff --git a/core/include/Variable/VariableController.h b/core/include/Variable/VariableController.h
index 3e18ad1..b198dcd 100644
--- a/core/include/Variable/VariableController.h
+++ b/core/include/Variable/VariableController.h
@@ -68,6 +68,8 @@ public:
      */
     void deleteVariables(const QVector<std::shared_ptr<Variable> > &variables) noexcept;
 
+    QByteArray mimeDataForVariables(const QList<std::shared_ptr<Variable> > &variables) const;
+    QList<std::shared_ptr<Variable> > variablesForMimeData(const QByteArray &mimeData) const;
 
     static AcquisitionZoomType getZoomType(const SqpRange &range, const SqpRange &oldRange);
 signals:
diff --git a/core/include/Variable/VariableModel.h b/core/include/Variable/VariableModel.h
index 960ea81..95da42c 100644
--- a/core/include/Variable/VariableModel.h
+++ b/core/include/Variable/VariableModel.h
@@ -18,6 +18,7 @@ enum VariableRoles { ProgressRole = Qt::UserRole };
 
 class IDataSeries;
 class Variable;
+class VariableController;
 
 /**
  * @brief The VariableModel class aims to hold the variables that have been created in SciQlop
@@ -25,7 +26,7 @@ class Variable;
 class SCIQLOP_CORE_EXPORT VariableModel : public QAbstractTableModel {
     Q_OBJECT
 public:
-    explicit VariableModel(QObject *parent = nullptr);
+    explicit VariableModel(VariableController *parent = nullptr);
 
     /**
      * Adds an existing variable in the model.
@@ -73,7 +74,20 @@ public:
     virtual QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
     virtual QVariant headerData(int section, Qt::Orientation orientation,
                                 int role = Qt::DisplayRole) const override;
-
+    virtual Qt::ItemFlags flags(const QModelIndex &index) const override;
+
+    // ///////////////// //
+    // Drag&Drop methods //
+    // ///////////////// //
+
+    virtual Qt::DropActions supportedDropActions() const override;
+    virtual Qt::DropActions supportedDragActions() const override;
+    virtual QStringList mimeTypes() const override;
+    virtual QMimeData *mimeData(const QModelIndexList &indexes) const override;
+    virtual bool canDropMimeData(const QMimeData *data, Qt::DropAction action, int row, int column,
+                                 const QModelIndex &parent) const override;
+    virtual bool dropMimeData(const QMimeData *data, Qt::DropAction action, int row, int column,
+                              const QModelIndex &parent) override;
 
     void abortProgress(const QModelIndex &index);
 
diff --git a/core/src/Variable/VariableController.cpp b/core/src/Variable/VariableController.cpp
index d1865e6..5377542 100644
--- a/core/src/Variable/VariableController.cpp
+++ b/core/src/Variable/VariableController.cpp
@@ -12,6 +12,7 @@
 #include <Data/VariableRequest.h>
 #include <Time/TimeController.h>
 
+#include <QDataStream>
 #include <QMutex>
 #include <QThread>
 #include <QUuid>
@@ -277,6 +278,46 @@ void VariableController::deleteVariables(
     }
 }
 
+QByteArray
+VariableController::mimeDataForVariables(const QList<std::shared_ptr<Variable> > &variables) const
+{
+    auto encodedData = QByteArray{};
+
+    QVariantList ids;
+    for (auto &var : variables) {
+        auto itVar = impl->m_VariableToIdentifierMap.find(var);
+        if (itVar == impl->m_VariableToIdentifierMap.cend()) {
+            qCCritical(LOG_VariableController())
+                << tr("Impossible to find the data for an unknown variable.");
+        }
+
+        ids << itVar->second.toByteArray();
+    }
+
+    QDataStream stream{&encodedData, QIODevice::WriteOnly};
+    stream << ids;
+
+    return encodedData;
+}
+
+QList<std::shared_ptr<Variable> >
+VariableController::variablesForMimeData(const QByteArray &mimeData) const
+{
+    auto variables = QList<std::shared_ptr<Variable> >{};
+    QDataStream stream{mimeData};
+
+    QVariantList ids;
+    stream >> ids;
+
+    for (auto id : ids) {
+        auto uuid = QUuid(id.toByteArray());
+        auto var = impl->findVariable(uuid);
+        variables << var;
+    }
+
+    return variables;
+}
+
 std::shared_ptr<Variable>
 VariableController::createVariable(const QString &name, const QVariantHash &metadata,
                                    std::shared_ptr<IDataProvider> provider) noexcept
diff --git a/core/src/Variable/VariableModel.cpp b/core/src/Variable/VariableModel.cpp
index e5f9947..6e13efd 100644
--- a/core/src/Variable/VariableModel.cpp
+++ b/core/src/Variable/VariableModel.cpp
@@ -1,11 +1,14 @@
 #include <Variable/Variable.h>
+#include <Variable/VariableController.h>
 #include <Variable/VariableModel.h>
 
 #include <Common/DateUtils.h>
+#include <Common/MimeTypesDef.h>
 #include <Common/StringUtils.h>
 
 #include <Data/IDataSeries.h>
 
+#include <QMimeData>
 #include <QSize>
 #include <unordered_map>
 
@@ -66,14 +69,16 @@ struct VariableModel::VariableModelPrivate {
     /// Variables created in SciQlop
     std::vector<std::shared_ptr<Variable> > m_Variables;
     std::unordered_map<std::shared_ptr<Variable>, double> m_VariableToProgress;
+    VariableController *m_VariableController;
 
     /// Return the row index of the variable. -1 if it's not found
     int indexOfVariable(Variable *variable) const noexcept;
 };
 
-VariableModel::VariableModel(QObject *parent)
+VariableModel::VariableModel(VariableController *parent)
         : QAbstractTableModel{parent}, impl{spimpl::make_unique_impl<VariableModelPrivate>()}
 {
+    impl->m_VariableController = parent;
 }
 
 void VariableModel::addVariable(std::shared_ptr<Variable> variable) noexcept
@@ -255,6 +260,57 @@ QVariant VariableModel::headerData(int section, Qt::Orientation orientation, int
     return QVariant{};
 }
 
+Qt::ItemFlags VariableModel::flags(const QModelIndex &index) const
+{
+    return QAbstractTableModel::flags(index) | Qt::ItemIsDragEnabled | Qt::ItemIsDropEnabled;
+}
+
+Qt::DropActions VariableModel::supportedDropActions() const
+{
+    return Qt::MoveAction;
+}
+
+Qt::DropActions VariableModel::supportedDragActions() const
+{
+    return Qt::CopyAction | Qt::MoveAction;
+}
+
+QStringList VariableModel::mimeTypes() const
+{
+    return {MIME_TYPE_VARIABLE_LIST};
+}
+
+QMimeData *VariableModel::mimeData(const QModelIndexList &indexes) const
+{
+    auto mimeData = new QMimeData;
+
+    QList<std::shared_ptr<Variable> > variableList;
+
+    for (const auto &index : indexes) {
+        auto variable = impl->m_Variables.at(index.row());
+        if (variable.get() && index.isValid()) {
+            variableList << variable;
+        }
+    }
+
+    auto encodedData = impl->m_VariableController->mimeDataForVariables(variableList);
+    mimeData->setData(MIME_TYPE_VARIABLE_LIST, encodedData);
+
+    return mimeData;
+}
+
+bool VariableModel::canDropMimeData(const QMimeData *data, Qt::DropAction action, int row,
+                                    int column, const QModelIndex &parent) const
+{
+    return false;
+}
+
+bool VariableModel::dropMimeData(const QMimeData *data, Qt::DropAction action, int row, int column,
+                                 const QModelIndex &parent)
+{
+    return false;
+}
+
 void VariableModel::abortProgress(const QModelIndex &index)
 {
     if (auto variable = impl->m_Variables.at(index.row())) {