]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PYTHIA6/AliGenPythia.cxx
Fix in order to eliminate SSD overlaps (E. Cattaruzza)
[u/mrichter/AliRoot.git] / PYTHIA6 / AliGenPythia.cxx
index bc9447fe90424281105cd6dc1d275f0f86d2ff62..649e123b03023e6ef1001b6e09582b5ecc154cd0 100644 (file)
@@ -42,7 +42,7 @@
 #include "AliStack.h"
 #include "AliRunLoader.h"
 #include "AliMC.h"
-#include "pyquenCommon.h"
+#include "PyquenCommon.h"
 
 ClassImp(AliGenPythia)
 
@@ -114,9 +114,12 @@ AliGenPythia::AliGenPythia():
     fFileName(0),
     fFragPhotonInCalo(kFALSE),
     fPi0InCalo(kFALSE) ,
+    fPhotonInCalo(kFALSE),
     fCheckEMCAL(kFALSE),
     fCheckPHOS(kFALSE),
+    fCheckPHOSeta(kFALSE),
     fFragPhotonOrPi0MinPt(0), 
+    fPhotonMinPt(0), 
     fPHOSMinPhi(219.),
     fPHOSMaxPhi(321.),
     fPHOSEta(0.13),
@@ -198,9 +201,12 @@ AliGenPythia::AliGenPythia(Int_t npart)
      fFileName(0),
      fFragPhotonInCalo(kFALSE),
      fPi0InCalo(kFALSE) ,
+     fPhotonInCalo(kFALSE),
      fCheckEMCAL(kFALSE),
      fCheckPHOS(kFALSE),
+     fCheckPHOSeta(kFALSE),
      fFragPhotonOrPi0MinPt(0),
+     fPhotonMinPt(0),
      fPHOSMinPhi(219.),
      fPHOSMaxPhi(321.),
      fPHOSEta(0.13),
@@ -312,9 +318,6 @@ void AliGenPythia::Init()
 //
     fParentWeight=1./Float_t(fNpart);
 //
-//  Forward Paramters to the AliPythia object
-    fDecayer->SetForceDecay(fForceDecay);    
-    fDecayer->Init();
 
 
     fPythia->SetCKIN(3,fPtHardMin);
@@ -338,7 +341,8 @@ void AliGenPythia::Init()
 //  pt - kick
     if (fPtKick > 0.) {
        fPythia->SetMSTP(91,1);
-       fPythia->SetPARP(91,fPtKick);
+       fPythia->SetPARP(91,fPtKick);   
+       fPythia->SetPARP(93, 4. * fPtKick);
     } else {
        fPythia->SetMSTP(91,0);
     }
@@ -351,14 +355,22 @@ void AliGenPythia::Init()
     } else {
        fRL = 0x0;
     }
+ //
+    fPythia->ProcInit(fProcess,fEnergyCMS,fStrucFunc);
+    //  Forward Paramters to the AliPythia object
+    fDecayer->SetForceDecay(fForceDecay);    
 // Switch off Heavy Flavors on request  
     if (fHFoff) {
+       // Maximum number of quark flavours used in pdf 
        fPythia->SetMSTP(58, 3);
+       // Maximum number of flavors that can be used in showers
        fPythia->SetMSTJ(45, 3);        
-       for (Int_t i = 156; i <= 160; i++) fPythia->SetMDME(i, 1, 0);
+       // Switch off g->QQbar splitting in decay table
+       ((AliDecayerPythia*) fDecayer)->HeavyFlavourOff();
     }
- //
-    fPythia->ProcInit(fProcess,fEnergyCMS,fStrucFunc);
+
+    fDecayer->Init();
+
 
 //  Parent and Children Selection
     switch (fProcess) 
@@ -425,6 +437,7 @@ void AliGenPythia::Init()
     case kPyJpsi:
        fParentSelect[0] = 443;
        break;
+    case kPyMbDefault:
     case kPyMb:
     case kPyMbNonDiffr:
     case kPyMbMSEL1:
