]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PYTHIA6/AliGenPythia.h
Option to switch pi0 decay off
[u/mrichter/AliRoot.git] / PYTHIA6 / AliGenPythia.h
index ba751f8977340211ff1cfb68a8debec812861ae6..0c82e8cf024c994af29c54137f277c439995f68f 100644 (file)
@@ -31,7 +31,7 @@ class AliGenPythia : public AliGenMC
 {
  public:
 
-    typedef enum {kFlavorSelection, kParentSelection} StackFillOpt_t;
+    typedef enum {kFlavorSelection, kParentSelection, kHeavyFlavor} StackFillOpt_t;
     typedef enum {kCountAll, kCountParents, kCountTrackables} CountMode_t;
     typedef enum {kCluster, kCell} JetRecMode_t;
          
@@ -44,6 +44,7 @@ class AliGenPythia : public AliGenMC
     virtual void    SetEventListRange(Int_t eventFirst=-1, Int_t eventLast=-1);
     // Select process type
     virtual void    SetProcess(Process_t proc = kPyCharm) {fProcess = proc;}
+    virtual void    SetTune(Int_t itune) {fItune = itune;}
 
     // Select structure function
     virtual void    SetStrucFunc(StrucFunc_t func =  kCTEQ5L) {fStrucFunc = func;}
@@ -67,6 +68,11 @@ class AliGenPythia : public AliGenMC
     virtual void    SetEnergyCMS(Float_t energy = 5500) {fEnergyCMS = energy;}
     // Treat protons as inside nuclei with mass numbers a1 and a2
     virtual void    SetNuclei(Int_t a1, Int_t a2, Int_t pdfset = 0);
+    // Set colliding nuclei ("p","n",...)
+    virtual void    SetCollisionSystem(TString projectile, TString target) { fProjectile = projectile; fTarget = target; }
+    virtual void    SetNuclearPDF(Int_t pdf) {fNucPdf = pdf;}
+    virtual void    SetUseNuclearPDF(Bool_t val) {fUseNuclearPDF = val;}
+    virtual void    SetUseLorentzBoost(Bool_t val) {fUseLorentzBoost = val;}
     //
     // Trigger options
     //
@@ -87,32 +93,78 @@ class AliGenPythia : public AliGenMC
     // Phi range for gamma trigger
     virtual void    SetGammaPhiRange(Float_t phimin = 0., Float_t phimax = 360.)
        {fPhiMinGamma = TMath::Pi()*phimin/180.; fPhiMaxGamma = TMath::Pi()*phimax/180.;}
-   // Select jets with fragmentation photon or pi0 going to PHOS or EMCAL
-    virtual void  SetFragPhotonInCalo(Bool_t b)  {fFragPhotonInCalo = b;}
-    virtual void  SetPi0InCalo       (Bool_t b)  {fPi0InCalo    = b;}
-    virtual void  SetPhotonInCalo(Bool_t b)      {fPhotonInCalo = b;}
-    virtual void  SetCheckPHOS (Bool_t b)        {fCheckPHOS    = b;}
-    virtual void  SetCheckEMCAL(Bool_t b)        {fCheckEMCAL   = b;}
-    virtual void  SetFragPhotonInEMCAL(Bool_t b) {fCheckEMCAL   = b; fFragPhotonInCalo = b;}
-    virtual void  SetFragPhotonInPHOS(Bool_t b)  {fCheckPHOS    = b; fFragPhotonInCalo = b;}
-    virtual void  SetPi0InEMCAL(Bool_t b)        {fCheckEMCAL   = b; fPi0InCalo        = b;}
-    virtual void  SetPi0InPHOS(Bool_t b)         {fCheckPHOS    = b; fPi0InCalo        = b;}
-    virtual void  SetPhotonInEMCAL(Bool_t b)     {fCheckEMCAL   = b; fPhotonInCalo     = b;}
-    virtual void  SetElectronInEMCAL(Bool_t b)   {fEleInEMCAL   = b;}
-    virtual void  SetPhotonInPHOS(Bool_t b)      {fCheckPHOS    = b; fPhotonInCalo     = b;}
+  
+    // Select events with fragmentation photon, decay photon, pi0 or eta going to PHOS or EMCAL and central barrel
+    virtual Bool_t TriggerOnSelectedParticles(Int_t np);
+  
+    virtual void  SetCheckPHOS         (Bool_t b) {fCheckPHOS    = b;}
+    virtual void  SetCheckEMCAL        (Bool_t b) {fCheckEMCAL   = b;}
+    virtual void  SetCheckBarrel       (Bool_t b) {fCheckBarrel  = b;}
+  
+    //virtual void  SetElectronInEMCAL   (Bool_t b) {fEleInEMCAL   = b;}
+    //virtual void  SetPhotonInPHOS      (Bool_t b) {fCheckPHOS    = b; fPhotonInCalo     = b;} // Not in use
 
+    virtual void  SetFragPhotonInCalo  (Bool_t b) {                   fFragPhotonInCalo = b;}
+    virtual void  SetFragPhotonInBarrel(Bool_t b) {fCheckBarrel  = b; fFragPhotonInCalo = b;}
+    virtual void  SetFragPhotonInEMCAL (Bool_t b) {fCheckEMCAL   = b; fFragPhotonInCalo = b;}
+    virtual void  SetFragPhotonInPHOS  (Bool_t b) {fCheckPHOS    = b; fFragPhotonInCalo = b;}
+  
+    virtual void  SetHadronInCalo      (Bool_t b) {                   fHadronInCalo     = b;}
+    virtual void  SetHadronInBarrel    (Bool_t b) {fCheckBarrel  = b; fHadronInCalo     = b;}
+    virtual void  SetHadronInEMCAL     (Bool_t b) {fCheckEMCAL   = b; fHadronInCalo     = b;}
+    virtual void  SetHadronInPHOS      (Bool_t b) {fCheckPHOS    = b; fHadronInCalo     = b;}
+  
+    virtual void  SetElectronInCalo    (Bool_t b) {                   fEleInCalo        = b;}
+    virtual void  SetElectronInBarrel  (Bool_t b) {fCheckBarrel  = b; fEleInCalo        = b;}
+    virtual void  SetElectronInEMCAL   (Bool_t b) {fCheckEMCAL   = b; fEleInCalo        = b;}
+    virtual void  SetElectronInPHOS    (Bool_t b) {fCheckPHOS    = b; fEleInCalo        = b;}
+  
+    virtual void  SetDecayPhotonInCalo (Bool_t d)  {fDecayPhotonInCalo = d;}
+    virtual void  SetDecayPhotonInBarrel(Bool_t d) {fDecayPhotonInCalo = d; fCheckBarrel  = d;}
+    virtual void  SetDecayPhotonInEMCAL(Bool_t d)  {fDecayPhotonInCalo = d; fCheckEMCAL   = d;}
+    virtual void  SetDecayPhotonInPHOS (Bool_t d)  {fDecayPhotonInCalo = d; fCheckPHOS    = d;}
+  
+    virtual void  SetPi0InCalo         (Bool_t b, Bool_t f = kFALSE) {fPi0InCalo = b; fForceNeutralMeson2PhotonDecay = f;}
+    virtual void  SetPi0InBarrel       (Bool_t b, Bool_t f = kFALSE) {fPi0InCalo = b; fForceNeutralMeson2PhotonDecay = f; fCheckBarrel= b; }
+    virtual void  SetPi0InEMCAL        (Bool_t b, Bool_t f = kFALSE) {fPi0InCalo = b; fForceNeutralMeson2PhotonDecay = f; fCheckEMCAL = b; }
+    virtual void  SetPi0InPHOS         (Bool_t b, Bool_t f = kFALSE) {fPi0InCalo = b; fForceNeutralMeson2PhotonDecay = f; fCheckPHOS  = b; }
+    virtual void  SetEtaInCalo         (Bool_t b, Bool_t f = kFALSE) {fEtaInCalo = b; fForceNeutralMeson2PhotonDecay = f;}
+    virtual void  SetEtaInBarrel       (Bool_t b, Bool_t f = kFALSE) {fEtaInCalo = b; fForceNeutralMeson2PhotonDecay = f; fCheckBarrel= b; }
+    virtual void  SetEtaInEMCAL        (Bool_t b, Bool_t f = kFALSE) {fEtaInCalo = b; fForceNeutralMeson2PhotonDecay = f; fCheckEMCAL = b; }
+    virtual void  SetEtaInPHOS         (Bool_t b, Bool_t f = kFALSE) {fEtaInCalo = b; fForceNeutralMeson2PhotonDecay = f; fCheckPHOS  = b; }
+
+    virtual void  SetPi0PhotonDecayInBarrel(Bool_t b, Bool_t f = kFALSE) {fPi0InCalo = b; fDecayPhotonInCalo = b; fForceNeutralMeson2PhotonDecay = f; fCheckBarrel  = b; }
+    virtual void  SetPi0PhotonDecayInEMCAL (Bool_t b, Bool_t f = kFALSE) {fPi0InCalo = b; fDecayPhotonInCalo = b; fForceNeutralMeson2PhotonDecay = f; fCheckEMCAL   = b; }
+    virtual void  SetPi0PhotonDecayInPHOS  (Bool_t b, Bool_t f = kFALSE) {fPi0InCalo = b; fDecayPhotonInCalo = b; fForceNeutralMeson2PhotonDecay = f; fCheckPHOS    = b; }
+
+    virtual void  SetEtaPhotonDecayInBarrel(Bool_t b, Bool_t f = kFALSE) {fEtaInCalo = b; fDecayPhotonInCalo = b; fForceNeutralMeson2PhotonDecay = f; fCheckBarrel  = b; }
+    virtual void  SetEtaPhotonDecayInEMCAL (Bool_t b, Bool_t f = kFALSE) {fEtaInCalo = b; fDecayPhotonInCalo = b; fForceNeutralMeson2PhotonDecay = f; fCheckEMCAL   = b; }
+    virtual void  SetEtaPhotonDecayInPHOS  (Bool_t b, Bool_t f = kFALSE) {fEtaInCalo = b; fDecayPhotonInCalo = b; fForceNeutralMeson2PhotonDecay = f; fCheckPHOS    = b; }
+
+  
     // Trigger on a minimum multiplicity
-    virtual void    SetTriggerChargedMultiplicity(Int_t multiplicity, Float_t etamax = 0) {fTriggerMultiplicity = multiplicity; fTriggerMultiplicityEta = etamax; }
+    virtual void  SetTriggerChargedMultiplicity(Int_t multiplicity, Float_t etamax = 0, Float_t ptmin = -1.) 
+    {fTriggerMultiplicity = multiplicity; fTriggerMultiplicityEta = etamax; 
+      fTriggerMultiplicityPtMin = ptmin;}
        
-    virtual void  SetPhotonInPHOSeta(Bool_t b)   {fCheckPHOSeta = b; fPhotonInCalo     = b;}
-    virtual void  SetFragPhotonOrPi0MinPt(Float_t pt)      {fFragPhotonOrPi0MinPt = pt;}
-    virtual void  SetPhotonMinPt(Float_t pt)     {fPhotonMinPt = pt;}
-    virtual void  SetElectronMinPt(Float_t pt)     {fElectronMinPt = pt;}
+    // Calorimeters acceptance
+    // Set Phi in degrees, and Eta coverage, should not be negative
+    virtual void  SetBarrelAcceptance(Float_t deta) {fTriggerEta = deta ;}
+    virtual void  SetEMCALAcceptance (Float_t phimin, Float_t phimax, Float_t deta) {fEMCALMinPhi = phimin ; fEMCALMaxPhi = phimax ; fEMCALEta = deta ; }
+    virtual void  SetPHOSAcceptance  (Float_t phimin, Float_t phimax, Float_t deta) {fPHOSMinPhi  = phimin ; fPHOSMaxPhi  = phimax ; fPHOSEta  = deta ; }
+    virtual void  SetRotateParticleInPHOSeta(Bool_t b) {fCheckPHOSeta = b;}
+  
+    virtual void  SetTriggerParticleMinPt(Float_t pt) {fTriggerParticleMinPt = pt;}
+//    virtual void  SetPhotonMinPt(Float_t pt)          {fPhotonMinPt = pt;}
+//    virtual void  SetElectronMinPt(Float_t pt)        {fElectronMinPt = pt;}
     // Trigger and rotate event 
-    void RotatePhi(Int_t iphcand, Bool_t& okdd);
-    // Trigger on a single particle
-    virtual void    SetTriggerParticle(Int_t particle = 0, Float_t etamax = 0.9) 
-       {fTriggerParticle = particle; fTriggerEta = etamax;}
+    void RotatePhi(Bool_t& okdd);
+  
+    // Trigger on a single particle (not related to calorimeter trigger above)
+    virtual void    SetTriggerParticle(Int_t particle = 0, Float_t etamax = 0.9, Float_t ptmin = -1, Float_t ptmax = 1000) 
+       {fTriggerParticle = particle; fTriggerEta = etamax; fTriggerMinPt = ptmin; fTriggerMaxPt = ptmax;}
+
     //
     // Heavy flavor options
     //
@@ -142,9 +194,13 @@ class AliGenPythia : public AliGenMC
     void SetQhat(Float_t qhat) {fQhat = qhat;}
     //Set initial medium length.
     void SetLength(Float_t length) {fLength = length;}
-
+    //set parameters for pyquen afterburner
+    virtual void SetPyquenPar(Float_t t0=1., Float_t tau0=0.1, Int_t nf=0,Int_t iengl=0, Int_t iangl=3)
+    {fpyquenT = t0; fpyquenTau = tau0; fpyquenNf=nf;fpyquenEloss=iengl;fpyquenAngle=iangl;}
     virtual void SetHadronisation(Int_t flag = 1) {fHadronisation = flag;}
-    virtual void SetReadFromFile(const Text_t *filname) {fFileName = filname;  fReadFromFile = 1;}    
+    virtual void SetPatchOmegaDalitz(Int_t flag = 1) {fPatchOmegaDalitz = flag;}
+    virtual void SetReadFromFile(const Text_t *filname) {fkFileName = filname;  fReadFromFile = 1;}    
+    virtual void SetReadLHEF(const Text_t *filename) {fkNameLHEF = filename; fReadLHEF = 1;}    
 
     //
     // Pile-up
@@ -162,7 +218,7 @@ class AliGenPythia : public AliGenMC
                             Float_t minet = 10., Float_t r = 1.);
     
   void LoadEvent(AliStack* stack, Int_t flag = 0, Int_t reHadr = 0);
