/**
  * @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 CustomViewNContamination;
}

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

private:
  bool calculateIcon(const AreaEffectSimulation::NContamination &nContamination, const QColor &color);

  Ui::CustomViewNContamination *m_ui;

};
} // namespace AreaEffectSimulation