TDPMjet/TDPMjet.cxx

   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 //*KEND.
  45
  46 //*KEEP,TROOT.
  47 #include "TROOT.h"
  48 //*KEND.
  49
  50 #ifndef WIN32
  51 # define dt_dtuini dt_dtuini_
  52 # define dt_getemu de_getemu_
  53 # define dt_kkinc  dt_kkinc_
  54 # define pho_phist pho_phist_
  55 # define dt_dtuout dt_dtuout_
  56 # define dt_rndm   dt_rndm_
  57 # define dt_rndmst dt_rndmst_
  58 # define dt_rndmin dt_rndmin_
  59 # define dt_rndmou dt_rndmou_
  60 # define type_of_call
  61 #else
  62 # define dt_dtuini DT_DTUINI
  63 # define dt_getemu DT_GETEMU
  64 # define dt_kkinc  DT_KKINC
  65 # define pho_phist PHO_PHIST
  66 # define dt_dtuout DT_DTUOUT
  67 # define dt_rndm   DT_RNDM
  68 # define dt_rndmst DT_RNDMST
  69 # define dt_rndmin DT_RNDMIN
  70 # define dt_rndmou DT_RNDMOU
  71 # define type_of_call _stdcall
  72 #endif
  73
  74 #ifndef WIN32
  75 extern "C" void   type_of_call dt_dtuini(Int_t & , Double_t &, Int_t & , Int_t &,
  76                         Int_t &, Int_t &, Int_t &, Int_t &);
  77 extern "C" double type_of_call dt_getemu(Int_t &, Int_t &, Int_t &, Int_t &);
  78 extern "C" void   type_of_call dt_kkinc(Int_t &, Int_t &, Int_t &, Int_t &,
  79                                     Int_t &, Double_t &, Int_t &, Int_t &);
  80 extern "C" void   type_of_call pho_phist(Int_t &, Double_t &);
  81 extern "C" void   type_of_call dt_dtuout();
  82 extern "C" void   type_of_call dt_rndm(Int_t &);
  83 extern "C" void   type_of_call dt_rndmst(Int_t &, Int_t &, Int_t &, Int_t &);
  84 extern "C" void   type_of_call dt_rndmin(Int_t &, Int_t &, Int_t &, Int_t &, Int_t &, Int_t &);
  85 extern "C" void   type_of_call dt_rndmou(Int_t &, Int_t &, Int_t &, Int_t &, Int_t &, Int_t &);
  86
  87 #else
  88
  89 #endif
  90
  91 ClassImp(TDPMjet)
  92
  93 //______________________________________________________________________________
  94 TDPMjet::TDPMjet() : TGenerator("dpmjet","dpmjet")
  95 {
  96     fNEvent  = 0;
  97     fIp      = 0;
  98     fIpz     = 0;
  99     fIt      = 0;
 100     fItz     = 0;
 101     fEpn     = 0;
 102     fCMEn    = 0;
 103     fIdp     = 0;
 104     fProcess = kDpmMb;
 105 }
 106
 107 //______________________________________________________________________________
 108 TDPMjet::TDPMjet(DpmProcess_t  iproc, Int_t Ip=208, Int_t Ipz=82, Int_t It=208, Int_t Itz=82,
 109         Double_t Epn=2700., Double_t CMEn=5400.)
 110         : TGenerator("dpmjet","dpmjet")
 111 {
 112     printf("TDPMJet Constructor %d %d %d %d \n", Ip, Ipz, It, Itz);
 113
 114    fNEvent  = 0;
 115    fIp      = Ip;
 116    fIpz     = Ipz;
 117    fIt      = It;
 118    fItz     = Itz;
 119    fEpn     = Epn;
 120    fCMEn    = CMEn;
 121    fIdp     = 0;
 122    fProcess = iproc;
 123 }
 124
 125
 126 //______________________________________________________________________________
 127 Int_t TDPMjet::ImportParticles(TClonesArray *particles, Option_t *option)
 128 {
 129 //
 130 //  Default primary creation method. It reads the /HEPEVT/ common block which
 131 //  has been filled by the GenerateEvent method. If the event generator does
 132 //  not use the HEPEVT common block, This routine has to be overloaded by
 133 //  the subclasses.
 134 //  The function loops on the generated particles and store them in
 135 //  the TClonesArray pointed by the argument particles.
 136 //  The default action is to store only the stable particles
 137 //  This can be demanded explicitly by setting the option = "Final"
 138 //  If the option = "All", all the particles are stored.
 139 //
 140   if(particles==0) return 0;
 141   TClonesArray &Particles = *particles;
 142   Particles.Clear();
 143   Int_t numpart = 0;     // Total number of produced particles
 144   Int_t numStabpart = 0; // Total number of produced stable particles
 145   Double_t entot = 0;    // Total energy in final state (from stable particles)
 146
 147   numpart = DTEVT1.nhkk;
 148   for(Int_t i=0; i<numpart; i++){
 149      if(DTEVT1.isthkk[i]==1 || DTEVT1.isthkk[i]==-1 || DTEVT1.isthkk[i]==1001){
 150         numStabpart++;
 151         entot += DTEVT1.phkk[i][3]; // PHKK[i][3] <-> PHKK(4,i)
 152      }
 153   }
 154   //printf("\n TDPMjet: DPMJET stack contains %d particles", numpart);
 155   // printf("\n TDPMjet: Final not decayed particles: %d",    numStabpart);
 156   //printf("\n TDPMjet: Total energy: %f GeV          \n",   entot);
 157   Int_t nump = 0;
 158
 159   if(!strcmp(option,"") || !strcmp(option,"Final")){
 160       for (Int_t i=0; i < numpart; i++) {
 161
 162           if (DTEVT1.isthkk[i] == 1) {
 163            //
 164            //  Use the common block values for the TParticle constructor
 165            //
 166               new(Particles[nump]) TParticle(
 167                   DTEVT1.idhkk[i],
 168                   DTEVT1.isthkk[i],
 169                   -1,
 170                   -1,
 171                   -1,
 172                   -1,
 173                   DTEVT1.phkk[i][0],
 174                   DTEVT1.phkk[i][1],
 175                   DTEVT1.phkk[i][2],
 176                   DTEVT1.phkk[i][3],
 177
 178                   DTEVT1.vhkk[i][0],
 179                   DTEVT1.vhkk[i][1],
 180                   DTEVT1.vhkk[i][2],
 181                   DTEVT1.vhkk[i][3]);
 182               nump++;
 183           }
 184       }
 185   }
 186   else if(!strcmp(option,"All")){
 187       nump = numpart;
 188       for (Int_t i=0; i <= numpart; i++){
 189
 190           // DTEVT1.JMOHKK[i][0] pointer to the entry of the 1st mother of entry i
 191           Int_t iParent = DTEVT1.jmohkk[i][0] - 1;
 192
 193           if(iParent >= 0){
 194               TParticle *mother = (TParticle*) (Particles.UncheckedAt(iParent));
 195               mother->SetLastDaughter(i);
 196               if(mother->GetFirstDaughter() == -1) mother->SetFirstDaughter(i);
 197           }
 198           // --- PDGcode for residual nuclei (idhkk=80000)
 199           // ---        10000*Z + 10*A
 200           // --- DPMJET -> idres = mass #, idxres = charge
 201           if(DTEVT1.idhkk[i] == 80000)
 202               DTEVT1.idhkk[i] = 10000*DTEVT2.idxres[i]+10*DTEVT2.idres[i];
 203 /*
 204           if(DTEVT2.idxres[i] != 0)
 205               printf("\n        pc#%d -> A = %d, Z = %d -> PDGcode = %d\n",
 206                      i,DTEVT2.idres[i],DTEVT2.idxres[i],DTEVT1.idhkk[i]);
 207 */
 208           new(Particles[i]) TParticle(
 209               DTEVT1.idhkk[i],
 210               DTEVT1.isthkk[i],
 211               iParent,
 212               -1,
 213               -1,
 214               -1,
 215
 216               DTEVT1.phkk[i][0],
 217               DTEVT1.phkk[i][1],
 218               DTEVT1.phkk[i][2],
 219               DTEVT1.phkk[i][3],
 220
 221               DTEVT1.vhkk[i][0],
 222               DTEVT1.vhkk[i][1],
 223               DTEVT1.vhkk[i][2],
 224               DTEVT1.vhkk[i][3]);
 225       } // Particle loop
 226   }
 227   return nump;
 228 }
 229
 230
 231 //====================== access to dpmjet subroutines =========================
 232 //______________________________________________________________________________
 233 void TDPMjet::Initialize()
 234 {
 235 //
 236 //  Write standard DPMJET input cards
 237 //
 238     FILE* out = fopen("dpmjet.inp","w");
 239 //  Projectile and Target definition
 240     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.);
 241     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.);
 242 //  Beam energy and crossing-angle
 243     fprintf(out, "BEAM      %10.1f%10.1f%10.1f%10.1f%10.1f%10.1f\n",fEpn, fEpn, 0., 0., 0., 0.);
 244 //  Centrality
 245     fprintf(out, "CENTRAL   %10.1f%10.1f%10.1f%10.1f%10.1f%10.1f\n",-1., fBmin, fBmax, 0., 0., 0.);
 246
 247 //
 248 //  PHOJET specific
 249     fprintf(out, "PHOINPUT\n");
 250     fprintf(out, "DEBUG      0 0 0 \n");
 251
 252     if (fProcess == kDpmMb) {
 253         fprintf(out, "PROCESS           1 0 1 1 1 1 1 1\n");
 254     } else if (fProcess == kDpmMbNonDiffr) {
 255         fprintf(out, "PROCESS           1 0 1 1 0 0 0 1\n");
 256     }
 257
 258     fprintf(out, "ENDINPUT\n");
 259 //
 260 //  START card
 261     fprintf(out, "START            1.0       0.0\n");
 262     fprintf(out, "STOP\n");
 263     fclose(out);
 264 //
 265 //  Call DPMJET initialisation
 266     Int_t iemu = 0; // No emulsion (default)
 267     Dt_Dtuini(1, fEpn, fIp, fIpz, fIt, fItz, fIdp, iemu);
 268
 269 }
 270
 271
 272 //______________________________________________________________________________
 273 void TDPMjet::GenerateEvent()
 274 {
 275    // Generates one event;
 276    fNEvent++;
 277    DTEVNO.nevent=fNEvent;
 278    Int_t kkmat=-1;
 279    Float_t Elab = fEpn;
 280    Int_t irej=0;
 281    Dt_Kkinc(fIp, fIpz, fIt, fItz, fIdp, Elab, kkmat, irej);
 282    if(irej!=0) return;
 283 }
 284 //______________________________________________________________________________
 285 void TDPMjet::Dt_Dtuini(int nevts, double epn, int npmass, int npchar,
 286                         int ntmass, int ntchar, int idp, int iemu)
 287 {
 288   // Call dmpjet routine DT_DTUINI passing the parameters
 289   // in a way accepted by Fortran routines
 290
 291
 292    printf("\n-------------------------------------------\n");
 293    printf("\n           Dt_Dtuini called with:\n\n");
 294    printf(" Projectile  -> A = %d, Z = %d \n",npmass, npchar);
 295    printf(" Target      -> A = %d, Z = %d \n",ntmass, ntchar);
 296    printf(" Proj. LAB E -> E = %f GeV \n",epn);
 297    printf(" nevts = %d, idp = %d, iemu = %d \n",nevts,idp,iemu);
 298    printf("\n-------------------------------------------\n");
 299
 300    dt_dtuini(nevts, epn, npmass, npchar, ntmass, ntchar, idp, iemu);
 301
 302 }
 303
 304 //______________________________________________________________________________
 305 void TDPMjet::Dt_Kkinc(int npmass, int npchar, int ntmass, int ntchar,
 306                        int idp, double elab, int kkmat, int irej)
 307 {
 308   // Call dmpjet routine DT_KKINC passing the parameters
 309   // in a way accepted by Fortran routines
 310   dt_kkinc(npmass, npchar, ntmass, ntchar, idp, elab, kkmat, irej);
 311 }
 312
 313 //______________________________________________________________________________
 314 void TDPMjet::Pho_Phist(int imode, double weight)
 315 {
 316   // Call dmpjet routine PHO_PHIST passing the parameters
 317   // in a way accepted by Fortran routines
 318
 319   pho_phist(imode,weight);
 320
 321 }
 322
 323 //______________________________________________________________________________
 324 void TDPMjet::Dt_Dtuout()
 325 {
 326   // Call dmpjet routine DT_DTUOT passing the parameters
 327   // in a way accepted by Fortran routines
 328
 329   dt_dtuout();
 330
 331 }
 332
 333 //______________________________________________________________________________
 334 Int_t TDPMjet::GetEvNum() const
 335 {
 336         return DTEVT1.nevhkk;
 337 }
 338 //______________________________________________________________________________
 339 Int_t TDPMjet::GetEntriesNum() const
 340 {
 341         return DTEVT1.nhkk;
 342 }
 343 //______________________________________________________________________________
 344 Int_t TDPMjet::GetNumStablePc() const
 345 {
 346         Int_t NumStablePc = 0;
 347         for(Int_t i=0; i<DTEVT1.nhkk; i++){
 348            if(DTEVT1.isthkk[i] == 1) NumStablePc++;
 349         }
 350         return NumStablePc;
 351 }
 352
 353 //______________________________________________________________________________
 354 Float_t TDPMjet::GetTotEnergy() const
 355 {
 356         Float_t TotEnergy = 0.;
 357         for(Int_t i=0; i<DTEVT1.nhkk; i++){
 358           if(DTEVT1.isthkk[i] == 1)
 359             TotEnergy += DTEVT1.phkk[i][3]; // PHKK[i][3] <-> PHKK(4,i)
 360         }
 361         return TotEnergy;
 362 }
 363
 364 //______________________________________________________________________________
 365 Int_t TDPMjet::GetStatusCode(Int_t evnum) const
 366 {
 367         return DTEVT1.isthkk[evnum];
 368 }
 369 //______________________________________________________________________________
 370 Int_t TDPMjet::GetPDGCode(Int_t evnum) const
 371 {
 372         return DTEVT1.idhkk[evnum];
 373 }
 374 //______________________________________________________________________________
 375 Double_t TDPMjet::Getpx(Int_t evnum) const
 376 {
 377         return DTEVT1.phkk[evnum][0];
 378 }
 379 //______________________________________________________________________________
 380 Double_t TDPMjet::Getpy(Int_t evnum) const
 381 {
 382         return DTEVT1.phkk[evnum][1];
 383 }
 384 //______________________________________________________________________________
 385 Double_t TDPMjet::Getpz(Int_t evnum) const
 386 {
 387         return DTEVT1.phkk[evnum][2];
 388 }
 389 //______________________________________________________________________________
 390 Double_t TDPMjet::GetEnergy(Int_t evnum) const
 391 {
 392         return DTEVT1.phkk[evnum][3];
 393 }
 394 //______________________________________________________________________________
 395 Double_t TDPMjet::GetMass(Int_t evnum) const
 396 {
 397         return DTEVT1.phkk[evnum][4];
 398 }
 399 //______________________________________________________________________________
 400 Int_t    TDPMjet::GetFragmentA(Int_t evnum) const
 401 {
 402         return DTEVT2.idres[evnum];
 403 }
 404 //______________________________________________________________________________
 405 Int_t    TDPMjet::GetFragmentZ(Int_t evnum) const
 406 {
 407         return DTEVT2.idxres[evnum];
 408 }
 409 //______________________________________________________________________________
 410 Double_t TDPMjet::GetXSFrac() const
 411 {
 412         return DTIMPA.xsfrac;
 413 }
 414 //______________________________________________________________________________
 415 Double_t TDPMjet::GetBImpac() const
 416 {
 417         return DTGLCP.bimpac;
 418 }
 419 //______________________________________________________________________________
 420 Double_t TDPMjet::GetProjRadius() const
 421 {
 422         return DTGLCP.rproj;
 423 }
 424 //______________________________________________________________________________
 425 Double_t TDPMjet::GetTargRadius() const
 426 {
 427         return DTGLCP.rtarg;
 428 }
 429 //______________________________________________________________________________
 430 Int_t TDPMjet::GetProjWounded() const
 431 {
 432         return DTGLCP.nwasam;
 433 }
 434 //______________________________________________________________________________
 435 Int_t TDPMjet::GetTargWounded() const
 436 {
 437         return DTGLCP.nwbsam;
 438 }
 439 //______________________________________________________________________________
 440 Int_t TDPMjet::GetProjSpectators() const
 441 {
 442         return DTGLCP.nwtaac;
 443 }
 444 //______________________________________________________________________________
 445 Int_t TDPMjet::GetTargSpectators() const
 446 {
 447         return DTGLCP.nwtbac;
 448 }
 449
 450 //______________________________________________________________________________
 451 void TDPMjet::Dt_Rndm(int idummy)
 452 {
 453         dt_rndm(idummy);
 454 }
 455
 456 //______________________________________________________________________________
 457 void TDPMjet::Dt_Rndmst(int na1, int na2, int na3, int nb1)
 458 {
 459         dt_rndmst(na1, na2, na3, nb1);
 460 }
 461
 462 //______________________________________________________________________________
 463 void TDPMjet::Dt_Rndmin(int u, int c, int cd, int cm, int i, int j)
 464 {
 465         dt_rndmin(u, c, cd, cm, i, j);
 466 }
 467
 468 //______________________________________________________________________________
 469 void TDPMjet::Dt_Rndmou(int u, int c, int cd, int cm, int i, int j)
 470 {
 471         dt_rndmou(u, c, cd, cm, i, j);
 472 }
 473