Generator for dimuon decays of low masses resonances in AliRoot.
[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};
14 AliGenMUONLMR();
15 ~AliGenMUONLMR();
16 static Double_t PtDistr(Double_t *x, Double_t *par);
17 static Double_t YDistr(Double_t *x, Double_t *par);
18 virtual void Decay2Body(TParticle *mother);
19 virtual void DalitzDecay(TParticle *mother);
20 virtual void DecayPiK(TParticle *mother, Bool_t &hadDecayed);
21 virtual Double_t FormFactor(Double_t q2, Int_t decay);
22 virtual void Generate();
23 virtual TParticle* GetMuon(Int_t i) {return fMu[i];}
24 virtual void SetNMuMin(Int_t nmin) {fNMuMin = nmin; }
25 virtual void GenerateSingleProcess(Int_t whichproc) { fGenSingleProc = whichproc;}
26 virtual void SetScaleMultiplicity(Int_t ipart, Double_t scale) { fScaleMult[ipart] = scale; }
27 static Double_t RhoLineShapeNew(Double_t *, Double_t *);
28 virtual void FinishRun();
29 private:
30 static const Int_t fgkNpart = 7; // number of particles to be generated
31 Int_t fNMuMin; // min. number of muons to accept the event for writing
32 Int_t fGenSingleProc; // flag to generate a single process (1) or the whole cocktail (0)
33 Int_t fPDG[7]; // pdg code of particle to be generated
34 Double_t fScaleMult[7]; // multiplicity scaling factor (w.r.t. pythia@7TeV)
35 TF1 *fPt[7]; // pt distribution
36 TF1 *fY[7]; // rapidity distribution
37 TF1 *fMult[7]; // multiplicity distribution
38 TF1 *fDecay[2]; // fDecay[0] = pion, fDecay[1] = kaon
39 TH1F *fDalitz[3]; // Dalitz decay form factor for eta, omega, etaprime
40 TF1 *fCosTheta; // function for polarized theta distributions
41 TF1 *fRhoLineShape; // rho line shape
42 TParticle* fParticle[7]; // TPaticle object for the particles to be generated
43 TParticle* fMu[2]; // fMu[0] = mu+ fMu[1] = mu-
44 TH1D *fHMultMu; // muon multiplicity
45 TH1D *fHNProc; // number of events generated per process
46 ClassDef(AliGenMUONLMR, 1) // example of analysis
47};
48
49#endif