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="" );
    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