]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - EVGEN/AliGenMUONLMR.cxx
New bversion
[u/mrichter/AliRoot.git] / EVGEN / AliGenMUONLMR.cxx
index 602991a8ee4bff966f2c0a1d31d095a8cf1cd848..5d59dbd0bb6eabf40db6695a8aa71cfe05726564 100644 (file)
@@ -3,7 +3,6 @@
 #include <TDatabasePDG.h>
 #include <TFile.h>
 #include "AliConst.h"
-#include "AliLog.h"
 #include "AliGenMUONLMR.h" 
 #include "AliMC.h" 
 #include "AliRun.h" 
 ClassImp(AliGenMUONLMR)
 
 
-AliGenMUONLMR::AliGenMUONLMR () : AliGenMC(), fNMuMin(2), fGenSingleProc(-1), fCosTheta(0), fRhoLineShape(0), fHMultMu(0), fHNProc(0) { 
+AliGenMUONLMR::AliGenMUONLMR (Double_t energy) : AliGenMC(), 
+                                                fNMuMin(2), 
+                                                fGenSingleProc(-1),
+                                                fCosTheta(0x0), 
+                                                fRhoLineShape(0x0),  
+                                                fHMultMu(0x0), 
+                                                fHNProc(0x0) { 
   //
   // default constructor 
   //
   // initialize pt and y distributions according to a fit to 
   // Pythia simulation at sqrt(s) = 7 TeV
-  printf ("Using AliGenMUONLMR as generator\n");
   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
@@ -27,50 +31,102 @@ AliGenMUONLMR::AliGenMUONLMR () : AliGenMC(), fNMuMin(2), fGenSingleProc(-1), fC
   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 ptparam[7][3] = { {1,0.427,2.52}, // pions from Pythia
-                            {1,0.58,2.57},  // kaons from Pythia
-                            {1,0.641,2.62},   // eta from Pythia
-                            {1,1.2,2.5},    // rho+omega from ALICE muon  
-                            {1,1.2,2.5},    // rho+omega from ALICE muon  
-                            {1,1.03,2.5},   // phi from ALICE muon  
-                            {1,0.72,2.5}};  // etaPrime from Pythia
+  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 7 TeV generation
+  if (energy==7.0) {
+    printf ("AliGenMUONLMR: 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
+                                 {1,1.44,3.16,0,0,0,0,0,0},  // rho+omega from ALICE muon  
+                                 {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
+                                {1,1.234,3.264,0,0,0,0,0,0},  // rho from pythia
+                                {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},
+                                {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] = ptparam7000[i][j];
+       yparam[i][j] = yparam7000[i][j];
+       nparam[i][j] = nparam7000[i][j];        
+      }
+    }
+  }  
   
-  Double_t yparam[7][3] = { {1, 0.8251, 3.657}, // pions from pythia
-                           {1, 1.83, 2.698},   // kaons from pythia
-                           {1, 1.169, 3.282},  // eta from pythia
-                           {1, 1.234, 3.264},  // rho from pythia
-                           {1, 1.311, 3.223},  // omega from pythia
-                           {1, 2.388, 2.129},  // phi from pythia
-                           {1, 1.13,3.3}};     // eta prime from pythia
-
-  // multiplicity parameters from pythia
-  Double_t nparam[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.e04, 1.167,  0,0,0,0,0,0,0}, 
-                           {2.9e04,0.714,  0,0,0,0,0,0,0}}; 
-
-  Double_t ctau[2] = {7.8045, 3.712};
-
-  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]);  
+  // 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");  
+    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
+                                {1,-0.01,3.409,0,0,0,0,0,0},       // rho from pythia
+                                {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
+                                 {1,1.3551,3.16,0,0,0,0,0,0},   // rho with <pt> scaled 
+                                 {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
+                                {1.e4, 0.2471, 0,0,0,0,0,0,0},                              // rho
+                                {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];
+       yparam[i][j] = yparam2760[i][j];
+       nparam[i][j] = nparam2760[i][j];        
+      }
     }
-    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]);
+  }
+  
+    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]); 
     }
   
-    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]); 
-  }
-
   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]);
@@ -122,11 +178,11 @@ AliGenMUONLMR::AliGenMUONLMR () : AliGenMC(), fNMuMin(2), fGenSingleProc(-1), fC
     }
     Double_t delta   = md3 * md3 / (mres * mres);
     Double_t epsilon = mumass * mumass / (mres * mres);
