]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - EVGEN/AliGenMUONlib.cxx
Quarkonia for pp @ 14 TeV added.
[u/mrichter/AliRoot.git] / EVGEN / AliGenMUONlib.cxx
index 53a4597f59cad7cb2c7d03cd5aecb83c48b7e087..73bc6d47e768a84ab67a7a1d6f658e8c79288304 100644 (file)
 
 /*
 $Log$
+Revision 1.20  2003/03/13 11:54:39  morsch
+Limited pT range for parameterized Upsilon and J/Psi pT distributions.
+
+Revision 1.19  2003/02/24 16:46:11  morsch
+New parameterisation for Psi and Upsilon (PbPb)
+
+Revision 1.18  2002/11/07 09:13:42  morsch
+Use "Vogt" to label new distributions.
+
+Revision 1.17  2002/11/07 09:06:10  morsch
+J/Psi and Upsilon pt and y distributions from R. Vogt 2002 added.
+
+Revision 1.16  2002/10/14 14:55:35  hristov
+Merging the VirtualMC branch to the main development branch (HEAD)
+
+Revision 1.14.6.1  2002/06/10 14:57:41  hristov
+Merged with v3-08-02
+
+Revision 1.15  2002/04/17 10:11:51  morsch
+Coding Rule violations corrected.
+
+Revision 1.14  2002/02/22 17:26:43  morsch
+Eta and omega added.
+
+Revision 1.13  2001/03/27 11:01:04  morsch
+Charm pt-distribution corrected. More realistic y-distribution for pi and K.
+
+Revision 1.12  2001/03/09 13:01:41  morsch
+- enum constants for paramterisation type (particle family) moved to AliGen*lib.h
+- use AliGenGSIlib::kUpsilon, AliGenPHOSlib::kEtaPrime to access the constants
+
+Revision 1.11  2000/11/30 07:12:50  alibrary
+Introducing new Rndm and QA classes
+
 Revision 1.10  2000/06/29 21:08:27  morsch
 All paramatrisation libraries derive from the pure virtual base class AliGenLib.
 This allows to pass a pointer to a library directly to AliGenParam and avoids the
@@ -34,6 +68,15 @@ Introduction of the Copyright and cvs Log
 
 */
 
+// Library class for particle pt and y distributions used for 
+// muon spectrometer simulations.
+// To be used with AliGenParam.
+// The following particle typed can be simulated:
+// pi, K, phi, omega, eta, J/Psi, Upsilon, charm and beauty mesons. 
+//
+// andreas.morsch@cern.ch
+//
+
 #include "TMath.h"
 #include "TRandom.h"
 
@@ -74,13 +117,15 @@ Double_t AliGenMUONlib::PtPion(Double_t *px, Double_t *dummy)
 Double_t AliGenMUONlib::YPion( Double_t *py, Double_t *dummy)
 {
 // Pion y
+  Double_t y=TMath::Abs(*py);
+/*
   const Double_t ka    = 7000.;
   const Double_t kdy   = 4.;
-
-  Double_t y=TMath::Abs(*py);
-  //
   Double_t ex = y*y/(2*kdy*kdy);
   return ka*TMath::Exp(-ex);
+*/
+  return 1.16526e+04+y*-3.79886e+03+y*y*4.31130e+02;
+  
 }
 //                 particle composition
 //
@@ -130,14 +175,16 @@ Double_t AliGenMUONlib::PtKaon( Double_t *px, Double_t *dummy)
 Double_t AliGenMUONlib::YKaon( Double_t *py, Double_t *dummy)
 {
 // Kaon y
+  Double_t y=TMath::Abs(*py);
+/*
   const Double_t ka    = 1000.;
   const Double_t kdy   = 4.;
-  
-
-  Double_t y=TMath::Abs(*py);
   //
   Double_t ex = y*y/(2*kdy*kdy);
   return ka*TMath::Exp(-ex);
+*/
+
+  return 1.16526e+04+y*-3.79886e+03+y*y*4.31130e+02;
 }
 
 //                 particle composition
