KVXtermBatch.cpp

#include "KVDataSetAnalyser.h"
#include "KVXtermBatch.h"
#include <cassert>
 
 
 
void KVXtermBatch::Run()
{
   //Processes the job requests for the batch system.
   //In multijobs mode, this submits one job for each run in the runlist associated to fAnalyser
 
   if (!CheckJobParameters()) return;
 
   //submit jobs for every GetRunsPerJob() runs in runlist
   KVDataSetAnalyser* ana = dynamic_cast<KVDataSetAnalyser*>(fAnalyser);
   assert(ana);
   auto runs = ana->GetRunList();
   Int_t remaining_runs = runs.GetNValues();
 
   // to optimize use of CPUs, some jobs will have multiple runs
   std::vector<int> runs_per_job(max_num_cpus);
   int job_index = 0;
 
   for (auto& r : runs) {
      ++runs_per_job[job_index++];
      if (job_index == max_num_cpus) job_index = 0;
   }
 
   fCurrJobRunList.Clear();
   job_index = 0;
   bool submitted = false;
   for (auto& run : runs) {
      if (!remaining_runs) break;
      remaining_runs--;
      fCurrJobRunList.Add(run);
      submitted = false;
      if (fCurrJobRunList.GetNValues() == runs_per_job[job_index]) {
         // submit job for run
         ana->SetRuns(fCurrJobRunList, kFALSE);
         ana->SetFullRunList(runs);
         std::cout << "Submitting job " << job_index + 1 << " for runs:" << fCurrJobRunList.AsString() << std::endl;
         SubmitJob();
         fCurrJobRunList.Clear();
         ++job_index;
         submitted = true;
      }
   }
   if (!submitted && !fCurrJobRunList.IsEmpty()) {
      // submit last job for run
      ana->SetRuns(fCurrJobRunList, kFALSE);
      ana->SetFullRunList(runs);
      std::cout << "Submitting job " << job_index + 1 << " for runs:" << fCurrJobRunList.AsString() << std::endl;
      SubmitJob();
      fCurrJobRunList.Clear();
      ++job_index;
   }
   ana->SetRuns(runs, kFALSE);
}
 
 
ClassImp(KVXtermBatch)
 
 
 
 
void KVXtermBatch::GetBatchSystemParameterList(KVNameValueList& nl)
{
   // Add to batch parameters the number of CPUs to use
   //
   // By default, it is the number of CPUs on the machine
 
   KVBatchSystem::GetBatchSystemParameterList(nl);
   nl.SetValue("MaxNumCPUs", max_num_cpus);
}
 
 
 
 
void KVXtermBatch::SetBatchSystemParameters(const KVNameValueList& nl)
{
   // Add to batch parameters the number of CPUs to use
   //
   // By default, it is the number of CPUs on the machine
 
   KVBatchSystem::SetBatchSystemParameters(nl);
   max_num_cpus = nl.GetIntValue("MaxNumCPUs");
}