/**
  * @RDE_HEADER@
  * HMI Plugin for AreaEffectSimulation - Functional Area Overview Dialog Model
  * $Id$
  * $HeadURL$
  * $Author$
  *
  * Created by Malte Diers, SLSE, 2017-04-06
  */

#pragma once

#include <Plugins/AreaEffectSimulation/Gui/Base/aes_base_model.h>

namespace LtsSession
{
class LtsSession;
}

namespace AreaEffectSimulation
{

class FunctionalAreasModel : public AesGuiBaseModel<FunctionalArea>
{
  Q_OBJECT

public:
  enum AdditionalColumns {ColType = BcCount};
  explicit FunctionalAreasModel(QObject *parent = nullptr)
    : AesGuiBaseModel<FunctionalArea>(parent, {tr("Type")})
    , m_gearClass(ProjectDataManagement::GearClass::FunctionalArea)
  {}

  bool initModel(LtsSession::LtsSession *session) override;

  ProjectDataManagement::GearClass gearClass() const { return m_gearClass; }

  QVariant data(const QModelIndex &, int) const override final;

protected:
  ProjectDataManagement::GearClass m_gearClass;
};



} //namespace AreaEffectSimulation