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

#ifndef ALIGENEMLIB_H
#define ALIGENEMLIB_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id: AliGenEMlib.h 30052 2008-11-25 14:54:18Z morsch $ */

/////////////////////////////////////////////////////////////////////////////
//                                                                         //
// Implementation of AliGenEMlib for electron, di-electron, and photon     //
// cocktail calculations.                                                  //
// It is based on AliGenGSIlib.                                            //
//                                                                         //
// Responsible: R.Averbeck@gsi.de                                          //
//                                                                         //
/////////////////////////////////////////////////////////////////////////////

#include "AliGenLib.h"
class TRandom;

class AliGenEMlib :public AliGenLib {
public:
    
  enum Particle_t{kPizero=0, kEta, kRho, kOmega, kEtaprime, kPhi, kJpsi, kDirectRealGamma, kDirectVirtGamma };
  enum Centrality_t{kpp=0x0, k0005=0x1, k0510=0x2, k1020=0x3, k2030=0x4, k3040=0x5, k4050=0x6, k5060=0x7, k0010=0x8, k2040=0x9, k4060=0xA, k6080=0xB, k0020=0xC, k0040=0xD, k2080=0xE, k4080=0xF, kCentralities=0x10};
  enum PtParamSet_t{kPizero7TeVpp=0x00, kPizeroEta7TeVpp=0x10, kPizero7TeVpplow=0x20, kPizeroEta7TeVpplow=0x30, kPizero7TeVpphigh=0x40, kPizeroEta7TeVpphigh=0x50, kPizero2760GeVpp=0x60, kPizeroEta2760GeVpp=0x70, kPizero2760GeVpplow=0x80, kPizeroEta2760GeVpplow=0x90, kPizero2760GeVpphigh=0xA0, kPizeroEta2760GeVpphigh=0xB0, kPiOldChargedPbPb=0xC0, kPichargedPbPb=0xD0, kPizeroPbPb=0xE0, kPichargedPPb=0xF0 };
  enum v2Sys_t{kLoV2Sys=-1, kNoV2Sys=0, kUpV2Sys=+1};
  
  AliGenEMlib() { } ;

  static void SelectParams(PtParamSet_t ptSelect, Centrality_t centSelect=kpp, Int_t v2sys=kNoV2Sys) 
  { fgSelectedPtParam=ptSelect; fgSelectedCentrality=centSelect; fgSelectedV2Systematic=v2sys; }

    GenFunc   GetPt(Int_t param, const char * tname=0) const;
    GenFunc   GetY(Int_t param, const char * tname=0) const;
    GenFuncIp GetIp(Int_t param, const char * tname=0) const;    
  GenFunc   GetV2(Int_t param, const char * tname=0) const;

  //private:

  // General functions

  static Int_t fgSelectedPtParam; // selected pT parameter
  static Int_t fgSelectedCentrality; // selected Centrality
  static Int_t fgSelectedV2Systematic; // selected v2 systematics, usefully values: -1,0,1


  static Double_t PtModifiedHagedornThermal(Double_t pt, 
					    Double_t c, 
					    Double_t p0, 
					    Double_t p1, 
					    Double_t n,
					    Double_t cT,
					    Double_t T);


  static Double_t PtModifiedHagedornExp(Double_t pt,
					Double_t c,
					Double_t p0,
					Double_t p1,
					Double_t p2,
					Double_t n); 


  static Double_t PtModifiedHagedornExp2(Double_t pt,
                                           Double_t c,
                                           Double_t a,
                                           Double_t b,
                                           Double_t p0,
                                           Double_t p1,
                                           Double_t d,
                                           Double_t n);


  static Double_t PtTsallis(Double_t pt,
			    Double_t m,
			    Double_t c,
			    Double_t T,
			    Double_t n);

  static Double_t PtExponential(const Double_t *pt, const Double_t *param);
  static Double_t PtModifiedHagedornPowerlaw(const Double_t *pt, const Double_t *param);
  static Double_t PtDoublePowerlaw(const Double_t *pt, const Double_t *param);
  static Double_t IntegratedKrollWada(const Double_t *mh, const Double_t *);

  // direct gamma
  static Double_t PtPromptRealGamma(const Double_t *px, const Double_t *dummy);
  static Double_t PtPromptVirtGamma(const Double_t *px, const Double_t *dummy);
  static Double_t PtThermalRealGamma(const Double_t *px, const Double_t *dummy);
  static Double_t PtThermalVirtGamma(const Double_t *px, const Double_t *dummy);

  static Int_t    IpDirectRealGamma(TRandom *ran);
  static Double_t PtDirectRealGamma(const Double_t *px, const Double_t *dummy);
  static Double_t YDirectRealGamma(const Double_t *py, const Double_t *dummy);
  static Double_t V2DirectRealGamma(const Double_t *px, const Double_t *dummy);

