Initial commit

Lts/Tools/VulnerabilityStateGenerator/main_handler.cpp

@@ -0,0 +1,128 @@
+#include "main_handler.h"
+#include <NGLLogger/NGLLoggerBootstrap.h>
+#include <DDS/DDSConnector/dds_connector.h>
+#include <QtCore/QThread>
+#include <QtCore/QCoreApplication>
+#include <QtCore/QDateTime>
+#include <QtCore/QElapsedTimer>
+#include "proxy_emulator.h"
+
+#include <QTimer>
+
+VulnerabilityStateGenerator::MainHandler::MainHandler(QObject *parent)
+  : QObject(parent)
+  , m_domainId(0)
+  , m_exerciseId(0) 
+  , m_proxyEmulator(nullptr)  
+  , m_genTimer(new QTimer())
+  , m_logger(NGL::NGLLoggerBootstrap::getNGLRootLogger().getNGLLogger("MainHandler"))
+{
+}
+
+VulnerabilityStateGenerator::MainHandler::~MainHandler()
+{  
+}
+
+bool VulnerabilityStateGenerator::MainHandler::init(int32_t domainId, int32_t exerciseId)
+{
+  m_domainId        = domainId;
+  m_exerciseId      = exerciseId;
+
+  delete m_proxyEmulator;
+  m_proxyEmulator   = nullptr;
+
+  // DdsConnector
+  DEBUG_LOG(m_logger) << "Creating and initializing DdsConnector ...";
+  m_ddsConnector.reset(new DDSConnector::DdsConnector(m_domainId, IDL::PartitionType::Exercise, m_exerciseId));
+  if (!m_ddsConnector->init())
+  {
+    ERROR_LOG(m_logger) << "Could not init DdsConnector";
+    return false;
+  }
+  m_ddsConnector->initPublisherSubscriber();
+  DEBUG_LOG(m_logger) << "... done";
+
+  // dtc
+  DEBUG_LOG(m_logger) << "Creating and initializing PPM DTC ...";
+  m_ppmDtc = ParticipantManagement::Interface::Dtc::create(m_ddsConnector);
+  QThread::msleep(100);
+  if (!m_ppmDtc->enableAll())
+  {
+    ERROR_LOG(m_logger) << "Could not init PPM DTC";
+    return false;
+  }
+  DEBUG_LOG(m_logger) << "... done";  
+
+  fillMapFromEnum();
+
+  DEBUG_LOG(m_logger) << "m_vsMap.size(): " << m_vsMap.size();
+
+  // ProxyEmulator
+  DEBUG_LOG(m_logger) << "Creating and initializing ProxyEmulator ...";
+  m_proxyEmulator = new ProxyEmulator(m_ddsConnector, this);
+  QThread::msleep(100);
+  if (!m_proxyEmulator->init())
+  {
+    ERROR_LOG(m_logger) << "Could not init ProxyEmulator";
+    return false;
+  }
+
+  m_partMap = m_ppmDtc->participants();
+
+  m_proxyEmulator->giveInitState(m_partMap);
+
+  connect(m_genTimer, &QTimer::timeout, this, &MainHandler::setNewState);
+  m_genTimer->start(1000);
+
+  return true;
+}
+
+
+void VulnerabilityStateGenerator::MainHandler::quit()
+{
+  ERROR_LOG(m_logger) << "Are the following procs running?\n"
+                      << "svcParticipantManagement -d " << m_domainId << " -e 0\n"
+                      << "svcParticipantManagement -d " << m_domainId << " -e " << m_exerciseId << "\n"
+                      << "... shutting down";
+  QCoreApplication::exit(-1);
+}
+
+
+ProjectDataManagement::VulnerabilityState VulnerabilityStateGenerator::MainHandler::randomVulnerabilityState()
+{
+  ProjectDataManagement::VulnerabilityState result;
+
+  qsrand(QDateTime::currentMSecsSinceEpoch());
+  unsigned int index = 0;
+  if (m_vsMap.size() > 0)
+    index = qrand() % m_vsMap.size();
+
+
+  return m_vsMap.value(index);
+
+}
+
+void VulnerabilityStateGenerator::MainHandler::setNewState()
+{
+  for (auto it = m_partMap.begin(); it != m_partMap.end(); ++it)
+  {
+    if (it.value().equipmentNumber().is_set() && it.value().equipmentNumber().get() > 0)
+      m_proxyEmulator->publishNewState(randomVulnerabilityState(), it.value());
+  }
+}
+
+void VulnerabilityStateGenerator::MainHandler::fillMapFromEnum()
+{
+  m_vsMap.insert(0, ProjectDataManagement::VulnerabilityState_def::CombatReady);
+  m_vsMap.insert(1, ProjectDataManagement::VulnerabilityState_def::LightDamage);
+  m_vsMap.insert(2, ProjectDataManagement::VulnerabilityState_def::MediumDamage);
+  m_vsMap.insert(3, ProjectDataManagement::VulnerabilityState_def::HeavyDamage);
+  m_vsMap.insert(4, ProjectDataManagement::VulnerabilityState_def::MobilityKill);
+  m_vsMap.insert(5, ProjectDataManagement::VulnerabilityState_def::VisibleMobilityKill);
+  m_vsMap.insert(6, ProjectDataManagement::VulnerabilityState_def::WeaponSystemDamage);
+  m_vsMap.insert(7, ProjectDataManagement::VulnerabilityState_def::VisibleWeaponSystemDamage);
+  m_vsMap.insert(8, ProjectDataManagement::VulnerabilityState_def::CatastrophicKill);
+  m_vsMap.insert(9, ProjectDataManagement::VulnerabilityState_def::TemporarySeverance);
+  m_vsMap.insert(10, ProjectDataManagement::VulnerabilityState_def::NearMiss);
+  m_vsMap.insert(11, ProjectDataManagement::VulnerabilityState_def::HitWithNoEffect);
+}