TopologicalSelector.cpp

Go to the documentation of this file.

#include "caf_util/TopologicalSelector.hpp"
#include "cafe/Stat.hpp"
#include "cafe/Config.hpp"

#include <sstream>

using namespace std ;
using namespace cafe ;
 
namespace caf_util {

  TopologicalSelector::TopologicalSelector(const char *name) : cafe::Processor(name),
                                                               _jetvars("fX","fY"),
                                                               _emvars("fX","fY"),
                                                               _muvars("fX","fY")
  {
    // get parameters from the configuration file
    cafe::Config config(name);

    _electronBranch = config.get("electronBranch", "EMscone") ;
    _muonBranch = config.get("muonBranch", "Muon") ;
    _jetBranch = config.get("jetBranch", "JCCB") ;

    _nelectrons = config.get("nelectrons", -1) ;
    _nmuons = config.get("nmuons", -1) ;
    _njets = config.get("njets", -1) ;

    _minHT = config.get("minHT", 0) ;
    _maxHT = config.get("maxHT", 0) ;

    _minHTleadL = config.get("minHTleadL", 0) ;
    _maxHTleadL = config.get("maxHTleadL", 0) ;
    
    _minHTleptons = config.get("minHTleptons", 0) ;
    _maxHTleptons = config.get("maxHTleptons", 0) ;

    _calcHT = config.get("CalculateHT", true);
    _calcHTLeadL = config.get("CalculateHTleadL", false);
    _calcHTLeptons = config.get("CalculateHTleptons", false);

    _htVarName = config.get("HTVarName", "HT");
    _htLeadLVarName = config.get("HTVarLeadLName", "HTleadL");
    _htLeptonsVarName = config.get("HTVarLeptonsName", "HTleptons");

  }
  
