Initial commit

Lts/Plugins/ExerciseManagement/Gui/ExerciseEdit/exm_phase_model.cpp

@@ -0,0 +1,104 @@
+/**
+  * @RDE_HEADER@
+  * Exercise Phase Overview Model
+  * $Id$
+  * $HeadURL$
+  * $Author$
+  * Created by Christian Ehrlicher, SLSE, 2017-04-11
+  */
+
+#include "exm_phase_model.h"
+
+namespace ExerciseManagement
+{
+
+//---------------------------------
+// ExercisePhaseModel
+//---------------------------------
+ExercisePhaseModel::ExercisePhaseModel(QObject *parent)
+  : LtsWidgets::LtsAbstractTableModel<ExerciseManagement::ExercisePhase>({tr("Phase"),
+                                                                          tr("Start"),
+                                                                          tr("End"),
+                                                                          tr("Description"),
+                                                                          tr("Exercise-Id")},
+                                                                          parent)
+  , m_exmDtc(nullptr)
+  , m_logger(getLogger("ExercisePhaseModel"))
+{}
+
+
+bool ExercisePhaseModel::initModel(int exerciseId, const std::shared_ptr<ExerciseManagement::Interface::Dtc> &dtc)
+{
+  m_exerciseId = exerciseId;
+  m_exmDtc = dtc;
+
+  connect(m_exmDtc.get(), &ExerciseManagement::Interface::Dtc::exercisePhaseChanged,
+          this, &ExercisePhaseModel::onDataChanged);
+
+  // insert Model
+  m_exmDtc->exercisePhases([this](const ExercisePhase &item) { insertRow(item); });
+  INFO_LOG(m_logger) << "initModel - #exercise phases: " << m_vecData.size();
+  return true;
+}
+
+
+QVariant ExercisePhaseModel::data(const QModelIndex &index, int role) const
+{
+  if (!index.isValid() || role != Qt::DisplayRole || index.row() >= m_vecData.size())
+    return QVariant();
+
+  if (role == Qt::DisplayRole)
+  {
+    const ExerciseManagement::ExercisePhase &exPhase = m_vecData.at(index.row());
+
+    switch (index.column())
+    {
+    case Name:
+      return QString::fromStdString(exPhase.name());
+    case Start:
+      return QDateTime::fromMSecsSinceEpoch(exPhase.beginTime());
+    case End:
+      return QDateTime::fromMSecsSinceEpoch(exPhase.endTime());
+    case Description:
+      return QString::fromStdString(exPhase.description());
+    case ExerciseId:
+      return (exPhase.exerciseId());
+    }
+  }
+
+  return QVariant();
+}
+
+//
+// ExercisePhaseProxyModel
+//
+void ExercisePhaseProxyModel::setSourceModel(QAbstractItemModel *sourceModel)
+{
+  m_sourceModel = qobject_cast<ExercisePhaseModel*>(sourceModel);
+  LtsWidgets::LtsColumnFilterProxyModel::setSourceModel(sourceModel);
+}
+
+bool ExercisePhaseProxyModel::lessThan(const QModelIndex &source_left, const QModelIndex &source_right) const
+{
+  if (!m_sourceModel ||
+      source_left.column() != source_right.column() ||
+      !source_left.isValid() ||
+      !source_right.isValid())
+    return LtsWidgets::LtsColumnFilterProxyModel::lessThan(source_left, source_right);
+  const auto &dataLeft = m_sourceModel->m_vecData.at(source_left.row());
+  const auto dataRight = m_sourceModel->m_vecData.at(source_right.row());
+  switch (source_left.column())
+  {
+  case ExercisePhaseModel::Start:
+    return dataLeft.beginTime() < dataRight.beginTime();
+  case ExercisePhaseModel::End:
+    return dataLeft.endTime() < dataRight.endTime();
+  case ExercisePhaseModel::Name:
+  case ExercisePhaseModel::Description:
+  case ExercisePhaseModel::ExerciseId:
+    break;
+  }
+  return LtsWidgets::LtsColumnFilterProxyModel::lessThan(source_left, source_right);
+}
+
+} // namespace ExerciseManagement