METSelector.cpp

Go to the documentation of this file.

#include "caf_util/METSelector.hpp"
#include "cafe/Stat.hpp"
#include "tmb_tree/TMBMet.hpp"
#include "met_util/BMetQualInfo.hpp"
#include "cafe/Config.hpp"

#include <sstream>

using namespace std ;
using namespace cafe ;
 
namespace caf_util {

  METSelector::METSelector(const char *name) : cafe::Processor(name),
                                               _vars("_metqualinfos"),
                                               _mcvars("_nchunks")
  {
    // get parameters from the configuration file
    cafe::Config config(name);

    _fromBranch = config.get("From", "Met");
    out() << "METSelector \"" << name << "\" Read MET From: " << _fromBranch << endl ;

    _minMET = config.get("minMET", 0) ;
    _maxMET = config.get("maxMET", 0) ;

    _minSET = config.get("minSET", 0) ;
    _maxSET = config.get("maxSET", 0) ;
    
    string jetAlgoType = config.get("jetAlgoType","JCCB") ;
    bool jesMuCorrected = config.get("jesMuCorrected", false) ;
    _muCorrected = config.get("MuCorrected", false) ; 
    _MCsmeared = config.get("MCsmeared", true) ; 

    if (jetAlgoType != "JCCA" && jetAlgoType != "JCCB") {
      err() << "METSelector ERROR: .jetAlgoType should be JCCA or JCCB, you value is \"" 
            << jetAlgoType << "\". WILL USE JCCB!" << endl ;
      jetAlgoType = "JCCB" ;
    }      

    _algo = jetAlgoType ;
    if (!jesMuCorrected)  _algo = "corr" + _algo ;
    else _algo = "corrmu" + _algo ;

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

    // For MC : MC smearing "a la d0correct" on EM/jet/muons is propagated to MET
    // There is no additional MET oversmearing    if ( isMC )
    // No smearing for data
    string algo = _algo ;
    if (_MCsmeared && event.getMCEventInfo(_mcvars)->nchunks() > 0) algo = "smear_" + _algo ;

    const metid::BMetQualInfo* metqual =  event.get<TMBMet>(_fromBranch, _vars)->getMetQualInfo(algo) ;

    if (!metqual) {
      err() << "METSelector: pointer to the metid::BMetQualInfo object is 0!" 
            << " Algo name \"" << algo 
            << endl ;
      return false ;   
    }

    float met = -10.0;       
    float metx = -10.0;       
    float mety = -10.0;       
    float set = -10.0 ;

    if (!_muCorrected) {
      // MET corrected for CH+EM+JES
      met = metqual->getMETcorrCALO().getmet();
      metx = metqual->getMETcorrCALO().getmex();
      mety = metqual->getMETcorrCALO().getmey();
      set = metqual->getMETcorrCALO().getset();
    } else {
      // MET corrected for CH+EM+JES+muon
      met = metqual->getMETcorrCALOMU().getmet();
      metx = metqual->getMETcorrCALOMU().getmex();
      mety = metqual->getMETcorrCALOMU().getmey();
      set = metqual->getMETcorrCALOMU().getset();
    }

    event.put("caf_util::MET", met) ;
    event.put("caf_util::SET", set) ;
    event.put("caf_util::METx", metx) ;
    event.put("caf_util::METy", mety) ;
    
    if (debug() > 1) 
      out() << "METSelector: " 
            << " MET = " << met 
            << " METx = " << metx 
            << " METy = " << mety 
            << " SET = " << set 
            << " GeV for algo name \"" << algo << "\""      
            << endl ;

    if (_minMET > 0) {
      if (met < _minMET) return false ;
      ostringstream st ;
      st << "MET >= " << _minMET << " GeV" ;
      stat.EventSelected(st.str()) ;
    }
    
    if (_maxMET > 0) {
      if (met > _maxMET) return false ;
      ostringstream st ;
      st << "MET <= " << _maxMET << " GeV" ;
      stat.EventSelected(st.str()) ;
    }

    if (_minSET > 0) {
      if (set < _minSET) return false ;
      ostringstream st ;
      st << "SET >= " << _minSET << " GeV" ;
      stat.EventSelected(st.str()) ;
    }
    
    if (_maxSET > 0) {
      if (set > _maxSET) return false ;
      ostringstream st ;
      st << "SET <= " << _maxSET << " GeV" ;
      stat.EventSelected(st.str()) ;
    }

    return true  ;      
  };
}
ClassImp(caf_util::METSelector) ;
  

---