SimpleNtp.C

Go to the documentation of this file.

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

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

class SimpleNtp {
public:
                
        SimpleNtp(TString name, TString title, int maxsize = 100) {
                fTree    = new TTree(name, title);
                fValues  = new Float_t[maxsize];
                fBranchMap.clear();
                fSize=0;
                fMax=maxsize;
        };
                
        ~SimpleNtp(){delete fTree; delete[] fValues;}
        
        void Column(TString name, Float_t value);
        void DumpData() { fTree->Fill(); }
        
private:
        TTree *fTree;
        std::map<TString, Int_t> fBranchMap;
        Float_t *fValues;
        int fSize;
        int fMax;
};

void SimpleNtp::Column(TString name, Float_t value)
{
        if (fBranchMap.find(name)==fBranchMap.end())
        {
                if (fSize<fMax) 
                {
                        fTree->Branch(name.Data(),&(fValues[fSize]));
                        fBranchMap[name]=fSize++;
                }
                else
                {
                        cout <<" - WARNING - max size reached, branch '"<<name.Data()<<"' is not created!"<<endl;
                        return;
                }
        }
        fValues[fBranchMap[name]]=value;
}