LLDriver.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration
*/

 
 
//
// includes
//
 
#include <EventLoop/LLDriver.h>
 
#include <AsgMessaging/StatusCode.h>
#include <EventLoop/Job.h>
#include <EventLoop/ManagerData.h>
#include <EventLoop/MessageCheck.h>
#include <RootCoreUtils/ShellExec.h>
#include <TSystem.h>
#include <sstream>
 
//
// method implementations
//
 
ClassImp(EL::LLDriver)
 
 
namespace EL
{
  void LLDriver ::
  testInvariant () const
  {}
 
 
 
  LLDriver ::
  LLDriver ()
  {
    RCU_NEW_INVARIANT (this);
  }
 
 
 
  ::StatusCode LLDriver ::
  doManagerStep (Detail::ManagerData& data) const
  {
    RCU_READ_INVARIANT (this);
    using namespace msgEventLoop;
    ANA_CHECK (BatchDriver::doManagerStep (data));
    switch (data.step)
    {
    case Detail::ManagerStep::batchScriptVar:
      {
        data.batchName = "run{JOBID}.cmd";
        data.batchInit =
          "#\n"
          "# @ job_name       = EventLoopAnalysis{JOBID}\n"
          "# @ job_type       = serial\n"
          "# @ error          = $(Cluster).err\n"
          "# @ output         = $(Cluster).out\n"
          "# @ class          = " + queue + "\n"
          "# @ resources = ConsumableCpus(1) ConsumableMemory(4gb)\n"
          "# @ wall_clock_limit = 00:20:00\n"
          "# @ queue\n";
      }
      break;
 
    case Detail::ManagerStep::submitJob:
    case Detail::ManagerStep::doResubmit:
      {
        // safely ignoring: resubmit
 
        // Submit n jobs with loadleveler
        for (std::size_t iter : data.batchJobIndices)
        {
          // Submit!
 
          std::ostringstream cmd;
          cmd << "cd " << RCU::Shell::quote (data.submitDir) << "/submit && llsubmit "
              << data.options.castString (Job::optSubmitFlags)
              << " run"<<iter<<".cmd";
 
          if (gSystem->Exec (cmd.str().c_str()) != 0)
          {
            ANA_MSG_ERROR ("failed to execute: " << cmd.str());
            return StatusCode::FAILURE;
          }
        }    
        data.submitted = true;
      }
      break;
 
    default:
      break;
    }
    return ::StatusCode::SUCCESS;
  }
}