NTupleHelper.cxx

Go to the documentation of this file.

 
#include "NTupleHelper/NTupleHelper.h"
#include <sstream>
 
// ********************************************************************
//    HELPER FUNCTIONS
// ********************************************************************
 
NTupleHelper::NTupleHelper(NTuple::Tuple* Tree){
    m_Tree = Tree;
    m_bookingStage = true;
 
    if( m_Tree == NULL )
        cout << "ERROR:  null tree pointer -- "
            << "check for duplicate final states in configuration"
            << endl; 
    assert( m_Tree != NULL );
}
 
 
void
NTupleHelper::write(){
 
    for (map<string,double>::iterator mapItr = m_doubleMap.begin();
            mapItr != m_doubleMap.end(); mapItr++){
        m_ntupleDoubleMap[mapItr->first] = mapItr->second;
    }
 
    for (map<string,long>::iterator mapItr = m_longMap.begin();
            mapItr != m_longMap.end(); mapItr++){
        m_ntupleLongMap[mapItr->first] = mapItr->second;
    }
 
    for (map<string,double*>::iterator mapItr = m_arrayMap.begin();
            mapItr != m_arrayMap.end(); mapItr++){
        for(int i=0; i< 100; i++) { 
            m_ntupleArrayMap[mapItr->first][i] =  *(mapItr->second + i);
        }
    }
 
    for (map<string,long*>::iterator mapItr = m_arrayLongMap.begin();
            mapItr != m_arrayLongMap.end(); mapItr++){
        for(int i=0; i< 100; i++) { 
            m_ntupleArrayLongMap[mapItr->first][i] =  *(mapItr->second + i);
        }
    }
 
    for (map<string,int*>::iterator mapItr = m_arrayIntMap.begin();
            mapItr != m_arrayIntMap.end(); mapItr++){
        for(int i=0; i< 100; i++) { 
            m_ntupleArrayIntMap[mapItr->first][i] =  *(mapItr->second + i);
        }
    }
 
    m_Tree->write();
 
    for (map<string,double*>::iterator mapItr = m_arrayMap.begin();
            mapItr != m_arrayMap.end(); mapItr++){
        delete mapItr->second;
    }
    for (map<string,long*>::iterator mapItr = m_arrayLongMap.begin();
            mapItr != m_arrayLongMap.end(); mapItr++){
        delete mapItr->second;
    }
    for (map<string,int*>::iterator mapItr = m_arrayIntMap.begin();
            mapItr != m_arrayIntMap.end(); mapItr++){
        delete mapItr->second;
    }
    m_bookingStage = false;
    //cout<<mapItr->first<<" "<<sizeof(*(mapItr->second))<<endl;
    //for(int i=0; i<m_ntupleLongMap["idx_mc"]; i++) {
}
 
 
void NTupleHelper::fillDouble(string name, double value){
    if (m_bookingStage && !containsEntry(name)){
        m_Tree->addItem(name,m_ntupleDoubleMap[name]);
    }
    if (!m_bookingStage && !containsEntry(name)){
        cout << "NTUPLEHELPER:  Variable " << name << " has not been booked." << endl;
        exit(0);
    }
    m_doubleMap[name] = value;
}
 
void NTupleHelper::fillLong(string name, long value){
    if (m_bookingStage && !containsEntry(name)){
        m_Tree->addItem(name,m_ntupleLongMap[name]);
    }
    if (!m_bookingStage && !containsEntry(name)){
        cout << "NTUPLEHELPER:  Variable " << name << " has not been booked." << endl;
        exit(0);
    }
    m_longMap[name] = value;
}
 
 
 
