ElectronSelector.cpp

Go to the documentation of this file.

#include "caf_util/ElectronSelector.hpp"
#include "caf_util/TRefFinder.hpp"
#include "cafe/Config.hpp"
#include <sstream>

using namespace cafe ;
using namespace std ;

namespace {
  bool moreThan(const TMBEMCluster& a, const TMBEMCluster& b) {
    return a.Pt() > b.Pt();    
  }
}

namespace caf_util {

  ElectronSelector::ElectronSelector(const char *name) : 
    cafe::SelectUserObjects<TMBEMCluster>(name), _nselected(0), 
    _event(0), _finder(0)
  {
    cafe::Config config(name);
    
    _quality = config.get("Quality", "Tight");
    _ver = config.get("Version", 1);
    _lorun = config.get("FromRun", 0);
    _hirun = config.get("ToRun", 10000000);
    _nelectrons = config.get("nElectrons", 0);
    _nelectronsmax = config.get("nElectronsMax", -1);
    _sort = config.get("Sort", 0);
    _emidfilepath = config.get("SelectionFile","");
    
    // lets look for the input file: emid_cuts.txt
    // First in the user specified path:
    if ( _emidfilepath != "" )
      _emidfilepath = FindFile(_emidfilepath);
    
    // then in the workin area
    if ( _emidfilepath=="" )
      if (gSystem->Getenv("SRT_PRIVATE_CONTEXT") && strcmp(gSystem->Getenv("SRT_PRIVATE_CONTEXT"),".") != 0)
        _emidfilepath = FindFile(string(gSystem->Getenv("SRT_PRIVATE_CONTEXT"))+string("/emid_cuts/support/emid_cuts.txt"));
    
    if ( _emidfilepath=="" )
      _emidfilepath = FindFile("./emid_cuts/support/emid_cuts.txt");
    
    if ( _emidfilepath == "" )
      _emidfilepath = FindFile("./emid_cuts.txt");
    
    if ( _emidfilepath == "") {
      err() << "ElectronSelector(" << name << ")\nError: can not find the file \"emid_cuts.txt\". Please specify the correct path in your config file:\n   "
            << name << ".SelectionFile: path/emid_cuts.txt\n";
      throw std::runtime_error("Can not find emid_cuts file");
    }
    
    out() << "\n======== ElectronSelector("<<name<<") ==================\nUsing ElectronID criteria from \""<<_emidfilepath<<"\"\n";
    _id = new emID(_emidfilepath, _quality, _ver, _lorun, _hirun, 0);
    out() << "===================================================\n\n";

  
  addVariable("_id") ;
  if (_id->cutOnIsolation()) addVariable("_iso") ;
  if (_id->cutOnEMFraction()) addVariable("_EMfrac") ;
  if (_id->cutOnpT() ) {
    addVariable("fX") ;
    addVariable("fY") ;
  }
  bool etavar = false ;
  if ( _id->cutOnEtaCC() || _id->cutOnEtaECmin() || _id->cutOnEtaECmax()) {
    addVariable("_CalDetectorEta") ;
    etavar = true ;
  }

  bool hmx7var = false ;
  if (_id->cutOnHMx7CC()) {
    if (!etavar) {
      addVariable("_CalDetectorEta") ;
      etavar = true ;
    }
    addVariable("_HMx7") ;
    hmx7var = true ;
  }

  if (_id->cutOnHMx8EC()) {
    if (!etavar) {
      addVariable("_CalDetectorEta") ;
      etavar = true ;
    }
    addVariable("_HMx8") ;
  }

  if (_id->cutOnHMx7()) {
    if (!hmx7var) {
      addVariable("_HMx7") ;
      hmx7var = true ;
    }
  }

  if (_id->cutOnTrkMatchChi2EOP())  {
    addVariable("_TrMatchChi2ProbBest") ;
    addVariable("_SpatialTrMatchChi2ProbBest") ;
  } if (_id->cutOnTrkMatchChi2()) addVariable("_SpatialTrMatchChi2ProbBest") ;


  if (_id->cutOnLHood()) {
    addVariable("_Lhood8") ;
    addVariable("_chptr") ;
  }
  if (_id->cutOnTrackPt()) {
    addVariable("_chptr") ;
    if (_id->cutOnTrkMatchChi2EOP() ) addVariable("_chptrBest") ;
    _finder = new TRefFinder() ;
    _finder->SetVariablesTrack(Variables("fUniqueID", "fX", "fY")) ;
  }
  _id->saveCuts() ;
  }

