BTagJetSelector.cpp

Go to the documentation of this file.

/* File BTagJetSelector.hpp
 *
 * Created       : Wed Feb 1 14:36:54 CST 2006
 * Author        : Aran GARCIA-BELLIDO, aran@fnal.gov
 * Purpose       : Select events that contain certain number of b-tagged jets
 * Last modified : 
 * Comments      :
 */

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

using namespace std;
using namespace cafe;
 
namespace caf_util {
  
  // Constructor, destructor: 
  BTagJetSelector::BTagJetSelector(const char *name) : cafe::SelectUserObjects<TMBJet>(name),
                                                       _nselected(0),_vars("_nchunks")
  {
    // Here you can read parameters from the config file:
    cafe::Config config(name);
    
    _nbjets = config.get("nBJets",1);
    _nbjetsmax = config.get("nBJetsMax",-1);
    _btagAlgo = config.get("BTagAlgo","EMPTY");
    _btagPoint = config.get("Cut","EMPTY");

    // Report
    cout << "\nBTagJetSelector instantiated"
         << "\n  Tagger: " << _btagAlgo
         << "\n  Cut level: " << _btagPoint
         << "\n  # b jets >= " << _nbjets << endl;
    if (_nbjetsmax!=0) cout << "  # b jets <= " << _nbjetsmax << endl;
    
  }
  
  BTagJetSelector::~BTagJetSelector()
  {
    
  }
  
  bool BTagJetSelector::processEvent(cafe::Event& event)
  {
    _isMC=false;
    if (event.getMCEventInfo(_vars)!=NULL) _isMC=true;
    // if (_isMC==true) return false;
    
    _nselected = 0;
    
    // Get pointer to statistics collector
    event.get("StatPointer", _stat) ;
    
    // Do the actual selection:
    SelectUserObjects<TMBJet>::processEvent(event) ;
    
    // Accounting:
    for (int i = 1; i <= _nbjets; i++) {
      if (_nselected < i) return false; 
      ostringstream st;
      st << "Number of b-jets >= " << i;
      _stat.EventSelected(st.str());
    }
    
    if (_nbjetsmax >= 0) {
      if (_nselected > _nbjetsmax) return false ; 
      ostringstream st;
      st << "Number of b-jets <= " << _nbjetsmax ;
      _stat.EventSelected(st.str());      
    }
    
    return true;
  }

  bool BTagJetSelector::selectObject(const TMBJet &jet)
  { 
    if (jet.taggable() == -1) {
      cout << "Taggability not run: BAD or EM jet" << endl;
      return false;
    }
    // Check taggability: 
    if (jet.taggable() == 0) return false;
    // Check if it is taggable and tagged:
    if (jet.taggable() == 1) {
      // ok, this is taggable, mark it in the Stat:
      if (_nbjets > 0) _stat.EventSelected("Taggable Jet");  
      // Instantiate the TMBBTag information for this jet:
      const TMBBTag* btaginfo = jet.GetBTag(_btagAlgo, _btagPoint); // e.g. ("NN", "L4")
      if (! btaginfo) {throw std::runtime_error("BTagJetSelector: Can't find required b-tag");}
      if (!btaginfo->is_tagged()) return false;
      ostringstream st;
      st << _btagAlgo << " " << _btagPoint <<  " Tagged Jet";
      if (_nbjets > 0) _stat.EventSelected(st.str());
      _nselected++;
    } 
    return true;
  };

  void BTagJetSelector::before(Collection<TMBJet>& from)
  {
    // empty
  };

}

ClassImp(caf_util::BTagJetSelector);

---