Initial commit

Lts/Tools/DataSampleReceiver/main.cpp

@@ -0,0 +1,141 @@
+/**
+  * @RDE_HEADER@
+  * main (TopicDataGenerator)
+  * $Id$
+  * $HeadURL$
+  * $Author$
+  */
+
+#include <nglapplication/nglcoreapplication.h>
+#include <NGLLogger/INGLLogger.h>
+
+#include "initialisation.h"
+
+#include <DDS/Generic/generic_base.h>
+#include <DDS/Generic/topic_type_plugin.h>
+#include <LtsPluginManager/lts_plugin_loader.h>
+
+#include <LtsCommandLineParser/LtsCommandLineParser.h>
+#include <QSet>
+
+// static
+static QVector<DDSGeneric::TopicDescription> getAvailableTopicTypes() // get topicTypes per plugin mechanism
+{
+  QVector<DDSGeneric::TopicDescription> topicTypeDescs;
+
+  LtsPluginManager::LtsPluginLoader::getInstance()->loadPlugins(LtsPluginManager::PluginType::TOPIC_TYPE, [&](const LtsPluginManager::PluginInstance &pi)
+  {
+    DDSGeneric::TopicTypePlugin *typePlugin = qobject_cast<DDSGeneric::TopicTypePlugin*>(pi.plugin);
+    if (!typePlugin)
+      return false;
+
+    QMap<std::string,DDSGeneric::TopicDescription> topics = typePlugin->getDataTopics();
+    topics.unite(typePlugin->getFunctionTopics());
+
+    QMap<std::string, DDSGeneric::TopicDescription>::iterator it = topics.begin();
+    for( ; it != topics.end(); ++it)
+    {
+      topicTypeDescs.append(it.value());
+    }
+
+    return false; // we don't need the plugins anymore -> unload
+  });
+
+  return topicTypeDescs;
+}
+
+
+class CmdLineParser : public LtsCommandLineParser::LtsCommandLineParser
+{
+public:
+  CmdLineParser()
+    : LtsCommandLineParser()
+  {
+    addOption( { QStringList() << QLatin1String("s") << QLatin1String("maxSamples"),
+                 QLatin1String("Default: 0 (infinite loop)"),
+                 QLatin1String("Max Samples"),
+                 QLatin1String("0") } );
+
+    addOption( { QStringList() << QLatin1String("t") << QLatin1String("doType"),
+                 QLatin1String("Default: all, can be a list of doTypes, i.e. 1,2,5 (more Infos on doType, use -h)"),
+                 QLatin1String("doType"),
+                 QLatin1String("0") } );
+  }
+
+  int maxSamples() const { return isSet(QLatin1String("s")) ? value(QLatin1String("s")).toInt() : -1; }
+
+  QVector<DDSGeneric::TopicDescription> getDoTypes(NGL::INGLLogger &logger) const
+  {
+    const QVector<DDSGeneric::TopicDescription> availDataTopics = getAvailableDoTypes();
+    if (!isSet(QLatin1String("t")))
+      return availDataTopics;
+
+    // Convert list of DoTypes, separated by comma (given by user)
+    const QString doTypeStr = value(QLatin1String("t"));
+    const QVector<QStringRef> strList = doTypeStr.splitRef(QLatin1Char(','), QString::SkipEmptyParts);
+
+    if (strList.isEmpty())
+      return availDataTopics;
+
+    QVector<DDSGeneric::TopicDescription> vDt;
+    vDt.reserve(strList.size());
+    for (const QStringRef &doTypeStr : strList)
+    {
+      bool ok = false;
+      int doTypeInt = doTypeStr.toInt(&ok);
+      INFO_LOG(logger) << "doType: " << doTypeInt;
+      if (ok &&  doTypeInt >= 0 && doTypeInt < availDataTopics.size())
+        vDt.append(availDataTopics.at(doTypeInt));
+      else
+        WARNING_LOG(logger) << doTypeStr.toString() << " is not a valid DoType, skipping.. ";
+    }
+    return vDt;
+  }
+
+  QVector<DDSGeneric::TopicDescription> getAvailableDoTypes() const
+  {
+    return getAvailableTopicTypes();
+  }
+
+  bool init()
+  {
+    LtsCommandLineParser::init(false);
+    if (isSet(QLatin1String("h")))
+    {
+      QVector<DDSGeneric::TopicDescription> doTypes = getAvailableDoTypes();
+      auto it = doTypes.begin();
+      for(int i=0 ; it != doTypes.end(); ++it, ++i)
+        fprintf(stdout, "\ndoType(%d): %s ", i, (*it).name.c_str());
+      fprintf(stdout, "\n\n");
+    }
+    return true;
+  }
+};
+
+int main ( int argc, char**argv )
+{
+  NGLCoreApplication app ( argc, argv );
+  app.createLogger("DataSampleGenerator.main", "log4cxx_tdsr.props");
+
+  CmdLineParser options;
+  if (!options.init())
+    return -1;
+
+  QVector<DDSGeneric::TopicDescription> doTypes = options.getDoTypes(app.baseLogger());
+  INFO_LOG(app.baseLogger()) << "size vDoTypes: " << doTypes.size();
+  DataSampleReceiver::Initialisation* initDtcr = new DataSampleReceiver::Initialisation(options.domainId(), options.exerciseId(), doTypes);
+  if (!initDtcr->init())
+    return -2;
+
+  STARTUP_LOG(app.baseLogger());
+
+  app.exec();
+  delete initDtcr;
+
+  SHUTDOWN_LOG(app.baseLogger());
+
+  return 0;
+}
+
+
+