]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - EVGEN/AliGenHijing.cxx
stream.h changed to iostream.h
[u/mrichter/AliRoot.git] / EVGEN / AliGenHijing.cxx
index fba6a3adf9903b21e06b183d6ba5d047cd128380..17b37c277e683d25ace4c7980c197c26dd6f4076 100644 (file)
 
 /*
 $Log$
+Revision 1.40  2002/02/12 11:05:53  morsch
+Get daughter indices right.
+
+Revision 1.39  2002/02/12 09:16:39  morsch
+Correction in SelectFlavor()
+
+Revision 1.38  2002/02/12 08:53:21  morsch
+SetNoGammas can be used to inhibit writing of gammas and pi0.
+
+Revision 1.37  2002/02/08 16:50:50  morsch
+Add name and title in constructor.
+
+Revision 1.36  2002/01/31 20:17:55  morsch
+Allow for triggered jets with simplified topology: Exact pT, back-to-back
+
+Revision 1.35  2001/12/13 07:56:25  hristov
+Set pointers to zero in the default constructor
+
+Revision 1.34  2001/12/11 16:55:42  morsch
+Correct initialization for jet phi-range.
+
+Revision 1.33  2001/12/05 10:18:51  morsch
+Possibility of kinematic biasing of jet phi range. (J. Klay)
+
+Revision 1.32  2001/11/28 13:51:11  morsch
+Introduce kinematic biasing (etamin, etamax) of jet trigger. Bookkeeping
+(number of trials) done in AliGenHijingEventHeader.
+
+Revision 1.31  2001/11/06 12:30:34  morsch
+Add Boost() method to boost all particles to LHC lab frame. Needed for asymmetric collision systems.
+
+Revision 1.30  2001/10/21 18:35:56  hristov
+Several pointers were set to zero in the default constructors to avoid memory management problems
+
+Revision 1.29  2001/10/15 08:12:24  morsch
+- Vertex smearing with truncated gaussian.
+- Store triggered jet info before and after final state radiation into mc-heade
+
+Revision 1.28  2001/10/08 11:55:25  morsch
+Store 4-momenta of trigegred jets in event header.
+Possibility to switch of initial and final state radiation.
+
 Revision 1.27  2001/10/08 07:13:14  morsch
 Add setter for minimum transverse momentum of triggered jet.
 
@@ -133,6 +175,9 @@ AliGenHijing::AliGenHijing()
                  :AliGenMC()
 {
 // Constructor
+    fHijing = 0;
+    fDsigmaDb = 0;
+    fDnDb = 0;
 }
 
 AliGenHijing::AliGenHijing(Int_t npart)
@@ -140,10 +185,17 @@ AliGenHijing::AliGenHijing(Int_t npart)
 {
 // Default PbPb collisions at 5. 5 TeV
 //
+    fName = "Hijing";
+    fTitle= "Particle Generator using HIJING";
+
     SetEnergyCMS();
     SetImpactParameterRange();
     SetTarget();
     SetProjectile();
+    SetBoostLHC();
+    SetJetEtaRange();
+    SetJetPhiRange();
+    
     fKeep       =  0;
     fQuench     =  1;
     fShadowing  =  1;
@@ -157,9 +209,16 @@ AliGenHijing::AliGenHijing(Int_t npart)
     fDnDb       =  0;
     fPtMinJet   = -2.5;        
     fRadiation  =  1;
+    fEventVertex.Set(3);
+//
+    SetSimpleJets();
+    SetNoGammas();
+    
 //
 // Set random number generator   
     sRandom = fRandom;
+    fHijing = 0;
+
 }
 
 AliGenHijing::AliGenHijing(const AliGenHijing & Hijing)
@@ -189,14 +248,49 @@ void AliGenHijing::Init()
     fHijing=(THijing*) fgMCEvGen;
     fHijing->SetIHPR2(2,  fRadiation);
     fHijing->SetIHPR2(3,  fTrigger);
-    fHijing->SetIHPR2(4,  fQuench);
     fHijing->SetIHPR2(6,  fShadowing);
     fHijing->SetIHPR2(12, fDecaysOff);    
     fHijing->SetIHPR2(21, fKeep);
     fHijing->SetHIPR1(10, fPtMinJet);  
-    fHijing->Initialize();
-
+    fHijing->SetHIPR1(50, fSimpleJet);
+//
+//  Quenching
+//
+//
+//  fQuench = 0:  no quenching
+//  fQuench = 1:  hijing default
+//  fQuench = 2:  new LHC  parameters for HIPR1(11) and HIPR1(14)
+//  fQuench = 3:  new RHIC parameters for HIPR1(11) and HIPR1(14)
+//  fQuench = 4:  new LHC  parameters with log(e) dependence
+//  fQuench = 5:  new RHIC parameters with log(e) dependence
+    fHijing->SetIHPR2(50, 0);
+    if (fQuench > 0) 
+       fHijing->SetIHPR2(4,  1);
+    else
+       fHijing->SetIHPR2(4,  0);
+// New LHC parameters from Xin-Nian Wang
+    if (fQuench == 2) {
+       fHijing->SetHIPR1(14, 1.1);
+       fHijing->SetHIPR1(11, 3.7);
+    } else if (fQuench == 3) {
+       fHijing->SetHIPR1(14, 0.20);
+       fHijing->SetHIPR1(11, 2.5);
+    } else if (fQuench == 4) {
+       fHijing->SetIHPR2(50, 1);
+       fHijing->SetHIPR1(14, 4.*0.34);
+       fHijing->SetHIPR1(11, 3.7);
+    } else if (fQuench == 5) {
+       fHijing->SetIHPR2(50, 1);
+       fHijing->SetHIPR1(14, 0.34);
+       fHijing->SetHIPR1(11, 2.5);
+    }
+    
     
+    
+//
+//  Initialize Hijing  
+//    
+    fHijing->Initialize();
 //
     if (fEvaluate) EvaluateCrossSections();
 //
@@ -228,11 +322,16 @@ void AliGenHijing::Generate()
   fTrials = 0;
   for (j = 0;j < 3; j++) origin0[j] = fOrigin[j];
   if(fVertexSmear == kPerEvent) {
-       Rndm(random,6);
-       for (j=0; j < 3; j++) {
-           origin0[j] += fOsigma[j]*TMath::Cos(2*random[2*j]*TMath::Pi())*
-               TMath::Sqrt(-2*TMath::Log(random[2*j+1]));
-       }
+      Float_t dv[3];
+      dv[2] = 1.e10;
+      while(TMath::Abs(dv[2]) > fCutVertexZ*fOsigma[2]) {
+         Rndm(random,6);
+         for (j=0; j < 3; j++) {
+             dv[j] = fOsigma[j]*TMath::Cos(2*random[2*j]*TMath::Pi())*
+                 TMath::Sqrt(-2*TMath::Log(random[2*j+1]));
+         }
+      }
+      for (j=0; j < 3; j++) origin0[j] += dv[j];
   } else if (fVertexSmear == kPerTrack) {
 //         fHijing->SetMSTP(151,0);
       for (j = 0; j < 3; j++) {
@@ -248,120 +347,130 @@ void AliGenHijing::Generate()
 // --------------------------------------------------------------------------
       fHijing->GenerateEvent();
       fTrials++;
+      if (fTrigger != kNoTrigger) {
+         if (!CheckTrigger()) continue;
+      }
+
       fHijing->ImportParticles(particles,"All");
+      if (fLHC) Boost(particles);
+      
+      
       Int_t np = particles->GetEntriesFast();
       printf("\n **************************************************%d\n",np);
       Int_t nc = 0;
       if (np == 0 ) continue;
       Int_t i;
-      Int_t * newPos = new Int_t[np];
+      Int_t* newPos     = new Int_t[np];
+      Int_t* pSelected  = new Int_t[np];
 
-      for (i = 0; i < np; i++) *(newPos+i) = i;
+      for (i = 0; i < np; i++) {
+         newPos[i]    = i;
+         pSelected[i] = 0;
+      }
+      
+//      Get event vertex
 //
-//      First write parent particles
+      TParticle *  iparticle = (TParticle *) particles->At(0);
+      fEventVertex[0] = origin0[0];
+      fEventVertex[1] = origin0[1];    
+      fEventVertex[2] = origin0[2];
+      
+//
+//      First select parent particles
 //
 
       for (i = 0; i < np; i++) {
-         TParticle *  iparticle       = (TParticle *) particles->At(i);
+         iparticle = (TParticle *) particles->At(i);
 // Is this a parent particle ?
-        if (Stable(iparticle)) continue;
+         if (Stable(iparticle)) continue;
 //
-        Bool_t  hasMother            =  (iparticle->GetFirstMother()   >=0);
-        Bool_t  selected             =  kTRUE;
-        Bool_t  hasSelectedDaughters =  kFALSE;
-           
-           
-        kf        = iparticle->GetPdgCode();
-        ks        = iparticle->GetStatusCode();
-        if (kf == 92) continue;
-           
-        if (!fSelectAll) selected = KinematicSelection(iparticle, 0)&&SelectFlavor(kf);
-        hasSelectedDaughters = DaughtersSelection(iparticle, particles);
-//
-// Put particle on the stack if it is either selected or it is the mother of at least one seleted particle
-//
-        if (selected || hasSelectedDaughters) {
-           nc++;
-           p[0] = iparticle->Px();
-           p[1] = iparticle->Py();
-           p[2] = iparticle->Pz();
-           origin[0] = origin0[0]+iparticle->Vx()/10;
-           origin[1] = origin0[1]+iparticle->Vy()/10;
-           origin[2] = origin0[2]+iparticle->Vz()/10;
-           tof = kconv*iparticle->T();
-           imo = -1;
-           TParticle* mother = 0;
-           if (hasMother) {
-               imo = iparticle->GetFirstMother();
-               mother = (TParticle *) particles->At(imo);
-               imo = (mother->GetPdgCode() != 92) ? imo =* (newPos+imo) : -1;
-               
-           }
-// Put particle on the stack ... 
-//             printf("\n set track mother: %d %d %d %d %d %d ",i,imo, kf, nt+1, selected, hasSelectedDaughters);
+         Bool_t  selected             =  kTRUE;
+         Bool_t  hasSelectedDaughters =  kFALSE;
+         
+         
+         kf        = iparticle->GetPdgCode();
+         ks        = iparticle->GetStatusCode();
+         if (kf == 92) continue;
            
-           SetTrack(0,imo,kf,p,origin,polar, tof,kPPrimary,nt);
-// ... and keep it there
-           KeepTrack(nt);
+         if (!fSelectAll) selected = KinematicSelection(iparticle, 0) && 
+                              SelectFlavor(kf);
+         hasSelectedDaughters = DaughtersSelection(iparticle, particles);
 //
-           *(newPos+i)=nt;
-        } // selected
+// Put particle on the stack if it is either selected or 
+// it is the mother of at least one seleted particle
+//
+         if (selected || hasSelectedDaughters) {
+             nc++;
+             pSelected[i] = 1;
+         } // selected
       } // particle loop parents
 //
-// Now write the final state particles
+// Now select the final state particles
 //
 
       for (i = 0; i<np; i++) {
-        TParticle *  iparticle       = (TParticle *) particles->At(i);
+         TParticle *  iparticle = (TParticle *) particles->At(i);
 // Is this a final state particle ?
-        if (!Stable(iparticle)) continue;
-
-        Bool_t  hasMother            =  (iparticle->GetFirstMother()   >=0);
-        Bool_t  selected             =  kTRUE;
-        kf        = iparticle->GetPdgCode();
-        ks        = iparticle->GetStatusCode();
+         if (!Stable(iparticle)) continue;
+      
+         Bool_t  selected             =  kTRUE;
+         kf        = iparticle->GetPdgCode();
+         ks        = iparticle->GetStatusCode();
 // --------------------------------------------------------------------------
 // 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 || ks == 10 || ks == 11){
+             if(kf == kNeutron) fSpecn += 1;
+             if(kf == kProton)  fSpecp += 1;
+         }
 // --------------------------------------------------------------------------
 //         
-        if (!fSelectAll) {
-           selected = KinematicSelection(iparticle,0)&&SelectFlavor(kf);
-           if (!fSpectators && selected) selected = (ks != 0 && ks != 1 && ks != 10
-                                                     && ks != 11);
-        }
+         if (!fSelectAll) {
+             selected = KinematicSelection(iparticle,0)&&SelectFlavor(kf);
+             if (!fSpectators && selected) selected = (ks != 0 && ks != 1 && ks != 10
+                                                       && ks != 11);
+         }
 //
 // Put particle on the stack if selected
 //
-        if (selected) {
-           nc++;
-           p[0] = iparticle->Px();
-           p[1] = iparticle->Py();
-           p[2] = iparticle->Pz();
-           origin[0] = origin0[0]+iparticle->Vx()/10;
-           origin[1] = origin0[1]+iparticle->Vy()/10;
-           origin[2] = origin0[2]+iparticle->Vz()/10;
-           tof = kconv*iparticle->T();
-           imo = -1;
-           TParticle* mother = 0;
-           if (hasMother) {
-               imo = iparticle->GetFirstMother();
-               mother = (TParticle *) particles->At(imo);
-               imo = (mother->GetPdgCode() != 92) ? imo=*(newPos+imo) : -1;
-           }   
-// Put particle on the stack
-           SetTrack(fTrackIt,imo,kf,p,origin,polar,
-                                                    tof,kPNoProcess,nt);
-           KeepTrack(nt);
-           *(newPos+i)=nt;
-        } // selected
+         if (selected) {
+             nc++;
+             pSelected[i] = 1;
+         } // selected
       } // particle loop final state
+//
+// Write particles to stack
+//
+      for (i = 0; i<np; i++) {
+         TParticle *  iparticle = (TParticle *) particles->At(i);
+         Bool_t  hasMother   = (iparticle->GetFirstMother()     >=0);
+         Bool_t  hasDaughter = (iparticle->GetFirstDaughter()   >=0);
+
+         if (pSelected[i]) {
+             kf   = iparticle->GetPdgCode();
+             p[0] = iparticle->Px();
+             p[1] = iparticle->Py();
+             p[2] = iparticle->Pz();
+             origin[0] = origin0[0]+iparticle->Vx()/10;
+             origin[1] = origin0[1]+iparticle->Vy()/10;
+             origin[2] = origin0[2]+iparticle->Vz()/10;
+             tof = kconv*iparticle->T();
+             imo = -1;
+             TParticle* mother = 0;
+             if (hasMother) {
+                 imo = iparticle->GetFirstMother();
+                 mother = (TParticle *) particles->At(imo);
+                 imo = (mother->GetPdgCode() != 92) ? imo = newPos[imo] : -1;
+             } // if has mother   
+             Bool_t tFlag = (fTrackIt && !hasDaughter);
+             SetTrack(tFlag,imo,kf,p,origin,polar,
+                      tof,kPNoProcess,nt);
+             KeepTrack(nt);
+             newPos[i] = nt;
+         } // if selected
+      } // particle loop
       delete[] newPos;
-
+      delete[] pSelected;
+      
       printf("\n I've put %i particles on the stack \n",nc);
       if (nc > 0) {
          jev += nc;
@@ -373,7 +482,6 @@ void AliGenHijing::Generate()
       }
   } // event loop
   MakeHeader();
-  
   SetHighWaterMark(nt);
 }
 
@@ -481,17 +589,28 @@ Bool_t AliGenHijing::SelectFlavor(Int_t pid)
 // 0: all
 // 4: charm and beauty
 // 5: beauty
-    if (fFlavor == 0) return kTRUE;
+    Bool_t res = 0;
     
-    Int_t ifl = TMath::Abs(pid/100);
-    if (ifl > 10) ifl/=10;
-    return (fFlavor == ifl);
+    if (fFlavor == 0) {
+       res = kTRUE;
+    } else {
+       Int_t ifl = TMath::Abs(pid/100);
+       if (ifl > 10) ifl/=10;
+       res = (fFlavor == ifl);
+    }
+//
+//  This part if gamma writing is inhibited
+    if (fNoGammas) 
+       res = res && (pid != kGamma && pid != kPi0);
+//
+    return res;
 }
 
 Bool_t AliGenHijing::Stable(TParticle*  particle)
 {
 // Return true for a stable particle
 //
+    
     if (particle->GetFirstDaughter() < 0 )
     {
        return kTRUE;
@@ -500,6 +619,39 @@ Bool_t AliGenHijing::Stable(TParticle*  particle)
     }
 }
 
+
+void AliGenHijing::Boost(TClonesArray* particles)
+{
+//
+// Boost cms into LHC lab frame
+//
+    Double_t dy    = - 0.5 * TMath::Log(Double_t(fZProjectile) * Double_t(fATarget) / 
+                                     (Double_t(fZTarget)    * Double_t(fAProjectile)));
+    Double_t beta  = TMath::TanH(dy);
+    Double_t gamma = 1./TMath::Sqrt(1.-beta*beta);
+    Double_t gb    = gamma * beta;
+
+    printf("\n Boosting particles to lab frame %f %f %f", dy, beta, gamma);
+    
+    Int_t i;
+    Int_t np = particles->GetEntriesFast();
+    for (i = 0; i < np; i++) 
+    {
+       TParticle* iparticle = (TParticle*) particles->At(i);
+
+       Double_t e   = iparticle->Energy();
+       Double_t px  = iparticle->Px();
+       Double_t py  = iparticle->Py();
+       Double_t pz  = iparticle->Pz();
+
+       Double_t eb  = gamma * e -      gb * pz;
+       Double_t pzb =   -gb * e +   gamma * pz;
+
+       iparticle->SetMomentum(px, py, pzb, eb);
+    }
+}
+
+
 void AliGenHijing::MakeHeader()
 {
 // Builds the event header, to be called after each event
@@ -515,6 +667,9 @@ void AliGenHijing::MakeHeader()
                                                       fHijing->GetN11());
     ((AliGenHijingEventHeader*) header)->SetSpectators(fSpecn, fSpecp);
 
+// 4-momentum vectors of the triggered jets.
+//
+// Before final state gluon radiation.
     TLorentzVector* jet1 = new TLorentzVector(fHijing->GetHINT1(21), 
                                              fHijing->GetHINT1(22),
                                              fHijing->GetHINT1(23),
@@ -524,11 +679,63 @@ void AliGenHijing::MakeHeader()
                                              fHijing->GetHINT1(32),
                                              fHijing->GetHINT1(33),
                                              fHijing->GetHINT1(34));
-
-    ((AliGenHijingEventHeader*) header)->SetJets(jet1, jet2);
+// After final state gluon radiation.
+    TLorentzVector* jet3 = new TLorentzVector(fHijing->GetHINT1(26), 
+                                             fHijing->GetHINT1(27),
+                                             fHijing->GetHINT1(28),
+                                             fHijing->GetHINT1(29));
+
+    TLorentzVector* jet4 = new TLorentzVector(fHijing->GetHINT1(36), 
+                                             fHijing->GetHINT1(37),
+                                             fHijing->GetHINT1(38),
+                                             fHijing->GetHINT1(39));
+    ((AliGenHijingEventHeader*) header)->SetJets(jet1, jet2, jet3, jet4);
+// Bookkeeping for kinematic bias
+    ((AliGenHijingEventHeader*) header)->SetTrials(fTrials);
+// Event Vertex
+    header->SetPrimaryVertex(fEventVertex);
     gAlice->SetGenEventHeader(header);    
 }
 
+Bool_t AliGenHijing::CheckTrigger()
+{
+// Check the kinematic trigger condition
+// 
+    TLorentzVector* jet1 = new TLorentzVector(fHijing->GetHINT1(26), 
+                                             fHijing->GetHINT1(27),
+                                             fHijing->GetHINT1(28),
+                                             fHijing->GetHINT1(29));
+
+    TLorentzVector* jet2 = new TLorentzVector(fHijing->GetHINT1(36), 
+                                             fHijing->GetHINT1(37),
+                                             fHijing->GetHINT1(38),
+                                             fHijing->GetHINT1(39));
+    Double_t eta1      = jet1->Eta();
+    Double_t eta2      = jet2->Eta();
+    Double_t phi1      = jet1->Phi();
+    Double_t phi2      = jet2->Phi();
+    Bool_t   triggered = kFALSE;
+//    printf("\n Trigger: %f %f %f %f",
+//        fEtaMinJet, fEtaMaxJet, fPhiMinJet, fPhiMaxJet);
+//    printf("\n Jet1: %f %f", phi1, eta1);
+//    printf("\n Jet2: %f %f", phi2, eta2);
+
+    
+    if (
+       (eta1 < fEtaMaxJet && eta1 > fEtaMinJet &&  
+        phi1 < fPhiMaxJet && phi1 > fPhiMinJet) 
+       ||
+       (eta2 < fEtaMaxJet && eta2 > fEtaMinJet &&  
+        phi2 < fPhiMaxJet && phi2 > fPhiMinJet)
+       ) 
+       triggered = kTRUE;
+
+    return triggered;
+}
+
+
+
+
 AliGenHijing& AliGenHijing::operator=(const  AliGenHijing& rhs)
 {
 // Assignment operator