PATMuonProducer.h

Go to the documentation of this file.

//
// $Id: PATMuonProducer.h,v 1.28.6.1 2012/09/04 17:31:56 vadler Exp $
//

#ifndef PhysicsTools_PatAlgos_PATMuonProducer_h
#define PhysicsTools_PatAlgos_PATMuonProducer_h

#include <string>

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EDProducer.h"
#include "DataFormats/PatCandidates/interface/Muon.h"
#include "CommonTools/Utils/interface/PtComparator.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "TrackingTools/IPTools/interface/IPTools.h"
#include "PhysicsTools/PatAlgos/interface/MultiIsolator.h"
#include "PhysicsTools/PatAlgos/interface/EfficiencyLoader.h"
#include "PhysicsTools/PatAlgos/interface/KinResolutionsLoader.h"

#include "DataFormats/PatCandidates/interface/UserData.h"
#include "PhysicsTools/PatAlgos/interface/PATUserDataHelper.h"
#include "TrackingTools/TransientTrack/interface/TransientTrackBuilder.h"


namespace pat {
  class TrackerIsolationPt;
  class CaloIsolationEnergy;

  class PATMuonProducer : public edm::EDProducer {
    
  public:
    explicit PATMuonProducer(const edm::ParameterSet & iConfig);
    ~PATMuonProducer();
    virtual void produce(edm::Event & iEvent, const edm::EventSetup & iSetup);
    static void fillDescriptions(edm::ConfigurationDescriptions & descriptions);

  private:
    typedef edm::RefToBase<reco::Muon> MuonBaseRef;
    typedef std::vector<edm::Handle<edm::Association<reco::GenParticleCollection> > > GenAssociations;
    typedef std::vector< edm::Handle< edm::ValueMap<IsoDeposit> > > IsoDepositMaps;
    typedef std::vector< edm::Handle< edm::ValueMap<double> > > IsolationValueMaps;
    typedef std::pair<pat::IsolationKeys,edm::InputTag> IsolationLabel;
    typedef std::vector<IsolationLabel> IsolationLabels;


    void fillMuon( Muon& aMuon, const MuonBaseRef& muonRef, const reco::CandidateBaseRef& baseRef, const GenAssociations& genMatches, const IsoDepositMaps& deposits, const IsolationValueMaps& isolationValues) const;
    void readIsolationLabels( const edm::ParameterSet & iConfig, const char* psetName, IsolationLabels& labels); 


    // embed various impact parameters with errors
    // embed high level selection
    void embedHighLevel( pat::Muon & aMuon,
             reco::TrackRef track,
             reco::TransientTrack & tt,
             reco::Vertex & primaryVertex,
             bool primaryVertexIsValid,
             reco::BeamSpot & beamspot,
             bool beamspotIsValid );

    
  private:
    edm::InputTag muonSrc_;

    bool embedBestTrack_;
    bool embedImprovedBestTrack_;
    bool embedTrack_;
    bool embedStandAloneMuon_;
    bool embedCombinedMuon_;
    bool embedCaloMETMuonCorrs_;
    edm::InputTag caloMETMuonCorrs_;
    bool embedTcMETMuonCorrs_;
    edm::InputTag tcMETMuonCorrs_;
    bool embedPickyMuon_;
    bool embedTpfmsMuon_;
    bool embedDytMuon_;
    bool addGenMatch_;
    std::vector<edm::InputTag> genMatchSrc_;
    bool embedGenMatch_;
    bool addResolutions_;
    pat::helper::KinResolutionsLoader resolutionLoader_;    
    bool useParticleFlow_;    
    edm::InputTag pfMuonSrc_;
    bool embedPFCandidate_;
    bool embedHighLevelSelection_;
    edm::InputTag beamLineSrc_;
    bool usePV_;
    edm::InputTag pvSrc_;
    IsolationLabels isoDepositLabels_;
    IsolationLabels isolationValueLabels_;
    bool addEfficiencies_;    
    bool useUserData_;

    GreaterByPt<Muon> pTComparator_;
    pat::helper::MultiIsolator isolator_; 
    pat::helper::MultiIsolator::IsolationValuePairs isolatorTmpStorage_;
    pat::helper::EfficiencyLoader efficiencyLoader_;
    pat::PATUserDataHelper<pat::Muon> userDataHelper_;    
  };

}

#endif