PndSimpleNtuple.h

Go to the documentation of this file.

// -------------------------------------------------------------------------------
// ---------                    PndSimpleNtuple                           --------
// -------------------------------------------------------------------------------
//
// Represents a simplified N-Tuple based on a ROOT TTree.
//
// Branches of type double, float or int are created on the fly with the command 
//
//   'Column(TString name, double/float/int value);'
//
// At the end of each event (i.e. when the N-Tuple should be filled) either
//
//   'DumpData()'    or 
//   'AcceptData()' 
//
// has to be called. The method 'AcceptData()' is a conditional fill in case a
// certain precut is fulfilled, which can be set either in the constructor with
//
//   PndSimpleNtuple(TString name, TString title, TString <precut>)
//
// or with the method
//
//   SetPrecut(TString precut)
//
// which can even be changed inbetween. The precut is a TFormula directly based
// on the branches of the TTree.
//
// -------------------------------------------------------------------------------

#ifndef PndSimpleNtuple_H
#define PndSimpleNtuple_H 1

#include "TString.h"
#include "TTree.h"
#include "TTreeFormula.h"
#include <iostream>
#include <map>
#include <vector>

class PndSimpleNtuple {
public:
        PndSimpleNtuple(TString name, TString title, TString precut="");
                
        ~PndSimpleNtuple(){if (fFml!=0) delete fFml;}
        
        void Column(const TString &name, double value);     // sets double value of a variable 'name'
        void Column(const TString &name, float value);      // sets float  value of a variable 'name'
        void Column(const TString &name, int value);        // sets integer value of a variable 'name'
        void Column(const TString &name, bool value);       // sets bool value of a variable 'name'

        void ColumnD(const TString &name, double value) {Column(name, value);} // sets double value of a variable 'name'
        void ColumnF(const TString &name, float value)  {Column(name, value);} // sets float  value of a variable 'name'
        void ColumnI(const TString &name, int value)    {Column(name, value);} // sets integer value of a variable 'name'
        void ColumnB(const TString &name, bool value)   {Column(name, value);} // sets bool value of a variable 'name'

        //void Column(TString name, Double_t *vpointer, TString idxvar);  // sets double value array of a variable 'name', index variable 'idxvar'
        //void Column(TString name, Float_t *vpointer,  TString idxvar);  // sets float value array of a variable 'name', index variable 'idxvar'
        //void Column(TString name, Int_t *vpointer,    TString idxvar);  // sets integer value array of a variable 'name', index variable 'idxvar'
        //void Column(TString name, Bool_t *vpointer,   TString idxvar);  // sets bool value of array a variable 'name', index variable 'idxvar'

        bool DumpData() { fTree->Fill(); fTmpTree->Reset(); return true; }           // writes the current event
        bool Accept();                                                               // check whether entry fulfills cut
        bool AcceptData() {if (Accept()) {DumpData(); return true;} return false; }  // writes current event if accepted
        
        TTree* GetTree() {return fTree;}
        double GetCurrentValue(TString name);
        void SetPrecut(TString precut); 
        bool BranchExists(TString name) { return (fBrTypes.find(name)!=fBrTypes.end()); }
        int  ShowBranches();
        
private:
        TTree     *fTree;
        TTree     *fTmpTree;
        
        std::map<TString, double>  fDValues;
        std::map<TString, float>   fFValues;
        std::map<TString, int>     fIValues;
        std::map<TString, char>    fBValues;
        
        std::map<TString, int>     fBrTypes;
                
        TString      fPrecut;       // precut in TTreeFormula notation
        TTreeFormula *fFml;         // the actual formula
};

#endif