-    Int_t nbins0 = fDalitz[index]->GetNbinsX();
-    Double_t xmin0 = fDalitz[index]->GetXaxis()->GetXmin(); 
+    nbins = fDalitz[index]->GetNbinsX();
+    xmin = fDalitz[index]->GetXaxis()->GetXmin(); 
     Double_t deltax =  fDalitz[index]->GetBinWidth(1);
-    Double_t xd = xmin0 - deltax/2.; 
-    for (Int_t ibin = 0; ibin< nbins0; ibin++) { 
+    Double_t xd = xmin - deltax/2.; 
+    for (Int_t ibin = 0; ibin< nbins; ibin++) { 
       Double_t dalval = 0; 
       xd += deltax; 
       if (xd > 4. *epsilon) { 
@@ -147,6 +203,55 @@ AliGenMUONLMR::AliGenMUONLMR () : AliGenMC(), fNMuMin(2), fGenSingleProc(-1), fC
   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), 
+                                                   fGenSingleProc(gen.fGenSingleProc),
+                                                   fCosTheta(gen.fCosTheta), 
+                                                   fRhoLineShape(gen.fRhoLineShape),  
+                                                   fHMultMu(gen.fHMultMu), 
+                                                   fHNProc(gen.fHNProc) {  
+  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(); 
+}
+
+//-----------------------------------------------------------
+
+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(); 
+  }
+  
+  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; 
+}
+
 //-----------------------------------------------------------
 
 AliGenMUONLMR::~AliGenMUONLMR()
