#pragma once

#include <LtsWidgets/lts_abstract_table_model.h>
#include <AssetManagement/asm_dtc.h>
#include <AssetManagement/logger.h>


namespace AssetManagement
{

class AsmTrainingAreaDialogModel
    : public LtsWidgets::LtsAbstractTableModel<AssetManagement::TrainingArea>
{
  Q_OBJECT

public:
  enum Columns
  {
    Id,
    Name,
    BottomLeft_x,
    BottomLeft_y,
    Height,
    Width,
    Image,
  };

  explicit AsmTrainingAreaDialogModel(QObject *parent = nullptr);
  ~AsmTrainingAreaDialogModel();
  bool initModel(const std::shared_ptr<AssetManagement::Interface::Dtc> &asmDtc, bool hideRemoved);

  QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override final;


private:
  bool m_hideRemoved;
  std::shared_ptr<Interface::Dtc> m_asmDtc;
  LOGGER_T m_logger;
};

}// namespace AssetManagement