#ifndef ALIGENPYTHIA_H
#define ALIGENPYTHIA_H
#include "AliGenMC.h"
#include "AliPythia.h"
class AliPythia;
class TParticle;
class AliGenPythiaEventHeader;
class AliGenEventHeader;
class AliStack;
class AliRunLoader;
class TObjArray;
class AliGenPythia : public AliGenMC
{
public:
typedef enum {kFlavorSelection, kParentSelection, kHeavyFlavor} StackFillOpt_t;
typedef enum {kCountAll, kCountParents, kCountTrackables} CountMode_t;
typedef enum {kCluster, kCell} JetRecMode_t;
AliGenPythia();
AliGenPythia(Int_t npart);
virtual ~AliGenPythia();
virtual void Generate();
virtual void Init();
virtual void SetEventListRange(Int_t eventFirst=-1, Int_t eventLast=-1);
virtual void SetProcess(Process_t proc = kPyCharm) {fProcess = proc;}
virtual void SetTune(Int_t itune) {fItune = itune;}
virtual void SetSeed(UInt_t seed);
virtual void SetStrucFunc(StrucFunc_t func = kCTEQ5L) {fStrucFunc = func;}
virtual void SetWeightPower(Float_t power = 0.) { fWeightPower = power; }
virtual void SetPtHard(Float_t ptmin = 0, Float_t ptmax = 1.e10)
{fPtHardMin = ptmin; fPtHardMax = ptmax; }
virtual void SetYHard(Float_t ymin = -1.e10, Float_t ymax = 1.e10)
{fYHardMin = ymin; fYHardMax = ymax; }
virtual void SetGluonRadiation(Int_t iIn, Int_t iFin)
{fGinit = iIn; fGfinal = iFin;}
virtual void SetColorReconnectionOff(Int_t iflag=0){fCRoff=iflag;}
virtual void SetPtKick(Float_t kt = 1.)
{fPtKick = kt;}
virtual void UseNewMultipleInteractionsScenario() {fNewMIS = kTRUE;}
virtual void SwitchHFOff() {fHFoff = kTRUE;}
virtual void SetEnergyCMS(Float_t energy = 5500) {fEnergyCMS = energy;}
virtual void SetNuclei(Int_t a1, Int_t a2, Int_t pdfset = 0);
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;}
virtual void SetJetEtRange(Float_t etmin = 0., Float_t etmax = 1.e4)
{fEtMinJet = etmin; fEtMaxJet = etmax;}
virtual void SetJetEtaRange(Float_t etamin = -20., Float_t etamax = 20.)
{fEtaMinJet = etamin; fEtaMaxJet = etamax;}
virtual void SetJetPhiRange(Float_t phimin = 0., Float_t phimax = 360.)
{fPhiMinJet = TMath::Pi()*phimin/180.; fPhiMaxJet = TMath::Pi()*phimax/180.;}
virtual void SetJetReconstructionMode(Int_t mode = kCell) {fJetReconstruction = mode;}
virtual void SetGammaEtaRange(Float_t etamin = -20., Float_t etamax = 20.)
{fEtaMinGamma = etamin; fEtaMaxGamma = etamax;}
virtual void SetGammaPhiRange(Float_t phimin = 0., Float_t phimax = 360.)
{fPhiMinGamma = TMath::Pi()*phimin/180.; fPhiMaxGamma = TMath::Pi()*phimax/180.;}
virtual void SetHeavyQuarkYRange(Float_t ymin=-20., Float_t ymax=20.){
fYMinHQ = ymin; fYMaxHQ=ymax; fUseYCutHQ=kTRUE;
}
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 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; }
virtual void SetTriggerChargedMultiplicity(Int_t multiplicity, Float_t etamax = 0, Float_t ptmin = -1.)
{fTriggerMultiplicity = multiplicity; fTriggerMultiplicityEta = etamax;
fTriggerMultiplicityPtMin = ptmin;}
virtual void SetTriggerMultiplicityEtaRange(Int_t multiplicity, Float_t etamin = 0., Float_t etamax = 0., Float_t ptmin = -1.)
{fTriggerMultiplicity = multiplicity; fTriggerMultiplicityEtaMin = etamin; fTriggerMultiplicityEtaMax = etamax;
fTriggerMultiplicityPtMin = ptmin;}
virtual void SetBarrelAcceptance(Float_t deta) {fTriggerEta = deta ;}
virtual void SetTriggerY(Float_t dy) {fTriggerY = dy;}
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;}
void RotatePhi(Bool_t& okdd);
virtual void SetTriggerParticle(Int_t particle = 0, Float_t etamax = 0.9, Float_t ptmin = -1, Float_t ptmax = 1000)
{fTriggerParticle = particle; fTriggerEta = etamax; fTriggerEtaMin = etamax; fTriggerMinPt = ptmin; fTriggerMaxPt = ptmax;}
virtual void SetTriggerParticle(Int_t particle, Float_t etamin, Float_t etamax, Float_t ptmin, Float_t ptmax)
{fTriggerParticle = particle; fTriggerEtaMin = etamin, fTriggerEta = etamax; fTriggerMinPt = ptmin; fTriggerMaxPt = ptmax;}
virtual void SetFeedDownHigherFamily(Bool_t opt) {
fFeedDownOpt = opt;
}
virtual void SetStackFillOpt(StackFillOpt_t opt) {
fStackFillOpt = opt;
}
virtual void SetFragmentation(Bool_t opt) {
fFragmentation = opt;
}
virtual void SetCountMode(CountMode_t mode) {
fCountMode = mode;
}
virtual void SetQuench(Int_t flag = 0) {fQuench = flag;}
void SetQhat(Float_t qhat) {fQhat = qhat;}
void SetLength(Float_t length) {fLength = length;}
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 SetPatchOmegaDalitz(Int_t flag = 1) {fPatchOmegaDalitz = flag;}
virtual void SetDecayerExodus(Int_t flag = 1) {fDecayerExodus = flag;}
virtual void SetReadFromFile(const Text_t *filname) {fkFileName = filname; fReadFromFile = 1;}
virtual void SetReadLHEF(const Text_t *filename) {fkNameLHEF = filename; fReadLHEF = 1;}
virtual void SetInteractionRate(Float_t rate,Float_t timewindow = 90.e-6);
virtual Float_t GetInteractionRate() const {return fInteractionRate;}
virtual Float_t GetXsection() const {return fXsection;}
void GetJets(Int_t& njets, Int_t& ntrig, Float_t jets[4][10]);
void RecJetsUA1(Int_t& njets, Float_t jets[4][50]);
void SetPycellParameters(Float_t etamax = 2., Int_t neta = 274, Int_t nphi = 432,
Float_t thresh = 0., Float_t etseed = 4.,
Float_t minet = 10., Float_t r = 1.);
void LoadEvent(AliStack* stack, Int_t flag = 0, Int_t reHadr = 0);
void LoadEvent(const TObjArray* stack, Int_t flag = 0, Int_t reHadr = 0);
virtual Process_t GetProcess() const {return fProcess;}
virtual StrucFunc_t GetStrucFunc() const {return fStrucFunc;}
virtual void GetPtHard(Float_t& ptmin, Float_t& ptmax) const
{ptmin = fPtHardMin; ptmax = fPtHardMax;}
virtual void GetNuclei(Int_t& a1, Int_t& a2) const
{a1 = fAProjectile; a2 = fATarget;}
virtual void GetJetEtRange(Float_t& etamin, Float_t& etamax) const
{etamin = fEtaMinJet; etamax = fEtaMaxJet;}
virtual void GetJetPhiRange(Float_t& phimin, Float_t& phimax) const
{phimin = fPhiMinJet*180./TMath::Pi(); phimax = fPhiMaxJet*180/TMath::Pi();}
virtual void GetGammaEtaRange(Float_t& etamin, Float_t& etamax) const
{etamin = fEtaMinGamma; etamax = fEtaMaxGamma;}
virtual void GetGammaPhiRange(Float_t& phimin, Float_t& phimax) const
{phimin = fPhiMinGamma*180./TMath::Pi(); phimax = fPhiMaxGamma*180./TMath::Pi();}
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(const TParticle* jet1, const TParticle* jet2);
Bool_t CheckKinematicsOnChild();
void GetSubEventTime();
void SetTuneForDiff(Bool_t a=kTRUE) {fkTuneForDiff=a;}
AliDecayer * GetDecayer(){return fDecayer;}
protected:
void AdjustWeights() const;
Int_t GenerateMB();
void MakeHeader();
void GeneratePileup();
Process_t fProcess;
Int_t fItune;
StrucFunc_t fStrucFunc;
Float_t fKineBias;
Int_t fTrials;
Int_t fTrialsRun;
Float_t fQ;
Float_t fX1;
Float_t fX2;
Float_t fEventTime;
Float_t fInteractionRate;
Float_t fTimeWindow;
Int_t fCurSubEvent;
TArrayF *fEventsTime;
Int_t fNev;
Int_t fFlavorSelect;
Float_t fXsection;
AliPythia *fPythia;
Float_t fWeightPower;
Float_t fPtHardMin;
Float_t fPtHardMax;
Float_t fYHardMin;
Float_t fYHardMax;
Int_t fGinit;
Int_t fGfinal;
Int_t fCRoff;
Int_t fHadronisation;
Bool_t fPatchOmegaDalitz;
Bool_t fDecayerExodus;
Int_t fNpartons;
Int_t fReadFromFile;
Int_t fReadLHEF;
Int_t fQuench;
Float_t fQhat;
Float_t fLength;
Float_t fpyquenT;
Float_t fpyquenTau;
Int_t fpyquenNf;
Int_t fpyquenEloss;
Int_t fpyquenAngle;
Float_t fImpact;
Float_t fPtKick;
Bool_t fFullEvent;
AliDecayer *fDecayer;
Int_t fDebugEventFirst;
Int_t fDebugEventLast;
Float_t fEtMinJet;
Float_t fEtMaxJet;
Float_t fEtaMinJet;
Float_t fEtaMaxJet;
Float_t fPhiMinJet;
Float_t fPhiMaxJet;
Int_t fJetReconstruction;
Float_t fEtaMinGamma;
Float_t fEtaMaxGamma;
Float_t fPhiMinGamma;
Float_t fPhiMaxGamma;
Bool_t fUseYCutHQ;
Float_t fYMinHQ;
Float_t fYMaxHQ;
Float_t fPycellEtaMax;
Int_t fPycellNEta;
Int_t fPycellNPhi;
Float_t fPycellThreshold;
Float_t fPycellEtSeed;
Float_t fPycellMinEtJet;
Float_t fPycellMaxRadius;
StackFillOpt_t fStackFillOpt;
Bool_t fFeedDownOpt;
Bool_t fFragmentation;
Bool_t fSetNuclei;
Bool_t fUseNuclearPDF;
Bool_t fUseLorentzBoost;
Bool_t fNewMIS;
Bool_t fHFoff;
Int_t fNucPdf;
Int_t fTriggerParticle;
Float_t fTriggerEta;
Float_t fTriggerY;
Float_t fTriggerEtaMin;
Float_t fTriggerMinPt;
Float_t fTriggerMaxPt;
Int_t fTriggerMultiplicity;
Float_t fTriggerMultiplicityEta;
Float_t fTriggerMultiplicityEtaMin;
Float_t fTriggerMultiplicityEtaMax;
Float_t fTriggerMultiplicityPtMin;
CountMode_t fCountMode;
AliGenPythiaEventHeader* fHeader;
AliRunLoader* fRL;
const Text_t* fkFileName;
const Text_t* fkNameLHEF;
Bool_t fFragPhotonInCalo;
Bool_t fHadronInCalo;
Bool_t fPi0InCalo;
Bool_t fEtaInCalo;
Bool_t fPhotonInCalo;
Bool_t fDecayPhotonInCalo;
Bool_t fForceNeutralMeson2PhotonDecay;
Bool_t fEleInCalo;
Bool_t fEleInEMCAL;
Bool_t fCheckBarrel;
Bool_t fCheckEMCAL;
Bool_t fCheckPHOS;
Bool_t fCheckPHOSeta;
Int_t fPHOSRotateCandidate;
Float_t fTriggerParticleMinPt;
Float_t fPhotonMinPt;
Float_t fElectronMinPt;
Float_t fPHOSMinPhi;
Float_t fPHOSMaxPhi;
Float_t fPHOSEta;
Float_t fEMCALMinPhi;
Float_t fEMCALMaxPhi;
Float_t fEMCALEta;
Bool_t fkTuneForDiff;
Int_t fProcDiff;
private:
AliGenPythia(const AliGenPythia &Pythia);
AliGenPythia & operator=(const AliGenPythia & rhs);
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, 15)
};
#endif