Add isolated charged hadron flag to RECO+AOD #18324
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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,101 @@ | ||
| /* | ||
| * ChargedHadronPFTrackIsolationProducer | ||
| * | ||
| * Author: Andreas Hinzmann | ||
| * | ||
| * Associates PF isolation flag to charged hadron candidates | ||
| * | ||
| */ | ||
|
|
||
| #include "FWCore/Framework/interface/stream/EDProducer.h" | ||
| #include "FWCore/Framework/interface/EventSetup.h" | ||
| #include "FWCore/Framework/interface/Event.h" | ||
| #include "FWCore/ParameterSet/interface/ParameterSet.h" | ||
| #include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h" | ||
| #include "FWCore/ParameterSet/interface/ParameterSetDescription.h" | ||
|
|
||
| #include "FWCore/Utilities/interface/Exception.h" | ||
| #include "FWCore/MessageLogger/interface/MessageLogger.h" | ||
|
|
||
| #include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h" | ||
| #include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h" | ||
| #include "DataFormats/Common/interface/ValueMap.h" | ||
| #include "DataFormats/ParticleFlowReco/interface/PFBlock.h" | ||
| #include "DataFormats/ParticleFlowReco/interface/PFBlockElement.h" | ||
|
|
||
| namespace edm { | ||
| class ConfigurationDescriptions; | ||
| } | ||
|
|
||
| class ChargedHadronPFTrackIsolationProducer : public edm::stream::EDProducer<> | ||
|
There was a problem hiding this comment. this module can be made There was a problem hiding this comment. made edm::global and changed the produce method to "void produce(edm::StreamID, edm::Event& evt, const edm::EventSetup& es) const" as defined in global/EDProducerBase.h. |
||
| { | ||
| public: | ||
|
|
||
| explicit ChargedHadronPFTrackIsolationProducer(const edm::ParameterSet& cfg); | ||
| ~ChargedHadronPFTrackIsolationProducer() {} | ||
| void produce(edm::Event& evt, const edm::EventSetup& es); | ||
|
There was a problem hiding this comment. please add |
||
| static void fillDescriptions(edm::ConfigurationDescriptions & descriptions); | ||
|
|
||
| private: | ||
| // input collection | ||
| edm::InputTag srcCandidates_; | ||
| edm::EDGetTokenT<edm::View<reco::PFCandidate> > Candidates_token; | ||
|
There was a problem hiding this comment. please keep the same style of data members: start with lower case, end with an underscore |
||
| double minTrackPt_; | ||
| double minRawCaloEnergy_; | ||
|
|
||
| }; | ||
|
|
||
| ChargedHadronPFTrackIsolationProducer::ChargedHadronPFTrackIsolationProducer(const edm::ParameterSet& cfg) | ||
| { | ||
| srcCandidates_ = cfg.getParameter<edm::InputTag>("src"); | ||
| Candidates_token = consumes<edm::View<reco::PFCandidate> >(srcCandidates_); | ||
| minTrackPt_ = cfg.getParameter<double>("minTrackPt"); | ||
| minRawCaloEnergy_ = cfg.getParameter<double>("minRawCaloEnergy"); | ||
|
|
||
| produces<edm::ValueMap<bool> >(); | ||
| } | ||
|
|
||
| void ChargedHadronPFTrackIsolationProducer::produce(edm::Event& evt, const edm::EventSetup& es) | ||
| { | ||
| // get a view of our Candidates via the base candidates | ||
| typedef edm::View<reco::PFCandidate> PFCandidateView; | ||
| edm::Handle<PFCandidateView> Candidates; | ||
| evt.getByToken(Candidates_token, Candidates); | ||
|
There was a problem hiding this comment. start local variables and data members with lower case letter. Capitalization is reserved for class names. |
||
|
|
||
| std::vector<bool> values; | ||
|
|
||
| for( PFCandidateView::const_iterator c = Candidates->begin(); c!=Candidates->end(); ++c) { | ||
|
There was a problem hiding this comment. for range loop seems more appropriate |
||
| // Check that there is only one track in the block. | ||
| unsigned int nTracks = 0; | ||
| if ((c->particleId()==1) && (c->pt()>minTrackPt_) && ((c->rawEcalEnergy()+c->rawHcalEnergy())>minRawCaloEnergy_)) { | ||
| const reco::PFCandidate::ElementsInBlocks& theElements = c->elementsInBlocks(); | ||
| if( theElements.empty() ) continue; | ||
| const reco::PFBlockRef blockRef = theElements[0].first; | ||
| const edm::OwnVector<reco::PFBlockElement>& elements = blockRef->elements(); | ||
| for(unsigned int iEle=0; iEle<elements.size(); iEle++) { // Find the tracks in the block | ||
|
There was a problem hiding this comment. similarly for (auto const& ele : elements){
... |
||
| reco::PFBlockElement::Type type = elements[iEle].type(); | ||
| if( type== reco::PFBlockElement::TRACK) | ||
| nTracks++; | ||
| } | ||
| } | ||
| values.push_back((nTracks==1)); | ||
| } | ||
|
|
||
| std::unique_ptr<edm::ValueMap<bool> > out(new edm::ValueMap<bool>()); | ||
| edm::ValueMap<bool>::Filler filler(*out); | ||
| filler.insert(Candidates,values.begin(),values.end()); | ||
| filler.fill(); | ||
| evt.put(std::move(out)); | ||
| } | ||
|
|
||
| void ChargedHadronPFTrackIsolationProducer::fillDescriptions(edm::ConfigurationDescriptions & descriptions) { | ||
| edm::ParameterSetDescription desc; | ||
| desc.add<edm::InputTag>("src", edm::InputTag("particleFlow")); | ||
| desc.add<double>("minTrackPt", 1); | ||
| desc.add<double>("minRawCaloEnergy", 0.5); | ||
| descriptions.add("chargedHadronPFTrackIsolation", desc); | ||
| } | ||
|
|
||
| #include "FWCore/Framework/interface/MakerMacros.h" | ||
|
|
||
| DEFINE_FWK_MODULE(ChargedHadronPFTrackIsolationProducer); | ||
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,9 @@ | ||
| import FWCore.ParameterSet.Config as cms | ||
|
|
||
|
|
||
| chargedHadronPFTrackIsolation = cms.EDProducer( | ||
|
There was a problem hiding this comment. this file is redundant and can be removed. |
||
| "ChargedHadronPFTrackIsolationProducer", | ||
| src = cms.InputTag("particleFlow"), | ||
| minTrackPt = cms.double(1), | ||
| minRawCaloEnergy = cms.double(0.5), | ||
| ) | ||
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
this forward declaration can be removed (there is an include for this class earlier in this file)