void NTupleHelper::fillArray(string name, string index_name, double* value, int size){
 
    if (m_bookingStage && !containsEntry(index_name)){
        m_Tree->addItem(index_name, m_ntupleLongMap[index_name], 0, 100);
    }
 
    m_longMap[index_name]=size;
 
    if (m_bookingStage && !containsEntry(name)){
        m_Tree->addIndexedItem(name, m_ntupleLongMap[index_name], m_ntupleArrayMap[name]);
    }
    
    if (!m_bookingStage && !containsEntry(name)){
        cout << "NTUPLEHELPER:  Variable " << name << " has not been booked." << endl;
        exit(0);
    }
    
    if (!m_bookingStage && !containsEntry(index_name)){
        cout << "NTUPLEHELPER:  Variable " << index_name << " has not been booked." << endl;
        exit(0);
    }
 
    m_arrayMap[name]= new double[100];
    for(int i=0; i<size; i++){
        //printf("value %2d = %10f\n", i, *(value+i));
        m_arrayMap[name][i] = *(value+i);
    }
}
 
void NTupleHelper::fillArrayLong(string name, string index_name, long* value, int size){
    
    if (m_bookingStage && !containsEntry(index_name)){
        m_Tree->addItem(index_name, m_ntupleLongMap[index_name], 0, 100);
    }
    
    m_longMap[index_name]=size;
 
    if (m_bookingStage && !containsEntry(name)){
        m_Tree->addIndexedItem(name, m_ntupleLongMap[index_name], m_ntupleArrayLongMap[name]);
    }
    
    if (!m_bookingStage && !containsEntry(name)){
        cout << "NTUPLEHELPER:  Variable " << name << " has not been booked." << endl;
        exit(0);
    }
    
    if (!m_bookingStage && !containsEntry(index_name)){
        cout << "NTUPLEHELPER:  Variable " << index_name << " has not been booked." << endl;
        exit(0);
    }
 
    m_arrayLongMap[name]= new long[100];
    for(int i=0; i<size; i++){
        //printf("value %2d = %10f\n", i, *(value+i));
        m_arrayLongMap[name][i] = *(value+i);
    }
}
 
void NTupleHelper::fillArrayInt(string name, string index_name, int* value, int size){
    
    if (m_bookingStage && !containsEntry(index_name)){
        m_Tree->addItem(index_name, m_ntupleLongMap[index_name], 0, 100);
    }
    
    m_longMap[index_name]=size;
 
    if (m_bookingStage && !containsEntry(name)){
        m_Tree->addIndexedItem(name, m_ntupleLongMap[index_name], m_ntupleArrayIntMap[name]);
    }
    
    if (!m_bookingStage && !containsEntry(name)){
        cout << "NTUPLEHELPER:  Variable " << name << " has not been booked." << endl;
        exit(0);
    }
    
    if (!m_bookingStage && !containsEntry(index_name)){
        cout << "NTUPLEHELPER:  Variable " << index_name << " has not been booked." << endl;
        exit(0);
    }
 
    m_arrayIntMap[name]= new int[100];
    for(int i=0; i<size; i++){
        //printf("value %2d = %10f\n", i, *(value+i));
        m_arrayIntMap[name][i] = *(value+i);
    }
}
 
 
 
bool
NTupleHelper::containsEntry(string name){
 
    map<string,double>::iterator mapItr1 = m_doubleMap.find(name);
    if (mapItr1 != m_doubleMap.end()) return true;
    
    map<string,long>::iterator mapItr2 = m_longMap.find(name);
    if (mapItr2 != m_longMap.end()) return true;
    
    map<string,double*>::iterator mapItr3 = m_arrayMap.find(name);
    if (mapItr3 != m_arrayMap.end()) return true;
    
    map<string,long*>::iterator mapItr4 = m_arrayLongMap.find(name);
    if (mapItr4 != m_arrayLongMap.end()) return true;
    
    map<string,int*>::iterator mapItr5 = m_arrayIntMap.find(name);
    if (mapItr5 != m_arrayIntMap.end()) return true;
    
    return false;
}
 
string 
NTupleHelper::concatName(string tag, string base, int index){
    stringstream name;
    name << tag;
    name << base;
    name << "P";
    name << index;
    return name.str();
}
 
string 
NTupleHelper::concatName(string tag, string base){
    stringstream name;
    name << tag;
    name << base;
    return name.str();
}