Possibility to switch off heavy flavor production added.

[u/mrichter/AliRoot.git] / THijing / AliGenHijing.cxx

diff --git a/THijing/AliGenHijing.cxx b/THijing/AliGenHijing.cxx
index 017cda2003011b3624761316dac54e41223d1723..f456abceef2779c3ac6039b02c726c53f5a3ba3d 100644 (file)
--- a/THijing/AliGenHijing.cxx
+++ b/THijing/AliGenHijing.cxx
@@ -76,6 +76,8 @@ AliGenHijing::AliGenHijing(Int_t npart)
     //
     SetSimpleJets();
     SetNoGammas();
+    SetRandomPz();
+    SwitchOffHeavyQuarks(kFALSE);
 //
     fParticles = new TClonesArray("TParticle",10000);    
 //
@@ -151,6 +153,15 @@ void AliGenHijing::Init()
        fHijing->SetHIPR1(11, 2.5);
     }
     
+//
+// Heavy quarks
+//    
+    if (fNoHeavyQuarks) {
+       fHijing->SetIHPR2(49, 1);
+    } else {
+       fHijing->SetIHPR2(49, 0);
+    }
+    
     
     AliGenMC::Init();
     
@@ -190,12 +201,16 @@ void AliGenHijing::Generate()
       for (j=0; j < 3; j++) origin0[j] = fVertex[j];
   } 
 
+
+  Float_t sign = (fRandomPz && (Rndm() < 0.5))? -1. : 1.;
   while(1)
   {
 //    Generate one event
 // --------------------------------------------------------------------------
-      fSpecn   = 0;  
-      fSpecp   = 0;
+      fProjectileSpecn    = 0;  
+      fProjectileSpecp    = 0;
+      fTargetSpecn        = 0;  
+      fTargetSpecp        = 0;
 // --------------------------------------------------------------------------
       fHijing->GenerateEvent();
       fTrials++;
@@ -271,9 +286,13 @@ void AliGenHijing::Generate()
          
 // --------------------------------------------------------------------------
 // Count spectator neutrons and protons
-         if(ks == 0 || ks == 1 || ks == 10 || ks == 11){
-             if(kf == kNeutron) fSpecn += 1;
-             if(kf == kProton)  fSpecp += 1;
+         if(ks == 0 || ks == 1){
+             if(kf == kNeutron) fProjectileSpecn += 1;
+             if(kf == kProton)  fProjectileSpecp += 1;
+         }
+         else if(ks == 10 || ks == 11){
+             if(kf == kNeutron) fTargetSpecn += 1;
+             if(kf == kProton)  fTargetSpecp += 1;
          }
 // --------------------------------------------------------------------------
 //         
@@ -303,7 +322,7 @@ void AliGenHijing::Generate()
              ks   = iparticle->GetStatusCode();
              p[0] = iparticle->Px();
              p[1] = iparticle->Py();
-             p[2] = iparticle->Pz();
+             p[2] = iparticle->Pz() * sign;
              origin[0] = origin0[0]+iparticle->Vx()/10;
              origin[1] = origin0[1]+iparticle->Vy()/10;
              origin[2] = origin0[2]+iparticle->Vz()/10;
@@ -490,7 +509,8 @@ void AliGenHijing::MakeHeader()
                                                       fHijing->GetN01(),
                                                       fHijing->GetN10(),
                                                       fHijing->GetN11());
-    ((AliGenHijingEventHeader*) header)->SetSpectators(fSpecn, fSpecp);
+    ((AliGenHijingEventHeader*) header)->SetSpectators(fProjectileSpecn, fProjectileSpecp,
+                                                      fTargetSpecn,fTargetSpecp);
 
 // 4-momentum vectors of the triggered jets.
 //