Updating E/gamma supercluster/ecal candidate filtering tools to allow future jetcore seeding by e/gamma #21818
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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,174 @@ | ||
| #ifndef RECOEGAMMA_EGAMMAHLTPRODUCERS_EGAMMAHLTFILTEREDOBJPRODUCER_H | ||
| #define RECOEGAMMA_EGAMMAHLTPRODUCERS_EGAMMAHLTFILTEREDOBJPRODUCER_H | ||
|
|
||
|
|
||
| #include "FWCore/Framework/interface/Frameworkfwd.h" | ||
| #include "FWCore/Framework/interface/stream/EDProducer.h" | ||
| #include "FWCore/Framework/interface/Event.h" | ||
| #include "FWCore/Framework/interface/MakerMacros.h" | ||
| #include "FWCore/Framework/interface/ESHandle.h" | ||
| #include "FWCore/Framework/interface/ConsumesCollector.h" | ||
| #include "FWCore/ParameterSet/interface/ParameterSet.h" | ||
| #include "FWCore/ParameterSet/interface/ParameterSetDescription.h" | ||
| #include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h" | ||
|
|
||
| #include "DataFormats/RecoCandidate/interface/RecoEcalCandidate.h" | ||
| #include "DataFormats/RecoCandidate/interface/RecoEcalCandidateFwd.h" | ||
| #include "DataFormats/RecoCandidate/interface/RecoEcalCandidateIsolation.h" | ||
|
|
||
| #include "HLTrigger/HLTcore/interface/defaultModuleLabel.h" | ||
|
|
||
| namespace { | ||
| template<typename T> edm::Handle<T> getHandle(const edm::Event& event,const edm::EDGetTokenT<T>& token) | ||
| { | ||
| edm::Handle<T> handle; | ||
| event.getByToken(token,handle); | ||
| return handle; | ||
| } | ||
|
|
||
| } | ||
|
|
||
| template <typename OutCollType> | ||
| class EgammaHLTFilteredObjProducer : public edm::stream::EDProducer<>{ | ||
| public: | ||
| class SelectionCut { | ||
| private: | ||
| struct CutValues { | ||
| float cut; | ||
| float cutOverE; | ||
| float cutOverE2; | ||
| bool useEt; | ||
| std::function<bool(float,float)> compFunc; | ||
|
|
||
| CutValues(const edm::ParameterSet& pset): | ||
| cut(pset.getParameter<double>("cut")), | ||
| cutOverE(pset.getParameter<double>("cutOverE")), | ||
| cutOverE2(pset.getParameter<double>("cutOverE2")), | ||
| useEt(pset.getParameter<bool>("useEt")), | ||
| compFunc(std::less<float>()) {} | ||
|
|
||
| bool operator()(const reco::RecoEcalCandidate& cand,float value)const{ | ||
| float energy = useEt ? cand.et() : cand.energy(); | ||
| return compFunc(value,cut) || compFunc(value/energy,cutOverE) || | ||
| compFunc(value/energy/energy,cutOverE2); | ||
|
There was a problem hiding this comment. it would be more effective to test |
||
| } | ||
| }; | ||
| public: | ||
| SelectionCut(const edm::ParameterSet& pset,edm::ConsumesCollector && iC): | ||
| ebCut_(pset.getParameter<edm::ParameterSet>("barrelCut")), | ||
| eeCut_(pset.getParameter<edm::ParameterSet>("endcapCut")), | ||
| varToken_(iC.consumes<reco::RecoEcalCandidateIsolationMap>(pset.getParameter<edm::InputTag>("var"))) | ||
| {} | ||
|
|
||
| ~SelectionCut()=default; | ||
|
|
||
| bool operator()(const reco::RecoEcalCandidateRef& cand)const{ | ||
| CutValues cut = std::abs(cand->eta())<1.479 ? ebCut_ : eeCut_; | ||
| return cut(*cand,getVar(cand)); | ||
| } | ||
|
|
||
| float getVar(const reco::RecoEcalCandidateRef& cand)const{ | ||
| auto res = varHandle_->find(cand); | ||
| if(res!=varHandle_->end()) return res->val; | ||
| else{ | ||
| //FIX ME: add some provenance info to this | ||
| throw cms::Exception("LogicError") <<" candidate not found in collection "; | ||
| } | ||
| } | ||
|
|
||
| void getHandles(const edm::Event& event){ | ||
| event.getByToken(varToken_,varHandle_); | ||
| } | ||
| private: | ||
| CutValues ebCut_; | ||
| CutValues eeCut_; | ||
| edm::EDGetTokenT<reco::RecoEcalCandidateIsolationMap> varToken_; | ||
| edm::Handle<reco::RecoEcalCandidateIsolationMap> varHandle_; | ||
| }; | ||
|
|
||
| explicit EgammaHLTFilteredObjProducer(const edm::ParameterSet& pset); | ||
| ~EgammaHLTFilteredObjProducer() override =default; | ||
| static void fillDescriptions(edm::ConfigurationDescriptions & descriptions); | ||
|
|
||
| void produce(edm::Event&, const edm::EventSetup&) override; | ||
|
|
||
| private: | ||
| //different collection types may need to specialise this function | ||
| //eg if you wanted superclusters it would output.push_back(cand->superCluster()) | ||
| static void addObj(const reco::RecoEcalCandidateRef& cand,OutCollType& output){ | ||
| output.push_back(cand); | ||
| } | ||
| private: | ||
|
There was a problem hiding this comment. this There was a problem hiding this comment. personal style, I use a new label out of habit every time I have a new group (aka member functions, member classes, data members). But fine, personal style and dont really feel strongly so changed it. |
||
| edm::EDGetTokenT<reco::RecoEcalCandidateCollection> candsToken_; | ||
| std::vector<SelectionCut> cuts_; | ||
| }; | ||
|
|
||
|
|
||
| template<typename OutCollType> | ||
| EgammaHLTFilteredObjProducer<OutCollType>:: | ||
| EgammaHLTFilteredObjProducer(const edm::ParameterSet& pset): | ||
| candsToken_(consumes<reco::RecoEcalCandidateCollection>(pset.getParameter<edm::InputTag>("cands"))) | ||
| { | ||
| const auto& cutPsets = pset.getParameter<std::vector<edm::ParameterSet> >("cuts"); | ||
| for(auto& cutPset : cutPsets){ | ||
| cuts_.push_back(SelectionCut(cutPset,consumesCollector())); | ||
| } | ||
|
|
||
| produces<OutCollType>(); | ||
|
|
||
| } | ||
|
|
||
| template<typename OutCollType> | ||
| void EgammaHLTFilteredObjProducer<OutCollType>:: | ||
| fillDescriptions(edm::ConfigurationDescriptions & descriptions) | ||
| { | ||
| edm::ParameterSetDescription desc; | ||
| desc.add<edm::InputTag>("cands",edm::InputTag("hltEgammaCandidates")); | ||
|
|
||
| edm::ParameterSetDescription cutsDesc; | ||
| edm::ParameterSetDescription regionCutsDesc; | ||
| regionCutsDesc.add<double>("cut",-1); | ||
| regionCutsDesc.add<double>("cutOverE",-1); | ||
| regionCutsDesc.add<double>("cutOverE2",-1); | ||
| regionCutsDesc.add<bool>("useEt",false); | ||
| edm::ParameterSet cutDefaults; | ||
| cutDefaults.addParameter<double>("cutOverE",0.2); | ||
| cutDefaults.addParameter<double>("useEt",false); | ||
|
|
||
| cutsDesc.add<edm::ParameterSetDescription>("barrelCut",regionCutsDesc); | ||
| cutsDesc.add<edm::ParameterSetDescription>("endcapCut",regionCutsDesc); | ||
| cutsDesc.add<edm::InputTag>("var",edm::InputTag("hltEgammaHoverE")); | ||
|
|
||
| edm::ParameterSet defaults; | ||
| defaults.addParameter<edm::InputTag>("var",edm::InputTag("hltEgammaHoverE")); | ||
| defaults.addParameter<edm::ParameterSet>("barrelCut",cutDefaults); | ||
| defaults.addParameter<edm::ParameterSet>("endcapCut",cutDefaults); | ||
| desc.addVPSet("cuts",cutsDesc,std::vector<edm::ParameterSet>{defaults}); | ||
| descriptions.add(defaultModuleLabel<EgammaHLTFilteredObjProducer<OutCollType>>(),desc); | ||
| } | ||
|
|
||
| template<typename OutCollType> | ||
| void EgammaHLTFilteredObjProducer<OutCollType>:: | ||
| produce(edm::Event& iEvent, const edm::EventSetup& iSetup) | ||
| { | ||
| for(auto& cut : cuts_) cut.getHandles(iEvent); | ||
| auto candsHandle = getHandle(iEvent,candsToken_); | ||
|
|
||
| auto output = std::make_unique<OutCollType>(); | ||
|
|
||
| for(size_t candNr=0;candNr<candsHandle->size();candNr++){ | ||
| reco::RecoEcalCandidateRef candRef(candsHandle,candNr); | ||
| bool passAllCuts=true; | ||
| for(const auto& cut: cuts_){ | ||
| if(!cut(candRef)){ | ||
| passAllCuts=false; | ||
| break; | ||
| } | ||
| } | ||
| if(passAllCuts) addObj(candRef,*output); | ||
| } | ||
|
|
||
| iEvent.put(std::move(output)); | ||
| } | ||
|
|
||
| #endif | ||
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,26 @@ | ||
|
|
||
| #include "RecoEgamma/EgammaHLTProducers/interface/EgammaHLTFilteredObjProducer.h" | ||
| #include "DataFormats/Common/interface/Ptr.h" | ||
| #include "DataFormats/Common/interface/RefToPtr.h" | ||
|
|
||
| template<> | ||
| void EgammaHLTFilteredObjProducer<reco::RecoEcalCandidateCollection>:: | ||
| addObj(const reco::RecoEcalCandidateRef& cand, reco::RecoEcalCandidateCollection& output) | ||
| { | ||
| output.push_back(*cand); | ||
| } | ||
|
|
||
| using EgammaHLTFilteredEcalCandProducer=EgammaHLTFilteredObjProducer<reco::RecoEcalCandidateCollection>; | ||
| DEFINE_FWK_MODULE(EgammaHLTFilteredEcalCandProducer); | ||
|
|
||
|
|
||
| template<> | ||
| void EgammaHLTFilteredObjProducer<std::vector<edm::Ptr<reco::Candidate> > >:: | ||
| addObj(const reco::RecoEcalCandidateRef& cand, std::vector<edm::Ptr<reco::Candidate> >& output) | ||
| { | ||
| output.push_back(edm::refToPtr(cand)); | ||
| } | ||
|
|
||
| using EgammaHLTFilteredEcalCandPtrProducer=EgammaHLTFilteredObjProducer<std::vector<edm::Ptr<reco::Candidate> > >; | ||
| DEFINE_FWK_MODULE(EgammaHLTFilteredEcalCandPtrProducer); | ||
|
|
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 |
|---|---|---|
| @@ -1,168 +1,15 @@ | ||
|
|
||
| #include "RecoEgamma/EgammaHLTProducers/interface/EgammaHLTFilteredObjProducer.h" | ||
| #include "DataFormats/EgammaReco/interface/SuperCluster.h" | ||
| #include "DataFormats/EgammaReco/interface/SuperClusterFwd.h" | ||
|
|
||
| template<> | ||
| void EgammaHLTFilteredObjProducer<std::vector<reco::SuperClusterRef>>:: | ||
| addObj(const reco::RecoEcalCandidateRef& cand, std::vector<reco::SuperClusterRef>& output) | ||
| { | ||
| output.push_back(cand->superCluster()); | ||
| } | ||
|
|
||
| using EgammaHLTFilteredSuperClusterProducer=EgammaHLTFilteredObjProducer<std::vector<reco::SuperClusterRef>>; | ||
| DEFINE_FWK_MODULE(EgammaHLTFilteredSuperClusterProducer); | ||
|
|
||
Oops, something went wrong.
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.
public should show up first
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.
done