@@ -167,6 +214,57 @@ Double_t AliGenMUONlib::PtJpsi( Double_t *px, Double_t *dummy)
   Double_t pass1 = 1.+(x/kpt0)*(x/kpt0);
   return x/TMath::Power(pass1,kxn);
 }
+
+Double_t AliGenMUONlib::PtJpsiPbPb( Double_t *px, Double_t *dummy)
+{
+// J/Psi pT spectrum
+//
+// R. Vogt 2002
+// PbPb 5.5 TeV
+// MRST HO
+// mc = 1.4 GeV, pt-kick 1 GeV
+//
+    Float_t x = px[0];
+    Float_t c[8] = {
+       -2.13098e+00, 9.46552e+00, -5.06799e+00, 1.27260e+00, 
+       -1.83806e-01, 1.55853e-02, -7.23241e-04, 1.42105e-05
+    };
+    
+    Double_t y;
+    if (x < 10.) {
+       Int_t j;
+       y = c[j = 7];
+       while (j > 0) y  = y * x +c[--j];
+       y = x * TMath::Exp(y);
+    } else {
+       y = 0.;
+    }
+    return y;
+}
+Double_t AliGenMUONlib::PtJpsiPP( Double_t *px, Double_t *dummy)
+{
+// J/Psi pT spectrum
+//
+// R. Vogt 2002
+// pp 14 TeV
+// MRST HO
+// mc = 1.4 GeV, pt-kick 1 GeV
+//
+    Float_t x = px[0];
+    Float_t c[4] = {8.47471e+00, -1.93567e+00, 1.50271e-01, -5.51212e-03};
+    Double_t y;
+    if (x < 10.) {
+       Int_t j;
+       y = c[j = 3];
+       while (j > 0) y  = y * x +c[--j];
+       y = x * TMath::Exp(y);
+    } else {
+       y = 0.;
+    }
+    return y;
+}
+
 //
 //               y-distribution
 //____________________________________________________________
@@ -184,6 +282,67 @@ Double_t AliGenMUONlib::YJpsi(Double_t *py, Double_t *dummy)
     yj=kb*TMath::Exp(-(y-ky0)*(y-ky0)/2);
   return yj;
 }
+
+
+Double_t AliGenMUONlib::YJpsiPbPb( Double_t *px, Double_t *dummy)
+{
+
+//
+// J/Psi y
+//
+//
+// R. Vogt 2002
+// PbPb 5.5 TeV
+// MRST HO
+// mc = 1.4 GeV, pt-kick 1 GeV
+//
+    Double_t c[5] = {-6.03425e+02, 4.98257e+02, -1.38794e+02, 1.62209e+01, -6.85955e-01};
+    Double_t x = TMath::Abs(px[0]);
+    Double_t y;
+    
+    if (x < 4.) {
+       y = 31.754;
+    } else if (x < 6) {
+       Int_t j;
+       y = c[j = 4];
+       while (j > 0) y  = y * x + c[--j];
+    } else {
+       y =0.;
+    }
+    
+    return y;
+}
+
+Double_t AliGenMUONlib::YJpsiPP( Double_t *px, Double_t *dummy)
+{
+
+//
+// J/Psi y
+//
+//
+// R. Vogt 2002
+// pp 14  TeV
+// MRST HO
+// mc = 1.4 GeV, pt-kick 1 GeV
+//
+
+    Double_t c[5] = {1.38532e+00, 1.00596e+02, -3.46378e+01, 3.94172e+00, -1.48319e-01};
+    Double_t x = TMath::Abs(px[0]);
+    Double_t y;
+    
+    if (x < 2.5) {
+       y = 96.455 - 0.8483 * x * x;
+    } else if (x < 7.9) {
+       Int_t j;
+       y = c[j = 4];
+       while (j > 0) y  = y * x + c[--j];
+    } else {
+       y =0.;
+    }
+    
+    return y;
+}
+
 //                 particle composition
 //
 Int_t AliGenMUONlib::IpJpsi(TRandom *)
