TSlurmCluster.h

Go to the documentation of this file.

 
// Copyright (C) 2013-2024 CEA/DES
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published
// by the Free Software Foundation, either version 3 of the License, or any
// later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
// $Id$
// $Author$
// $Date$
// $Revision$
// $State$
 
#ifndef TSLURMCLUSTER_H
#define TSLURMCLUSTER_H
 
#include <iostream>
#include <sstream>
 
using std::cout;
using std::endl;
using std::istringstream;
using std::string;
#define BUF_SIZE 128
//ROOT
#include "Rtypes.h"
 
// Uranie
#include "TCluster.h"
 
namespace URANIE
{
namespace Launcher
{
 
class TSlurmCluster: public TCluster
{
private:
    static const char *nodeList();
 
public:
 
    static bool eligible();
 
    //---------------------------------------------
 
    TSlurmCluster();
#ifdef WITH_LIBSSH
    TSlurmCluster(URANIE::DataServer::TDataServer *tds, TCode *code);
    TSlurmCluster(TCode *code);
#endif
    
    virtual ~TSlurmCluster();
 
    //---------------------------------------------
 
    virtual void init();
    virtual void getCommandPrefix(char *, char *);
 
    //---------------------------------------------
    void setLog()
    {
        _blog = kTRUE;
    }
    void unsetLog()
    {
        _blog = kFALSE;
    }
    void changeLog()
    {
        _blog = _blog ? kFALSE : kTRUE;
    }
    Bool_t getLog()
    {
        return _blog;
    }
    virtual void printLog(Option_t *option = "") ;
 
#ifdef WITH_LIBSSH
 
    void setCluster(TString cluster_name); 
 
    void selectHeader(TString SkeletonHeaderName="header_skeleton.in",TString localHeaderFolderPath=gSystem->pwd() );
    void generateHeader(Option_t *option = "");
    void generateHeaderSplit(Option_t *option = "");
    TString getJobStatusCommand(TString JobIDs);
    TString checkJobStatus(TString JobId, TString Command);
 
#endif
    
     ClassDef(URANIE::Launcher::TSlurmCluster , ID_LAUNCHER)
 
};
 
} // Fin du namespace Launcher
} // Fin du namespace URANIE
#endif