  static Int_t    IpDirectVirtGamma(TRandom *ran);
  static Double_t PtDirectVirtGamma(const Double_t *px, const Double_t *dummy);
  static Double_t YDirectVirtGamma(const Double_t *py, const Double_t *dummy);
  static Double_t V2DirectVirtGamma(const Double_t *px, const Double_t *dummy);

  // Pizero
    static Int_t    IpPizero(TRandom *ran);
  static Double_t PtPizero(const Double_t *px, const Double_t *dummy);
    static Double_t YPizero(const Double_t *py, const Double_t *dummy);
  static Double_t V2Pizero(const Double_t *px, const Double_t *dummy);

  // Eta
    static Int_t    IpEta(TRandom *ran);
  static Double_t PtEta(const Double_t *px, const Double_t *dummy);
    static Double_t YEta(const Double_t *py, const Double_t *dummy);
  static Double_t V2Eta(const Double_t *px, const Double_t *dummy);

  // Rho
    static Int_t    IpRho(TRandom *ran);
  static Double_t PtRho(const Double_t *px, const Double_t *dummy);
    static Double_t YRho(const Double_t *py, const Double_t *dummy);
  static Double_t V2Rho(const Double_t *py, const Double_t *dummy);

  // Omega
    static Int_t    IpOmega(TRandom *ran);
  static Double_t PtOmega(const Double_t *px, const Double_t *dummy);
    static Double_t YOmega(const Double_t *py, const Double_t *dummy);
  static Double_t V2Omega(const Double_t *py, const Double_t *dummy);

  // Etaprime
    static Int_t    IpEtaprime(TRandom *ran);
  static Double_t PtEtaprime(const Double_t *px, const Double_t *dummy);
    static Double_t YEtaprime(const Double_t *py, const Double_t *dummy);
  static Double_t V2Etaprime(const Double_t *py, const Double_t *dummy);

  // Phi
    static Int_t    IpPhi(TRandom *ran);
  static Double_t PtPhi(const Double_t *px, const Double_t *dummy);
    static Double_t YPhi(const Double_t *py, const Double_t *dummy);
  static Double_t V2Phi(const Double_t *py, const Double_t *dummy);

  // Jpsi
  static Int_t    IpJpsi(TRandom *ran);
  static Double_t PtJpsi(const Double_t *px, const Double_t *dummy);
  static Double_t YJpsi(const Double_t *py, const Double_t *dummy);
  static Double_t V2Jpsi(const Double_t *py, const Double_t *dummy);

  // General
  //static Double_t PtFlat(const Double_t *px, const Double_t *dummy);
    static Double_t YFlat(Double_t y);
  static Double_t MtScal(Double_t pt, Int_t np);
  static Double_t V2Param(const Double_t *px, const Double_t *param);
  static Double_t V2Flat(const Double_t *px, const Double_t *param);
  static Double_t KEtScal(Double_t pt, Int_t np);
  static Double_t GetTAA(Int_t cent);

  static Double_t CrossOverLc(double a, double b, double x);
  static Double_t CrossOverRc(double a, double b, double x);

  static const Double_t fgkPtParam[kCentralities][10];          // parameters of pi pt spectrum
  static const Double_t fgkV2param[kCentralities][16];          // parameters of pi v2 spectrum
  static const Double_t fgkRawPtOfV2Param[kCentralities][10];   // parameters of the raw pt spectrum of v2 analysis
  static const Double_t fgkThermPtParam[kCentralities][2];      // parameters of thermal gamma pt
  static const Double_t fgkHM[8];                    // particle masses
  static const Double_t fgkMtFactor[2][8];           // mt scaling factor

  ClassDef(AliGenEMlib,0)
};


