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 "TLauncherEnv.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 TLauncherEnv
{
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:
    TString _sdraw;             // Variables to visualize
    TString _sopt;              // Option of visualisation
    TString _scut;              // Selection of visualisation
    TString _sPreProCommand;     
    TString _sPostProCommand;    
    TString _sInputNames;
    TString _sInputTypes;
    Bool_t _bsave;        
    Bool_t _bclean;     
    Bool_t _DoLogger;               
    Bool_t _DoMultiDump;               
    Bool_t _bdrawProgressBar; 
    Bool_t _bnoIntermediateSaved; 
    Bool_t _bemptyMasterNode;
    Int_t _nprocs;             
    Int_t _nMaxSave;           
    TList *_listOfSaveFile; 
    URANIE::DataServer::TDataServer *_tds;      
    vector<URANIE::DataServer::UEntry*> _InputVector, _OutputVector;
    
    //---------------------------------------------
    // ENV 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 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;
    }

    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