Initial commit

Lts/Tools/DBAdministration/local_databases_model.cpp

@@ -0,0 +1,165 @@
+/**
+  * @RDE_HEADER@
+  * PresentationManangement - Script Overview Dialog Model
+  * $Id$
+  * $HeadURL$
+  * $Author$
+  *
+  * Created by Mike Wille, SLSE, 2017-03-15
+  */
+
+#include "local_databases_model.h"
+
+#include <ngltools/ngltools_dtg.h>
+#include <ngldb/nsqlquery.h>
+
+#include <QDateTime>
+#include <QPalette>
+
+namespace DBAdministration
+{
+
+LocalDatabasesModel::LocalDatabasesModel(QObject *parent)
+  : QAbstractTableModel(parent)
+, m_logger(getLogger("LocalDatabasesModel"))
+{}
+
+
+LocalDatabasesModel::~LocalDatabasesModel()
+{}
+
+
+bool LocalDatabasesModel::initModel(const QSqlDatabase &db)
+{
+  beginResetModel();
+  m_vecData.clear();
+  endResetModel();
+
+  NSqlQuery q(db);
+
+  QString stmt = QStringLiteral("SELECT id, name, db_name, begin_time, end_time FROM local_databases");
+
+  if (!q.exec(stmt))
+    return false;
+
+  while (q.next())
+  {
+    LocalExercise e;
+    e.id        = q.value(0).toInt();
+    e.name      = q.value(1).toString();
+    e.dbName    = q.value(2).toString();
+    e.startTime = q.value(3).toLongLong();
+    e.endTime   = q.value(4).toLongLong();
+    insertRow(e);
+  }
+
+  INFO_LOG(m_logger) << "initModel - #exercises: " << m_vecData.size();
+
+  return true;
+}
+
+
+QVariant LocalDatabasesModel::data(const QModelIndex &index, int role) const
+{
+  if (role != Qt::DisplayRole || !index.isValid() || index.row() >= m_vecData.size())
+    return QVariant();
+
+  const LocalExercise &e = m_vecData.at(index.row());
+
+  switch (index.column())
+  {
+  case Id:
+    return e.id;
+  case Name:
+    return QString(e.name);
+  case DbName:
+    return QString(e.dbName);
+  case StartTime:
+    return DTG::toStringMSec(e.startTime);
+  case EndTime:
+    return DTG::toStringMSec(e.endTime);
+  }
+
+  return QVariant();
+}
+
+int LocalDatabasesModel::rowCount(const QModelIndex &parent) const
+{
+  return parent.isValid() ? 0 : m_vecData.size();
+}
+
+int LocalDatabasesModel::columnCount(const QModelIndex &parent) const
+{
+  return parent.isValid() ? 0 : EndTime + 1;
+}
+
+QVariant LocalDatabasesModel::headerData(int section, Qt::Orientation orientation, int role) const
+{
+  if (orientation == Qt::Horizontal
+      && role == Qt::DisplayRole)
+  {
+    Columns col = static_cast<Columns>(section);
+    switch (col)
+    {
+    case Id:
+      return tr("Id");
+    case Name:
+      return tr("Name");
+    case DbName:
+      return tr("Database Name");
+    case StartTime:
+      return tr("Start Time");
+    case EndTime:
+      return tr("End Time");
+    }
+  }
+  return QAbstractTableModel::headerData(section, orientation, role);
+}
+
+LocalExercise LocalDatabasesModel::getRowData(int row) const
+{
+  if (row < 0 || row >= m_vecData.size())
+    return LocalExercise();
+  return m_vecData.at(row);
+}
+
+bool LocalDatabasesModel::insertRow(const LocalExercise &item)
+{
+  beginInsertRows(QModelIndex(), m_vecData.size(), m_vecData.size());
+  m_vecData.append(item);
+  endInsertRows();
+  return true;
+}
+
+void LocalDatabasesProxyModel::setSourceModel(QAbstractItemModel *sourceModel)
+{
+  m_sourceModel = qobject_cast<LocalDatabasesModel*>(sourceModel);
+  QSortFilterProxyModel::setSourceModel(sourceModel);
+}
+
+bool LocalDatabasesProxyModel::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 QSortFilterProxyModel::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 LocalDatabasesModel::StartTime:
+    return dataLeft.startTime < dataRight.startTime;
+  case LocalDatabasesModel::EndTime:
+    return dataLeft.endTime < dataRight.endTime;
+  case LocalDatabasesModel::Id:
+  case LocalDatabasesModel::Name:
+  case LocalDatabasesModel::DbName:
+
+    break;
+  }
+  return QSortFilterProxyModel::lessThan(source_left, source_right);
+}
+
+
+} // namespace DBAdministration