CMS 3D CMS Logo

 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
HLTTauRefProducer.cc
Go to the documentation of this file.
2 #include "TLorentzVector.h"
3 // TAU includes
8 // ELECTRON includes
17 // MUON includes
20 #include "TLorentzVector.h"
22 //CaloTower includes
26 #include "Math/GenVector/VectorUtil.h"
27 
28 
29 using namespace edm;
30 using namespace reco;
31 using namespace std;
32 
34 {
35 
36 
37  //One Parameter Set per Collection
38 
39  ParameterSet pfTau = iConfig.getUntrackedParameter<edm::ParameterSet>("PFTaus");
40  PFTaus_ = consumes<reco::PFTauCollection>(pfTau.getUntrackedParameter<InputTag>("PFTauProducer"));
41  auto discs = pfTau.getUntrackedParameter<std::vector<InputTag> >("PFTauDiscriminators");
42  for(edm::InputTag& tag: discs) {
43  PFTauDis_.push_back(consumes<reco::PFTauDiscriminator>(tag));
44  }
45  doPFTaus_ = pfTau.getUntrackedParameter<bool>("doPFTaus",false);
46  ptMinPFTau_= pfTau.getUntrackedParameter<double>("ptMin",15.);
47 
49  Electrons_ = consumes<reco::GsfElectronCollection>(electrons.getUntrackedParameter<InputTag>("ElectronCollection"));
50  doElectrons_ = electrons.getUntrackedParameter<bool>("doElectrons",false);
51  e_doID_ = electrons.getUntrackedParameter<bool>("doID",false);
52  if(e_doID_) {
53  e_idAssocProd_ = consumes<reco::ElectronIDAssociationCollection>(electrons.getUntrackedParameter<InputTag>("IdCollection"));
54  }
55  e_ctfTrackCollectionSrc_ = electrons.getUntrackedParameter<InputTag>("TrackCollection");
56  e_ctfTrackCollection_ = consumes<reco::TrackCollection>(e_ctfTrackCollectionSrc_);
57  ptMinElectron_= electrons.getUntrackedParameter<double>("ptMin",15.);
58  e_doTrackIso_ = electrons.getUntrackedParameter<bool>("doTrackIso",false);
59  e_trackMinPt_= electrons.getUntrackedParameter<double>("ptMinTrack",1.5);
60  e_lipCut_= electrons.getUntrackedParameter<double>("lipMinTrack",1.5);
61  e_minIsoDR_= electrons.getUntrackedParameter<double>("InnerConeDR",0.02);
62  e_maxIsoDR_= electrons.getUntrackedParameter<double>("OuterConeDR",0.6);
63  e_isoMaxSumPt_= electrons.getUntrackedParameter<double>("MaxIsoVar",0.02);
64 
66  Muons_ = consumes<reco::MuonCollection>(muons.getUntrackedParameter<InputTag>("MuonCollection"));
67  doMuons_ = muons.getUntrackedParameter<bool>("doMuons",false);
68  ptMinMuon_= muons.getUntrackedParameter<double>("ptMin",15.);
69 
71  Jets_ = consumes<reco::CaloJetCollection>(jets.getUntrackedParameter<InputTag>("JetCollection"));
72  doJets_ = jets.getUntrackedParameter<bool>("doJets");
73  ptMinJet_= jets.getUntrackedParameter<double>("etMin");
74 
76  Towers_ = consumes<CaloTowerCollection>(towers.getUntrackedParameter<InputTag>("TowerCollection"));
77  doTowers_ = towers.getUntrackedParameter<bool>("doTowers");
78  ptMinTower_= towers.getUntrackedParameter<double>("etMin");
79  towerIsol_= towers.getUntrackedParameter<double>("towerIsolation");
80 
82  Photons_ = consumes<reco::PhotonCollection>(photons.getUntrackedParameter<InputTag>("PhotonCollection"));
83  doPhotons_ = photons.getUntrackedParameter<bool>("doPhotons");
84  ptMinPhoton_= photons.getUntrackedParameter<double>("etMin");
85  photonEcalIso_= photons.getUntrackedParameter<double>("ECALIso");
86 
88  MET_ = consumes<reco::CaloMETCollection>(met.getUntrackedParameter<InputTag>("METCollection"));
89  doMET_ = met.getUntrackedParameter<bool>("doMET",false);
90  ptMinMET_= met.getUntrackedParameter<double>("ptMin",15.);
91 
92 
93  etaMax = iConfig.getUntrackedParameter<double>("EtaMax",2.5);
94 
95 
96  //recoCollections
97  produces<LorentzVectorCollection>("PFTaus");
98  produces<LorentzVectorCollection>("Electrons");
99  produces<LorentzVectorCollection>("Muons");
100  produces<LorentzVectorCollection>("Jets");
101  produces<LorentzVectorCollection>("Photons");
102  produces<LorentzVectorCollection>("Towers");
103  produces<LorentzVectorCollection>("MET");
104 
105 }
106 
108 
110 {
111  if(doPFTaus_)
112  doPFTaus(iEvent,iES);
113  if(doElectrons_)
114  doElectrons(iEvent,iES);
115  if(doMuons_)
116  doMuons(iEvent,iES);
117  if(doJets_)
118  doJets(iEvent,iES);
119  if(doPhotons_)
120  doPhotons(iEvent,iES);
121  if(doTowers_)
122  doTowers(iEvent,iES);
123  if(doMET_)
124  doMET(iEvent,iES);
125 }
126 
127 void
129 {
130  auto_ptr<LorentzVectorCollection> product_PFTaus(new LorentzVectorCollection);
131  //Retrieve the collection
133  if(iEvent.getByToken(PFTaus_,pftaus)) {
134  for(unsigned int i=0; i<pftaus->size(); ++i) {
135  if((*pftaus)[i].pt()>ptMinPFTau_&&fabs((*pftaus)[i].eta())<etaMax)
136  {
137  reco::PFTauRef thePFTau(pftaus,i);
138  bool passAll = true;
141  if(iEvent.getByToken(token, pftaudis)) {
142  if((*pftaudis)[thePFTau] < 0.5) {
143  passAll = false;
144  break;
145  }
146  }else{
147  passAll = false;
148  break;
149  }
150  }
151  if(passAll) {
152  LorentzVector vec((*pftaus)[i].px(),(*pftaus)[i].py(),(*pftaus)[i].pz(),(*pftaus)[i].energy());
153  product_PFTaus->push_back(vec);
154  }
155  }
156  }
157  }
158  iEvent.put(product_PFTaus,"PFTaus");
159 }
160 
161 
162 void
164 {
165  auto_ptr<LorentzVectorCollection> product_Electrons(new LorentzVectorCollection);
166  //Retrieve the collections
167 
169  if(e_doID_){//UGLY HACK UNTIL GET ELETRON ID WORKING IN 210
170 
171  iEvent.getByToken(e_idAssocProd_, pEleID);
172 
173  if (!pEleID.isValid()){
174  edm::LogInfo("")<< "Error! Can't get electronIDAssocProducer by label. ";
175  e_doID_ = false;
176  }
177  }
179  iEvent.getByToken(e_ctfTrackCollection_, pCtfTracks);
180  if (!pCtfTracks.isValid()) {
181  edm::LogInfo("")<< "Error! Can't get " << e_ctfTrackCollectionSrc_.label() << " by label. ";
182  iEvent.put(product_Electrons,"Electrons");
183  return;
184  }
185  const reco::TrackCollection * ctfTracks = pCtfTracks.product();
187  if(iEvent.getByToken(Electrons_,electrons))
188  for(size_t i=0;i<electrons->size();++i)
189  {
190  edm::Ref<reco::GsfElectronCollection> electronRef(electrons,i);
191  bool idDec=false;
192  if(e_doID_){
194  tagIDAssocItr = pEleID->find(electronRef);
195  const reco::ElectronIDRef& id_tag = tagIDAssocItr->val;
196  idDec=id_tag->cutBasedDecision();
197  }else idDec=true;
198  if((*electrons)[i].pt()>ptMinElectron_&&fabs((*electrons)[i].eta())<etaMax&&idDec)
199  {
200  if(e_doTrackIso_){
201  reco::TrackCollection::const_iterator tr = ctfTracks->begin();
202  double sum_of_pt_ele=0;
203  for(;tr != ctfTracks->end();++tr)
204  {
205  double lip = (*electrons)[i].gsfTrack()->dz() - tr->dz();
206  if(tr->pt() > e_trackMinPt_ && fabs(lip) < e_lipCut_){
207  double dphi=fabs(tr->phi()-(*electrons)[i].trackMomentumAtVtx().phi());
208  if(dphi>acos(-1.))dphi=2*acos(-1.)-dphi;
209  double deta=fabs(tr->eta()-(*electrons)[i].trackMomentumAtVtx().eta());
210  double dr_ctf_ele = sqrt(deta*deta+dphi*dphi);
211  if((dr_ctf_ele>e_minIsoDR_) && (dr_ctf_ele<e_maxIsoDR_)){
212  double cft_pt_2 = (tr->pt())*(tr->pt());
213  sum_of_pt_ele += cft_pt_2;
214  }
215  }
216  }
217  double isolation_value_ele = sum_of_pt_ele/((*electrons)[i].trackMomentumAtVtx().Rho()*(*electrons)[i].trackMomentumAtVtx().Rho());
218  if(isolation_value_ele<e_isoMaxSumPt_){
219  LorentzVector vec((*electrons)[i].px(),(*electrons)[i].py(),(*electrons)[i].pz(),(*electrons)[i].energy());
220  product_Electrons->push_back(vec);
221  }
222 
223  }
224  else{
225  LorentzVector vec((*electrons)[i].px(),(*electrons)[i].py(),(*electrons)[i].pz(),(*electrons)[i].energy());
226  product_Electrons->push_back(vec);
227  }
228  }
229  }
230 
231  iEvent.put(product_Electrons,"Electrons");
232 }
233 
234 void
236 {
237  auto_ptr<LorentzVectorCollection> product_Muons(new LorentzVectorCollection);
238  //Retrieve the collection
240  if(iEvent.getByToken(Muons_,muons))
241 
242  for(size_t i = 0 ;i<muons->size();++i)
243  {
244 
245  if((*muons)[i].pt()>ptMinMuon_&&fabs((*muons)[i].eta())<etaMax)
246  {
247  LorentzVector vec((*muons)[i].px(),(*muons)[i].py(),(*muons)[i].pz(),(*muons)[i].energy());
248  product_Muons->push_back(vec);
249  }
250  }
251 
252 
253  iEvent.put(product_Muons,"Muons");
254 
255 }
256 
257 
258 void
260 {
261  auto_ptr<LorentzVectorCollection> product_Jets(new LorentzVectorCollection);
262  //Retrieve the collection
264  if(iEvent.getByToken(Jets_,jets))
265  for(size_t i = 0 ;i<jets->size();++i)
266  {
267  if((*jets)[i].et()>ptMinJet_&&fabs((*jets)[i].eta())<etaMax)
268  {
269  LorentzVector vec((*jets)[i].px(),(*jets)[i].py(),(*jets)[i].pz(),(*jets)[i].energy());
270  product_Jets->push_back(vec);
271  }
272  }
273  iEvent.put(product_Jets,"Jets");
274 }
275 
276 void
278 {
279  auto_ptr<LorentzVectorCollection> product_Towers(new LorentzVectorCollection);
280  //Retrieve the collection
282  if(iEvent.getByToken(Towers_,towers))
283  for(size_t i = 0 ;i<towers->size();++i)
284  {
285  if((*towers)[i].pt()>ptMinTower_&&fabs((*towers)[i].eta())<etaMax)
286  {
287  //calculate isolation
288  double isolET=0;
289  for(unsigned int j=0;j<towers->size();++j)
290  {
291  if(ROOT::Math::VectorUtil::DeltaR((*towers)[i].p4(),(*towers)[j].p4())<0.5)
292  isolET+=(*towers)[j].pt();
293  }
294  isolET-=(*towers)[i].pt();
295  if(isolET<towerIsol_)
296  {
297  LorentzVector vec((*towers)[i].px(),(*towers)[i].py(),(*towers)[i].pz(),(*towers)[i].energy());
298  product_Towers->push_back(vec);
299  }
300  }
301  }
302  iEvent.put(product_Towers,"Towers");
303 }
304 
305 
306 void
308 {
309  auto_ptr<LorentzVectorCollection> product_Gammas(new LorentzVectorCollection);
310  //Retrieve the collection
312  if(iEvent.getByToken(Photons_,photons))
313  for(size_t i = 0 ;i<photons->size();++i)
314  if((*photons)[i].ecalRecHitSumEtConeDR04()<photonEcalIso_)
315  {
316  if((*photons)[i].et()>ptMinPhoton_&&fabs((*photons)[i].eta())<etaMax)
317  {
318  LorentzVector vec((*photons)[i].px(),(*photons)[i].py(),(*photons)[i].pz(),(*photons)[i].energy());
319  product_Gammas->push_back(vec);
320  }
321  }
322  iEvent.put(product_Gammas,"Photons");
323 }
324 
325 void
327 {
328  auto_ptr<LorentzVectorCollection> product_MET(new LorentzVectorCollection);
329  //Retrieve the collection
331  if(iEvent.getByToken(MET_,met)){
332  double px = met->front().p4().Px();
333  double py = met->front().p4().Py();
334  double pt = met->front().p4().Pt();
335  LorentzVector vec(px,py,0,pt);
336  product_MET->push_back(vec);
337  }
338  iEvent.put(product_MET,"MET");
339 }
340 
T getUntrackedParameter(std::string const &, T const &) const
int i
Definition: DBlmapReader.cc:9
void doPFTaus(edm::Event &, const edm::EventSetup &)
virtual void produce(edm::Event &, const edm::EventSetup &)
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:462
const std::string id_tag("id")
std::vector< Track > TrackCollection
collection of Tracks
Definition: TrackFwd.h:14
HLTTauRefProducer(const edm::ParameterSet &)
void doTowers(edm::Event &, const edm::EventSetup &)
void doJets(edm::Event &, const edm::EventSetup &)
math::XYZTLorentzVectorD LorentzVector
int iEvent
Definition: GenABIO.cc:230
OrphanHandle< PROD > put(std::auto_ptr< PROD > product)
Put a new product.
Definition: Event.h:121
T sqrt(T t)
Definition: SSEVec.h:18
double p4[4]
Definition: TauolaWrapper.h:92
vector< PseudoJet > jets
void doMET(edm::Event &, const edm::EventSetup &)
int j
Definition: DBlmapReader.cc:9
bool isValid() const
Definition: HandleBase.h:75
void doMuons(edm::Event &, const edm::EventSetup &)
T const * product() const
Definition: Handle.h:81
std::vector< LorentzVector > LorentzVectorCollection
void doPhotons(edm::Event &, const edm::EventSetup &)
tuple muons
Definition: patZpeak.py:38
void doElectrons(edm::Event &, const edm::EventSetup &)