/**
  * @RDE_HEADER@
  * HMI Plugin for AreaEffectSimulation - Booby Trap Overview Dialog Model
  * $Id$
  * $HeadURL$
  * $Author$
  *
  * Created by Hanna E. Rathje, SLSE, 2023-07-14
  */

#pragma once

#include <QSet>
#include <Plugins/AreaEffectSimulation/Gui/Base/aes_base_model.h>
#include <ProjectDataManagement/pdm_enums.hpp>

namespace LtsSession
{
class LtsSession;
}

namespace AreaEffectSimulation
{


/************  Abstract Class for Mine Single Virtual ***************/
class BoobyTrapModel : public AesGuiBaseModel<BoobyTrap>
{
  Q_OBJECT

public:
  enum AdditionalColumns {ColTriggerRadius = BcCount,
                          ColImpactRadius  = BcCount + 1,
                          ColRotationAngle = BcCount + 2,
                          ColCentralAngle  = BcCount + 3};

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

  explicit BoobyTrapModel(QObject *parent = nullptr)
    : AesGuiBaseModel<BoobyTrap>(parent, {tr("Trigger Radius"), tr("Impact Radius"), tr("Rotation Angle"), tr("Central Angle")})
  {}  

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

} //namespace AreaEffectSimulation