THijing/AliGenHijing.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  * Author: The ALICE Off-line Project.                                    *
   5  * Contributors are mentioned in the code where appropriate.              *
   6  *                                                                        *
   7  * Permission to use, copy, modify and distribute this software and its   *
   8  * documentation strictly for non-commercial purposes is hereby granted   *
   9  * without fee, provided that the above copyright notice appears in all   *
  10  * copies and that both the copyright notice and this permission notice   *
  11  * appear in the supporting documentation. The authors make no claims     *
  12  * about the suitability of this software for any purpose. It is          *
  13  * provided "as is" without express or implied warranty.                  *
  14  **************************************************************************/
  15
  16 /* $Id$ */
  17
  18 // Generator using HIJING as an external generator
  19 // The main HIJING options are accessable for the user through this interface.
  20 // Uses the THijing implementation of TGenerator.
  21 // Author:
  22 // Andreas Morsch    (andreas.morsch@cern.ch)
  23 //
  24
  25 #include <TGraph.h>
  26 #include <THijing.h>
  27 #include <TLorentzVector.h>
  28 #include <TPDGCode.h>
  29 #include <TParticle.h>
  30
  31 #include "AliGenHijing.h"
  32 #include "AliGenHijingEventHeader.h"
  33 #include "AliRun.h"
  34 #include "AliHijingRndm.h"
  35
  36 ClassImp(AliGenHijing)
  37
  38 AliGenHijing::AliGenHijing()
  39                  :AliGenMC()
  40 {
  41 // Constructor
  42     fParticles = 0;
  43     fHijing    = 0;
  44     fDsigmaDb  = 0;
  45     fDnDb      = 0;
  46     AliHijingRndm::SetHijingRandom(GetRandom());
  47 }
  48
  49 AliGenHijing::AliGenHijing(Int_t npart)
  50     :AliGenMC(npart)
  51 {
  52 // Default PbPb collisions at 5. 5 TeV
  53 //
  54     fName = "Hijing";
  55     fTitle= "Particle Generator using HIJING";
  56
  57     SetEnergyCMS();
  58     SetImpactParameterRange();
  59     SetBoostLHC();
  60     SetJetEtaRange();
  61     SetJetPhiRange();
  62
  63     fKeep       =  0;
  64     fQuench     =  1;
  65     fShadowing  =  1;
  66     fTrigger    =  0;
  67     fDecaysOff  =  1;
  68     fEvaluate   =  0;
  69     fSelectAll  =  0;
  70     fFlavor     =  0;
  71     fSpectators =  1;
  72     fDsigmaDb   =  0;
  73     fDnDb       =  0;
  74     fPtMinJet   = -2.5;
  75     fRadiation  =  3;
  76     //
  77     SetSimpleJets();
  78     SetNoGammas();
  79     SetRandomPz();
  80     SwitchOffHeavyQuarks(kFALSE);
  81 //
  82     fParticles = new TClonesArray("TParticle",10000);
  83 //
  84 // Set random number generator
  85     AliHijingRndm::SetHijingRandom(GetRandom());
  86     fHijing = 0;
  87
  88 }
  89
  90 AliGenHijing::AliGenHijing(const AliGenHijing & hijing):
  91     AliGenMC(hijing)
  92 {
  93 // copy constructor
  94 }
  95
  96
  97 AliGenHijing::~AliGenHijing()
  98 {
  99 // Destructor
 100     if ( fDsigmaDb) delete  fDsigmaDb;
 101     if ( fDnDb)     delete  fDnDb;
 102     delete fParticles;
 103 }
 104
 105 void AliGenHijing::Init()
 106 {
 107 // Initialisation
 108     fFrame.Resize(8);
 109     fTarget.Resize(8);
 110     fProjectile.Resize(8);
 111
 112     SetMC(new THijing(fEnergyCMS, fFrame, fProjectile, fTarget,
 113                       fAProjectile, fZProjectile, fATarget, fZTarget,
 114                       fMinImpactParam, fMaxImpactParam));
 115
 116     fHijing=(THijing*) fMCEvGen;
 117     fHijing->SetIHPR2(2,  fRadiation);
 118     fHijing->SetIHPR2(3,  fTrigger);
 119     fHijing->SetIHPR2(6,  fShadowing);
 120     fHijing->SetIHPR2(12, fDecaysOff);
 121     fHijing->SetIHPR2(21, fKeep);
 122     fHijing->SetHIPR1(10, fPtMinJet);
 123     fHijing->SetHIPR1(50, fSimpleJet);
 124 //
 125 //  Quenching
 126 //
 127 //
 128 //  fQuench = 0:  no quenching
 129 //  fQuench = 1:  hijing default
 130 //  fQuench = 2:  new LHC  parameters for HIPR1(11) and HIPR1(14)
 131 //  fQuench = 3:  new RHIC parameters for HIPR1(11) and HIPR1(14)
 132 //  fQuench = 4:  new LHC  parameters with log(e) dependence
 133 //  fQuench = 5:  new RHIC parameters with log(e) dependence
 134     fHijing->SetIHPR2(50, 0);
 135     if (fQuench > 0)
 136         fHijing->SetIHPR2(4,  1);
 137     else
 138         fHijing->SetIHPR2(4,  0);
 139 // New LHC parameters from Xin-Nian Wang
 140     if (fQuench == 2) {
 141         fHijing->SetHIPR1(14, 1.1);
 142         fHijing->SetHIPR1(11, 3.7);
 143     } else if (fQuench == 3) {
 144         fHijing->SetHIPR1(14, 0.20);
 145         fHijing->SetHIPR1(11, 2.5);
 146     } else if (fQuench == 4) {
 147         fHijing->SetIHPR2(50, 1);
 148         fHijing->SetHIPR1(14, 4.*0.34);
 149         fHijing->SetHIPR1(11, 3.7);
 150     } else if (fQuench == 5) {
 151         fHijing->SetIHPR2(50, 1);
 152         fHijing->SetHIPR1(14, 0.34);
 153         fHijing->SetHIPR1(11, 2.5);
 154     }
 155
 156 //
 157 // Heavy quarks
 158 //
 159     if (fNoHeavyQuarks) {
 160         fHijing->SetIHPR2(49, 1);
 161     } else {
 162         fHijing->SetIHPR2(49, 0);
 163     }
 164
 165
 166     AliGenMC::Init();
 167
 168 //
 169 //  Initialize Hijing
 170 //
 171     fHijing->Initialize();
 172 //
 173     if (fEvaluate) EvaluateCrossSections();
 174 //
 175 }
 176
 177 void AliGenHijing::Generate()
 178 {
 179 // Generate one event
 180
 181   Float_t polar[3]    =   {0,0,0};
 182   Float_t origin[3]   =   {0,0,0};
 183   Float_t origin0[3]  =   {0,0,0};
 184   Float_t p[3];
 185   Float_t tof;
 186
 187 //  converts from mm/c to s
 188   const Float_t kconv = 0.001/2.999792458e8;
 189 //
 190   Int_t nt  = 0;
 191   Int_t jev = 0;
 192   Int_t j, kf, ks, imo;
 193   kf = 0;
 194
 195
 196
 197   fTrials = 0;
 198   for (j = 0;j < 3; j++) origin0[j] = fOrigin[j];
 199   if(fVertexSmear == kPerEvent) {
 200       Vertex();
 201       for (j=0; j < 3; j++) origin0[j] = fVertex[j];
 202   }
 203
 204
 205   Float_t sign = (fRandomPz && (Rndm() < 0.5))? -1. : 1.;
 206   while(1)
 207   {
 208 //    Generate one event
 209 // --------------------------------------------------------------------------
 210       fProjectileSpecn    = 0;
 211       fProjectileSpecp    = 0;
 212       fTargetSpecn        = 0;
 213       fTargetSpecp        = 0;
 214 // --------------------------------------------------------------------------
 215       fHijing->GenerateEvent();
 216       fTrials++;
 217       fHijing->ImportParticles(fParticles,"All");
 218       if (fTrigger != kNoTrigger) {
 219           if (!CheckTrigger()) continue;
 220       }
 221       if (fLHC) Boost();
 222
 223
 224       Int_t np = fParticles->GetEntriesFast();
 225       printf("\n **************************************************%d\n",np);
 226       Int_t nc = 0;
 227       if (np == 0 ) continue;
 228       Int_t i;
 229       Int_t* newPos     = new Int_t[np];
 230       Int_t* pSelected  = new Int_t[np];
 231
 232       for (i = 0; i < np; i++) {
 233           newPos[i]    = i;
 234           pSelected[i] = 0;
 235       }
 236
 237 //      Get event vertex
 238 //
 239       TParticle *  iparticle = (TParticle *) fParticles->At(0);
 240       fVertex[0] = origin0[0];
 241       fVertex[1] = origin0[1];
 242       fVertex[2] = origin0[2];
 243
 244 //
 245 //      First select parent particles
 246 //
 247
 248       for (i = 0; i < np; i++) {
 249           iparticle = (TParticle *) fParticles->At(i);
 250
 251 // Is this a parent particle ?
 252           if (Stable(iparticle)) continue;
 253 //
 254           Bool_t  selected             =  kTRUE;
 255           Bool_t  hasSelectedDaughters =  kFALSE;
 256
 257
 258           kf        = iparticle->GetPdgCode();
 259           ks        = iparticle->GetStatusCode();
 260           if (kf == 92) continue;
 261
 262           if (!fSelectAll) selected = KinematicSelection(iparticle, 0) &&
 263                                SelectFlavor(kf);
 264           hasSelectedDaughters = DaughtersSelection(iparticle);
 265 //
 266 // Put particle on the stack if it is either selected or
 267 // it is the mother of at least one seleted particle
 268 //
 269           if (selected || hasSelectedDaughters) {
 270               nc++;
 271               pSelected[i] = 1;
 272           } // selected
 273       } // particle loop parents
 274 //
 275 // Now select the final state particles
 276 //
 277
 278       for (i = 0; i<np; i++) {
 279           TParticle *  iparticle = (TParticle *) fParticles->At(i);
 280 // Is this a final state particle ?
 281           if (!Stable(iparticle)) continue;
 282
 283           Bool_t  selected             =  kTRUE;
 284           kf        = iparticle->GetPdgCode();
 285           ks        = iparticle->GetStatusCode();
 286
 287 // --------------------------------------------------------------------------
 288 // Count spectator neutrons and protons
 289           if(ks == 0 || ks == 1){
 290               if(kf == kNeutron) fProjectileSpecn += 1;
 291               if(kf == kProton)  fProjectileSpecp += 1;
 292           }
 293           else if(ks == 10 || ks == 11){
 294               if(kf == kNeutron) fTargetSpecn += 1;
 295               if(kf == kProton)  fTargetSpecp += 1;
 296           }
 297 // --------------------------------------------------------------------------
 298 //
 299           if (!fSelectAll) {
 300               selected = KinematicSelection(iparticle,0)&&SelectFlavor(kf);
 301               if (!fSpectators && selected) selected = (ks != 0 && ks != 1 && ks != 10
 302                                                         && ks != 11);
 303           }
 304 //
 305 // Put particle on the stack if selected
 306 //
 307           if (selected) {
 308               nc++;
 309               pSelected[i] = 1;
 310           } // selected
 311       } // particle loop final state
 312 //
 313 // Write particles to stack
 314 //
 315       for (i = 0; i<np; i++) {
 316           TParticle *  iparticle = (TParticle *) fParticles->At(i);
 317           Bool_t  hasMother   = (iparticle->GetFirstMother()     >=0);
 318           Bool_t  hasDaughter = (iparticle->GetFirstDaughter()   >=0);
 319           if (pSelected[i]) {
 320               kf   = iparticle->GetPdgCode();
 321               ks   = iparticle->GetStatusCode();
 322               p[0] = iparticle->Px();
 323               p[1] = iparticle->Py();
 324               p[2] = iparticle->Pz() * sign;
 325               origin[0] = origin0[0]+iparticle->Vx()/10;
 326               origin[1] = origin0[1]+iparticle->Vy()/10;
 327               origin[2] = origin0[2]+iparticle->Vz()/10;
 328               tof = kconv*iparticle->T();
 329               imo = -1;
 330               TParticle* mother = 0;
 331               if (hasMother) {
 332                   imo = iparticle->GetFirstMother();
 333                   mother = (TParticle *) fParticles->At(imo);
 334                   imo = (mother->GetPdgCode() != 92) ? newPos[imo] : -1;
 335               } // if has mother
 336               Bool_t tFlag = (fTrackIt && !hasDaughter);
 337               PushTrack(tFlag,imo,kf,p,origin,polar,
 338                        tof,kPNoProcess,nt, 1., ks);
 339
 340
 341               KeepTrack(nt);
 342               newPos[i] = nt;
 343           } // if selected
 344       } // particle loop
 345       delete[] newPos;
 346       delete[] pSelected;
 347
 348       printf("\n I've put %i particles on the stack \n",nc);
 349       if (nc > 0) {
 350           jev += nc;
 351           if (jev >= fNpart || fNpart == -1) {
 352               fKineBias = Float_t(fNpart)/Float_t(fTrials);
 353               printf("\n Trials: %i %i %i\n",fTrials, fNpart, jev);
 354               break;
 355           }
 356       }
 357   } // event loop
 358   MakeHeader();
 359   SetHighWaterMark(nt);
 360 }
 361
 362 void AliGenHijing::KeepFullEvent()
 363 {
 364     fKeep=1;
 365 }
 366
 367 void AliGenHijing::EvaluateCrossSections()
 368 {
 369 //     Glauber Calculation of geometrical x-section
 370 //
 371     Float_t xTot       = 0.;          // barn
 372     Float_t xTotHard   = 0.;          // barn
 373     Float_t xPart      = 0.;          // barn
 374     Float_t xPartHard  = 0.;          // barn
 375     Float_t sigmaHard  = 0.1;         // mbarn
 376     Float_t bMin       = 0.;
 377     Float_t bMax       = fHijing->GetHIPR1(34)+fHijing->GetHIPR1(35);
 378     const Float_t kdib = 0.2;
 379     Int_t   kMax       = Int_t((bMax-bMin)/kdib)+1;
 380
 381
 382     printf("\n Projectile Radius (fm): %f \n",fHijing->GetHIPR1(34));
 383     printf("\n Target     Radius (fm): %f \n",fHijing->GetHIPR1(35));
 384     Int_t i;
 385     Float_t oldvalue= 0.;
 386
 387     Float_t* b   = new Float_t[kMax];
 388     Float_t* si1 = new Float_t[kMax];
 389     Float_t* si2 = new Float_t[kMax];
 390
 391     for (i = 0; i < kMax; i++)
 392     {
 393         Float_t xb  = bMin+i*kdib;
 394         Float_t ov;
 395         ov=fHijing->Profile(xb);
 396         Float_t gb  =  2.*0.01*fHijing->GetHIPR1(40)*kdib*xb*(1.-TMath::Exp(-fHijing->GetHINT1(12)*ov));
 397         Float_t gbh =  2.*0.01*fHijing->GetHIPR1(40)*kdib*xb*sigmaHard*ov;
 398         xTot+=gb;
 399         xTotHard += gbh;
 400         printf("profile %f %f %f\n", xb, ov, fHijing->GetHINT1(12));
 401
 402         if (xb > fMinImpactParam && xb < fMaxImpactParam)
 403         {
 404             xPart += gb;
 405             xPartHard += gbh;
 406         }
 407
 408         if(oldvalue) if ((xTot-oldvalue)/oldvalue<0.0001) break;
 409         oldvalue = xTot;
 410         printf("\n Total cross section (barn): %d %f %f \n",i, xb, xTot);
 411         printf("\n Hard  cross section (barn): %d %f %f \n\n",i, xb, xTotHard);
 412         if (i>0) {
 413             si1[i] = gb/kdib;
 414             si2[i] = gbh/gb;
 415             b[i]  = xb;
 416         }
 417     }
 418
 419     printf("\n Total cross section (barn): %f \n",xTot);
 420     printf("\n Hard  cross section (barn): %f \n \n",xTotHard);
 421     printf("\n Partial       cross section (barn): %f %f \n",xPart, xPart/xTot*100.);
 422     printf("\n Partial  hard cross section (barn): %f %f \n",xPartHard, xPartHard/xTotHard*100.);
 423
 424 //  Store result as a graph
 425     b[0] = 0;
 426     si1[0] = 0;
 427     si2[0]=si2[1];
 428
 429     fDsigmaDb  = new TGraph(i, b, si1);
 430     fDnDb      = new TGraph(i, b, si2);
 431 }
 432
 433 Bool_t AliGenHijing::DaughtersSelection(TParticle* iparticle)
 434 {
 435 //
 436 // Looks recursively if one of the daughters has been selected
 437 //
 438 //    printf("\n Consider daughters %d:",iparticle->GetPdgCode());
 439     Int_t imin = -1;
 440     Int_t imax = -1;
 441     Int_t i;
 442     Bool_t hasDaughters = (iparticle->GetFirstDaughter() >=0);
 443     Bool_t selected = kFALSE;
 444     if (hasDaughters) {
 445         imin = iparticle->GetFirstDaughter();
 446         imax = iparticle->GetLastDaughter();
 447         for (i = imin; i <= imax; i++){
 448             TParticle *  jparticle = (TParticle *) fParticles->At(i);
 449             Int_t ip = jparticle->GetPdgCode();
 450             if (KinematicSelection(jparticle,0)&&SelectFlavor(ip)) {
 451                 selected=kTRUE; break;
 452             }
 453             if (DaughtersSelection(jparticle)) {selected=kTRUE; break; }
 454         }
 455     } else {
 456         return kFALSE;
 457     }
 458     return selected;
 459 }
 460
 461
 462 Bool_t AliGenHijing::SelectFlavor(Int_t pid)
 463 {
 464 // Select flavor of particle
 465 // 0: all
 466 // 4: charm and beauty
 467 // 5: beauty
 468     Bool_t res = 0;
 469
 470     if (fFlavor == 0) {
 471         res = kTRUE;
 472     } else {
 473         Int_t ifl = TMath::Abs(pid/100);
 474         if (ifl > 10) ifl/=10;
 475         res = (fFlavor == ifl);
 476     }
 477 //
 478 //  This part if gamma writing is inhibited
 479     if (fNoGammas)
 480         res = res && (pid != kGamma && pid != kPi0);
 481 //
 482     return res;
 483 }
 484
 485 Bool_t AliGenHijing::Stable(TParticle*  particle) const
 486 {
 487 // Return true for a stable particle
 488 //
 489
 490     if (particle->GetFirstDaughter() < 0 )
 491     {
 492         return kTRUE;
 493     } else {
 494         return kFALSE;
 495     }
 496 }
 497
 498
 499
 500 void AliGenHijing::MakeHeader()
 501 {
 502 // Builds the event header, to be called after each event
 503     AliGenEventHeader* header = new AliGenHijingEventHeader("Hijing");
 504     ((AliGenHijingEventHeader*) header)->SetNProduced(fHijing->GetNATT());
 505     ((AliGenHijingEventHeader*) header)->SetImpactParameter(fHijing->GetHINT1(19));
 506     ((AliGenHijingEventHeader*) header)->SetTotalEnergy(fHijing->GetEATT());
 507     ((AliGenHijingEventHeader*) header)->SetHardScatters(fHijing->GetJATT());
 508     ((AliGenHijingEventHeader*) header)->SetParticipants(fHijing->GetNP(), fHijing->GetNT());
 509     ((AliGenHijingEventHeader*) header)->SetCollisions(fHijing->GetN0(),
 510                                                        fHijing->GetN01(),
 511                                                        fHijing->GetN10(),
 512                                                        fHijing->GetN11());
 513     ((AliGenHijingEventHeader*) header)->SetSpectators(fProjectileSpecn, fProjectileSpecp,
 514                                                        fTargetSpecn,fTargetSpecp);
 515
 516 // 4-momentum vectors of the triggered jets.
 517 //
 518 // Before final state gluon radiation.
 519     TLorentzVector* jet1 = new TLorentzVector(fHijing->GetHINT1(21),
 520                                               fHijing->GetHINT1(22),
 521                                               fHijing->GetHINT1(23),
 522                                               fHijing->GetHINT1(24));
 523
 524     TLorentzVector* jet2 = new TLorentzVector(fHijing->GetHINT1(31),
 525                                               fHijing->GetHINT1(32),
 526                                               fHijing->GetHINT1(33),
 527                                               fHijing->GetHINT1(34));
 528 // After final state gluon radiation.
 529     TLorentzVector* jet3 = new TLorentzVector(fHijing->GetHINT1(26),
 530                                               fHijing->GetHINT1(27),
 531                                               fHijing->GetHINT1(28),
 532                                               fHijing->GetHINT1(29));
 533
 534     TLorentzVector* jet4 = new TLorentzVector(fHijing->GetHINT1(36),
 535                                               fHijing->GetHINT1(37),
 536                                               fHijing->GetHINT1(38),
 537                                               fHijing->GetHINT1(39));
 538     ((AliGenHijingEventHeader*) header)->SetJets(jet1, jet2, jet3, jet4);
 539 // Bookkeeping for kinematic bias
 540     ((AliGenHijingEventHeader*) header)->SetTrials(fTrials);
 541 // Event Vertex
 542     header->SetPrimaryVertex(fVertex);
 543     AddHeader(header);
 544     fCollisionGeometry = (AliGenHijingEventHeader*)  header;
 545 }
 546
 547 void AliGenHijing::AddHeader(AliGenEventHeader* header)
 548 {
 549     // Passes header either to the container or to gAlice
 550     if (fContainer) {
 551         fContainer->AddHeader(header);
 552     } else {
 553         gAlice->SetGenEventHeader(header);
 554     }
 555 }
 556
 557
 558 Bool_t AliGenHijing::CheckTrigger()
 559 {
 560 // Check the kinematic trigger condition
 561 //
 562     Bool_t   triggered = kFALSE;
 563
 564     if (fTrigger == 1) {
 565 //
 566 //  jet-jet Trigger
 567
 568         TLorentzVector* jet1 = new TLorentzVector(fHijing->GetHINT1(26),
 569                                                   fHijing->GetHINT1(27),
 570                                                   fHijing->GetHINT1(28),
 571                                                   fHijing->GetHINT1(29));
 572
 573         TLorentzVector* jet2 = new TLorentzVector(fHijing->GetHINT1(36),
 574                                                   fHijing->GetHINT1(37),
 575                                                   fHijing->GetHINT1(38),
 576                                                   fHijing->GetHINT1(39));
 577         Double_t eta1      = jet1->Eta();
 578         Double_t eta2      = jet2->Eta();
 579         Double_t phi1      = jet1->Phi();
 580         Double_t phi2      = jet2->Phi();
 581 //    printf("\n Trigger: %f %f %f %f",
 582 //         fEtaMinJet, fEtaMaxJet, fPhiMinJet, fPhiMaxJet);
 583         if (
 584             (eta1 < fEtaMaxJet && eta1 > fEtaMinJet &&
 585              phi1 < fPhiMaxJet && phi1 > fPhiMinJet)
 586             ||
 587             (eta2 < fEtaMaxJet && eta2 > fEtaMinJet &&
 588              phi2 < fPhiMaxJet && phi2 > fPhiMinJet)
 589             )
 590             triggered = kTRUE;
 591     } else if (fTrigger == 2) {
 592 //  Gamma Jet
 593 //
 594         Int_t np = fParticles->GetEntriesFast();
 595         for (Int_t i = 0; i < np; i++) {
 596             TParticle* part = (TParticle*) fParticles->At(i);
 597             Int_t kf = part->GetPdgCode();
 598             Int_t ks = part->GetStatusCode();
 599             if (kf == 22 && ks == 40) {
 600                 Float_t phi = part->Phi();
 601                 Float_t eta = part->Eta();
 602                 if  (eta < fEtaMaxJet &&
 603                      eta > fEtaMinJet &&
 604                      phi < fPhiMaxJet &&
 605                      phi > fPhiMinJet) {
 606                     triggered = 1;
 607                     break;
 608                 } // check phi,eta within limits
 609             } // direct gamma ?
 610         } // particle loop
 611     } // fTrigger == 2
 612     return triggered;
 613 }
 614
 615
 616 void AliGenHijing::Copy(TObject &) const
 617 {
 618   Fatal("Copy","Not implemented!\n");
 619 }
 620
 621 AliGenHijing& AliGenHijing::operator=(const  AliGenHijing& rhs)
 622 {
 623     rhs.Copy(*this);
 624     return (*this);
 625 }
 626