Flat J/Psi and Upsilon distributions added (Raphael Tieulent <tieulent@in2p3.fr)

diff --git a/EVGEN/AliGenMUONlib.cxx b/EVGEN/AliGenMUONlib.cxx
index 447d50edb19d2d5ed26139b1abb396784dc381e9..8eb5fe2326eb958a2792c66099a802e2b74c9365 100644 (file)
--- a/EVGEN/AliGenMUONlib.cxx
+++ b/EVGEN/AliGenMUONlib.cxx
@@ -173,6 +173,11 @@ Double_t AliGenMUONlib::PtJpsiCDFscaled( Double_t *px, Double_t */*dummy*/)
   return x/TMath::Power(pass1,kxn);
 }
 
+Double_t AliGenMUONlib::PtJpsiFlat( Double_t *px, Double_t */*dummy*/ )
+{
+  return 1.;
+}
+
 Double_t AliGenMUONlib::PtJpsiPbPb( Double_t *px, Double_t */*dummy*/)
 {
 // J/Psi pT spectrum
@@ -256,6 +261,11 @@ Double_t AliGenMUONlib::YJpsi(Double_t *py, Double_t */*dummy*/)
   return yj;
 }
 
+Double_t AliGenMUONlib::YJpsiFlat( Double_t *py, Double_t */*dummy*/ )
+{
+  return 1.;
+}
+
 
 Double_t AliGenMUONlib::YJpsiPbPb( Double_t *px, Double_t */*dummy*/)
 {
@@ -404,6 +414,11 @@ Double_t AliGenMUONlib::PtUpsilonCDFscaled( Double_t *px, Double_t */*dummy*/ )
   return x/TMath::Power(pass1,kxn);
 }
 
+Double_t AliGenMUONlib::PtUpsilonFlat( Double_t *px, Double_t */*dummy*/ )
+{
+  return 1.;
+}
+
 Double_t AliGenMUONlib::PtUpsilonPbPb( Double_t *px, Double_t */*dummy*/)
 {
 
@@ -510,6 +525,12 @@ Double_t AliGenMUONlib::YUpsilonCDFscaled( Double_t *px, Double_t *dummy)
     // Upsilon y
     return AliGenMUONlib::YUpsilonPbPb(px, dummy);
     
+}
+Double_t AliGenMUONlib::YUpsilonFlat( Double_t *px, Double_t *dummy)
+{
+    // Upsilon y
+    return 1.;
+    
 }
 
 Double_t AliGenMUONlib::YUpsilonPP( Double_t *px, Double_t */*dummy*/)
@@ -839,6 +860,8 @@ GenFunc AliGenMUONlib::GetPt(Int_t param,  const char* tname) const
            func=PtJpsiPP;
        } else if (sname == "CDF scaled") {
            func=PtJpsiCDFscaled;
+       } else if (sname == "Flat") {
+           func=PtJpsiFlat;
        } else {
            func=PtJpsi;
        }
@@ -856,6 +879,8 @@ GenFunc AliGenMUONlib::GetPt(Int_t param,  const char* tname) const
            func=PtUpsilonPP;
        } else if (sname == "CDF scaled") {
            func=PtUpsilonCDFscaled;
+       } else if (sname == "Flat") {
+           func=PtUpsilonFlat;
        } else {
            func=PtUpsilon;
        }
@@ -914,6 +939,8 @@ GenFunc AliGenMUONlib::GetY(Int_t param, const char* tname) const
            func=YJpsiPP;
        } else if (sname == "CDF scaled") {
            func=YJpsiCDFscaled;
+       } else if (sname == "Flat") {
+           func=YJpsiFlat;
        } else {
            func=YJpsi;
        }
@@ -931,6 +958,8 @@ GenFunc AliGenMUONlib::GetY(Int_t param, const char* tname) const
            func = YUpsilonPP;
        } else if (sname == "CDF scaled") {
            func=YUpsilonCDFscaled;
+       } else if (sname == "Flat") {
+           func=YUpsilonFlat;
        } else {
            func=YUpsilon;
        }

diff --git a/EVGEN/AliGenMUONlib.h b/EVGEN/AliGenMUONlib.h
index b743971e8d8e9c234dbee3803f0f86f0ed28a7c9..2cf40258fe0c6e58b3c56a96fde50c59678483af 100644 (file)
--- a/EVGEN/AliGenMUONlib.h
+++ b/EVGEN/AliGenMUONlib.h
@@ -56,6 +56,8 @@ class AliGenMUONlib :
     static Int_t    IpJpsi(TRandom *ran);
     static Int_t    IpJpsiFamily(TRandom *ran);
     static Int_t    IpPsiP(TRandom *ran);
+    static Double_t PtJpsiFlat( Double_t *px, Double_t *dummy );
+    static Double_t YJpsiFlat(Double_t *py, Double_t *dummy);
 
 // Upsilon    
     static Double_t PtUpsilon( Double_t *px, Double_t *dummy );
@@ -70,6 +72,8 @@ class AliGenMUONlib :
     static Int_t    IpUpsilonFamily(TRandom *ran);
     static Int_t    IpUpsilonP(TRandom *ran);
     static Int_t    IpUpsilonPP(TRandom *ran);
+    static Double_t PtUpsilonFlat( Double_t *px, Double_t *dummy );
+    static Double_t YUpsilonFlat(Double_t *py, Double_t *dummy);
 //
 // Charm    
     static Double_t PtCharm( Double_t *px, Double_t *dummy );