pat::PackedCandidateMuonSelectorProducer Class Reference

Inheritance diagram for pat::PackedCandidateMuonSelectorProducer:

Public Member Functions
	PackedCandidateMuonSelectorProducer (const edm::ParameterSet &iConfig)
void	produce (edm::Event &, const edm::EventSetup &) override
	~PackedCandidateMuonSelectorProducer () override=default
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
	EDProducer (const EDProducer &)=delete
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginProcessBlocks () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndProcessBlocks () const final
bool	hasAbilityToProduceInEndRuns () const final
const EDProducer &	operator= (const EDProducer &)=delete

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &)

Private Attributes
const edm::EDGetTokenT < edm::Association < reco::PFCandidateCollection > >	candidate2PFToken_
const edm::EDGetTokenT < pat::PackedCandidateCollection >	candidateToken_
const edm::EDGetTokenT < pat::PackedCandidateCollection >	lostTrackToken_
std::map< std::string, std::unique_ptr < StringCutObjectSelector < reco::Muon > > >	muonIDMap_
const std::vector< std::string >	muonIDs_
const std::vector< std::string >	muonSelectors_
const edm::EDGetTokenT < reco::MuonCollection >	muonToken_
const edm::EDGetTokenT < edm::Association < pat::PackedCandidateCollection > >	track2LostTrackToken_

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
using	CacheTypes = CacheContexts< T...>
using	GlobalCache = typename CacheTypes::GlobalCache
using	HasAbility = AbilityChecker< T...>
using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache
using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache
using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >
using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache
using	RunCache = typename CacheTypes::RunCache
using	RunContext = RunContextT< RunCache, GlobalCache >
using	RunSummaryCache = typename CacheTypes::RunSummaryCache

Detailed Description

Definition at line 15 of file PackedCandidateMuonSelectorProducer.cc.

Constructor & Destructor Documentation

pat::PackedCandidateMuonSelectorProducer::PackedCandidateMuonSelectorProducer ( const edm::ParameterSet &  iConfig )

inlineexplicit

Definition at line 17 of file PackedCandidateMuonSelectorProducer.cc.

References muonIDMap_, muonIDs_, muonSelectors_, and EgammaValidation_Wenu_cff::sel.

        : muonToken_(consumes<reco::MuonCollection>(iConfig.getParameter<edm::InputTag>("muons"))),
          candidateToken_(consumes<pat::PackedCandidateCollection>(iConfig.getParameter<edm::InputTag>("candidates"))),
          candidate2PFToken_(consumes<edm::Association<reco::PFCandidateCollection>>(
              iConfig.getParameter<edm::InputTag>("candidates"))),
          track2LostTrackToken_(consumes<edm::Association<pat::PackedCandidateCollection>>(
              iConfig.getParameter<edm::InputTag>("lostTracks"))),
          muonSelectors_(iConfig.getParameter<std::vector<std::string>>("muonSelectors")),
          muonIDs_(iConfig.getParameter<std::vector<std::string>>("muonIDs")) {
      for (const auto& sel : muonSelectors_) {
        produces<pat::PackedCandidateRefVector>("lostTracks" + sel);
        produces<pat::PackedCandidateRefVector>("pfCandidates" + sel);
      }
      for (const auto& sel : muonIDs_) {
        muonIDMap_[sel] = std::make_unique<StringCutObjectSelector<reco::Muon>>("passed('" + sel + "')");
        produces<pat::PackedCandidateRefVector>("lostTracks" + sel);
        produces<pat::PackedCandidateRefVector>("pfCandidates" + sel);
      }
    }

pat::PackedCandidateMuonSelectorProducer::~PackedCandidateMuonSelectorProducer ( )

overridedefault

Member Function Documentation

void pat::PackedCandidateMuonSelectorProducer::fillDescriptions ( edm::ConfigurationDescriptions &  descriptions )

static

Definition at line 120 of file PackedCandidateMuonSelectorProducer.cc.

References edm::ConfigurationDescriptions::add(), edm::ParameterSetDescription::add(), submitPVResolutionJobs::desc, and HLT_FULL_cff::InputTag.

                                                                                                        {
  edm::ParameterSetDescription desc;
  desc.add<edm::InputTag>("muons", edm::InputTag("muons"))->setComment("muon input collection");
  desc.add<edm::InputTag>("candidates", edm::InputTag("packedPFCandidates"))
      ->setComment("packed PF candidate input collection");
  desc.add<edm::InputTag>("lostTracks", edm::InputTag("lostTracks"))->setComment("lost track input collection");
  desc.add<std::vector<std::string>>("muonSelectors", {"AllTrackerMuons", "TMOneStationTight"})
      ->setComment("muon selectors");
  desc.add<std::vector<std::string>>("muonIDs", {})->setComment("muon IDs");
  descriptions.add("packedCandidateMuonID", desc);
}

void pat::PackedCandidateMuonSelectorProducer::produce ( edm::Event &  iEvent, const edm::EventSetup &  iSetup  )

override

Definition at line 55 of file PackedCandidateMuonSelectorProducer.cc.

