]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PYTHIA6/AliGenPythia.cxx
kPyLhwgMb added.
[u/mrichter/AliRoot.git] / PYTHIA6 / AliGenPythia.cxx
index 0a90e0bd8fdb97f8e601be2d559a97322e5ec9bd..2bc4744bbcbc98eff9c456ec33450b6eb7f21786 100644 (file)
 
 ClassImp(AliGenPythia)
 
-AliGenPythia::AliGenPythia()
-                 :AliGenMC()
+
+AliGenPythia::AliGenPythia():
+    AliGenMC(),
+    fProcess(kPyCharm),          
+    fStrucFunc(kCTEQ5L), 
+    fEnergyCMS(5500.),
+    fKineBias(0.),
+    fTrials(0),
+    fTrialsRun(0),
+    fQ(0.),
+    fX1(0.),
+    fX2(0.),
+    fEventTime(0.),
+    fInteractionRate(0.),
+    fTimeWindow(0.),
+    fCurSubEvent(0),
+    fEventsTime(0),
+    fNev(0),
+    fFlavorSelect(0),
+    fXsection(0.),
+    fPythia(0),
+    fPtHardMin(0.),
+    fPtHardMax(1.e4),
+    fYHardMin(-1.e10),
+    fYHardMax(1.e10),
+    fGinit(1),
+    fGfinal(1),
+    fHadronisation(1),
+    fNpartons(0),
+    fReadFromFile(0),
+    fQuench(0),
+    fPtKick(1.),
+    fFullEvent(kTRUE),
+    fDecayer(new AliDecayerPythia()),
+    fDebugEventFirst(-1),
+    fDebugEventLast(-1),
+    fEtMinJet(0.),      
+    fEtMaxJet(1.e4),      
+    fEtaMinJet(-20.),     
+    fEtaMaxJet(20.),     
+    fPhiMinJet(0.),     
+    fPhiMaxJet(2.* TMath::Pi()),     
+    fJetReconstruction(kCell),
+    fEtaMinGamma(-20.),      
+    fEtaMaxGamma(20.),      
+    fPhiMinGamma(0.),      
+    fPhiMaxGamma(2. * TMath::Pi()),      
+    fPycellEtaMax(2.),     
+    fPycellNEta(274),       
+    fPycellNPhi(432),       
+    fPycellThreshold(0.),  
+    fPycellEtSeed(4.),     
+    fPycellMinEtJet(10.),  
+    fPycellMaxRadius(1.), 
+    fStackFillOpt(kFlavorSelection),   
+    fFeedDownOpt(kTRUE),    
+    fFragmentation(kTRUE),
+    fSetNuclei(kFALSE),
+    fNewMIS(kFALSE),   
+    fHFoff(kFALSE),    
+    fTriggerParticle(0),
+    fTriggerEta(0.9),     
+    fCountMode(kCountAll),      
+    fHeader(0),  
+    fRL(0),      
+    fFileName(0)
 {
 // Default Constructor
-  fParticles = 0;
-  fPythia    = 0;
-  fHeader = 0;
-  fReadFromFile = 0;
-  fEventTime = 0.;
-  fInteractionRate = 0.;
-  fTimeWindow = 0.;
-  fEventsTime = 0;
-  fCurSubEvent = 0;
-  fDecayer   = new AliDecayerPythia();
-  SetEventListRange();
-  SetJetPhiRange();
-  SetJetEtaRange();
-  SetJetEtRange();
-  SetGammaPhiRange();
-  SetGammaEtaRange();
-  SetPtKick();
-  SetQuench();
-  SetHadronisation();  
-  fSetNuclei = kFALSE;
-  fNewMIS    = kFALSE;
-  fHFoff     = kFALSE;
+  SetNuclei(0,0);
   if (!AliPythiaRndm::GetPythiaRandom()) 
-    AliPythiaRndm::SetPythiaRandom(GetRandom());
+      AliPythiaRndm::SetPythiaRandom(GetRandom());
 }
 
 AliGenPythia::AliGenPythia(Int_t npart)
