anajob.cc

Go to the documentation of this file.

#include "lcio.h"

#include "IO/LCReader.h"
#include "IMPL/LCTOOLS.h"
#include "EVENT/LCRunHeader.h" 

#include <cstdlib>
#include <sstream>

static std::vector<std::string> FILEN ; 

using namespace std ;
using namespace lcio ;

int main(int argc, char** argv ){

  // read file names from command line (only argument) 
  if( argc < 2) {
    cout << " usage:  anajob <input-file1> [[input-file2],...]" << endl << endl ;
    cout << "  set the environment variable LCIO_READ_COL_NAMES to a space separated list" << endl ;
    cout << "  of collection names that you would like to read (all are read if not set)" << endl ;
    exit(1) ;
  }
  for(int i=1 ; i < argc ; i++){
      FILEN.push_back( argv[i] )  ;
  }
  int nFiles = argc-1 ;
  
  LCReader* lcReader = LCFactory::getInstance()->createLCReader() ;
  
    // ------ check if LCIO_READ_COL_NAMES is set -------------
    
    char* rColChar = getenv ("LCIO_READ_COL_NAMES");

    if ( rColChar != 0 ) {
      
    std::vector< std::string > colSubset ;
      std::stringstream sts( rColChar ) ;
      std::string colName;
      
      while( sts >> colName) {
    
    colSubset.push_back( colName ) ;
      }
      
      lcReader->setReadCollectionNames(  colSubset ) ;
    }
    //-----------------------------------------------------------



  // first we read the run information
  
  cout << "anajob:  will open and read from files: " << endl ;  

  for(int i=0 ; i < nFiles ; i++){

    lcReader->open( FILEN[i] ) ;
    
    cout  << endl <<  "     "  << FILEN[i] 
      <<  "     [ number of runs: "    <<  lcReader->getNumberOfRuns() 
      <<       ", number of events: "  <<  lcReader->getNumberOfEvents() << " ] "   
      << endl 
      << endl ; 
    
    lcReader->close() ;
  }  
  
  
  // open list of files
  lcReader->open( FILEN ) ;

  LCRunHeader *runHdr ;
  
  // use a try catch block here: if sth. went wrong with reading the run data we 
  // still can try and read the event data - see below
  try{  

    // loop over all run headers
    while( ( runHdr = lcReader->readNextRunHeader() ) != 0 ){
      
      LCTOOLS::dumpRunHeader( runHdr ) ;
//       cout << "  Run : " << runHdr->getRunNumber() 
//     << " - "      << runHdr->getDetectorName() 
//     << ":  "      << runHdr->getDescription()  << endl ;
    }
    
  }catch(IOException& e){
    cout << " io error when reading run data : " << e.what() << endl ;
  }
  cout << endl ;
  
  lcReader->close() ;
  
  
  // now loop over the file again and dump event data

  lcReader->open( FILEN ) ;

  //  cout << " reopened " << FILEN << " for reading " << endl ; 
  cout << " will reopen and read from files: " << endl ;  
  for(int i=0 ; i < nFiles ; i++){
    cout  << "     "  << FILEN[i] << endl ; 
  }  
  

  LCEvent* evt ;
  int nEvents = 0 ;
  
  //----------- the event loop -----------
  while( (evt = lcReader->readNextEvent()) != 0 ) {
    
    LCTOOLS::dumpEvent( evt ) ;
    nEvents ++ ;
  } 
  // -------- end of event loop -----------
  
  cout << endl <<  "  " <<  nEvents << " events read from files: " << endl  ;
  for(int i=0 ; i < nFiles ; i++){
    cout  << "     "  << FILEN[i] << endl ; 
  }  

  
  lcReader->close() ;
  delete lcReader ;
  return 0 ;
}