  bool TopologicalSelector::processEvent(cafe::Event &event)
  {
    //get pointer to statistics collector
    StatPointer stat ;
    event.get("StatPointer", stat) ;

    //---------  HT cut -----------------------

    if (_njets != 0 && (_calcHT || _minHT >0 || _maxHT > 0) ) {
      
      Collection<TMBJet> from(event.getCollection<TMBJet>(_jetBranch.c_str(), _jetvars));
      if (from.size() == 0) {
        out() << "TOPOLOGICAL SELECTOR WARNING! Jet branch " 
              << _jetBranch << " is empty or is not existing! "
              << endl ;
        return false ;
      }
      
      float HT = 0 ;      
      for(Collection<TMBJet>::const_iterator it = from.begin() ; 
          it != from.end(); ++it) 
        if (_njets < 0 || it - from.begin() < _njets) HT += it->Pt() ;
      
      if(_htVarName!="") event.put(_htVarName, HT);

      if (_minHT > 0) {
        if (HT < _minHT) return false ;
        
        ostringstream st ;
        st << "HT >= " << _minHT << " GeV" ;
        stat.EventSelected(st.str()) ;     
      }

      if (_maxHT > 0) {
        if (HT > _maxHT) return false ;
        
        ostringstream st ;
        st << "HT <= " << _maxHT << " GeV" ;
        stat.EventSelected(st.str()) ;     
      }
    }


    //---------  HT with leptons cut -----------------------

    if (_calcHTLeptons || _minHTleptons >0 || _maxHTleptons > 0) {

      float HT = 0 ;      
      if (_njets != 0) {
        Collection<TMBJet> from(event.getCollection<TMBJet>(_jetBranch.c_str(), _emvars));
        if (from.size() == 0) {
          out() << "TOPOLOGICAL SELECTOR WARNING! Jet branch " 
                << _jetBranch << " is empty or is not existing! "
                << endl ;
          return false ;
        }
        
        for(Collection<TMBJet>::const_iterator it = from.begin() ; 
            it != from.end(); ++it) 
          if (_njets < 0 || it - from.begin() < _njets) HT += it->Pt() ;
      }
      
      if (_nmuons != 0) {
        Collection<TMBMuon> from(event.getCollection<TMBMuon>(_muonBranch.c_str(), _muvars));
        if (from.size() == 0) {
          out() << "TOPOLOGICAL SELECTOR WARNING! Muon branch " 
                << _muonBranch << " is empty or is not existing! "
                << endl ;
          return false ;
        }
      
        for(Collection<TMBMuon>::const_iterator it = from.begin() ; 
            it != from.end(); ++it) 
          if (_nmuons < 0 || it - from.begin() <= _nmuons) HT += it->Pt() ;     
      } 


      if (_nelectrons != 0) {
        Collection<TMBEMCluster> from(event.getCollection<TMBEMCluster>(_electronBranch.c_str(),_emvars));
        if (from.size() == 0) {
          out() << "TOPOLOGICAL SELECTOR WARNING! Electron branch " 
                << _electronBranch << " is empty or is not existing! "
                << endl ;
          return false ;
        }
      
        for(Collection<TMBEMCluster>::const_iterator it = from.begin() ; 
            it != from.end(); ++it) 
          if (_nelectrons < 0 || it - from.begin() <= _nelectrons) HT += it->Pt() ;     
      } 
      
      if(_htLeptonsVarName!="") event.put(_htLeptonsVarName, HT);

      if (_minHTleptons > 0) {
        if (HT < _minHTleptons) return false ;
        
        ostringstream st ;
        st << "HT (with leptons) >= " << _minHTleptons << " GeV" ;
        stat.EventSelected(st.str()) ;     
      }

      if (_maxHTleptons > 0) {
        if (HT > _maxHTleptons) return false ;
        
        ostringstream st ;
        st << "HT (with leptons) <= " << _maxHTleptons << " GeV" ;
        stat.EventSelected(st.str()) ;     
      }
    }


    //---------  HT with leading lepton cut -----------------------

    if (_calcHTLeadL || _minHTleadL >0 || _maxHTleadL > 0) {

      float HT = 0 ;      
      if (_njets != 0) {
        Collection<TMBJet> from(event.getCollection<TMBJet>(_jetBranch.c_str(),_jetvars));
        if (from.size() == 0) {
          out() << "TOPOLOGICAL SELECTOR WARNING! Jet branch " 
                << _jetBranch << " is empty or is not existing! "
                << endl ;
          return false ;
        }
        
        for(Collection<TMBJet>::const_iterator it = from.begin() ; 
            it != from.end(); ++it) 
          if (_njets < 0 || it - from.begin() < _njets) HT += it->Pt() ;
      }

      float leadPT = 0 ;
      if (_nmuons != 0) {
        Collection<TMBMuon> from(event.getCollection<TMBMuon>(_muonBranch.c_str(),_muvars));
        if (from.size() == 0) {
          out() << "TOPOLOGICAL SELECTOR WARNING! Muon branch " 
                << _muonBranch << " is empty or is not existing! "
                << endl ;
          return false ;
        }
      
        for(Collection<TMBMuon>::const_iterator it = from.begin() ; 
            it != from.end(); ++it) 
          if (it->Pt() > leadPT ) leadPT = it->Pt() ;
      } 


      if (_nelectrons != 0) {
        Collection<TMBEMCluster> from(event.getCollection<TMBEMCluster>(_electronBranch.c_str(),_emvars));
        if (from.size() == 0) {
          out() << "TOPOLOGICAL SELECTOR WARNING! Electron branch " 
                << _electronBranch << " is empty or is not existing! "
                << endl ;
          return false ;
        }
      
        for(Collection<TMBEMCluster>::const_iterator it = from.begin() ; 
            it != from.end(); ++it) 
          if (it->Pt() > leadPT ) leadPT = it->Pt() ;      
      } 
      
      HT += leadPT ;

      if(_htLeadLVarName!="") event.put(_htLeadLVarName, HT);

      if (_minHTleadL > 0) {
        if (HT < _minHTleadL) return false ;
        
        ostringstream st ;
        st << "HT (with leading lepton) >= " << _minHTleadL << " GeV" ;
        stat.EventSelected(st.str()) ;     
      }

      if (_maxHTleadL > 0) {
        if (HT > _maxHTleadL) return false ;
        
        ostringstream st ;
        st << "HT (with leading lepton) <= " << _maxHTleadL << " GeV" ;
        stat.EventSelected(st.str()) ;     
      }
    }


  return true  ;        

  }
}
ClassImp(caf_util::TopologicalSelector) ;
  

---