-                 :AliGenMC(npart)
+    :AliGenMC(npart),
+     fProcess(kPyCharm),          
+     fStrucFunc(kCTEQ5L), 
+     fEnergyCMS(5500.),
+     fKineBias(0.),
+     fTrials(0),
+     fTrialsRun(0),
+     fQ(0.),
+     fX1(0.),
+     fX2(0.),
+     fEventTime(0.),
+     fInteractionRate(0.),
+     fTimeWindow(0.),
+     fCurSubEvent(0),
+     fEventsTime(0),
+     fNev(0),
+     fFlavorSelect(0),
+     fXsection(0.),
+     fPythia(0),
+     fPtHardMin(0.),
+     fPtHardMax(1.e4),
+     fYHardMin(-1.e10),
+     fYHardMax(1.e10),
+     fGinit(kTRUE),
+     fGfinal(kTRUE),
+     fHadronisation(kTRUE),
+     fNpartons(0),
+     fReadFromFile(kFALSE),
+     fQuench(kFALSE),
+     fPtKick(1.),
+     fFullEvent(kTRUE),
+     fDecayer(new AliDecayerPythia()),
+     fDebugEventFirst(-1),
+     fDebugEventLast(-1),
+     fEtMinJet(0.),      
+     fEtMaxJet(1.e4),      
+     fEtaMinJet(-20.),     
+     fEtaMaxJet(20.),     
+     fPhiMinJet(0.),     
+     fPhiMaxJet(2.* TMath::Pi()),     
+     fJetReconstruction(kCell),
+     fEtaMinGamma(-20.),      
+     fEtaMaxGamma(20.),      
+     fPhiMinGamma(0.),      
+     fPhiMaxGamma(2. * TMath::Pi()),      
+     fPycellEtaMax(2.),     
+     fPycellNEta(274),       
+     fPycellNPhi(432),       
+     fPycellThreshold(0.),  
+     fPycellEtSeed(4.),     
+     fPycellMinEtJet(10.),  
+     fPycellMaxRadius(1.), 
+     fStackFillOpt(kFlavorSelection),   
+     fFeedDownOpt(kTRUE),    
+     fFragmentation(kTRUE),
+     fSetNuclei(kFALSE),
+     fNewMIS(kFALSE),   
+     fHFoff(kFALSE),    
+     fTriggerParticle(0),
+     fTriggerEta(0.9),     
+     fCountMode(kCountAll),      
+     fHeader(0),  
+     fRL(0),      
+     fFileName(0)  
 {
 // default charm production at 5. 5 TeV
 // semimuonic decay
@@ -84,53 +190,79 @@ AliGenPythia::AliGenPythia(Int_t npart)
 //
     fName = "Pythia";
     fTitle= "Particle Generator using PYTHIA";
-    fXsection  = 0.;
-    fReadFromFile = 0;
-    fEventTime = 0.;
-    fInteractionRate = 0.;
-    fTimeWindow = 0.;
-    fEventsTime = 0;
-    fCurSubEvent = 0;
-    SetProcess();
-    SetStrucFunc();
     SetForceDecay();
-    SetPtHard();
-    SetYHard();
-    SetEnergyCMS();
-    fDecayer = new AliDecayerPythia();
     // Set random number generator 
     if (!AliPythiaRndm::GetPythiaRandom()) 
       AliPythiaRndm::SetPythiaRandom(GetRandom());
-    fFlavorSelect   = 0;
-    // Produced particles  
     fParticles = new TClonesArray("TParticle",1000);
-    fHeader = 0;
-    SetEventListRange();
-    SetJetPhiRange();
-    SetJetEtaRange();
-    SetJetEtRange();
-    SetGammaPhiRange();
-    SetGammaEtaRange();
-    SetJetReconstructionMode();
-    SetQuench();
-    SetHadronisation();
-    SetPtKick();
-    // Options determining what to keep in the stack (Heavy flavour generation)
-    fStackFillOpt = kFlavorSelection; // Keep particle with selected flavor
-    fFeedDownOpt = kTRUE;             // allow feed down from higher family
-    // Fragmentation on/off
-    fFragmentation = kTRUE;
-    // Default counting mode
-    fCountMode = kCountAll;
-    // Pycel
-    SetPycellParameters();
-    fSetNuclei = kFALSE;
-    fNewMIS    = kFALSE;
-    fHFoff     = kFALSE;
-}
+    SetNuclei(0,0);
+ }
 
 AliGenPythia::AliGenPythia(const AliGenPythia & Pythia)
