Add task for single muon flow estimation with the EP method (Diego)
[u/mrichter/AliRoot.git] / EVGEN / AliGenMUONLMR.h
CommitLineData
368fc61e 1#ifndef AliGenMUONLMR_h
2#define AliGenMUONLMR_h
3
4#include <TH1F.h>
5#include <TH1D.h>
6#include <TF1.h>
7#include <TParticle.h>
8#include <TLorentzVector.h>
9#include "AliGenMC.h"
10
11class AliGenMUONLMR : public AliGenMC {
12 public:
13 enum parttype_t {kPionLMR, kKaonLMR, kEtaLMR, kRhoLMR, kOmegaLMR, kPhiLMR, kEtaPrimeLMR};
35e2c739 14 AliGenMUONLMR(Double_t energy=7.0);
15 AliGenMUONLMR(AliGenMUONLMR &gen);
16 AliGenMUONLMR &operator=(const AliGenMUONLMR &gen);
368fc61e 17 ~AliGenMUONLMR();
35e2c739 18 static Double_t PtDistr(Double_t *x, Double_t *par);
19 static Double_t YDistr(Double_t *x, Double_t *par);
20 virtual void SetPtParams(Int_t iproc, Double_t p1, Double_t p2, Double_t p3) {fPt[iproc]->SetParameters(p1,p2,p3);}
21 virtual void SetYParams(Int_t iproc, Double_t p1, Double_t p2, Double_t p3) {fY[iproc]->SetParameters(p1,p2,p3);}
22 virtual void Decay2Body(TParticle *mother);
23 virtual void DalitzDecay(TParticle *mother);
368fc61e 24 virtual void DecayPiK(TParticle *mother, Bool_t &hadDecayed);
25 virtual Double_t FormFactor(Double_t q2, Int_t decay);
26 virtual void Generate();
27 virtual TParticle* GetMuon(Int_t i) {return fMu[i];}
35e2c739 28 virtual void SetNMuMin(Int_t nmin) {fNMuMin = nmin;}
368fc61e 29 virtual void GenerateSingleProcess(Int_t whichproc) { fGenSingleProc = whichproc;}
30 virtual void SetScaleMultiplicity(Int_t ipart, Double_t scale) { fScaleMult[ipart] = scale; }
35e2c739 31 static Double_t RhoLineShapeNew(Double_t *, Double_t *);
368fc61e 32 virtual void FinishRun();
35e2c739 33 virtual TF1* GetRapidity(Int_t iproc) { return fY[iproc]; }
34 virtual TF1* GetPt(Int_t iproc) { return fPt[iproc]; }
368fc61e 35 private:
36 static const Int_t fgkNpart = 7; // number of particles to be generated
37 Int_t fNMuMin; // min. number of muons to accept the event for writing
38 Int_t fGenSingleProc; // flag to generate a single process (1) or the whole cocktail (0)
39 Int_t fPDG[7]; // pdg code of particle to be generated
40 Double_t fScaleMult[7]; // multiplicity scaling factor (w.r.t. pythia@7TeV)
41 TF1 *fPt[7]; // pt distribution
42 TF1 *fY[7]; // rapidity distribution
43 TF1 *fMult[7]; // multiplicity distribution
44 TF1 *fDecay[2]; // fDecay[0] = pion, fDecay[1] = kaon
45 TH1F *fDalitz[3]; // Dalitz decay form factor for eta, omega, etaprime
46 TF1 *fCosTheta; // function for polarized theta distributions
47 TF1 *fRhoLineShape; // rho line shape
48 TParticle* fParticle[7]; // TPaticle object for the particles to be generated
49 TParticle* fMu[2]; // fMu[0] = mu+ fMu[1] = mu-
50 TH1D *fHMultMu; // muon multiplicity
51 TH1D *fHNProc; // number of events generated per process
35e2c739 52 ClassDef(AliGenMUONLMR, 1) // low mass dimuons parametric generator
368fc61e 53};
54
55#endif