TauSelector.cpp

Go to the documentation of this file.

/* File TauSelector.cpp
 *
 * Created       : Wed Oct 12 11:08:04 CDT 2005
 * Author        : Mark OWEN, markowen@fnal.gov
 * Purpose       : Select Basic Tau Objects
 * Last modified : 10-24-05
 * Comments      : 
 */

#include "caf_util/TauSelector.hpp"
#include "cafe/Config.hpp"

#include <string>
#include <sstream>

using namespace std;
using namespace cafe;

namespace caf_util {
  
  // Constructor: 
  TauSelector::TauSelector(const char *name) : cafe::SelectUserObjects<TMBTau>(name)
  {
    // get parameters from the configuration file
    cafe::Config config(name);
    
    //minimal number of taus to be selected per event
    _ntaus = config.get("nTaus",1);

    //maximal number of taus to be selected per event
    _ntausmax = config.get("nTausMax",-1);
    
    //Remove charge 0 taus
    _remove_q0 = config.get("RemoveQ0",0);

    //Tau detector eta cut
    _tau_eta_cut = config.get("EtaCut",-1.0);

    cout << "Tau Eta Cut for " << name << " = " << _tau_eta_cut << endl; 

    //Tau Et cuts
    _taut1_ET_cut = config.get("Type1Et",10.0);
    _taut2_ET_cut = config.get("Type2Et",5.0);
    _taut3_ET_cut = config.get("Type3Et",10.0);
    
    cout << "Tau Type 1 Et cut for " << name << " = " << _taut1_ET_cut << endl;
    cout << "Tau Type 2 Et cut for " << name << " = " << _taut2_ET_cut << endl;
    cout << "Tau Type 3 Et cut for " << name << " = " << _taut3_ET_cut << endl;

    //Tau Track pT cuts
    _taut1_trkpt_cut = config.get("Type1TrkPt",7.0);
    _taut2_trkpt_cut = config.get("Type2TrkPt",5.0);
    _taut3_trkpt_cut = config.get("Type3TrkPt",7.0);

    cout << "Tau Type 1 Track Pt cut for " << name << " = " << _taut1_trkpt_cut << endl;
    cout << "Tau Type 2 Track Pt cut for " << name << " = " << _taut2_trkpt_cut << endl;
    cout << "Tau Type 3 Track Pt cut for " << name << " = " << _taut3_trkpt_cut << endl;

    //Tau NN Output cuts
    _taut1_nn_cut = config.get("Type1NNCut",-1.0);
    _taut2_nn_cut = config.get("Type2NNCut",-1.0);
    _taut3_nn_cut = config.get("Type3NNCut",-1.0);

    if(_taut1_nn_cut>0) cout << "Tau type 1 NN cut for " << name << " = " << _taut1_nn_cut << endl;
    if(_taut2_nn_cut>0) cout << "Tau type 2 NN cut for " << name << " = " << _taut2_nn_cut << endl;
    if(_taut3_nn_cut>0) cout << "Tau type 3 NN cut for " << name << " = " << _taut3_nn_cut << endl;
    
    _nselected = 0;
    
  }
  
  //destructor
  TauSelector::~TauSelector()
  {
    
  }

  bool TauSelector::processEvent(cafe::Event& event)
  {
    //get pointer to statistics collector
    event.get("StatPointer", _stat) ;
   
    //Open tracks branch in case autoloading does not work
    Collection<TMBTrack> tracks = event.getTracks();
    
    _nselected = 0 ;
    SelectUserObjects<TMBTau>::processEvent(event);
    
    if(_ntaus > 0 && _nselected < 1) return false;
    
    for(int i = 2; i <= _ntaus; i++) {
      if(_nselected < i) return false;
      ostringstream st;
      st << i << " taus";
      _stat.EventSelected(st.str());
    }
    
    if (_ntausmax >= 0) {
      if (_nselected > _ntausmax) return false ; 
      ostringstream st ;
      st << "Number of taus <= " << _ntausmax ;
      _stat.EventSelected(st.str()) ;      
    }

    return true;
  }
  