@@ -207,6 +366,64 @@ Double_t AliGenMUONlib::PtUpsilon( Double_t *px, Double_t *dummy )
   Double_t pass1 = 1.+(x/kpt0)*(x/kpt0);
   return x/TMath::Power(pass1,kxn);
 }
+
+Double_t AliGenMUONlib::PtUpsilonPbPb( Double_t *px, Double_t *dummy)
+{
+
+//
+// Upsilon pT
+//
+//
+// R. Vogt 2002
+// PbPb 5.5 TeV
+// MRST HO
+// mc = 1.4 GeV, pt-kick 1 GeV
+//
+    Float_t x = px[0];
+    Double_t c[8] = {
+       -1.03488e+01, 1.28065e+01, -6.60500e+00, 1.66140e+00,       
+       -2.34293e-01, 1.86925e-02, -7.80708e-04, 1.30610e-05
+    };
+    Double_t y;
+    if (x < 10.) {
+       Int_t j;
+       y = c[j = 7];
+       while (j > 0) y  = y * x +c[--j];
+       y = x * TMath::Exp(y);
+    } else {
+       y = 0.;
+    }
+    return y;
+}
+
+Double_t AliGenMUONlib::PtUpsilonPP( Double_t *px, Double_t *dummy)
+{
+
+//
+// Upsilon pT
+//
+//
+// R. Vogt 2002
+// pp 14 TeV
+// MRST HO
+// mc = 1.4 GeV, pt-kick 1 GeV
+//
+    Float_t x = px[0];
+    Double_t c[8] = {-7.93955e+00, 1.06306e+01, -5.21392e+00, 1.19703e+00,   
+                    -1.45718e-01, 8.95151e-03, -2.04806e-04, -1.13053e-06};
+    
+    Double_t y;
+    if (x < 10.) {
+       Int_t j;
+       y = c[j = 7];
+       while (j > 0) y  = y * x +c[--j];
+       y = x * TMath::Exp(y);
+    } else {
+       y = 0.;
+    }
+    return y;
+}
+
 //
 //                    y-distribution
 //
@@ -225,6 +442,55 @@ Double_t AliGenMUONlib::YUpsilon(Double_t *py, Double_t *dummy)
     yu=kb*TMath::Exp(-(y-ky0)*(y-ky0)/2);
   return yu;
 }
+
+
+Double_t AliGenMUONlib::YUpsilonPbPb( Double_t *px, Double_t *dummy)
+{
+
+//
+// Upsilon y
+//
+//
+// R. Vogt 2002
+// PbPb 5.5 TeV
+// MRST HO
+// mc = 1.4 GeV, pt-kick 1 GeV
+//
+
+    Double_t c[7] = {3.40036e-01, -3.98882e-07, -4.48398e-03, 8.46411e-08, -6.10854e-04,
+                    -2.99753e-09, 1.28895e-05};
+        
+    Double_t x = px[0];
+    if (TMath::Abs(x) > 5.55) return 0.;
+    Int_t j;
+    Double_t y = c[j = 6];
+    while (j > 0) y  = y * x +c[--j];
+    return y;
+}
+
+Double_t AliGenMUONlib::YUpsilonPP( Double_t *px, Double_t *dummy)
+{
+
+//
+// Upsilon y
+//
+//
+// R. Vogt 2002
+// p p  14. TeV
+// MRST HO
+// mc = 1.4 GeV, pt-kick 1 GeV
+//
+    Double_t c[7] = {8.91936e-01, -6.46645e-07, -1.52774e-02, 4.28677e-08, -7.01517e-04, 
+                    -6.20539e-10, 1.29943e-05};
+                
+    Double_t x = px[0];
+    if (TMath::Abs(x) > 6.2) return 0.;
+    Int_t j;
+    Double_t y = c[j = 6];
+    while (j > 0) y  = y * x +c[--j];
+    return y;
+}
+
 //                 particle composition
 //
 Int_t AliGenMUONlib::IpUpsilon(TRandom *)
