-
Notifications
You must be signed in to change notification settings - Fork 54
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 #8 from perrozzi/74X
first implementation of miniIsolation
- Loading branch information
Showing
4 changed files
with
208 additions
and
2 deletions.
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,150 @@ | ||
| // system include files | ||
| #include <memory> | ||
| #include <cmath> | ||
|
|
||
| // user include files | ||
| #include "FWCore/Framework/interface/Frameworkfwd.h" | ||
| #include "FWCore/Framework/interface/EDProducer.h" | ||
|
|
||
| #include "FWCore/Framework/interface/Event.h" | ||
| #include "FWCore/Framework/interface/MakerMacros.h" | ||
| #include "FWCore/Framework/interface/ESHandle.h" | ||
|
|
||
| #include "FWCore/ParameterSet/interface/ParameterSet.h" | ||
| #include "FWCore/ServiceRegistry/interface/Service.h" | ||
|
|
||
| #include "DataFormats/Math/interface/deltaR.h" | ||
| #include "DataFormats/Common/interface/ValueMap.h" | ||
| #include "DataFormats/MuonReco/interface/Muon.h" | ||
| #include "DataFormats/EgammaCandidates/interface/GsfElectron.h" | ||
| #include "DataFormats/VertexReco/interface/VertexFwd.h" | ||
| #include "DataFormats/VertexReco/interface/Vertex.h" | ||
| #include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h" | ||
|
|
||
| // | ||
| // class declaration | ||
| // | ||
|
|
||
| class MuonMiniIso : public edm::EDProducer { | ||
| public: | ||
|
|
||
| typedef std::vector< edm::FwdPtr<reco::PFCandidate> > PFCollection; | ||
|
|
||
| explicit MuonMiniIso(const edm::ParameterSet&); | ||
| ~MuonMiniIso(); | ||
|
|
||
| private: | ||
| virtual void produce(edm::Event&, const edm::EventSetup&); | ||
|
|
||
| // ----------member data --------------------------- | ||
| const edm::InputTag probes_; | ||
| const edm::InputTag pfCandidates_; | ||
| double dRCandProbeVeto_; | ||
| double CandPtThreshold_; | ||
| double ChargedPVdZ_; | ||
| bool usePUcands_; | ||
|
|
||
| /// Store extra information in a ValueMap | ||
| template<typename Hand, typename T> | ||
| void storeMap(edm::Event &iEvent, | ||
| const Hand & handle, | ||
| const std::vector<T> & values, | ||
| const std::string & label) const ; | ||
| }; | ||
|
|
||
| // | ||
| // constants, enums and typedefs | ||
| // | ||
|
|
||
|
|
||
| // | ||
| // static data member definitions | ||
| // | ||
|
|
||
| // | ||
| // constructors and destructor | ||
| // | ||
| MuonMiniIso::MuonMiniIso(const edm::ParameterSet& iConfig): | ||
| probes_(iConfig.getParameter<edm::InputTag>("probes")), | ||
| pfCandidates_(iConfig.getParameter<edm::InputTag>("pfCandidates")), | ||
| dRCandProbeVeto_(iConfig.getParameter<double>("dRCandProbeVeto")), | ||
| CandPtThreshold_(iConfig.getParameter<double>("CandPtThreshold")) | ||
| { | ||
| produces<edm::ValueMap<float> >(); | ||
| } | ||
|
|
||
|
|
||
| MuonMiniIso::~MuonMiniIso() | ||
| { | ||
| } | ||
|
|
||
|
|
||
| // | ||
| // member functions | ||
| // | ||
|
|
||
| // ------------ method called to produce the data ------------ | ||
| void | ||
| MuonMiniIso::produce(edm::Event& iEvent, const edm::EventSetup& iSetup) | ||
| { | ||
| using namespace edm; | ||
|
|
||
| // read input | ||
| Handle<View<reco::Muon> > probes; | ||
| iEvent.getByLabel(probes_, probes); | ||
|
|
||
| //Handle<View<reco::PFCandidate> > pfCandidates; | ||
| Handle<PFCollection> pfCandidates; | ||
| iEvent.getByLabel(pfCandidates_, pfCandidates); | ||
|
|
||
| View<reco::Muon>::const_iterator probe, endprobes=probes->end(); | ||
| PFCollection::const_iterator iP, beginpf = pfCandidates->begin(), endpf=pfCandidates->end(); | ||
| unsigned int n = probes->size(); | ||
|
|
||
| std::vector<float> iso(n,0); | ||
|
|
||
| // loop on PROBES | ||
| unsigned int imu = 0; | ||
| for (probe = probes->begin(); probe != endprobes; ++probe, ++imu) { | ||
| const reco::Muon &mu = *probe; | ||
|
|
||
| // loop on PF candidates | ||
| int ipf=0; | ||
| for (iP = beginpf; iP != endpf; ++iP, ++ipf) { | ||
|
|
||
| // check pf candidate threshold | ||
| if(iP->get()->pt() < CandPtThreshold_) continue; | ||
|
|
||
| double dr = deltaR( *(iP->get() ) , mu ); | ||
| // check dr min | ||
| if (dr < dRCandProbeVeto_) continue; | ||
|
|
||
| // miniiso definition | ||
| if (mu.pt()<=50 && dr > 0.2) continue; | ||
| if (50 < mu.pt() && mu.pt() < 200 && dr > 10/mu.pt()) continue; | ||
| if (mu.pt()>200 && dr > 0.05) continue; | ||
|
|
||
| iso[imu] += iP->get()->pt(); | ||
|
|
||
| } | ||
| }// end loop on probes | ||
|
|
||
| storeMap(iEvent, probes, iso, ""); | ||
| } | ||
|
|
||
| template<typename Hand, typename T> | ||
| void | ||
| MuonMiniIso::storeMap(edm::Event &iEvent, | ||
| const Hand & handle, | ||
| const std::vector<T> & values, | ||
| const std::string & label) const { | ||
| using namespace edm; using namespace std; | ||
| auto_ptr<ValueMap<T> > valMap(new ValueMap<T>()); | ||
| typename edm::ValueMap<T>::Filler filler(*valMap); | ||
| filler.insert(handle, values.begin(), values.end()); | ||
| filler.fill(); | ||
| iEvent.put(valMap, label); | ||
| } | ||
|
|
||
| //define this as a plug-in | ||
| DEFINE_FWK_MODULE(MuonMiniIso); |
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