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

#include "aes_explosion.h"
#include "ui_aes_explosion.h"
#include "aes_explosions_area_model.h"
#include "aes_explosions_single_model.h"

#include <Services/AreaEffectSimulation/Utils/polygons.h>
#include <ProjectDataManagement/pdm_data_plugin_interface.h>

namespace AreaEffectSimulation
{
CustomViewExplosion::CustomViewExplosion(LtsSession::LtsSession *session, QWidget *parent)
  : AbstractDisplayWidget(parent)
  , m_session(session)
  , m_ui(new Ui::CustomViewExplosion)
{
  m_ui->setupUi(this);
}

CustomViewExplosion::~CustomViewExplosion()
{
  delete m_ui;
}

bool CustomViewExplosion::init(const QModelIndex &idx)
{
  if (!idx.isValid())
    return false;

  m_ui->lbName->setText(rowData(idx, AreaEffectSimulation::ExplosionsAreaModel::BcName).toString());
  m_ui->lbEffectState->setText(rowData(idx, AreaEffectSimulation::ExplosionsAreaModel::BcEffectState).toString());
  m_ui->lbEffectStart->setText(rowData(idx, AreaEffectSimulation::ExplosionsAreaModel::BcEffectStart).toString());
  m_ui->lbChargeType->setText(rowData(idx, AreaEffectSimulation::ExplosionsAreaModel::BcChargeType).toString());

  QString durationString = rowData(idx, AreaEffectSimulation::ExplosionsAreaModel::BcEffectDuration).toString();
  if (!durationString.isEmpty() && durationString != QLatin1String("0"))
  {
    m_ui->lbEffectDuration->setText(durationString);
  }
  else
  {
    m_ui->label_4->hide();
    m_ui->lbEffectDuration->hide();
  }
  m_ui->lblIcon->clear();
  m_ui->lblIcon->setAutoFillBackground(true);

  const AesItemSortFilterProxyModel *sfpm = qobject_cast<const AesItemSortFilterProxyModel*>(idx.model());
  if (!sfpm)
    return false;
  const auto srcIdx = sfpm->mapToSource(idx);
  const ExplosionsAreaModel *aesModelArea = qobject_cast<const ExplosionsAreaModel*>(sfpm->sourceModel());
  if (aesModelArea)
  {
    ExplosionArea ea = aesModelArea->getRowData(srcIdx.row());
    calculateIcon(ea, sfpm->partyColor(idx.row()));
    return true;
  }
  else
  {
    const ExplosionsSingleModel *aesModelSingle = qobject_cast<const ExplosionsSingleModel*>(sfpm->sourceModel());
    if (aesModelSingle)
    {
      ExplosionSingle es = aesModelSingle->getRowData(srcIdx.row());
      calculateIcon(es, sfpm->partyColor(idx.row()));
      return true;
    }
  }
  return false;
}

bool CustomViewExplosion::calculateIcon(const ExplosionArea &area, const QColor &color)
{
  Database::AesExplosionArea ea;
  ea.impactPosition = QPointF(area.area().cornerBottomLeft().lon(),
                              area.area().cornerBottomLeft().lat());
  ea.rotationAngle = area.area().rotation();
  ea.impactWidth = area.area().width();
  ea.impactLength = area.area().height(); // Todo: Correct?

  QPolygonF poly = AreaEffectSimulation::Polygons::explosionArea(ea);
  Qt::BrushStyle bStyle = PixmapLabel::checkForBrushStyle(area.effectStateId());
  const auto pix = Polygons::pixmapFromPolygon(PixmapLabel::geographic2Utm(poly), color, ICON_RESOLUTION, bStyle);
  m_ui->lblIcon->setPixmap(pix);

  return true;
}

bool CustomViewExplosion::calculateIcon(const ExplosionSingle &single, const QColor &color)
{
  Database::AesExplosionSingle es;
  es.impactPosition = QPointF(single.impactPosition().lon(), single.impactPosition().lat());

  std::shared_ptr<ProjectDataManagement::Interface::Dtc> pdmDtc;
  if (!m_session->getDtc<ProjectDataManagement::DataInterface>(pdmDtc))
    return false;

  double radius = pdmDtc->impactRadius(single.chargeStruct().chargeTypeId(),
                                       single.chargeStruct().igniterTypeId(),
                                       single.chargeStruct().igniterFunctionId());
  QPolygonF poly = AreaEffectSimulation::Polygons::explosionSingle(es, radius);
  Qt::BrushStyle bStyle = PixmapLabel::checkForBrushStyle(single.effectStateId());
  const auto pix = Polygons::pixmapFromPolygon(PixmapLabel::geographic2Utm(poly), color, ICON_RESOLUTION, bStyle);
  m_ui->lblIcon->setPixmap(pix);

  return true;
}
} // namespace AreaEffectSimulation