X-Git-Url: http://git.uio.no/git/?p=u%2Fmrichter%2FAliRoot.git;a=blobdiff_plain;f=PYTHIA6%2FAliPythia.cxx;h=85fd6d166ca84df47da38eee4665a75a750d509f;hp=f4747604bccc57ffd817a7faaf04d786ce657346;hb=6e90ad26e1a089242d111e1ce6ce2eec2b38aff9;hpb=adf4d898e02daf4e16db15bc3efb5186d8be83c5 diff --git a/PYTHIA6/AliPythia.cxx b/PYTHIA6/AliPythia.cxx index f4747604bcc..85fd6d166ca 100644 --- a/PYTHIA6/AliPythia.cxx +++ b/PYTHIA6/AliPythia.cxx @@ -13,114 +13,33 @@ * provided "as is" without express or implied warranty. * **************************************************************************/ -/* -$Log$ -Revision 1.1 2003/03/15 15:00:48 morsch -Classed imported from EVGEN. - -Revision 1.28 2002/12/09 08:22:56 morsch -UA1 jet finder (Pycell) for software triggering added. - -Revision 1.27 2002/11/15 00:39:37 morsch -- Correct initialisation of sRandom. -- QCD Jets with initial and final state gluon radiation is default -- pt kick for jets default -- Interface to Pyclus added. - -Revision 1.26 2002/11/14 00:37:32 morsch -Warning message for kPyJets added. - -Revision 1.25 2002/10/14 14:55:35 hristov -Merging the VirtualMC branch to the main development branch (HEAD) - -Revision 1.20.6.1 2002/06/10 14:57:41 hristov -Merged with v3-08-02 - -Revision 1.24 2002/05/22 13:22:53 morsch -Process kPyMbNonDiffr added. - -Revision 1.23 2002/05/06 07:17:29 morsch -Pyr gives random number r in interval 0 < r < 1. - -Revision 1.22 2002/04/26 10:28:48 morsch -Option kPyBeautyPbMNR added (N. Carrer). - -Revision 1.21 2002/03/25 14:46:16 morsch -Case kPyD0PbMNR added (N. Carrer). - -Revision 1.20 2002/03/03 13:48:50 morsch -Option kPyCharmPbMNR added. Produce charm pairs in agreement with MNR -NLO calculations (Nicola Carrer). - -Revision 1.19 2002/02/20 08:52:20 morsch -Correct documentation of SetNuclei method. - -Revision 1.18 2002/02/07 10:43:06 morsch -Tuned pp-min.bias settings (M.Monteno, R.Ugoccioni and N.Carrer) - -Revision 1.17 2001/12/19 15:40:43 morsch -For kPyJets enforce simple jet topology, i.e no initial or final state -gluon radiation and no primordial pT. - -Revision 1.16 2001/10/12 11:13:59 morsch -Missing break statements added (thanks to Nicola Carrer) - -Revision 1.15 2001/03/27 10:54:50 morsch -Add ResetDecayTable() and SsetDecayTable() methods. - -Revision 1.14 2001/03/09 13:03:40 morsch -Process_t and Struc_Func_t moved to AliPythia.h - -Revision 1.13 2000/12/18 08:55:35 morsch -Make AliPythia dependent generartors work with new scheme of random number generation - -Revision 1.12 2000/11/30 07:12:50 alibrary -Introducing new Rndm and QA classes - -Revision 1.11 2000/10/20 06:30:06 fca -Use version 0 to avoid streamer generation - -Revision 1.10 2000/10/06 14:18:44 morsch -Upper cut of prim. pT distribution set to 5. GeV - -Revision 1.9 2000/09/18 10:41:35 morsch -Add possibility to use nuclear structure functions from PDF library V8. - -Revision 1.8 2000/09/06 14:26:24 morsch -Decayer functionality of AliPythia has been moved to AliDecayerPythia. -Class is now a singleton. - -Revision 1.7 2000/06/09 20:34:50 morsch -All coding rule violations except RS3 corrected - -Revision 1.6 1999/11/09 07:38:48 fca -Changes for compatibility with version 2.23 of ROOT - -Revision 1.5 1999/11/03 17:43:20 fca -New version from G.Martinez & A.Morsch - -Revision 1.4 1999/09/29 09:24:14 fca -Introduction of the Copyright and cvs Log - -*/ - +/* $Id$ */ #include "AliPythia.h" +#include "AliPythiaRndm.h" +#include "../FASTSIM/AliFastGlauber.h" +#include "../FASTSIM/AliQuenchingWeights.h" +#include "TVector3.h" ClassImp(AliPythia) #ifndef WIN32 # define pyclus pyclus_ # define pycell pycell_ +# define pyshow pyshow_ +# define pyrobo pyrobo_ # define type_of_call #else # define pyclus PYCLUS # define pycell PYCELL +# define pyrobo PYROBO # define type_of_call _stdcall #endif extern "C" void type_of_call pyclus(Int_t & ); extern "C" void type_of_call pycell(Int_t & ); +extern "C" void type_of_call pyshow(Int_t &, Int_t &, Double_t &); +extern "C" void type_of_call pyrobo(Int_t &, Int_t &, Double_t &, Double_t &, Double_t &, Double_t &, Double_t &); //_____________________________________________________________________________ @@ -131,14 +50,17 @@ AliPythia::AliPythia() // Default Constructor // // Set random number - if (!sRandom) sRandom=fRandom; - + if (!AliPythiaRndm::GetPythiaRandom()) + AliPythiaRndm::SetPythiaRandom(GetRandom()); + fGlauber = 0; + fQuenchingWeights = 0; } void AliPythia::ProcInit(Process_t process, Float_t energy, StrucFunc_t strucfunc) { // Initialise the process to generate - if (!sRandom) sRandom = gRandom; + if (!AliPythiaRndm::GetPythiaRandom()) + AliPythiaRndm::SetPythiaRandom(GetRandom()); fProcess = process; fEcms = energy; @@ -218,30 +140,51 @@ void AliPythia::ProcInit(Process_t process, Float_t energy, StrucFunc_t strucfun // // select Pythia min. bias model SetMSEL(0); - SetMSUB(92,1); // single diffraction AB-->XB - SetMSUB(93,1); // single diffraction AB-->AX - SetMSUB(94,1); // double diffraction - SetMSUB(95,1); // low pt production - SetMSTP(81,1); // multiple interactions switched on - SetMSTP(82,3); // model with varying impact param. & a single Gaussian - SetPARP(82,3.47); // set value pT_0 for turn-off of the cross section of - // multiple interaction at a reference energy = 14000 GeV - SetPARP(89,14000.); // reference energy for the above parameter - SetPARP(90,0.174); // set exponent for energy dependence of pT_0 + SetMSUB(92,1); // single diffraction AB-->XB + SetMSUB(93,1); // single diffraction AB-->AX + SetMSUB(94,1); // double diffraction + SetMSUB(95,1); // low pt production + +// +// ATLAS Tuning +// + SetMSTP(51,7); // CTEQ5L pdf + SetMSTP(81,1); // Multiple Interactions ON + SetMSTP(82,4); // Double Gaussian Model + + SetPARP(82,1.8); // [GeV] PT_min at Ref. energy + SetPARP(89,1000.); // [GeV] Ref. energy + SetPARP(90,0.16); // 2*epsilon (exponent in power law) + SetPARP(83,0.5); // Core density in proton matter distribution (def.value) + SetPARP(84,0.5); // Core radius + SetPARP(85,0.33); // Regulates gluon prod. mechanism + SetPARP(86,0.66); // Regulates gluon prod. mechanism + SetPARP(67,1); // Regulates Initial State Radiation + break; case kPyMbNonDiffr: // Minimum Bias pp-Collisions // // // select Pythia min. bias model SetMSEL(0); - SetMSUB(95,1); // low pt production - SetMSTP(81,1); // multiple interactions switched on - SetMSTP(82,3); // model with varying impact param. & a single Gaussian - SetPARP(82,3.47); // set value pT_0 for turn-off of the cross section of - // multiple interaction at a reference energy = 14000 GeV - SetPARP(89,14000.); // reference energy for the above parameter - SetPARP(90,0.174); // set exponent for energy dependence of pT_0 - + SetMSUB(95,1); // low pt production + +// +// ATLAS Tuning +// + + SetMSTP(51,7); // CTEQ5L pdf + SetMSTP(81,1); // Multiple Interactions ON + SetMSTP(82,4); // Double Gaussian Model + + SetPARP(82,1.8); // [GeV] PT_min at Ref. energy + SetPARP(89,1000.); // [GeV] Ref. energy + SetPARP(90,0.16); // 2*epsilon (exponent in power law) + SetPARP(83,0.5); // Core density in proton matter distribution (def.value) + SetPARP(84,0.5); // Core radius + SetPARP(85,0.33); // Regulates gluon prod. mechanism + SetPARP(86,0.66); // Regulates gluon prod. mechanism + SetPARP(67,1); // Regulates Initial State Radiation break; case kPyJets: // @@ -532,17 +475,15 @@ void AliPythia::PrintParticles() { // Print list of particl properties Int_t np = 0; - + char* name = new char[16]; for (Int_t kf=0; kf<1000000; kf++) { for (Int_t c = 1; c > -2; c-=2) { - Int_t kc = Pycomp(c*kf); if (kc) { Float_t mass = GetPMAS(kc,1); Float_t width = GetPMAS(kc,2); Float_t tau = GetPMAS(kc,4); - - char* name = new char[8]; + Pyname(kf,name); np++; @@ -586,34 +527,551 @@ void AliPythia::Pycell(Int_t& njet) pycell(njet); } +void AliPythia::Pyshow(Int_t ip1, Int_t ip2, Double_t qmax) +{ +// Call Pythia jet reconstruction algorithm +// + Int_t numpart = fPyjets->N; + for (Int_t i = 0; i < numpart; i++) + { + if (fPyjets->K[2][i] == 7) ip1 = i+1; + if (fPyjets->K[2][i] == 8) ip2 = i+1; + } + + qmax = 2. * GetVINT(51); + printf("Pyshow %d %d %f", ip1, ip2, qmax); + + pyshow(ip1, ip2, qmax); +} -#ifndef WIN32 -#define pyr pyr_ -#define pyrset pyrset_ -#define pyrget pyrget_ -#define pyclus pyclus_ -#define pycell pycell_ -#else -#define pyr PYR -#define pyrset PYRSET -#define pyrget PYRGET -#define pyclus PYCLUS -#define pycell PYCELL -#endif - -extern "C" { - Double_t pyr(Int_t*) +void AliPythia::Pyrobo(Int_t imi, Int_t ima, Double_t the, Double_t phi, Double_t bex, Double_t bey, Double_t bez) { - Float_t r; - do r=sRandom->Rndm(); while(0 >= r || r >= 1); - return r; + pyrobo(imi, ima, the, phi, bex, bey, bez); } - void pyrset(Int_t*,Int_t*) {} - void pyrget(Int_t*,Int_t*) {} + + + +void AliPythia::InitQuenching(Float_t cMin, Float_t cMax, Float_t qTransport, Float_t maxLength, Int_t iECMethod) +{ +// Initializes +// (1) The quenching model using quenching weights according to C. Salgado and U. Wiedemann +// (2) The nuclear geometry using the Glauber Model +// + + + fGlauber = new AliFastGlauber(); + fGlauber->Init(2); + fGlauber->SetCentralityClass(cMin, cMax); + + fQuenchingWeights = new AliQuenchingWeights(); + fQuenchingWeights->InitMult(); + fQuenchingWeights->SetQTransport(qTransport); + fQuenchingWeights->SetECMethod(AliQuenchingWeights::kECMethod(iECMethod)); + fQuenchingWeights->SetLengthMax(Int_t(maxLength)); + fQuenchingWeights->SampleEnergyLoss(); + } +void AliPythia::Quench() +{ +// +// +// Simple Jet Quenching routine: +// ============================= +// The jet formed by all final state partons radiated by the parton created +// in the hard collisions is quenched by a factor (1-z) using light cone variables in +// the initial parton reference frame: +// (E + p_z)new = (1-z) (E + p_z)old +// +// +// +// +// The lost momentum is first balanced by one gluon with virtuality > 0. +// Subsequently the gluon splits to yield two gluons with E = p. +// +// +// + const Int_t kGluons = 1; + + Double_t p0[2][5]; + Double_t p1[2][5]; + Double_t p2[2][5]; + Int_t klast[2] = {-1, -1}; + Int_t kglu[2]; + + Int_t numpart = fPyjets->N; + Double_t px = 0., py = 0., pz = 0., e = 0., m = 0., p = 0., pt = 0., theta = 0.; + Double_t pxq[2], pyq[2], pzq[2], eq[2], yq[2], mq[2], pq[2], phiq[2], thetaq[2], ptq[2]; + Bool_t quenched[2]; + Double_t phi; + Double_t zInitial[2], wjtKick[2]; + Int_t imo, kst, pdg; +// +// Primary partons +// + + for (Int_t i = 6; i <= 7; i++) { + Int_t j = i - 6; + + pxq[j] = fPyjets->P[0][i]; + pyq[j] = fPyjets->P[1][i]; + pzq[j] = fPyjets->P[2][i]; + eq[j] = fPyjets->P[3][i]; + mq[j] = fPyjets->P[4][i]; + yq[j] = 0.5 * TMath::Log((e + pz + 1.e-14) / (e - pz + 1.e-14)); + pq[j] = TMath::Sqrt(pxq[j] * pxq[j] + pyq[j] * pyq[j] + pzq[j] * pzq[j]); + phiq[j] = TMath::Pi()+TMath::ATan2(-pyq[j], -pxq[j]); + ptq[j] = TMath::Sqrt(pxq[j] * pxq[j] + pyq[j] * pyq[j]); + thetaq[j] = TMath::ATan2(ptq[j], pzq[j]); + phi = phiq[j]; + + // Quench only central jets + if (TMath::Abs(yq[j]) > 2.5) { + zInitial[j] = 0.; + } else { + pdg = fPyjets->K[1][i]; + + // Get length in nucleus + Double_t l; + fGlauber->GetLengthsForPythia(1, &phi, &l, -1.); + // + // Energy loss for given length and parton typr + Int_t itype = (pdg == 21) ? 2 : 1; + Double_t eloss = fQuenchingWeights->GetELossRandom(itype, l, eq[j]); + // + // Extra pt + wjtKick[j] = TMath::Sqrt(l * fQuenchingWeights->GetQTransport()); + // + // Fractional energy loss + zInitial[j] = eloss / eq[j]; + // + // Avoid complete loss + // + if (zInitial[j] == 1.) zInitial[j] = 0.95; + // + // Some debug printing + printf("Initial parton # %3d, Type %3d Energy %10.3f Phi %10.3f Length %10.3f Loss %10.3f Kick %10.3f\n", + j, itype, eq[j], phi, l, eloss, wjtKick[j]); + } + quenched[j] = (zInitial[j] > 0.01); + } + +// +// Radiated partons +// + zInitial[0] = 1. - TMath::Power(1. - zInitial[0], 1./Double_t(kGluons)); + zInitial[1] = 1. - TMath::Power(1. - zInitial[1], 1./Double_t(kGluons)); + wjtKick[0] = wjtKick[0] / TMath::Sqrt(Double_t(kGluons)); + wjtKick[1] = wjtKick[1] / TMath::Sqrt(Double_t(kGluons)); +// this->Pylist(1); + +// Arrays to store particle 4-momenta to be changed +// +/* + Double_t** pNew = new Double_t* [numpart]; + for (Int_t i = 0; i < numpart; i++) pNew[i] = new Double_t [4]; + Int_t* kNew = new Int_t [numpart]; +*/ + Double_t pNew[1000][4]; + Int_t kNew[1000]; + Int_t icount = 0; +// +// Radiation Loop + for (Int_t iglu = 0; iglu < kGluons; iglu++) { + for (Int_t isys = 0; isys < 2; isys++) { +// Skip to next system if not quenched. + + Double_t zHeavy = zInitial[isys]; + + if (!quenched[isys]) continue; +// + while (1) { + icount = 0; + for (Int_t k = 0; k < 4; k++) + { + p0[isys][k] = 0.; + p1[isys][k] = 0.; + p2[isys][k] = 0.; + } +// Loop over partons + for (Int_t i = 0; i < numpart; i++) + { + imo = fPyjets->K[2][i]; + kst = fPyjets->K[0][i]; + pdg = fPyjets->K[1][i]; + + + +// Quarks and gluons only + if (pdg != 21 && TMath::Abs(pdg) > 6) continue; +// Particles from hard scattering only + if (imo > 8 && imo < 1000) imo = fPyjets->K[2][imo - 1]; + if (imo != isys + 7 && imo != 1000 + isys + 7) continue; + +// Skip comment lines + if (kst != 1 && kst != 2) continue; +// +// Parton kinematic + px = fPyjets->P[0][i]; + py = fPyjets->P[1][i]; + pz = fPyjets->P[2][i]; + e = fPyjets->P[3][i]; + m = fPyjets->P[4][i]; + pt = TMath::Sqrt(px * px + py * py); + p = TMath::Sqrt(px * px + py * py + pz * pz); + phi = TMath::Pi() + TMath::ATan2(-py, -px); + theta = TMath::ATan2(pt, pz); + +// +// Save 4-momentum sum for balancing + Int_t index = imo - 7; + if (index >= 1000) index -= 1000; + + p0[index][0] += px; + p0[index][1] += py; + p0[index][2] += pz; + p0[index][3] += e; + +// Don't quench radiated gluons +// + if (imo == 1000 + isys + 7) { + p1[index][0] += px; + p1[index][1] += py; + p1[index][2] += pz; + p1[index][3] += e; + continue; + } + +// + + klast[index] = i; + +// +// Fractional energy loss + Double_t z = zInitial[index]; + if (m > 0.) z = zHeavy; + + // + // + // Transform into frame in which initial parton is along z-axis + // + TVector3 v(px, py, pz); + v.RotateZ(-phiq[index]); v.RotateY(-thetaq[index]); + Double_t pxs = v.X(); Double_t pys = v.Y(); Double_t pl = v.Z(); + + Double_t jt = TMath::Sqrt(pxs * pxs + pys * pys); + Double_t mt2 = jt * jt + m * m; + Double_t zmax = 1.; + // + // Kinematic limit on z + // + + if (m > 0.) { + zmax = 1. - m / TMath::Sqrt(m * m + jt * jt); + if (z > zmax) { + printf("We have to put z to the kinematic limit %e %e \n", z, zmax); + z = 0.9999 * zmax; + } // z > zmax + + if (z < 0.01) { +// +// If z is too small, there is no phase space for quenching +// + printf("No phase space for quenching ! %e \n", z); + p1[index][0] += px; + p1[index][1] += py; + p1[index][2] += pz; + p1[index][3] += e; + continue; + } + } // massive particles + + // + // Change light-cone kinematics rel. to initial parton + // + Double_t eppzOld = e + pl; + Double_t empzOld = e - pl; + + Double_t eppzNew = (1. - z) * eppzOld; + Double_t empzNew = empzOld - mt2 * z / eppzOld; + Double_t eNew = 0.5 * (eppzNew + empzNew); + Double_t plNew = 0.5 * (eppzNew - empzNew); + + Double_t jtNew; + // + // if mt very small (or sometimes even < 0 for numerical reasons) set it to 0 + Double_t mt2New = eppzNew * empzNew; + if (mt2New < 1.e-8) mt2New = 0.; + + if (m * m > mt2New) { + // + // This should not happen + // + Fatal("Quench()", "This should never happen %e %e %e!", m, eppzNew, empzNew); + jtNew = 0; + } else { + jtNew = TMath::Sqrt(mt2New - m * m); + } + + + // + // Calculate new px, py + // + Double_t pxNew = jtNew / jt * pxs; + Double_t pyNew = jtNew / jt * pys; + +// Double_t dpx = pxs - pxNew; +// Double_t dpy = pys - pyNew; +// Double_t dpz = pl - plNew; +// Double_t de = e - eNew; +// Double_t dmass2 = de * de - dpx * dpx - dpy * dpy - dpz * dpz; +// printf("New mass (1) %e %e %e %e %e %e %e \n", dmass2, jt, jtNew, pl, plNew, e, eNew); +// printf("New mass (2) %e %e \n", pxNew, pyNew); + // + // Rotate back + // + TVector3 w(pxNew, pyNew, plNew); + w.RotateY(thetaq[index]); w.RotateZ(phiq[index]); + pxNew = w.X(); pyNew = w.Y(); plNew = w.Z(); + + p1[index][0] += pxNew; + p1[index][1] += pyNew; + p1[index][2] += plNew; + p1[index][3] += eNew; + // + // Updated 4-momentum vectors + // + pNew[icount][0] = pxNew; + pNew[icount][1] = pyNew; + pNew[icount][2] = plNew; + pNew[icount][3] = eNew; + kNew[icount] = i; + icount++; + } // parton loop + // + // Check if there was phase-space for quenching + // + + if (icount == 0) quenched[isys] = kFALSE; + if (!quenched[isys]) break; + + for (Int_t j = 0; j < 4; j++) + { + p2[isys][j] = p0[isys][j] - p1[isys][j]; + } + p2[isys][4] = p2[isys][3] * p2[isys][3] - p2[isys][0] * p2[isys][0] - p2[isys][1] * p2[isys][1] - p2[isys][2] * p2[isys][2]; + + if (p2[isys][4] > 0.) { + p2[isys][4] = TMath::Sqrt(p2[isys][4]); + break; + } else { + printf("Warning negative mass squared in system %d %f ! \n", isys, zInitial[isys]); + + printf("Kinematics %10.3e %10.3e %10.3e %10.3e %10.3e \n", p2[isys][0], p2[isys][1], p2[isys][2], p2[isys][3], p2[isys][4]); + if (p2[isys][4] < -0.01) { + printf("Negative mass squared ! Let's try to fix this by decreasing z\n"); +// this->Pylist(1); + + } else { + p2[isys][4] = 0.; + break; + } + } + // + // jt-kick + // + /* + TVector3 v(p2[k][0], p2[k][1], p2[k][2]); + v.RotateZ(-phiq[k]); + v.RotateY(-thetaq[k]); + Double_t px = v.X(); Double_t py = v.Y(); Double_t pz = v.Z(); + Double_t r = AliPythiaRndm::GetPythiaRandom()->Rndm(); + Double_t jtKick = wjtKick[k] * TMath::Sqrt(-TMath::Log(r)); + Double_t phiKick = 2. * TMath::Pi() * AliPythiaRndm::GetPythiaRandom()->Rndm(); + px += jtKick * TMath::Cos(phiKick); + py += jtKick * TMath::Sin(phiKick); + TVector3 w(px, py, pz); + w.RotateY(thetaq[k]); + w.RotateZ(phiq[k]); + p2[k][0] = w.X(); p2[k][1] = w.Y(); p2[k][2] = w.Z(); + p2[k][3] = TMath::Sqrt(p2[k][0] * p2[k][0] + p2[k][1] * p2[k][1] + p2[k][2] * p2[k][2] + p2[k][4] * p2[k][4]); + */ + zHeavy *= 0.98; + printf("zHeavy lowered to %f\n", zHeavy); + if (zHeavy < 0.01) { + printf("No success ! \n"); + icount = 0; + quenched[isys] = kFALSE; + break; + } + } // iteration on z + +// Update event record + for (Int_t k = 0; k < icount; k++) { +// printf("%6d %6d %10.3e %10.3e %10.3e %10.3e\n", k, kNew[k], pNew[k][0],pNew[k][1], pNew[k][2], pNew[k][3] ); + fPyjets->P[0][kNew[k]] = pNew[k][0]; + fPyjets->P[1][kNew[k]] = pNew[k][1]; + fPyjets->P[2][kNew[k]] = pNew[k][2]; + fPyjets->P[3][kNew[k]] = pNew[k][3]; + } + } // System + // + // Add the gluons + // + Int_t ish = 0; + for (Int_t i = 0; i < 2; i++) { + Int_t jmin, jmax, iGlu, iNew; + if (!quenched[i]) continue; +// +// Last parton from shower i + Int_t in = klast[i]; +// +// Continue if no parton in shower i selected + if (in == -1) continue; +// +// If this is the second initial parton and it is behind the first move pointer by previous ish + if (i == 1 && klast[1] > klast[0]) in += ish; +// +// Starting index + + jmin = in - 1; +// How many additional gluons will be generated + ish = 1; + if (p2[i][4] > 0.05) ish = 2; +// +// Position of gluons + iGlu = in; + iNew = in + ish; + jmax = numpart + ish - 1; + + if (fPyjets->K[0][in-1] == 1 || fPyjets->K[0][in-1] == 21 || fPyjets->K[0][in-1] == 11) { + jmin = in; + iGlu = in + 1; + iNew = in; + } + + kglu[i] = iGlu; +// +// Shift stack +// + for (Int_t j = jmax; j > jmin; j--) + { + for (Int_t k = 0; k < 5; k++) { + fPyjets->K[k][j] = fPyjets->K[k][j-ish]; + fPyjets->P[k][j] = fPyjets->P[k][j-ish]; + fPyjets->V[k][j] = fPyjets->V[k][j-ish]; + } + } // end shifting + + numpart += ish; + (fPyjets->N) += ish; + + if (ish == 1) { + fPyjets->P[0][iGlu] = p2[i][0]; + fPyjets->P[1][iGlu] = p2[i][1]; + fPyjets->P[2][iGlu] = p2[i][2]; + fPyjets->P[3][iGlu] = p2[i][3]; + fPyjets->P[4][iGlu] = p2[i][4]; + + fPyjets->K[0][iGlu] = 2; + fPyjets->K[1][iGlu] = 21; + fPyjets->K[2][iGlu] = fPyjets->K[2][iNew] + 1000; + fPyjets->K[3][iGlu] = -1; + fPyjets->K[4][iGlu] = -1; + } else { + // + // Split gluon in rest frame. + // + Double_t bx = p2[i][0] / p2[i][3]; + Double_t by = p2[i][1] / p2[i][3]; + Double_t bz = p2[i][2] / p2[i][3]; + Double_t pst = p2[i][4] / 2.; + // + // Isotropic decay ???? + Double_t cost = 2. * gRandom->Rndm() - 1.; + Double_t sint = TMath::Sqrt(1. - cost * cost); + Double_t phi = 2. * TMath::Pi() * gRandom->Rndm(); + + Double_t pz1 = pst * cost; + Double_t pz2 = -pst * cost; + Double_t pt1 = pst * sint; + Double_t pt2 = -pst * sint; + Double_t px1 = pt1 * TMath::Cos(phi); + Double_t py1 = pt1 * TMath::Sin(phi); + Double_t px2 = pt2 * TMath::Cos(phi); + Double_t py2 = pt2 * TMath::Sin(phi); + + fPyjets->P[0][iGlu] = px1; + fPyjets->P[1][iGlu] = py1; + fPyjets->P[2][iGlu] = pz1; + fPyjets->P[3][iGlu] = pst; + fPyjets->P[4][iGlu] = 0.; + + fPyjets->K[0][iGlu] = 2; + fPyjets->K[1][iGlu] = 21; + fPyjets->K[2][iGlu] = fPyjets->K[2][iNew] + 1000; + fPyjets->K[3][iGlu] = -1; + fPyjets->K[4][iGlu] = -1; + + fPyjets->P[0][iGlu+1] = px2; + fPyjets->P[1][iGlu+1] = py2; + fPyjets->P[2][iGlu+1] = pz2; + fPyjets->P[3][iGlu+1] = pst; + fPyjets->P[4][iGlu+1] = 0.; + + fPyjets->K[0][iGlu+1] = 2; + fPyjets->K[1][iGlu+1] = 21; + fPyjets->K[2][iGlu+1] = fPyjets->K[2][iNew] + 1000; + fPyjets->K[3][iGlu+1] = -1; + fPyjets->K[4][iGlu+1] = -1; + SetMSTU(1,0); + SetMSTU(2,0); + // + // Boost back + // + Pyrobo(iGlu + 1, iGlu + 2, 0., 0., bx, by, bz); + } + } + + // Check energy conservation + Double_t pxs = 0.; + Double_t pys = 0.; + Double_t pzs = 0.; + Double_t es = 14000.; + + for (Int_t i = 0; i < numpart; i++) + { + kst = fPyjets->K[0][i]; + if (kst != 1 && kst != 2) continue; + pxs += fPyjets->P[0][i]; + pys += fPyjets->P[1][i]; + pzs += fPyjets->P[2][i]; + es -= fPyjets->P[3][i]; + } + if (TMath::Abs(pxs) > 1.e-2 || + TMath::Abs(pys) > 1.e-2 || + TMath::Abs(pzs) > 1.e-1) { + printf("%e %e %e %e\n", pxs, pys, pzs, es); + this->Pylist(1); + Fatal("Quench()", "4-Momentum non-conservation"); + } +// this->Pylist(1); + + } // end quenching loop +// Clean-up + for (Int_t i = 0; i < numpart; i++) + { + imo = fPyjets->K[2][i]; + if (imo > 1000) fPyjets->K[2][i] = fPyjets->K[2][i] % 1000; + } + + +// delete[] kNew; +// delete[] pNew; + +} // end quench