concurrentRun.cpp

Go to the documentation of this file.

#include "GaudiKernel/Bootstrap.h"
#include "GaudiKernel/IAppMgrUI.h"
#include "GaudiKernel/IProperty.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/Property.h"
#include "GaudiKernel/SmartIF.h"
#include <iostream>
#include <string>

// boost includes
#include "boost/program_options.hpp"

using namespace boost::program_options;

//-----------------------------------------------------------------------------
int main( int argc, char** argv )
{
  // Declare the command line options and read them in
  options_description desc( "Allowed options" );
  desc.add_options()( "help,h", "produce help message" )( "config", value<std::string>(),
                                                          "comma separated list of python configuration files" )(
      "post-option", value<std::string>(), "python command to be run after ConfigurableUser has been applied" )(
      "param", value<std::vector<std::string>>(), "parameters to the configuration file, multiple can be given" );
  positional_options_description p;
  p.add( "config", 1 );
  variables_map vm;
  try {
    store( command_line_parser( argc, argv ).options( desc ).positional( p ).run(), vm );
  } catch ( boost::exception_detail::clone_impl<
            boost::exception_detail::error_info_injector<boost::program_options::unknown_option>> ) {
    std::cerr << "Unknown option(s) detected." << std::endl << "Usage:" << std::endl << desc << std::endl;
    return ( 1 );
  }
  notify( vm );

  std::string fileName;
  // handle the retrieved options
  if ( vm.count( "help" ) or vm.count( "h" ) ) {
    std::cout << desc << std::endl;
    return 1;
  }
  if ( !vm.count( "config" ) ) {
    std::cout << "Please specify a config file" << std::endl;
    return 1;
  } else {
    fileName = vm["config"].as<std::string>();
  }

  std::stringstream params;
  if ( 0 != vm.count( "param" ) ) {
    std::vector<std::string> vParams = vm["param"].as<std::vector<std::string>>();
    std::cout << "Configuration parameters are:" << std::endl;
    for ( std::string s : vParams ) {
      params << s << "\n";
      std::cout << "\t" << s << std::endl;
    }
  }

  std::string postAction;
  if ( 0 != vm.count( "post-option" ) ) {
    postAction = vm["post-option"].as<std::string>();
  }
  // end of options handling

  IInterface*        iface = Gaudi::createApplicationMgr();
  SmartIF<IProperty> propMgr( iface );
  SmartIF<IAppMgrUI> appUI( iface );
  propMgr->setProperty( "JobOptionsType", "PYTHON" );
  propMgr->setProperty( "JobOptionsPath", fileName );
  propMgr->setProperty( "JobOptionsPreAction", params.str() );
  propMgr->setProperty( "JobOptionsPostAction", postAction ); // TODO: grep this from command line
  return appUI->run().getCode();
}