[u/mrichter/AliRoot.git] / EVGEN / AliGenMUONLMR.h

#ifndef AliGenMUONLMR_h
#define AliGenMUONLMR_h

#include <TH1F.h> 
#include <TH1D.h> 
#include <TF1.h> 
#include <TParticle.h> 
#include <TLorentzVector.h> 
#include "AliGenMC.h" 
 
class AliGenMUONLMR : public AliGenMC { 
 public:
  enum parttype_t {kPionLMR, kKaonLMR, kEtaLMR, kRhoLMR, kOmegaLMR, kPhiLMR, kEtaPrimeLMR};
  enum CMSEnergies { kCMS2760GeV, kCMS7000GeV, kNCMSEnergies };    
  AliGenMUONLMR(); 
  AliGenMUONLMR(AliGenMUONLMR &gen); 
  AliGenMUONLMR &operator=(const AliGenMUONLMR &gen);  
  ~AliGenMUONLMR(); 
  static Double_t PtDistr(Double_t *x, Double_t *par); 
  static Double_t YDistr(Double_t *x, Double_t *par); 
  virtual void SetPtParams(Int_t iproc, Double_t p1, Double_t p2, Double_t p3) {fPt[iproc]->SetParameters(p1,p2,p3);}
  virtual void SetYParams(Int_t iproc, Double_t p1, Double_t p2, Double_t p3) {fY[iproc]->SetParameters(p1,p2,p3);}
  virtual void Decay2Body(TParticle *mother);
  virtual void DalitzDecay(TParticle *mother);
  virtual void DecayPiK(TParticle *mother, Bool_t &hadDecayed);
  virtual Double_t FormFactor(Double_t q2, Int_t decay); 
  virtual void Generate(); 
  virtual TParticle* GetMuon(Int_t i) {return fMu[i];} 
  virtual void SetNMuMin(Int_t nmin) {fNMuMin = nmin;}
  virtual void GenerateSingleProcess(Int_t whichproc) { fGenSingleProc = whichproc;}
  virtual void SetScaleMultiplicity(Int_t ipart, Double_t scale) { fScaleMult[ipart] = scale; } 
  static Double_t RhoLineShapeNew(Double_t *, Double_t *); 
  virtual void FinishRun(); 
  virtual TF1* GetRapidity(Int_t iproc) { return fY[iproc]; }
  virtual TF1* GetPt(Int_t iproc) { return fPt[iproc]; }
  void SetCMSEnergy(CMSEnergies energy);
 private: 
  static const Int_t fgkNpart = 7; // number of particles to be generated 
  Int_t fNMuMin;                   // min. number of muons to accept the event for writing
  CMSEnergies fCMSEnergy;          // CMS Energy 
  Int_t fGenSingleProc;            // flag to generate a single process (1) or the whole cocktail (0)
  Int_t fPDG[7];                   // pdg code of particle to be generated 
  Double_t fScaleMult[7];          // multiplicity scaling factor (w.r.t. pythia@7TeV)
  TF1 *fPt[7];                     // pt distribution
  TF1 *fY[7];                      // rapidity distribution
  TF1 *fMult[7];                   // multiplicity distribution 
  TF1 *fDecay[2];                  // fDecay[0] = pion, fDecay[1] = kaon
  TH1F *fDalitz[3];                // Dalitz decay form factor for eta, omega, etaprime
  TF1 *fCosTheta;                  // function for polarized theta distributions
  TF1 *fRhoLineShape;              // rho line shape 
  TParticle* fParticle[7];         // TPaticle object for the particles to be generated
  TParticle* fMu[2];               // fMu[0] = mu+    fMu[1] = mu-
  TH1D *fHMultMu;                  // muon multiplicity 
  TH1D *fHNProc;                   // number of events generated per process
  ClassDef(AliGenMUONLMR, 1)       // low mass dimuons parametric generator
}; 

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