-  void LoadEvent(TObjArray* stack, Int_t flag = 0, Int_t reHadr = 0);
+  void LoadEvent(const TObjArray* stack, Int_t flag = 0, Int_t reHadr = 0);
     // Getters
     virtual Process_t    GetProcess() const {return fProcess;}
     virtual StrucFunc_t  GetStrucFunc() const {return fStrucFunc;}
@@ -179,15 +235,21 @@ class AliGenPythia : public AliGenMC
     virtual void         GetGammaPhiRange(Float_t& phimin, Float_t& phimax) const
        {phimin = fPhiMinGamma*180./TMath::Pi(); phimax = fPhiMaxGamma*180./TMath::Pi();}
     //
-    Bool_t IsInEMCAL(Float_t phi, Float_t eta);
-    Bool_t IsInPHOS(Float_t phi, Float_t eta);
+    Bool_t CheckDetectorAcceptance(Float_t phi, Float_t eta, Int_t iparticle);
+    Bool_t IsInEMCAL (Float_t phi, Float_t eta) const;
+    Bool_t IsInPHOS  (Float_t phi, Float_t eta, Int_t iparticle) ;
+    Bool_t IsInBarrel(Float_t eta) const;
+    Bool_t IsFromHeavyFlavor(Int_t ipart);
     //
     virtual void FinishRun();
