TLauncherByStep.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 TLAUNCHERBYSTEP_H
#define TLAUNCHERBYSTEP_H

#include "TNamed.h"

#include <iostream>
#include <fstream>

using std::cout;
using std::endl;

// ROOT
#include "Rtypes.h"

// Uranie
#include "TLauncher.h"
#include "Launcher.h"

namespace URANIE
{
namespace Launcher
{
class TLauncherByStep: public TLauncher
{

public:
    //---------------------------------------------
    TLauncherByStep(URANIE::DataServer::TDataServer *tds, TCode *code, Option_t *option="");
    virtual ~TLauncherByStep();
    //---------------------------------------------

    virtual void preTreatment(Option_t *option = "");

    virtual void run(Option_t *option = "");

    virtual void postTreatment(Option_t *option = "");

    //---------------------------------------------
    virtual void printLog(Option_t *option = "");

    //-----------------------------------------------

    void setNumberOfCores(Int_t ncores)
    {
        _nAllocatedCores = ncores;
    }
    Int_t getNumberOfCores() const
    {
        return _nAllocatedCores;
    }

    void setAllocatedTime(Int_t nseconds)
    {
        _nAllocatedSeconds = nseconds;
    }
    Int_t getAllocatedTime() const
    {
        return _nAllocatedSeconds;
    }

    void setProjectName(TString project_name)
    {
        _sProjectName = project_name;
    }
    TString getProjectName() const
    {
        return _sProjectName;
    }

    void setClusterName(TString cluster_name)
    {
        _sClusterName = cluster_name;
    }
    TString getClusterName() const
    {
        return _sClusterName;
    }

    void addExtraDirective(const TString& directive)
    {
        _sListExtraDirectives.push_back(directive);
    }

private:
    Int_t _nAllocatedSeconds;
    Int_t _nAllocatedCores;
    TString _sProjectName;
    TString _sClusterName;
    vector<TString> _sListExtraDirectives;

    ClassDef(URANIE::Launcher::TLauncherByStep, ID_LAUNCHER)
    //The description
};

} // End Of namespace Launcher
} // End Of namespace URANIE

#endif
//  End Of File TLauncherByStep.h