New files from Ralf Kliemt

git-svn-id: https://subversion.gsi.de/fairroot/fairbase/trunk@22606 0381ead4-6506-0410-b988-94b70fbc4730
2025-10-15 09:31:45 +00:00 · 2013-11-01 11:24:47 +00:00
parent 5121fe3ae5
commit 9c4d64f3b1
4 changed files with 260 additions and 23 deletions
--- a/fairmq/runNToOneMerger.cxx
+++ b/fairmq/runNToOneMerger.cxx
@@ -0,0 +1,124 @@
+/*
+ * runMerger.cxx
+ *
+ *  Created on: Dec 6, 2012
+ *      Author: dklein
+ */
+
+#include <iostream>
+#include <csignal>
+
+#include "FairMQLogger.h"
+#include "FairMQStandaloneMerger.h"
+
+
+FairMQStandaloneMerger merger;
+
+static void s_signal_handler (int signal)
+{
+  std::cout << std::endl << "Caught signal " << signal << std::endl;
+
+  merger.ChangeState(FairMQStandaloneMerger::STOP);
+  merger.ChangeState(FairMQStandaloneMerger::END);
+
+  std::cout << "Shutdown complete. Bye!" << std::endl;
+  exit(1);
+}
+
+static void s_catch_signals (void)
+{
+  struct sigaction action;
+  action.sa_handler = s_signal_handler;
+  action.sa_flags = 0;
+  sigemptyset(&action.sa_mask);
+  sigaction(SIGINT, &action, NULL);
+  sigaction(SIGTERM, &action, NULL);
+}
+
+int main(int argc, char** argv)
+{
+  if ( argc < 16 || (argc-8)%4!=0 ) {
+    std::cout << "Usage: merger \tID numIoTreads numInputs\n"
+              << "\t\tinputSocketType inputRcvBufSize inputMethod inputAddress\n"
+              << "\t\tinputSocketType inputRcvBufSize inputMethod inputAddress\n"
+              << "\t\t...\n"
+              << "\t\toutputSocketType outputSndBufSize outputMethod outputAddress\n"
+              << argc << std::endl;
+    return 1;
+  }
+
+  s_catch_signals();
+
+  std::stringstream logmsg;
+  logmsg << "PID: " << getpid();
+  FairMQLogger::GetInstance()->Log(FairMQLogger::INFO, logmsg.str());
+
+  int i = 1;
+
+  merger.SetProperty(FairMQStandaloneMerger::Id, argv[i]);
+  ++i;
+
+  int numIoThreads;
+  std::stringstream(argv[i]) >> numIoThreads;
+  merger.SetProperty(FairMQStandaloneMerger::NumIoThreads, numIoThreads);
+  ++i;
+
+  int numInputs;
+  std::stringstream(argv[i]) >> numInputs;
+  merger.SetProperty(FairMQStandaloneMerger::NumInputs, numInputs);
+  ++i;
+
+  merger.SetProperty(FairMQStandaloneMerger::NumOutputs, 1);
+
+
+  merger.ChangeState(FairMQStandaloneMerger::INIT);
+
+
+  int inputSocketType;
+  for (int iInput = 0; iInput < numInputs; iInput++ ) {
+    inputSocketType = ZMQ_SUB;
+    if (strcmp(argv[i], "pull") == 0) {
+      inputSocketType = ZMQ_PULL;
+    }
+    merger.SetProperty(FairMQStandaloneMerger::InputSocketType, inputSocketType, iInput);
+    ++i;
+    int inputRcvBufSize;
+    std::stringstream(argv[i]) >> inputRcvBufSize;
+    merger.SetProperty(FairMQStandaloneMerger::InputRcvBufSize, inputRcvBufSize, iInput);
+    ++i;
+    merger.SetProperty(FairMQStandaloneMerger::InputMethod, argv[i], iInput);
+    ++i;
+    merger.SetProperty(FairMQStandaloneMerger::InputAddress, argv[i], iInput);
+    ++i;
+  }
+
+  int outputSocketType = ZMQ_PUB;
+  if (strcmp(argv[i], "push") == 0) {
+    outputSocketType = ZMQ_PUSH;
+  }
+  merger.SetProperty(FairMQStandaloneMerger::OutputSocketType, outputSocketType, 0);
+  ++i;
+  int outputSndBufSize;
+  std::stringstream(argv[i]) >> outputSndBufSize;
+  merger.SetProperty(FairMQStandaloneMerger::OutputSndBufSize, outputSndBufSize, 0);
+  ++i;
+  merger.SetProperty(FairMQStandaloneMerger::OutputMethod, argv[i], 0);
+  ++i;
+  merger.SetProperty(FairMQStandaloneMerger::OutputAddress, argv[i], 0);
+  ++i;
+
+
+  merger.ChangeState(FairMQStandaloneMerger::SETOUTPUT);
+  merger.ChangeState(FairMQStandaloneMerger::SETINPUT);
+  merger.ChangeState(FairMQStandaloneMerger::RUN);
+
+
+  char ch;
+  std::cin.get(ch);
+
+  merger.ChangeState(FairMQStandaloneMerger::STOP);
+  merger.ChangeState(FairMQStandaloneMerger::END);
+
+  return 0;
+}
+