CMS 3D CMS Logo

 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
PATPhotonSlimmer.cc
Go to the documentation of this file.
1 
10 
13 
15 
17 
24 
25 namespace pat {
26 
28  public:
29  explicit PATPhotonSlimmer(const edm::ParameterSet & iConfig);
30  virtual ~PATPhotonSlimmer() { }
31 
32  virtual void produce(edm::Event & iEvent, const edm::EventSetup & iSetup);
33  virtual void beginLuminosityBlock(const edm::LuminosityBlock&, const edm::EventSetup&) override final;
34 
35  private:
36  edm::EDGetTokenT<edm::View<pat::Photon> > src_;
37 
39 
40  edm::EDGetTokenT<edm::ValueMap<std::vector<reco::PFCandidateRef>>> reco2pf_;
41  edm::EDGetTokenT<edm::Association<pat::PackedCandidateCollection>> pf2pc_;
42  edm::EDGetTokenT<pat::PackedCandidateCollection> pc_;
47  std::unique_ptr<pat::ObjectModifier<pat::Photon> > photonModifier_;
48  };
49 
50 } // namespace
51 
53  src_(consumes<edm::View<pat::Photon> >(iConfig.getParameter<edm::InputTag>("src"))),
54  dropSuperClusters_(iConfig.getParameter<std::string>("dropSuperCluster")),
55  dropBasicClusters_(iConfig.getParameter<std::string>("dropBasicClusters")),
56  dropPreshowerClusters_(iConfig.getParameter<std::string>("dropPreshowerClusters")),
57  dropSeedCluster_(iConfig.getParameter<std::string>("dropSeedCluster")),
58  dropRecHits_(iConfig.getParameter<std::string>("dropRecHits")),
59  linkToPackedPF_(iConfig.getParameter<bool>("linkToPackedPFCandidates")),
60  saveNonZSClusterShapes_(iConfig.getParameter<std::string>("saveNonZSClusterShapes")),
61  reducedBarrelRecHitCollectionToken_(consumes<EcalRecHitCollection>(iConfig.getParameter<edm::InputTag>("reducedBarrelRecHitCollection"))),
62  reducedEndcapRecHitCollectionToken_(consumes<EcalRecHitCollection>(iConfig.getParameter<edm::InputTag>("reducedEndcapRecHitCollection"))),
63  modifyPhoton_(iConfig.getParameter<bool>("modifyPhotons"))
64 {
66  if( modifyPhoton_ ) {
67  const edm::ParameterSet& mod_config = iConfig.getParameter<edm::ParameterSet>("modifierConfig");
68  photonModifier_.reset(new pat::ObjectModifier<pat::Photon>(mod_config) );
69  photonModifier_->setConsumes(sumes);
70  } else {
71  photonModifier_.reset(nullptr);
72  }
73 
74  produces<std::vector<pat::Photon> >();
75  if (linkToPackedPF_) {
76  reco2pf_ = consumes<edm::ValueMap<std::vector<reco::PFCandidateRef>>>(iConfig.getParameter<edm::InputTag>("recoToPFMap"));
77  pf2pc_ = consumes<edm::Association<pat::PackedCandidateCollection>>(iConfig.getParameter<edm::InputTag>("packedPFCandidates"));
78  pc_ = consumes<pat::PackedCandidateCollection>(iConfig.getParameter<edm::InputTag>("packedPFCandidates"));
79  }
80  mayConsume<EcalRecHitCollection>(edm::InputTag("reducedEcalRecHitsEB"));
81  mayConsume<EcalRecHitCollection>(edm::InputTag("reducedEcalRecHitsEE"));
82 }
83 
84 void
86 }
87 
88 void
90  using namespace edm;
91  using namespace std;
92 
94  iEvent.getByToken(src_, src);
95 
99  if (linkToPackedPF_) {
100  iEvent.getByToken(reco2pf_, reco2pf);
101  iEvent.getByToken(pf2pc_, pf2pc);
102  iEvent.getByToken(pc_, pc);
103  }
105 
106  auto_ptr<vector<pat::Photon> > out(new vector<pat::Photon>());
107  out->reserve(src->size());
108 
109  if( modifyPhoton_ ) { photonModifier_->setEvent(iEvent); }
110  if( modifyPhoton_ ) photonModifier_->setEventContent(iSetup);
111 
112  std::vector<unsigned int> keys;
113  for (View<pat::Photon>::const_iterator it = src->begin(), ed = src->end(); it != ed; ++it) {
114  out->push_back(*it);
115  pat::Photon & photon = out->back();
116 
117  if( modifyPhoton_ ) { photonModifier_->modify(photon); }
118 
119  if (dropSuperClusters_(photon)) { photon.superCluster_.clear(); photon.embeddedSuperCluster_ = false; }
120  if (dropBasicClusters_(photon)) { photon.basicClusters_.clear(); }
121  if (dropPreshowerClusters_(photon)) { photon.preshowerClusters_.clear(); }
122  if (dropSeedCluster_(photon)) { photon.seedCluster_.clear(); photon.embeddedSeedCluster_ = false; }
123  if (dropRecHits_(photon)) { photon.recHits_ = EcalRecHitCollection(); photon.embeddedRecHits_ = false; }
124 
125  if (linkToPackedPF_) {
126  //std::cout << " PAT photon in " << src.id() << " comes from " << photon.refToOrig_.id() << ", " << photon.refToOrig_.key() << std::endl;
127  keys.clear();
128  for(auto const& pf: (*reco2pf)[photon.refToOrig_]) {
129  if( pf2pc->contains(pf.id()) ) {
130  keys.push_back( (*pf2pc)[pf].key());
131  }
132  }
134  keys.begin(), keys.end());
135  //std::cout << "Photon with pt " << photon.pt() << " associated to " << photon.associatedPackedFCandidateIndices_.size() << " PF Candidates\n";
136  //if there's just one PF Cand then it's me, otherwise I have no univoque parent so my ref will be null
137  if (keys.size() == 1) {
138  photon.refToOrig_ = photon.sourceCandidatePtr(0);
139  } else {
140  photon.refToOrig_ = reco::CandidatePtr(pc.id());
141  }
142  }
143  if (saveNonZSClusterShapes_(photon)) {
144  std::vector<float> vCov = lazyToolsNoZS.localCovariances(*( photon.superCluster()->seed()));
145  float r9 = lazyToolsNoZS.e3x3( *( photon.superCluster()->seed())) / photon.superCluster()->rawEnergy() ;
146  float sigmaIetaIeta = ( !edm::isNotFinite(vCov[0]) ) ? sqrt(vCov[0]) : 0;
147  float sigmaIetaIphi = vCov[1];
148  float sigmaIphiIphi = ( !edm::isNotFinite(vCov[2]) ) ? sqrt(vCov[2]) : 0;
149  float e15o55 = lazyToolsNoZS.e1x5( *( photon.superCluster()->seed()) ) / lazyToolsNoZS.e5x5( *( photon.superCluster()->seed()) );
150  photon.addUserFloat("sigmaIetaIeta_NoZS", sigmaIetaIeta);
151  photon.addUserFloat("sigmaIetaIphi_NoZS", sigmaIetaIphi);
152  photon.addUserFloat("sigmaIphiIphi_NoZS", sigmaIphiIphi);
153  photon.addUserFloat("r9_NoZS", r9);
154  photon.addUserFloat("e1x5_over_e5x5_NoZS", e15o55);
155  }
156  }
157 
158  iEvent.put(out);
159 }
160 
162 using namespace pat;
T getParameter(std::string const &) const
StringCutObjectSelector< pat::Photon > dropSeedCluster_
Analysis-level Photon class.
Definition: Photon.h:47
reco::SuperClusterRef superCluster() const
override the superCluster method from CaloJet, to access the internal storage of the supercluster ...
float e5x5(const reco::BasicCluster &cluster)
StringCutObjectSelector< pat::Photon > dropPreshowerClusters_
edm::EDGetTokenT< edm::View< pat::Photon > > src_
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:457
StringCutObjectSelector< pat::Photon > dropSuperClusters_
#define DEFINE_FWK_MODULE(type)
Definition: MakerMacros.h:17
std::vector< pat::PackedCandidate > PackedCandidateCollection
void setAssociatedPackedPFCandidates(const edm::RefProd< pat::PackedCandidateCollection > &refprod, T beginIndexItr, T endIndexItr)
References to PFCandidates linked to this object (e.g. for isolation vetos or masking before jet recl...
Definition: Photon.h:309
std::vector< reco::SuperCluster > superCluster_
Definition: Photon.h:330
std::vector< reco::CaloCluster > preshowerClusters_
Place to store electron&#39;s preshower clusters internally.
Definition: Photon.h:336
float e3x3(const reco::BasicCluster &cluster)
edm::EDGetTokenT< edm::ValueMap< std::vector< reco::PFCandidateRef > > > reco2pf_
std::vector< reco::CaloCluster > seedCluster_
Place to store electron&#39;s seed cluster internally.
Definition: Photon.h:340
StringCutObjectSelector< pat::Photon > dropRecHits_
EDGetTokenT< ProductType > consumes(edm::InputTag const &tag)
int iEvent
Definition: GenABIO.cc:230
slimmer of PAT Taus
bool isNotFinite(T x)
Definition: isFinite.h:10
reco::CandidatePtr sourceCandidatePtr(size_type i) const
get the source candidate pointer with index i
OrphanHandle< PROD > put(std::auto_ptr< PROD > product)
Put a new product.
Definition: Event.h:115
void addUserFloat(const std::string &label, float data)
Set user-defined float.
Definition: PATObject.h:765
ConsumesCollector consumesCollector()
Use a ConsumesCollector to gather consumes information from helper functions.
T sqrt(T t)
Definition: SSEVec.h:48
edm::EDGetTokenT< EcalRecHitCollection > reducedEndcapRecHitCollectionToken_
edm::Ref< PFCandidateCollection > PFCandidateRef
persistent reference to a PFCandidate
string key
FastSim: produces sample of signal events, overlayed with premixed minbias events.
edm::EDGetTokenT< EcalRecHitCollection > reducedBarrelRecHitCollectionToken_
bool embeddedSuperCluster_
Definition: Photon.h:329
virtual void beginLuminosityBlock(const edm::LuminosityBlock &, const edm::EventSetup &) overridefinal
tuple out
Definition: dbtoconf.py:99
EcalRecHitCollection recHits_
Place to store electron&#39;s RecHits internally (5x5 around seed+ all RecHits)
Definition: Photon.h:344
bool embeddedRecHits_
True if RecHits stored internally.
Definition: Photon.h:342
edm::Ptr< Candidate > CandidatePtr
persistent reference to an object in a collection of Candidate objects
Definition: CandidateFwd.h:25
std::vector< float > localCovariances(const reco::BasicCluster &cluster, float w0=4.7)
edm::EDGetTokenT< edm::Association< pat::PackedCandidateCollection > > pf2pc_
string const
Definition: compareJSON.py:14
PATPhotonSlimmer(const edm::ParameterSet &iConfig)
#define private
Definition: FWFileEntry.h:17
edm::EDGetTokenT< pat::PackedCandidateCollection > pc_
StringCutObjectSelector< pat::Photon > saveNonZSClusterShapes_
boost::indirect_iterator< typename seq_t::const_iterator > const_iterator
Definition: View.h:85
float e1x5(const reco::BasicCluster &cluster)
bool embeddedSeedCluster_
True if seed cluster is stored internally.
Definition: Photon.h:338
virtual void produce(edm::Event &iEvent, const edm::EventSetup &iSetup)
std::unique_ptr< pat::ObjectModifier< pat::Photon > > photonModifier_
StringCutObjectSelector< pat::Photon > dropBasicClusters_
edm::Ptr< reco::Candidate > refToOrig_
Definition: PATObject.h:407
std::vector< reco::CaloCluster > basicClusters_
Place to store electron&#39;s basic clusters internally.
Definition: Photon.h:334
edm::SortedCollection< EcalRecHit > EcalRecHitCollection