-
Notifications
You must be signed in to change notification settings - Fork 4.3k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #42762 from vlimant/13_0_backport_42646
Add mu beamspot constrained pT. Increase mu eta and phi precision to 16 bits [13.0.X]
- Loading branch information
Showing
4 changed files
with
156 additions
and
1 deletion.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
140 changes: 140 additions & 0 deletions
140
RecoMuon/GlobalTrackingTools/plugins/MuonBeamspotConstraintValueMapProducer.cc
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,140 @@ | ||
| /** \class MuonBeamspotConstraintValueMapProducer | ||
| * Compute muon pt and ptErr after beamspot constraint. | ||
| * | ||
| */ | ||
| #include "FWCore/Framework/interface/Frameworkfwd.h" | ||
| #include "FWCore/Framework/interface/MakerMacros.h" | ||
| #include "FWCore/Framework/interface/Event.h" | ||
| #include "DataFormats/PatCandidates/interface/Muon.h" | ||
| #include "DataFormats/Common/interface/ValueMap.h" | ||
| #include "FWCore/Framework/interface/global/EDProducer.h" | ||
| #include "FWCore/ParameterSet/interface/ParameterSet.h" | ||
| #include "TrackingTools/TransientTrack/interface/TransientTrackBuilder.h" | ||
| #include "TrackingTools/Records/interface/TransientTrackRecord.h" | ||
| #include "TrackingTools/TransientTrack/interface/TransientTrack.h" | ||
| #include "RecoVertex/KalmanVertexFit/interface/SingleTrackVertexConstraint.h" | ||
| #include "DataFormats/VertexReco/interface/Vertex.h" | ||
|
|
||
| class MuonBeamspotConstraintValueMapProducer : public edm::global::EDProducer<> { | ||
| public: | ||
| explicit MuonBeamspotConstraintValueMapProducer(const edm::ParameterSet& config) | ||
| : muonToken_(consumes<pat::MuonCollection>(config.getParameter<edm::InputTag>("src"))), | ||
| beamSpotToken_(consumes<reco::BeamSpot>(config.getParameter<edm::InputTag>("beamspot"))), | ||
| PrimaryVertexToken_(consumes<reco::VertexCollection>(config.getParameter<edm::InputTag>("vertices"))), | ||
| ttbToken_(esConsumes(edm::ESInputTag("", "TransientTrackBuilder"))) { | ||
| produces<edm::ValueMap<float>>("muonBSConstrainedPt"); | ||
| produces<edm::ValueMap<float>>("muonBSConstrainedPtErr"); | ||
| produces<edm::ValueMap<float>>("muonBSConstrainedChi2"); | ||
| } | ||
|
|
||
| ~MuonBeamspotConstraintValueMapProducer() override = default; | ||
|
|
||
| static void fillDescriptions(edm::ConfigurationDescriptions& descriptions) { | ||
| edm::ParameterSetDescription desc; | ||
| desc.add<edm::InputTag>("src", edm::InputTag("muons"))->setComment("Muon collection"); | ||
| desc.add<edm::InputTag>("beamspot", edm::InputTag("offlineBeamSpot"))->setComment("Beam spot collection"); | ||
| desc.add<edm::InputTag>("vertices", edm::InputTag("offlineSlimmedPrimaryVertices")) | ||
| ->setComment("Primary vertex collection"); | ||
|
|
||
| descriptions.addWithDefaultLabel(desc); | ||
| } | ||
|
|
||
| private: | ||
| void produce(edm::StreamID streamID, edm::Event& event, const edm::EventSetup& setup) const override { | ||
| edm::Handle<pat::MuonCollection> muons; | ||
| event.getByToken(muonToken_, muons); | ||
|
|
||
| edm::Handle<reco::BeamSpot> beamSpotHandle; | ||
| event.getByToken(beamSpotToken_, beamSpotHandle); | ||
|
|
||
| edm::ESHandle<TransientTrackBuilder> ttkb = setup.getHandle(ttbToken_); | ||
|
|
||
| std::vector<float> pts, ptErrs, chi2s; | ||
| pts.reserve(muons->size()); | ||
| ptErrs.reserve(muons->size()); | ||
| chi2s.reserve(muons->size()); | ||
|
|
||
| for (const auto& muon : *muons) { | ||
| bool tbd = true; | ||
| if (beamSpotHandle.isValid()) { | ||
| double BeamWidthX = beamSpotHandle->BeamWidthX(); | ||
| double BeamWidthXError = beamSpotHandle->BeamWidthXError(); | ||
| double BeamWidthY = beamSpotHandle->BeamWidthY(); | ||
| double BeamWidthYError = beamSpotHandle->BeamWidthYError(); | ||
| // Protect for mis-reconstructed beamspots (note that | ||
| // SingleTrackVertexConstraint uses the width for the constraint, | ||
| // not the error) | ||
| if ((BeamWidthXError / BeamWidthX < 0.3) && (BeamWidthYError / BeamWidthY < 0.3)) { | ||
| SingleTrackVertexConstraint::BTFtuple btft = | ||
| stvc.constrain(ttkb->build(muon.muonBestTrack()), *beamSpotHandle); | ||
| if (std::get<0>(btft)) { | ||
| const reco::Track& trkBS = std::get<1>(btft).track(); | ||
| pts.push_back(trkBS.pt()); | ||
| ptErrs.push_back(trkBS.ptError()); | ||
| chi2s.push_back(std::get<2>(btft)); | ||
| tbd = false; | ||
| } | ||
| } | ||
| } | ||
|
|
||
| if (tbd) { | ||
| // Invalid BS; use PV instead | ||
| edm::Handle<reco::VertexCollection> pvHandle; | ||
| event.getByToken(PrimaryVertexToken_, pvHandle); | ||
|
|
||
| if (pvHandle.isValid() && !pvHandle->empty()) { | ||
| auto pv = pvHandle->at(0); | ||
| VertexState pvs = VertexState(GlobalPoint(Basic3DVector<float>(pv.position())), GlobalError(pv.covariance())); | ||
|
|
||
| SingleTrackVertexConstraint::BTFtuple btft = stvc.constrain(ttkb->build(muon.muonBestTrack()), pvs); | ||
| if (std::get<0>(btft)) { | ||
| const reco::Track& trkBS = std::get<1>(btft).track(); | ||
| pts.push_back(trkBS.pt()); | ||
| ptErrs.push_back(trkBS.ptError()); | ||
| chi2s.push_back(std::get<2>(btft)); | ||
| tbd = false; | ||
| } | ||
| } | ||
| } | ||
|
|
||
| if (tbd) { | ||
| // Fall-back case, keep the unconstrained values | ||
| pts.push_back(muon.pt()); | ||
| ptErrs.push_back(muon.bestTrack()->ptError()); | ||
| chi2s.push_back(-1.f); | ||
| } | ||
| } | ||
|
|
||
| { | ||
| std::unique_ptr<edm::ValueMap<float>> valueMap(new edm::ValueMap<float>()); | ||
| edm::ValueMap<float>::Filler filler(*valueMap); | ||
| filler.insert(muons, pts.begin(), pts.end()); | ||
| filler.fill(); | ||
| event.put(std::move(valueMap), "muonBSConstrainedPt"); | ||
| } | ||
|
|
||
| { | ||
| std::unique_ptr<edm::ValueMap<float>> valueMap(new edm::ValueMap<float>()); | ||
| edm::ValueMap<float>::Filler filler(*valueMap); | ||
| filler.insert(muons, ptErrs.begin(), ptErrs.end()); | ||
| filler.fill(); | ||
| event.put(std::move(valueMap), "muonBSConstrainedPtErr"); | ||
| } | ||
|
|
||
| { | ||
| std::unique_ptr<edm::ValueMap<float>> valueMap(new edm::ValueMap<float>()); | ||
| edm::ValueMap<float>::Filler filler(*valueMap); | ||
| filler.insert(muons, chi2s.begin(), chi2s.end()); | ||
| filler.fill(); | ||
| event.put(std::move(valueMap), "muonBSConstrainedChi2"); | ||
| } | ||
| } | ||
|
|
||
| edm::EDGetTokenT<pat::MuonCollection> muonToken_; | ||
| edm::EDGetTokenT<reco::BeamSpot> beamSpotToken_; | ||
| edm::EDGetTokenT<reco::VertexCollection> PrimaryVertexToken_; | ||
| edm::ESGetToken<TransientTrackBuilder, TransientTrackRecord> ttbToken_; | ||
| SingleTrackVertexConstraint stvc; | ||
| }; | ||
|
|
||
| DEFINE_FWK_MODULE(MuonBeamspotConstraintValueMapProducer); |