-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathGenSelection.cc
More file actions
76 lines (49 loc) · 3.56 KB
/
GenSelection.cc
File metadata and controls
76 lines (49 loc) · 3.56 KB
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
#include "GenSelection.h"
GenSel::GenSel() {};
GenSel::~GenSel() {};
void GenSel::GenSelection(std::vector<Double_t> Eta, std::vector<Double_t> Pt, std::vector<Double_t> Px, std::vector<Double_t> Py, std::vector<Double_t> Pz, std::vector<Double_t> E, std::vector<Double_t> Trkdx, std::vector<Double_t> Trkdy, std::vector<Double_t> Trkdz, Double_t Vertex_X, Double_t Vertex_Y, Double_t Vertex_Z, std::vector<Int_t> pdgId, std::vector<Int_t> status, std::vector<Int_t> ndaug, std::vector<Int_t> mother, std::vector<GenParticle>& genColl) {
dz_cut=1.0;
for (UInt_t ilep=0; ilep<Pt.size(); ilep++) {
dz = fabs(Trkdz[ilep]-Vertex_Z);
dxy = sqrt(pow(Trkdx[ilep]-Vertex_X,2)+pow(Trkdy[ilep]-Vertex_Y,2));
D0 = 0.0;
D0Error = 0.0;
Int_t Charge = 0.0;
Double_t TrkIso = 0.0;
LeptonRelIso = 0.0;
LeptonchiNdof = 0.0;
vLepton.SetPxPyPzE(Px[ilep], Py[ilep], Pz[ilep], E[ilep]);
fakeType = GenParticle::unknown;
looseTight = GenParticle::Other;
leptonType = GenParticle::Muon;
if ( !( (fabs(pdgId[ilep])==13 || fabs(pdgId[ilep])==15) && status[ilep]==3 && (fabs(pdgId[mother[ilep]])==24 || fabs(pdgId[mother[ilep]])==23 || fabs(pdgId[mother[ilep]])==90) ) ) continue;
// boh if ( fabs(pdgId[ilep]==13) && fabs(pdgId[mother[ilep]])==24 || status[ilep]==3 || fabs(pdgId[mother[ilep]])==23 ) continue;
//cout << "idx "<< ilep << " pdgID " << pdgId[ilep] << " madre " << mother[ilep] << " stato " << status[ilep]<< " ndaug " <<ndaug[ilep]<<endl;
if ( fabs(Eta[ilep]) < eta_cut && Pt[ilep] >= pt_cut_min && Pt[ilep] < pt_cut_max )
genColl.push_back( GenParticle(leptonType, ilep, vLepton, Eta[ilep], LeptonchiNdof, D0, D0Error, dxy, dz, Charge, fakeType, looseTight, TrkIso, LeptonRelIso, pdgId[ilep], mother[ilep]) );
}
std::sort( genColl.begin(), genColl.end(), GenPTSorter );
}
void GenSel::GenSelectionB(std::vector<Double_t> Eta, std::vector<Double_t> Pt, std::vector<Double_t> Px, std::vector<Double_t> Py, std::vector<Double_t> Pz, std::vector<Double_t> E, std::vector<Double_t> Trkdx, std::vector<Double_t> Trkdy, std::vector<Double_t> Trkdz, Double_t Vertex_X, Double_t Vertex_Y, Double_t Vertex_Z, std::vector<Int_t> pdgId, std::vector<Int_t> status, std::vector<Int_t> ndaug, std::vector<Int_t> mother, std::vector<GenParticle>& genColl) {
dz_cut=1.0;
for (UInt_t ilep=0; ilep<Pt.size(); ilep++) {
dz = fabs(Trkdz[ilep]-Vertex_Z);
dxy = sqrt(pow(Trkdx[ilep]-Vertex_X,2)+pow(Trkdy[ilep]-Vertex_Y,2));
D0 = 0.0;
D0Error = 0.0;
Int_t Charge = 0.0;
Double_t TrkIso = 0.0;
LeptonRelIso = 0.0;
LeptonchiNdof = 0.0;
vLepton.SetPxPyPzE(Px[ilep], Py[ilep], Pz[ilep], E[ilep]);
fakeType = GenParticle::unknown;
looseTight = GenParticle::Other;
leptonType = GenParticle::Muon;
if ( !( status[ilep]==3 && (fabs(pdgId[ilep])==1 || fabs(pdgId[ilep])==2 || fabs(pdgId[ilep])==3 || fabs(pdgId[ilep])==4 || fabs(pdgId[ilep])==5 || fabs(pdgId[ilep])==21) ) ) continue;
// boh if ( fabs(pdgId[ilep]==13) && fabs(pdgId[mother[ilep]])==24 || status[ilep]==3 || fabs(pdgId[mother[ilep]])==23 ) continue;
//cout << "idx "<< ilep << " pdgID " << pdgId[ilep] << " madre " << mother[ilep] << " stato " << status[ilep]<< " ndaug " <<ndaug[ilep]<<endl;
if ( fabs(Eta[ilep]) < eta_cut && Pt[ilep] >= pt_cut_min && Pt[ilep] < pt_cut_max )
genColl.push_back( GenParticle(leptonType, ilep, vLepton, Eta[ilep], LeptonchiNdof, D0, D0Error, dxy, dz, Charge, fakeType, looseTight, TrkIso, LeptonRelIso, pdgId[ilep], mother[ilep]) );
}
std::sort( genColl.begin(), genColl.end(), GenPTSorter );
}