-    Bool_t CheckTrigger(TParticle* jet1, TParticle* jet2);
+    Bool_t CheckTrigger(const TParticle* jet1, const TParticle* jet2);
     //Used in some processes to selected child properties
     Bool_t CheckKinematicsOnChild();
     void     GetSubEventTime();
 
+    void SetTuneForDiff(Bool_t a=kTRUE) {fkTuneForDiff=a;}
+    AliDecayer * GetDecayer(){return fDecayer;}
+
  protected:
     // adjust the weight from kinematic cuts
     void     AdjustWeights() const;
@@ -195,6 +257,7 @@ class AliGenPythia : public AliGenMC
     void     MakeHeader();    
     void     GeneratePileup();
     Process_t   fProcess;           //Process type
+    Int_t       fItune;             // Pythia tune > 6.4
     StrucFunc_t fStrucFunc;         //Structure Function
     Float_t     fKineBias;          //!Bias from kinematic selection
     Int_t       fTrials;            //!Number of trials for current event
@@ -218,11 +281,18 @@ class AliGenPythia : public AliGenMC
     Int_t       fGinit;             //initial state gluon radiation
     Int_t       fGfinal;            //final state gluon radiation
     Int_t       fHadronisation;     //hadronisation
+    Bool_t      fPatchOmegaDalitz;  //flag for omega dalitz decay patch
     Int_t       fNpartons;          //Number of partons before hadronisation
     Int_t       fReadFromFile;      //read partons from file
