|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
#include "GeneratorInterface/GenFilters/plugins/BCToEFilterAlgo.h"
#include "DataFormats/HepMCCandidate/interface/GenParticle.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include <cmath>
#include <cstdlib>
BCToEFilterAlgo::BCToEFilterAlgo(const edm::ParameterSet& iConfig, edm::ConsumesCollector&& iC)
: //set constants
maxAbsEta_((float)iConfig.getParameter<double>("maxAbsEta")),
eTThreshold_((float)iConfig.getParameter<double>("eTThreshold")),
genParSource_(iC.consumes<reco::GenParticleCollection>(iConfig.getParameter<edm::InputTag>("genParSource"))) {}
BCToEFilterAlgo::~BCToEFilterAlgo() {}
//look for status==1 e coming from b or c hadron
//there is an eT threshold on the electron (configurable)
bool BCToEFilterAlgo::filter(const edm::Event& iEvent) const {
bool result = false;
edm::Handle<reco::GenParticleCollection> genParsHandle;
iEvent.getByToken(genParSource_, genParsHandle);
reco::GenParticleCollection genPars = *genParsHandle;
for (uint32_t ig = 0; ig < genPars.size(); ig++) {
reco::GenParticle gp = genPars.at(ig);
if (gp.status() == 1 && std::abs(gp.pdgId()) == 11 && gp.et() > eTThreshold_ && std::fabs(gp.eta()) < maxAbsEta_) {
if (hasBCAncestors(gp)) {
result = true;
}
}
}
return result;
}
//does this particle have an ancestor (mother, mother of mother, etc.) that is a b or c hadron?
//attention: the GenParticle argument must have the motherRef correctly filled for this
//to work. That is, you had better have got it out of the genParticles collection
bool BCToEFilterAlgo::hasBCAncestors(const reco::GenParticle& gp) const {
//stopping condition: this particle is a b or c hadron
if (isBCHadron(gp))
return true;
//stopping condition: this particle has no mothers
if (gp.numberOfMothers() == 0)
return false;
//otherwise continue
bool retval = false;
for (uint32_t im = 0; im < gp.numberOfMothers(); im++) {
retval = retval || hasBCAncestors(*gp.motherRef(im));
}
return retval;
}
bool BCToEFilterAlgo::isBCHadron(const reco::GenParticle& gp) const { return isBCMeson(gp) || isBCBaryon(gp); }
bool BCToEFilterAlgo::isBCMeson(const reco::GenParticle& gp) const {
uint32_t pdgid = std::abs(gp.pdgId());
uint32_t hundreds = pdgid % 1000;
if (hundreds >= 400 && hundreds < 600) {
return true;
} else {
return false;
}
}
bool BCToEFilterAlgo::isBCBaryon(const reco::GenParticle& gp) const {
uint32_t pdgid = std::abs(gp.pdgId());
uint32_t thousands = pdgid % 10000;
if (thousands >= 4000 && thousands < 6000) {
return true;
} else {
return false;
}
}
|