NTupleHelper.h

Go to the documentation of this file.

#ifndef NTUPLEHELPER_H
#define NTUPLEHELPER_H
 
using namespace std;
 
#include <string>
#include <map>
#include "GaudiKernel/NTuple.h"
 
 
// ***************************************************************
//
//  CLASS NTupleHelper
//
//    a utility to keep track of and fill trees
//
// ***************************************************************
 
class NTupleHelper{
 
    public:
 
        // constructor takes a Tuple object created by BOSS
 
        NTupleHelper(NTuple::Tuple* Tree);
 
 
        // generic routine to fill a tree variable;
        //   if the variable doesn't yet exist, it is created
 
        void fillDouble(string name, double  value);
        void fillLong  (string name, long    value);
        void fillArray (string name, string index_name, double* value, int size);
        void fillArrayLong(string name, string index_name, long* value, int size);
        void fillArrayInt(string name, string index_name, int* value, int size);
 
        // write the tree
 
        void write();
 
 
    private:
 
        bool containsEntry(string name);
 
        NTuple::Tuple* m_Tree;
 
 
        map<string, NTuple::Item<double> > m_ntupleDoubleMap;
        map<string, double> m_doubleMap;
        
        map<string, NTuple::Item<long> > m_ntupleLongMap;
        map<string, long> m_longMap;
        
        map<string, NTuple::Array<double> > m_ntupleArrayMap;
        map<string, double*> m_arrayMap;
 
        map<string, NTuple::Array<long> > m_ntupleArrayLongMap;
        map<string, long*> m_arrayLongMap;
        
        map<string, NTuple::Array<int> > m_ntupleArrayIntMap;
        map<string, int*> m_arrayIntMap;
        
        bool m_bookingStage;
 
        string concatName(string prefix, string base, int index);
        string concatName(string prefix, string base);
 
};
 
#endif