@@ -256,7 +522,60 @@ Double_t AliGenMUONlib::YPhi( Double_t *px, Double_t *dummy)
 Int_t AliGenMUONlib::IpPhi(TRandom *)
 {
 // Phi composition
-    return 41;
+    return 333;
+}
+
+//
+//                        omega
+//
+//
+//    pt-distribution (by scaling of pion distribution)
+//____________________________________________________________
+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)
+{
+// Omega y
+    Double_t *dum=0;
+    return YJpsi(px,dum);
+}
+//                 particle composition
+//
+Int_t AliGenMUONlib::IpOmega(TRandom *)
+{
+// Omega composition
+    return 223;
+}
+
+
+//
+//                        Eta
+//
+//
+//    pt-distribution (by scaling of pion distribution)
+//____________________________________________________________
+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)
+{
+// Eta y
+    Double_t *dum=0;
+    return YJpsi(px,dum);
+}
+//                 particle composition
+//
+Int_t AliGenMUONlib::IpEta(TRandom *)
+{
+// Eta composition
+    return 221;
 }
 
 //
@@ -270,9 +589,10 @@ Double_t AliGenMUONlib::PtCharm( Double_t *px, Double_t *dummy)
 // Charm pT
   const Double_t kpt0 = 4.08;
   const Double_t kxn  = 9.40;
+
   Double_t x=*px;
   //
-  Double_t pass1 = 1.+(x/kpt0)*(x/kpt0);
+  Double_t pass1 = 1.+(x/kpt0);
   return x/TMath::Power(pass1,kxn);
 }
 //                  y-distribution
@@ -350,31 +670,50 @@ Int_t AliGenMUONlib::IpBeauty(TRandom *ran)
 }
 
 typedef Double_t (*GenFunc) (Double_t*,  Double_t*);