@@ -567,7 +580,7 @@ void AliGenPythia::Generate()
 //
        Int_t i;
        
-
+       fNprimaries = 0;
        Int_t np = fParticles->GetEntriesFast();
        
        if (np == 0) continue;
@@ -586,12 +599,16 @@ void AliGenPythia::Generate()
        Int_t nc = 0;        // Total n. of selected particles
        Int_t nParents = 0;  // Selected parents
        Int_t nTkbles = 0;   // Trackable particles
-       if (fProcess != kPyMb && fProcess != kPyJets && 
+       if (fProcess != kPyMbDefault && 
+           fProcess != kPyMb && 
+           fProcess != kPyJets && 
            fProcess != kPyDirectGamma &&
            fProcess != kPyMbNonDiffr  &&
            fProcess != kPyMbMSEL1     &&
-           fProcess != kPyW && fProcess != kPyZ &&
-           fProcess != kPyCharmppMNRwmi && fProcess != kPyBeautyppMNRwmi) {
+           fProcess != kPyW && 
+           fProcess != kPyZ &&
+           fProcess != kPyCharmppMNRwmi && 
+           fProcess != kPyBeautyppMNRwmi) {
            
            for (i = 0; i < np; i++) {
                TParticle* iparticle = (TParticle *) fParticles->At(i);
@@ -750,7 +767,8 @@ void AliGenPythia::Generate()
                              polar[0], polar[1], polar[2],
                              kPPrimary, nt, 1., ks);
                    pParent[i] = nt;
-                   KeepTrack(nt); 
+                   KeepTrack(nt);
+                   fNprimaries++;
                } //  PushTrack loop
            }
        } else {
@@ -858,6 +876,41 @@ Int_t  AliGenPythia::GenerateMB()
        return 0;
     }
     
