hal_jobs.cpp

#include "Cout.h"
#include "Jobs.h"
#include "Std.h"
#include "StdString.h"
 
#include <TString.h>
#include <TSystem.h>
#include <iostream>
#include <vector>
 
 
void printHelp() {
  std::cout << "Class for managing job arrays/job groups" << std::endl;
  std::cout << " Usage:" << std::endl;
  std::cout << "\thal-jobs [file] where file is name of xml file with settings" << std::endl;
  std::cout << "\thal-jobs --debug=[file] where file is name of xml file with settings, submit commands are displayed"
            << std::endl;
  std::cout << "hal-jobs --prepare=[X] where X" << std::endl;
  std::cout << "\t pbs_single for torque" << std::endl;
  std::cout << "\t pbs_array for torque array" << std::endl;
  std::cout << "\t sbatch_single for sbatch" << std::endl;
  std::cout << "\t sbatch_array for sbatch array" << std::endl;
  std::cout << "hal-jobs --submit=[Y] --stage=[X]    prepares jobs" << std::endl;
  std::cout << "\t Y - name of  the files with configured jobs " << std::endl;
  std::cout << "\t X stage of deploying:" << std::endl;
  std::cout << "\t create - only creates job file(s)" << std::endl;
  std::cout << "\t deploy - only submit job(s)" << std::endl;
  std::cout << "--export=[X] --id=[Y] --par=[Z]  - prints the name of the value from "
               "configuration file"
            << std::endl;
  std::cout << "X - configuration parm, Y job id, par parameter number" << std::endl;
}
int main(int argc, char* argv[]) {
  if (argc < 2) {
    Hal::Cout::PrintInfo("No arguments! run: hal-jobs --help to get help", Hal::EInfo::kCriticalError);
    return 0;
  }
  std::vector<std::pair<TString, TString>> args = Hal::Std::ConvertMainArgs(argc, argv);
 
  Int_t nArgs = args.size();
  if (nArgs == 1) {
    TString flag1 = args[0].first;
    TString val1  = args[0].second;
    if (flag1 == "help") {
      printHelp();
      return 0;
    } else if (flag1 == "" && val1.EndsWith(".xml")) {
      Hal::Jobs jobs(val1);
      jobs.CreateJobs();
      jobs.SubmitJobs();
      return 0;
    } else if (flag1 == "debug" && val1.EndsWith(".xml")) {
      Hal::Jobs jobs(val1);
      jobs.DebugCommands();
      jobs.CreateJobs();
      jobs.SubmitJobs();
    } else if (flag1 == "prepare") {
      TString name = val1 + ".xml";
      TString path = Hal::Std::GetHalrootPlus();
      Hal::Std::CopyFiles(Form("%s/job_patterns/%s", path.Data(), name.Data()), name);
      gSystem->mkdir("jobs");
      gSystem->mkdir("logs");
      gSystem->mkdir("errors");
      return 0;
    }
  } else if (nArgs == 2) {
    TString flag1 = args[0].first;
    TString val1  = args[0].second;
    TString flag2 = args[1].first;
    TString val2  = args[1].second;
    if (flag1 == "submit" && val2 == "create") {
      Hal::Jobs jobs(val1);
      jobs.CreateJobs();
      return 0;
    } else if (flag1 == "submit" && val2 == "deploy") {
      Hal::Jobs jobs(val1);
      jobs.SubmitJobs();
      return 0;
    }
  } else if (nArgs == 3) {
    TString flag1 = args[0].first;
    TString flag2 = args[1].first;
    TString flag3 = args[2].first;
    TString val1  = args[0].second;
    TString val2  = args[1].second;
    TString val3  = args[2].second;
    if (flag1 == "export" && flag2 == "id" && flag3 == "par") {
      return 0;
      TString job_id       = val1;
      TString parameter_id = val2;
      TString textfile     = val3;
      std::cout << Hal::Jobs::GetParameter(textfile, job_id.Atoi(), parameter_id.Atoi()) << std::endl;
      return 0;
    }
  }
  std::cout << "Uknown parameter try --help" << std::endl;
  return 1;
}