Replace the PHOS version from IHEP to Run1 and fix the compilation, when possibletest...
[u/mrichter/AliRoot.git] / test / embedding / Config.C
CommitLineData
2858736f 1// One can use the configuration macro in compiled mode by
2// root [0] gSystem->Load("libgeant321");
5889a25e 3// root [1] gSystem->Load("libpythia6.4.25.so");
4// root [2] gSystem->Load("libqpythia.so");
5// root [3] gSystem->SetIncludePath("-I$ROOTSYS/include -I$ALICE_ROOT/include\
6// -I$ALICE/geant3/TGeant3");
7// root [4] AliSimulation sim
8// root [5] sim.SetConfigFile("Config.C++")
9// root [6] sim.Run()
2858736f 10
11#if !defined(__CINT__) || defined(__MAKECINT__)
12#include <Riostream.h>
13#include <TRandom.h>
14#include <TDatime.h>
15#include <TSystem.h>
16#include <TVirtualMC.h>
17#include <TGeant3TGeo.h>
5889a25e 18#include <TGeoGlobalMagField.h>
19#include "AliRunLoader.h"
20#include "AliRun.h"
21#include "AliConfig.h"
22#include "AliDecayerPythia.h"
23#include "AliGenPythia.h"
24#include "AliGenDPMjet.h"
25#include "AliMagF.h"
26#include "AliBODY.h"
27#include "AliMAG.h"
28#include "AliABSOv3.h"
29#include "AliDIPOv3.h"
30#include "AliHALLv3.h"
31#include "AliFRAMEv2.h"
32#include "AliSHILv3.h"
33#include "AliPIPEv3.h"
34#include "AliITSv11.h"
35#include "AliTPCv2.h"
36#include "AliTOFv6T0.h"
37#include "AliHMPIDv3.h"
38#include "AliZDCv4.h"
39#include "AliTRDv1.h"
40#include "AliTRDgeometry.h"
41#include "AliFMDv1.h"
42#include "AliMUONv1.h"
43#include "AliPHOSv1.h"
44#include "AliPMDv1.h"
45#include "AliT0v1.h"
46#include "AliEMCALv2.h"
47#include "AliACORDEv1.h"
48#include "AliVZEROv7.h"
49#include "AliSimulation.h"
50#include "AliGenCocktail.h"
51#include "AliGenTherminator.h"
52#include "AliGenerator.h"
2858736f 53#endif
54
55
56enum PDC06Proc_t
57{
58 kPythia6, kPhojet, kRunMax
59};
60
61const char * pprRunName[] = {
62 "kPythia6", "kPhojet"
63};
64
65//--- Magnetic Field ---
66enum Mag_t
67{
68 kNoField, k5kG, kFieldMax
69};
70
71const char * pprField[] = {
72 "kNoField", "k5kG"
73};
74
75//--- Embedding run ---
76enum EmbedRun_t
77{
78 kBackground, kMerged, kSignal, kEmbRunMax
79};
80
81const char *embedRun[] = {
82 "kBackground", "kMerged", "kSignal"
83};
84
85//--- Functions ---
86class AliGenPythia;
87AliGenerator *MbPythia();
88AliGenerator *MbPhojet();
89void ProcessEnvironmentVars();
90
91// Geterator, field, beam energy
92static PDC06Proc_t proc = kPhojet;
93static Mag_t mag = k5kG;
94static Float_t energy = 10000; // energy in CMS
95static EmbedRun_t embedrun = kBackground;
96//========================//
97// Set Random Number seed //
98//========================//
99TDatime dt;
100static UInt_t seed = dt.Get();
101
102// Comment line
103static TString comment;
104
105Float_t EtaToTheta(Float_t arg);
106
107void Config()
108{
109
110
111 // Get settings from environment variables
112 ProcessEnvironmentVars();
113
114 gRandom->SetSeed(seed);
115 cerr<<"Seed for random number generation= "<<seed<<endl;
116
117 // Libraries required by geant321
118#if defined(__CINT__)
119 gSystem->Load("liblhapdf"); // Parton density functions
120 gSystem->Load("libEGPythia6"); // TGenerator interface
121 gSystem->Load("libpythia6"); // Pythia
122 gSystem->Load("libAliPythia6"); // ALICE specific implementations
123 gSystem->Load("libgeant321");
124 gSystem->Load("libTTherminator");
125#endif
126
127 new TGeant3TGeo("C++ Interface to Geant3");
128
129 //=======================================================================
130 // Create the output file
131
132
133 AliRunLoader* rl=0x0;
134
135 cout<<"Config.C: Creating Run Loader ..."<<endl;
136 rl = AliRunLoader::Open("galice.root",
137 AliConfig::GetDefaultEventFolderName(),
138 "recreate");
139 if (rl == 0x0)
140 {
141 gAlice->Fatal("Config.C","Can not instatiate the Run Loader");
142 return;
143 }
144 rl->SetCompressionLevel(2);
145 rl->SetNumberOfEventsPerFile(3);
146 gAlice->SetRunLoader(rl);
147
148
149 // Set the trigger configuration
150 if ((embedrun == kBackground) || (embedrun == kMerged)) {
28da60d3 151 AliSimulation::Instance()->SetTriggerConfig("Pb-Pb");
2858736f 152 cout<<"Trigger configuration is set to Pb-Pb"<<endl;
153 }
154 else {
155 // Set the trigger configuration: proton-proton
28da60d3 156 AliSimulation::Instance()->SetTriggerConfig("p-p");
2858736f 157 }
158
2858736f 159 //=======================================================================
160 // ************* STEERING parameters FOR ALICE SIMULATION **************
161 // --- Specify event type to be tracked through the ALICE setup
162 // --- All positions are in cm, angles in degrees, and P and E in GeV
163
164
165 gMC->SetProcess("DCAY",1);
166 gMC->SetProcess("PAIR",1);
167 gMC->SetProcess("COMP",1);
168 gMC->SetProcess("PHOT",1);
169 gMC->SetProcess("PFIS",0);
170 gMC->SetProcess("DRAY",0);
171 gMC->SetProcess("ANNI",1);
172 gMC->SetProcess("BREM",1);
173 gMC->SetProcess("MUNU",1);
174 gMC->SetProcess("CKOV",1);
175 gMC->SetProcess("HADR",1);
176 gMC->SetProcess("LOSS",2);
177 gMC->SetProcess("MULS",1);
178 gMC->SetProcess("RAYL",1);
179
180 Float_t cut = 1.e-3; // 1MeV cut by default
181 Float_t tofmax = 1.e10;
182
183 gMC->SetCut("CUTGAM", cut);
184 gMC->SetCut("CUTELE", cut);
185 gMC->SetCut("CUTNEU", cut);
186 gMC->SetCut("CUTHAD", cut);
187 gMC->SetCut("CUTMUO", cut);
188 gMC->SetCut("BCUTE", cut);
189 gMC->SetCut("BCUTM", cut);
190 gMC->SetCut("DCUTE", cut);
191 gMC->SetCut("DCUTM", cut);
192 gMC->SetCut("PPCUTM", cut);
193 gMC->SetCut("TOFMAX", tofmax);
194
195 //======================//
196 // Set External decayer //
197 //======================//
198 TVirtualMCDecayer* decayer = new AliDecayerPythia();
199 decayer->SetForceDecay(kAll);
200 decayer->Init();
201 gMC->SetExternalDecayer(decayer);
202
5889a25e 203 AliGenerator* gener = 0x0;
2858736f 204 if ((embedrun == kMerged) || (embedrun == kSignal)) {
205 //=========================//
206 // Generator Configuration //
207 //=========================//
2858736f 208
209 if (proc == kPythia6) {
210 gener = MbPythia();
211 } else if (proc == kPhojet) {
212 gener = MbPhojet();
213 }
214 }
215 else {
216 AliGenCocktail *gener = new AliGenCocktail();
217 gener->SetPhiRange(0, 360);
218 // Set pseudorapidity range from -8 to 8.
219 Float_t thmin = EtaToTheta(1); // theta min. <---> eta max
220 Float_t thmax = EtaToTheta(-1); // theta max. <---> eta min
221 gener->SetThetaRange(thmin,thmax);
ae8c1d56 222 gener->SetProjectile("A",208,82);
223 gener->SetTarget("A",208,82);
2858736f 224
225 AliGenTherminator *genther = new AliGenTherminator();
226 genther->SetFileName("event.out");
227 genther->SetEventNumberInFile(1);
228 genther->SetTemperature(0.145);
229 genther->SetMiuI(-0.0009);
230 genther->SetMiuS(0.000);
231 genther->SetMiuB(0.0008);
232 genther->SetAlfaRange(8.0);
233 genther->SetRapRange(4.0);
234 genther->SetRhoMax(7.74);
235 genther->SetTau(9.74);
236 genther->SetModel("Lhyquid3D");
237 genther->SetLhyquidSet("LHC500C2030");
238
239 gener->AddGenerator(genther, "THERMINATOR LHYQUID3D", 1);
240 }
241
242
243
244 // PRIMARY VERTEX
245 //
246 gener->SetOrigin(0., 0., 0.); // vertex position
247 //
248 //
249 // Size of the interaction diamond
250 // Longitudinal
251 Float_t sigmaz;
252
253 if (embedrun == kBackground) {
254 sigmaz = 7.55 / TMath::Sqrt(2.); // [cm]
255 }
256 else {
257 Float_t sigmaz = 5.4 / TMath::Sqrt(2.); // [cm]
258 if (energy == 900)
259 sigmaz = 10.5 / TMath::Sqrt(2.); // [cm]
260 }
261
262 //
263 // Transverse
264 Float_t betast = 10; // beta* [m]
265 Float_t eps = 3.75e-6; // emittance [m]
266 Float_t gamma = energy / 2.0 / 0.938272; // relativistic gamma [1]
267 Float_t sigmaxy = TMath::Sqrt(eps * betast / gamma) / TMath::Sqrt(2.) * 100.; // [cm]
268 printf("\n \n Diamond size x-y: %10.3e z: %10.3e\n \n", sigmaxy, sigmaz);
269
270 gener->SetSigma(sigmaxy, sigmaxy, sigmaz); // Sigma in (X,Y,Z) (cm) on IP position
271 gener->SetCutVertexZ(3.); // Truncate at 3 sigma
272 gener->SetVertexSmear(kPerEvent);
273
274 gener->Init();
275
276 // FIELD
277 //
ae8c1d56 278// Field
279
280 // AliMagF* field = 0x0;
2858736f 281 if (mag == kNoField) {
282 comment = comment.Append(" | L3 field 0.0 T");
4642ac4b 283 TGeoGlobalMagField::Instance()->SetField(new AliMagF("Maps","Maps", 0., 0., AliMagF::k5kGUniform));
2858736f 284 } else if (mag == k5kG) {
285 comment = comment.Append(" | L3 field 0.5 T");
4642ac4b 286 TGeoGlobalMagField::Instance()->SetField(new AliMagF("Maps","Maps", -1., -1., AliMagF::k5kG));
2858736f 287 }
288 printf("\n \n Comment: %s \n \n", comment.Data());
ae8c1d56 289 // TGeoGlobalMagField::Instance()->SetField(field);
2858736f 290
291 rl->CdGAFile();
f7a1cc68 292
2858736f 293 Int_t iABSO = 1;
294 Int_t iACORDE= 0;
295 Int_t iDIPO = 1;
ae8c1d56 296 Int_t iEMCAL = 1;
2858736f 297 Int_t iFMD = 1;
298 Int_t iFRAME = 1;
299 Int_t iHALL = 1;
300 Int_t iITS = 1;
301 Int_t iMAG = 1;
302 Int_t iMUON = 1;
303 Int_t iPHOS = 1;
304 Int_t iPIPE = 1;
ae8c1d56 305 Int_t iPMD = 1;
2858736f 306 Int_t iHMPID = 1;
307 Int_t iSHIL = 1;
308 Int_t iT0 = 1;
309 Int_t iTOF = 1;
310 Int_t iTPC = 1;
311 Int_t iTRD = 1;
312 Int_t iVZERO = 1;
313 Int_t iZDC = 1;
314
315
316 //=================== Alice BODY parameters =============================
317 AliBODY *BODY = new AliBODY("BODY", "Alice envelop");
318
319
320 if (iMAG)
321 {
322 //=================== MAG parameters ============================
323 // --- Start with Magnet since detector layouts may be depending ---
324 // --- on the selected Magnet dimensions ---
325 AliMAG *MAG = new AliMAG("MAG", "Magnet");
326 }
327
328
329 if (iABSO)
330 {
331 //=================== ABSO parameters ============================
332 AliABSO *ABSO = new AliABSOv3("ABSO", "Muon Absorber");
333 }
334
335 if (iDIPO)
336 {
337 //=================== DIPO parameters ============================
338
339 AliDIPO *DIPO = new AliDIPOv3("DIPO", "Dipole version 3");
340 }
341
342 if (iHALL)
343 {
344 //=================== HALL parameters ============================
345
346 AliHALL *HALL = new AliHALLv3("HALL", "Alice Hall");
347 }
348
349
350 if (iFRAME)
351 {
352 //=================== FRAME parameters ============================
353
354 AliFRAMEv2 *FRAME = new AliFRAMEv2("FRAME", "Space Frame");
355 FRAME->SetHoles(1);
356 }
357
358 if (iSHIL)
359 {
360 //=================== SHIL parameters ============================
361
362 AliSHIL *SHIL = new AliSHILv3("SHIL", "Shielding Version 3");
363 }
364
365
366 if (iPIPE)
367 {
368 //=================== PIPE parameters ============================
369
370 AliPIPE *PIPE = new AliPIPEv3("PIPE", "Beam Pipe");
371 }
372
373 if (iITS)
374 {
375 //=================== ITS parameters ============================
376
3a5fcc75 377 AliITS *ITS = new AliITSv11("ITS","ITS v11");
2858736f 378 }
379
380 if (iTPC)
381 {
382 //============================ TPC parameters =====================
383
384 AliTPC *TPC = new AliTPCv2("TPC", "Default");
385 }
386
387
388 if (iTOF) {
389 //=================== TOF parameters ============================
390
391 AliTOF *TOF = new AliTOFv6T0("TOF", "normal TOF");
392 }
393
394
395 if (iHMPID)
396 {
397 //=================== HMPID parameters ===========================
398
399 AliHMPID *HMPID = new AliHMPIDv3("HMPID", "normal HMPID");
400
401 }
402
403
404 if (iZDC)
405 {
406 //=================== ZDC parameters ============================
407
047922b1 408 AliZDC *ZDC = new AliZDCv4("ZDC", "normal ZDC");
2858736f 409 }
410
411 if (iTRD)
412 {
413 //=================== TRD parameters ============================
414
415 AliTRD *TRD = new AliTRDv1("TRD", "TRD slow simulator");
2858736f 416 }
417
418 if (iFMD)
419 {
420 //=================== FMD parameters ============================
421
422 AliFMD *FMD = new AliFMDv1("FMD", "normal FMD");
423 }
424
425 if (iMUON)
426 {
427 //=================== MUON parameters ===========================
428 // New MUONv1 version (geometry defined via builders)
429
430 AliMUON *MUON = new AliMUONv1("MUON", "default");
431 }
432
433 if (iPHOS)
434 {
435 //=================== PHOS parameters ===========================
5889a25e 436 AliPHOS *PHOS = new AliPHOSv1("PHOS", "Run1");
2858736f 437 }
438
439
440 if (iPMD)
441 {
442 //=================== PMD parameters ============================
443
444 AliPMD *PMD = new AliPMDv1("PMD", "normal PMD");
445 }
446
447 if (iT0)
448 {
449 //=================== T0 parameters ============================
450 AliT0 *T0 = new AliT0v1("T0", "T0 Detector");
451 }
452
453 if (iEMCAL)
454 {
455 //=================== EMCAL parameters ============================
456
c76b9217 457 AliEMCAL *EMCAL = new AliEMCALv2("EMCAL", "EMCAL_COMPLETEV1");
2858736f 458 }
459
460 if (iACORDE)
461 {
462 //=================== ACORDE parameters ============================
463
464 AliACORDE *ACORDE = new AliACORDEv1("ACORDE", "normal ACORDE");
465 }
466
467 if (iVZERO)
468 {
469 //=================== ACORDE parameters ============================
470
471 AliVZERO *VZERO = new AliVZEROv7("VZERO", "normal VZERO");
472 }
473}
474//
475// PYTHIA
476//
477
478AliGenerator* MbPythia()
479{
480 comment = comment.Append(" pp at 14 TeV: Pythia low-pt");
481//
482// Pythia
483 AliGenPythia* pythia = new AliGenPythia(-1);
484 pythia->SetMomentumRange(0, 999999.);
485 pythia->SetThetaRange(0., 180.);
486 pythia->SetYRange(-12.,12.);
487 pythia->SetPtRange(0,1000.);
488 pythia->SetProcess(kPyMb);
489 pythia->SetEnergyCMS(energy);
490
491 return pythia;
492}
493
494AliGenerator* MbPhojet()
495{
496 comment = comment.Append(" pp at 14 TeV: Pythia low-pt");
497//
498// DPMJET
499#if defined(__CINT__)
2858736f 500#endif
257e1b2b 501 gSystem->Load("libDPMJET"); // Parton density functions
ae8c1d56 502 gSystem->Load("libTDPMjet"); // Parton density functions
503
2858736f 504 AliGenDPMjet* dpmjet = new AliGenDPMjet(-1);
505 dpmjet->SetMomentumRange(0, 999999.);
506 dpmjet->SetThetaRange(0., 180.);
507 dpmjet->SetYRange(-12.,12.);
508 dpmjet->SetPtRange(0,1000.);
509 dpmjet->SetProcess(kDpmMb);
510 dpmjet->SetEnergyCMS(energy);
511
512 return dpmjet;
513}
514
515void ProcessEnvironmentVars()
516{
517 // Run type
518 if (gSystem->Getenv("CONFIG_RUN_TYPE")) {
519 for (Int_t iRun = 0; iRun < kRunMax; iRun++) {
520 if (strcmp(gSystem->Getenv("CONFIG_RUN_TYPE"), pprRunName[iRun])==0) {
521 proc = (PDC06Proc_t)iRun;
522 cout<<"Run type set to "<<pprRunName[iRun]<<endl;
523 }
524 }
525 }
526
527 // Field
528 if (gSystem->Getenv("CONFIG_FIELD")) {
529 for (Int_t iField = 0; iField < kFieldMax; iField++) {
530 if (strcmp(gSystem->Getenv("CONFIG_FIELD"), pprField[iField])==0) {
531 mag = (Mag_t)iField;
532 cout<<"Field set to "<<pprField[iField]<<endl;
533 }
534 }
535 }
536
537 // Energy
538 if (gSystem->Getenv("CONFIG_ENERGY")) {
539 energy = atoi(gSystem->Getenv("CONFIG_ENERGY"));
540 cout<<"Energy set to "<<energy<<" GeV"<<endl;
541 }
542
543 // Random Number seed
544 if (gSystem->Getenv("CONFIG_SEED")) {
545 seed = atoi(gSystem->Getenv("CONFIG_SEED"));
546 }
547
548 // Embedding job type
549 if (gSystem->Getenv("CONFIG_EMBEDDING")) {
550 for (Int_t iEmb = 0; iEmb < kEmbRunMax; iEmb++) {
551 if (strcmp(gSystem->Getenv("CONFIG_EMBEDDING"), embedRun[iEmb])==0) {
552 embedrun = (EmbedRun_t)iEmb;
553 cout<<"Embedding run set to "<<embedRun[embedrun]<<endl;
554 }
555 }
556
557 }
558
559}
560
561Float_t EtaToTheta(Float_t arg){
562 return (180./TMath::Pi())*2.*atan(exp(-arg));
563}