Initial commit

Lts/Plugins/Evaluations/Gui/SchemaEdit/evl_edit_treemodel_schema.cpp

@@ -0,0 +1,314 @@
+/**
+  * @RDE_HEADER@
+  * $Id$
+  * $HeadURL$
+  * $Author$
+  *
+  * Created by Philipp Last, NLSE, 05.01.2018
+  */
+
+#include "evl_edit_treemodel_schema.h"
+#include <ngltools/ngltools_dtg.h>
+
+#include <QJsonArray>
+#include <QJsonObject>
+
+namespace Evaluations
+{
+TreeModelSchema::TreeModelSchema(QObject *parent, NGL::INGLLogger &logger)
+  : TreeModelGeneral(parent, logger)
+{
+  qRegisterMetaTypeStreamOperators<Evaluations::ElementData>();
+}
+
+
+void TreeModelSchema::resetModel()
+{
+  TreeModelGeneral::resetModel();
+  m_elementId = 1;
+  m_cntComments = 0;
+  m_cntSentComments = 0;
+}
+
+
+bool TreeModelSchema::dropMimeData(const QMimeData *data, Qt::DropAction action,
+                                   int row, int column, const QModelIndex &parent)
+{
+  /*
+   * Prevent dropping/appending elements to other elements!
+   * Only dropping to folders is allowed!
+   */
+  ElementData p = getSelectedIndexDataAsElement(parent);
+
+  if (!p.isFolder())
+  {
+    emit signalTriedToDropFolderToElement();
+    return false;
+  }
+
+  bool success = TreeModelGeneral::dropMimeData(data, action, row, column, parent);
+
+  if (success)
+  {
+    /*
+     * User may have moved a folder with some elements into another folder.
+     * Hence, we have to update the parent ids.
+     */
+    int parentId = getSelectedIndexDataAsElement(parent).m_id;
+    iterateModel(parent, [this, parentId](const QModelIndex & idx, const QVariant & v)
+    {
+      Evaluations::ElementData e = v.value<Evaluations::ElementData>();
+      e.m_parentId = parentId;
+      updateElementData(idx, e);
+    }, false);
+
+    refreshAllIcons();
+    emit signalSuccessfulDropPerformed(parent);
+  }
+
+  return success;
+}
+
+
+QVariant TreeModelSchema::data(const QModelIndex &index, int role) const
+{
+  QVariant v = TreeModelGeneral::data(index, role);
+
+  if (v.isNull())
+    return v;
+
+  if (v.canConvert<Evaluations::ElementData>())
+  {
+    Evaluations::ElementData e = v.value<Evaluations::ElementData>();
+    return e.getText();
+  }
+  else if (v.canConvert<Evaluations::ItemComment>())
+  {
+    Evaluations::ItemComment c = v.value<Evaluations::ItemComment>();
+    QString str(DTG::toStringMSec(c.getTimeStamp().toMSecsSinceEpoch()));
+    str += QLatin1Char(' ');
+    str += c.getText();
+    return str;
+  }
+
+  return v;
+}
+
+
+ElementData TreeModelSchema::getSelectedIndexDataAsElement(const QModelIndex &idx) const
+{
+  if (!idx.isValid())
+    ElementData();
+
+  QVariant v = TreeModelGeneral::dataAsVariant(idx);
+  return v.value<Evaluations::ElementData>();
+}
+
+
+QVector<QString> TreeModelSchema::getSelectedIndexHierarchy(const QModelIndex &idx) const
+{
+  if (!idx.isValid())
+    QString();
+
+  QVector<QString> vecHierarchy;
+
+  QModelIndex iterIdx = idx;
+
+  while (iterIdx.parent() != QModelIndex())
+  {
+    Evaluations::ElementData e = getSelectedIndexDataAsElement(iterIdx);
+    vecHierarchy.push_front(e.getText());
+
+    iterIdx = iterIdx.parent();
+  }
+
+  Evaluations::ElementData e = getSelectedIndexDataAsElement(iterIdx);
+  vecHierarchy.push_front(e.getText());
+
+  return vecHierarchy;
+}
+
+
+QJsonDocument TreeModelSchema::toJsonDocument() const
+{
+  QJsonArray arr;
+
+  QModelIndex root;
+  iterateModel(root, [this, &arr](const QModelIndex & idx, const QVariant & v)
+  {
+    Q_UNUSED(idx);
+    Evaluations::ElementData e = v.value<Evaluations::ElementData>();
+    INFO_LOG(m_logger) << "toJsonDocument: Name: " << e.getText() << ", elementId: " << e.m_id << ", parent id: " << e.m_parentId;
+
+    QJsonObject obj;
+    obj[QStringLiteral("id")] = e.m_id;
+    obj[QStringLiteral("parentId")] = e.m_parentId;
+    obj[QStringLiteral("text")] = e.getText();
+    obj[QStringLiteral("state")] = int(e.getState());
+
+    arr.append(obj);
+  });
+
+  QJsonDocument doc(arr);
+  return doc;
+}
+
+
+bool TreeModelSchema::fromQJsonDocument(const QJsonDocument &doc)
+{
+  resetModel();
+
+  const QJsonArray arr(doc.array());
+
+  if (!arr.size())
+    return false;
+
+  QHash<int, QPersistentModelIndex> mapId2Index;
+
+  for (const QJsonValue &val : arr)
+  {
+    const QJsonObject obj = val.toObject();
+
+    ElementData d(obj[QStringLiteral("text")].toString());
+    d.m_id = obj[QStringLiteral("id")].toInt();
+    d.m_parentId = obj[QStringLiteral("parentId")].toInt();
+    d.switchState((ElementData::State)obj[QStringLiteral("state")].toInt());
+
+    QPersistentModelIndex parent = (d.m_parentId == 0) ? QPersistentModelIndex() : mapId2Index[d.m_parentId];
+    QModelIndex newIdx = TreeModelGeneral::insertRow(parent, QVariant::fromValue(d), d.getIcon());
+
+    mapId2Index[d.m_id] = newIdx;
+
+    /*
+     * Get current max element id. This is required since
+     * the user will most likely add new elements to the loaded treeview / element list
+     */
+    m_elementId = std::max(m_elementId, d.m_id);
+    INFO_LOG(m_logger) << "fromQJsonDocument: " << d.getText() << ", own id = " << d.m_id << ", parent id = " << d.m_parentId;
+  }
+
+  ++m_elementId;
+  return true;
+}
+
+
+QModelIndex TreeModelSchema::insertComment(const QModelIndex &parent, const ItemComment &data)
+{
+  QModelIndex idx = TreeModelGeneral::insertRow(parent, QVariant::fromValue(data), data.getIcon());
+
+  if (idx.isValid())
+  {
+    ++m_cntComments;
+
+    if (data.isSent())
+      ++m_cntSentComments;
+
+    refreshElementDataIconState(parent);
+  }
+
+  return idx;
+}
+
+
+void TreeModelSchema::insertComment(const ItemComment &data)
+{
+  QModelIndex root = QModelIndex();
+  iterateModel(root, [&data, this](const QModelIndex & idx, const QVariant & v)
+  {
+    if (v.canConvert<Evaluations::ElementData>())
+    {
+      ElementData e = v.value<Evaluations::ElementData>();
+      if (e.m_id == data.elementId())
+        TreeModelSchema::insertComment(idx, data);
+    }
+  });
+}
+
+
+bool TreeModelSchema::updateComment(const QModelIndex &index, const ItemComment &data)
+{
+  bool success = TreeModelGeneral::updateRow(index, QVariant::fromValue(data), data.getIcon());
+
+  if (success)
+  {
+    if (data.isSent())
+      ++m_cntSentComments;
+
+    refreshElementDataIconState(index.parent());
+  }
+
+  return success;
+}
+
+
+void TreeModelSchema::removeComment(const QModelIndex &idx)
+{
+  if (!idx.isValid())
+    return;
+
+  --m_cntComments;
+  QModelIndex idxParentElement = idx.parent();
+  removeRow(idx);
+  refreshElementDataIconState(idxParentElement);
+}
+
+
+QModelIndex TreeModelSchema::insertElementData(const QModelIndex &parent, const ElementData &data)
+{
+  ElementData d(data);
+  d.m_id = m_elementId++;
+  d.m_parentId = !parent.isValid() ? 0 : getSelectedIndexDataAsElement(parent).m_id;
+
+  return TreeModelGeneral::insertRow(parent, QVariant::fromValue(d), d.getIcon());
+}
+
+
+bool TreeModelSchema::updateElementData(const QModelIndex &index, const ElementData &data)
+{
+  return TreeModelGeneral::updateRow(index, QVariant::fromValue(data), data.getIcon());
+}
+
+
+void TreeModelSchema::refreshElementDataIconState(const QModelIndex &idx)
+{
+  ElementData e = getSelectedIndexDataAsElement(idx);
+
+  if (e.isFolder())
+    return;
+
+  if (!rowCount(idx))
+    e.switchState(ElementData::State::ElementWithoutComments);
+  else
+  {
+    bool hasUnsentComments = false;
+    iterateModel(idx, [&hasUnsentComments](const QModelIndex & idx, const QVariant & v)
+    {
+      Q_UNUSED(idx);
+      const Evaluations::ItemComment d = v.value<Evaluations::ItemComment>();
+      if (!d.isSent())
+        hasUnsentComments = true;
+    });
+
+    e.switchState(hasUnsentComments ? ElementData::State::ElementWithUnsentComments : ElementData::State::ElementAllCommentsSent);
+  }
+
+  TreeModelGeneral::updateRow(idx, QVariant::fromValue(e), e.getIcon());
+}
+
+
+void TreeModelSchema::refreshAllIcons()
+{
+  QModelIndex root;
+  iterateModel(root, [this](const QModelIndex & idx, const QVariant & v)
+  {
+    Evaluations::ElementData d = v.value<Evaluations::ElementData>();
+    d.refreshIcon();
+
+    NGLTreeItem *item = privateData(idx);
+    item->setIcon(d.getIcon());
+  });
+
+  emit dataChanged(QModelIndex(), QModelIndex());
+}
+} // namespace Evaluations
+