Initial commit

Lts/Plugins/AreaEffectSimulation/Gui/TechnicalArea/aes_technical_areas_model.cpp

@@ -0,0 +1,66 @@
+/**
+  * @RDE_HEADER@
+  * AreaEffectManangement - Technical Area Overview Dialog Model
+  * $Id$
+  * $HeadURL$
+  * $Author$
+  * Created by Philipp Dassow, SLEA, 2017-06-09
+  */
+
+#include "aes_technical_areas_model.h"
+#include <Plugins/ProjectDataManagement/pdm_data_plugin_interface.h>
+#include <QLocale>
+#include <QSet>
+
+namespace AreaEffectSimulation
+{
+
+TechnicalAreasModel::TechnicalAreasModel(QObject *parent)
+  : AesGuiBaseModel<TechnicalArea>(parent)
+{}
+
+bool TechnicalAreasModel::initModel(LtsSession::LtsSession *session)
+{
+  if (!AesGuiBaseModel::initModel(session))
+    return false;
+
+  connect(m_aesDtc.get(), &Interface::Dtc::technicalAreaChanged,
+          this, &TechnicalAreasModel::onDataChanged);
+
+  connect(m_aesDtc.get(), &Interface::Dtc::technicalAreaChanged,
+          this, &TechnicalAreasModel::dataChanged);
+
+  m_aesDtc->technicalAreas([this](const TechnicalArea & item) { insertRow(item); });
+  return true;
+}
+
+QVariant TechnicalAreasModel::headerData(int section, Qt::Orientation orientation, int role) const
+{
+  if (section == BcChargeType && orientation == Qt::Horizontal && role == Qt::DisplayRole)
+    return tr("Type");
+  if (section == BcEffectState && orientation == Qt::Horizontal && role == Qt::DisplayRole)
+    return tr("State");
+  return AesGuiBaseModel::headerData(section, orientation, role);
+}
+
+QVariant TechnicalAreasModel::data(const QModelIndex &idx, int role) const
+{
+  if (!indexIsValid(idx))
+    return QVariant();
+  // misissue ChargeType for Type :)
+  if (role == Qt::DisplayRole && idx.column() == BcChargeType)
+  {
+    const TechnicalArea &ta = m_vecData.at(idx.row());
+    return AreaEffectSimulation::toString(ta.type());
+  }
+  if (role == Qt::DisplayRole && idx.column() == BcEffectState)
+  {
+    const TechnicalArea &ta = m_vecData.at(idx.row());
+    return AreaEffectSimulation::toString(ta.effectStateId());
+  }
+  return AesGuiBaseModel::data(idx, role);
+}
+
+
+
+} // namespace AreaEffectSimulation