1 | /*
|
---|
2 | ---- Delphes ----
|
---|
3 | A Fast Simulator for general purpose LHC detector
|
---|
4 | S. Ovyn ~~~~ severine.ovyn@uclouvain.be
|
---|
5 |
|
---|
6 | Center for Particle Physics and Phenomenology (CP3)
|
---|
7 | Universite Catholique de Louvain (UCL)
|
---|
8 | Louvain-la-Neuve, Belgium
|
---|
9 | */
|
---|
10 |
|
---|
11 | /// \file Delphes.cpp
|
---|
12 | /// \brief executable for the Delphes
|
---|
13 |
|
---|
14 | #include "TChain.h"
|
---|
15 | #include "TApplication.h"
|
---|
16 | #include "TStopwatch.h"
|
---|
17 | #include "TFile.h"
|
---|
18 |
|
---|
19 | #include "ExRootTreeReader.h"
|
---|
20 | #include "ExRootTreeWriter.h"
|
---|
21 | #include "ExRootTreeBranch.h"
|
---|
22 |
|
---|
23 | #include "DataConverter.h"
|
---|
24 | #include "HEPEVTConverter.h"
|
---|
25 | #include "LHEFConverter.h"
|
---|
26 | #include "STDHEPConverter.h"
|
---|
27 |
|
---|
28 | #include "SmearUtil.h"
|
---|
29 | #include "BFieldProp.h"
|
---|
30 | #include "TriggerUtil.h"
|
---|
31 | #include "VeryForward.h"
|
---|
32 | #include "JetsUtil.h"
|
---|
33 | #include "FrogUtil.h"
|
---|
34 |
|
---|
35 | #include <vector>
|
---|
36 | #include <iostream>
|
---|
37 |
|
---|
38 | using namespace std;
|
---|
39 |
|
---|
40 | //------------------------------------------------------------------------------
|
---|
41 | void todo(string filename) {
|
---|
42 | ifstream infile(filename.c_str());
|
---|
43 | cout << "** TODO list ..." << endl;
|
---|
44 | while(infile.good()) {
|
---|
45 | string temp;
|
---|
46 | getline(infile,temp);
|
---|
47 | cout << "*" << temp << endl;
|
---|
48 | }
|
---|
49 | cout << "** done...\n";
|
---|
50 | }
|
---|
51 |
|
---|
52 | //------------------------------------------------------------------------------
|
---|
53 |
|
---|
54 | int main(int argc, char *argv[])
|
---|
55 | {
|
---|
56 | int appargc = 2;
|
---|
57 | char *appName= new char[20];
|
---|
58 | char *appOpt= new char[20];
|
---|
59 | sprintf(appName,"Delphes");
|
---|
60 | sprintf(appOpt,"-b");
|
---|
61 | char *appargv[] = {appName,appOpt};
|
---|
62 | TApplication app(appName, &appargc, appargv);
|
---|
63 | delete [] appName;
|
---|
64 | delete [] appOpt;
|
---|
65 |
|
---|
66 |
|
---|
67 | if(argc != 4 && argc != 3 && argc != 5) {
|
---|
68 | cout << " Usage: " << argv[0] << " input_file output_file [detector_card] [trigger_card] " << endl;
|
---|
69 | cout << " input_list - list of files in Ntpl, StdHep of LHEF format," << endl;
|
---|
70 | cout << " output_file - output file." << endl;
|
---|
71 | cout << " detector_card - Datacard containing resolution variables for the detector simulation (optional) "<<endl;
|
---|
72 | cout << " trigger_card - Datacard containing the trigger algorithms (optional) "<<endl;
|
---|
73 | exit(1);
|
---|
74 | }
|
---|
75 |
|
---|
76 |
|
---|
77 |
|
---|
78 |
|
---|
79 | cout << endl << endl;
|
---|
80 |
|
---|
81 | cout <<"*********************************************************************"<< endl;
|
---|
82 | cout <<"*********************************************************************"<< endl;
|
---|
83 | cout <<"** **"<< endl;
|
---|
84 | cout <<"** Welcome to **"<< endl;
|
---|
85 | cout <<"** **"<< endl;
|
---|
86 | cout <<"** **"<< endl;
|
---|
87 | cout <<"** .ddddddd- lL hH **"<< endl;
|
---|
88 | cout <<"** -Dd` `dD: Ll hH` **"<< endl;
|
---|
89 | cout <<"** dDd dDd eeee. lL .pp+pp Hh+hhh` -eeee- `sssss **"<< endl;
|
---|
90 | cout <<"** -Dd `DD ee. ee Ll .Pp. PP Hh. HH. ee. ee sSs **"<< endl;
|
---|
91 | cout <<"** dD` dDd eEeee: lL. pP. pP hH hH` eEeee:` -sSSSs. **"<< endl;
|
---|
92 | cout <<"** .Dd :dd eE. LlL PpppPP Hh Hh eE sSS **"<< endl;
|
---|
93 | cout <<"** dddddd:. eee+: lL. pp. hh. hh eee+ sssssS **"<< endl;
|
---|
94 | cout <<"** Pp **"<< endl;
|
---|
95 | cout <<"** **"<< endl;
|
---|
96 | cout <<"** Delphes, a framework for the fast simulation **"<< endl;
|
---|
97 | cout <<"** of a generic collider experiment **"<< endl;
|
---|
98 | cout <<"** **"<< endl;
|
---|
99 | cout <<"** --- Version 1.3beta of Delphes --- **"<< endl;
|
---|
100 | cout <<"** Last date of change: 29 January 2009 **"<< endl;
|
---|
101 | cout <<"** **"<< endl;
|
---|
102 | cout <<"** **"<< endl;
|
---|
103 | cout <<"** This package uses: **"<< endl;
|
---|
104 | cout <<"** ------------------ **"<< endl;
|
---|
105 | cout <<"** FastJet algorithm: Phys. Lett. B641 (2006) [hep-ph/0512210] **"<< endl;
|
---|
106 | cout <<"** Hector: JINST 2:P09005 (2007) [physics.acc-ph:0707.1198v2] **"<< endl;
|
---|
107 | cout <<"** FROG: **"<< endl;
|
---|
108 | cout <<"** **"<< endl;
|
---|
109 | cout <<"**-----------------------------------------------------------------**"<< endl;
|
---|
110 | cout <<"** **"<< endl;
|
---|
111 | cout <<"** Main authors: **"<< endl;
|
---|
112 | cout <<"** ------------- **"<< endl;
|
---|
113 | cout <<"** **"<< endl;
|
---|
114 | cout <<"** Séverine Ovyn Xavier Rouby **"<< endl;
|
---|
115 | cout <<"** severine.ovyn@uclouvain.be xavier.rouby@cern **"<< endl;
|
---|
116 | cout <<"** Center for Particle Physics and Phenomenology (CP3) **"<< endl;
|
---|
117 | cout <<"** Universite Catholique de Louvain (UCL) **"<< endl;
|
---|
118 | cout <<"** Louvain-la-Neuve, Belgium **"<< endl;
|
---|
119 | cout <<"** **"<< endl;
|
---|
120 | cout <<"**-----------------------------------------------------------------**"<< endl;
|
---|
121 | cout <<"** **"<< endl;
|
---|
122 | cout <<"** Former Delphes versions and documentation can be found on : **"<< endl;
|
---|
123 | cout <<"** http://www.fynu.ucl.ac.be/delphes.html **"<< endl;
|
---|
124 | cout <<"** **"<< endl;
|
---|
125 | cout <<"** **"<< endl;
|
---|
126 | cout <<"** Disclaimer: this program is a beta version of Delphes and **"<< endl;
|
---|
127 | cout <<"** therefore comes without guarantees. Beware of errors and please **"<< endl;
|
---|
128 | cout <<"** give us your feedbacks about potential bugs **"<< endl;
|
---|
129 | cout <<"** **"<< endl;
|
---|
130 | cout <<"*********************************************************************"<< endl;
|
---|
131 | cout <<"*********************************************************************"<< endl;
|
---|
132 |
|
---|
133 | // 1. ********** initialisation ***********
|
---|
134 |
|
---|
135 | srand (time (NULL)); /* Initialisation du générateur */
|
---|
136 | TStopwatch globalwatch, loopwatch, triggerwatch, frogwatch;
|
---|
137 | globalwatch.Start();
|
---|
138 |
|
---|
139 |
|
---|
140 | //read the input TROOT file
|
---|
141 | string inputFileList(argv[1]), outputfilename(argv[2]);
|
---|
142 | if(outputfilename.find(".root") > outputfilename.length() ) {
|
---|
143 | cout << "output_file should be a .root file!\n";
|
---|
144 | exit(1);
|
---|
145 | }
|
---|
146 | //create output log-file name
|
---|
147 | string forLog = outputfilename;
|
---|
148 | string LogName = forLog.erase(forLog.find(".root"));
|
---|
149 | LogName = LogName+"_run.log";
|
---|
150 |
|
---|
151 | TFile *outputFile = TFile::Open(outputfilename.c_str(), "RECREATE"); // Creates the file, but should be closed just after
|
---|
152 | outputFile->Close();
|
---|
153 |
|
---|
154 | string line;
|
---|
155 | ifstream infile(inputFileList.c_str());
|
---|
156 | infile >> line; // the first line determines the type of input files
|
---|
157 |
|
---|
158 | //read the datacard input file
|
---|
159 | string DetDatacard("");
|
---|
160 | if(argc>=4)DetDatacard =argv[3];
|
---|
161 |
|
---|
162 | //Smearing information
|
---|
163 | RESOLution *DET = new RESOLution();
|
---|
164 | cout <<"** **"<< endl;
|
---|
165 | cout <<"** ####### Start reading DETECTOR parameters ####### **"<< endl;
|
---|
166 | cout << left << setw(40) <<"** Opening configuration card: "<<""
|
---|
167 | << left << setw(20) << DetDatacard <<""
|
---|
168 | << right << setw(9) <<" **"<<""<<endl;
|
---|
169 | DET->ReadDataCard(DetDatacard);
|
---|
170 | DET->Logfile(LogName);
|
---|
171 | cout << left << setw(42) <<"** Parameters summarised in file: "<<""
|
---|
172 | << left << setw(20) << LogName <<""
|
---|
173 | << right << setw(7) <<"**"<<""<<endl;
|
---|
174 | cout <<"** **"<< endl;
|
---|
175 |
|
---|
176 | //read the trigger input file
|
---|
177 | string TrigDatacard("data/trigger.dat");
|
---|
178 | if(argc==5) TrigDatacard =argv[4];
|
---|
179 |
|
---|
180 | //Trigger information
|
---|
181 | TriggerTable *TRIGT = new TriggerTable();
|
---|
182 | TRIGT->TriggerCardReader(TrigDatacard.c_str());
|
---|
183 | TRIGT->PrintTriggerTable(LogName);
|
---|
184 | if(DET->FLAG_trigger == 1)
|
---|
185 | {
|
---|
186 | cout <<"** ########### Start reading TRIGGER card ########## **"<< endl;
|
---|
187 | cout << left << setw(40) <<"** Opening configuration card: "<<""
|
---|
188 | << left << setw(20) << TrigDatacard <<""
|
---|
189 | << right << setw(9) <<" **"<<""<<endl;
|
---|
190 | cout <<"** **"<< endl;
|
---|
191 | }
|
---|
192 |
|
---|
193 | //Propagation of tracks in the B field
|
---|
194 | TrackPropagation *TRACP = new TrackPropagation(DET);
|
---|
195 | //TrackPropagation *TRACP = new TrackPropagation(DetDatacard);
|
---|
196 |
|
---|
197 | //Jet information
|
---|
198 | JetsUtil *JETRUN = new JetsUtil(DET);
|
---|
199 | //JetsUtil *JETRUN = new JetsUtil(DetDatacard);
|
---|
200 |
|
---|
201 | //VFD information
|
---|
202 | VeryForward * VFD = new VeryForward(DET);
|
---|
203 | //VeryForward * VFD = new VeryForward(DetDatacard);
|
---|
204 |
|
---|
205 | // data converters
|
---|
206 | DataConverter *converter=NULL;
|
---|
207 | cout <<"** **"<<endl;
|
---|
208 | cout <<"** ####### Start convertion to TRoot format ######## **"<< endl;
|
---|
209 |
|
---|
210 | if(strstr(line.c_str(),".hep"))
|
---|
211 | {
|
---|
212 | cout <<"** StdHEP file format detected **"<<endl;
|
---|
213 | cout <<"** This can take several minutes **"<< endl;
|
---|
214 | converter = new STDHEPConverter(inputFileList,outputfilename);//case ntpl file in input list
|
---|
215 | }
|
---|
216 | else if(strstr(line.c_str(),".lhe"))
|
---|
217 | {
|
---|
218 | cout <<"** LHEF file format detected **"<<endl;
|
---|
219 | cout <<"** This can take several minutes **"<< endl;
|
---|
220 | converter = new LHEFConverter(inputFileList,outputfilename);//case ntpl file in input list
|
---|
221 | }
|
---|
222 | else if(strstr(line.c_str(),".root"))
|
---|
223 | {
|
---|
224 | cout <<"** h2root file format detected **"<<endl;
|
---|
225 | cout <<"** This can take several minutes **"<< endl;
|
---|
226 | converter = new HEPEVTConverter(inputFileList,outputfilename);//case ntpl file in input list
|
---|
227 | }
|
---|
228 | else {
|
---|
229 | cout << left << setw(4) <<"** "<<""
|
---|
230 | << left << setw(63) << line.c_str() <<""
|
---|
231 | << right << setw(2) <<"**"<<endl;
|
---|
232 | cout <<"** File format not identified -- Exiting... **"<< endl;
|
---|
233 | cout <<"** **"<< endl;
|
---|
234 | cout <<"*********************************************************************"<< endl;
|
---|
235 | return -1;};
|
---|
236 | cout <<"** Exiting conversion... **"<< endl;
|
---|
237 |
|
---|
238 | TChain chain("GEN");
|
---|
239 | chain.Add(outputfilename.c_str());
|
---|
240 | ExRootTreeReader *treeReader = new ExRootTreeReader(&chain);
|
---|
241 | const TClonesArray *branchGen = treeReader->UseBranch("Particle");
|
---|
242 | TIter itGen((TCollection*)branchGen);
|
---|
243 |
|
---|
244 | //Output file : contents of the analysis object data
|
---|
245 | ExRootTreeWriter *treeWriter = new ExRootTreeWriter(outputfilename, "Analysis");
|
---|
246 | ExRootTreeBranch *branchJet = treeWriter->NewBranch("Jet", TRootJet::Class());
|
---|
247 | ExRootTreeBranch *branchTauJet = treeWriter->NewBranch("TauJet", TRootTauJet::Class());
|
---|
248 | ExRootTreeBranch *branchElectron = treeWriter->NewBranch("Electron", TRootElectron::Class());
|
---|
249 | ExRootTreeBranch *branchMuon = treeWriter->NewBranch("Muon", TRootMuon::Class());
|
---|
250 | ExRootTreeBranch *branchPhoton = treeWriter->NewBranch("Photon", TRootPhoton::Class());
|
---|
251 | ExRootTreeBranch *branchTracks = treeWriter->NewBranch("Tracks", TRootTracks::Class());
|
---|
252 | ExRootTreeBranch *branchETmis = treeWriter->NewBranch("ETmis", TRootETmis::Class());
|
---|
253 | ExRootTreeBranch *branchCalo = treeWriter->NewBranch("CaloTower", TRootCalo::Class());
|
---|
254 | ExRootTreeBranch *branchZDC = treeWriter->NewBranch("ZDChits", TRootZdcHits::Class());
|
---|
255 | ExRootTreeBranch *branchRP220 = treeWriter->NewBranch("RP220hits", TRootRomanPotHits::Class());
|
---|
256 | ExRootTreeBranch *branchFP420 = treeWriter->NewBranch("FP420hits", TRootRomanPotHits::Class());
|
---|
257 |
|
---|
258 | TRootGenParticle *particle;
|
---|
259 | TRootETmis *elementEtmis;
|
---|
260 | TRootElectron *elementElec;
|
---|
261 | TRootMuon *elementMu;
|
---|
262 | TRootPhoton *elementPhoton;
|
---|
263 | TRootTracks *elementTracks;
|
---|
264 | TRootCalo *elementCalo;
|
---|
265 |
|
---|
266 | TLorentzVector genMomentum(0,0,0,0); // four-momentum at the vertex
|
---|
267 | TLorentzVector genMomentumBfield(0,0,0,0); // four-momentum at the exit of the tracks
|
---|
268 | TLorentzVector momentumCaloSegmentation(0,0,0,0); // four-momentum in the calo, after applying the calo segmentation
|
---|
269 | LorentzVector jetMomentum;
|
---|
270 |
|
---|
271 | vector<fastjet::PseudoJet> input_particles;//for FastJet algorithm
|
---|
272 | vector<fastjet::PseudoJet> sorted_jets;
|
---|
273 | vector<TLorentzVector> TrackCentral;
|
---|
274 | vector<PhysicsTower> towers;
|
---|
275 | vector<ParticleUtil> electron;
|
---|
276 | vector<ParticleUtil> muon;
|
---|
277 | vector<ParticleUtil> gamma;
|
---|
278 |
|
---|
279 | TSimpleArray<TRootGenParticle> NFCentralQ;
|
---|
280 | float iPhi=0,iEta=0;
|
---|
281 |
|
---|
282 |
|
---|
283 |
|
---|
284 | // 2. ********** Loop over all events ***********
|
---|
285 | Long64_t entry, allEntries = treeReader->GetEntries();
|
---|
286 | cout <<"** **"<<endl;
|
---|
287 | cout <<"** ####### Start fast detector simulation ######## **"<< endl;
|
---|
288 | cout << left << setw(52) <<"** Total number of events to run: "<<""
|
---|
289 | << left << setw(15) << allEntries <<""
|
---|
290 | << right << setw(2) <<"**"<<endl;
|
---|
291 |
|
---|
292 | loopwatch.Start();
|
---|
293 |
|
---|
294 | // loop on all events
|
---|
295 | for(entry = 0; entry < allEntries; ++entry)
|
---|
296 | {
|
---|
297 | TLorentzVector PTmis(0,0,0,0);
|
---|
298 | treeReader->ReadEntry(entry);
|
---|
299 | treeWriter->Clear();
|
---|
300 | if((entry % 100) == 0 && entry > 0 ) {
|
---|
301 | cout << left << setw(52) <<"** Processing element # "<<""
|
---|
302 | << left << setw(15) << entry <<""
|
---|
303 | << right << setw(2) <<"**"<<endl;
|
---|
304 | }
|
---|
305 |
|
---|
306 | electron.clear();
|
---|
307 | muon.clear();
|
---|
308 | gamma.clear();
|
---|
309 | NFCentralQ.Clear();
|
---|
310 |
|
---|
311 | TrackCentral.clear();
|
---|
312 | towers.clear();
|
---|
313 | input_particles.clear();
|
---|
314 |
|
---|
315 | // 2.1 Loop over all particles in event
|
---|
316 | itGen.Reset();
|
---|
317 | while( (particle = (TRootGenParticle*) itGen.Next()) ) {
|
---|
318 | int pid = abs(particle->PID);
|
---|
319 |
|
---|
320 |
|
---|
321 | // 2.1.1********************* preparation for the b-tagging
|
---|
322 | //// This subarray is needed for the B-jet algorithm
|
---|
323 | // optimization for speed : put first PID condition, then ETA condition, then either pt or status
|
---|
324 | if( (pid <= pB || pid == pGLUON) &&// is it a light quark or a gluon, i.e. is it one of these : u,d,c,s,b,g ?
|
---|
325 | fabs(particle->Eta) < DET->CEN_max_tracker &&
|
---|
326 | particle->Status != 1 &&
|
---|
327 | particle->PT > DET->PT_QUARKS_MIN ) {
|
---|
328 | NFCentralQ.Add(particle);
|
---|
329 | }
|
---|
330 |
|
---|
331 | // 2.1.2 ********************* central detector: keep only visible particles
|
---|
332 | // keeps only final particles, visible by the central detector, including the fiducial volume
|
---|
333 | // the ordering of conditions have been optimised for speed : put first the STATUS condition
|
---|
334 | if( (particle->Status == 1) &&
|
---|
335 | ((pid != pNU1) && (pid != pNU2) && (pid != pNU3)) &&
|
---|
336 | (fabs(particle->Eta) < DET->CEN_max_calo_fwd)
|
---|
337 | )
|
---|
338 | {
|
---|
339 | genMomentum.SetPxPyPzE(particle->Px, particle->Py, particle->Pz, particle->E);
|
---|
340 | genMomentumBfield = genMomentum;
|
---|
341 |
|
---|
342 | // 2.1.2.1 ********************* central detector: magnetic field
|
---|
343 | // genMomentumBfield is then changed with respect to the magnetic field
|
---|
344 | if(DET->FLAG_bfield==1) TRACP->Propagation(particle,genMomentumBfield);
|
---|
345 | float eta=fabs(genMomentumBfield.Eta());
|
---|
346 |
|
---|
347 |
|
---|
348 | // 2.1.2.2 ********************* central detector: smearing (calorimeters, muon chambers)
|
---|
349 | switch(pid) {
|
---|
350 |
|
---|
351 | case pE: // all electrons with eta < DET->MAX_CALO_FWD
|
---|
352 | DET->SmearElectron(genMomentumBfield);
|
---|
353 | if(genMomentumBfield.E()!=0 && eta < DET->CEN_max_tracker && genMomentumBfield.Pt() > DET->PTCUT_elec){
|
---|
354 | electron.push_back(ParticleUtil(genMomentumBfield,particle->PID));
|
---|
355 | }
|
---|
356 | break; // case pE
|
---|
357 | case pGAMMA: // all photons with eta < DET->MAX_CALO_FWD
|
---|
358 | DET->SmearElectron(genMomentumBfield);
|
---|
359 | if(genMomentumBfield.E()!=0 && eta < DET->CEN_max_tracker && genMomentumBfield.Pt() > DET->PTCUT_gamma) {
|
---|
360 | gamma.push_back(ParticleUtil(genMomentumBfield,particle->PID));
|
---|
361 | }
|
---|
362 | break; // case pGAMMA
|
---|
363 | case pMU: // all muons with eta < DET->MAX_MU
|
---|
364 | DET->SmearMu(genMomentumBfield);
|
---|
365 | if(genMomentumBfield.E()!=0 && eta < DET->CEN_max_mu && genMomentumBfield.Pt() > DET->PTCUT_muon){
|
---|
366 | muon.push_back(ParticleUtil(genMomentumBfield,particle->PID));
|
---|
367 | }
|
---|
368 | break; // case pMU
|
---|
369 | case pLAMBDA: // all lambdas with eta < DET->MAX_CALO_FWD
|
---|
370 | case pK0S: // all K0s with eta < DET->MAX_CALO_FWD
|
---|
371 | DET->SmearHadron(genMomentumBfield, 0.7);
|
---|
372 | break; // case hadron
|
---|
373 | default: // all other final particles with eta < DET->MAX_CALO_FWD
|
---|
374 | DET->SmearHadron(genMomentumBfield, 1.0);
|
---|
375 | break;
|
---|
376 | } // 2.1.2.2 switch (pid)
|
---|
377 |
|
---|
378 |
|
---|
379 | // 2.1.2.3 ********************* central detector: calotowers
|
---|
380 | // all final particles but muons and neutrinos
|
---|
381 | // for calorimetric towers and missing PT
|
---|
382 | int charge=Charge(pid);
|
---|
383 | if(genMomentumBfield.E() !=0 && pid != pMU) {
|
---|
384 | // in case the Bfield is not simulated, checks that charged particles have enough pt to reach the calos
|
---|
385 | if ( !DET->FLAG_bfield && charge!=0 && genMomentumBfield.Pt() <= DET->TRACK_ptmin ) { /* particules do not reach calos */ }
|
---|
386 | else { // particles reach calos
|
---|
387 | // applies the calo segmentation and returns iEta & iPhi
|
---|
388 | DET->BinEtaPhi(genMomentumBfield.Phi(), genMomentumBfield.Eta(), iPhi, iEta);
|
---|
389 | if(iEta != -100 && iPhi != -100) {
|
---|
390 | momentumCaloSegmentation.SetPtEtaPhiE(genMomentumBfield.Pt(),iEta,iPhi,genMomentumBfield.E());
|
---|
391 | elementCalo = (TRootCalo*) branchCalo->NewEntry();
|
---|
392 | elementCalo->Set(momentumCaloSegmentation);
|
---|
393 | PhysicsTower Tower(LorentzVector(momentumCaloSegmentation.Px(),momentumCaloSegmentation.Py(),momentumCaloSegmentation.Pz(),momentumCaloSegmentation.E()));
|
---|
394 | towers.push_back(Tower);
|
---|
395 | } // if iEta != -100
|
---|
396 | } // else : when particles reach the calos
|
---|
397 | } // 2.1.2.3 calotowers
|
---|
398 |
|
---|
399 |
|
---|
400 | // 2.1.2.4 ********************* central detector: tracks
|
---|
401 | // all final charged particles
|
---|
402 | if(
|
---|
403 | (genMomentumBfield.E()!=0) &&
|
---|
404 | (fabs(genMomentumBfield.Eta()) < DET->CEN_max_tracker) &&
|
---|
405 | (DET->FLAG_bfield || ( !DET->FLAG_bfield && genMomentumBfield.Pt() > DET->TRACK_ptmin )) &&
|
---|
406 | // if bfield not simulated, pt should be high enough to be taken into account
|
---|
407 | ((rand()%100) < DET->TRACK_eff) &&
|
---|
408 | (charge!=0)
|
---|
409 | )
|
---|
410 | {
|
---|
411 | elementTracks = (TRootTracks*) branchTracks->NewEntry();
|
---|
412 | elementTracks->Set(genMomentum); // fills px,py,pz,pt,e,eta,phi at vertex
|
---|
413 | elementTracks->Etaout = genMomentumBfield.Eta();
|
---|
414 | elementTracks->Phiout = genMomentumBfield.Phi();
|
---|
415 | // TODO!!! apply a smearing on the position of the origin of the track
|
---|
416 | // elementTracks->SetPosition(particle->X,particle->Y,particle->Z);
|
---|
417 | // uses the output of the bfield computation : Xout=... Yout=... Zout...
|
---|
418 | // TODO!!! elementTrakcs->SetPositionOut(Xout,Yout,Zout);
|
---|
419 | TrackCentral.push_back(genMomentum); // tracks at vertex!
|
---|
420 | } // 2.1.2.4 tracks
|
---|
421 |
|
---|
422 | } // 2.1.2 central detector
|
---|
423 |
|
---|
424 | // 2.1.3 ********************* forward detectors: zdc
|
---|
425 | if(DET->FLAG_vfd==1) {
|
---|
426 | VFD->ZDC(treeWriter,branchZDC,particle);
|
---|
427 | VFD->RomanPots(treeWriter,branchRP220,branchFP420,particle);
|
---|
428 | }
|
---|
429 |
|
---|
430 | } // 2.1 while : loop on all particles of the event.
|
---|
431 |
|
---|
432 |
|
---|
433 |
|
---|
434 | // 2.2 ********** Output preparation & complex objects ***********
|
---|
435 |
|
---|
436 | // 2.2.1 ********************* sorting collections by decreasing pt
|
---|
437 | DET->SortedVector(electron);
|
---|
438 | for(unsigned int i=0; i < electron.size(); i++) {
|
---|
439 | elementElec = (TRootElectron*) branchElectron->NewEntry();
|
---|
440 | elementElec->Set(electron[i].Px(),electron[i].Py(),electron[i].Pz(),electron[i].E());
|
---|
441 | elementElec->Charge = sign(electron[i].PID());
|
---|
442 | elementElec->IsolFlag = DET->Isolation(electron[i].Phi(),electron[i].Eta(),TrackCentral,2.0);//isolation based on tracks
|
---|
443 | }
|
---|
444 | DET->SortedVector(muon);
|
---|
445 | for(unsigned int i=0; i < muon.size(); i++) {
|
---|
446 | elementMu = (TRootMuon*) branchMuon->NewEntry();
|
---|
447 | elementMu->Charge = sign(muon[i].PID());
|
---|
448 | elementMu->Set(muon[i].Px(),muon[i].Py(),muon[i].Pz(),muon[i].E());
|
---|
449 | elementMu->IsolFlag = DET->Isolation(muon[i].Phi(),muon[i].Eta(),TrackCentral,2.0);
|
---|
450 | }
|
---|
451 | DET->SortedVector(gamma);
|
---|
452 | for(unsigned int i=0; i < gamma.size(); i++) {
|
---|
453 | elementPhoton = (TRootPhoton*) branchPhoton->NewEntry();
|
---|
454 | elementPhoton->Set(gamma[i].Px(),gamma[i].Py(),gamma[i].Pz(),gamma[i].E());
|
---|
455 | }
|
---|
456 |
|
---|
457 | // 2.2.2 ************* computes the Missing Transverse Momentum
|
---|
458 | TLorentzVector Att(0.,0.,0.,0.);
|
---|
459 | for(unsigned int i=0; i < towers.size(); i++)
|
---|
460 | {
|
---|
461 | Att.SetPxPyPzE(towers[i].fourVector.px, towers[i].fourVector.py, towers[i].fourVector.pz, towers[i].fourVector.E);
|
---|
462 | if(fabs(Att.Eta()) < DET->CEN_max_calo_fwd)
|
---|
463 | {
|
---|
464 | PTmis = PTmis + Att;
|
---|
465 | // create a fastjet::PseudoJet with these components and put it onto
|
---|
466 | // back of the input_particles vector
|
---|
467 | input_particles.push_back(fastjet::PseudoJet(towers[i].fourVector.px,towers[i].fourVector.py,towers[i].fourVector.pz,towers[i].fourVector.E));
|
---|
468 | }
|
---|
469 | }
|
---|
470 | elementEtmis = (TRootETmis*) branchETmis->NewEntry();
|
---|
471 | elementEtmis->ET = (PTmis).Pt();
|
---|
472 | elementEtmis->Phi = (-PTmis).Phi();
|
---|
473 | elementEtmis->Px = (-PTmis).Px();
|
---|
474 | elementEtmis->Py = (-PTmis).Py();
|
---|
475 |
|
---|
476 | // 2.2.3 ************* B-tag, tau jets
|
---|
477 | sorted_jets=JETRUN->RunJets(input_particles);
|
---|
478 | JETRUN->RunJetBtagging(treeWriter, branchJet,sorted_jets,NFCentralQ);
|
---|
479 | JETRUN->RunTauJets(treeWriter,branchTauJet,sorted_jets,towers, TrackCentral);
|
---|
480 |
|
---|
481 | treeWriter->Fill();
|
---|
482 | } // 2. Loop over all events ('for' loop)
|
---|
483 |
|
---|
484 | cout <<"** **"<< endl;
|
---|
485 | cout <<"** Exiting detector simulation... **"<< endl;
|
---|
486 |
|
---|
487 |
|
---|
488 | treeWriter->Write();
|
---|
489 | delete treeWriter;
|
---|
490 | loopwatch.Stop();
|
---|
491 |
|
---|
492 |
|
---|
493 |
|
---|
494 | // 3. ********** Trigger & Frog ***********
|
---|
495 | // 3.1 ************ running the trigger in case the FLAG trigger is put to 1 in the datacard
|
---|
496 | triggerwatch.Start();
|
---|
497 | if(DET->FLAG_trigger == 1)
|
---|
498 | {
|
---|
499 | cout <<"** **"<<endl;
|
---|
500 | cout <<"** ########### Start Trigger selection ########### **"<< endl;
|
---|
501 |
|
---|
502 | // input
|
---|
503 | TChain chainT("Analysis");
|
---|
504 | chainT.Add(outputfilename.c_str());
|
---|
505 | ExRootTreeReader *treeReaderT = new ExRootTreeReader(&chainT);
|
---|
506 |
|
---|
507 | // output
|
---|
508 | TClonesArray *branchElecTrig = treeReaderT->UseBranch("Electron");
|
---|
509 | TClonesArray *branchMuonTrig = treeReaderT->UseBranch("Muon");
|
---|
510 | TClonesArray *branchJetTrig = treeReaderT->UseBranch("Jet");
|
---|
511 | TClonesArray *branchTauJetTrig = treeReaderT->UseBranch("TauJet");
|
---|
512 | TClonesArray *branchPhotonTrig = treeReaderT->UseBranch("Photon");
|
---|
513 | TClonesArray *branchETmisTrig = treeReaderT->UseBranch("ETmis");
|
---|
514 |
|
---|
515 | ExRootTreeWriter *treeWriterT = new ExRootTreeWriter(outputfilename, "Trigger");
|
---|
516 | ExRootTreeBranch *branchTrigger = treeWriterT->NewBranch("TrigResult", TRootTrigger::Class());
|
---|
517 |
|
---|
518 |
|
---|
519 | Long64_t entryT, allEntriesT = treeReaderT->GetEntries();
|
---|
520 | // loop on all entries
|
---|
521 | for(entryT = 0; entryT < allEntriesT; ++entryT) {
|
---|
522 | treeWriterT->Clear();
|
---|
523 | treeReaderT->ReadEntry(entryT);
|
---|
524 | TRIGT->GetGlobalResult(branchElecTrig, branchMuonTrig,branchJetTrig, branchTauJetTrig,branchPhotonTrig, branchETmisTrig,branchTrigger);
|
---|
525 | treeWriterT->Fill();
|
---|
526 | } // loop on all entries
|
---|
527 | cout <<"** Exiting trigger simulation... **"<< endl;
|
---|
528 |
|
---|
529 | treeWriterT->Write();
|
---|
530 | delete treeWriterT;
|
---|
531 | delete treeReaderT;
|
---|
532 | } // trigger
|
---|
533 | triggerwatch.Stop();
|
---|
534 |
|
---|
535 |
|
---|
536 | // 3.2 ************** FROG display
|
---|
537 | frogwatch.Start();
|
---|
538 | if(DET->FLAG_frog == 1) {
|
---|
539 | cout <<"** **"<<endl;
|
---|
540 | cout <<"** ################## Start FROG ################# **"<< endl;
|
---|
541 |
|
---|
542 | FrogDisplay *FROG = new FrogDisplay(DET);
|
---|
543 | FROG->BuildEvents(outputfilename);
|
---|
544 | FROG->BuildGeom();
|
---|
545 | delete FROG;
|
---|
546 | }
|
---|
547 | frogwatch.Stop();
|
---|
548 |
|
---|
549 |
|
---|
550 |
|
---|
551 |
|
---|
552 | // 4. ********** End & Exit ***********
|
---|
553 |
|
---|
554 | globalwatch.Stop();
|
---|
555 | cout <<"** **"<< endl;
|
---|
556 | cout <<"** ################## Time report ################# **"<< endl;
|
---|
557 | cout << left << setw(32) <<"** Time report for "<<""
|
---|
558 | << left << setw(15) << allEntries <<""
|
---|
559 | << right << setw(22) <<"events **"<<endl;
|
---|
560 | cout <<"** **"<< endl;
|
---|
561 | cout << left << setw(10) <<"**"<<""
|
---|
562 | << left << setw(15) <<"Time (s):"<<""
|
---|
563 | << right << setw(15) <<"CPU"<<""
|
---|
564 | << right << setw(15) <<"Real"<<""
|
---|
565 | << right << setw(14) <<"**"<<endl;
|
---|
566 | cout << left << setw(10) <<"**"<<""
|
---|
567 | << left << setw(15) <<" + Global:"<<""
|
---|
568 | << right << setw(15) <<globalwatch.CpuTime()<<""
|
---|
569 | << right << setw(15) <<globalwatch.RealTime()<<""
|
---|
570 | << right << setw(14) <<"**"<<endl;
|
---|
571 | cout << left << setw(10) <<"**"<<""
|
---|
572 | << left << setw(15) <<" + Events:"<<""
|
---|
573 | << right << setw(15) <<loopwatch.CpuTime()<<""
|
---|
574 | << right << setw(15) <<loopwatch.RealTime()<<""
|
---|
575 | << right << setw(14) <<"**"<<endl;
|
---|
576 | if(DET->FLAG_trigger == 1)
|
---|
577 | {
|
---|
578 | cout << left << setw(10) <<"**"<<""
|
---|
579 | << left << setw(15) <<" + Trigger:"<<""
|
---|
580 | << right << setw(15) <<triggerwatch.CpuTime()<<""
|
---|
581 | << right << setw(15) <<triggerwatch.RealTime()<<""
|
---|
582 | << right << setw(14) <<"**"<<endl;
|
---|
583 | }
|
---|
584 | if(DET->FLAG_frog == 1)
|
---|
585 | {
|
---|
586 | cout << left << setw(10) <<"**"<<""
|
---|
587 | << left << setw(15) <<" + Frog:"<<""
|
---|
588 | << right << setw(15) <<frogwatch.CpuTime()<<""
|
---|
589 | << right << setw(15) <<frogwatch.RealTime()<<""
|
---|
590 | << right << setw(14) <<"**"<<endl;
|
---|
591 |
|
---|
592 | }
|
---|
593 |
|
---|
594 | cout <<"** **"<< endl;
|
---|
595 | cout <<"** Exiting Delphes ... **"<< endl;
|
---|
596 | cout <<"** **"<< endl;
|
---|
597 | cout <<"*********************************************************************"<< endl;
|
---|
598 | cout <<"*********************************************************************"<< endl;
|
---|
599 |
|
---|
600 |
|
---|
601 | delete treeReader;
|
---|
602 | delete DET;
|
---|
603 | delete TRIGT;
|
---|
604 | delete TRACP;
|
---|
605 | delete JETRUN;
|
---|
606 | delete VFD;
|
---|
607 | delete converter;
|
---|
608 |
|
---|
609 | // todo("TODO");
|
---|
610 | }
|
---|