]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - EVGEN/AliGenMUONLMR.cxx
doxy: consider comments after last line
[u/mrichter/AliRoot.git] / EVGEN / AliGenMUONLMR.cxx
index 5d59dbd0bb6eabf40db6695a8aa71cfe05726564..5408ec8f500fbeb66be11a3787428b88ffebb9c3 100644 (file)
 #include "AliGenMUONLMR.h" 
 #include "AliMC.h" 
 #include "AliRun.h" 
+#include "AliLog.h" 
 #include "AliGenEventHeader.h"
 
 ClassImp(AliGenMUONLMR)
 
+  AliGenMUONLMR::AliGenMUONLMR () : AliGenMC(), 
+                                   fNMuMin(2), 
+                                   fCMSEnergy(kNCMSEnergies),
+                                   fGenSingleProc(-1),
+                                   fYCM(0), 
+                                   fCosTheta(0x0), 
+                                   fRhoLineShape(0x0),  
+                                   fHMultMu(0x0), 
+                                   fHNProc(0x0) { 
+    //
+    // default constructor 
+    //
+    for (int i=0; i<fgkNpart; i++) {
+      fPDG[i] = 0;
+      fScaleMult[i] = 1.;
+      fPt[i] = NULL;
+      fY[i] = NULL;
+      fMult[i] = NULL;
+      fParticle[i] = NULL;
+    }
+    for (int i=0; i<2; i++) {
+      fMu[i]     = NULL;
+      fDecay[i] = NULL;
+    }                                           
 
-AliGenMUONLMR::AliGenMUONLMR (Double_t energy) : AliGenMC(), 
-                                                fNMuMin(2), 
-                                                fGenSingleProc(-1),
-                                                fCosTheta(0x0), 
-                                                fRhoLineShape(0x0),  
-                                                fHMultMu(0x0), 
-                                                fHNProc(0x0) { 
-  //
-  // default constructor 
-  //
+    for (int i=0; i<3; i++) {
+      fDalitz[i] = NULL;
+    }                                           
+
+  }
+
+//-----------------------------------------------------------
+
+void AliGenMUONLMR::SetCMSEnergy(CMSEnergies energy){
+  fCMSEnergy = energy;
   // initialize pt and y distributions according to a fit to 
   // Pythia simulation at sqrt(s) = 7 TeV
   for (Int_t ipart=0; ipart < fgkNpart; ipart++) fScaleMult[ipart] = 1; 
   fScaleMult[kPionLMR] = 0; // set pion multiplicity to zero 
   fScaleMult[kKaonLMR] = 0; // set kaon multiplicity to zero
+  const char* fdname[2] = {"fDecPion","fDecKaon"};
+  Double_t ctau[2] = {7.8045, 3.712};  
   Int_t pdg[7] = {211, 321, 221, 113, 223, 333, 331}; 
   const char* fptname[7] = {"fPtPion","fPtKaon","fPtEta","fPtRho","fPtOmega","fPtPhi","fPtEtaPrime"};
   const char* fyname[7] = {"fYPion","fYKaon","fYEta","fYRho","fYOmega","fYPhi","fYEtaPrime"}; 
   const char* fnname[7] = {"fMultPion","fMultKaon","fMultEta","fMultRho","fMultOmega","fMultPhi","fMultEtaPrime"};
-  const char* fdname[2] = {"fDecPion","fDecKaon"};
-  Double_t ctau[2] = {7.8045, 3.712};  
   Double_t ptparam[7][9];
   Double_t yparam[7][9];
   Double_t nparam[7][9];
 
+  // parameters for 8 TeV generation
+  if (fCMSEnergy==kCMS8000GeV) {
+    AliInfo ("Using pp parameterization at 8 TeV\n");
+
+  // Parameters of transverse momentum spectra
+  Double_t ptparam8000[7][9] = {{1,0.427,2.52,0,0,0,0,0,0},   // pions from 7 TeV section of code
+                               {1,0.58,2.57,0,0,0,0,0,0},    // kaons from 7 TeV section of code
+                               {1,0.657,2.685,0,0,0,0,0,0},  // eta from PYTHIA6.4 ATLAS-CSC at 8 TeV
+                               {1,1.44,3.16,0,0,0,0,0,0},    // rho+omega from 7 TeV section of code
+                               {1,1.44,3.16,0,0,0,0,0,0},    // rho+omega from 7 TeV section of code
+                               {1,1.16,2.74,0,0,0,0,0,0},    // phi from 7 TeV section of code
+                               {1,0.755,2.578,0,0,0,0,0,0}}; // etaPrime from PYTHIA 6.4 ATLAS-CSC at 8 TeV
+
+  // Parameters of rapidity spectra
+  Double_t yparam8000[7][9] = {{1,0.8251,3.657,0,0,0,0,0,0},  // pions from 7 TeV section of code
+                              {1,1.83,2.698,0,0,0,0,0,0},    // kaons from 7 TeV section of code
+                              {1,0.0509,3.96,0,0,0,0,0,0},   // eta from PYTHIA6.4 ATLAS-CSC at 8 TeV
+                              {1,0.0489,3.961,0,0,0,0,0,0},  // rho from PYTHIA6.4 ATLAS-CSC at 8 TeV
+                              {1,0.0650,3.966,0,0,0,0,0,0},  // omega from PYTHIA 6.4 ATLAS-CSC at 8 TeV
+                              {1,1.279,2.745,0,0,0,0,0,0},   // phi from from PYTHIA6.4 ATLAS-CSC at 8 TeV
+                              {1,0.1627,3.883,0,0,0,0,0,0}}; // eta prime from PYTHIA6.4 ATLAS-CSC at 8 TeV
+
+  // Parameters of multiplicity spectra
+  Double_t nparam8000[7][9] = {{353.582, 6.76263, 1.66979, 998.445, 9.73281, 12.6704, 175.187, 29.08, 40.2531}, //pions from 7 TeV section of code
+                               {1.e4,    0.2841, 0,0,0,0,0,0,0},     // kaons from 7 TeV section of code
+                               {2.279e4, 0.2622, 0,0,0,0,0,0,0},  // eta from PYTHIA6.4 ATLAS-CSC at 8 TeV
+                               {1.564e4, 0.1713, 0,0,0,0,0,0,0},  // rho from PYTHIA6.4 ATLAS-CSC at 8 TeV
+                               {1.662e4, 0.183,  0,0,0,0,0,0,0},   // omega from PYTHIA6.4 ATLAS-CSC at 8 TeV 
+                               {6.723e4, 1.121,  0,0,0,0,0,0,0},  // phi from PYTHIA6.4 ATLAS-CSC at 8 TeV
+                               {5.005e4, 0.6971, 0,0,0,0,0,0,0}}; // eta prime from PYTHIA6.4 ATLAS-CSC at 8 TeV
+
+    for (Int_t i=0; i<fgkNpart; i++) {
+      for (Int_t j=0; j<9; j++) {
+       ptparam[i][j] = ptparam8000[i][j];
+       yparam[i][j] = yparam8000[i][j];
+       nparam[i][j] = nparam8000[i][j];
+      }
+    }
+  }
+       
   // parameters for 7 TeV generation
-  if (energy==7.0) {
-    printf ("AliGenMUONLMR: using pp parameterization at 7 TeV\n");  
+  else if (fCMSEnergy==kCMS7000GeV) {
+    AliInfo ("Using pp parameterization at 7 TeV\n");  
     Double_t ptparam7000[7][9] = {{1,0.427,2.52,0,0,0,0,0,0}, // pions from Pythia
                                  {1,0.58,2.57,0,0,0,0,0,0},  // kaons from Pythia
                                  {1,0.641,2.62,0,0,0,0,0,0}, // eta from Pythia
@@ -46,7 +111,7 @@ AliGenMUONLMR::AliGenMUONLMR (Double_t energy) : AliGenMC(),
                                  {1,1.44,3.16,0,0,0,0,0,0},  // rho+omega from ALICE muon  
                                  {1,1.16,2.74,0,0,0,0,0,0},  // phi from ALICE muon  
                                  {1,0.72,2.5,0,0,0,0,0,0}};  // etaPrime from Pythia    
-    
+               
     Double_t yparam7000[7][9] = {{1,0.8251,3.657,0,0,0,0,0,0}, // pions from pythia
                                 {1,1.83,2.698,0,0,0,0,0,0},   // kaons from pythia
                                 {1,1.169,3.282,0,0,0,0,0,0},  // eta from pythia
@@ -54,7 +119,7 @@ AliGenMUONLMR::AliGenMUONLMR (Double_t energy) : AliGenMC(),
                                 {1,1.311,3.223,0,0,0,0,0,0},  // omega from pythia
                                 {1,2.388,2.129,0,0,0,0,0,0},  // phi from pythia
                                 {1,1.13,3.3,0,0,0,0,0,0}};    // eta prime from pythia
-    
+               
     // multiplicity parameters from pythia
     Double_t nparam7000[7][9] = {{353.582, 6.76263, 1.66979, 998.445, 9.73281, 12.6704, 175.187, 29.08, 40.2531},
                                 {1.e4,  0.2841, 0,0,0,0,0,0,0},
@@ -63,7 +128,7 @@ AliGenMUONLMR::AliGenMUONLMR (Double_t energy) : AliGenMC(),
                                 {7500,  0.1896, 0,0,0,0,0,0,0},
                                 {5.e4,  1.167,  0,0,0,0,0,0,0}, 
                                 {2.9e4, 0.714,  0,0,0,0,0,0,0}};
-    
+               
     for (Int_t i=0; i<fgkNpart; i++) { 
       for (Int_t j=0; j<9; j++) {
        ptparam[i][j] = ptparam7000[i][j];
@@ -72,11 +137,47 @@ AliGenMUONLMR::AliGenMUONLMR (Double_t energy) : AliGenMC(),
       }
     }
   }  
-  
-  // parameters for 2.76 generation
-  // pt params has been determined as <pt>ALICE_2.76 = <pt>ALICE_7 * <pt>PYTHIA_2.76 / <pt>PYTHIA_7
-  if (energy == 2.76){
-    printf ("AliGenMUONLMR: using pp parameterization at 2.76 TeV\n");  
+  else if (fCMSEnergy==kCMS5020GeVpPb || fCMSEnergy==kCMS5020GeVPbp) {
+    AliInfo ("Using pPb parameterization at 5.02 TeV\n");  
+    Double_t ptparam5020[7][9] = {{1,0.427,2.52,0,0,0,0,0,0}, // pions from Pythia at 7 TeV
+                                 {1,0.58,2.57,0,0,0,0,0,0},  // kaons from Pythia at 7 TeV
+                                 {1,0.665,2.796,0,0,0,0,0,0}, // eta from Pythia at 5.02 TeV
+                                 {1,1.66,3.12,0,0,0,0,0,0},  // rho+omega from ALICE muon  
+                                 {1,1.66,3.12,0,0,0,0,0,0},  // rho+omega from ALICE muon  
+                                 {1,2.03,3.13,0,0,0,0,0,0},  // phi from ALICE muon  
+                                 {1,0.767,2.713,0,0,0,0,0,0}};  // etaPrime from Pythia at 5.02 TeV
+               
+    Double_t yparam5020[7][9] = {{1,0.8251,3.657,0,0,0,0,0,0}, // pions from pythia at 7 TeV
+                                {1,1.83,2.698,0,0,0,0,0,0},   // kaons from pythia at 7 TeV
+                                {1,1.169,3.282,0,0,0,0,0,0},  // eta from pythia at 7 TeV
+                                {1,1.234,3.264,0,0,0,0,0,0},  // rho from pythia at 7 TeV
+                                {1,1.311,3.223,0,0,0,0,0,0},  // omega from pythia at 7 TeV
+                                {1,2.388,2.129,0,0,0,0,0,0},  // phi from pythia at 7 TeV
+                                {1,1.13,3.3,0,0,0,0,0,0}};    // eta prime from pythia at 7 TeV
+               
+    // multiplicity parameters from pythia at 7 TeV
+    Double_t nparam5020[7][9] = {{353.582, 6.76263, 1.66979, 998.445, 9.73281, 12.6704, 175.187, 29.08, 40.2531},
+                                {1.e4,  0.2841, 0,0,0,0,0,0,0},
+                                {1.e4,  0.2647, 0,0,0,0,0,0,0},
+                                {7055,  0.1786, 0,0,0,0,0,0,0},
+                                {7500,  0.1896, 0,0,0,0,0,0,0},
+                                {5.e4,  1.167,  0,0,0,0,0,0,0}, 
+                                {2.9e4, 0.714,  0,0,0,0,0,0,0}};
+               
+    for (Int_t i=0; i<fgkNpart; i++) { 
+      for (Int_t j=0; j<9; j++) {
+       ptparam[i][j] = ptparam5020[i][j];
+       yparam[i][j] = yparam5020[i][j];
+       nparam[i][j] = nparam5020[i][j];        
+      }
+    }
+    if (fCMSEnergy==kCMS5020GeVpPb) fYCM = -0.4654;   
+    else fYCM = 0.4654;
+  }  
+  else if (fCMSEnergy==kCMS2760GeV){
+    // parameters for 2.76 generation
+    // pt params has been determined as <pt>ALICE_2.76 = <pt>ALICE_7 * <pt>PYTHIA_2.76 / <pt>PYTHIA_7
+    AliInfo ("Using pp parameterization at 2.76 TeV\n");  
     Double_t yparam2760[7][9] = {{1,0.8251,3.657,0,0,0,0,0,0},      // pions from pythia
                                 {1,1.83,2.698,0,0,0,0,0,0},        // kaons from pythia
                                 {1,0.011,3.474,0,0,0,0,0,0},       // eta from pythia
@@ -84,7 +185,7 @@ AliGenMUONLMR::AliGenMUONLMR (Double_t energy) : AliGenMC(),
                                 {1,-0.037,3.294,0,0,0,0,0,0},      // omega from pythia
                                 {1,-0.016,2.717,0,0,0,0,0,0},      // phi from pythia
                                 {1,-0.010,3.312,0,0,0,0,0,0}};     // eta prime from pythia  
-    
+               
     Double_t ptparam2760[7][9] = {{1,0.1665,8.878,0,0,0,0,0,0},  // pions from Pythia
                                  {1,0.1657,8.591,0,0,0,0,0,0},  // kaons from Pythia
                                  {1,0.641,2.62,0,0,0,0,0,0},    // eta from ALICE 7 TeV
@@ -92,7 +193,7 @@ AliGenMUONLMR::AliGenMUONLMR (Double_t energy) : AliGenMC(),
                                  {1,1.3551,3.16,0,0,0,0,0,0},   // omega with <pt> scaled 
                                  {1,1.0811,2.74,0,0,0,0,0,0},   // phi with <pt> scaled 
                                  {1,0.72,2.5,0,0,0,0,0,0}};     // etaPrime from ALICE 7 TeV
-    
+               
     Double_t nparam2760[7][9] = {{9752,-2.693,3.023,9.5e9,-84.68,16.75,-14.06,635.3,-423.2}, // pions
                                 {1.e5, 1.538,  0,0,0,0,0,0,0},                              // kaons
                                 {1.e4, 0.351,  0,0,0,0,0,0,0},                              // eta
@@ -100,7 +201,7 @@ AliGenMUONLMR::AliGenMUONLMR (Double_t energy) : AliGenMC(),
                                 {1.e4, 0.2583, 0,0,0,0,0,0,0},                              // omega
                                 {1.e5, 1.393,  0,0,0,0,0,0,0},                              // phi
                                 {1.e4, 0.9005, 0,0,0,0,0,0,0}};                             // etaPrime
-
+               
     for (Int_t i=0; i<fgkNpart; i++) { 
       for (Int_t j=0; j<9; j++) {
        ptparam[i][j] = ptparam2760[i][j];
@@ -108,35 +209,36 @@ AliGenMUONLMR::AliGenMUONLMR (Double_t energy) : AliGenMC(),
        nparam[i][j] = nparam2760[i][j];        
       }
     }
-  }
-  
-    for (Int_t i=0; i<fgkNpart; i++) { 
-      fPDG[i] = pdg[i]; 
-      if (i!=0) { 
-        fMult[i] = new TF1(fnname[i],"[0]*exp(-[1]*x)",0,30);
-        fMult[i]->SetParameters(nparam[i][0],nparam[i][1]);  
-      }
-      else { 
-        fMult[i] = new TF1(fnname[i],"gaus(0)+gaus(3)+gaus(6)",0,150);
-        for (Int_t j=0; j<9; j++) fMult[i]->SetParameter(j,nparam[i][j]);
-      }
-  
-      fPt[i] = new TF1(fptname[i],AliGenMUONLMR::PtDistr,0,20,3);
-      fPt[i]->SetParameters(ptparam[i][0], ptparam[i][1], ptparam[i][2]);  
-      fY[i] = new TF1(fyname[i],AliGenMUONLMR::YDistr,-10,10,3);
-      fY[i]->SetParameters(yparam[i][0], yparam[i][1], yparam[i][2]); 
+  } 
+  else AliFatal("Energy not correctly defined");
+       
+  for (Int_t i=0; i<fgkNpart; i++) { 
+    fPDG[i] = pdg[i]; 
+    if (i!=0) { 
+      fMult[i] = new TF1(fnname[i],"[0]*exp(-[1]*x)",0,30);
+      fMult[i]->SetParameters(nparam[i][0],nparam[i][1]);  
     }
-  
+    else { 
+      fMult[i] = new TF1(fnname[i],"gaus(0)+gaus(3)+gaus(6)",0,150);
+      for (Int_t j=0; j<9; j++) fMult[i]->SetParameter(j,nparam[i][j]);
+    }
+               
+    fPt[i] = new TF1(fptname[i],AliGenMUONLMR::PtDistr,0,20,3);
+    fPt[i]->SetParameters(ptparam[i][0], ptparam[i][1], ptparam[i][2]);  
+    fY[i] = new TF1(fyname[i],AliGenMUONLMR::YDistr,-10,10,4);
+    fY[i]->SetParameters(yparam[i][0], yparam[i][1], yparam[i][2],fYCM); 
+  }
+       
   for(Int_t i = 0; i<2; i++){
     fDecay[i] = new TF1(fdname[i],"exp(-x/[0])",0,150);
     fDecay[i]->SetParameter(0,ctau[i]);
   }
-  
+       
   for (Int_t ipart = 0; ipart < fgkNpart; ipart++) { 
     fParticle[ipart] = new TParticle(); 
     fParticle[ipart]->SetPdgCode(fPDG[ipart]); 
   }
-
+       
   TDatabasePDG *pdgdb = TDatabasePDG::Instance(); 
   Double_t mumass = pdgdb->GetParticle(13)->Mass();
   fMu[0] = new TParticle(); 
@@ -145,24 +247,24 @@ AliGenMUONLMR::AliGenMUONLMR (Double_t energy) : AliGenMC(),
   fMu[1] = new TParticle(); 
   fMu[1]->SetPdgCode(13); 
   fMu[1]->SetCalcMass(mumass); 
-  
+       
   // function for polarized theta distributions
   fCosTheta = new TF1 ("fCosTheta","1+[0]*x*x",-1,1);
   fCosTheta->SetParameter(0,1);
-
+       
   // Dalitz decays 
   Int_t nbins = 1000;
   Double_t xmin = 0, xmax = 2; 
   fDalitz[0] = new TH1F("hDalitzEta","",nbins,xmin,xmax);
   fDalitz[1] = new TH1F("hDalitzOmega","",nbins,xmin,xmax);
   fDalitz[2] = new TH1F("hDalitzEtaPrime","",nbins,xmin,xmax);
-  
+       
   Double_t meta   = pdgdb->GetParticle("eta")->Mass(); 
   Double_t momega = pdgdb->GetParticle("omega")->Mass(); 
   Double_t metaPrime = pdgdb->GetParticle("eta'")->Mass(); 
   Double_t mpi0   = pdgdb->GetParticle("pi0")->Mass(); 
   Double_t md3 = 0, mres = 0; 
-  
+       
   for (Int_t index = 0; index < 3; index++) { 
     if (index == 0) { 
       mres = meta; 
@@ -197,17 +299,19 @@ AliGenMUONLMR::AliGenMUONLMR (Double_t energy) : AliGenMC(),
       }
     }
   }
-
+       
   fRhoLineShape = new TF1("fRhoLineShape",RhoLineShapeNew,0,2,2); 
   fHMultMu = new TH1D("fHMultMu","Muon multiplicity",20,-0.5,19.5); 
   fHNProc = new TH1D("fHNProc","Number of gen. evts. per process in 4 pi",9,-0.5,8.5); 
+       
 }
-
 //-----------------------------------------------------------
 
 AliGenMUONLMR::AliGenMUONLMR (AliGenMUONLMR &gen) : AliGenMC(), 
                                                    fNMuMin(gen.fNMuMin), 
+                                                   fCMSEnergy(gen.fCMSEnergy), 
                                                    fGenSingleProc(gen.fGenSingleProc),
+                                                   fYCM(gen.fYCM),
                                                    fCosTheta(gen.fCosTheta), 
                                                    fRhoLineShape(gen.fRhoLineShape),  
                                                    fHMultMu(gen.fHMultMu), 
@@ -229,25 +333,30 @@ AliGenMUONLMR::AliGenMUONLMR (AliGenMUONLMR &gen) : AliGenMC(),
 //-----------------------------------------------------------
 
 AliGenMUONLMR& AliGenMUONLMR::operator=(const AliGenMUONLMR &gen) {
-  fNMuMin = gen.fNMuMin; 
-  fGenSingleProc = gen.fGenSingleProc; 
-  fCosTheta = (TF1*) gen.fCosTheta->Clone();  
-  fRhoLineShape = (TF1*) gen.fRhoLineShape->Clone();
-  fHMultMu = (TH1D*) gen.fHMultMu->Clone();
-  fHNProc = (TH1D*) gen.fHNProc->Clone();  
-  
-  for (Int_t i=0; i < fgkNpart; i++) { 
-    fPDG[i] = gen.fPDG[i]; 
-    fScaleMult[i] = gen.fScaleMult[i]; 
-    fPt[i] = (TF1*) gen.fPt[i]->Clone(); 
-    fY[i] = (TF1*) gen.fY[i]->Clone();  
-    fMult[i] = (TF1*) gen.fMult[i]->Clone(); 
-    fParticle[i] = (TParticle*) gen.fParticle[i]->Clone(); 
+  // Assignment operator
+  if (this!=&gen) {
+    fNMuMin = gen.fNMuMin; 
+    fCMSEnergy = gen.fCMSEnergy; 
+    fGenSingleProc = gen.fGenSingleProc; 
+    fYCM = gen.fYCM; 
+    fCosTheta = (TF1*) gen.fCosTheta->Clone();  
+    fRhoLineShape = (TF1*) gen.fRhoLineShape->Clone();
+    fHMultMu = (TH1D*) gen.fHMultMu->Clone();
+    fHNProc = (TH1D*) gen.fHNProc->Clone();  
+    
+    for (Int_t i=0; i < fgkNpart; i++) { 
+      fPDG[i] = gen.fPDG[i]; 
+      fScaleMult[i] = gen.fScaleMult[i]; 
+      fPt[i] = (TF1*) gen.fPt[i]->Clone(); 
+      fY[i] = (TF1*) gen.fY[i]->Clone();  
+      fMult[i] = (TF1*) gen.fMult[i]->Clone(); 
+      fParticle[i] = (TParticle*) gen.fParticle[i]->Clone(); 
+    }
+    
+    for(Int_t i = 0; i<2; i++) fDecay[i] = (TF1*) gen.fDecay[i]->Clone(); 
+    for(Int_t i = 0; i<3; i++) fDalitz[i] = (TH1F*) gen.fDalitz[i]->Clone(); 
+    for(Int_t i = 0; i<2; i++) fMu[i] = (TParticle*) gen.fMu[i]->Clone(); 
   }
-  
-  for(Int_t i = 0; i<2; i++) fDecay[i] = (TF1*) gen.fDecay[i]->Clone(); 
-  for(Int_t i = 0; i<3; i++) fDalitz[i] = (TH1F*) gen.fDalitz[i]->Clone(); 
-  for(Int_t i = 0; i<2; i++) fMu[i] = (TParticle*) gen.fMu[i]->Clone(); 
   return *this; 
 }
 
@@ -282,9 +391,9 @@ AliGenMUONLMR::~AliGenMUONLMR()
 void AliGenMUONLMR::FinishRun(){ 
   // save some histograms to an output file 
   Int_t nbins = fHNProc->GetNbinsX(); 
-  for (Int_t ibin=1; ibin <= nbins; ibin++) printf ("ibin = %d nEvProc = %g\n",
-                                                     ibin,fHNProc->GetBinContent(ibin));
-    TFile *fout = new TFile("AliGenMUONLMR_histos.root","recreate"); 
+  for (Int_t ibin=1; ibin <= nbins; ibin++) AliInfo (Form("ibin = %d nEvProc = %g",
+                                                         ibin,fHNProc->GetBinContent(ibin)));
+  TFile *fout = new TFile("AliGenMUONLMR_histos.root","recreate"); 
   fHMultMu->Write(); 
   fHNProc->Write(); 
   fout->Close(); 
@@ -295,11 +404,16 @@ void AliGenMUONLMR::FinishRun(){
 Double_t AliGenMUONLMR::YDistr(Double_t *px, Double_t *par){ 
   // function for rapidity distribution: plateau at par[0] +
   // gaussian tails centered at par[1] and with par[2]=sigma  
-  Double_t x = TMath::Abs(px[0]);
+  Double_t ylab = px[0]; 
+  Double_t y0 = par[3]; // center of mass rapidity  
   Double_t func = 0;
-  if (x<par[1]) func = par[0]; 
-  else { 
-    Double_t z = (x-par[1])/(par[2]); 
+  if (ylab<y0+par[1] && ylab>y0-par[1]) func = par[0]; 
+  else if (ylab>y0+par[1]) { 
+    Double_t z = (ylab-(par[1]+y0) )/(par[2]); 
+    func = par[0] * TMath::Exp(-0.5 * z * z); 
+  }
+  else {
+    Double_t z = (ylab-(-par[1]+y0) )/(par[2]); 
     func = par[0] * TMath::Exp(-0.5 * z * z); 
   }
   return func; 
@@ -334,7 +448,6 @@ void AliGenMUONLMR::Generate() {
     for (Int_t j=0;j<3;j++) origin0[j]=fVertex[j];
   }
   
-  printf ("In Generate()\n");
   TParticle *mother; 
   TDatabasePDG* pdg = TDatabasePDG::Instance();
 
@@ -373,7 +486,7 @@ void AliGenMUONLMR::Generate() {
     }
     
     for (Int_t iproc = 0; iproc < nproc; iproc++) { 
-//       printf ("Multiplicity for process %d is %d\n",iproc,mult[iproc]); 
+      //       printf ("Multiplicity for process %d is %d\n",iproc,mult[iproc]); 
       for (Int_t imult=0; imult<mult[iproc]; imult++) { 
        if (gRandom->Rndm() < BR[iproc]) { 
          fHNProc->Fill(iproc); 
@@ -499,9 +612,9 @@ void AliGenMUONLMR::Decay2Body(TParticle *mother){
   v1.Boost(betaParent);
   v2.Boost(betaParent);
 
-//   TLorentzVector vtot = v1 + v2; 
-//   printf ("mother: %g   %g    %g     %g\n",vmother.Px(), vmother.Py(), vmother.Pz(), vmother.E());
-//   printf ("vtot  : %g   %g    %g     %g\n",vtot.Px(), vtot.Py(), vtot.Pz(), vtot.E());
+  //   TLorentzVector vtot = v1 + v2; 
+  //   printf ("mother: %g   %g    %g     %g\n",vmother.Px(), vmother.Py(), vmother.Pz(), vmother.E());
+  //   printf ("vtot  : %g   %g    %g     %g\n",vtot.Px(), vtot.Py(), vtot.Pz(), vtot.E());
 
   fMu[0]->SetMomentum(v1.Px(),v1.Py(),v1.Pz(),v1.E());
   fMu[1]->SetMomentum(v2.Px(),v2.Py(),v2.Pz(),v2.E());
@@ -539,7 +652,7 @@ void AliGenMUONLMR::DecayPiK(TParticle *mother, Bool_t &hasDecayed){
   if (mother->GetPdgCode()>0) fMu[0]->SetMomentum(v1.Px(),v1.Py(),v1.Pz(),v1.E());
   else fMu[1]->SetMomentum(v1.Px(),v1.Py(),v1.Pz(),v1.E());
 
-  Int_t idmother = -1
+  Int_t idmother = 0
   if (TMath::Abs(mother->GetPdgCode())== 211) idmother = 0; 
   if (TMath::Abs(mother->GetPdgCode())== 321) idmother = 1; 
   Double_t gammaRes = mother->Energy()/mres;
@@ -547,7 +660,6 @@ void AliGenMUONLMR::DecayPiK(TParticle *mother, Bool_t &hasDecayed){
   Double_t zResLab = gammaRes*zResCM;  
   if(zResLab > 0.938) hasDecayed = 0; // 0.938: distance from IP to absorber + lambda_i
   else hasDecayed = 1;
-
 } 
 
 //-------------------------------------------------------------------
@@ -561,7 +673,7 @@ void AliGenMUONLMR::DalitzDecay(TParticle *mother){
   Double_t mumass  = fMu[0]->GetMass(); 
   Double_t md3  = 0;  // unless differently specified, third particle is a photon 
   if (mother->GetPdgCode() == 223) md3 = 0.134977; // if mother is an omega, third particle is a pi0
-  Int_t index = -1
+  Int_t index = 0
   if (mother->GetPdgCode() == 221) index = 0;  // eta
   else if (mother->GetPdgCode() == 223) index = 1; // omega  
   else if (mother->GetPdgCode() == 331) index = 2; // etaPrime  
@@ -587,7 +699,7 @@ void AliGenMUONLMR::DalitzDecay(TParticle *mother){
   Double_t e1 = TMath::Sqrt(mvirt2)/2.; // energy of mu1 in the virtual photon frame
   Double_t psquare = (e1 + mumass)*(e1 - mumass); 
   if (psquare<0) {
-    printf("Error in AliGenMUONLMR::DalitzDecay: sqrt of psquare = %f put to 0\n",psquare); 
+    AliError(Form("sqrt of psquare = %f put to 0\n",psquare)); 
     psquare = 0;
   }
   Double_t p1 = TMath::Sqrt(psquare);
@@ -618,7 +730,7 @@ void AliGenMUONLMR::DalitzDecay(TParticle *mother){
   Double_t e3 = (mres * mres + md3 * md3 - mvirt2) / (2.*mres);
   Double_t psquare3 = (e3 + md3)*(e3 - md3); 
   if (psquare3<0) {
-    printf("Error in AliGenMUONLMR::DalitzDecay: sqrt of psquare3 = %f put to 0\n",psquare3); 
+    AliError(Form("Sqrt of psquare3 = %f put to 0\n",psquare3)); 
     psquare3 = 0;
   }
   Double_t p3 = TMath::Sqrt(psquare3);
@@ -674,14 +786,14 @@ void AliGenMUONLMR::DalitzDecay(TParticle *mother){
 
   fMu[0]->SetMomentum(v1.Px(),v1.Py(),v1.Pz(),v1.E());
   fMu[1]->SetMomentum(v2.Px(),v2.Py(),v2.Pz(),v2.E());
-//   part3->SetMomentum(v3.Px(),v3.Py(),v3.Pz(),v3.E());
+  //   part3->SetMomentum(v3.Px(),v3.Py(),v3.Pz(),v3.E());
 
-//   TLorentzVector vtot = v1 + v2 + v3; 
-//   TLorentzVector vdimu = v1 + v2; 
-//   printf ("mother: %g   %g    %g     %g\n",vmother.Px(), vmother.Py(), vmother.Pz(), vmother.E());
-//   printf ("vtot  : %g   %g    %g     %g\n",vtot.Px(), vtot.Py(), vtot.Pz(), vtot.E());
-//   printf ("vvirt : %g   %g    %g     %g\n",vvirt.Px(), vvirt.Py(), vvirt.Pz(), vvirt.E());
-//   printf ("vdimu : %g   %g    %g     %g\n",vdimu.Px(), vdimu.Py(), vdimu.Pz(), vdimu.E());
+  //   TLorentzVector vtot = v1 + v2 + v3; 
+  //   TLorentzVector vdimu = v1 + v2; 
+  //   printf ("mother: %g   %g    %g     %g\n",vmother.Px(), vmother.Py(), vmother.Pz(), vmother.E());
+  //   printf ("vtot  : %g   %g    %g     %g\n",vtot.Px(), vtot.Py(), vtot.Pz(), vtot.E());
+  //   printf ("vvirt : %g   %g    %g     %g\n",vvirt.Px(), vvirt.Py(), vvirt.Pz(), vvirt.E());
+  //   printf ("vdimu : %g   %g    %g     %g\n",vdimu.Px(), vdimu.Py(), vdimu.Pz(), vdimu.E());
 
 }
 
@@ -700,7 +812,7 @@ Double_t AliGenMUONLMR::FormFactor(Double_t q2, Int_t decay){
   Double_t lambda2inv = 0;
   switch (decay) { 
   case 0:   // eta -> mu mu gamma  
-  // eta   -> l+ l- gamma: pole approximation
+    // eta   -> l+ l- gamma: pole approximation
     lambda2inv = 1.95; 
     mass2 = fParticle[kEtaLMR]->GetMass() * fParticle[kEtaLMR]->GetMass(); 
     if (q2 < mass2) ff2 = TMath::Power(1./(1.-lambda2inv*q2),2);
@@ -723,7 +835,7 @@ Double_t AliGenMUONLMR::FormFactor(Double_t q2, Int_t decay){
     else ff2 = 0; 
     break;
   default:
-    printf ("FormFactor: Decay not found\n"); 
+    AliError ("FormFactor: Decay not found"); 
     return 0; 
     break; 
   }
@@ -771,3 +883,4 @@ Double_t AliGenMUONLMR::RhoLineShapeNew(Double_t *x, Double_t */*para*/){
 
   return r;
 }
+