lumiParser.h

Go to the documentation of this file.

/* emacs: this is -*- c++ -*- */
#ifndef  LUMIPARSER_H
#define  LUMIPARSER_H
 
#include <iostream>
#include <fstream>
#include <sstream>
#include <string>
#include <vector>
#include <set>
#include <cstdlib>
#include <cmath>
 
#include "lumiList.h"
#include "computils.h"
 
 
class lumiParser : public lumiList {
 
public:
  
  lumiParser() { }
 
  lumiParser( const std::string& file ) { read( file ); }
 
  void read( const std::string& f ) {
 
    std::string file = f;
    size_t pos = file.find("http");
    if ( pos!=std::string::npos ) { 
      std::string cmd = "wget ";
      cmd += file;
      tail( file, "/" );
      if ( exists( file ) ) { 
    std::string mvcmd = "mv ";
    mvcmd += file + " " + file + ".bak";
    std::system( mvcmd.c_str() );
      }
      std::system( cmd.c_str() ); 
    }
 
    std::cout << "lumiParser file: " << file << std::endl;
 
    std::ifstream input( file.c_str() );
 
    int run;
    std::vector<int> lbstart;
    std::vector<int> lbend;
    
    for( std::string line; getline( input, line ); ) { 
      
      //    std::string s;
      //    std::cout << line << std::endl;
      
      std::string inputline = tolower(line);
      
      if ( inputline.find("run:")!=std::string::npos || 
       inputline.find("<run>")!=std::string::npos ) {
    
 
    std::string label = line;
    chop(  label, "Run:" );
    chop(  label, "<Run>" );
    chomp( label, "</Run>" );
    clean( label, " " );
    
    run = std::atoi(label.c_str());
    lbstart.clear();
    lbend.clear();
    
    //      std::cout << "\n\n" << line << "\trun: " << label << std::endl;
 
    for( std::string fline ; getline( input, fline ) ; ) { 
      if ( tolower(fline).find("from:")!=std::string::npos ) { 
        chop(  fline, "From:" );
        clean( fline, " ");
        std::string start = fline; 
        chomp( start, "to");
        chop(  fline, "to" );
        lbstart.push_back( std::atoi(start.c_str()) );
        lbend.push_back( std::atoi(fline.c_str()) );
      }
      else if ( tolower(fline).find("lbrange")!=std::string::npos ) {
        clean( fline, "\"" );
        clean( fline, " " );
        chop(  fline, "LBRange");
        chomp( fline, "/>" );
        chop(  fline, "Start=" );
        std::string start = fline; 
        chomp( start, "End");
        chop(  fline, "End=");
        lbstart.push_back( std::atoi(start.c_str()) );
        lbend.push_back( std::atoi(fline.c_str()) );
      }
      else break;
    }
        
    for ( unsigned  il=0 ; il<lbstart.size() ; il++ ) this->addRange( run, lbstart[il], lbend[il] );
    
      }
    }    
 
    std::cout << "lumiParser: read " << size() << " runs" << std::endl;
  }
 
  virtual ~lumiParser() { } 
 
private:
 
   
  void chomp( std::string& s, const std::string& regex ) { 
    size_t pos = s.find( regex );
    if ( pos!=std::string::npos ) s.erase( pos, s.size() );
  }
    
  void chop( std::string& s, const std::string& regex ) { 
    size_t pos = s.find( regex );
    if ( pos!=std::string::npos ) s.erase( 0, pos+regex.size() );  
  }
 
  void tail( std::string& s, const std::string& regex ) { 
    size_t pos = s.find( regex );
    while( pos!=std::string::npos ) { 
      s.erase( 0, pos+regex.size() );
      pos = s.find( regex );
    }
  }
  
  void clean( std::string& s, const std::string& regex ) { 
    size_t pos = s.find( regex );
    while( pos!=std::string::npos ) { 
      s.erase( pos, regex.size() );
      pos = s.find( regex );
    }
  }
  
  std::string toupper( const std::string& s ) { 
    std::string t = s;
    for ( unsigned i=0 ; i<t.size() ; i++ ) if ( t[i]>='a' && t[i]<='z' ) t[i] += 'A'-'a';  
    return t;
  }
  
  
  std::string tolower( const std::string& s ) { 
    std::string t = s;
    for ( unsigned i=0 ; i<t.size() ; i++ ) if ( t[i]>='A' && t[i]<='Z' ) t[i] -= 'A'-'a';  
    return t;
  }
  
  
};
 
 
 
inline std::ostream& operator<<( std::ostream& s, const lumiParser& lp ) {
  // return s << *dynamic_cast<const lumiList*>(&_l);
  return s << (const lumiList&)lp;
}
 
 
#endif  // LUMIPARSER_H