MuonProducer.h

Go to the documentation of this file.

#ifndef RecoMuon_MuonIdentification_MuonProducer_H
#define RecoMuon_MuonIdentification_MuonProducer_H
 
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Utilities/interface/InputTag.h"
 
#include "DataFormats/MuonReco/interface/MuonFwd.h"
#include "DataFormats/MuonReco/interface/MuonTimeExtra.h"
 
namespace reco {
  class Track;
}
#include "FWCore/Framework/interface/ESHandle.h"
#include "Geometry/CommonDetUnit/interface/GlobalTrackingGeometry.h"
 
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
 
#include "DataFormats/MuonReco/interface/Muon.h"
#include "DataFormats/MuonReco/interface/MuonTrackLinks.h"
#include "DataFormats/MuonReco/interface/MuonTimeExtraMap.h"
#include "DataFormats/MuonReco/interface/MuonShower.h"
#include "DataFormats/MuonReco/interface/MuonCosmicCompatibility.h"
#include "DataFormats/MuonReco/interface/MuonToMuonMap.h"
 
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
 
#include "DataFormats/Common/interface/ValueMap.h"
#include "DataFormats/RecoCandidate/interface/IsoDepositFwd.h"
#include "DataFormats/RecoCandidate/interface/IsoDeposit.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
 
class MuPFIsoHelper;
 
class MuonProducer : public edm::stream::EDProducer<> {
public:
  MuonProducer(const edm::ParameterSet&);
 
  ~MuonProducer() override;
 
  void produce(edm::Event&, const edm::EventSetup&) override;
 
  typedef std::vector<edm::InputTag> InputTags;
 
protected:
private:
  template <typename TYPE>
  void fillMuonMap(edm::Event& event,
                   const edm::OrphanHandle<reco::MuonCollection>& muonHandle,
                   const std::vector<TYPE>& muonExtra,
                   const std::string& label);
 
  std::string theAlias;
 
  void setAlias(std::string alias) {
    alias.erase(alias.size() - 1, alias.size());
    theAlias = alias;
  }
 
  std::string labelOrInstance(const edm::InputTag&) const;
 
private:
  bool debug_;
  bool fastLabelling_;
 
  edm::InputTag theMuonsCollectionLabel;
  edm::EDGetTokenT<reco::MuonCollection> theMuonsCollectionToken_;
 
  edm::InputTag thePFCandLabel;
  edm::EDGetTokenT<reco::PFCandidateCollection> thePFCandToken_;
 
  bool fillIsolation_;
  bool writeIsoDeposits_;
  bool fillSelectors_;
  bool fillCosmicsIdMap_;
  bool fillPFMomentum_;
  bool fillPFIsolation_;
  bool fillDetectorBasedIsolation_;
  bool fillShoweringInfo_;
  bool fillTimingInfo_;
  bool computeStandardSelectors_;
 
  edm::InputTag theTrackDepositName;
  edm::InputTag theEcalDepositName;
  edm::InputTag theHcalDepositName;
  edm::InputTag theHoDepositName;
  edm::InputTag theJetDepositName;
 
  edm::EDGetTokenT<reco::IsoDepositMap> theTrackDepositToken_;
  edm::EDGetTokenT<reco::IsoDepositMap> theEcalDepositToken_;
  edm::EDGetTokenT<reco::IsoDepositMap> theHcalDepositToken_;
  edm::EDGetTokenT<reco::IsoDepositMap> theHoDepositToken_;
  edm::EDGetTokenT<reco::IsoDepositMap> theJetDepositToken_;
 
  InputTags theSelectorMapNames;
  std::vector<edm::EDGetTokenT<edm::ValueMap<bool> > > theSelectorMapTokens_;
 
  edm::InputTag theShowerMapName;
  edm::EDGetTokenT<edm::ValueMap<reco::MuonShower> > theShowerMapToken_;
 
  edm::InputTag theCosmicCompMapName;
  edm::EDGetTokenT<edm::ValueMap<unsigned int> > theCosmicIdMapToken_;
  edm::EDGetTokenT<edm::ValueMap<reco::MuonCosmicCompatibility> > theCosmicCompMapToken_;
  std::string theMuToMuMapName;
 
  MuPFIsoHelper* thePFIsoHelper;
 
  edm::EDGetTokenT<reco::MuonTimeExtraMap> timeMapCmbToken_;
  edm::EDGetTokenT<reco::MuonTimeExtraMap> timeMapDTToken_;
  edm::EDGetTokenT<reco::MuonTimeExtraMap> timeMapCSCToken_;
 
  std::vector<std::map<std::string, edm::InputTag> > pfIsoMapNames;
  std::vector<std::map<std::string, edm::EDGetTokenT<edm::ValueMap<double> > > > pfIsoMapTokens_;
 
  edm::EDGetTokenT<reco::VertexCollection> vertexes_;
};
#endif