NtupleReader/src/MyDatasetInfo.cxx

//+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
// 09/2007, AUTHOR: Mandfred Groh, Steffen Kaiser
//+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

// standard C++ //
#include <iostream>
#include <fstream>

// STL //
#include <string>

// ROOT //
#include "TObjString.h"

// NtupleReader reader //
#include "NtupleReader/MyDatasetInfo.h"

//::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::
//:: IMPLEMENTATION OF METHODS DEFINED IN THE CLASS MyDatasetInfo ::
//::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::

//::::::::::::::::::::::::
//:: NAMESPACE SETTINGS ::
//::::::::::::::::::::::::

using namespace std;

//*****************************************************************************

//::::::::::::::::::::::::::::::::
//:: METHOD set/get_AtlFastFlag ::
//::::::::::::::::::::::::::::::::

void MyDatasetInfo::set_AtlFastFlag(const bool & flag){
    
    m_atlfastflag = flag;
}

bool MyDatasetInfo::get_AtlFastFlag(void) const {
    
    return m_atlfastflag;
}

//*****************************************************************************

//::::::::::::::::::::::::::::::::::
//:: METHOD set/get_trackjet_flag ::
//::::::::::::::::::::::::::::::::::

void MyDatasetInfo::set_trackjet_flag(const bool & flag){
    
    m_trackjetflag = flag;
}

bool MyDatasetInfo::get_trackjet_flag(void) const {
    
    return m_trackjetflag;
}

//*****************************************************************************

//::::::::::::::::::::::::::::::::
//:: METHOD set/get_vertex_flag ::
//::::::::::::::::::::::::::::::::

void MyDatasetInfo::set_vertex_flag(const bool & flag){
    
    m_vertexflag = flag;
}

bool MyDatasetInfo::get_vertex_flag(void) const {
    
    return m_vertexflag;
}

//*****************************************************************************

//::::::::::::::::::::::::::::
//:: METHOD set/get_release ::
//::::::::::::::::::::::::::::

void MyDatasetInfo::set_release(const int & release){
    
    m_release = release;
}

int MyDatasetInfo::get_release(void) const {
    
    return m_release;
}

//*****************************************************************************

//:::::::::::::::::::::::::::::::
//:: METHOD set/get_svnversion ::
//:::::::::::::::::::::::::::::::

void MyDatasetInfo::set_svnversion(const int & svnversion){
    
    m_svnversion = svnversion;
}

int MyDatasetInfo::get_svnversion(void) const {
    
    return m_svnversion;
}

//*****************************************************************************

//::::::::::::::::::::::::::
//:: METHOD set/get_scale ::
//::::::::::::::::::::::::::

void MyDatasetInfo::set_scale(const double & scale){
    
    m_scale = scale;
}

double MyDatasetInfo::get_scale(void) const {
    
    return m_scale;
}

//*****************************************************************************

//::::::::::::::::::::::
//:: METHOD set_TFile ::
//::::::::::::::::::::::

void MyDatasetInfo::set_TFile(TFile* aan_file) {
        
    m_aan_file = aan_file;
    return;

}

//*****************************************************************************

//:::::::::::::::::::::::::
//:: METHOD get_aan_file ::
//:::::::::::::::::::::::::

TFile* 
MyDatasetInfo::get_aan_file(void) const {

    if (m_aan_file==NULL){
        cout << "Error In DatasetInfo: TFile not found!"<<endl;
    }
    return m_aan_file;

}

//*****************************************************************************

//:::::::::::::::::::::
//:: METHOD get_tree ::
//:::::::::::::::::::::

TTree* 
MyDatasetInfo::get_tree(string tree_name){
   
    if (m_aan_file==NULL){
        cout << "In DatasetInfo: TFile not found!"<<endl;
        return NULL;
    }

    TTree* tmp_tree = NULL;
    tmp_tree = (TTree*)m_aan_file->Get(tree_name.c_str());
    if (tmp_tree==NULL){
        cout<<"In DatasetInfo: TTree with name "<<endl
            <<tree_name<<" not found!"<<endl;
        //exit(1);
    }
    
    return tmp_tree;

  }

//*****************************************************************************

//::::::::::::::::::::::::::
//:: METHOD get_generator ::
//::::::::::::::::::::::::::

string 
MyDatasetInfo::get_generator(){

    TString path       = "unknown";
    TString aod_file   = "unknown";
    TString tmp_gen    = "unknown";
    //TTree* tmp_tree    = (TTree*)m_aan_file->Get("aod_file_collection");
    TTree* tmp_tree    = get_tree("aod_file_collection");
    Int_t n_generators = 0;

    // if AOD filename tree is not in the aan file
    if (tmp_tree==NULL){
        cerr<<"In DatasetInfo: TTree \"aod_file_collection\" not found!"<<endl
            <<"=> Can not determine generator!"<<endl;
    }
    // get the AOD filename
    else{
        char ch_array_helper[250];
        tmp_tree->SetBranchAddress("aod_files",&ch_array_helper);
        tmp_tree->GetEntry(0);
        path = ch_array_helper;
    }
    //cout << "path:" << path <<  endl;

    TObjArray*  TOarray  = path.Tokenize("/");
    TObjString* TOstring = (TObjString*)TOarray->Last();
    aod_file = TOstring->GetString();
    aod_file = path;
    aod_file.ToLower();
    delete TOarray; 
    // cout << "aod_file:" << aod_file << endl;
        
    // look for a known generator name in the filename
    if (aod_file.Contains("herwig")){
        tmp_gen += "Herwig";
        n_generators++;
    }
    if (aod_file.Contains("alpgen")){
        tmp_gen += "Alpgen";
        n_generators++;
    }
    if (aod_file.Contains("mcatnlo")){
        tmp_gen += "McAtNlo";
        n_generators++;
    }
    if (aod_file.Contains("pythia")){
        tmp_gen += "Pythia";
        n_generators++;
    }
    if (aod_file.Contains("sherpa")){
        tmp_gen += "Sherpa";
        n_generators++;
    }
    if (aod_file.Contains("jimmy_susy")){
        tmp_gen += "jimmy_susy";
        n_generators++;
    }
    if (aod_file.Contains("A10")||aod_file.Contains("a3")){
        cout <<"ASD1"<<endl;
        tmp_gen += "Pythia";
        n_generators++;
    }

    //if generator was found remove "unknown"
    if(!tmp_gen.EndsWith("unknown")){
        tmp_gen.ReplaceAll("unknown","");
    }
    
    if(n_generators>1){
        cout << endl
             << "Warning In DatasetInfo: More then one generator found in" << endl
             << "\tfile:      " << aod_file << endl
             << "\tgenerator: " << tmp_gen  << endl
             << endl;
    }

    return string(tmp_gen);

}

---