  bool TauSelector::selectObject(const TMBTau &tau)
  {
    int tau_type = tau.type();

    //Reject tau type 0s
    if(tau_type<1) return false;
    {
      ostringstream st;
      st << "Remove Tau Type 0";
      if(_ntaus>0) _stat.EventSelected(st.str());
    }

    //Remove charge 0 taus
    if(_remove_q0) {
      if(tau.charge()==0) return false;
      ostringstream st;
      st << "Remove Taus with charge = 0";
      if(_ntaus>0) _stat.EventSelected(st.str());
    }

    //Tau Eta selection
    if(_tau_eta_cut>0) {
      if(fabs(tau.etad()) >= _tau_eta_cut) return false;
      ostringstream st;
      st << "Tau eta < " << _tau_eta_cut ;
      if(_ntaus>0) _stat.EventSelected(st.str());
    }

    //Et selection
    if(tau_type==1 && tau.ET()<_taut1_ET_cut) return false; 
    if(tau_type==2 && tau.ET()<_taut2_ET_cut) return false;
    if(tau_type==3 && tau.ET()<_taut3_ET_cut) return false;
    
    {
      ostringstream st;
      st << "Tau ET cut (type 1 > " << _taut1_ET_cut << ", type 2 > " << _taut2_ET_cut << ", type 3 > " << _taut3_ET_cut << ")";
      if(_ntaus>0) _stat.EventSelected(st.str());
    }      

    //Track Pt selection
    if(tau_type==1 || tau_type==2) {
      if(tau.ntrk()>1) {
        cerr << "ERROR: Tau type 1 or 2 has more than 1 track, tau will be rejected" << endl;
        return false;
      }
      const TMBTrack *track = tau.GetChargedTrack(0);
      if(track==0) {
        cerr << "ERROR: No track associated with tau, tau will be rejected" << endl;
        return false;
      }
      if(tau_type==1 && track->Pt()<_taut1_trkpt_cut) return false;
      if(tau_type==2 && track->Pt()<_taut2_trkpt_cut) return false;
    } //end if type 1 or 2
    
    if(tau_type==3) {
      float sum_pt=0;
      for(int itrk=0; itrk<tau.ntrk(); itrk++) {
        const TMBTrack *track = tau.GetChargedTrack(itrk);
        if(track==0) {
        cerr << "ERROR: No track associated with tau, tau will be rejected" << endl;
        return false;
        }
        sum_pt+=track->Pt();
      }//end loop over tau tracks       
      if(sum_pt<_taut3_trkpt_cut) return false;
    } //end if type 3
      
    {
      ostringstream st;
      st << "Tau Track Pt cut (type 1 > " << _taut1_trkpt_cut << ", type 2 > " << _taut2_trkpt_cut << ", type 3 > " << _taut3_trkpt_cut << ")";
      if(_ntaus>0) _stat.EventSelected(st.str());
    }
  
    //Tau NN output cut
    if(tau_type==1&&_taut1_nn_cut>0) if(tau.nnout()<_taut1_nn_cut) return false;
    if(tau_type==2&&_taut2_nn_cut>0) if(tau.nnout()<_taut2_nn_cut) return false;
    if(tau_type==3&&_taut3_nn_cut>0) if(tau.nnout()<_taut3_nn_cut) return false;

    if( (_taut1_nn_cut>0) || (_taut2_nn_cut>0) || (_taut3_nn_cut>0) ) {
      ostringstream st;
      st << "Tau NN (type 1 > " << _taut1_nn_cut << ", type 2 > " << _taut2_nn_cut << ", type 3 > " << _taut3_nn_cut << ")";
      if(_ntaus>0) _stat.EventSelected(st.str());
    }

      
    _nselected++;
    
    return true;
    
  }
   
  
}



ClassImp(caf_util::TauSelector);

---