SoftLepton.h

Go to the documentation of this file.

#ifndef RecoBTag_SoftLepton_SoftLepton_h
#define RecoBTag_SoftLepton_SoftLepton_h

// -*- C++ -*-
//
// Package:    SoftLepton
// Class:      SoftLepton
// 
//
// Original Author:  fwyzard
//         Created:  Wed Oct 18 18:02:07 CEST 2006
// $Id: SoftLepton.h,v 1.14 2010/08/19 23:14:52 fwyzard Exp $
//

// system include files
#include <memory>
#include <map>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDProducer.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/Common/interface/RefToBase.h"
#include "DataFormats/Math/interface/Vector3D.h"
#include "DataFormats/GeometryVector/interface/GlobalVector.h"
#include "DataFormats/JetReco/interface/Jet.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/MuonReco/interface/Muon.h"
#include "DataFormats/MuonReco/interface/MuonSelectors.h"
#include "DataFormats/BTauReco/interface/SoftLeptonTagInfo.h"

class TransientTrackBuilder;

class SoftLepton : public edm::EDProducer {
public:
  explicit SoftLepton(const edm::ParameterSet& iConfig);
  ~SoftLepton();

  struct TrackCompare :
    public std::binary_function<edm::RefToBase<reco::Track>,
                                edm::RefToBase<reco::Track>, bool> {
    inline bool operator () (const edm::RefToBase<reco::Track> &t1,
                             const edm::RefToBase<reco::Track> &t2) const
    { return t1.key() < t2.key();}
  };

  typedef std::map<unsigned int, float> LeptonIds;
  typedef std::map<edm::RefToBase<reco::Track>, LeptonIds, TrackCompare> Leptons;

  // generic interface, using a TrackRefVector for lepton tracks
  reco::SoftLeptonTagInfo tag (
      const edm::RefToBase<reco::Jet> & jet,
      const reco::TrackRefVector      & tracks,
      const Leptons                   & leptons,
      const reco::Vertex              & primaryVertex
  ) const;

protected:
  // generic interface, using a TrackRefVector for lepton tracks

  GlobalVector refineJetAxis (
      const edm::RefToBase<reco::Jet>   & jet,
      const reco::TrackRefVector        & tracks,
      const edm::RefToBase<reco::Track> & exclude = edm::RefToBase<reco::Track>()
  ) const;

  static double relativeEta(
      const math::XYZVector& vector,
      const math::XYZVector& axis
  );

  static double boostedPPar(
      const math::XYZVector& vector,
      const math::XYZVector& axis
  );
  
private:
  virtual void produce(edm::Event & event, const edm::EventSetup & setup);

  // configuration   
  const edm::InputTag m_jets;
  const edm::InputTag m_primaryVertex;
  const edm::InputTag m_leptons;
  const edm::InputTag m_leptonCands;
  const edm::InputTag m_leptonId;

  // service used to make transient tracks from tracks
  const TransientTrackBuilder * m_transientTrackBuilder;

  // algorithm configuration
  unsigned int  m_refineJetAxis;
  double        m_deltaRCut;
  double        m_chi2Cut;
  
  // specific for reco::Muons
  muon::SelectionType m_muonSelection;

  // nominal beam spot position
  static const reco::Vertex s_nominalBeamSpot;
};

#endif // RecoBTag_SoftLepton_SoftLepton_h