+    
+     // Select events with a photon  pt > min pt going to PHOS eta acceptance or exactly PHOS eta phi
+    if ((fProcess == kPyJets || fProcess == kPyDirectGamma) && fPhotonInCalo && (fCheckPHOSeta || fCheckPHOS)){
+
+      Bool_t okd = kFALSE;
+
+      Int_t pdg  = 22; 
+      Int_t iphcand = -1;
+      for (i=0; i< np; i++) {
+        TParticle* iparticle = (TParticle *) fParticles->At(i);
+        Float_t phi = iparticle->Phi()*180./TMath::Pi(); //Convert to degrees
+        Float_t eta =TMath::Abs(iparticle->Eta());//in calos etamin=-etamax 
+        
+        if(iparticle->GetStatusCode() == 1 
+           && iparticle->GetPdgCode() == pdg   
+           && iparticle->Pt() > fPhotonMinPt    
+           && eta < fPHOSEta){                 
+           
+           // first check if the photon is in PHOS phi
+           if(IsInPHOS(phi,eta)){ 
+               okd = kTRUE;
+               break;
+           } 
+           if(fCheckPHOSeta) iphcand = i; // candiate photon to rotate in phi
+            
+        }
+      }
+      
+      if(!okd && iphcand != -1) // execute rotation in phi 
+          RotatePhi(iphcand,okd);
+      
+      if(!okd)
+       return 0;
+    }
+    
     if (fTriggerParticle) {
        Bool_t triggered = kFALSE;
        for (i = 0; i < np; i++) {
@@ -899,7 +952,11 @@ Int_t  AliGenPythia::GenerateMB()
     }
 
     //Introducing child cuts in case kPyW, kPyZ, kPyMb, and kPyMbNonDiff
-    if ( (fProcess == kPyW || fProcess == kPyZ || fProcess == kPyMb || fProcess == kPyMbNonDiffr)  
+    if ( (fProcess == kPyW ||
+         fProcess == kPyZ ||
+         fProcess == kPyMbDefault ||
+         fProcess == kPyMb ||
+         fProcess == kPyMbNonDiffr)  
         && (fCutOnChild == 1) ) {
       if ( !CheckKinematicsOnChild() ) {
        delete[] pParent;
@@ -942,6 +999,7 @@ Int_t  AliGenPythia::GenerateMB()
                      origin[0], origin[1], origin[2], tof, 
                      polar[0], polar[1], polar[2],
                      kPPrimary, nt, 1., ks);
+           fNprimaries++;
            //
            // Special Treatment to store color-flow
            //
@@ -1030,6 +1088,10 @@ void AliGenPythia::MakeHeader()
 //
 // Event Vertex 
     fHeader->SetPrimaryVertex(fVertex);
+    
+//
+// Number of primaries
+    fHeader->SetNProduced(fNprimaries);
 //
 // Jets that have triggered
 
@@ -1091,19 +1153,9 @@ void AliGenPythia::MakeHeader()
 //  Pass header
 //
     AddHeader(fHeader);
+    fHeader = 0x0;
 }
 
-void AliGenPythia::AddHeader(AliGenEventHeader* header)
-{
-    // Add header to container or runloader
-    if (fContainer) {
-       fContainer->AddHeader(header);
-    } else {
-       AliRunLoader::GetRunLoader()->GetHeader()->SetGenEventHeader(header);   
-    }
-}
-
-
 Bool_t AliGenPythia::CheckTrigger(TParticle* jet1, TParticle* jet2)
 {
 // Check the kinematic trigger condition
@@ -1359,6 +1411,56 @@ Bool_t AliGenPythia::IsInPHOS(Float_t phi, Float_t eta)
     return kFALSE;
 }
 
+void AliGenPythia::RotatePhi(Int_t iphcand, Bool_t& okdd)
+{
+  //calculate the new position random between fPHOSMinPhi and fPHOSMaxPhi 
+  Double_t phiPHOSmin = TMath::Pi()*fPHOSMinPhi/180;
+  Double_t phiPHOSmax = TMath::Pi()*fPHOSMaxPhi/180;
+  Double_t phiPHOS = gRandom->Uniform(phiPHOSmin,phiPHOSmax);
+  
+  //calculate deltaphi
+  TParticle* ph = (TParticle *) fParticles->At(iphcand);
+  Double_t phphi = ph->Phi();
+  Double_t deltaphi = phiPHOS - phphi;
+
+  
+  
+  //loop for all particles and produce the phi rotation
+  Int_t np = (fHadronisation) ? fParticles->GetEntriesFast() : fNpartons;
+  Double_t oldphi, newphi;
+  Double_t newVx, newVy, R, Vz, time; 
+  Double_t newPx, newPy, pt, Pz, e;
+  for(Int_t i=0; i< np; i++) {
+      TParticle* iparticle = (TParticle *) fParticles->At(i);
+      oldphi = iparticle->Phi();
+      newphi = oldphi + deltaphi;
+      if(newphi < 0) newphi = 2*TMath::Pi() + newphi; // correct angle 
+      if(newphi > 2*TMath::Pi()) newphi = newphi - 2*TMath::Pi(); // correct angle
+      
+      R = iparticle->R();
+      newVx = R*TMath::Cos(newphi);
+      newVy = R*TMath::Sin(newphi);
+      Vz = iparticle->Vz(); // don't transform
+      time = iparticle->T(); // don't transform
+      
+      pt = iparticle->Pt();
+      newPx = pt*TMath::Cos(newphi);
+      newPy = pt*TMath::Sin(newphi);
+      Pz = iparticle->Pz(); // don't transform
+      e = iparticle->Energy(); // don't transform
+      
+      // apply rotation 
+      iparticle->SetProductionVertex(newVx, newVy, Vz, time);
+      iparticle->SetMomentum(newPx, newPy, Pz, e);
+      
+  } //end particle loop 
+  
+   // now let's check that we put correctly the candidate photon in PHOS
+   Float_t phi = ph->Phi()*180./TMath::Pi(); //Convert to degrees
+   Float_t eta =TMath::Abs(ph->Eta());//in calos etamin=-etamax 
+   if(IsInPHOS(phi,eta)) 
+      okdd = kTRUE;
+}
 
 
 #ifdef never