X-Git-Url: http://git.uio.no/git/?p=u%2Fmrichter%2FAliRoot.git;a=blobdiff_plain;f=PYTHIA6%2FAliPythia.cxx;h=85fd6d166ca84df47da38eee4665a75a750d509f;hp=798e76a9e8230c73aece941d82d2dfb1cbdaaa0d;hb=6e90ad26e1a089242d111e1ce6ce2eec2b38aff9;hpb=452af8c788be92fcf9c3d726d0453b34867512ab diff --git a/PYTHIA6/AliPythia.cxx b/PYTHIA6/AliPythia.cxx index 798e76a9e82..85fd6d166ca 100644 --- a/PYTHIA6/AliPythia.cxx +++ b/PYTHIA6/AliPythia.cxx @@ -17,6 +17,9 @@ #include "AliPythia.h" #include "AliPythiaRndm.h" +#include "../FASTSIM/AliFastGlauber.h" +#include "../FASTSIM/AliQuenchingWeights.h" +#include "TVector3.h" ClassImp(AliPythia) @@ -49,7 +52,8 @@ AliPythia::AliPythia() // Set random number if (!AliPythiaRndm::GetPythiaRandom()) AliPythiaRndm::SetPythiaRandom(GetRandom()); - + fGlauber = 0; + fQuenchingWeights = 0; } void AliPythia::ProcInit(Process_t process, Float_t energy, StrucFunc_t strucfunc) @@ -136,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: // @@ -527,6 +552,28 @@ void AliPythia::Pyrobo(Int_t imi, Int_t ima, Double_t the, Double_t phi, Double_ +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() { // @@ -534,256 +581,497 @@ 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 z using: +// 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. // - Float_t p0[2][5]; - Float_t p1[2][5]; - Float_t p2[2][5]; +// +// + 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]; - for (Int_t i = 0; i < 4; i++) - { - p0[0][i] = 0.; - p0[1][i] = 0.; - p1[0][i] = 0.; - p1[1][i] = 0.; - p2[0][i] = 0.; - p2[1][i] = 0.; - } Int_t numpart = fPyjets->N; - - for (Int_t i = 0; i < numpart; i++) - { - Int_t imo = fPyjets->K[2][i]; - Int_t kst = fPyjets->K[0][i]; - Int_t pdg = fPyjets->K[1][i]; - -// Quarks and gluons only - if (pdg != 21 && TMath::Abs(pdg) > 6) continue; - -// Particles from hard scattering only - - - Float_t px = fPyjets->P[0][i]; - Float_t py = fPyjets->P[1][i]; - Float_t pz = fPyjets->P[2][i]; - Float_t e = fPyjets->P[3][i]; - Float_t m = fPyjets->P[4][i]; - Float_t pt = TMath::Sqrt(px * px + py * py); -// Skip comment lines - if (kst != 1 && kst != 2) continue; - - Float_t mt = TMath::Sqrt(px * px + py * py + m * m); - - // - // Some cuts to be in a save kinematic region - // - if (imo != 7 && imo != 8) continue; - Int_t index = imo - 7; - klast[index] = i; - - p0[index][0] += px; - p0[index][1] += py; - p0[index][2] += pz; - p0[index][3] += e; + 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; - // - // Fix z - // - - Float_t z = 0.2; - Float_t eppzOld = e + pz; - Float_t empzOld = e - pz; - - - // - // Kinematics of the original parton - // - - Float_t eppzNew = (1. - z) * eppzOld; - Float_t empzNew = empzOld - mt * mt * z / eppzOld; - Float_t eNew0 = 0.5 * (eppzNew + empzNew); - Float_t pzNew0 = 0.5 * (eppzNew - empzNew); - // - // Skip if pt too small - // - if (m * m > eppzNew * empzNew) continue; - Float_t ptNew = TMath::Sqrt(eppzNew * empzNew - m * m); - Float_t pxNew0 = ptNew / pt * px; - Float_t pyNew0 = ptNew / pt * py; - - p1[index][0] += pxNew0; - p1[index][1] += pyNew0; - p1[index][2] += pzNew0; - p1[index][3] += eNew0; - // - // Update event record - // - fPyjets->P[0][i] = pxNew0; - fPyjets->P[1][i] = pyNew0; - fPyjets->P[2][i] = pzNew0; - fPyjets->P[3][i] = eNew0; + 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]; - } - - // - // Gluons - // - - for (Int_t k = 0; k < 2; k++) - { - for (Int_t j = 0; j < 4; j++) - { - p2[k][j] = p0[k][j] - p1[k][j]; - } - p2[k][4] = p2[k][3] * p2[k][3] - p2[k][0] * p2[k][0] - p2[k][1] * p2[k][1] - p2[k][2] * p2[k][2]; - - if (p2[k][4] > 0.) - { - + // 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.); // - // Bring gluon back to mass shell via momentum scaling - // (momentum will not be conserved, but energy) + // Energy loss for given length and parton typr + Int_t itype = (pdg == 21) ? 2 : 1; + Double_t eloss = fQuenchingWeights->GetELossRandom(itype, l, eq[j]); // - // not used anymore -/* - Float_t psq = p2[k][0] * p2[k][0] + p2[k][1] * p2[k][1] + p2[k][2] * p2[k][2]; - Float_t fact = TMath::Sqrt(1. + p2[k][4] / psq); - p2[k][0] *= fact; - p2[k][1] *= fact; - p2[k][2] *= fact; - p2[k][3] = TMath::Sqrt(psq) * fact; - p2[k][4] = 0.; -*/ + // 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); } - - if (p2[0][4] > 0.) { - p2[0][4] = TMath::Sqrt(p2[0][4]); - } else { - printf("Warning negative mass squared ! \n"); - } - - if (p2[1][4] > 0.) { - p2[1][4] = TMath::Sqrt(p2[1][4]); - } else { - printf("Warning negative mass squared ! \n"); - } - - // - // Add the gluons - // - + +// +// 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); - for (Int_t i = 0; i < 2; i++) { - Int_t ish, jmin, jmax, iGlu, iNew; - Int_t in = klast[i]; - ish = 0; - - if (in == -1) continue; - if (i == 1 && klast[1] > klast[0]) in += ish; - - jmin = in - 1; - ish = 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]; +*/ - if (p2[i][4] > 0) ish = 2; - - 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; - } + 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. - kglu[i] = iGlu; - - 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]; - 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 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); + } + - Float_t pst = p2[i][4] / 2.; + // + // 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 - Float_t cost = 2. * gRandom->Rndm() - 1.; - Float_t sint = TMath::Sqrt(1. - cost * cost); - Float_t phi = 2. * TMath::Pi() * gRandom->Rndm(); +// 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 - Float_t pz1 = pst * cost; - Float_t pz2 = -pst * cost; - Float_t pt1 = pst * sint; - Float_t pt2 = -pst * sint; - Float_t px1 = pt1 * TMath::Cos(phi); - Float_t py1 = pt1 * TMath::Sin(phi); - Float_t px2 = pt2 * TMath::Cos(phi); - Float_t py2 = pt2 * TMath::Sin(phi); + 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; - 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.; + 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; + } - fPyjets->K[0][iGlu] = 2; - fPyjets->K[1][iGlu] = 21; - fPyjets->K[2][iGlu] = fPyjets->K[2][iNew]; - 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.; + 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 - fPyjets->K[0][iGlu+1] = 2; - fPyjets->K[1][iGlu+1] = 21; - fPyjets->K[2][iGlu+1] = fPyjets->K[2][iNew]; - 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); - + 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); + } } - } // end adding gluons -} // end quench - + + // 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