+    Int_t       fReadLHEF;          //read lhef file
     Int_t       fQuench;            //Flag for quenching
     Float_t     fQhat;              //Transport coefficient (GeV^2/fm)
     Float_t     fLength;            //Medium length (fm)
+    Float_t     fpyquenT;           //Pyquen initial temperature 
+    Float_t     fpyquenTau;         //Pyquen initial proper time
+    Int_t       fpyquenNf;          //Pyquen number of flavours into the game
+    Int_t       fpyquenEloss;       //Pyquen type of energy loss 
+    Int_t       fpyquenAngle;       //Pyquen radiation angle for gluons
     Float_t     fImpact;            //Impact parameter for quenching simulation (q-pythia) 
     Float_t     fPtKick;            //Transverse momentum kick
     Bool_t      fFullEvent;         //!Write Full event if true
@@ -254,13 +324,18 @@ class AliGenPythia : public AliGenMC
                                     // quark families (e.g. b->c)
     Bool_t  fFragmentation;         // Option to activate fragmentation by Pythia
     Bool_t  fSetNuclei;             // Flag indicating that SetNuclei has been called
+    Bool_t  fUseNuclearPDF;         // flag if nuclear pdf should be applied
+    Bool_t  fUseLorentzBoost;       // flag if lorentz boost should be applied
     Bool_t  fNewMIS;                // Flag for the new multipple interactions scenario
     Bool_t  fHFoff;                 // Flag for switching heafy flavor production off
     Int_t   fNucPdf;                // Nuclear pdf 0: EKS98 1: EPS08
     Int_t   fTriggerParticle;       // Trigger on this particle ...
     Float_t fTriggerEta;            // .. within |eta| < fTriggerEta