-    :AliGenMC(Pythia)
+    :AliGenMC(Pythia),
+     fProcess(kPyCharm),          
+     fStrucFunc(kCTEQ5L), 
+     fEnergyCMS(5500.),
+     fKineBias(0.),
+     fTrials(0),
+     fTrialsRun(0),
+     fQ(0.),
+     fX1(0.),
+     fX2(0.),
+     fEventTime(0.),
+     fInteractionRate(0.),
+     fTimeWindow(0.),
+     fCurSubEvent(0),
+     fEventsTime(0),
+     fNev(0),
+     fFlavorSelect(0),
+     fXsection(0.),
+     fPythia(0),
+     fPtHardMin(0.),
+     fPtHardMax(1.e4),
+     fYHardMin(-1.e10),
+     fYHardMax(1.e10),
+     fGinit(kTRUE),
+     fGfinal(kTRUE),
+     fHadronisation(kTRUE),
+     fNpartons(0),
+     fReadFromFile(kFALSE),
+     fQuench(kFALSE),
+     fPtKick(1.),
+     fFullEvent(kTRUE),
+     fDecayer(new AliDecayerPythia()),
+     fDebugEventFirst(-1),
+     fDebugEventLast(-1),
+     fEtMinJet(0.),      
+     fEtMaxJet(1.e4),      
+     fEtaMinJet(-20.),     
+     fEtaMaxJet(20.),     
+     fPhiMinJet(0.),     
+     fPhiMaxJet(2.* TMath::Pi()),     
+     fJetReconstruction(kCell),
+     fEtaMinGamma(-20.),      
+     fEtaMaxGamma(20.),      
+     fPhiMinGamma(0.),      
+     fPhiMaxGamma(2. * TMath::Pi()),      
+     fPycellEtaMax(2.),     
+     fPycellNEta(274),       
+     fPycellNPhi(432),       
+     fPycellThreshold(0.),  
+     fPycellEtSeed(4.),     
+     fPycellMinEtJet(10.),  
+     fPycellMaxRadius(1.), 
+     fStackFillOpt(kFlavorSelection),   
+     fFeedDownOpt(kTRUE),    
+     fFragmentation(kTRUE),
+     fSetNuclei(kFALSE),
+     fNewMIS(kFALSE),   
+     fHFoff(kFALSE),    
+     fTriggerParticle(0),
+     fTriggerEta(0.9),     
+     fCountMode(kCountAll),      
+     fHeader(0),  
+     fRL(0),      
+     fFileName(0)  
 {
 // copy constructor
     Pythia.Copy(*this);
@@ -284,8 +416,9 @@ void AliGenPythia::Init()
     case kPyCharm:
     case kPyCharmUnforced:
     case kPyCharmPbPbMNR:
-    case kPyCharmppMNR:
     case kPyCharmpPbMNR:
+    case kPyCharmppMNR:
+    case kPyCharmppMNRwmi:
        fParentSelect[0] =   411;
        fParentSelect[1] =   421;
        fParentSelect[2] =   431;
@@ -304,10 +437,17 @@ void AliGenPythia::Init()
        fParentSelect[0] =   411;
        fFlavorSelect    =   4; 
        break;
+    case kPyDPlusStrangePbPbMNR:
+    case kPyDPlusStrangepPbMNR:
+    case kPyDPlusStrangeppMNR:
+       fParentSelect[0] =   431;
+       fFlavorSelect    =   4; 
+       break;
     case kPyBeauty:
     case kPyBeautyPbPbMNR:
     case kPyBeautypPbMNR:
     case kPyBeautyppMNR:
+    case kPyBeautyppMNRwmi:
        fParentSelect[0]=  511;
        fParentSelect[1]=  521;
        fParentSelect[2]=  531;
@@ -333,8 +473,10 @@ void AliGenPythia::Init()
        break;
     case kPyMb:
     case kPyMbNonDiffr:
+    case kPyMbMSEL1:
     case kPyJets:
     case kPyDirectGamma:
+    case kPyLhwgMb:    
        break;
     case kPyW:
     case kPyZ:
@@ -346,17 +488,17 @@ void AliGenPythia::Init()
 //
 //  Configure detector (EMCAL like)
 //
-       fPythia->SetPARU(51, fPycellEtaMax);
-       fPythia->SetMSTU(51, fPycellNEta);
-       fPythia->SetMSTU(52, fPycellNPhi);
+    fPythia->SetPARU(51, fPycellEtaMax);
+    fPythia->SetMSTU(51, fPycellNEta);
+    fPythia->SetMSTU(52, fPycellNPhi);
 //
 //  Configure Jet Finder
 //  
-       fPythia->SetPARU(58,  fPycellThreshold);
-       fPythia->SetPARU(52,  fPycellEtSeed);
-       fPythia->SetPARU(53,  fPycellMinEtJet);
-       fPythia->SetPARU(54,  fPycellMaxRadius);
-       fPythia->SetMSTU(54,  2);
+    fPythia->SetPARU(58,  fPycellThreshold);
+    fPythia->SetPARU(52,  fPycellEtSeed);
+    fPythia->SetPARU(53,  fPycellMinEtJet);
+    fPythia->SetPARU(54,  fPycellMaxRadius);
+    fPythia->SetMSTU(54,  2);
 //
 //  This counts the total number of calls to Pyevnt() per run.
     fTrialsRun = 0;
@@ -375,11 +517,22 @@ void AliGenPythia::Init()
        fDyBoost = 0;
        Warning("Init","SetNuclei used. Use SetProjectile + SetTarget instead. fDyBoost has been reset to 0\n");
     }
