/**
  * @RDE_HEADER@
  * CustomView for AES NContamination
  * $Id$
  * $HeadURL$
  * $Author$
  * Created by Mike Wille, SLSE, 2016-04-08
  */
#pragma once

#include <Viewer/LtsDataView/abstract_display_widget.h>

namespace Ui
{
class CustomViewDangerArea;
}

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

private:
  bool calculateIcon(const AreaEffectSimulation::DangerArea &area, const QColor &color);

  Ui::CustomViewDangerArea *m_ui;
};
} // namespace AreaEffectSimulation