/**
  * @RDE_HEADER@
  * CustomView for AES Mine Single Virtual
  * $Id$
  * $HeadURL$
  * $Author$
  * Created by Hanna E. Rathje, SLSE, 2023-07-14
  */
#pragma once

#include <Viewer/LtsDataView/abstract_display_widget.h>

namespace Ui
{
class CustomViewBoobyTrap;
}

namespace AreaEffectSimulation
{
class BoobyTrap;
class CustomViewBoobyTrap final : public LtsDataView::AbstractDisplayWidget
{
  Q_OBJECT
public:
  explicit CustomViewBoobyTrap(QWidget *parent = nullptr);
  ~CustomViewBoobyTrap();
  bool init(const QModelIndex &idx) override final;

private:
  bool calculateIcon(const AreaEffectSimulation::BoobyTrap &bt, const QColor &color);

  Ui::CustomViewBoobyTrap *m_ui;

};
} // namespace AreaEffectSimulation