-
+    
     if (fQuench) {
        fPythia->InitQuenching(0., 0.1, 0.6e6, 0);
     }
-    
+    fPythia->SetPARJ(200, 0.0);
+
+    if (fQuench == 3) {
+       // Nestor's change of the splittings
+       fPythia->SetPARJ(200, 0.8);
+       fPythia->SetMSTJ(41, 1);  // QCD radiation only
+       fPythia->SetMSTJ(42, 2);  // angular ordering
+       fPythia->SetMSTJ(44, 2);  // option to run alpha_s
+       fPythia->SetMSTJ(47, 0);  // No correction back to hard scattering element
+       fPythia->SetMSTJ(50, 0);  // No coherence in first branching
+       fPythia->SetPARJ(82, 1.); // Cut off for parton showers
+    }
 }
 
 void AliGenPythia::Generate()
@@ -440,7 +593,10 @@ void AliGenPythia::Generate()
            fPythia->Quench();
        } else if (fQuench == 2){
            fPythia->Pyquen(208., 0, 0.);
+       } else if (fQuench == 3) {
+           // Quenching is via multiplicative correction of the splittings
        }
+       
 //
 // Switch hadronisation on
 //
@@ -479,7 +635,9 @@ void AliGenPythia::Generate()
        if (fProcess != kPyMb && fProcess != kPyJets && 
            fProcess != kPyDirectGamma &&
            fProcess != kPyMbNonDiffr  &&
-           fProcess != kPyW && fProcess != kPyZ ) {
+           fProcess != kPyMbMSEL1     &&
+           fProcess != kPyW && fProcess != kPyZ &&
+           fProcess != kPyCharmppMNRwmi && fProcess != kPyBeautyppMNRwmi) {
            
            for (i = 0; i < np; i++) {
                TParticle* iparticle = (TParticle *) fParticles->At(i);
@@ -647,9 +805,9 @@ void AliGenPythia::Generate()
        
        GetSubEventTime();
 
-       if (pParent)   delete[] pParent;
-       if (pSelected) delete[] pSelected;
-       if (trackIt)   delete[] trackIt;
+       delete[] pParent;
+       delete[] pSelected;
+       delete[] trackIt;
 
        if (nc > 0) {
          switch (fCountMode) {
@@ -668,8 +826,7 @@ void AliGenPythia::Generate()
          }
            if (jev >= fNpart || fNpart == -1) {
                fKineBias=Float_t(fNpart)/Float_t(fTrials);
-               printf("\n Trials: %i %i %i\n",fTrials, fNpart, jev);
-
+               
                fQ  += fPythia->GetVINT(51);
                fX1 += fPythia->GetVINT(41);
                fX2 += fPythia->GetVINT(42);
@@ -705,20 +862,63 @@ Int_t  AliGenPythia::GenerateMB()
     Int_t np = (fHadronisation) ? fParticles->GetEntriesFast() : fNpartons;
 
 
+
     Int_t* pParent = new Int_t[np];
     for (i=0; i< np; i++) pParent[i] = -1;
     if (fProcess == kPyJets || fProcess == kPyDirectGamma) {
        TParticle* jet1 = (TParticle *) fParticles->At(6);
        TParticle* jet2 = (TParticle *) fParticles->At(7);
-       if (!CheckTrigger(jet1, jet2)) return 0;
+       if (!CheckTrigger(jet1, jet2)) {
+         delete [] pParent;
+         return 0;
+       }
+    }
+
+    if (fTriggerParticle) {
+       Bool_t triggered = kFALSE;
+       for (i = 0; i < np; i++) {
+           TParticle *  iparticle = (TParticle *) fParticles->At(i);
+           kf = CheckPDGCode(iparticle->GetPdgCode());
+           if (kf != fTriggerParticle) continue;
+           if (iparticle->Pt() == 0.) continue;
+           if (TMath::Abs(iparticle->Eta()) > fTriggerEta) continue;
+           triggered = kTRUE;
+           break;
+       }
+       if (!triggered) {
+         delete [] pParent;
+         return 0;
+       }
+    }
+       
+
+    // Check if there is a ccbar or bbbar pair with at least one of the two
+    // in fYMin < y < fYMax
+    if (fProcess == kPyCharmppMNRwmi || fProcess == kPyBeautyppMNRwmi) {
+      TParticle *hvq;
+      Bool_t  theQ=kFALSE,theQbar=kFALSE,inYcut=kFALSE;
+      Float_t yQ;  
+      Int_t   pdgQ;
+      for(i=0; i<np; i++) {
+       hvq = (TParticle*)fParticles->At(i);
+       pdgQ = hvq->GetPdgCode();  
+       if(TMath::Abs(pdgQ) != fFlavorSelect) continue; 
+       if(pdgQ>0) { theQ=kTRUE; } else { theQbar=kTRUE; }
+       yQ = 0.5*TMath::Log((hvq->Energy()+hvq->Pz()+1.e-13)/
+                           (hvq->Energy()-hvq->Pz()+1.e-13));
+       if(yQ>fYMin && yQ<fYMax) inYcut=kTRUE;
+      }
+      if (!theQ || !theQbar || !inYcut) {
+       delete[] pParent;
+       return 0;
+      }
     }
-    
 
     //Introducing child cuts in case kPyW, kPyZ, kPyMb, and kPyMbNonDiff
     if ( (fProcess == kPyW || fProcess == kPyZ || fProcess == kPyMb || fProcess == kPyMbNonDiffr)  
         && (fCutOnChild == 1) ) {
       if ( !CheckKinematicsOnChild() ) {
-       if (pParent)   delete[] pParent;
+       delete[] pParent;
        return 0;
       }
     }
@@ -779,9 +979,8 @@ Int_t  AliGenPythia::GenerateMB()
        } // select particle
     } // particle loop 
 
-    if (pParent) delete[] pParent;
+    delete[] pParent;
     
-    printf("\n I've put %i particles on the stack \n",nc);
     return 1;
 }
 
@@ -801,7 +1000,7 @@ void AliGenPythia::FinishRun()
     printf("\nMean Q, x1, x2: %f %f %f\n", fQ, fX1, fX2);
 }
 
