TLauncher.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 TLAUNCHER_H
#define TLAUNCHER_H
 
#define MAX_PATTERN 980
 
//ROOT
#include "Rtypes.h"
#include "TString.h"
 
// Uranie
#include "Launcher.h"
#include "TCode.h"
#include "TDataServer.h"
#include "UMessageLogger.h"
#include "UExceptions.h"
 
#include <iostream>
#include <fstream>
#include <iomanip>
 
using std::cout;
using std::endl;
using std::cerr;
using std::setw;
 
namespace URANIE
{
namespace Launcher
{
class TLauncher: public TNamed
{
protected:
    // the mutable declaration is needed to use the logger in const methods
    URANIE::DataServer::UMessageLogger *_fLogger; // message logger
 
 
    void setNoIntermediateSaved(Bool_t bbool = kTRUE)
    {
        _bnoIntermediateSaved = bbool;
    }
    
public:
    TCode* _code;
    TString _sdraw;             // Variables to visualize
    TString _sopt;              // Option of visualisation
    TString _scut;              // Selection of visualisation
    TString _sdirectory;        
    TString _sworkingDirectory; 
    TString _sPreProCommand;     
    TString _sPostProCommand;    
    TString _sInputNames;
    TString _sInputTypes;
    Bool_t _bsave;        
    Bool_t _bclean;     
    Bool_t _blog;               
    Bool_t _DoLogger;               
    Bool_t _DoMultiDump;               
    Bool_t _bdrawProgressBar; 
    Bool_t _bnoIntermediateSaved; 
    Bool_t _bemptyMasterNode;
    Int_t _nprocs;             
    Int_t _nMaxSave;           
    Int_t _delay; 
    TList *_listOfSaveFile; 
    URANIE::DataServer::TDataServer *_tds;      
    vector<URANIE::DataServer::UEntry*> _InputVector, _OutputVector;
    
    //---------------------------------------------
 
    virtual void createResultsDirectory(Option_t *option = "");
 
    URANIE::DataServer::TDSNtupleD* createNtupleDFile(TString &sntlname, TString &slabelvarname, TString &slabelvartype);
 
 
    void initiateTheEntryVector();
 
public:
    //---------------------------------------------
 
    TLauncher();
    TLauncher(URANIE::DataServer::TDataServer *tds, TCode *code, Option_t *option="");
    virtual ~TLauncher();
 
    //---------------------------------------------
 
    virtual void run(Option_t *option = "");
 
protected:
    virtual void init(Option_t *option = "");
    virtual void terminate(Option_t *option = "");
    virtual void clean(Option_t *option = "");
 
    //---------------------------------------------
 
public:
    void setWorkingDirectory(TString str);
 
    TString getWorkingDirectory()
    {
        return _sworkingDirectory;
    }
 
 
    void setPreProCommand(TString sCommand) { _sPreProCommand = sCommand;}
 
    TString getPreProCommand() { return _sPreProCommand;}
 
 
    void setPostProCommand(TString sCommand) { _sPostProCommand = sCommand;}
    TString getPostProCommand() { return _sPostProCommand;}
 
 
    void setSave(Int_t nSave = -1);
 
    void unsetSave()
    {
        _nMaxSave = MAX_PATTERN;
        _bsave = kFALSE;
    }
 
 
    Bool_t getSave()
    {
        return _bsave;
    }
 
 
    void setClean(Bool_t bbool = kTRUE)
    {
        _bclean = bbool;
    }
 
 
    Bool_t getClean()
    {
        return _bclean;
    }
 
 
    void setDrawProgressBar(Bool_t bbool = kTRUE)
    {
        _bdrawProgressBar = bbool;
    }
 
 
    Bool_t getDrawProgressBar()
    {
        return _bdrawProgressBar;
    }
 
 
    Bool_t getNoIntermediateSaved()
    {
      return _bnoIntermediateSaved;
    }
    
 
    void addSaveFile(TString sfile);
 
    void setVarDraw(TString str, TString select = "", TString opt = "")
    {
        _sdraw = str;
        _scut = select;
        _sopt = opt;
    }
 
 
    //---------------------------------------------
    inline void setProcsPerJob(int procs)
    {
        _nprocs = procs;
    }
    inline int getProcsPerJob()
    {
        return _nprocs;
    }
    void setDelay(int delay)
    {
        _delay = delay;
    }
 
    //---------------------------------------------
    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 = "");
 
    void setEmptyMasterNode()
    {
        _bemptyMasterNode = kTRUE;
    }
    void unsetEmptyMasterNode()
    {
        _bemptyMasterNode = kFALSE;
    }
 
    ClassDef(URANIE::Launcher::TLauncher, ID_LAUNCHER)
    //Classe de
};
 
} // Fin du namespace Launcher
} // Fin du namespace URANIE
#endif