TCode.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/>.

#ifndef TCODE_H
#define TCODE_H

// URANIE
#include "TAttribute.h"
#include "TDataServer.h"

#include "TInputFile.h"
#include "TOutputFile.h"

namespace URANIE {
namespace Launcher {


class TCode: public TNamed {
public:
    //---------------------------------------------
    TCode();
    TCode(const TString &name, const TString &title);
    TCode(URANIE::DataServer::TDataServer *tds, TString scmd);

    TCode(URANIE::Launcher::TCode &mycode);

    virtual ~TCode();

    //---------------------------------------------
    TInputFile * getInputFile(TString sfile);
    TInputFile * getInputFile(Int_t nfile);
    TInputFile * getLastInputFile();
    TOutputFile * getOutputFile(Int_t nfile);

    TOutputFile * getOutputFile(TString sfile);


    //---------------------------------------------
    void setWorkingDirectory(TString sPathName = "");

    TString getWorkingDirectory() {
        return _sWorkingDirectory;
    };
    void setReferenceDirectory(TString str) {
        _sReferenceDirectory = str;
    };
    TString getReferenceDirectory() {
        return _sReferenceDirectory;
    };
    void setCommand(TString str) {
        _sCommand = str;
    };
    TString getCommand() {
        return _sCommand;
    };
    Int_t getNOutputFiles() {
        return (Int_t)(_outputFiles.size());
    };
    Int_t getNOutputAttributes();
    Int_t getNOutputAttributes(Int_t nfile);
    Int_t getNInputFiles() {
        return (Int_t)(_listOfInputFiles->GetEntries());
    };
    Int_t getNInputAttributes();
    Int_t getNInputAttributes(Int_t nfile);
    URANIE::DataServer::TAttribute * getInputAttribute(Int_t nfile, Int_t natt);
    URANIE::DataServer::TAttribute * getAttribute(Int_t nfile, Int_t natt);
    TString getInputVarNames();
    TString getOutputVarNames();
    URANIE::DataServer::TDataServer * getTDataServer() {
        return _tds;
    };
    void setTDataServer(URANIE::DataServer::TDataServer *atds) {
        _tds = atds;
        //      UpdateInputList();
    };

    //---------------------------------------------
    virtual void runTest(Option_t *option = "");
    virtual void initFileType(Option_t *option = "");
    virtual void init(Option_t *option = "");
    virtual void initRun(forkcode *fjob, Option_t *option = "");
    virtual void preTraitment(forkcode *fjob, Double_t *val, Int_t *index = NULL, Option_t *option = "");
    virtual void prepareEntries(forkcode *fjob, vector<URANIE::DataServer::UEntry*> *input);
    virtual void preTraitment(forkcode *fjob, vector<URANIE::DataServer::UEntry*> *input, Option_t *option = "");
    virtual void run(forkcode *fjob, Option_t *option = "");
    virtual void start(forkcode *fjob, Option_t *option = "");
    virtual Bool_t postTraitment(forkcode *fjob, Int_t &nout, Double_t *dvalout, Option_t *option = "", TList *lstOut = NULL);
    virtual Bool_t postTraitment(forkcode *fjob, vector<URANIE::DataServer::UEntry*> *out, Option_t *option = "");
    virtual void terminate(Option_t *option = "");

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

    void createDirectory(TString sPathName, TString sDirectoryName = "");

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

    void resetCurrentSimulation();
    virtual void cleanOutputFiles(forkcode *fjob, Option_t *option = "");

    //---------------------------------------------
    void addInputFile(TInputFile *ifile);
    void addInputFile(TString str);
    void addOutputFile(TOutputFile *ifile);

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

    ClassDef(URANIE::Launcher::TCode, ID_LAUNCHER)

    //---------------------------------------------
    protected:
        vector<TOutputFile *> _outputFiles;         
        TList *_listOfInputFiles;                   
        TString _sWorkingDirectory;                 
        TString _sReferenceDirectory;               
        TString _sCommand;                          
        Int_t _nCurrentSimulation;            
        Int_t _nOfInput;              
        Int_t _nOfInputToSubstitute; 
        Int_t *_nIndexOfInputToSubstitute;    
        Bool_t _blog;                               
        URANIE::DataServer::TDataServer *_tds;      

};

} // End of namespace Launcher
} // End of namespace URANIE

#endif
// End of File