-void AliGenPythia::AdjustWeights()
+void AliGenPythia::AdjustWeights() const
 {
 // Adjust the weights after generation of all events
 //
@@ -827,6 +1026,9 @@ void AliGenPythia::SetNuclei(Int_t a1, Int_t a2)
 
 void AliGenPythia::MakeHeader()
 {
+//
+// Make header for the simulated event
+// 
   if (gAlice) {
     if (gAlice->GetEvNumber()>=fDebugEventFirst &&
        gAlice->GetEvNumber()<=fDebugEventLast) fPythia->Pylist(2);
@@ -942,7 +1144,7 @@ Bool_t AliGenPythia::CheckTrigger(TParticle* jet1, TParticle* jet2)
 //
        GetJets(njets, ntrig, jets);
        
-       if (ntrig) triggered = kTRUE;
+       if (ntrig || fEtMinJet == 0.) triggered = kTRUE;
 //
     } else {
        Int_t ij = 0;
@@ -967,9 +1169,11 @@ Bool_t AliGenPythia::CheckTrigger(TParticle* jet1, TParticle* jet2)
 }
 
 
-//Checking Kinematics on Child (status code 1, particle code ?, kin cuts
-Bool_t AliGenPythia::CheckKinematicsOnChild(){
 
+Bool_t AliGenPythia::CheckKinematicsOnChild(){
+//
+//Checking Kinematics on Child (status code 1, particle code ?, kin cuts
+//
     Bool_t checking = kFALSE;
     Int_t j, kcode, ks, km;
     Int_t nPartAcc = 0; //number of particles in the acceptance range
@@ -1021,12 +1225,12 @@ void  AliGenPythia::LoadEvent(AliStack* stack, Int_t flag, Int_t reHadr)
     
     
     for (Int_t part = 0; part < npart; part++) {
-       TParticle *MPart = stack->Particle(part);
+       TParticle *mPart = stack->Particle(part);
        
-       Int_t kf     =  MPart->GetPdgCode();
-       Int_t ks     =  MPart->GetStatusCode();
-       Int_t idf    =  MPart->GetFirstDaughter();
-       Int_t idl    =  MPart->GetLastDaughter();
+       Int_t kf     =  mPart->GetPdgCode();
+       Int_t ks     =  mPart->GetStatusCode();
+       Int_t idf    =  mPart->GetFirstDaughter();
+       Int_t idl    =  mPart->GetLastDaughter();
        
        if (reHadr) {
            if (ks == 11 || ks == 12) {
@@ -1036,11 +1240,11 @@ void  AliGenPythia::LoadEvent(AliStack* stack, Int_t flag, Int_t reHadr)
            }
        }
        
-       Float_t px = MPart->Px();
-       Float_t py = MPart->Py();
-       Float_t pz = MPart->Pz();
-       Float_t e  = MPart->Energy();
-       Float_t m  = MPart->GetCalcMass();
+       Float_t px = mPart->Px();
+       Float_t py = mPart->Py();
+       Float_t pz = mPart->Pz();
+       Float_t e  = mPart->Energy();
+       Float_t m  = mPart->GetCalcMass();
        
        
        (fPythia->GetPyjets())->P[0][part+n0] = px;
@@ -1053,7 +1257,7 @@ void  AliGenPythia::LoadEvent(AliStack* stack, Int_t flag, Int_t reHadr)
        (fPythia->GetPyjets())->K[0][part+n0] = ks;
        (fPythia->GetPyjets())->K[3][part+n0] = idf + 1;
        (fPythia->GetPyjets())->K[4][part+n0] = idl + 1;
-       (fPythia->GetPyjets())->K[2][part+n0] = MPart->GetFirstMother() + 1;
+       (fPythia->GetPyjets())->K[2][part+n0] = mPart->GetFirstMother() + 1;
     }
 }