-    Int_t       fTriggerMultiplicity;    // Trigger on events with a minimum charged multiplicity
-    Float_t     fTriggerMultiplicityEta; // in a given eta range
+    Float_t fTriggerMinPt;          // .. within pt > fTriggerMinPt
+    Float_t fTriggerMaxPt;          // .. within pt < fTriggerMaxPt
+    Int_t       fTriggerMultiplicity;       // Trigger on events with a minimum charged multiplicity
+    Float_t     fTriggerMultiplicityEta;    // in a given eta range
+    Float_t     fTriggerMultiplicityPtMin;  // above this pT 
     CountMode_t fCountMode;         // Options for counting when the event will be finished.     
     // fCountMode = kCountAll         --> All particles that end up in the
     //                                    stack are counted
@@ -272,32 +347,45 @@ class AliGenPythia : public AliGenMC
 
     AliGenPythiaEventHeader* fHeader;  //! Event header
     AliRunLoader*            fRL;      //! Run Loader
-    const Text_t* fFileName;           //! Name of file to read from
-
-
+    const Text_t* fkFileName;          //! Name of file to read from
+    const Text_t* fkNameLHEF;          //! Name of lhef file to read from
     Bool_t fFragPhotonInCalo; // Option to ask for Fragmentation Photon in calorimeters acceptance
+    Bool_t fHadronInCalo;     // Option to ask for hadron (not pi0) in calorimeters acceptance
     Bool_t fPi0InCalo;        // Option to ask for Pi0 in calorimeters acceptance
