1 |
#include "ZHAnalysis.h"
|
2 |
|
3 |
using namespace std;
|
4 |
|
5 |
void ZHAnalysis::BookTree() {
|
6 |
myevent_ = new myevent;
|
7 |
t=new TTree("t","tree");
|
8 |
t->Branch("myevent","myevent",&myevent_,256000,1);
|
9 |
}
|
10 |
|
11 |
|
12 |
bool ZHAnalysis::Run(UltraFastSim * ufs){
|
13 |
|
14 |
vector<TParticle> GenTaus_ = ufs->genTauList();
|
15 |
vector<TParticle> BQuarks_ = ufs->bQuarkList();
|
16 |
vector<TParticle> CQuarks_ = ufs->cQuarkList();
|
17 |
vector<TParticle> Photons_ = ufs->photonList();
|
18 |
vector<TParticle> Electrons_ = ufs->electronList();
|
19 |
vector<TParticle> Muons_ = ufs->muonList();
|
20 |
vector<TParticle> MuonsStdGeom_ = ufs->muonListStdGeom();
|
21 |
vector<TParticle> Taus_ = ufs->tauList();
|
22 |
vector<TParticle> ChargedHadrons_ = ufs->chargedHadronList();
|
23 |
vector<TParticle> NeutralHadrons_ = ufs->neutralHadronList();
|
24 |
vector<TLorentzVector> Jets_ = ufs->jetList();
|
25 |
vector<TLorentzVector> BJets_ = ufs->bJetList();
|
26 |
vector<TLorentzVector> BJetsStdGeom_ = ufs->bJetListStdGeom();
|
27 |
|
28 |
myevent_->clear();
|
29 |
|
30 |
for(vector<TParticle>::const_iterator it=GenTaus_.begin();it!=GenTaus_.end();it++){
|
31 |
myobject obj;
|
32 |
obj.px=it->Px();
|
33 |
obj.py=it->Py();
|
34 |
obj.pz=it->Pz();
|
35 |
obj.E=it->Energy();
|
36 |
obj.pt=it->Pt();
|
37 |
obj.eta=it->Eta();
|
38 |
obj.phi=it->Phi();
|
39 |
(myevent_->GenTaus).push_back(obj);
|
40 |
}
|
41 |
for(vector<TParticle>::const_iterator it=BQuarks_.begin();it!=BQuarks_.end();it++){
|
42 |
myobject obj;
|
43 |
obj.px=it->Px();
|
44 |
obj.py=it->Py();
|
45 |
obj.pz=it->Pz();
|
46 |
obj.E=it->Energy();
|
47 |
obj.pt=it->Pt();
|
48 |
obj.eta=it->Eta();
|
49 |
obj.phi=it->Phi();
|
50 |
(myevent_->BQuarks).push_back(obj);
|
51 |
}
|
52 |
for(vector<TParticle>::const_iterator it=CQuarks_.begin();it!=CQuarks_.end();it++){
|
53 |
myobject obj;
|
54 |
obj.px=it->Px();
|
55 |
obj.py=it->Py();
|
56 |
obj.pz=it->Pz();
|
57 |
obj.E=it->Energy();
|
58 |
obj.pt=it->Pt();
|
59 |
obj.eta=it->Eta();
|
60 |
obj.phi=it->Phi();
|
61 |
(myevent_->CQuarks).push_back(obj);
|
62 |
}
|
63 |
for(vector<TParticle>::const_iterator it=Photons_.begin();it!=Photons_.end();it++){
|
64 |
myobject obj;
|
65 |
obj.px=it->Px();
|
66 |
obj.py=it->Py();
|
67 |
obj.pz=it->Pz();
|
68 |
obj.E=it->Energy();
|
69 |
obj.pt=it->Pt();
|
70 |
obj.eta=it->Eta();
|
71 |
obj.phi=it->Phi();
|
72 |
(myevent_->Photons).push_back(obj);
|
73 |
}
|
74 |
for(vector<TParticle>::const_iterator it=Electrons_.begin();it!=Electrons_.end();it++){
|
75 |
myobject obj;
|
76 |
obj.px=it->Px();
|
77 |
obj.py=it->Py();
|
78 |
obj.pz=it->Pz();
|
79 |
obj.E=it->Energy();
|
80 |
obj.pt=it->Pt();
|
81 |
obj.eta=it->Eta();
|
82 |
obj.phi=it->Phi();
|
83 |
(myevent_->Electrons).push_back(obj);
|
84 |
}
|
85 |
for(vector<TParticle>::const_iterator it=Muons_.begin();it!=Muons_.end();it++){
|
86 |
myobject obj;
|
87 |
obj.px=it->Px();
|
88 |
obj.py=it->Py();
|
89 |
obj.pz=it->Pz();
|
90 |
obj.E=it->Energy();
|
91 |
obj.pt=it->Pt();
|
92 |
obj.eta=it->Eta();
|
93 |
obj.phi=it->Phi();
|
94 |
(myevent_->Muons).push_back(obj);
|
95 |
}
|
96 |
for(vector<TParticle>::const_iterator it=MuonsStdGeom_.begin();it!=MuonsStdGeom_.end();it++){
|
97 |
myobject obj;
|
98 |
obj.px=it->Px();
|
99 |
obj.py=it->Py();
|
100 |
obj.pz=it->Pz();
|
101 |
obj.E=it->Energy();
|
102 |
obj.pt=it->Pt();
|
103 |
obj.eta=it->Eta();
|
104 |
obj.phi=it->Phi();
|
105 |
(myevent_->MuonsStdGeom).push_back(obj);
|
106 |
}
|
107 |
for(vector<TParticle>::const_iterator it=Taus_.begin();it!=Taus_.end();it++){
|
108 |
myobject obj;
|
109 |
obj.px=it->Px();
|
110 |
obj.py=it->Py();
|
111 |
obj.pz=it->Pz();
|
112 |
obj.E=it->Energy();
|
113 |
obj.pt=it->Pt();
|
114 |
obj.eta=it->Eta();
|
115 |
obj.phi=it->Phi();
|
116 |
(myevent_->Taus).push_back(obj);
|
117 |
}
|
118 |
for(vector<TParticle>::const_iterator it=ChargedHadrons_.begin();it!=ChargedHadrons_.end();it++){
|
119 |
myobject obj;
|
120 |
obj.px=it->Px();
|
121 |
obj.py=it->Py();
|
122 |
obj.pz=it->Pz();
|
123 |
obj.E=it->Energy();
|
124 |
obj.pt=it->Pt();
|
125 |
obj.eta=it->Eta();
|
126 |
obj.phi=it->Phi();
|
127 |
(myevent_->ChargedHadrons).push_back(obj);
|
128 |
}
|
129 |
for(vector<TParticle>::const_iterator it=NeutralHadrons_.begin();it!=NeutralHadrons_.end();it++){
|
130 |
myobject obj;
|
131 |
obj.px=it->Px();
|
132 |
obj.py=it->Py();
|
133 |
obj.pz=it->Pz();
|
134 |
obj.E=it->Energy();
|
135 |
obj.pt=it->Pt();
|
136 |
obj.eta=it->Eta();
|
137 |
obj.phi=it->Phi();
|
138 |
(myevent_->NeutralHadrons).push_back(obj);
|
139 |
}
|
140 |
for(vector<TLorentzVector>::const_iterator it=Jets_.begin();it!=Jets_.end();it++){
|
141 |
myobject obj;
|
142 |
obj.px=it->Px();
|
143 |
obj.py=it->Py();
|
144 |
obj.pz=it->Pz();
|
145 |
obj.E=it->E();
|
146 |
obj.pt=it->Pt();
|
147 |
obj.eta=it->Eta();
|
148 |
obj.phi=it->Phi();
|
149 |
(myevent_->Jets).push_back(obj);
|
150 |
}
|
151 |
for(vector<TLorentzVector>::const_iterator it=BJets_.begin();it!=BJets_.end();it++){
|
152 |
myobject obj;
|
153 |
obj.px=it->Px();
|
154 |
obj.py=it->Py();
|
155 |
obj.pz=it->Pz();
|
156 |
obj.E=it->E();
|
157 |
obj.pt=it->Pt();
|
158 |
obj.eta=it->Eta();
|
159 |
obj.phi=it->Phi();
|
160 |
(myevent_->BJets).push_back(obj);
|
161 |
}
|
162 |
for(vector<TLorentzVector>::const_iterator it=BJetsStdGeom_.begin();it!=BJetsStdGeom_.end();it++){
|
163 |
myobject obj;
|
164 |
obj.px=it->Px();
|
165 |
obj.py=it->Py();
|
166 |
obj.pz=it->Pz();
|
167 |
obj.E=it->E();
|
168 |
obj.pt=it->Pt();
|
169 |
obj.eta=it->Eta();
|
170 |
obj.phi=it->Phi();
|
171 |
(myevent_->BJetsStdGeom).push_back(obj);
|
172 |
}
|
173 |
|
174 |
t->Fill();
|
175 |
|
176 |
return true;
|
177 |
|
178 |
}//Run
|