TDSNtupleD.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 TDSNtupleD_H
#define TDSNtupleD_H

// ROOT
#include "TROOT.h"
#include "TTree.h"
#include "TMatrixD.h"

// URANIE
#include "DataServer.h"
#include "UEntry.h"

#define SIZENAME std::string("size")

namespace URANIE
{
namespace DataServer
{
class TDSNtupleD: public TTree
{
public:

    TDSNtupleD();
    
    ~TDSNtupleD();


    TDSNtupleD(TTree* ntd, const char* varlist="*", const char* cut="", bool NewIsMainOne=false);


    TDSNtupleD(const char* name, const char* title, const char* varlist,
               const char *vartype, Int_t bufsize = 32000);


    TDSNtupleD(const char* name, const char* title, const char* varlist,
               Int_t bufsize = 32000);


    TDSNtupleD(const char* name, const char* title, const char* varlist,
               TMatrixD data, Int_t bufsize = 32000);

    

    virtual Int_t  Fill();


    Int_t     Fill(Double_t* AllValues);
    

    Int_t     Fill(Double_t x0, Double_t x1, Double_t x2=0, Double_t x3=0, Double_t x4=0,
                   Double_t x5=0, Double_t x6=0, Double_t x7=0, Double_t x8=0, Double_t x9=0,
                   Double_t x10=0, Double_t x11=0, Double_t x12=0, Double_t x13=0, Double_t x14=0);

    

    Int_t     Fill(std::string line, int IterIndex=-1);
    

    Int_t     Fill(std::vector<URANIE::DataServer::UEntry*> *event);


    Int_t     Fill(std::vector<URANIE::DataServer::UEntry*> *vinput, std::vector<URANIE::DataServer::UEntry*> *voutput);


    


    TBranch* Branch(const char* name, void* address, const char* leaflist,
                    Int_t bufsize = 32000);
    

    void ResetBranchAddress(TBranch *branch);

    void ResetBranchAddresses();
    
 


    Double_t *GetArgs();
    

    Double_t* getBranchData(const char* branchName, const char* selection = "",
                            Int_t* nbDataPtr = NULL);
    

    void copyBranchData(Double_t* copyArray, int arraySize,
                        const char* branchName, const char* selection = "",
                        Int_t* nbDataPtr = NULL);
    

    void extractData(Double_t* dataOut, Int_t arraySize,
                     const char* varexpinput = "*", const char* selection = "",
                     const Option_t* option = "line", Int_t *nbDataPtr = NULL);


    void extractData(TMatrixD* matrix, const char* varexpinput = "*",
                     const char* selection = "");


    Long64_t ReadFile(const char* filename);
    

    void Merger(TDSNtupleD *file, const std::string newbrname, const std::string oldbrname, bool tdsmerge=false, const char* iter1="", const char* iter2="");

    void Browse(TBrowser *b);  
    


    virtual Int_t     GetNvar() const { return _fNvar; }
    UInt_t getNVar(){return _fNvar;};
    std::string getVarName(){return _varname;};
    std::string getVarType(){return _vartype;};
    int getVectSize(){return _vfArgs.size();};
    
    ClassDef(URANIE::DataServer::TDSNtupleD, ID_DATASERVER);    
protected:

    UInt_t       _i;                     
    UInt_t       _fNvar;                  
    std::vector<UEntry*> _vfArgs; 
    std::vector<Double_t>    _vfDArgs;                
    std::vector<std::string*> _vfSArgs;     
    std::vector<std::vector<Double_t>* > _vfVArgs;  
    std::string   _varname; 
    std::string   _vartype; 
    
    std::map<std::string,unsigned int> _mTypeNb;      
    std::map<unsigned int,unsigned int> _mVarTypeNb;  
    
    

    void _initTTree(const char* name, const char* title);
    

    void _branchConstruction(Int_t bufsize);


    void ClearVectorsAndMaps();


    //Flag special because windows complains a lot
#ifdef WIN32 
    Int_t Branch(TCollection* list, Int_t bufsize = 32000, Int_t splitlevel = 99, const char* name = "")
        {
            return TTree::Branch(list, bufsize, splitlevel, name );
        }
    Int_t Branch(TList* list, Int_t bufsize = 32000, Int_t splitlevel = 99)
        {
            return TTree::Branch(list, bufsize , splitlevel );    
        }
    Int_t Branch(const char* folder, Int_t bufsize = 32000, Int_t splitlevel = 99)
        {
            return TTree::Branch(folder, bufsize , splitlevel );
        }
    TBranch* Branch(const char* name, const char* classname, void* addobj, Int_t bufsize = 32000, Int_t splitlevel = 99)
        {
            return TTree::Branch(name, classname, addobj, bufsize, splitlevel );
        }
#endif
    
};

}
}

#endif