eventloop_exec_worker.cxx File Reference

#include <AsgMessaging/MessageCheck.h>
#include <EventLoop/Driver.h>
#include <RootCoreUtils/ShellExec.h>
#include <TSystem.h>
#include <xAODRootAccess/Init.h>
#include <fstream>
#include <sstream>
#include <string>
#include <vector>

Include dependency graph for eventloop_exec_worker.cxx:

Go to the source code of this file.

Functions
int	main (int argc, char **argv)

Function Documentation

main()

int main	(	int	argc,
		char **	argv )

Definition at line 15 of file eventloop_exec_worker.cxx.

{
  using namespace asg::msgUserCode;
  ANA_CHECK_SET_TYPE (int);
 
  ANA_CHECK (xAOD::Init ());
 
  // arguments: <submitDir> <resubmit:0|1> <index>...
  if (argc < 3)
  {
    ANA_MSG_ERROR ("invalid number of arguments");
    return -1;
  }
 
  const std::string submitDir = argv[1];
  const bool resubmit = (std::string (argv[2]) == "1");
 
  std::vector<std::size_t> indices;
  for (int arg = 3; arg != argc; ++ arg)
  {
    try
    {
      indices.push_back (std::stoul (argv[arg]));
    } catch (std::exception& e)
    {
      ANA_MSG_ERROR ("failed to parse job index \"" << argv[arg] << "\": " << e.what());
      return -1;
    }
  }
 
  const std::string basedirName = submitDir + "/tmp";
 
  // on resubmit the tmp directory already exists from the first submission
  if (!resubmit && gSystem->MakeDirectory (basedirName.c_str()) != 0)
  {
    ANA_MSG_ERROR ("failed to create directory " + basedirName);
    return -1;
  }
 
  auto submitSingle = [&] (std::size_t index) noexcept -> StatusCode
  {
    try
    {
      const std::string dirName = basedirName + "/" + std::to_string (index);
      if (gSystem->MakeDirectory (dirName.c_str()) != 0 && !resubmit)
      {
        ANA_MSG_ERROR ("failed to create directory " + dirName);
        return StatusCode::FAILURE;
      }
 
      std::ostringstream cmd;
      cmd << "cd " << RCU::Shell::quote (dirName) << " && ";
      cmd << RCU::Shell::quote (submitDir) << "/submit/run " << index;
      RCU::Shell::exec (cmd.str());
    } catch (std::exception& e)
    {
      ANA_MSG_ERROR ("exception in job " << index << ": " << e.what());
      return StatusCode::FAILURE;
    }
    return StatusCode::SUCCESS;
  };
  for (std::size_t index : indices)
  {
    if (submitSingle (index).isFailure())
      return EXIT_FAILURE;
  }
  // this particular file can be checked to see if a job has
  // been submitted successfully.
  std::ofstream ((submitDir + "/submitted").c_str());
  if (!EL::Driver::retrieve (submitDir))
  {
    ANA_MSG_ERROR ("failed to retrieve job output in " + submitDir);
    return EXIT_FAILURE;
  }
  return 0;
}