/Users/jmonk/Physics/ForIA/src/EventLoop.cxx

Go to the documentation of this file.

#include "ForIA/EventLoop.hh"

#include "ForIA/DataConverter.hh"

#include <boost/numeric/conversion/cast.hpp>

#include <iostream>

namespace ForIA{
 
  EventLoop::EventLoop(DataConverter *converter, IHistogrammer *histogrammer): 
  m_converter(converter), m_histogrammer(histogrammer), m_weightSum(0.0){
    
  }
  
  bool EventLoop::addAnalysis(const string &name){
    
    m_analyses.push_back(m_factory.createAnalysis(name, m_histogrammer));
    
    return true;
  }
  
  void EventLoop::printAnalyses(std::ostream &out)const{
    
    vector<string> analyses = m_factory.listAnalyses();
    out<<std::endl<<"List of available analyses:"<<std::endl;
    
    for(vector<string>::const_iterator name = analyses.begin();
        name != analyses.end(); ++name){
      out<<*name<<std::endl;
    }
    
    return;
  }
  
  bool EventLoop::initialise(){
    
    bool success = true;
    
    for(vector<AnalysisPtr>::iterator analysis = m_analyses.begin();
        analysis != m_analyses.end(); ++analysis){
      success = success && (*analysis)->initialise();
    }
        
    return success;
  }
  
  bool EventLoop::finalise(){
    bool success = true;
    
    for(vector<AnalysisPtr>::iterator analysis = m_analyses.begin();
        analysis != m_analyses.end(); ++analysis){
      success = success && (*analysis)->finalise();
    }
    
    m_histogrammer->bookHistogram1D("/EventLoop", "sumOfEventWeights", "Sum of event weights in loop", 1, 0.5, 1.5);
    m_histogrammer->fillHistogram1D("/EventLoop", "sumOfEventWeights", 1.0, m_weightSum);
    
    return success;
  }
  
  bool EventLoop::run(long nEvents){
   
    long counter=0;
    long counterThresh=10;
    long counterThreshMax=10000;
    unsigned int smallCounter = 1;
    unsigned int incr=1;
    
    unsigned long vetoCounter=0;
    unsigned long vetoMax=50;
    
    string th = "th";
    
    bool success = true;
    
    while(m_converter->nextEvent() && counter != nEvents){
      
      Event ev = m_converter->loadEvent();
      
      if(smallCounter==incr){
        try{
          std::cout<<"Processing "<<counter<<th<<" event. "<<ev<<std::endl;
        }catch(boost::numeric::negative_overflow negEr){
          std::cout<<"EventLoop::run : Looks like the event didn't contain enough info."<<std::endl;
          std::cout<<"EventLoop::run : Check that the event run/lb/number is present and read in for this data format"<<std::endl;
          throw;
        }
        
        if(counter == 0){
          th="st";
        }else if(counter == 1){
          th="nd";
        }else if(counter == 2){
          th="rd";
        }else if(counter ==3){
          th="th";
        }
        smallCounter=0;
      }
      
      if(counter == counterThresh){
        incr *= 10;
        if(counterThresh != counterThreshMax){
          counterThresh *=10;
        }
      }

      ++smallCounter;
      ++counter;
      
      if(!ev.goodLumiBlock()){
        if(vetoCounter < vetoMax){
          std::cout<<"vetoing event on bad lumi block"<<std::endl;
        }else if(vetoCounter == vetoMax){
          std::cout<<"reached max number of bad lumi block messages"<<std::endl;
        }
        ++vetoCounter;
        continue;
      }
      m_weightSum += ev.weight();
      
      bool storeEvent = false;
      for(vector<AnalysisPtr>::iterator analysis=m_analyses.begin();
          analysis != m_analyses.end(); ++analysis){
        (*analysis)->executeClear(ev);
        if((*analysis)->storesEvent())storeEvent = true;
      }
      
      if(storeEvent) m_converter->storeEvent();
      
    }
    
    if(counter == nEvents){
      std::cout<<"Event limit reached - finished"<<std::endl;
    }
    
    return success;
  }
  
}