-    Bool_t fPhotonInCalo;     // Option to ask for Decay Photon in calorimeter acceptance
-    Bool_t fEleInEMCAL;       // Option to ask for Electron in EMCAL acceptance
-    Bool_t fCheckEMCAL;       // Option to ask for FragPhoton or Pi0 in calorimeters EMCAL acceptance
-    Bool_t fCheckPHOS;        // Option to ask for FragPhoton or Pi0 in calorimeters PHOS acceptance
-    Bool_t fCheckPHOSeta;     // Option to ask for PHOS eta acceptance
-    Float_t fFragPhotonOrPi0MinPt; // Minimum momentum of Fragmentation Photon or Pi0
-    Float_t fPhotonMinPt;          // Minimum momentum of Photon 
-    Float_t fElectronMinPt;        // Minimum momentum of Electron 
+    Bool_t fEtaInCalo;        // Option to ask for Eta in calorimeters acceptance
+    Bool_t fPhotonInCalo;     // Option to ask for Photon in calorimeter acceptance (not in use)
+    Bool_t fDecayPhotonInCalo;// Option to ask for Decay Photon in calorimeter acceptance
+    Bool_t fForceNeutralMeson2PhotonDecay; // Option to ask for Pi0/Eta in calorimeters acceptance when decay into 2 photons
+    Bool_t fEleInCalo;        // Option to ask for Electron in EMCAL acceptance
+    Bool_t fEleInEMCAL;       // Option to ask for Electron in EMCAL acceptance (not in use)
+    Bool_t fCheckBarrel;      // Option to ask for FragPhoton or Pi0 or Eta or gamma decays in central barrel acceptance
+    Bool_t fCheckEMCAL;       // Option to ask for FragPhoton or Pi0 or Eta or gamma decays in calorimeters EMCAL acceptance
+    Bool_t fCheckPHOS;        // Option to ask for FragPhoton or Pi0 or Eta or gamma decays in calorimeters PHOS acceptance
+    Bool_t fCheckPHOSeta;     // Option to ask for rotate event particles in phi to have in PHOS acceptance a requested particle that previously had the good eta
+    Int_t  fPHOSRotateCandidate; // Internal member to select the particle candidate to trigger the event phi rotation, to put it in PHOS phi acceptance
+    Float_t fTriggerParticleMinPt; // Minimum momentum of Fragmentation Photon or Pi0 or other hadron
+    Float_t fPhotonMinPt;          // Minimum momentum of Photon  (not in use)
+    Float_t fElectronMinPt;        // Minimum momentum of Electron (not in use) 
     //Calorimeters eta-phi acceptance 
-    Float_t fPHOSMinPhi;           // Minimum phi PHOS
-    Float_t fPHOSMaxPhi;           // Maximum phi PHOS
-    Float_t fPHOSEta;              // Minimum eta PHOS
-    Float_t fEMCALMinPhi;          // Minimum phi EMCAL
-    Float_t fEMCALMaxPhi;          // Maximum phi EMCAL
-    Float_t fEMCALEta;             // Maximum eta EMCAL
+    Float_t fPHOSMinPhi;           // Minimum phi PHOS, degrees
+    Float_t fPHOSMaxPhi;           // Maximum phi PHOS, degrees
+    Float_t fPHOSEta;              // Minimum eta PHOS, coverage delta eta
+    Float_t fEMCALMinPhi;          // Minimum phi EMCAL, degrees
+    Float_t fEMCALMaxPhi;          // Maximum phi EMCAL, degrees
+    Float_t fEMCALEta;             // Maximum eta EMCAL, coverage delta eta
 
+    Bool_t fkTuneForDiff;    // Pythia tune 
+    Int_t  fProcDiff;
  private:
     AliGenPythia(const AliGenPythia &Pythia);
     AliGenPythia & operator=(const AliGenPythia & rhs);
 
-    ClassDef(AliGenPythia,9) // AliGenerator interface to Pythia
+
+    Bool_t CheckDiffraction();
+    Bool_t GetWeightsDiffraction(Double_t M, Double_t &Mmin, Double_t &Mmax, 
+                                              Double_t &wSD, Double_t &wDD, Double_t &wND);
+
+    ClassDef(AliGenPythia, 14) // AliGenerator interface to Pythia
 };
 #endif