]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - EVGEN/AliGenMUONlib.cxx
All the necessary for fast Chi_c generation. (P. Ladron de Guevara)
[u/mrichter/AliRoot.git] / EVGEN / AliGenMUONlib.cxx
index 32a62077a0da53816949855db1a75042bdc7f203..6b9b9a7b76d77ad57d27e6d821a8acf884264a55 100644 (file)
@@ -32,7 +32,7 @@
 ClassImp(AliGenMUONlib)
 //
 //  Pions
-Double_t AliGenMUONlib::PtPion(Double_t *px, Double_t *dummy)
+Double_t AliGenMUONlib::PtPion(Double_t *px, Double_t* /*dummy*/)
 {
 //
 //     PT-PARAMETERIZATION CDF, PRL 61(88) 1819
@@ -61,7 +61,7 @@ Double_t AliGenMUONlib::PtPion(Double_t *px, Double_t *dummy)
 //
 // y-distribution
 //
-Double_t AliGenMUONlib::YPion( Double_t *py, Double_t *dummy)
+Double_t AliGenMUONlib::YPion( Double_t *py, Double_t */*dummy*/)
 {
 // Pion y
   Double_t y=TMath::Abs(*py);
@@ -111,7 +111,7 @@ Double_t AliGenMUONlib::PtScal(Double_t pt, Int_t np)
 //
 //                pt-distribution
 //____________________________________________________________
-Double_t AliGenMUONlib::PtKaon( Double_t *px, Double_t *dummy)
+Double_t AliGenMUONlib::PtKaon( Double_t *px, Double_t */*dummy*/)
 {
 // Kaon pT
   return PtScal(*px,2);
@@ -119,7 +119,7 @@ Double_t AliGenMUONlib::PtKaon( Double_t *px, Double_t *dummy)
 
 // y-distribution
 //____________________________________________________________
-Double_t AliGenMUONlib::YKaon( Double_t *py, Double_t *dummy)
+Double_t AliGenMUONlib::YKaon( Double_t *py, Double_t */*dummy*/)
 {
 // Kaon y
   Double_t y=TMath::Abs(*py);
@@ -151,7 +151,7 @@ Int_t AliGenMUONlib::IpKaon(TRandom *ran)
 //
 //                pt-distribution
 //____________________________________________________________
-Double_t AliGenMUONlib::PtJpsi( Double_t *px, Double_t *dummy)
+Double_t AliGenMUONlib::PtJpsi( Double_t *px, Double_t */*dummy*/)
 {
 // J/Psi pT
   const Double_t kpt0 = 4.;
@@ -162,7 +162,39 @@ Double_t AliGenMUONlib::PtJpsi( Double_t *px, Double_t *dummy)
   return x/TMath::Power(pass1,kxn);
 }
 
-Double_t AliGenMUONlib::PtJpsiPbPb( Double_t *px, Double_t *dummy)
+Double_t AliGenMUONlib::PtJpsiCDFscaled( Double_t *px, Double_t */*dummy*/)
+{
+// J/Psi pT
+  const Double_t kpt0 = 4.703;
+  const Double_t kxn  = 3.826;
+  Double_t x=*px;
+  //
+  Double_t pass1 = 1.+(x/kpt0)*(x/kpt0);
+  return x/TMath::Power(pass1,kxn);
+}
+
+Double_t AliGenMUONlib::PtJpsiCDFscaledPP( Double_t *px, Double_t */*dummy*/)
+{
+// J/Psi pT
+//
+// pp 14 TeV
+//
+// scaled from CDF data at 2 TeV
+
+  const Double_t kpt0 = 5.355;
+  const Double_t kxn  = 3.821;
+  Double_t x=*px;
+  //
+  Double_t pass1 = 1.+(x/kpt0)*(x/kpt0);
+  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
 //
@@ -188,7 +220,22 @@ Double_t AliGenMUONlib::PtJpsiPbPb( Double_t *px, Double_t *dummy)
     }
     return y;
 }
-Double_t AliGenMUONlib::PtJpsiPP( Double_t *px, Double_t *dummy)
+
+Double_t AliGenMUONlib::PtJpsiBPbPb( Double_t *px, Double_t */*dummy*/)
+{
+// J/Psi pT spectrum
+// B -> J/Psi X
+    Double_t x0 =   4.0384;
+    Double_t  n =   3.0288;
+    
+    Double_t x = px[0];
+    Double_t y = x / TMath::Power((1. + (x/x0)*(x/x0)), n);
+    
+    return y;
+}
+
+
+Double_t AliGenMUONlib::PtJpsiPP( Double_t *px, Double_t */*dummy*/)
 {
 // J/Psi pT spectrum
 //
@@ -215,7 +262,7 @@ Double_t AliGenMUONlib::PtJpsiPP( Double_t *px, Double_t *dummy)
 //
 //               y-distribution
 //____________________________________________________________
-Double_t AliGenMUONlib::YJpsi(Double_t *py, Double_t *dummy)
+Double_t AliGenMUONlib::YJpsi(Double_t *py, Double_t */*dummy*/)
 {
 // J/psi y
   const Double_t ky0 = 4.;
@@ -230,8 +277,13 @@ 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)
+
+Double_t AliGenMUONlib::YJpsiPbPb( Double_t *px, Double_t */*dummy*/)
 {
 
 //
@@ -260,7 +312,19 @@ Double_t AliGenMUONlib::YJpsiPbPb( Double_t *px, Double_t *dummy)
     return y;
 }
 
-Double_t AliGenMUONlib::YJpsiPP( Double_t *px, Double_t *dummy)
+Double_t AliGenMUONlib::YJpsiCDFscaled( Double_t *px, Double_t* dummy)
+{
+    // J/Psi y 
+    return AliGenMUONlib::YJpsiPbPb(px, dummy);
+}
+
+Double_t AliGenMUONlib::YJpsiCDFscaledPP( Double_t *px, Double_t* dummy)
+{
+    // J/Psi y 
+    return AliGenMUONlib::YJpsiPP(px, dummy);
+}
+
+Double_t AliGenMUONlib::YJpsiPP( Double_t *px, Double_t */*dummy*/)
 {
 
 //
@@ -290,6 +354,33 @@ Double_t AliGenMUONlib::YJpsiPP( Double_t *px, Double_t *dummy)
     return y;
 }
 
+Double_t AliGenMUONlib::YJpsiBPbPb( Double_t *px, Double_t */*dummy*/)
+{
+
+//
+// J/Psi from B->J/Psi X
+//
+//
+    
+
+    Double_t c[7] = {7.37025e-02, 0., -2.94487e-03, 0., 6.07953e-06, 0., 5.39219e-07};
+    
+    Double_t x = TMath::Abs(px[0]);
+    Double_t y;
+    
+    if (x > 6.) {
+       y = 0.;
+    } else {
+       Int_t j;
+       y = c[j = 6];
+       while (j > 0) y  = y * x + c[--j];
+    } 
+    
+    return y;
+}
+
+
+
 //                 particle composition
 //
 Int_t AliGenMUONlib::IpJpsi(TRandom *)
@@ -297,13 +388,32 @@ Int_t AliGenMUONlib::IpJpsi(TRandom *)
 // J/Psi composition
     return 443;
 }
+Int_t AliGenMUONlib::IpPsiP(TRandom *)
+{
+// Psi prime composition
+    return 100443;
+}
+Int_t AliGenMUONlib::IpJpsiFamily(TRandom *)
+{
+// J/Psi composition
+  Int_t ip;
+  Float_t r = gRandom->Rndm();
+  if (r < 0.98) {
+    ip = 443;
+  } else {
+    ip = 100443;
+  }
+  return ip;
+}
+
+
 
 //                      Upsilon
 //
 //
 //                  pt-distribution
 //____________________________________________________________
-Double_t AliGenMUONlib::PtUpsilon( Double_t *px, Double_t *dummy )
+Double_t AliGenMUONlib::PtUpsilon( Double_t *px, Double_t */*dummy*/ )
 {
 // Upsilon pT
   const Double_t kpt0 = 5.3;
@@ -314,7 +424,39 @@ Double_t AliGenMUONlib::PtUpsilon( Double_t *px, Double_t *dummy )
   return x/TMath::Power(pass1,kxn);
 }
 
-Double_t AliGenMUONlib::PtUpsilonPbPb( Double_t *px, Double_t *dummy)
+Double_t AliGenMUONlib::PtUpsilonCDFscaled( Double_t *px, Double_t */*dummy*/ )
+{
+// Upsilon pT
+  const Double_t kpt0 = 7.753;
+  const Double_t kxn  = 3.042;
+  Double_t x=*px;
+  //
+  Double_t pass1 = 1.+(x/kpt0)*(x/kpt0);
+  return x/TMath::Power(pass1,kxn);
+}
+
+Double_t AliGenMUONlib::PtUpsilonCDFscaledPP( Double_t *px, Double_t */*dummy*/ )
+{
+// Upsilon pT
+//
+// pp 14 TeV
+//
+// scaled from CDF data at 2 TeV
+
+  const Double_t kpt0 = 8.610;
+  const Double_t kxn  = 3.051;
+  Double_t x=*px;
+  //
+  Double_t pass1 = 1.+(x/kpt0)*(x/kpt0);
+  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*/)
 {
 
 //
@@ -343,7 +485,7 @@ Double_t AliGenMUONlib::PtUpsilonPbPb( Double_t *px, Double_t *dummy)
     return y;
 }
 
-Double_t AliGenMUONlib::PtUpsilonPP( Double_t *px, Double_t *dummy)
+Double_t AliGenMUONlib::PtUpsilonPP( Double_t *px, Double_t */*dummy*/)
 {
 
 //
@@ -375,7 +517,7 @@ Double_t AliGenMUONlib::PtUpsilonPP( Double_t *px, Double_t *dummy)
 //                    y-distribution
 //
 //____________________________________________________________
-Double_t AliGenMUONlib::YUpsilon(Double_t *py, Double_t *dummy)
+Double_t AliGenMUONlib::YUpsilon(Double_t *py, Double_t */*dummy*/)
 {
 // Upsilon y
   const Double_t ky0 = 3.;
@@ -391,7 +533,7 @@ Double_t AliGenMUONlib::YUpsilon(Double_t *py, Double_t *dummy)
 }
 
 
-Double_t AliGenMUONlib::YUpsilonPbPb( Double_t *px, Double_t *dummy)
+Double_t AliGenMUONlib::YUpsilonPbPb( Double_t *px, Double_t */*dummy*/)
 {
 
 //
@@ -415,7 +557,28 @@ Double_t AliGenMUONlib::YUpsilonPbPb( Double_t *px, Double_t *dummy)
     return y;
 }
 
-Double_t AliGenMUONlib::YUpsilonPP( Double_t *px, Double_t *dummy)
+Double_t AliGenMUONlib::YUpsilonCDFscaled( Double_t *px, Double_t *dummy)
+{
+    // Upsilon y
+    return AliGenMUONlib::YUpsilonPbPb(px, dummy);
+    
+}
+
+Double_t AliGenMUONlib::YUpsilonCDFscaledPP( Double_t *px, Double_t *dummy)
+{
+    // Upsilon y
+    return AliGenMUONlib::YUpsilonPP(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*/)
 {
 
 //
@@ -445,6 +608,32 @@ Int_t AliGenMUONlib::IpUpsilon(TRandom *)
 // y composition
     return 553;
 }
+Int_t AliGenMUONlib::IpUpsilonP(TRandom *)
+{
+// y composition
+    return 100553;
+}
+Int_t AliGenMUONlib::IpUpsilonPP(TRandom *)
+{
+// y composition
+    return 200553;
+}
+Int_t AliGenMUONlib::IpUpsilonFamily(TRandom *)
+{
+// y composition
+  Int_t ip;
+  Float_t r = gRandom->Rndm();
+  
+  if (r < 0.712) {
+    ip = 553;
+  } else if (r < 0.896) {
+    ip = 100553;
+  } else {
+    ip = 200553;
+  }
+  return ip;
+}
+
 
 //
 //                        Phi
@@ -452,13 +641,13 @@ Int_t AliGenMUONlib::IpUpsilon(TRandom *)
 //
 //    pt-distribution (by scaling of pion distribution)
 //____________________________________________________________
-Double_t AliGenMUONlib::PtPhi( Double_t *px, Double_t *dummy)
+Double_t AliGenMUONlib::PtPhi( Double_t *px, Double_t */*dummy*/)
 {
 // Phi pT
   return PtScal(*px,7);
 }
 //    y-distribution
-Double_t AliGenMUONlib::YPhi( Double_t *px, Double_t *dummy)
+Double_t AliGenMUONlib::YPhi( Double_t *px, Double_t */*dummy*/)
 {
 // Phi y
     Double_t *dum=0;
@@ -478,13 +667,13 @@ Int_t AliGenMUONlib::IpPhi(TRandom *)
 //
 //    pt-distribution (by scaling of pion distribution)
 //____________________________________________________________
-Double_t AliGenMUONlib::PtOmega( Double_t *px, Double_t *dummy)
+Double_t AliGenMUONlib::PtOmega( Double_t *px, Double_t */*dummy*/)
 {
 // Omega pT
   return PtScal(*px,5);
 }
 //    y-distribution
-Double_t AliGenMUONlib::YOmega( Double_t *px, Double_t *dummy)
+Double_t AliGenMUONlib::YOmega( Double_t *px, Double_t */*dummy*/)
 {
 // Omega y
     Double_t *dum=0;
@@ -505,13 +694,13 @@ Int_t AliGenMUONlib::IpOmega(TRandom *)
 //
 //    pt-distribution (by scaling of pion distribution)
 //____________________________________________________________
-Double_t AliGenMUONlib::PtEta( Double_t *px, Double_t *dummy)
+Double_t AliGenMUONlib::PtEta( Double_t *px, Double_t */*dummy*/)
 {
 // Eta pT
   return PtScal(*px,3);
 }
 //    y-distribution
-Double_t AliGenMUONlib::YEta( Double_t *px, Double_t *dummy)
+Double_t AliGenMUONlib::YEta( Double_t *px, Double_t */*dummy*/)
 {
 // Eta y
     Double_t *dum=0;
@@ -531,25 +720,52 @@ Int_t AliGenMUONlib::IpEta(TRandom *)
 //
 //                    pt-distribution
 //____________________________________________________________
-Double_t AliGenMUONlib::PtCharm( Double_t *px, Double_t *dummy)
+Double_t AliGenMUONlib::PtCharm( Double_t *px, Double_t */*dummy*/)
+{
+// Charm pT
+  const Double_t kpt0 = 2.25;
+  const Double_t kxn  = 3.17;
+
+  Double_t x=*px;
+  //
+  Double_t pass1 = 1.+(x/kpt0)*(x/kpt0);
+  return x/TMath::Power(pass1,kxn);
+}
+
+Double_t AliGenMUONlib::PtCharmCentral( Double_t *px, Double_t */*dummy*/)
 {
 // Charm pT
-  const Double_t kpt0 = 4.08;
-  const Double_t kxn  = 9.40;
+  const Double_t kpt0 = 2.12;
+  const Double_t kxn  = 2.78;
 
   Double_t x=*px;
   //
-  Double_t pass1 = 1.+(x/kpt0);
+  Double_t pass1 = 1.+(x/kpt0)*(x/kpt0);
   return x/TMath::Power(pass1,kxn);
 }
 //                  y-distribution
-Double_t AliGenMUONlib::YCharm( Double_t *px, Double_t *dummy)
+Double_t AliGenMUONlib::YCharm( Double_t *px, Double_t */*dummy*/)
 {
-// Charm y
-    Double_t *dum=0;
-    return YJpsi(px,dum);
+// Charm y :: Carrer & Dainese : ALICE-INT-2003-019 v.3 (hep-ph/0311225) 
+// Pythia tuned to reproduce the distribution given by the HVQMNR program based on NLO calculations (pQCD)
+// shadowing + kt broadening 
+
+    Double_t x=px[0];
+    Double_t c[2]={-2.42985e-03,-2.31001e-04};
+    Double_t y=1+(c[0]*TMath::Power(x,2))+(c[1]*TMath::Power(x,4));
+    Double_t ycharm;
+    
+    if (TMath::Abs(x)>8) {
+      ycharm=0.;
+    }
+    else {
+      ycharm=TMath::Power(y,3);
+    }
+    
+    return ycharm;
 }
 
+
 Int_t AliGenMUONlib::IpCharm(TRandom *ran)
 {  
 // Charm composition
@@ -557,65 +773,112 @@ Int_t AliGenMUONlib::IpCharm(TRandom *ran)
     Int_t ip;
 //    411,421,431,4122
     random = ran->Rndm();
-    if (random < 0.5) {
-       ip=411;
-    } else if (random < 0.75) {
-       ip=421;
-    } else if (random < 0.90) {
-       ip=431;
+//  Taux de production Carrer & Dainese : ALICE-INT-2003-019 v.3  
+//  >>>>> cf. tab 4 p 11
+  
+    if (random < 0.30) {                       
+        ip=421;
+    } else if (random < 0.60) {
+        ip=-421;
+    } else if (random < 0.70) {
+        ip=411;
+    } else if (random < 0.80) {
+        ip=-411;
+    } else if (random < 0.86) {
+        ip=431;
+    } else if (random < 0.92) {
+        ip=-431;       
+    } else if (random < 0.96) {
+        ip=4122;
     } else {
-       ip=4122;
+        ip=-4122;
     }
-    if (ran->Rndm() < 0.5) {ip=-ip;}
     
     return ip;
 }
 
-
 //
 //                        Beauty
 //
 //
 //                    pt-distribution
 //____________________________________________________________
-Double_t AliGenMUONlib::PtBeauty( Double_t *px, Double_t *dummy)
+Double_t AliGenMUONlib::PtBeauty( Double_t *px, Double_t */*dummy*/)
 {
 // Beauty pT
-  const Double_t kpt0 = 4.;
-  const Double_t kxn  = 3.6;
+  const Double_t kpt0 = 6.53;
+  const Double_t kxn  = 3.59;
+  Double_t x=*px;
+  //
+  Double_t pass1 = 1.+(x/kpt0)*(x/kpt0);
+  return x/TMath::Power(pass1,kxn);
+}
+
+Double_t AliGenMUONlib::PtBeautyCentral( Double_t *px, Double_t */*dummy*/)
+{
+// Beauty pT
+  const Double_t kpt0 = 6.14;
+  const Double_t kxn  = 2.93;
   Double_t x=*px;
   //
   Double_t pass1 = 1.+(x/kpt0)*(x/kpt0);
   return x/TMath::Power(pass1,kxn);
 }
 //                     y-distribution
-Double_t AliGenMUONlib::YBeauty( Double_t *px, Double_t *dummy)
+Double_t AliGenMUONlib::YBeauty( Double_t *px, Double_t */*dummy*/)
 {
-// Beauty y
-    Double_t *dum=0;
-    return YJpsi(px,dum);
+// Beauty y :: Carrer & Dainese : ALICE-INT-2003-019 v.3 (hep-ph/0311225) 
+// Pythia tuned to reproduce the distribution given by the HVQMNR program based on NLO calculations (pQCD)
+// shadowing + kt broadening 
+
+    Double_t x=px[0];
+    Double_t c[2]={-1.27590e-02,-2.42731e-04};
+    Double_t y=1+c[0]*TMath::Power(x,2)+c[1]*TMath::Power(x,4);
+    Double_t ybeauty;
+    
+    if (TMath::Abs(x)>6) {
+      ybeauty=0.;
+    }
+    else {
+      ybeauty=TMath::Power(y,3);
+    }
+    
+    return ybeauty;
 }
 
+
 Int_t AliGenMUONlib::IpBeauty(TRandom *ran)
 {  
 // Beauty Composition
     Float_t random;
     Int_t ip;
-    random = ran->Rndm();
-    if (random < 0.5) {
-       ip=511;
-    } else if (random < 0.75) {
-       ip=521;
-    } else if (random < 0.90) {
-       ip=531;
+    random = ran->Rndm(); 
+    
+//  Taux de production Carrer & Dainese : ALICE-INT-2003-019 v.3  
+//  >>>>> cf. tab 4 p 11
+    
+ if (random < 0.20) {                       
+        ip=511;
+    } else if (random < 0.40) {
+        ip=-511;
+    } else if (random < 0.605) {
+        ip=521;
+    } else if (random < 0.81) {
+        ip=-521;
+    } else if (random < 0.87) {
+        ip=531;
+    } else if (random < 0.93) {
+        ip=-531;       
+    } else if (random < 0.965) {
+        ip=5122;
     } else {
-       ip=5122;
+        ip=-5122;
     }
-    if (ran->Rndm() < 0.5) {ip=-ip;}
     
   return ip;
+ return ip;
 }
 
+
 typedef Double_t (*GenFunc) (Double_t*,  Double_t*);
 GenFunc AliGenMUONlib::GetPt(Int_t param,  const char* tname) const
 {
@@ -633,29 +896,57 @@ GenFunc AliGenMUONlib::GetPt(Int_t param,  const char* tname) const
     case kEta:
        func=PtEta;
        break;
+    case kJpsiFamily:
+    case kPsiP:
     case kJpsi:
        if (sname == "Vogt" || sname == "Vogt PbPb") {
            func=PtJpsiPbPb;
        } else if (sname == "Vogt pp") {
            func=PtJpsiPP;
+       } else if (sname == "CDF scaled") {
+           func=PtJpsiCDFscaled;
+       } else if (sname == "CDF pp") {
+           func=PtJpsiCDFscaledPP;
+       } else if (sname == "Flat") {
+           func=PtJpsiFlat;
        } else {
            func=PtJpsi;
        }
        break;
+    case kJpsiFromB:
+       func = PtJpsiBPbPb;
+       break;
+    case kUpsilonFamily:
+    case kUpsilonP:
+    case kUpsilonPP:
     case kUpsilon:
        if (sname == "Vogt" || sname == "Vogt PbPb") {
            func=PtUpsilonPbPb;
        } else if (sname == "Vogt pp") {
            func=PtUpsilonPP;
+       } else if (sname == "CDF scaled") {
+           func=PtUpsilonCDFscaled;
+       } else if (sname == "CDF pp") {
+           func=PtUpsilonCDFscaledPP;
+       } else if (sname == "Flat") {
+           func=PtUpsilonFlat;
        } else {
            func=PtUpsilon;
        }
-       break;
+       break;  
     case kCharm:
-       func=PtCharm;
+       if (sname == "central") {
+           func=PtCharmCentral;
+       } else {
+           func=PtCharm;
+       }
        break;
     case kBeauty:
-       func=PtBeauty;
+       if (sname == "central") {
+           func=PtBeautyCentral;
+       } else {
+           func=PtBeauty;
+       }
        break;
     case kPion:
        func=PtPion;
@@ -663,6 +954,18 @@ GenFunc AliGenMUONlib::GetPt(Int_t param,  const char* tname) const
     case kKaon:
        func=PtKaon;
        break;
+    case kChi_c0:
+       func=PtChi_c0;
+       break;
+    case kChi_c1:
+       func=PtChi_c1;
+       break;
+    case kChi_c2:
+       func=PtChi_c2;
+       break;
+    case kChi_c:
+       func=PtChi_c;
+       break;
     default:
         func=0;
         printf("<AliGenMUONlib::GetPt> unknown parametrisation\n");
@@ -672,9 +975,10 @@ GenFunc AliGenMUONlib::GetPt(Int_t param,  const char* tname) const
 
 GenFunc AliGenMUONlib::GetY(Int_t param, const char* tname) const
 {
+  //    
+  // Return pointer to y- parameterisation
+  //
     TString sname = TString(tname);
-    
-// Return pointer to y- parameterisation
     GenFunc func;
     switch (param) 
     {
@@ -687,21 +991,40 @@ GenFunc AliGenMUONlib::GetY(Int_t param, const char* tname) const
     case kOmega:
        func=YOmega;
        break;
+    case kJpsiFamily:
+    case kPsiP:
     case kJpsi:
        if (sname == "Vogt" || sname == "Vogt PbPb") {
            func=YJpsiPbPb;
        } else if (sname == "Vogt pp"){
            func=YJpsiPP;
+       } else if (sname == "CDF scaled") {
+           func=YJpsiCDFscaled;
+       } else if (sname == "CDF pp") {
+           func=YJpsiCDFscaledPP;
+       } else if (sname == "Flat") {
+           func=YJpsiFlat;
        } else {
            func=YJpsi;
        }
-       
        break;
+    case kJpsiFromB:
+       func = YJpsiBPbPb;
+       break;
+    case kUpsilonFamily:
+    case kUpsilonP:
+    case kUpsilonPP:
     case kUpsilon:
        if (sname == "Vogt" || sname == "Vogt PbPb") {
            func=YUpsilonPbPb;
        } else if (sname == "Vogt pp") {
            func = YUpsilonPP;
+       } else if (sname == "CDF scaled") {
+           func=YUpsilonCDFscaled;
+       } else if (sname == "CDF pp") {
+           func=YUpsilonCDFscaledPP;
+       } else if (sname == "Flat") {
+           func=YUpsilonFlat;
        } else {
            func=YUpsilon;
        }
@@ -718,14 +1041,173 @@ GenFunc AliGenMUONlib::GetY(Int_t param, const char* tname) const
     case kKaon:
        func=YKaon;
        break;
+    case kChi_c0:
+       func=YChi_c0;
+       break;
+    case kChi_c1:
+       func=YChi_c1;
+       break;
+    case kChi_c2:
+       func=YChi_c2;
+       break;
+    case kChi_c:
+       func=YChi_c;
+       break;
     default:
         func=0;
         printf("<AliGenMUONlib::GetY> unknown parametrisation\n");
     }
     return func;
 }
+
+//
+//                    Chi
+//
+//
+//                pt-distribution
+//____________________________________________________________
+Double_t AliGenMUONlib::PtChi_c0( Double_t *px, Double_t */*dummy*/)
+{
+// Chi_c1 pT
+  const Double_t kpt0 = 4.;
+  const Double_t kxn  = 3.6;
+  Double_t x=*px;
+  //
+  Double_t pass1 = 1.+(x/kpt0)*(x/kpt0);
+  return x/TMath::Power(pass1,kxn);
+}
+Double_t AliGenMUONlib::PtChi_c1( Double_t *px, Double_t */*dummy*/)
+{
+// Chi_c1 pT
+  const Double_t kpt0 = 4.;
+  const Double_t kxn  = 3.6;
+  Double_t x=*px;
+  //
+  Double_t pass1 = 1.+(x/kpt0)*(x/kpt0);
+  return x/TMath::Power(pass1,kxn);
+}
+Double_t AliGenMUONlib::PtChi_c2( Double_t *px, Double_t */*dummy*/)
+{
+// Chi_c2 pT
+  const Double_t kpt0 = 4.;
+  const Double_t kxn  = 3.6;
+  Double_t x=*px;
+  //
+  Double_t pass1 = 1.+(x/kpt0)*(x/kpt0);
+  return x/TMath::Power(pass1,kxn);
+}
+Double_t AliGenMUONlib::PtChi_c( Double_t *px, Double_t */*dummy*/)
+{
+// Chi_c family pT
+  const Double_t kpt0 = 4.;
+  const Double_t kxn  = 3.6;
+  Double_t x=*px;
+  //
+  Double_t pass1 = 1.+(x/kpt0)*(x/kpt0);
+  return x/TMath::Power(pass1,kxn);
+}
+
+//
+//               y-distribution
+//____________________________________________________________
+Double_t AliGenMUONlib::YChi_c0(Double_t *py, Double_t */*dummy*/)
+{
+// Chi-1c y
+  const Double_t ky0 = 4.;
+ const Double_t kb=1.;
+  Double_t yj;
+  Double_t y=TMath::Abs(*py);
+  //
+  if (y < ky0)
+    yj=kb;
+  else
+    yj=kb*TMath::Exp(-(y-ky0)*(y-ky0)/2);
+  return yj;
+}
+
+Double_t AliGenMUONlib::YChi_c1(Double_t *py, Double_t */*dummy*/)
+{
+// Chi-1c y
+  const Double_t ky0 = 4.;
+  const Double_t kb=1.;
+  Double_t yj;
+  Double_t y=TMath::Abs(*py);
+  //
+  if (y < ky0)
+    yj=kb;
+  else
+    yj=kb*TMath::Exp(-(y-ky0)*(y-ky0)/2);
+  return yj;
+}
+
+Double_t AliGenMUONlib::YChi_c2(Double_t *py, Double_t */*dummy*/)
+{
+// Chi-2c y
+  const Double_t ky0 = 4.;
+  const Double_t kb=1.;
+  Double_t yj;
+  Double_t y=TMath::Abs(*py);
+  //
+  if (y < ky0)
+    yj=kb;
+  else
+    yj=kb*TMath::Exp(-(y-ky0)*(y-ky0)/2);
+  return yj;
+}
+
+Double_t AliGenMUONlib::YChi_c(Double_t *py, Double_t */*dummy*/)
+{
+// Chi_c family y
+  const Double_t ky0 = 4.;
+  const Double_t kb=1.;
+  Double_t yj;
+  Double_t y=TMath::Abs(*py);
+  //
+  if (y < ky0)
+    yj=kb;
+  else
+    yj=kb*TMath::Exp(-(y-ky0)*(y-ky0)/2);
+  return yj;
+}
+
+//                 particle composition
+//
+Int_t AliGenMUONlib::IpChi_c0(TRandom *)
+{
+// Chi composition
+    return 10441;
+}
+//
+Int_t AliGenMUONlib::IpChi_c1(TRandom *)
+{
+// Chi composition
+    return 20443;
+}
+Int_t AliGenMUONlib::IpChi_c2(TRandom *)
+{
+// Chi_c2 prime composition
+    return 445;
+}
+Int_t AliGenMUONlib::IpChi_c(TRandom *)
+{
+// Chi composition
+  Int_t ip;
+  Float_t r = gRandom->Rndm();
+  if (r < 0.001) {
+    ip = 10441;
+  } else if( r < 0.377 ) {
+    ip = 20443;
+  } else {
+    ip = 445;
+  }
+  return ip;
+}
+
+
+//_____________________________________________________________
+
 typedef Int_t (*GenFuncIp) (TRandom *);
-GenFuncIp AliGenMUONlib::GetIp(Int_t param,  const char* tname) const
+GenFuncIp AliGenMUONlib::GetIp(Int_t param,  const char* /*tname*/) const
 {
 // Return pointer to particle type parameterisation
     GenFuncIp func;
@@ -740,12 +1222,28 @@ GenFuncIp AliGenMUONlib::GetIp(Int_t param,  const char* tname) const
     case kOmega:
        func=IpOmega;
        break;
+    case kJpsiFamily:
+       func=IpJpsiFamily;
+       break;
+    case kPsiP:
+       func=IpPsiP;
+       break;
     case kJpsi:
+    case kJpsiFromB:
        func=IpJpsi;
        break;
     case kUpsilon:
        func=IpUpsilon;
        break;
+    case kUpsilonFamily:
+      func=IpUpsilonFamily;
+      break;
+    case kUpsilonP:
+       func=IpUpsilonP;
+       break;
+    case kUpsilonPP:
+       func=IpUpsilonPP;
+       break;
     case kCharm:
        func=IpCharm;
        break;
@@ -758,6 +1256,18 @@ GenFuncIp AliGenMUONlib::GetIp(Int_t param,  const char* tname) const
     case kKaon:
        func=IpKaon;
        break;
+    case kChi_c0:
+       func=IpChi_c0;
+       break;
+    case kChi_c1:
+       func=IpChi_c1;
+       break;
+    case kChi_c2:
+       func=IpChi_c2;
+       break;
+    case kChi_c:
+        func=IpChi_c;
+        break;
     default:
         func=0;
         printf("<AliGenMUONlib::GetIp> unknown parametrisation\n");