1) decouple the rapidity cut used in AliGenPythia to select the events with c or...

[u/mrichter/AliRoot.git] / PYTHIA6 / AliGenPythiaPlus.cxx

diff --git a/PYTHIA6/AliGenPythiaPlus.cxx b/PYTHIA6/AliGenPythiaPlus.cxx
index 0f91b97fcce8f5933e52f5a29aea32c3347d784c..76b6fce9fd9e8ce1b8b530bd4b589a15329016db 100644 (file)
--- a/PYTHIA6/AliGenPythiaPlus.cxx
+++ b/PYTHIA6/AliGenPythiaPlus.cxx
@@ -92,6 +92,9 @@ AliGenPythiaPlus::AliGenPythiaPlus():
     fEtaMaxGamma(20.),      
     fPhiMinGamma(0.),      
     fPhiMaxGamma(2. * TMath::Pi()),      
+    fUseYCutHQ(kFALSE),
+    fYMinHQ(-20.),
+    fYMaxHQ(20.),
     fPycellEtaMax(2.),     
     fPycellNEta(274),       
     fPycellNPhi(432),       
@@ -180,6 +183,9 @@ AliGenPythiaPlus::AliGenPythiaPlus(AliPythiaBase* pythia)
      fEtaMaxGamma(20.),      
      fPhiMinGamma(0.),      
      fPhiMaxGamma(2. * TMath::Pi()),      
+     fUseYCutHQ(kFALSE),
+     fYMinHQ(-20.),
+     fYMaxHQ(20.),
      fPycellEtaMax(2.),     
      fPycellNEta(274),       
      fPycellNPhi(432),       
@@ -949,7 +955,8 @@ Int_t  AliGenPythiaPlus::GenerateMB()
          if(pdg>0) { theQ=kTRUE; } else { theQbar=kTRUE; }
          y = 0.5*TMath::Log((partCheck->Energy()+partCheck->Pz()+1.e-13)/
                             (partCheck->Energy()-partCheck->Pz()+1.e-13));
-         if(y>fYMin && y<fYMax) inYcut=kTRUE;
+         if(fUseYCutHQ && y>fYMinHQ && y<fYMaxHQ) inYcut=kTRUE;
+         if(!fUseYCutHQ && y>fYMin && y<fYMax) inYcut=kTRUE;
        }
 
        if(fCutOnChild && TMath::Abs(pdg) == fPdgCodeParticleforAcceptanceCut) {
@@ -1113,6 +1120,8 @@ void AliGenPythiaPlus::MakeHeader()
 // Builds the event header, to be called after each event
     if (fHeader) delete fHeader;
     fHeader = new AliGenPythiaEventHeader("Pythia");
+    fHeader->SetTitle(GetTitle());
+
 //
 // Event type  
     ((AliGenPythiaEventHeader*) fHeader)->SetProcessType(fPythia->ProcessCode());