]> git.uio.no Git - u/mrichter/AliRoot.git/blame_incremental - TDPMjet/TDPMjet.cxx
Merge branch 'master' of https://git.cern.ch/reps/AliRoot
[u/mrichter/AliRoot.git] / TDPMjet / TDPMjet.cxx
... / ...
CommitLineData
1//*KEEP,CopyRight,T=C.
2/*************************************************************************
3 * Copyright (C) 1995-2000, Rene Brun and Fons Rademakers. *
4 * All rights reserved. *
5 * *
6 * For the licensing terms see $ROOTSYS/LICENSE. *
7 * For the list of contributors see $ROOTSYS/README/CREDITS. *
8 *************************************************************************/
9//*KEND.
10
11//* +-------------------------------------------------------------+
12//* | |
13//* | |
14//* | DPMJET 3.0 |
15//* | |
16//* | |
17//* | S. Roesler+), R. Engel#), J. Ranft*) |
18//* | |
19//* | +) CERN, TIS-RP |
20//* | CH-1211 Geneva 23, Switzerland |
21//* | Email: Stefan.Roesler@cern.ch |
22//* | |
23//* | #) University of Delaware, BRI |
24//* | Newark, DE 19716, USA |
25//* | |
26//* | *) University of Siegen, Dept. of Physics |
27//* | D-57068 Siegen, Germany |
28//* | |
29//* | |
30//* | http://home.cern.ch/sroesler/dpmjet3.html |
31//* | |
32//* | |
33//* | Monte Carlo models used for event generation: |
34//* | PHOJET 1.12, JETSET 7.4 and LEPTO 6.5.1 |
35//* | |
36//* +-------------------------------------------------------------+
37
38//*KEEP,TDPMjet.
39#include "TDPMjet.h"
40//*KEEP,DPMCOMMON.
41#include "DPMcommon.h"
42//*KEEP,TParticle,T=C++
43#include "TParticle.h"
44#include "TClonesArray.h"
45//*KEND.
46
47//*KEEP,TROOT.
48#include "TROOT.h"
49//*KEND.
50
51#ifndef WIN32
52# define dt_dtuini dt_dtuini_
53# define dt_getemu de_getemu_
54# define dt_kkinc dt_kkinc_
55# define pho_phist pho_phist_
56# define dt_dtuout dt_dtuout_
57# define dt_rndm dt_rndm_
58# define dt_rndmst dt_rndmst_
59# define dt_rndmin dt_rndmin_
60# define dt_rndmou dt_rndmou_
61# define dpmjet_openinp dpmjet_openinp_
62# define dt_evtout dt_evtout_
63# define type_of_call
64#else
65# define dt_dtuini DT_DTUINI
66# define dt_getemu DT_GETEMU
67# define dt_kkinc DT_KKINC
68# define pho_phist PHO_PHIST
69# define dt_dtuout DT_DTUOUT
70# define dt_rndm DT_RNDM
71# define dt_rndmst DT_RNDMST
72# define dt_rndmin DT_RNDMIN
73# define dt_rndmou DT_RNDMOU
74# define dt_evtout DT_EVTOUT_
75# define dpmjet_openinp DPMJET_OPENINP
76# define type_of_call _stdcall
77#endif
78
79
80#ifndef WIN32
81extern "C" void type_of_call dt_dtuini(Int_t & , Double_t &, Int_t & , Int_t &,
82 Int_t &, Int_t &, Int_t &, Int_t &);
83extern "C" double type_of_call dt_getemu(Int_t &, Int_t &, Int_t &, Int_t &);
84extern "C" void type_of_call dt_kkinc(Int_t &, Int_t &, Int_t &, Int_t &,
85 Int_t &, Double_t &, Int_t &, Int_t &);
86extern "C" void type_of_call pho_phist(Int_t &, Double_t &);
87extern "C" void type_of_call dt_dtuout();
88extern "C" int type_of_call dt_evtout(const Int_t &);
89extern "C" void type_of_call dt_rndm(Int_t &);
90extern "C" void type_of_call dt_rndmst(Int_t &, Int_t &, Int_t &, Int_t &);
91extern "C" void type_of_call dt_rndmin(Int_t &, Int_t &, Int_t &, Int_t &, Int_t &, Int_t &);
92extern "C" void type_of_call dt_rndmou(Int_t &, Int_t &, Int_t &, Int_t &, Int_t &, Int_t &);
93extern "C" void type_of_call dpmjet_openinp();
94
95#else
96
97#endif
98
99ClassImp(TDPMjet)
100
101
102//______________________________________________________________________________
103 TDPMjet::TDPMjet() :
104 TGenerator("dpmjet","dpmjet"),
105 fNEvent(0),
106 fIp(0),
107 fIpz(0),
108 fIt(0),
109 fItz(0),
110 fEpn(0.),
111 fPpn(0.),
112 fCMEn(0.),
113 fIdp(0),
114 fBmin(0.),
115 fBmax(0.),
116 fFCentr(0),
117 fPi0Decay(0),
118 fDecayAll(0),
119 fProcess(kDpmMb),
120 fFragmentation(kFALSE)
121{
122// Default Constructor
123}
124
125//______________________________________________________________________________
126TDPMjet::TDPMjet(DpmProcess_t iproc, Int_t Ip=208, Int_t Ipz=82, Int_t It=208, Int_t Itz=82,
127 Double_t Epn=2700., Double_t CMEn=5400.)
128 : TGenerator("dpmjet","dpmjet"),
129 fNEvent(0),
130 fIp(Ip),
131 fIpz(Ipz),
132 fIt(It),
133 fItz(Itz),
134 fEpn(Epn),
135 fPpn(0.),
136 fCMEn(CMEn),
137 fIdp(0),
138 fBmin(0.),
139 fBmax(0.),
140 fFCentr(0),
141 fPi0Decay(0),
142 fDecayAll(0),
143 fProcess(iproc),
144 fFragmentation(kFALSE)
145{
146 printf("TDPMJet Constructor %d %d %d %d \n", Ip, Ipz, It, Itz);
147}
148
149
150//______________________________________________________________________________
151Int_t TDPMjet::ImportParticles(TClonesArray *particles, Option_t *option)
152{
153//
154// Default primary creation method. It reads the /HEPEVT/ common block which
155// has been filled by the GenerateEvent method. If the event generator does
156// not use the HEPEVT common block, This routine has to be overloaded by
157// the subclasses.
158// The function loops on the generated particles and store them in
159// the TClonesArray pointed by the argument particles.
160// The default action is to store only the stable particles
161// This can be demanded explicitly by setting the option = "Final"
162// If the option = "All", all the particles are stored.
163//
164 if(particles==0) return 0;
165 TClonesArray &Particles = *particles;
166 Particles.Clear();
167 Int_t numpart = 0; // Total number of produced particles
168 Int_t numStabpart = 0; // Total number of produced stable particles
169 Double_t entot = 0; // Total energy in final state (from stable particles)
170
171 numpart = DTEVT1.nhkk;
172 for(Int_t i=0; i<numpart; i++){
173 if(DTEVT1.isthkk[i]==1 || DTEVT1.isthkk[i]==-1 || DTEVT1.isthkk[i]==1001){
174 numStabpart++;
175 entot += DTEVT1.phkk[i][3]; // PHKK[i][3] <-> PHKK(4,i)
176 }
177 }
178 Int_t nump = 0;
179
180 if(!strcmp(option,"") || !strcmp(option,"Final")){
181 for (Int_t i=0; i < numpart; i++) {
182
183 if (DTEVT1.isthkk[i] == 1) {
184 //
185 // Use the common block values for the TParticle constructor
186 //
187 new(Particles[nump]) TParticle(
188 DTEVT1.idhkk[i],
189 DTEVT1.isthkk[i],
190 -1,
191 -1,
192 -1,
193 -1,
194 DTEVT1.phkk[i][0],
195 DTEVT1.phkk[i][1],
196 DTEVT1.phkk[i][2],
197 DTEVT1.phkk[i][3],
198
199 DTEVT1.vhkk[i][0],
200 DTEVT1.vhkk[i][1],
201 DTEVT1.vhkk[i][2],
202 DTEVT1.vhkk[i][3]);
203 nump++;
204 }
205 }
206 }
207 else if(!strcmp(option,"All")){
208 nump = numpart;
209 for (Int_t i=0; i <= numpart; i++){
210
211 // DTEVT1.JMOHKK[i][0] pointer to the entry of the 1st mother of entry i
212 Int_t iParent = DTEVT1.jmohkk[i][0] - 1;
213
214 if(iParent >= 0){
215 TParticle *mother = (TParticle*) (Particles.UncheckedAt(iParent));
216 mother->SetLastDaughter(i);
217 if(mother->GetFirstDaughter() == -1) mother->SetFirstDaughter(i);
218 }
219 // --- PDGcode for residual nuclei (idhkk=80000)
220 // --- 10000*Z + 10*A
221 // --- DPMJET -> idres = mass #, idxres = charge
222 if(DTEVT1.idhkk[i] == 80000)
223 DTEVT1.idhkk[i] = 10000*DTEVT2.idxres[i]+10*DTEVT2.idres[i];
224/*
225 if(DTEVT2.idxres[i] != 0)
226 printf("\n pc#%d -> A = %d, Z = %d -> PDGcode = %d\n",
227 i,DTEVT2.idres[i],DTEVT2.idxres[i],DTEVT1.idhkk[i]);
228*/
229/*
230 printf("%5d %5d %5d %5d %13.3f %13.3f %13.3f %13.3f \n", i,
231 DTEVT1.idhkk[i],
232 DTEVT1.isthkk[i],
233 iParent,
234 DTEVT1.phkk[i][0],
235 DTEVT1.phkk[i][1],
236 DTEVT1.phkk[i][2],
237 DTEVT1.phkk[i][3]
238 );
239*/
240 new(Particles[i]) TParticle(
241 DTEVT1.idhkk[i],
242 DTEVT1.isthkk[i],
243 iParent,
244 -1,
245 -1,
246 -1,
247
248 DTEVT1.phkk[i][0],
249 DTEVT1.phkk[i][1],
250 DTEVT1.phkk[i][2],
251 DTEVT1.phkk[i][3],
252
253 DTEVT1.vhkk[i][0],
254 DTEVT1.vhkk[i][1],
255 DTEVT1.vhkk[i][2],
256 DTEVT1.vhkk[i][3]);
257 } // Particle loop
258 }
259 return nump;
260
261}
262
263
264//====================== access to dpmjet subroutines =========================
265//______________________________________________________________________________
266void TDPMjet::Initialize()
267{
268//
269// Write standard DPMJET input cards
270//
271 if(fFragmentation) printf("\tTDPMJet fragmentation/evaporation applied\n");
272
273 FILE* out = fopen("dpmjet.inp","w");
274// Projectile and Target definition
275 if (fIp == 1 && fIpz ==1) {
276 fprintf(out, "PROJPAR PROTON\n");
277 } else if (fIp == 1 && fIpz == -1) {
278 fprintf(out, "PROJPAR APROTON\n");
279 } else {
280 fprintf(out, "PROJPAR %10.1f%10.1f%10.1f%10.1f%10.1f%10.1f\n", (Float_t) fIp, (Float_t) fIpz, 0., 0., 0., 0.);
281 }
282
283 if (fIt == 1 && fItz ==1) {
284 fprintf(out, "TARPAR PROTON\n");
285 } else if (fIt == 1 && fItz == -1) {
286 fprintf(out, "TARPAR APROTON\n");
287 } else {
288 fprintf(out, "TARPAR %10.1f%10.1f%10.1f%10.1f%10.1f%10.1f\n", (Float_t) fIt, (Float_t) fItz, 0., 0., 0., 0.);
289 }
290
291// Beam energy and crossing-angle
292 fprintf(out, "CMENERGY %10.1f%10.1f%10.1f%10.1f%10.1f%10.1f\n",fCMEn, 0., 0., 0., 0., 0.);
293 if(fIt == 1 && fIp ==1){
294 fprintf(out, "BEAM %10.1f%10.1f%10.1f%10.1f%10.1f%10.1f\n",fEpn, fEpn, 0., 0., 0., 0.); //p-p
295 }
296 else if(fIp > 1 || fIt > 1){
297 if(fIp>1 && fIt>1) fprintf(out, "BEAM %10.1f%10.1f%10.1f%10.1f%10.1f%10.1f\n",fEpn, fEpn, 0., 0., 0., 0.);//A-A
298 else if(fIp==1 && fIt>1){ // proton towards A side (directed z>0)
299 fprintf(out, "BEAM %10.1f%10.1f%10.1f%10.1f%10.1f%10.1f\n", fEpn,fEpn*fItz/fIt, 0., 0., 0., 0.);//pA
300 printf("\n TDPMjet::Initialize() -> p-A: projectile (p) energy = %10.1f, CMS energy = %10.1f\n\n",fEpn,fCMEn);
301 }
302 else if(fIt==1 && fIp>1){ // proton towards C side (directed z<0)
303 fprintf(out, "BEAM %10.1f%10.1f%10.1f%10.1f%10.1f%10.1f\n", fEpn, fEpn*fIp/fIpz, 0., 0., 0., 0.);//A-p
304 printf("\n TDPMjet::Initialize() -> A-p: projectile (A) energy = %10.1f, CMS energy = %10.1f\n\n",fEpn,fCMEn);
305 }
306 }
307// Centrality
308 if((fIp > 1 || fIt > 1) && fFragmentation)
309 fprintf(out, "CENTRAL %10.1f%10.1f%10.1f%10.1f%10.1f%10.1f\n",-2., fBmin, fBmax, 0., 0., 0.);
310 else if((fIp > 1 || fIt > 1) && !fFragmentation)
311 fprintf(out, "CENTRAL %10.1f%10.1f%10.1f%10.1f%10.1f%10.1f\n",-1., fBmin, fBmax, 0., 0., 0.);
312// Particle decays
313 if (fPi0Decay)
314 fprintf(out, "PARDECAY %10.1f%10.1f%10.1f%10.1f%10.1f%10.1f\n", 2., 0., 0., 0., 0., 0.);
315
316
317//
318// PHOJET specific
319 fprintf(out, "PHOINPUT\n");
320 fprintf(out, "DEBUG 0 0 0 \n");
321
322 if (fProcess == kDpmMb) {
323 fprintf(out, "PROCESS 1 0 1 1 1 1 1 1\n");
324 } else if (fProcess == kDpmMbNonDiffr) {
325 fprintf(out, "PROCESS 1 0 1 0 0 0 0 1\n");
326 } else if (fProcess == kDpmDiffr) {
327 fprintf(out, "PROCESS 0 0 0 1 1 1 1 0\n");
328 }else if (fProcess == kDpmSingleDiffr) {
329 fprintf(out, "PROCESS 0 0 0 0 1 1 0 0\n");
330 }else if (fProcess == kDpmDoubleDiffr) {
331 fprintf(out, "PROCESS 0 0 0 0 0 0 1 0\n");
332 } else if (fProcess == kDpmCentralDiffr){
333 fprintf(out, "PROCESS 0 0 0 1 0 0 0 0\n");
334 }
335
336 Int_t iPDG[18] =
337 {
338// K0s pi0 lam sig+ sig- tet0
339 310, 111, 3122, 3222, 3112, 3322,
340// tet- om- D+ D0 Ds+
341 3312, 3334, 411, 421, 431,
342// etac lamc+ sigc++ sigc+ sigc0 Ksic+
343 441, 4122, 4222, 4212, 4112, 4232,
344// Ksic0 sig0
345 4132
346 };
347
348
349 Int_t iON = (fDecayAll) ? 1:0;
350 for (Int_t i = 0; i < 8; i++) {
351 fprintf(out, "LUND-DECAY%5d %5d\n", iPDG[i], iON);
352 }
353
354 fprintf(out, "LUND-MSTJ %5d %5d\n", 22, 1);
355
356 fprintf(out, "ENDINPUT\n");
357//
358// START card
359 fprintf(out, "START 1.0 0.0\n");
360 fprintf(out, "STOP\n");
361 fclose(out);
362 dpmjet_openinp();
363
364//
365// Call DPMJET initialisation
366 Int_t iemu = 0; // No emulsion (default)
367 Dt_Dtuini(1, fEpn, fIp, fIpz, fIt, fItz, fIdp, iemu);
368
369}
370
371
372//______________________________________________________________________________
373void TDPMjet::GenerateEvent()
374{
375 // Generates one event;
376 fNEvent++;
377 DTEVNO.nevent=fNEvent;
378 Int_t kkmat=-1;
379 Float_t Elab = fEpn;
380 Int_t irej=0;
381 Dt_Kkinc(fIp, fIpz, fIt, fItz, fIdp, Elab, kkmat, irej);
382}
383//______________________________________________________________________________
384void TDPMjet::Dt_Dtuini(int nevts, double epn, int npmass, int npchar,
385 int ntmass, int ntchar, int idp, int iemu)
386{
387 // Call dmpjet routine DT_DTUINI passing the parameters
388 // in a way accepted by Fortran routines
389
390
391 printf("\n-------------------------------------------\n");
392 printf("\n Dt_Dtuini called with:\n\n");
393 printf(" Projectile -> A = %d, Z = %d \n",npmass, npchar);
394 printf(" Target -> A = %d, Z = %d \n",ntmass, ntchar);
395 printf(" Proj. LAB E -> E = %f GeV \n",epn);
396 printf(" nevts = %d, idp = %d, iemu = %d \n",nevts,idp,iemu);
397 printf("\n-------------------------------------------\n");
398
399 dt_dtuini(nevts, epn, npmass, npchar, ntmass, ntchar, idp, iemu);
400
401}
402
403//______________________________________________________________________________
404void TDPMjet::Dt_Kkinc(int npmass, int npchar, int ntmass, int ntchar,
405 int idp, double elab, int kkmat, int irej)
406{
407 // Call dmpjet routine DT_KKINC passing the parameters
408 // in a way accepted by Fortran routines
409 dt_kkinc(npmass, npchar, ntmass, ntchar, idp, elab, kkmat, irej);
410}
411
412//______________________________________________________________________________
413void TDPMjet::Pho_Phist(int imode, double weight)
414{
415 // Call dmpjet routine PHO_PHIST passing the parameters
416 // in a way accepted by Fortran routines
417
418 pho_phist(imode,weight);
419
420}
421
422//______________________________________________________________________________
423void TDPMjet::Dt_Dtuout()
424{
425 // Call dmpjet routine DT_DTUOT passing the parameters
426 // in a way accepted by Fortran routines
427
428 dt_dtuout();
429
430}
431
432//______________________________________________________________________________
433Int_t TDPMjet::GetEvNum() const
434{
435 return DTEVT1.nevhkk;
436}
437//______________________________________________________________________________
438Int_t TDPMjet::GetEntriesNum() const
439{
440 return DTEVT1.nhkk;
441}
442//______________________________________________________________________________
443Int_t TDPMjet::GetNumStablePc() const
444{
445 Int_t NumStablePc = 0;
446 for(Int_t i=0; i<DTEVT1.nhkk; i++){
447 if(DTEVT1.isthkk[i] == 1) NumStablePc++;
448 }
449 return NumStablePc;
450}
451
452//______________________________________________________________________________
453Float_t TDPMjet::GetTotEnergy() const
454{
455 Float_t TotEnergy = 0.;
456 for(Int_t i=0; i<DTEVT1.nhkk; i++){
457 if(DTEVT1.isthkk[i] == 1)
458 TotEnergy += DTEVT1.phkk[i][3]; // PHKK[i][3] <-> PHKK(4,i)
459 }
460 return TotEnergy;
461}
462
463//______________________________________________________________________________
464Int_t TDPMjet::GetStatusCode(Int_t evnum) const
465{
466 return DTEVT1.isthkk[evnum];
467}
468//______________________________________________________________________________
469Int_t TDPMjet::GetPDGCode(Int_t evnum) const
470{
471 return DTEVT1.idhkk[evnum];
472}
473//______________________________________________________________________________
474Double_t TDPMjet::Getpx(Int_t evnum) const
475{
476 return DTEVT1.phkk[evnum][0];
477}
478//______________________________________________________________________________
479Double_t TDPMjet::Getpy(Int_t evnum) const
480{
481 return DTEVT1.phkk[evnum][1];
482}
483//______________________________________________________________________________
484Double_t TDPMjet::Getpz(Int_t evnum) const
485{
486 return DTEVT1.phkk[evnum][2];
487}
488//______________________________________________________________________________
489Double_t TDPMjet::GetEnergy(Int_t evnum) const
490{
491 return DTEVT1.phkk[evnum][3];
492}
493//______________________________________________________________________________
494Double_t TDPMjet::GetMass(Int_t evnum) const
495{
496 return DTEVT1.phkk[evnum][4];
497}
498//______________________________________________________________________________
499Int_t TDPMjet::GetFragmentA(Int_t evnum) const
500{
501 return DTEVT2.idres[evnum];
502}
503//______________________________________________________________________________
504Int_t TDPMjet::GetFragmentZ(Int_t evnum) const
505{
506 return DTEVT2.idxres[evnum];
507}
508//______________________________________________________________________________
509Double_t TDPMjet::GetXSFrac() const
510{
511 return DTIMPA.xsfrac;
512}
513//______________________________________________________________________________
514Double_t TDPMjet::GetBImpac() const
515{
516 return DTGLCP.bimpac;
517}
518//______________________________________________________________________________
519Double_t TDPMjet::GetProjRadius() const
520{
521 return DTGLCP.rproj;
522}
523//______________________________________________________________________________
524Double_t TDPMjet::GetTargRadius() const
525{
526 return DTGLCP.rtarg;
527}
528//______________________________________________________________________________
529Int_t TDPMjet::GetProjWounded() const
530{
531 return DTGLCP.nwasam;
532}
533//______________________________________________________________________________
534Int_t TDPMjet::GetTargWounded() const
535{
536 return DTGLCP.nwbsam;
537}
538
539//______________________________________________________________________________
540Int_t TDPMjet::GetProjParticipants() const
541{
542 return DTGLCP.nwtaac;
543}
544//______________________________________________________________________________
545Int_t TDPMjet::GetTargParticipants() const
546{
547 return DTGLCP.nwtbac;
548}
549//______________________________________________________________________________
550Int_t TDPMjet::GetProcessCode() const
551{
552 return POPRCS.iproce;
553}
554//______________________________________________________________________________
555void TDPMjet::Dt_Rndm(int idummy)
556{
557 dt_rndm(idummy);
558}
559
560//______________________________________________________________________________
561void TDPMjet::Dt_Rndmst(int na1, int na2, int na3, int nb1)
562{
563 dt_rndmst(na1, na2, na3, nb1);
564}
565
566//______________________________________________________________________________
567void TDPMjet::Dt_Rndmin(int u, int c, int cd, int cm, int i, int j)
568{
569 dt_rndmin(u, c, cd, cm, i, j);
570}
571
572//______________________________________________________________________________
573void TDPMjet::Dt_Rndmou(int u, int c, int cd, int cm, int i, int j)
574{
575 dt_rndmou(u, c, cd, cm, i, j);
576}
577
578
579Int_t TDPMjet::NHEP() const {return POEVT1.nhep;}
580Int_t TDPMjet::ISTHEP(Int_t i) const {return POEVT1.isthep[i];}
581Int_t TDPMjet::IDHEP(Int_t i) const {return POEVT1.idhep[i];}
582Double_t TDPMjet::PHEP(Int_t i, Int_t j) const {return POEVT1.phep[i][j];}
583