  std::string ElectronSelector::FindFile(std::string path) {
    ifstream emid_file;
    emid_file.open(path.c_str());
    
    if (!emid_file) {
      out() << "ElectronSelector("<<name()<<"): Did not find the file emid_cuts.txt in:\n  \""<< path << "\"" << endl;
      return "";
    }
    
    emid_file.close();
    return path;
  }

bool ElectronSelector::selectObject(const TMBEMCluster &electron)
{
  float trkpt = 1000.0 ;
  if (_finder) {
    const TMBTrack* trk = 0 ;
    //    cout << (electron.GetChargedTrackRef().GetUniqueID() & 0xffffff )
    //   << " " << (electron.getPtrChpSpatialRef().GetUniqueID() & 0xffffff )
    //   << " " << (electron.getPtrChpRef().GetUniqueID() & 0xffffff)  << endl ;
    if (_id->cutOnTrkMatchChi2EOP() && electron.track_match_chi2prob()>0) {
      trk = _finder->FindTrack(*_event, electron.getPtrChpRef()) ;
    } else if (electron.track_match_spatialchi2prob()>0) {
      //      getPtrChpSpatialRef() always 0. Bug?
      //      trk = _finder->FindTrack(*_event, electron.getPtrChpSpatialRef()) ;
      trk = _finder->FindTrack(*_event, electron.GetChargedTrackRef()) ;
    }

    if (!trk) 
      trkpt = -1.0 ;
    else 
      trkpt =trk->Pt() ;
  }



  bool pass = _id->select(1, 
                          electron.id(), 
                          electron.iso(), 
                          electron.emfrac(), 
                          electron.Pt(), 
                          electron.CalDetectorEta(), 
                          electron.HMx7(), 
                          electron.HMx8(), 
                          electron.track_match_spatialchi2prob(), 
                          electron.Lhood8(),
                          trkpt,
                          electron.track_match_chi2prob()
                          ) ;


  const vector<string>* cuts = _id->passCuts() ;
  if (cuts && _nelectrons > 0 ) 
    for (vector<string>::const_iterator it = cuts->begin();
         it != cuts->end(); it++) _stat.EventSelected(*it) ;

  if (!pass) return false ;
  
  _nselected++ ;

  if (debug()>1) 
    cout << "PROCESSOR \"" << name() 
         << "\" SELECTED OBJECT: pT = " << electron.Pt() 
         << " eta = " << electron.Eta() 
         << " phi = " << electron.Phi() 
         << " charge = " << electron.charge() 
         << endl ;
  return true ;
}

void ElectronSelector::finish() {
  if (debug() >=1 ) _id->Stat() ;
}

  bool ElectronSelector::processEvent(cafe::Event &event)
  {

    //get pointer to statistics collector
    event.get("StatPointer", _stat)  ;
    _event = &event ;
    _nselected = 0 ;
    SelectUserObjects<TMBEMCluster>::processEvent(event);

    if (debug() >=2 ) 
      out() << " ELECTRON SELECTED: " << _nselected 
            << std::endl ;
    
    for (int i = 1; i <= _nelectrons; i++) {
      if (_nselected < i) return false ; 
      ostringstream st ;
      st << " N electrons >= "  << i;
      _stat.EventSelected(st.str()) ;
    }

    if (_nelectronsmax >= 0) {
      if (_nselected > _nelectronsmax) return false ; 
      ostringstream st ;
      st << "Number of electrons <= " <<  _nelectronsmax ;
      _stat.EventSelected(st.str()) ;      
    }

    return true  ;      
  }

  void ElectronSelector::before(cafe::Collection<TMBEMCluster>& from) {
      if(_sort) {
          from.sort(from.begin(),from.end(),::moreThan) ;
      }
  }

}

ClassImp(caf_util::ElectronSelector) ;

---