References candidate2PFToken_, HLT_FULL_cff::candidates, candidateToken_, edm::Event::get(), edm::Event::getHandle(), mps_fire::i, muon::isGoodMuon(), eostools::move(), HLT_FULL_cff::muon, muonIDMap_, muonIDs_, patZpeak::muons, muonSelectors_, muonToken_, edm::Event::put(), reco::TrackBase::qualityByName(), alignCSCRings::s, EgammaValidation_Wenu_cff::sel, muon::selectionTypeFromString(), and track2LostTrackToken_.

                                                                                                  {
  const auto& muons = iEvent.get(muonToken_);
  const auto& candidates = iEvent.getHandle(candidateToken_);
  const auto& candidate2PF = iEvent.get(candidate2PFToken_);
  const auto& track2LostTrack = iEvent.get(track2LostTrackToken_);

  std::map<std::string, std::unique_ptr<pat::PackedCandidateRefVector>> lostTrackMap, candMap;
  for (const auto& sel : muonSelectors_) {
    lostTrackMap.emplace(sel, new pat::PackedCandidateRefVector());
    candMap.emplace(sel, new pat::PackedCandidateRefVector());
  }
  for (const auto& sel : muonIDs_) {
    lostTrackMap.emplace(sel, new pat::PackedCandidateRefVector());
    candMap.emplace(sel, new pat::PackedCandidateRefVector());
  }

  // loop over muons
  for (const auto& muon : muons) {
    const auto& muonTrack = muon.innerTrack();
    // ignore muons without high purity inner track
    if (muonTrack.isNull() || !muonTrack->quality(reco::TrackBase::qualityByName("highPurity")))
      continue;

    // find lost track associated to muon
    const auto& lostTrack = track2LostTrack[muonTrack];
    if (lostTrack.isNonnull()) {
      for (const auto& sel : muonSelectors_) {
        if (muon::isGoodMuon(muon, muon::selectionTypeFromString(sel)))
          lostTrackMap[sel]->push_back(lostTrack);
      }
      for (const auto& sel : muonIDs_) {
        if ((*muonIDMap_.at(sel))(muon))
          lostTrackMap[sel]->push_back(lostTrack);
      }
      continue;
    }

    // find PF candidate associated to muon
    for (size_t i = 0; i < candidates->size(); i++) {
      const auto& cand = pat::PackedCandidateRef(candidates, i);
      const auto& candTrack = candidate2PF[cand]->trackRef();
      // check if candidate and muon are compatible
      if (candTrack.isNonnull() && muonTrack == candTrack) {
        for (const auto& sel : muonSelectors_) {
          if (muon::isGoodMuon(muon, muon::selectionTypeFromString(sel)))
            candMap[sel]->push_back(cand);
        }
        for (const auto& sel : muonIDs_) {
          if ((*muonIDMap_.at(sel))(muon))
            candMap[sel]->push_back(cand);
        }
        break;
      }
    }
  }

  for (auto& s : lostTrackMap) {
    iEvent.put(std::move(s.second), "lostTracks" + s.first);
  }
  for (auto& s : candMap) {
    iEvent.put(std::move(s.second), "pfCandidates" + s.first);
  }
}

Member Data Documentation

const edm::EDGetTokenT<edm::Association<reco::PFCandidateCollection> > pat::PackedCandidateMuonSelectorProducer::candidate2PFToken_

private

Definition at line 46 of file PackedCandidateMuonSelectorProducer.cc.

Referenced by produce().

const edm::EDGetTokenT<pat::PackedCandidateCollection> pat::PackedCandidateMuonSelectorProducer::candidateToken_

private

Definition at line 44 of file PackedCandidateMuonSelectorProducer.cc.

Referenced by produce().

const edm::EDGetTokenT<pat::PackedCandidateCollection> pat::PackedCandidateMuonSelectorProducer::lostTrackToken_

private

Definition at line 45 of file PackedCandidateMuonSelectorProducer.cc.

std::map<std::string, std::unique_ptr<StringCutObjectSelector<reco::Muon> > > pat::PackedCandidateMuonSelectorProducer::muonIDMap_

private

Definition at line 50 of file PackedCandidateMuonSelectorProducer.cc.

Referenced by PackedCandidateMuonSelectorProducer(), and produce().

const std::vector<std::string> pat::PackedCandidateMuonSelectorProducer::muonIDs_

private

Definition at line 49 of file PackedCandidateMuonSelectorProducer.cc.

Referenced by PackedCandidateMuonSelectorProducer(), and produce().

const std::vector<std::string> pat::PackedCandidateMuonSelectorProducer::muonSelectors_

private

Definition at line 48 of file PackedCandidateMuonSelectorProducer.cc.

Referenced by PackedCandidateMuonSelectorProducer(), and produce().

const edm::EDGetTokenT<reco::MuonCollection> pat::PackedCandidateMuonSelectorProducer::muonToken_

private

Definition at line 43 of file PackedCandidateMuonSelectorProducer.cc.

Referenced by produce().

const edm::EDGetTokenT<edm::Association<pat::PackedCandidateCollection> > pat::PackedCandidateMuonSelectorProducer::track2LostTrackToken_

private

Definition at line 47 of file PackedCandidateMuonSelectorProducer.cc.

Referenced by produce().