#endif
Commit	Line	Data
e40b9538	1	#ifndef ALIGENEMLIB_H
	2	#define ALIGENEMLIB_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	/* $Id: AliGenEMlib.h 30052 2008-11-25 14:54:18Z morsch $ */
	7
	8	/////////////////////////////////////////////////////////////////////////////
	9	// //
	10	// Implementation of AliGenEMlib for electron, di-electron, and photon //
	11	// cocktail calculations. //
	12	// It is based on AliGenGSIlib. //
	13	// //
	14	// Responsible: R.Averbeck@gsi.de //
	15	// //
	16	/////////////////////////////////////////////////////////////////////////////
	17
	18	#include "AliGenLib.h"
	19	class TRandom;
	20
	21	class AliGenEMlib :public AliGenLib {
6078e216	22	public:
4ae1c9f0	23
6a8b015a	24	enum Particle_t{kPizero=0, kEta, kRho, kOmega, kEtaprime, kPhi, kJpsi, kDirectRealGamma, kDirectVirtGamma };
71443190	25	enum Centrality_t{kpp=0x0, k0005=0x1, k0510=0x2, k1020=0x3, k2030=0x4, k3040=0x5, k4050=0x6, k5060=0x7, k0010=0x8, k2040=0x9, k4060=0xA, k6080=0xB, k0020=0xC, k0040=0xD, k2080=0xE, k4080=0xF, kCentralities=0x10};
1b589c8a	26	enum PtParamSet_t{kPizero7TeVpp=0x00, kPizeroEta7TeVpp=0x10, kPizero7TeVpplow=0x20, kPizeroEta7TeVpplow=0x30, kPizero7TeVpphigh=0x40, kPizeroEta7TeVpphigh=0x50, kPizero2760GeVpp=0x60, kPizeroEta2760GeVpp=0x70, kPizero2760GeVpplow=0x80, kPizeroEta2760GeVpplow=0x90, kPizero2760GeVpphigh=0xA0, kPizeroEta2760GeVpphigh=0xB0, kPiOldChargedPbPb=0xC0, kPichargedPbPb=0xD0, kPizeroPbPb=0xE0, kPichargedPPb=0xF0 };
71443190	27	enum v2Sys_t{kLoV2Sys=-1, kNoV2Sys=0, kUpV2Sys=+1};
4ae1c9f0	28
	29	AliGenEMlib() { } ;
	30
1b589c8a	31	static void SelectParams(PtParamSet_t ptSelect, Centrality_t centSelect=kpp, Int_t v2sys=kNoV2Sys)
4ae1c9f0	32	{ fgSelectedPtParam=ptSelect; fgSelectedCentrality=centSelect; fgSelectedV2Systematic=v2sys; }
4ae1c9f0	33
e40b9538	34	GenFunc GetPt(Int_t param, const char * tname=0) const;
	35	GenFunc GetY(Int_t param, const char * tname=0) const;
	36	GenFuncIp GetIp(Int_t param, const char * tname=0) const;
6078e216	37	GenFunc GetV2(Int_t param, const char * tname=0) const;
e40b9538	38
4ae1c9f0	39	//private:
	40
	41	// General functions
	42
	43	static Int_t fgSelectedPtParam; // selected pT parameter
	44	static Int_t fgSelectedCentrality; // selected Centrality
	45	static Int_t fgSelectedV2Systematic; // selected v2 systematics, usefully values: -1,0,1
	46
	47
5a516e0a	48	static Double_t PtModifiedHagedornThermal(Double_t pt,
	49	Double_t c,
	50	Double_t p0,
	51	Double_t p1,
	52	Double_t n,
	53	Double_t cT,
	54	Double_t T);
4ae1c9f0	55
	56
	57
5a516e0a	58	static Double_t PtModifiedHagedornExp(Double_t pt,
	59	Double_t c,
	60	Double_t p0,
	61	Double_t p1,
	62	Double_t p2,
	63	Double_t n);
	64
	65
	66	static Double_t PtModifiedHagedornExp2(Double_t pt,
	67	Double_t c,
	68	Double_t a,
	69	Double_t b,
	70	Double_t p0,
	71	Double_t p1,
	72	Double_t d,
	73	Double_t n);
	74
	75
	76	static Double_t PtTsallis(Double_t pt,
	77	Double_t m,
	78	Double_t c,
	79	Double_t T,
	80	Double_t n);
4ae1c9f0	81
71443190	82	static Double_t PtExponential(const Double_t pt, const Double_t param);
71443190	83	static Double_t PtModifiedHagedornPowerlaw(const Double_t pt, const Double_t param);
6a8b015a	84	static Double_t PtDoublePowerlaw(const Double_t pt, const Double_t param);
6a8b015a	85	static Double_t IntegratedKrollWada(const Double_t mh, const Double_t );
4ae1c9f0	86
6a8b015a	87	// direct gamma
71443190	88	static Double_t PtPromptRealGamma(const Double_t px, const Double_t dummy);
71443190	89	static Double_t PtPromptVirtGamma(const Double_t px, const Double_t dummy);
6a8b015a	90	static Double_t PtThermalRealGamma(const Double_t px, const Double_t dummy);
	91	static Double_t PtThermalVirtGamma(const Double_t px, const Double_t dummy);
	92
	93	static Int_t IpDirectRealGamma(TRandom *ran);
	94	static Double_t PtDirectRealGamma(const Double_t px, const Double_t dummy);
	95	static Double_t YDirectRealGamma(const Double_t py, const Double_t dummy);
	96	static Double_t V2DirectRealGamma(const Double_t px, const Double_t dummy);
	97
	98	static Int_t IpDirectVirtGamma(TRandom *ran);
	99	static Double_t PtDirectVirtGamma(const Double_t px, const Double_t dummy);
	100	static Double_t YDirectVirtGamma(const Double_t py, const Double_t dummy);
	101	static Double_t V2DirectVirtGamma(const Double_t px, const Double_t dummy);
e40b9538	102
6078e216	103	// Pizero
e40b9538	104	static Int_t IpPizero(TRandom *ran);
4ae1c9f0	105	static Double_t PtPizero(const Double_t px, const Double_t dummy);
e40b9538	106	static Double_t YPizero(const Double_t py, const Double_t dummy);
6078e216	107	static Double_t V2Pizero(const Double_t px, const Double_t dummy);
e40b9538	108
6078e216	109	// Eta
e40b9538	110	static Int_t IpEta(TRandom *ran);
4ae1c9f0	111	static Double_t PtEta(const Double_t px, const Double_t dummy);
e40b9538	112	static Double_t YEta(const Double_t py, const Double_t dummy);
6078e216	113	static Double_t V2Eta(const Double_t px, const Double_t dummy);
e40b9538	114
6078e216	115	// Rho
e40b9538	116	static Int_t IpRho(TRandom *ran);
4ae1c9f0	117	static Double_t PtRho(const Double_t px, const Double_t dummy);
e40b9538	118	static Double_t YRho(const Double_t py, const Double_t dummy);
4ae1c9f0	119	static Double_t V2Rho(const Double_t py, const Double_t dummy);
e40b9538	120
6078e216	121	// Omega
e40b9538	122	static Int_t IpOmega(TRandom *ran);
4ae1c9f0	123	static Double_t PtOmega(const Double_t px, const Double_t dummy);
e40b9538	124	static Double_t YOmega(const Double_t py, const Double_t dummy);
4ae1c9f0	125	static Double_t V2Omega(const Double_t py, const Double_t dummy);
e40b9538	126
6078e216	127	// Etaprime
e40b9538	128	static Int_t IpEtaprime(TRandom *ran);
4ae1c9f0	129	static Double_t PtEtaprime(const Double_t px, const Double_t dummy);
e40b9538	130	static Double_t YEtaprime(const Double_t py, const Double_t dummy);
4ae1c9f0	131	static Double_t V2Etaprime(const Double_t py, const Double_t dummy);
e40b9538	132
6078e216	133	// Phi
e40b9538	134	static Int_t IpPhi(TRandom *ran);
4ae1c9f0	135	static Double_t PtPhi(const Double_t px, const Double_t dummy);
e40b9538	136	static Double_t YPhi(const Double_t py, const Double_t dummy);
4ae1c9f0	137	static Double_t V2Phi(const Double_t py, const Double_t dummy);
	138
	139	// Jpsi
	140	static Int_t IpJpsi(TRandom *ran);
	141	static Double_t PtJpsi(const Double_t px, const Double_t dummy);
	142	static Double_t YJpsi(const Double_t py, const Double_t dummy);
	143	static Double_t V2Jpsi(const Double_t py, const Double_t dummy);
e40b9538	144
6078e216	145	// General
4ae1c9f0	146	//static Double_t PtFlat(const Double_t px, const Double_t dummy);
e40b9538	147	static Double_t YFlat(Double_t y);
6078e216	148	static Double_t MtScal(Double_t pt, Int_t np);
	149	static Double_t V2Param(const Double_t px, const Double_t param);
	150	static Double_t V2Flat(const Double_t px, const Double_t param);
4ae1c9f0	151	static Double_t KEtScal(Double_t pt, Int_t np);
71443190	152	static Double_t GetTAA(Int_t cent);
e40b9538	153
5a516e0a	154	static Double_t CrossOverLc(double a, double b, double x);
5a516e0a	155	static Double_t CrossOverRc(double a, double b, double x);
6078e216	156
1b589c8a	157	static const Double_t fgkPtParam[kCentralities][10]; // parameters of pi pt spectrum
	158	static const Double_t fgkV2param[kCentralities][16]; // parameters of pi v2 spectrum
	159	static const Double_t fgkRawPtOfV2Param[kCentralities][10]; // parameters of the raw pt spectrum of v2 analysis
	160	static const Double_t fgkThermPtParam[kCentralities][2]; // parameters of thermal gamma pt
71443190	161	static const Double_t fgkHM[8]; // particle masses
71443190	162	static const Double_t fgkMtFactor[2][8]; // mt scaling factor
e40b9538	163
	164	ClassDef(AliGenEMlib,0)
	165	};
	166
4ae1c9f0	167
e40b9538	168	#endif