X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=EVGEN%2FAliGenHIJINGpara.cxx;h=3596b4deb8b27d9ca3a4b1b52b6965ea4df7dfa5;hb=88f9563f489a1f774f263067cf17614bfa5c66b6;hp=8086a3ad5d0191f307343d8816865e0657baa0ad;hpb=b9d0a01d7a0723a09071b0b56200d72f59a9c2b6;p=u%2Fmrichter%2FAliRoot.git diff --git a/EVGEN/AliGenHIJINGpara.cxx b/EVGEN/AliGenHIJINGpara.cxx index 8086a3ad5d0..3596b4deb8b 100644 --- a/EVGEN/AliGenHIJINGpara.cxx +++ b/EVGEN/AliGenHIJINGpara.cxx @@ -13,54 +13,7 @@ * provided "as is" without express or implied warranty. * **************************************************************************/ -/* -$Log$ -Revision 1.11.4.1 2002/07/24 08:56:28 alibrary -Updating EVGEN on TVirtulaMC - -Revision 1.12 2002/06/19 06:56:34 hristov -Memory leak corrected - -Revision 1.11 2002/03/20 10:21:13 hristov -Set fPtMax to 15 GeV in order to avoid some numerical problems - -Revision 1.10 2001/10/15 16:44:46 morsch -- Possibility for vertex distribution truncation. -- Write mc header with vertex position. - -Revision 1.9 2001/07/27 17:09:36 morsch -Use local SetTrack, KeepTrack and SetHighWaterMark methods -to delegate either to local stack or to stack owned by AliRun. -(Piotr Skowronski, A.M.) - -Revision 1.8 2001/07/20 11:03:58 morsch -Issue warning message if used outside allowed eta range (-8 to 8). - -Revision 1.7 2001/07/17 12:41:01 morsch -- Calculation of fraction of event corresponding to selected pt-range corrected -(R. Turrisi) -- Parent weight corrected. - -Revision 1.6 2001/05/16 14:57:10 alibrary -New files for folders and Stack - -Revision 1.5 2000/12/21 16:24:06 morsch -Coding convention clean-up - -Revision 1.4 2000/11/30 07:12:50 alibrary -Introducing new Rndm and QA classes - -Revision 1.3 2000/10/02 21:28:06 fca -Removal of useless dependecies via forward declarations - -Revision 1.2 2000/07/11 18:24:55 fca -Coding convention corrections + few minor bug fixes - -Revision 1.1 2000/06/09 20:20:30 morsch -Same class as previously in AliSimpleGen.cxx -All coding rule violations except RS3 corrected (AM) - -*/ +/* $Id$ */ // Parameterisation of pi and K, eta and pt distributions // used for the ALICE TDRs. @@ -84,21 +37,27 @@ All coding rule violations except RS3 corrected (AM) // // /////////////////////////////////////////////////////////////////// -#include "AliGenHIJINGpara.h" +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include "AliConst.h" +#include "AliDecayer.h" #include "AliGenEventHeader.h" +#include "AliGenHIJINGpara.h" +#include "AliLog.h" #include "AliRun.h" -#include "AliConst.h" -#include "AliPDG.h" - -#include -#include ClassImp(AliGenHIJINGpara) -AliGenHIJINGpara::AliGenHIJINGpara(const AliGenHIJINGpara & para) -{ -// copy constructor -} + //_____________________________________________________________________________ static Double_t ptpi(Double_t *px, Double_t *) @@ -108,23 +67,24 @@ static Double_t ptpi(Double_t *px, Double_t *) // POWER LAW FOR PT > 500 MEV // MT SCALING BELOW (T=160 MEV) // - const Double_t kp0 = 1.3; - const Double_t kxn = 8.28; - const Double_t kxlim=0.5; - const Double_t kt=0.160; - const Double_t kxmpi=0.139; - const Double_t kb=1.; + const Double_t kp0 = 1.3; + const Double_t kxn = 8.28; + const Double_t kxlim = 0.5; + const Double_t kt = 0.160; + const Double_t kxmpi = 0.139; + const Double_t kb = 1.; Double_t y, y1, xmpi2, ynorm, a; - Double_t x=*px; + Double_t x = *px; // - y1=TMath::Power(kp0/(kp0+kxlim),kxn); - xmpi2=kxmpi*kxmpi; - ynorm=kb*(TMath::Exp(-sqrt(kxlim*kxlim+xmpi2)/kt)); - a=ynorm/y1; + y1 = TMath::Power(kp0 / (kp0 + kxlim), kxn); + xmpi2 = kxmpi * kxmpi; + ynorm = kb * (TMath::Exp(-sqrt(kxlim * kxlim + xmpi2) / kt )); + a = ynorm / y1; if (x > kxlim) - y=a*TMath::Power(kp0/(kp0+x),kxn); + y = a * TMath::Power(kp0 / (kp0 + x), kxn); else - y=kb*TMath::Exp(-sqrt(x*x+xmpi2)/kt); + y = kb* TMath::Exp(-sqrt(x * x + xmpi2) / kt); + return y*x; } @@ -198,32 +158,44 @@ static Double_t etakac( Double_t *py, Double_t *) //_____________________________________________________________________________ AliGenHIJINGpara::AliGenHIJINGpara() - :AliGenerator() + :AliGenerator(), + fNt(-1), + fNpartProd(0), + fPi0Decays(kFALSE), + fPtWgtPi(0.), + fPtWgtKa(0.), + fPtpi(0), + fPtka(0), + fETApic(0), + fETAkac(0), + fDecayer(0) { // // Default constructor // - fPtpi = 0; - fPtka = 0; - fETApic = 0; - fETAkac = 0; SetCutVertexZ(); SetPtRange(); } //_____________________________________________________________________________ AliGenHIJINGpara::AliGenHIJINGpara(Int_t npart) - :AliGenerator(npart) + :AliGenerator(npart), + fNt(-1), + fNpartProd(npart), + fPi0Decays(kFALSE), + fPtWgtPi(0.), + fPtWgtKa(0.), + fPtpi(0), + fPtka(0), + fETApic(0), + fETAkac(0), + fDecayer(0) { // // Standard constructor // - fName="HIGINGpara"; + fName="HIJINGpara"; fTitle="HIJING Parametrisation Particle Generator"; - fPtpi = 0; - fPtka = 0; - fETApic = 0; - fETAkac = 0; SetCutVertexZ(); SetPtRange(); } @@ -251,15 +223,21 @@ void AliGenHIJINGpara::Init() Float_t etaMax = -TMath::Log(TMath::Tan( TMath::Max((Double_t)fThetaMin/2,1.e-10))); fPtpi = new TF1("ptpi",&ptpi,0,20,0); + gROOT->GetListOfFunctions()->Remove(fPtpi); fPtka = new TF1("ptka",&ptka,0,20,0); + gROOT->GetListOfFunctions()->Remove(fPtka); + fPtpi->SetNpx(1000); + fPtka->SetNpx(1000); fETApic = new TF1("etapic",&etapic,etaMin,etaMax,0); + gROOT->GetListOfFunctions()->Remove(fETApic); fETAkac = new TF1("etakac",&etakac,etaMin,etaMax,0); + gROOT->GetListOfFunctions()->Remove(fETAkac); - TF1 etaPic0("etapic",&etapic,-7,7,0); - TF1 etaKac0("etakac",&etakac,-7,7,0); + TF1 etaPic0("etaPic0",&etapic,-7,7,0); + TF1 etaKac0("etaKac0",&etakac,-7,7,0); - TF1 ptPic0("ptpi",&ptpi,0.,15.,0); - TF1 ptKac0("ptka",&ptka,0.,15.,0); + TF1 ptPic0("ptPic0",&ptpi,0.,15.,0); + TF1 ptKac0("ptKac0",&ptka,0.,15.,0); Float_t intETApi = etaPic0.Integral(-0.5, 0.5); Float_t intETAka = etaKac0.Integral(-0.5, 0.5); @@ -279,21 +257,40 @@ void AliGenHIJINGpara::Init() // Fraction of events corresponding to the selected phi-range Float_t phiFrac = (fPhiMax-fPhiMin)/2/TMath::Pi(); - fParentWeight = Float_t(fNpart)/(intETASel*ptFrac*phiFrac); - printf("%s: The number of particles in the selected kinematic region corresponds to %f percent of a full event\n ", - ClassName(),100.*fParentWeight); + fParentWeight = (intETASel*ptFrac*phiFrac) / Float_t(fNpart); + + if (fAnalog != 0) { + fPtWgtPi = (fPtMax - fPtMin) / fPtpi->Integral(0., 20.); + fPtWgtKa = (fPtMax - fPtMin) / fPtka->Integral(0., 20.); + fParentWeight = (intETASel*phiFrac) / Float_t(fNpart); + } + + + AliInfo(Form("The number of particles in the selected kinematic region corresponds to %f percent of a full event", + 100./ fParentWeight)); // Issue warning message if etaMin or etaMax are outside the alowed range // of the parametrization if (etaMin < -8.001 || etaMax > 8.001) { - printf("\n \n WARNING FROM AliGenHIJINGPara !"); - printf("\n YOU ARE USING THE PARAMETERISATION OUTSIDE "); - printf("\n THE ALLOWED PSEUDORAPIDITY RANGE (-8. - 8.)"); - printf("\n YOUR LIMITS: %f %f \n \n ", etaMin, etaMax); + AliWarning("\nYOU ARE USING THE PARAMETERISATION OUTSIDE "); + AliWarning("THE ALLOWED PSEUDORAPIDITY RANGE (-8. - 8.)"); + AliWarning(Form("YOUR LIMITS: %f %f \n ", etaMin, etaMax)); + } +// +// + if (fPi0Decays && gMC) + fDecayer = gMC->GetDecayer(); + + if (fPi0Decays) + { + fDecayer->SetForceDecay(kNeutralPion); + fDecayer->Init(); } + } + //_____________________________________________________________________________ void AliGenHIJINGpara::Generate() { @@ -310,10 +307,10 @@ void AliGenHIJINGpara::Generate() const Int_t kKaons[4] = {kK0Long, kK0Short, kKPlus, kKMinus}; // Float_t origin[3]; - Float_t pt, pl, ptot; + Float_t pt, pl, ptot, wgt; Float_t phi, theta; Float_t p[3]; - Int_t i, part, nt, j; + Int_t i, part, j; // TF1 *ptf; TF1 *etaf; @@ -323,39 +320,40 @@ void AliGenHIJINGpara::Generate() for (j=0;j<3;j++) origin[j]=fOrigin[j]; if(fVertexSmear == kPerEvent) { - Float_t dv[3]; - dv[2] = 1.e10; - while(TMath::Abs(dv[2]) > fCutVertexZ*fOsigma[2]) { - Rndm(random,6); - for (j=0; j < 3; j++) { - dv[j] = fOsigma[j]*TMath::Cos(2*random[2*j]*TMath::Pi())* - TMath::Sqrt(-2*TMath::Log(random[2*j+1])); - } - } - for (j=0; j < 3; j++) origin[j] += dv[j]; + Vertex(); + for (j=0; j < 3; j++) origin[j] = fVertex[j]; } // if kPerEvent TArrayF eventVertex; eventVertex.Set(3); eventVertex[0] = origin[0]; eventVertex[1] = origin[1]; eventVertex[2] = origin[2]; - + for(i=0;iGetRandom())); if(thetafThetaMax) continue; - pt=ptf->GetRandom(); + + if (fAnalog == 0) { + pt = ptf->GetRandom(); + } else { + pt = fPtMin + random[3] * (fPtMax - fPtMin); + } + + pl=pt/TMath::Tan(theta); ptot=TMath::Sqrt(pt*pt+pl*pl); if(ptotfPMax) continue; @@ -369,24 +367,94 @@ void AliGenHIJINGpara::Generate() TMath::Sqrt(-2*TMath::Log(random[2*j+1])); } } - SetTrack(fTrackIt,-1,part,p,origin,polar,0,kPPrimary,nt,fParentWeight); + + if (fAnalog == 0) { + wgt = fParentWeight; + } else { + wgt *= (fParentWeight * ptf->Eval(pt)); + } + + + if (part == kPi0 && fPi0Decays){ +// +// Decay pi0 if requested + PushTrack(0,-1,part,p,origin,polar,0,kPPrimary,fNt,wgt); + KeepTrack(fNt); + DecayPi0(origin, p); + } else { + // printf("fNt %d", fNt); + PushTrack(fTrackIt,-1,part,p,origin,polar,0,kPPrimary,fNt,wgt); + + KeepTrack(fNt); + } + break; } + SetHighWaterMark(fNt); } +// + // Header AliGenEventHeader* header = new AliGenEventHeader("HIJINGparam"); // Event Vertex header->SetPrimaryVertex(eventVertex); + header->SetNProduced(fNpartProd); gAlice->SetGenEventHeader(header); } -AliGenHIJINGpara& AliGenHIJINGpara::operator=(const AliGenHIJINGpara& rhs) -{ -// Assignment operator - return *this; +void AliGenHIJINGpara::SetPtRange(Float_t ptmin, Float_t ptmax) { + AliGenerator::SetPtRange(ptmin, ptmax); } -void AliGenHIJINGpara::SetPtRange(Float_t ptmin, Float_t ptmax) { - AliGenerator::SetPtRange(ptmin, ptmax); +void AliGenHIJINGpara::DecayPi0(Float_t* orig, Float_t * p) +{ +// +// Decay the pi0 +// and put decay products on the stack +// + static TClonesArray *particles; + if(!particles) particles = new TClonesArray("TParticle",1000); +// + const Float_t kMass = TDatabasePDG::Instance()->GetParticle(kPi0)->Mass(); + Float_t e = TMath::Sqrt(p[0] * p[0] + p[1] * p[1] + p[2] * p[2]+ kMass * kMass); +// +// Decay the pi0 + TLorentzVector pmom(p[0], p[1], p[2], e); + fDecayer->Decay(kPi0, &pmom); + +// +// Put decay particles on the stack +// + Float_t polar[3] = {0., 0., 0.}; + Int_t np = fDecayer->ImportParticles(particles); + fNpartProd += (np-1); + Int_t nt = 0; + for (Int_t i = 1; i < np; i++) + { + TParticle* iParticle = (TParticle *) particles->At(i); + p[0] = iParticle->Px(); + p[1] = iParticle->Py(); + p[2] = iParticle->Pz(); + Int_t part = iParticle->GetPdgCode(); + + PushTrack(fTrackIt, fNt, part, p, orig, polar, 0, kPDecay, nt, fParentWeight); + KeepTrack(nt); + } + fNt = nt; } +void AliGenHIJINGpara::Draw( const char * /*opt*/) +{ + // + // Draw the pT and y Distributions + // + TCanvas *c0 = new TCanvas("c0","Canvas 0",400,10,600,700); + c0->Divide(2,1); + c0->cd(1); + fPtpi->Draw(); + fPtpi->GetHistogram()->SetXTitle("p_{T} (GeV)"); + c0->cd(2); + fPtka->Draw(); + fPtka->GetHistogram()->SetXTitle("p_{T} (GeV)"); + +}