qt-migration-legatus/qt6-after/qt6-migrated-legatus/Lts/Services/AreaEffectSimulation/handler_off_route_mine.cpp

/**
  * Handler for all off route mines
  * $Id$
  * $HeadURL$
  * $Author$
  * Created by Hanna E. Rathje, SLSE, 2023-07-13
  */

#include "handler_off_route_mine.h"

namespace AreaEffectSimulation
{

HandlerOffRouteMine::HandlerOffRouteMine(MainHandler *mainHandler)
  : HandlerBaseTmpl<ObjectOffRouteMine>(mainHandler, QStringLiteral("OffRouteMine"), s_areaCount[static_cast<int>(AesType::AtOffRouteMine)], s_areaCountCentral[static_cast<int>(AesType::AtOffRouteMine)])
  , Interface::Fts::OffRouteMineModifyRequestHandler(mainHandler->dds()->fts())
  , Interface::Fts::OffRouteMineRemoveRequestHandler(mainHandler->dds()->fts())
  , Interface::Fts::OffRouteMineTriggerRequestHandler(mainHandler->dds()->fts())
{}

bool HandlerOffRouteMine::init()
{
  bool bRet = true;
  auto cb = [this, &bRet] (const Database::AesOffRouteMine &data)
  {
    ObjectOffRouteMine *ifs(new ObjectOffRouteMine(m_mainHandler, this));
    connect(ifs, &ObjectOffRouteMine::effectStateChanged, this, &HandlerOffRouteMine::onEffectStateChanged);
    Core::FunctionReplyObject ret = ifs->createObject(data);
    if (ret.error())
    {
      delete ifs;
      ERROR_LOG(m_logger) << ret();
      bRet = false;
      return;
    }
    if (totalAreasCentral() >= s_maxActiveAreasCentral[static_cast<int>(AesType::AtOffRouteMine)])
    {
      ERROR_LOG(m_logger) << "More active off route mines in database than allowed - ignoring the rest!";
      ifs->removeObject();
      delete ifs;
      return;
    }
    m_objectMap.insert(ifs->id(), ifs);
    connect(ifs, &ObjectOffRouteMine::newAiv,             this, &HandlerOffRouteMine::newAiv);
    connect(ifs, &ObjectOffRouteMine::removeAiv,          this, &HandlerOffRouteMine::removeAiv);
    if (ifs->isObjectInTrainingArea())
    {
      ++m_areaCount;
      DEBUG_LOG(m_logger) << "s_areaCount[AesType::AtBCContamination] = " << m_areaCount;
    }
  };
  if (!m_mainHandler->db()->select(cb) || !bRet)
  {
    ERROR_LOG(m_logger) << "Error reading data from database!";
    return true;
  }

  DEBUG_LOG(m_logger) << "m_exeObjMap.size() = " << m_objectMap.size();

  return true;
}

int HandlerOffRouteMine::totalAreas() const
{
  return s_areaCount[static_cast<int>(AesType::AtOffRouteMine)];
}

int HandlerOffRouteMine::totalAreasCentral() const
{
  return s_areaCountCentral[static_cast<int>(AesType::AtOffRouteMine)];
}

static void fillData(const OffRouteMineModifyRequest &src, Database::AesOffRouteMine &dst)
{
  if (src.id < 0) // is the id given as negative? set id -> used by OC-laptop
    dst.id = src.id;

  dst.name = QString::fromStdString(src.name);
  dst.effectStart = src.effectStart.is_set() && src.effectStart.get() > 0 ? QDateTime::fromMSecsSinceEpoch(src.effectStart.get()) : QDateTime();
  dst.effectDuration = src.effectDuration;
  dst.impactPosition.rx() = src.area.cornerBottomLeft.lon;
  dst.impactPosition.ry() = src.area.cornerBottomLeft.lat;
  dst.width = src.area.width;
  dst.length = src.area.height;
  dst.rotationAngle = src.area.rotation;
  dst.impactRadius = src.impactRadius.is_set() ? src.impactRadius.get() : 0;
  dst.partyId = src.partyId.is_set() ? src.partyId.get() : 0;
  dst.chargeTypeId = src.chargeTypeId;
  dst.isCentral = true;
  dst.triggered = false;
  //dst.participantId = src.participantId.is_set() ? src.participantId.get() : 0;
  //dst.externalId = src.externalId.is_set() ? src.externalId.get() : 0;

}

void HandlerOffRouteMine::onOffRouteMineModifyRequest(const rti::core::SampleIdentity &cookie,
                                                      const OffRouteMineModifyRequest &request)
{
  DEBUG_LOG(m_logger) << "onOffRouteMineModifyRequest: " << request;
  const Core::FunctionReplyObject ret = offRouteMineModifyRequest(request);
  if (ret.error())
    WARNING_LOG(m_logger) << ret();
  m_mainHandler->dds()->fts()->offRouteMineModifyReply(cookie, ret);
}

Core::FunctionReplyObject HandlerOffRouteMine::offRouteMineCreateRequest(const OffRouteMineModifyRequest &request)
{
  if (totalAreasCentral() >= (s_maxActiveAreasCentral[static_cast<int>(AesType::AtOffRouteMine)]) && request.effectStart.is_set())
    return Core::FunctionReplyObject(1, tr_lts("dds", "Too many active off route mines."));

  Database::AesOffRouteMine newData;
  fillData(request, newData);

  ObjectOffRouteMine *ifs(new ObjectOffRouteMine(m_mainHandler, this));
  connect(ifs, &ObjectOffRouteMine::effectStateChanged, this, &HandlerOffRouteMine::onEffectStateChanged);
  Core::FunctionReplyObject ret = ifs->createObject(newData);
  if (ret.error())
  {
    delete ifs;
    return ret;
  }

  connect(ifs, &ObjectOffRouteMine::newAiv,             this, &HandlerOffRouteMine::newAiv);
  connect(ifs, &ObjectOffRouteMine::removeAiv,          this, &HandlerOffRouteMine::removeAiv);
  if (ifs->isObjectInTrainingArea())
  {
    ++m_areaCountCentral;
    DEBUG_LOG(m_logger) << "s_areaCount[AesType::AtOffRouteMine] = " << m_areaCount;
  }
  m_objectMap.insert(ifs->id(), ifs);
  return ret;
}

Core::FunctionReplyObject HandlerOffRouteMine::offRouteMineModifyRequest(const OffRouteMineModifyRequest &request)
{
  Core::FunctionReplyObject ret;
  ObjectOffRouteMine *ifs = nullptr;
  if (request.id == 0 || ((request.id < 0) && !getObjectFromMap(m_objectMap, std::abs(request.id), ret)))
    return offRouteMineCreateRequest(request);

  ifs = getObjectFromMap(m_objectMap, std::abs(request.id), ret);

  if (!ifs)
    return ret;

  if (request.effectStart.is_set() && (request.effectStart.get() + request.effectDuration < m_mainHandler->dds()->currentPartitionTime().toMSecsSinceEpoch()))
    return removeObject(ifs);

  // Anzahl aktiver Areas ist erreicht, der Wirkbeginn ist (soll) gesetzt sein, aber die Area war
  // vorher noch ohne Wirkbeginn
  /*!
   *  If the effect start added after creation, the max amount need to be checked
  !*/
  if (totalAreasCentral() >= (s_maxActiveAreasCentral[static_cast<int>(AesType::AtOffRouteMine)])
      && request.effectStart.is_set()
      && !ifs->isActivationTimeSet() // check if time before change was not setted.
      /*&& !ifs->isObjectModifiable()*/)
  {
    return Core::FunctionReplyObject(request.id, tr_lts("dds", "Too many active off route mines."));
  }
  Database::AesOffRouteMine aea(ifs->data());
  fillData(request, aea);
  if (request.id < 0 && checkStateForSfm(ifs))
  {
    DEBUG_LOG(m_logger) << "Overwrite off route mine Effect State for OC";
    aea.effectStateId = AesEffectState::EsPrepared;
  }
  return ifs->updateObject(aea);
}

void HandlerOffRouteMine::onOffRouteMineRemoveRequest(const rti::core::SampleIdentity &cookie,
                                                      const Core::RemoveElementRequest &request)
{
  m_mainHandler->dds()->fts()->offRouteMineRemoveReply(cookie, removeReply(&DatabaseInterface::removeAesOffRouteMine, request));
}

void HandlerOffRouteMine::onOffRouteMineTriggerRequest(const rti::core::SampleIdentity &cookie,
                                                       const OffRouteMineTriggerRequest &request)
{
  DEBUG_LOG(m_logger) << "onOffRouteMineTriggerRequest: " << request;
  const Core::FunctionReplyObject ret = offRouteMineTriggerRequest(request);
  if (ret.error())
    WARNING_LOG(m_logger) << ret();
  m_mainHandler->dds()->fts()->offRouteMineTriggerReply(cookie, ret);
}

Core::FunctionReplyObject HandlerOffRouteMine::offRouteMineTriggerRequest(const OffRouteMineTriggerRequest &request)
{
  Core::FunctionReplyObject ret;
  ObjectOffRouteMine *ifs = getObjectFromMap(m_objectMap, std::abs(request.id), ret);

  if (!ifs)
    return ret;

  Database::AesOffRouteMine aea(ifs->data());
  DEBUG_LOG(m_logger) << "Overwrite off route mine Effect State ";
  aea.effectStateId = AesEffectState::EsExpired;
  aea.triggered     = request.triggered;
  aea.effectDuration = aea.effectStart.msecsTo(QDateTime::currentDateTimeUtc());
  if (aea.effectDuration < 0)
    aea.effectDuration = 1;

  return ifs->updateObject(aea);
}

} // Namespace AreaEffectSimulation