mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-13 00:31:14 +00:00
200 lines
6.6 KiB
C++
200 lines
6.6 KiB
C++
/********************************************************************************
|
|
* Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH *
|
|
* *
|
|
* This software is distributed under the terms of the *
|
|
* GNU Lesser General Public Licence version 3 (LGPL) version 3, *
|
|
* copied verbatim in the file "LICENSE" *
|
|
********************************************************************************/
|
|
/**
|
|
* runSplitter.cxx
|
|
*
|
|
* @since 2012-12-06
|
|
* @author D. Klein, A. Rybalchenko
|
|
*/
|
|
|
|
#include <iostream>
|
|
#include <csignal>
|
|
|
|
#include "boost/program_options.hpp"
|
|
|
|
#include "FairMQLogger.h"
|
|
#include "FairMQSplitter.h"
|
|
|
|
#ifdef NANOMSG
|
|
#include "FairMQTransportFactoryNN.h"
|
|
#else
|
|
#include "FairMQTransportFactoryZMQ.h"
|
|
#endif
|
|
|
|
using namespace std;
|
|
|
|
FairMQSplitter splitter;
|
|
|
|
static void s_signal_handler(int signal)
|
|
{
|
|
cout << endl << "Caught signal " << signal << endl;
|
|
|
|
splitter.ChangeState(FairMQSplitter::STOP);
|
|
splitter.ChangeState(FairMQSplitter::END);
|
|
|
|
cout << "Shutdown complete. Bye!" << 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);
|
|
}
|
|
|
|
typedef struct DeviceOptions
|
|
{
|
|
string id;
|
|
int ioThreads;
|
|
int numOutputs;
|
|
string inputSocketType;
|
|
int inputBufSize;
|
|
string inputMethod;
|
|
string inputAddress;
|
|
vector<string> outputSocketType;
|
|
vector<int> outputBufSize;
|
|
vector<string> outputMethod;
|
|
vector<string> outputAddress;
|
|
} DeviceOptions_t;
|
|
|
|
inline bool parse_cmd_line(int _argc, char* _argv[], DeviceOptions* _options)
|
|
{
|
|
if (_options == NULL)
|
|
throw std::runtime_error("Internal error: options' container is empty.");
|
|
|
|
namespace bpo = boost::program_options;
|
|
bpo::options_description desc("Options");
|
|
desc.add_options()
|
|
("id", bpo::value<string>()->required(), "Device ID")
|
|
("io-threads", bpo::value<int>()->default_value(1), "Number of I/O threads")
|
|
("num-outputs", bpo::value<int>()->required(), "Number of Splitter output sockets")
|
|
("input-socket-type", bpo::value<string>()->required(), "Input socket type: sub/pull")
|
|
("input-buff-size", bpo::value<int>()->required(), "Input buffer size in number of messages (ZeroMQ)/bytes(nanomsg)")
|
|
("input-method", bpo::value<string>()->required(), "Input method: bind/connect")
|
|
("input-address", bpo::value<string>()->required(), "Input address, e.g.: \"tcp://localhost:5555\"")
|
|
("output-socket-type", bpo::value< vector<string> >()->required(), "Output socket type: pub/push")
|
|
("output-buff-size", bpo::value< vector<int> >()->required(), "Output buffer size in number of messages (ZeroMQ)/bytes(nanomsg)")
|
|
("output-method", bpo::value< vector<string> >()->required(), "Output method: bind/connect")
|
|
("output-address", bpo::value< vector<string> >()->required(), "Output address, e.g.: \"tcp://localhost:5555\"")
|
|
("help", "Print help messages");
|
|
|
|
bpo::variables_map vm;
|
|
bpo::store(bpo::parse_command_line(_argc, _argv, desc), vm);
|
|
|
|
if ( vm.count("help") )
|
|
{
|
|
LOG(INFO) << "FairMQ Splitter" << endl << desc;
|
|
return false;
|
|
}
|
|
|
|
bpo::notify(vm);
|
|
|
|
if ( vm.count("id") )
|
|
_options->id = vm["id"].as<string>();
|
|
|
|
if ( vm.count("io-threads") )
|
|
_options->ioThreads = vm["io-threads"].as<int>();
|
|
|
|
if ( vm.count("num-outputs") )
|
|
_options->numOutputs = vm["num-outputs"].as<int>();
|
|
|
|
if ( vm.count("input-socket-type") )
|
|
_options->inputSocketType = vm["input-socket-type"].as<string>();
|
|
|
|
if ( vm.count("input-buff-size") )
|
|
_options->inputBufSize = vm["input-buff-size"].as<int>();
|
|
|
|
if ( vm.count("input-method") )
|
|
_options->inputMethod = vm["input-method"].as<string>();
|
|
|
|
if ( vm.count("input-address") )
|
|
_options->inputAddress = vm["input-address"].as<string>();
|
|
|
|
if ( vm.count("output-socket-type") )
|
|
_options->outputSocketType = vm["output-socket-type"].as< vector<string> >();
|
|
|
|
if ( vm.count("output-buff-size") )
|
|
_options->outputBufSize = vm["output-buff-size"].as< vector<int> >();
|
|
|
|
if ( vm.count("output-method") )
|
|
_options->outputMethod = vm["output-method"].as< vector<string> >();
|
|
|
|
if ( vm.count("output-address") )
|
|
_options->outputAddress = vm["output-address"].as< vector<string> >();
|
|
|
|
return true;
|
|
}
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
s_catch_signals();
|
|
|
|
DeviceOptions_t options;
|
|
try
|
|
{
|
|
if (!parse_cmd_line(argc, argv, &options))
|
|
return 0;
|
|
}
|
|
catch (exception& e)
|
|
{
|
|
LOG(ERROR) << e.what();
|
|
return 1;
|
|
}
|
|
|
|
LOG(INFO) << "PID: " << getpid();
|
|
|
|
#ifdef NANOMSG
|
|
FairMQTransportFactory* transportFactory = new FairMQTransportFactoryNN();
|
|
#else
|
|
FairMQTransportFactory* transportFactory = new FairMQTransportFactoryZMQ();
|
|
#endif
|
|
|
|
splitter.SetTransport(transportFactory);
|
|
|
|
splitter.SetProperty(FairMQSplitter::Id, options.id);
|
|
splitter.SetProperty(FairMQSplitter::NumIoThreads, options.ioThreads);
|
|
|
|
splitter.SetProperty(FairMQSplitter::NumInputs, 1);
|
|
splitter.SetProperty(FairMQSplitter::NumOutputs, options.numOutputs);
|
|
|
|
splitter.ChangeState(FairMQSplitter::INIT);
|
|
|
|
splitter.SetProperty(FairMQSplitter::InputSocketType, options.inputSocketType);
|
|
splitter.SetProperty(FairMQSplitter::InputRcvBufSize, options.inputBufSize);
|
|
splitter.SetProperty(FairMQSplitter::InputMethod, options.inputMethod);
|
|
splitter.SetProperty(FairMQSplitter::InputAddress, options.inputAddress);
|
|
|
|
for (int i = 0; i < options.numOutputs; ++i)
|
|
{
|
|
splitter.SetProperty(FairMQSplitter::OutputSocketType, options.outputSocketType.at(i), i);
|
|
splitter.SetProperty(FairMQSplitter::OutputSndBufSize, options.outputBufSize.at(i), i);
|
|
splitter.SetProperty(FairMQSplitter::OutputMethod, options.outputMethod.at(i), i);
|
|
splitter.SetProperty(FairMQSplitter::OutputAddress, options.outputAddress.at(i), i);
|
|
}
|
|
|
|
splitter.ChangeState(FairMQSplitter::SETOUTPUT);
|
|
splitter.ChangeState(FairMQSplitter::SETINPUT);
|
|
splitter.ChangeState(FairMQSplitter::RUN);
|
|
|
|
// wait until the running thread has finished processing.
|
|
boost::unique_lock<boost::mutex> lock(splitter.fRunningMutex);
|
|
while (!splitter.fRunningFinished)
|
|
{
|
|
splitter.fRunningCondition.wait(lock);
|
|
}
|
|
|
|
splitter.ChangeState(FairMQSplitter::STOP);
|
|
splitter.ChangeState(FairMQSplitter::END);
|
|
|
|
return 0;
|
|
}
|