NtupleReader/src/MyTextFileReader.cxx

//+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
// 10.12.2006, AUTHOR: STEFFEN KAISER
//+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

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

//::::::::::::::::::
//:: HEADER FILES ::
//::::::::::::::::::

#include "NtupleReader/MyTextFileReader.h"


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

using namespace std;


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

//:::::::::::::::::
//:: METHOD init ::
//:::::::::::::::::

void MyTextFileReader::init(TString textfilename) {
  
    m_textfilename = textfilename;
    return;
}

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

//:::::::::::::::::::::
//:: METHOD destruct ::
//:::::::::::::::::::::

void MyTextFileReader::destruct(void) {
    
    return;
    
}

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

//:::::::::::::::::::::::::::::
//:: METHOD get_textfilename ::
//:::::::::::::::::::::::::::::


TString 
MyTextFileReader::get_textfilename(void) {

    return m_textfilename;

}

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

//::::::::::::::::::::::::::::::::::::::::
//:: TEMPLATE METHOD retrieve_from_file ::
//::::::::::::::::::::::::::::::::::::::::

template <class MyType> MyType
MyTextFileReader::retrieve_from_file(TString key_string, TString filename){

    // variables // 

    MyType value;
    TString aBuffer;
    TString dummy;
    
    TString m_filename = filename;
    
    // if no filename is given read the default (global) file //
  
    if(m_filename=="")  
        m_filename = m_textfilename;
  
    ifstream myFile(m_filename);
    
    // exit if file can't be opened //

    if (myFile.fail()){
        std::cerr << std::endl
                  << "Error: Could not open " << m_filename
                  << std::endl;
        exit(1);
    }

    // loop over lines and read value after 'key_string = ' //

    while ( !myFile.eof() ){
    
        aBuffer.ReadToken( myFile );
        //if ( aBuffer.Index(key_string) != kNPOS ){
        if ( aBuffer==key_string){
            myFile >> dummy;
            myFile >> value;
        }
        
        if (myFile.eof()) {
            break;
        }
    
    }
  
  
    return value;
}

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

//:::::::::::::::::::::::
//:: METHOD get_string ::
//:::::::::::::::::::::::

TString 
MyTextFileReader::get_string(TString key_string){

    // the template method above can only be called from an other class //
    // if this method is in this class -> very strange!                  //   

    TString string;
    string = retrieve_from_file<TString>(key_string);
  
    return string;
  
}

Bool_t 
MyTextFileReader::get_bool(TString key_string){

    // the template method above can only be called from an other class //
    // if this method is in this class -> very strange!                  //   

    Bool_t flag;
    flag = retrieve_from_file<Bool_t>(key_string);
  
    return flag;
  
}

int 
MyTextFileReader::get_int(TString key_string){

    // the template method above can only be called from an other class //
    // if this method is in this class -> very strange!                  //   

    int flag;
    flag = retrieve_from_file<int>(key_string);
  
    return flag;
  
}




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

//::::::::::::::::::::::::::::::
//:: METHOD fill_cutvalue_map ::
//::::::::::::::::::::::::::::::

void 
MyTextFileReader::fill_cutvalue_map(TString cutstring, TString filename){
    m_cutvalue_map.clear();
    // variables //

    TString sdummy;
    TString eqdummy;
    Double_t vdummy;
  
    int nvalues = -1;
    TString aBuffer;
  
    TString m_filename = filename;
  
    // if no filename is given read the default (global) file // 

    if(m_filename=="")  
        m_filename = m_textfilename;
  
    ifstream myFile(m_filename);
  
    // exit if file can't be opened //

    if (myFile.fail()){
        std::cerr << std::endl
                  << "Error: Could not open " << m_filename
                  << std::endl;
        exit(1);
    }
  
    // loop over lines and fill cutvalues and names in map     //
    // between 'cutstring:' and 'end_of_cutstring' in textfile //
  
    while ( !myFile.eof() ){
    
        if ( nvalues < 0 ){
      
            aBuffer.ReadToken( myFile );
            if ( aBuffer.Index(cutstring+":") != kNPOS ){ 
                nvalues = 0;
            }
        }
        else{
   
            myFile >> sdummy;
            if(sdummy=="#end_of_"+cutstring) break;
            
            if(sdummy.Contains("#") || sdummy==""){ 
                continue;
            }

            myFile >> eqdummy;
            myFile >> vdummy;
      
//            cout << sdummy << "\t" << eqdummy << "\t" << vdummy << endl;
            
            m_cutvalue_map.insert(ValuePair(sdummy,vdummy));

            nvalues++;
      
        }
    
        if (myFile.eof()) {
            break;
        }
    
    }

//     // print values stored in the map // 
// 
 //    std::cout << "\n**********************************************" << std::endl;
//     std::cout << "** in MyTextFileReader::fill_cutvalue_map() " 
//            << std::endl;
//     std::cout << "** cutstring: " << cutstring 
//            << std::endl;
//     std::cout << "** cutvalues in m_cutvalue_map:             " 
//            << std::endl;
    
//     maptype::const_iterator it;
//     for( it=m_cutvalue_map.begin(); it!=m_cutvalue_map.end(); it++ ){
        
//      std::cout << "** value: " << (*it).first << " : " 
//                << (*it).second << std::endl;
        
//     }
//     std::cout << "**********************************************\n" << std::endl;


}

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

//:::::::::::::::::::::::::
//:: METHOD get_cutvalue ::
//:::::::::::::::::::::::::


Double_t 
MyTextFileReader::get_cutvalue(TString variable){

    // return value with name 'variable' from map //
    // if value doesn't exists than exit          //
   
    Double_t value;
 
    maptype::const_iterator iter = m_cutvalue_map.find(variable);
    if( iter != m_cutvalue_map.end() ){
        value = (*iter).second;
    }
    else{
        std::cerr << std::endl
                  << "Error: Could not find variable: " 
                  <<  variable
                  << std::endl;
        exit(1);
    }

    return value;

}



// ifstream* 
// MyTextFileReader::read_textfile(TString filename){
  
  
//   ifstream file(filename);

//   if (file.fail()){
//     std::cerr << std::endl
//            << "Error in MyTextFileReader::read_textfile(): Could not open " 
//            << filename
//            << std::endl;
//     exit(1);
//   }
  
//   return &file;

// }

---