mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-12 16:21:13 +00:00
93 lines
3.0 KiB
C++
93 lines
3.0 KiB
C++
/********************************************************************************
|
|
* Copyright (C) 2017 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH *
|
|
* *
|
|
* This software is distributed under the terms of the *
|
|
* GNU Lesser General Public Licence (LGPL) version 3, *
|
|
* copied verbatim in the file "LICENSE" *
|
|
********************************************************************************/
|
|
|
|
#include <FairMQLogger.h>
|
|
#include <options/FairMQProgOptions.h>
|
|
#include <FairMQDevice.h>
|
|
#include <tools/runSimpleMQStateMachine.h>
|
|
#include <fairmq/PluginManager.h>
|
|
#include <boost/program_options.hpp>
|
|
|
|
template <typename R>
|
|
class GenericFairMQDevice : public FairMQDevice
|
|
{
|
|
public:
|
|
GenericFairMQDevice(R func) : r(func) {}
|
|
|
|
protected:
|
|
virtual bool ConditionalRun() { return r(*static_cast<FairMQDevice*>(this)); }
|
|
|
|
private:
|
|
R r;
|
|
};
|
|
|
|
template <typename R>
|
|
FairMQDevice* makeDeviceWithConditionalRun(R r)
|
|
{
|
|
return new GenericFairMQDevice<R>(r);
|
|
}
|
|
|
|
using FairMQDevicePtr = FairMQDevice*;
|
|
|
|
// to be implemented by the user to return a child class of FairMQDevice
|
|
FairMQDevicePtr getDevice(const FairMQProgOptions& config);
|
|
|
|
// to be implemented by the user to add custom command line options (or just with empty body)
|
|
void addCustomOptions(boost::program_options::options_description&);
|
|
|
|
int main(int argc, const char** argv)
|
|
{
|
|
try
|
|
{
|
|
boost::program_options::options_description customOptions("Custom options");
|
|
addCustomOptions(customOptions);
|
|
|
|
// Plugin manager needs to be destroyed after config !
|
|
// TODO Investigate, why
|
|
auto pluginManager = fair::mq::PluginManager::MakeFromCommandLineOptions(fair::mq::tools::ToStrVector(argc, argv));
|
|
FairMQProgOptions config;
|
|
config.AddToCmdLineOptions(customOptions);
|
|
|
|
pluginManager->ForEachPluginProgOptions([&config](boost::program_options::options_description options){
|
|
config.AddToCmdLineOptions(options);
|
|
});
|
|
config.AddToCmdLineOptions(pluginManager->ProgramOptions());
|
|
|
|
config.ParseAll(argc, argv, true);
|
|
|
|
std::unique_ptr<FairMQDevice> device(getDevice(config));
|
|
if (!device)
|
|
{
|
|
LOG(ERROR) << "getDevice(): no valid device provided. Exiting.";
|
|
return 1;
|
|
}
|
|
|
|
pluginManager->EmplacePluginServices(config, *device);
|
|
pluginManager->InstantiatePlugins();
|
|
|
|
int result = runStateMachine(*device, config);
|
|
|
|
if (result > 0)
|
|
{
|
|
return 1;
|
|
}
|
|
}
|
|
catch (std::exception& e)
|
|
{
|
|
LOG(ERROR) << "Unhandled exception reached the top of main: " << e.what() << ", application will now exit";
|
|
return 1;
|
|
}
|
|
catch (...)
|
|
{
|
|
LOG(ERROR) << "Non-exception instance being thrown. Please make sure you use std::runtime_exception() instead. Application will now exit.";
|
|
return 1;
|
|
}
|
|
|
|
return 0;
|
|
}
|