-GenFunc AliGenMUONlib::GetPt(Param_t param,  const char* tname)
+GenFunc AliGenMUONlib::GetPt(Int_t param,  const char* tname) const
 {
 // Return pointer to pT parameterisation
+    TString sname = TString(tname);
     GenFunc func;
     switch (param) 
     {
-    case phi_p:
+    case kPhi:
        func=PtPhi;
        break;
-    case jpsi_p:
-       func=PtJpsi;
+    case kOmega:
+       func=PtOmega;
+       break;
+    case kEta:
+       func=PtEta;
+       break;
+    case kJpsi:
+       if (sname == "Vogt" || sname == "Vogt PbPb") {
+           func=PtJpsiPbPb;
+       } else if (sname == "Vogt pp") {
+           func=PtJpsiPP;
+       } else {
+           func=PtJpsi;
+       }
        break;
-    case upsilon_p:
-       func=PtUpsilon;
+    case kUpsilon:
+       if (sname == "Vogt" || sname == "Vogt PbPb") {
+           func=PtUpsilonPbPb;
+       } else if (sname == "Vogt pp") {
+           func=PtUpsilonPP;
+       } else {
+           func=PtUpsilon;
+       }
        break;
-    case charm_p:
+    case kCharm:
        func=PtCharm;
        break;
-    case beauty_p:
+    case kBeauty:
        func=PtBeauty;
        break;
-    case pion_p:
+    case kPion:
        func=PtPion;
        break;
-    case kaon_p:
+    case kKaon:
        func=PtKaon;
        break;
     default:
@@ -384,31 +723,52 @@ GenFunc AliGenMUONlib::GetPt(Param_t param,  const char* tname)
     return func;
 }
 
-GenFunc AliGenMUONlib::GetY(Param_t param, const char* tname)
+GenFunc AliGenMUONlib::GetY(Int_t param, const char* tname) const
 {
+    TString sname = TString(tname);
+    
 // Return pointer to y- parameterisation
     GenFunc func;
     switch (param) 
     {
-    case phi_p:
+    case kPhi:
        func=YPhi;
        break;
-    case jpsi_p:
-       func=YJpsi;
+    case kEta:
+       func=YEta;
+       break;
+    case kOmega:
+       func=YOmega;
+       break;
+    case kJpsi:
+       if (sname == "Vogt" || sname == "Vogt PbPb") {
+           func=YJpsiPbPb;
+       } else if (sname == "Vogt pp"){
+           func=YJpsiPP;
+       } else {
+           func=YJpsi;
+       }
+       
        break;
-    case upsilon_p:
-       func=YUpsilon;
+    case kUpsilon:
+       if (sname == "Vogt" || sname == "Vogt PbPb") {
+           func=YUpsilonPbPb;
+       } else if (sname == "Vogt pp") {
+           func = YUpsilonPP;
+       } else {
+           func=YUpsilon;
+       }
        break;
-    case charm_p:
+    case kCharm:
        func=YCharm;
        break;
-    case beauty_p:
+    case kBeauty:
        func=YBeauty;
        break;
-    case pion_p:
+    case kPion:
        func=YPion;
        break;
-    case kaon_p:
+    case kKaon:
        func=YKaon;
        break;
     default:
@@ -418,31 +778,37 @@ GenFunc AliGenMUONlib::GetY(Param_t param, const char* tname)
     return func;
 }
 typedef Int_t (*GenFuncIp) (TRandom *);
-GenFuncIp AliGenMUONlib::GetIp(Param_t param,  const char* tname)
+GenFuncIp AliGenMUONlib::GetIp(Int_t param,  const char* tname) const
 {
 // Return pointer to particle type parameterisation
     GenFuncIp func;
     switch (param) 
     {
-    case phi_p:
+    case kPhi:
        func=IpPhi;
        break;
-    case jpsi_p:
+    case kEta:
+       func=IpEta;
+       break;
+    case kOmega:
+       func=IpOmega;
+       break;
+    case kJpsi:
        func=IpJpsi;
        break;
-    case upsilon_p:
+    case kUpsilon:
        func=IpUpsilon;
        break;
-    case charm_p:
+    case kCharm:
        func=IpCharm;
        break;
-    case beauty_p:
+    case kBeauty:
        func=IpBeauty;
        break;
-    case pion_p:
+    case kPion:
        func=IpPion;
        break;
-    case kaon_p:
+    case kKaon:
        func=IpKaon;
        break;
     default:
@@ -454,4 +820,54 @@ GenFuncIp AliGenMUONlib::GetIp(Param_t param,  const char* tname)
 
 
 
+Float_t AliGenMUONlib::Interpolate(Float_t x, Float_t* y, Float_t x0, 
+                                  Float_t dx,
+                                  Int_t n, Int_t no)
+{
+//
+// Neville's alorithm for interpolation
+//
+// x:  x-value
+// y:  Input array
+// x0: minimum x 
+// dx: step size
+//  n: number of data points
+// no: order of polynom 
+//
+    Float_t*  c = new Float_t[n];
+    Float_t*  d = new Float_t[n];
+    Int_t m, i;
+    for (i = 0; i < n; i++) {
+       c[i] = y[i];
+       d[i] = y[i];
+    }
+    
+    Int_t   ns  = int((x - x0)/dx);
+    
+    Float_t y1  = y[ns];
+    ns--;    
+    for (m = 0; m < no; m++) {
+       for (i = 0; i < n-m; i++) {     
+           Float_t ho = x0 + Float_t(i) * dx - x;
+           Float_t hp = x0 + Float_t(i+m+1) * dx - x;
+           Float_t w  = c[i+1] - d[i];
+           Float_t den = ho-hp;
+           den = w/den;
+           d[i] = hp * den;
+           c[i] = ho * den;
+       }
+       Float_t dy;
+       
+       if (2*ns < (n-m-1)) {
+           dy  = c[ns+1];
+       } else {
+           dy  = d[ns--];
+       }
+       y1 += dy;}
+    delete[] c;
+    delete[] d;
+
+    return y1;
+}
+