KV_CCIN2P3_Slurm.h

 
#ifndef __KV_CCIN2P3_SLURM_H
#define __KV_CCIN2P3_SLURM_H
 
#include "KVBatchSystem.h"
 
class KV_CCIN2P3_Slurm : public KVBatchSystem {
protected:
 
    Bool_t fMultiJobs;
    Int_t fRunsPerJob;
    KVString fDefJobTime;
    KVString fDefJobMem;
   Bool_t fTimeSet;
   Bool_t fMemSet;
 
   void ChangeDefJobOpt(KVDataAnalyser*) override;
 
public:
 
   KV_CCIN2P3_Slurm(const Char_t* name);
 
   void SetJobTime(const Char_t* h = "");      /* Set CPU time for batch job */
   void SetJobMemory(const Char_t* h = ""); /* Set max memory for batch job */
   void PrintJobs(Option_t* opt = "") override;  /* Print list of all jobs submitted to system */
 
   Bool_t CheckJobParameters() override;        /* Checks job parameters */
   void SetMultiJobsMode(Bool_t on = kTRUE)
   {
      fMultiJobs = on;
   }
   Bool_t MultiJobsMode() const override
   {
      return fMultiJobs;
   }
 
   void SetRunsPerJob(Int_t n)
   {
      fRunsPerJob = n;
   }
   Int_t GetRunsPerJob() const
   {
      return fRunsPerJob;
   }
 
   const Char_t* GetJobTime(void) const;    /* Get CPU time for batch job */
   const Char_t* GetJobMemory(void) const;       /* Get max memory for batch job */
 
   void ChooseJobTime(void);    /* Choose CPU time for batch job */
   void ChooseJobMemory(void);       /* Choose max memory for batch job */
 
   void Clear(Option_t* opt = "") override;
 
   void WriteBatchEnvFile(TEnv*) override;
   void ReadBatchEnvFile(TEnv*) override;
   void Print(Option_t* /*option*/ = "") const override;
 
   void SanitizeJobName() const override;
 
   void Run() override;
   void GetBatchSystemParameterList(KVNameValueList&) override;
   void SetBatchSystemParameters(const KVNameValueList&) override;
 
   ClassDefOverride(KV_CCIN2P3_Slurm, 1) //Interface to new CCIN2P3 Slurm batch system (2022)
};
 
#endif