CMS 3D CMS Logo

HGCalTriggerNtupleGenTau.cc
Go to the documentation of this file.
1 #include <vector>
6 
8 
10 public:
12 
13  void initialize(TTree&, const edm::ParameterSet&, edm::ConsumesCollector&&) final;
14  void fill(const edm::Event&, const edm::EventSetup&) final;
15 
16 private:
17  void clear() final;
18 
19  bool isGoodTau(const reco::GenParticle& candidate) const;
20  bool isStableLepton(const reco::GenParticle& daughter) const;
21  bool isElectron(const reco::GenParticle& daughter) const;
22  bool isMuon(const reco::GenParticle& daughter) const;
23  bool isChargedHadron(const reco::GenParticle& daughter) const;
24  bool isChargedHadronFromResonance(const reco::GenParticle& daughter) const;
25  bool isNeutralPion(const reco::GenParticle& daughter) const;
26  bool isNeutralPionFromResonance(const reco::GenParticle& daughter) const;
27  bool isIntermediateResonance(const reco::GenParticle& daughter) const;
28  bool isGamma(const reco::GenParticle& daughter) const;
29  bool isStableNeutralHadron(const reco::GenParticle& daughter) const;
30 
33 
34  std::vector<float> gentau_pt_;
35  std::vector<float> gentau_eta_;
36  std::vector<float> gentau_phi_;
37  std::vector<float> gentau_energy_;
38  std::vector<float> gentau_mass_;
39 
40  std::vector<float> gentau_vis_pt_;
41  std::vector<float> gentau_vis_eta_;
42  std::vector<float> gentau_vis_phi_;
43  std::vector<float> gentau_vis_energy_;
44  std::vector<float> gentau_vis_mass_;
45  std::vector<int> gentau_decayMode_;
46  std::vector<int> gentau_totNproducts_;
47  std::vector<int> gentau_totNgamma_;
48  std::vector<int> gentau_totNpiZero_;
49  std::vector<int> gentau_totNcharged_;
50 
51  std::vector<std::vector<float> > gentau_products_pt_;
52  std::vector<std::vector<float> > gentau_products_eta_;
53  std::vector<std::vector<float> > gentau_products_phi_;
54  std::vector<std::vector<float> > gentau_products_energy_;
55  std::vector<std::vector<float> > gentau_products_mass_;
56  std::vector<std::vector<int> > gentau_products_id_;
57 };
58 
60 
62 
64  const edm::ParameterSet& conf,
65  edm::ConsumesCollector&& collector) {
66  gen_token_ = collector.consumes<reco::GenParticleCollection>(conf.getParameter<edm::InputTag>("GenParticles"));
67  isPythia8generator_ = conf.getParameter<bool>("isPythia8");
68 
69  tree.Branch("gentau_pt", &gentau_pt_);
70  tree.Branch("gentau_eta", &gentau_eta_);
71  tree.Branch("gentau_phi", &gentau_phi_);
72  tree.Branch("gentau_energy", &gentau_energy_);
73  tree.Branch("gentau_mass", &gentau_mass_);
74  tree.Branch("gentau_vis_pt", &gentau_vis_pt_);
75  tree.Branch("gentau_vis_eta", &gentau_vis_eta_);
76  tree.Branch("gentau_vis_phi", &gentau_vis_phi_);
77  tree.Branch("gentau_vis_energy", &gentau_vis_energy_);
78  tree.Branch("gentau_vis_mass", &gentau_vis_mass_);
79  tree.Branch("gentau_products_pt", &gentau_products_pt_);
80  tree.Branch("gentau_products_eta", &gentau_products_eta_);
81  tree.Branch("gentau_products_phi", &gentau_products_phi_);
82  tree.Branch("gentau_products_energy", &gentau_products_energy_);
83  tree.Branch("gentau_products_mass", &gentau_products_mass_);
84  tree.Branch("gentau_products_id", &gentau_products_id_);
85  tree.Branch("gentau_decayMode", &gentau_decayMode_);
86  tree.Branch("gentau_totNproducts", &gentau_totNproducts_);
87  tree.Branch("gentau_totNgamma", &gentau_totNgamma_);
88  tree.Branch("gentau_totNpiZero", &gentau_totNpiZero_);
89  tree.Branch("gentau_totNcharged", &gentau_totNcharged_);
90 }
91 
93  return (std::abs(candidate.pdgId()) == 15 && candidate.status() == 2);
94 }
95 
97  return ((std::abs(candidate.pdgId()) == 211 || std::abs(candidate.pdgId()) == 321) && candidate.status() == 1 &&
98  candidate.isDirectPromptTauDecayProductFinalState() && candidate.isLastCopy());
99 }
100 
102  return ((std::abs(candidate.pdgId()) == 211 || std::abs(candidate.pdgId()) == 321) && candidate.status() == 1 &&
103  candidate.isLastCopy());
104 }
105 
107  return ((std::abs(candidate.pdgId()) == 11 || std::abs(candidate.pdgId()) == 13) && candidate.status() == 1 &&
108  candidate.isDirectPromptTauDecayProductFinalState() && candidate.isLastCopy());
109 }
110 
112  return (std::abs(candidate.pdgId()) == 11 && candidate.isDirectPromptTauDecayProductFinalState() &&
113  candidate.isLastCopy());
114 }
115 
117  return (std::abs(candidate.pdgId()) == 13 && candidate.isDirectPromptTauDecayProductFinalState() &&
118  candidate.isLastCopy());
119 }
120 
122  return (std::abs(candidate.pdgId()) == 111 && candidate.status() == 2 &&
124 }
125 
127  return (std::abs(candidate.pdgId()) == 111 && candidate.status() == 2 && candidate.statusFlags().isTauDecayProduct());
128 }
129 
131  return (std::abs(candidate.pdgId()) == 22 && candidate.status() == 1 && candidate.statusFlags().isTauDecayProduct() &&
132  !candidate.isDirectPromptTauDecayProductFinalState() && candidate.isLastCopy());
133 }
134 
136  return ((std::abs(candidate.pdgId()) == 213 || std::abs(candidate.pdgId()) == 20213 ||
137  std::abs(candidate.pdgId()) == 24) &&
138  candidate.status() == 2);
139 }
140 
142  return (!(std::abs(candidate.pdgId()) > 10 && std::abs(candidate.pdgId()) < 17) && !isChargedHadron(candidate) &&
143  candidate.status() == 1);
144 }
145 
148  e.getByToken(gen_token_, gen_particles_h);
149  const reco::GenParticleCollection& gen_particles = *gen_particles_h;
150 
151  clear();
152 
153  for (const auto& particle : gen_particles) {
154  /* select good taus */
155  if (isGoodTau(particle)) {
156  LorentzVector tau_p4vis(0., 0., 0., 0.);
157  gentau_pt_.emplace_back(particle.pt());
158  gentau_eta_.emplace_back(particle.eta());
159  gentau_phi_.emplace_back(particle.phi());
160  gentau_energy_.emplace_back(particle.energy());
161  gentau_mass_.emplace_back(particle.mass());
162 
163  int n_pi = 0;
164  int n_piZero = 0;
165  int n_gamma = 0;
166  int n_ele = 0;
167  int n_mu = 0;
168 
169  std::vector<float> tau_products_pt;
170  std::vector<float> tau_products_eta;
171  std::vector<float> tau_products_phi;
172  std::vector<float> tau_products_energy;
173  std::vector<float> tau_products_mass;
174  std::vector<int> tau_products_id;
175 
176  /* loop over tau daughters */
177  const reco::GenParticleRefVector& daughters = particle.daughterRefVector();
178 
179  for (const auto& daughter : daughters) {
180  reco::GenParticleRefVector finalProds;
181 
182  if (isStableLepton(*daughter)) {
183  if (isElectron(*daughter)) {
184  n_ele++;
185  } else if (isMuon(*daughter)) {
186  n_mu++;
187  }
188  tau_p4vis += (daughter->p4());
189  finalProds.push_back(daughter);
190  }
191 
192  else if (isChargedHadron(*daughter)) {
193  n_pi++;
194  tau_p4vis += (daughter->p4());
195  finalProds.push_back(daughter);
196  }
197 
198  else if (isNeutralPion(*daughter)) {
199  n_piZero++;
200  const reco::GenParticleRefVector& granddaughters = daughter->daughterRefVector();
201  for (const auto& granddaughter : granddaughters) {
202  if (isGamma(*granddaughter)) {
203  n_gamma++;
204  tau_p4vis += (granddaughter->p4());
205  finalProds.push_back(granddaughter);
206  }
207  }
208  }
209 
210  else if (isStableNeutralHadron(*daughter)) {
211  tau_p4vis += (daughter->p4());
212  finalProds.push_back(daughter);
213  }
214 
215  else {
216  const reco::GenParticleRefVector& granddaughters = daughter->daughterRefVector();
217 
218  for (const auto& granddaughter : granddaughters) {
219  if (isStableNeutralHadron(*granddaughter)) {
220  tau_p4vis += (granddaughter->p4());
221  finalProds.push_back(granddaughter);
222  }
223  }
224  }
225 
226  /* Here the selection of the decay product according to the Pythia6 decayTree */
227  if (!isPythia8generator_) {
228  if (isIntermediateResonance(*daughter)) {
229  const reco::GenParticleRefVector& grandaughters = daughter->daughterRefVector();
230  for (const auto& grandaughter : grandaughters) {
231  if (isChargedHadron(*grandaughter) || isChargedHadronFromResonance(*grandaughter)) {
232  n_pi++;
233  tau_p4vis += (grandaughter->p4());
234  finalProds.push_back(daughter);
235  } else if (isNeutralPion(*grandaughter) || isNeutralPionFromResonance(*grandaughter)) {
236  n_piZero++;
237  const reco::GenParticleRefVector& descendants = grandaughter->daughterRefVector();
238  for (const auto& descendant : descendants) {
239  if (isGamma(*descendant)) {
240  n_gamma++;
241  tau_p4vis += (descendant->p4());
242  finalProds.push_back(daughter);
243  }
244  }
245  }
246  }
247  }
248  }
249 
250  /* Fill daughter informations */
251  for (const auto& prod : finalProds) {
252  tau_products_pt.emplace_back(prod->pt());
253  tau_products_eta.emplace_back(prod->eta());
254  tau_products_phi.emplace_back(prod->phi());
255  tau_products_energy.emplace_back(prod->energy());
256  tau_products_mass.emplace_back(prod->mass());
257  tau_products_id.emplace_back(prod->pdgId());
258  }
259  }
260 
261  /* assign the tau-variables */
262  gentau_vis_pt_.emplace_back(tau_p4vis.Pt());
263  gentau_vis_eta_.emplace_back(tau_p4vis.Eta());
264  gentau_vis_phi_.emplace_back(tau_p4vis.Phi());
265  gentau_vis_energy_.emplace_back(tau_p4vis.E());
266  gentau_vis_mass_.emplace_back(tau_p4vis.M());
267  gentau_totNproducts_.emplace_back(n_pi + n_gamma);
268  gentau_totNgamma_.emplace_back(n_gamma);
269  gentau_totNpiZero_.emplace_back(n_piZero);
270  gentau_totNcharged_.emplace_back(n_pi);
271 
272  gentau_products_pt_.emplace_back(tau_products_pt);
273  gentau_products_eta_.emplace_back(tau_products_eta);
274  gentau_products_phi_.emplace_back(tau_products_phi);
275  gentau_products_energy_.emplace_back(tau_products_energy);
276  gentau_products_mass_.emplace_back(tau_products_mass);
277  gentau_products_id_.emplace_back(tau_products_id);
278 
279  /* leptonic tau decays */
280  if (n_pi == 0 && n_piZero == 0 && n_ele == 1) {
281  gentau_decayMode_.emplace_back(11);
282  } else if (n_pi == 0 && n_piZero == 0 && n_mu == 1) {
283  gentau_decayMode_.emplace_back(13);
284  }
285  /* 1-prong */
286  else if (n_pi == 1 && n_piZero == 0) {
287  gentau_decayMode_.emplace_back(0);
288  }
289  /* 1-prong + pi0s */
290  else if (n_pi == 1 && n_piZero >= 1) {
291  gentau_decayMode_.emplace_back(1);
292  }
293  /* 3-prongs */
294  else if (n_pi == 3 && n_piZero == 0) {
295  gentau_decayMode_.emplace_back(4);
296  }
297  /* 3-prongs + pi0s */
298  else if (n_pi == 3 && n_piZero >= 1) {
299  gentau_decayMode_.emplace_back(5);
300  }
301  /* other decays */
302  else {
303  gentau_decayMode_.emplace_back(-1);
304  }
305  }
306  }
307 }
308 
310  gentau_pt_.clear();
311  gentau_eta_.clear();
312  gentau_phi_.clear();
313  gentau_energy_.clear();
314  gentau_mass_.clear();
315  gentau_decayMode_.clear();
316  gentau_vis_pt_.clear();
317  gentau_vis_eta_.clear();
318  gentau_vis_phi_.clear();
319  gentau_vis_energy_.clear();
320  gentau_vis_mass_.clear();
321  gentau_totNproducts_.clear();
322  gentau_totNgamma_.clear();
323  gentau_totNcharged_.clear();
324  gentau_products_pt_.clear();
325  gentau_products_eta_.clear();
326  gentau_products_phi_.clear();
327  gentau_products_energy_.clear();
328  gentau_products_mass_.clear();
329  gentau_products_id_.clear();
330 }
HGCalTriggerNtupleGenTau::gentau_totNcharged_
std::vector< int > gentau_totNcharged_
Definition: HGCalTriggerNtupleGenTau.cc:49
HGCalTriggerNtupleGenTau::gentau_products_phi_
std::vector< std::vector< float > > gentau_products_phi_
Definition: HGCalTriggerNtupleGenTau.cc:53
reco::LeafCandidate::status
int status() const final
status word
Definition: LeafCandidate.h:180
reco::GenParticle
Definition: GenParticle.h:21
HGCalTriggerNtupleGenTau::isStableNeutralHadron
bool isStableNeutralHadron(const reco::GenParticle &daughter) const
Definition: HGCalTriggerNtupleGenTau.cc:141
reco::GenParticle::isLastCopy
bool isLastCopy() const
Definition: GenParticle.h:101
tree
Definition: tree.py:1
reco::GenParticleCollection
std::vector< GenParticle > GenParticleCollection
collection of GenParticles
Definition: GenParticleFwd.h:13
HGCalTriggerNtupleGenTau::gentau_vis_mass_
std::vector< float > gentau_vis_mass_
Definition: HGCalTriggerNtupleGenTau.cc:44
HGCalTriggerNtupleGenTau::gentau_totNproducts_
std::vector< int > gentau_totNproducts_
Definition: HGCalTriggerNtupleGenTau.cc:46
HGCalTriggerNtupleGenTau::isIntermediateResonance
bool isIntermediateResonance(const reco::GenParticle &daughter) const
Definition: HGCalTriggerNtupleGenTau.cc:135
HGCalTriggerNtupleGenTau::gentau_pt_
std::vector< float > gentau_pt_
Definition: HGCalTriggerNtupleGenTau.cc:34
HGCalTriggerNtupleGenTau::isChargedHadronFromResonance
bool isChargedHadronFromResonance(const reco::GenParticle &daughter) const
Definition: HGCalTriggerNtupleGenTau.cc:101
HGCalTriggerNtupleGenTau::gentau_vis_eta_
std::vector< float > gentau_vis_eta_
Definition: HGCalTriggerNtupleGenTau.cc:41
edm::RefVector< GenParticleCollection >
edm::Handle< reco::GenParticleCollection >
HGCalTriggerNtupleGenTau::initialize
void initialize(TTree &, const edm::ParameterSet &, edm::ConsumesCollector &&) final
Definition: HGCalTriggerNtupleGenTau.cc:63
HGCalTriggerNtupleGenTau::gentau_vis_pt_
std::vector< float > gentau_vis_pt_
Definition: HGCalTriggerNtupleGenTau.cc:40
HGCalTriggerNtupleGenTau::gentau_products_mass_
std::vector< std::vector< float > > gentau_products_mass_
Definition: HGCalTriggerNtupleGenTau.cc:55
HGCalTriggerNtupleGenTau::gentau_totNgamma_
std::vector< int > gentau_totNgamma_
Definition: HGCalTriggerNtupleGenTau.cc:47
HGCalTriggerNtupleGenTau::isGoodTau
bool isGoodTau(const reco::GenParticle &candidate) const
Definition: HGCalTriggerNtupleGenTau.cc:92
GenParticle.h
HGCalTriggerNtupleGenTau::gentau_totNpiZero_
std::vector< int > gentau_totNpiZero_
Definition: HGCalTriggerNtupleGenTau.cc:48
HGCalTriggerNtupleBase
Definition: HGCalTriggerNtupleBase.h:10
HGCalTriggerNtupleGenTau::gentau_decayMode_
std::vector< int > gentau_decayMode_
Definition: HGCalTriggerNtupleGenTau.cc:45
HGCalTriggerNtupleGenTau::isNeutralPionFromResonance
bool isNeutralPionFromResonance(const reco::GenParticle &daughter) const
Definition: HGCalTriggerNtupleGenTau.cc:126
HGCalTriggerNtupleGenTau::gentau_energy_
std::vector< float > gentau_energy_
Definition: HGCalTriggerNtupleGenTau.cc:37
BPhysicsValidation_cfi.daughters
daughters
Definition: BPhysicsValidation_cfi.py:11
dumpMFGeometry_cfg.prod
prod
Definition: dumpMFGeometry_cfg.py:24
HGCalTriggerNtupleGenTau::gentau_mass_
std::vector< float > gentau_mass_
Definition: HGCalTriggerNtupleGenTau.cc:38
HGCalTriggerNtupleGenTau::isGamma
bool isGamma(const reco::GenParticle &daughter) const
Definition: HGCalTriggerNtupleGenTau.cc:130
HGCalTriggerNtupleGenTau::gentau_products_id_
std::vector< std::vector< int > > gentau_products_id_
Definition: HGCalTriggerNtupleGenTau.cc:56
DEFINE_EDM_PLUGIN
#define DEFINE_EDM_PLUGIN(factory, type, name)
Definition: PluginFactory.h:124
HGCalTriggerNtupleGenTau::isElectron
bool isElectron(const reco::GenParticle &daughter) const
Definition: HGCalTriggerNtupleGenTau.cc:111
edm::ParameterSet
Definition: ParameterSet.h:36
HGCalTriggerNtupleGenTau::gentau_phi_
std::vector< float > gentau_phi_
Definition: HGCalTriggerNtupleGenTau.cc:36
HGCalTriggerNtupleGenTau::isNeutralPion
bool isNeutralPion(const reco::GenParticle &daughter) const
Definition: HGCalTriggerNtupleGenTau.cc:121
HGCalTriggerNtupleGenTau::isMuon
bool isMuon(const reco::GenParticle &daughter) const
Definition: HGCalTriggerNtupleGenTau.cc:116
LorentzVector
math::XYZTLorentzVector LorentzVector
Definition: HGCalTriggerNtupleGenTau.cc:7
edmplugin::PluginFactory
Definition: PluginFactory.h:34
reco::LeafCandidate::pdgId
int pdgId() const final
PDG identifier.
Definition: LeafCandidate.h:176
reco::GenParticle::statusFlags
const GenStatusFlags & statusFlags() const
Definition: GenParticle.h:38
HGCalTriggerNtupleBase.h
reco::GenParticle::isDirectPromptTauDecayProductFinalState
bool isDirectPromptTauDecayProductFinalState() const
Definition: GenParticle.h:59
LorentzVector.h
edm::EventSetup
Definition: EventSetup.h:57
HGCalTriggerNtupleGenTau::fill
void fill(const edm::Event &, const edm::EventSetup &) final
Definition: HGCalTriggerNtupleGenTau.cc:146
HGCalTriggerNtupleGenTau::isChargedHadron
bool isChargedHadron(const reco::GenParticle &daughter) const
Definition: HGCalTriggerNtupleGenTau.cc:96
edm::EDGetToken
Definition: EDGetToken.h:35
edm::RefVector::push_back
void push_back(value_type const &ref)
Add a Ref<C, T> to the RefVector.
Definition: RefVector.h:67
HGCalTriggerNtupleGenTau::isStableLepton
bool isStableLepton(const reco::GenParticle &daughter) const
Definition: HGCalTriggerNtupleGenTau.cc:106
HGCalTriggerNtupleGenTau
Definition: HGCalTriggerNtupleGenTau.cc:9
edm::ParameterSet::getParameter
T getParameter(std::string const &) const
HGCalTriggerNtupleGenTau::gen_token_
edm::EDGetToken gen_token_
Definition: HGCalTriggerNtupleGenTau.cc:31
HGCalTriggerNtupleGenTau::HGCalTriggerNtupleGenTau
HGCalTriggerNtupleGenTau(const edm::ParameterSet &)
Definition: HGCalTriggerNtupleGenTau.cc:61
HGCalTriggerNtupleGenTau::gentau_vis_phi_
std::vector< float > gentau_vis_phi_
Definition: HGCalTriggerNtupleGenTau.cc:42
HGCalTriggerNtupleGenTau::gentau_vis_energy_
std::vector< float > gentau_vis_energy_
Definition: HGCalTriggerNtupleGenTau.cc:43
HGCalTriggerNtupleGenTau::gentau_eta_
std::vector< float > gentau_eta_
Definition: HGCalTriggerNtupleGenTau.cc:35
math::XYZTLorentzVector
XYZTLorentzVectorD XYZTLorentzVector
Lorentz vector with cylindrical internal representation using pseudorapidity.
Definition: LorentzVector.h:29
LorentzVector
math::XYZTLorentzVector LorentzVector
Definition: HLTMuonMatchAndPlot.h:49
HGCalTriggerNtupleGenTau::gentau_products_energy_
std::vector< std::vector< float > > gentau_products_energy_
Definition: HGCalTriggerNtupleGenTau.cc:54
reco::GenStatusFlags::isTauDecayProduct
bool isTauDecayProduct() const
Definition: GenStatusFlags.h:50
HGCalTriggerNtupleGenTau::gentau_products_pt_
std::vector< std::vector< float > > gentau_products_pt_
Definition: HGCalTriggerNtupleGenTau.cc:51
funct::abs
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
HGCalTriggerNtupleGenTau::isPythia8generator_
bool isPythia8generator_
Definition: HGCalTriggerNtupleGenTau.cc:32
edm::Event
Definition: Event.h:73
HGCalTriggerNtupleGenTau::gentau_products_eta_
std::vector< std::vector< float > > gentau_products_eta_
Definition: HGCalTriggerNtupleGenTau.cc:52
edm::InputTag
Definition: InputTag.h:15
HGCalTriggerNtupleGenTau::clear
void clear() final
Definition: HGCalTriggerNtupleGenTau.cc:309
edm::ConsumesCollector
Definition: ConsumesCollector.h:39
MillePedeFileConverter_cfg.e
e
Definition: MillePedeFileConverter_cfg.py:37