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