@@ -187,7 +292,7 @@ void AliGenMUONLMR::FinishRun(){
 
 //-----------------------------------------------------------
 
-Double_t AliGenMUONLMR::YDistr(const Double_t *px, const Double_t *par){ 
+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]);
@@ -202,7 +307,7 @@ Double_t AliGenMUONLMR::YDistr(const Double_t *px, const Double_t *par){
 
 //-----------------------------------------------------------
 
-Double_t AliGenMUONLMR::PtDistr(const Double_t *px, const Double_t *par){
+Double_t AliGenMUONLMR::PtDistr(Double_t *px, Double_t *par){
   // pt distribution: power law 
   Double_t x = px[0];
   Double_t func = par[0] * x / TMath::Power((1+(x/par[1])*(x/par[1])),par[2]); 
@@ -222,14 +327,11 @@ void AliGenMUONLMR::Generate() {
   Int_t nmuons = -1, npartPushed = 0, pdgPushed[100]; 
   Double_t polar[3]= {0,0,0};  // Polarisation of the parent particle (for GEANT tracking)
   Double_t origin0[3];         // Origin of the generated parent particle (for GEANT tracking)
-  Double_t time0;              // Time0  of the generated parent particle
   // Calculating vertex position per event
   for (Int_t j=0;j<3;j++) origin0[j]=fOrigin[j];
-  time0 = fTimeOrigin;
   if(fVertexSmear==kPerEvent) {
     Vertex();
     for (Int_t j=0;j<3;j++) origin0[j]=fVertex[j];
-    time0 = fTime;
   }
   
   printf ("In Generate()\n");
@@ -240,7 +342,7 @@ void AliGenMUONLMR::Generate() {
 
   const Int_t nproc = 9; 
   Int_t idRes[nproc] = {kEtaLMR, kEtaLMR, kRhoLMR, kOmegaLMR, kOmegaLMR, kPhiLMR, kEtaPrimeLMR, kPionLMR, kKaonLMR}; 
-  Double_t BR[nproc] = {5.8e-6, 3.1e-4, 4.55e-5, 7.28e-4, 1.3e-4, 2.86e-4, 1.04e-4, 1, 0.6344};
+  Double_t BR[nproc] = {5.8e-6, 3.1e-4, 4.55e-5, 7.28e-5, 1.3e-4, 2.86e-4, 1.04e-4, 1, 0.6344};
   //  Double_t BR[nproc] = {1, 1, 1, 1, 1, 1, 1, 1, 1};
   Int_t idDec[nproc] = {0, 1, 0, 0, 1, 0, 1, 2, 2}; // 0:2body, 1:Dalitz, 2:pi/K 
   Int_t mult[nproc] = {0,0,0,0,0,0,0,0,0}; 
@@ -341,7 +443,7 @@ void AliGenMUONLMR::Generate() {
     if (TMath::Abs(pdgPushed[ipart]) != 13) { // particle is not a muon, hence it's a mother
       PushTrack(0,-1,pdgPushed[ipart],
                pxPushed[ipart],pyPushed[ipart],pzPushed[ipart],ePushed[ipart],
-               origin0[0],origin0[1],origin0[2],time0,
+               origin0[0],origin0[1],origin0[2],0.,
                polar[0],polar[1],polar[2],
                kPPrimary,ntmother,1,11);
       KeepTrack(ntmother); 
@@ -349,7 +451,7 @@ void AliGenMUONLMR::Generate() {
     else { 
       PushTrack(1,ntmother,pdgPushed[ipart],
                pxPushed[ipart],pyPushed[ipart],pzPushed[ipart],ePushed[ipart],
-               origin0[0],origin0[1],origin0[2],time0,
+               origin0[0],origin0[1],origin0[2],0.,
                polar[0],polar[1],polar[2],
                kPDecay,ntchild,1,1);
       KeepTrack(ntchild); 
@@ -358,14 +460,13 @@ void AliGenMUONLMR::Generate() {
   SetHighWaterMark(ntchild); 
   AliGenEventHeader* header = new AliGenEventHeader("LMR");
   header->SetPrimaryVertex(fVertex);
-  header->SetInteractionTime(fTime);
   header->SetNProduced(fNprimaries);
   AddHeader(header); 
 }
 
 //------------------------------------------------------------------
 
-void AliGenMUONLMR::Decay2Body(const TParticle *mother){ 
+void AliGenMUONLMR::Decay2Body(TParticle *mother){ 
   // performs decay in two muons of the low mass resonances
   Double_t md1 = fMu[0]->GetMass(); 
   Int_t pdg = mother->GetPdgCode(); 
@@ -438,15 +539,9 @@ 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 = 0; 
-  if (TMath::Abs(mother->GetPdgCode())== 211) {
-      idmother = 0; 
-  } else if (TMath::Abs(mother->GetPdgCode())== 321) {
-      idmother = 1; 
-  } else {
-      AliWarning("Mother particle is not a pion or kaon \n");
-  }
-  
+  Int_t idmother = -1; 
+  if (TMath::Abs(mother->GetPdgCode())== 211) idmother = 0; 
+  if (TMath::Abs(mother->GetPdgCode())== 321) idmother = 1; 
   Double_t gammaRes = mother->Energy()/mres;
   Double_t zResCM = fDecay[idmother]->GetRandom();
   Double_t zResLab = gammaRes*zResCM;  
@@ -457,7 +552,7 @@ void AliGenMUONLMR::DecayPiK(TParticle *mother, Bool_t &hasDecayed){
 
 //-------------------------------------------------------------------
 
-void AliGenMUONLMR::DalitzDecay(const TParticle *mother){
+void AliGenMUONLMR::DalitzDecay(TParticle *mother){
   //
   // perform dalitz decays of eta, omega and etaprime 
   //
@@ -466,21 +561,10 @@ void AliGenMUONLMR::DalitzDecay(const 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 = 0; 
-  if (TMath::Abs(mother->GetPdgCode())== 221) {
-    // eta
-      index = 0; 
-  } else if (TMath::Abs(mother->GetPdgCode())== 223) {
-    // omega
-      index = 1; 
-  } else if (mother->GetPdgCode() == 331) {
-    // eta'
-      index = 2; 
-  } else {
-      AliWarning("Mother particle is not a eta, eta' or omega \n");
-  }
-
+  Int_t index = -1; 
+  if (mother->GetPdgCode() == 221) index = 0;  // eta
+  else if (mother->GetPdgCode() == 223) index = 1; // omega  
+  else if (mother->GetPdgCode() == 331) index = 2; // etaPrime  
   Int_t flag = 0; 
   Double_t xd=0, mvirt2=0; 
   Double_t countIt = 0;
@@ -648,7 +732,7 @@ Double_t AliGenMUONLMR::FormFactor(Double_t q2, Int_t decay){
 
 //____________________________________________________________
 
-Double_t AliGenMUONLMR::RhoLineShapeNew(const Double_t *x, const Double_t* /*para*/){
+Double_t AliGenMUONLMR::RhoLineShapeNew(Double_t *x, Double_t */*para*/){
   //new parameterization implemented by Hiroyuki Sako (GSI)
   Double_t mass = *x;
   double r, GammaTot;
@@ -658,6 +742,7 @@ Double_t AliGenMUONLMR::RhoLineShapeNew(const Double_t *x, const Double_t* /*par
   Double_t Gamma0  = TDatabasePDG::Instance()->GetParticle("rho0")->Width();
 
   const double Norm = 0.0744416*1.01;  
+
   // 0.0744416 at m = 0.72297
   // is the max number with Norm=1 (for rho)
   
@@ -686,5 +771,3 @@ Double_t AliGenMUONLMR::RhoLineShapeNew(const Double_t *x, const Double_t* /*par
 
   return r;
 }
-
-