#include "AliPythia.h"
#include "AliPythiaRndm.h"
+#include "../FASTSIM/AliFastGlauber.h"
+#include "../FASTSIM/AliQuenchingWeights.h"
+#include "TVector3.h"
ClassImp(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)
//
// 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, kCTEQ5L); // 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:
//
// Set c-quark mass
SetPMAS(4,1,1.2);
+ break;
+ case kPyDPlusPbPbMNR:
+ // Tuning of Pythia parameters aimed to get a resonable agreement
+ // between with the NLO calculation by Mangano, Nason, Ridolfi for the
+ // c-cbar single inclusive and double differential distributions.
+ // This parameter settings are meant to work with Pb-Pb collisions
+ // (AliGenPythia::SetNuclei) and with kCTEQ4L PDFs.
+ // To get a good agreement the minimum ptHard (AliGenPythia::SetPtHard)
+ // has to be set to 2.1GeV. Example in ConfigCharmPPR.C.
+
+ // All QCD processes
+ SetMSEL(1);
+
+ // No multiple interactions
+ SetMSTP(81,0);
+ SetPARP(81,0.0);
+ SetPARP(82,0.0);
+
+ // Initial/final parton shower on (Pythia default)
+ SetMSTP(61,1);
+ SetMSTP(71,1);
+
+ // 2nd order alpha_s
+ SetMSTP(2,2);
+
+ // QCD scales
+ SetMSTP(32,2);
+ SetPARP(34,1.0);
+
+ // Intrinsic <kT>
+ SetMSTP(91,1);
+ SetPARP(91,1.304);
+ SetPARP(93,6.52);
+
+ // Set c-quark mass
+ SetPMAS(4,1,1.2);
+
break;
case kPyCharmpPbMNR:
case kPyD0pPbMNR:
// Set c-quark mass
SetPMAS(4,1,1.2);
+ break;
+ case kPyDPluspPbMNR:
+ // Tuning of Pythia parameters aimed to get a resonable agreement
+ // between with the NLO calculation by Mangano, Nason, Ridolfi for the
+ // c-cbar single inclusive and double differential distributions.
+ // This parameter settings are meant to work with p-Pb collisions
+ // (AliGenPythia::SetNuclei) and with kCTEQ4L PDFs.
+ // To get a good agreement the minimum ptHard (AliGenPythia::SetPtHard)
+ // has to be set to 2.1GeV. Example in ConfigCharmPPR.C.
+
+ // All QCD processes
+ SetMSEL(1);
+
+ // No multiple interactions
+ SetMSTP(81,0);
+ SetPARP(81,0.0);
+ SetPARP(82,0.0);
+
+ // Initial/final parton shower on (Pythia default)
+ SetMSTP(61,1);
+ SetMSTP(71,1);
+
+ // 2nd order alpha_s
+ SetMSTP(2,2);
+
+ // QCD scales
+ SetMSTP(32,2);
+ SetPARP(34,1.0);
+
+ // Intrinsic <kT>
+ SetMSTP(91,1);
+ SetPARP(91,1.16);
+ SetPARP(93,5.8);
+
+ // Set c-quark mass
+ SetPMAS(4,1,1.2);
+
break;
case kPyCharmppMNR:
case kPyD0ppMNR:
// Set c-quark mass
SetPMAS(4,1,1.2);
+ break;
+ case kPyDPlusppMNR:
+ // Tuning of Pythia parameters aimed to get a resonable agreement
+ // between with the NLO calculation by Mangano, Nason, Ridolfi for the
+ // c-cbar single inclusive and double differential distributions.
+ // This parameter settings are meant to work with pp collisions
+ // (AliGenPythia::SetNuclei) and with kCTEQ4L PDFs.
+ // To get a good agreement the minimum ptHard (AliGenPythia::SetPtHard)
+ // has to be set to 2.1GeV. Example in ConfigCharmPPR.C.
+
+ // All QCD processes
+ SetMSEL(1);
+
+ // No multiple interactions
+ SetMSTP(81,0);
+ SetPARP(81,0.0);
+ SetPARP(82,0.0);
+
+ // Initial/final parton shower on (Pythia default)
+ SetMSTP(61,1);
+ SetMSTP(71,1);
+
+ // 2nd order alpha_s
+ SetMSTP(2,2);
+
+ // QCD scales
+ SetMSTP(32,2);
+ SetPARP(34,1.0);
+
+ // Intrinsic <kT^2>
+ SetMSTP(91,1);
+ SetPARP(91,1.);
+ SetPARP(93,5.);
+
+ // Set c-quark mass
+ SetPMAS(4,1,1.2);
+
break;
case kPyBeautyPbPbMNR:
// Tuning of Pythia parameters aimed to get a resonable agreement
{
// 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);
}
+void AliPythia::InitQuenching(Float_t cMin, Float_t cMax, Float_t k, 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->SetK(k);
+ fQuenchingWeights->SetECMethod(AliQuenchingWeights::kECMethod(iECMethod));
+}
+
+
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];
- 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.;
- }
+//
+//
+ static Float_t eMean = 0.;
+ static Int_t icall = 0;
+
+ Double_t p0[4][5];
+ Double_t p1[4][5];
+ Double_t p2[4][5];
+ Int_t klast[4] = {-1, -1, -1, -1};
Int_t numpart = fPyjets->N;
+ Double_t px = 0., py = 0., pz = 0., e = 0., m = 0., p = 0., pt = 0., theta = 0., phi = 0.;
+ Double_t pxq[4], pyq[4], pzq[4], eq[4], yq[4], mq[4], pq[4], phiq[4], thetaq[4], ptq[4];
+ Bool_t quenched[4];
+ Double_t zInitial[4], wjtKick[4];
+ Int_t nGluon[4];
+ Int_t qPdg[4];
+ Int_t imo, kst, pdg;
+//
+// Sore information about Primary partons
+//
+// j =
+// 0, 1 partons from hard scattering
+// 2, 3 partons from initial state radiation
+//
+ for (Int_t i = 2; i <= 7; i++) {
+ Int_t j = 0;
+ // Skip gluons that participate in hard scattering
+ if (i == 4 || i == 5) continue;
+ // Gluons from hard Scattering
+ if (i == 6 || i == 7) {
+ 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];
+ } else {
+ // Gluons from initial state radiation
+ //
+ // Obtain 4-momentum vector from difference between original parton and parton after gluon
+ // radiation. Energy is calculated independently because initial state radition does not
+ // conserve strictly momentum and energy for each partonic system independently.
+ //
+ // Not very clean. Should be improved !
+ //
+ //
+ j = i;
+ pxq[j] = fPyjets->P[0][i] - fPyjets->P[0][i+2];
+ pyq[j] = fPyjets->P[1][i] - fPyjets->P[1][i+2];
+ pzq[j] = fPyjets->P[2][i] - fPyjets->P[2][i+2];
+ mq[j] = fPyjets->P[4][i];
+ eq[j] = TMath::Sqrt(pxq[j] * pxq[j] + pyq[j] * pyq[j] + pzq[j] * pzq[j] + mq[j] * mq[j]);
+ }
+//
+// Calculate some kinematic variables
+//
+ yq[j] = 0.5 * TMath::Log((eq[j] + pzq[j] + 1.e-14) / (eq[j] - pzq[j] + 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]);
+ qPdg[j] = fPyjets->K[1][i];
+ }
+
+ Double_t int0[4];
+ Double_t int1[4];
- 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;
-
+ fGlauber->GetI0I1ForPythia(4, phiq, int0, int1, 15.);
+
+ for (Int_t j = 0; j < 4; j++) {
//
- // Fix z
+ // Quench only central jets and with E > 10.
//
- Float_t z = 0.2;
- Float_t eppzOld = e + pz;
- Float_t empzOld = e - pz;
+ Int_t itype = (qPdg[j] == 21) ? 2 : 1;
+ Double_t eloss = fQuenchingWeights->GetELossRandomKFast(itype, int0[j], int1[j], eq[j]);
- //
- // Kinematics of the original parton
- //
+ if (TMath::Abs(yq[j]) > 2.5 || eq[j] < 10.) {
+ zInitial[j] = 0.;
+ } else {
+ if (eq[j] > 40. && TMath::Abs(yq[j]) < 0.5) {
+ icall ++;
+ eMean += eloss;
+ }
+ //
+ // Extra pt
+ Double_t l = fQuenchingWeights->CalcLk(int0[j], int1[j]);
+ wjtKick[j] = TMath::Sqrt(l * fQuenchingWeights->CalcQk(int0[j], int1[j]));
+ //
+ // Fractional energy loss
+ zInitial[j] = eloss / eq[j];
+ //
+ // Avoid complete loss
+ //
+ if (zInitial[j] == 1.) zInitial[j] = 0.95;
+ //
+ // Some debug printing
- 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;
+
+ printf("Initial parton # %3d, Type %3d Energy %10.3f Phi %10.3f Length %10.3f Loss %10.3f Kick %10.3f Mean: %10.3f %10.3f\n",
+ j, itype, eq[j], phiq[j], l, eloss, wjtKick[j], eMean / Float_t(icall+1), yq[j]);
+
+// zInitial[j] = 0.8;
+// while (zInitial[j] >= 0.95) zInitial[j] = gRandom->Exp(0.2);
+ }
- }
-
- //
- // Gluons
- //
+ quenched[j] = (zInitial[j] > 0.01);
+ } // primary partons
- 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.)
- {
+ Double_t pNew[1000][4];
+ Int_t kNew[1000];
+ Int_t icount = 0;
+//
+// System Loop
+ for (Int_t isys = 0; isys < 4; isys++) {
+// Skip to next system if not quenched.
+ if (!quenched[isys]) continue;
+
+ nGluon[isys] = 1 + Int_t(zInitial[isys] / (1. - zInitial[isys]));
+ if (nGluon[isys] > 6) nGluon[isys] = 6;
+ zInitial[isys] = 1. - TMath::Power(1. - zInitial[isys], 1./Double_t(nGluon[isys]));
+ wjtKick[isys] = wjtKick[isys] / TMath::Sqrt(Double_t(nGluon[isys]));
- //
- // Bring gluon back to mass shell via momentum scaling
- // (momentum will not be conserved, but energy)
- //
- // 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.;
-*/
- }
- }
- 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
- //
-
-
- 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;
+ Int_t igMin = -1;
+ Int_t igMax = -1;
+ Double_t pg[4] = {0., 0., 0., 0.};
- jmin = in - 1;
- ish = 1;
+//
+// Loop on radiation events
+
+ for (Int_t iglu = 0; iglu < nGluon[isys]; iglu++) {
+ 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];
+ Int_t imom = imo % 1000;
+ if ((isys == 0 || isys == 1) && ((imom != (isys + 7)))) continue;
+ if ((isys == 2 || isys == 3) && ((imom != (isys + 1)))) 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 = isys;
+
+ p0[index][0] += px;
+ p0[index][1] += py;
+ p0[index][2] += pz;
+ p0[index][3] += e;
+
+ klast[index] = i;
+
+//
+// Fractional energy loss
+ Double_t z = zInitial[index];
+
+
+// Don't fully quench radiated gluons
+//
+ if (imo > 1000) {
+// This small factor makes sure that the gluons are not too close in phase space to avoid recombination
+//
- 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;
- }
+ z = 0.02;
+ }
+// printf("z: %d %f\n", imo, z);
+
- 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];
+//
+
+ //
+ //
+ // 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);
+ //
+ // 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 (z < zmax) {
+ 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);
+ }
+ } else {
+ // If pT is to small (probably a leading massive particle) we scale only the energy
+ // This can cause negative masses of the radiated gluon
+ // Let's hope for the best ...
+ jtNew = jt;
+ eNew = TMath::Sqrt(plNew * plNew + mt2);
+
+ }
+ //
+ // 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("4-Momentum: %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 !\n");
+ // Here we have to put the gluon back to mass shell
+ // This will lead to a small energy imbalance
+ p2[isys][4] = 0.;
+ p2[isys][3] = TMath::Sqrt(p2[isys][0] * p2[isys][0] + p2[isys][1] * p2[isys][1] + p2[isys][2] * p2[isys][2]);
+ break;
+ } else {
+ p2[isys][4] = 0.;
+ break;
+ }
+ }
+ /*
+ 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 (while)
+
+// 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];
}
- } // 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.
+ // Add the gluons
//
- 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];
-
- Float_t pst = p2[i][4] / 2.;
-
- Float_t cost = 2. * gRandom->Rndm() - 1.;
- Float_t sint = TMath::Sqrt(1. - cost * cost);
- Float_t phi = 2. * TMath::Pi() * gRandom->Rndm();
-
- 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);
-
- 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.;
+ Int_t ish = 0;
+ Int_t iGlu;
+ if (!quenched[isys]) continue;
+//
+// Last parton from shower i
+ Int_t in = klast[isys];
+//
+// 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 (isys == 1 && klast[1] > klast[0]) in += ish;
+//
+// Starting index
- 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.;
+// jmin = in - 1;
+// How many additional gluons will be generated
+ ish = 1;
+ if (p2[isys][4] > 0.05) ish = 2;
+//
+// Position of gluons
+ iGlu = numpart;
+ if (iglu == 0) igMin = iGlu;
+ igMax = iGlu;
+ numpart += ish;
+ (fPyjets->N) += ish;
- 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);
-
+ if (ish == 1) {
+ fPyjets->P[0][iGlu] = p2[isys][0];
+ fPyjets->P[1][iGlu] = p2[isys][1];
+ fPyjets->P[2][iGlu] = p2[isys][2];
+ fPyjets->P[3][iGlu] = p2[isys][3];
+ fPyjets->P[4][iGlu] = p2[isys][4];
+
+ fPyjets->K[0][iGlu] = 1;
+ if (iglu == nGluon[isys] - 1) fPyjets->K[0][iGlu] = 1;
+ fPyjets->K[1][iGlu] = 21;
+ fPyjets->K[2][iGlu] = fPyjets->K[2][in] + 1000;
+ fPyjets->K[3][iGlu] = -1;
+ fPyjets->K[4][iGlu] = -1;
+
+ pg[0] += p2[isys][0];
+ pg[1] += p2[isys][1];
+ pg[2] += p2[isys][2];
+ pg[3] += p2[isys][3];
+ } else {
+ //
+ // Split gluon in rest frame.
+ //
+ Double_t bx = p2[isys][0] / p2[isys][3];
+ Double_t by = p2[isys][1] / p2[isys][3];
+ Double_t bz = p2[isys][2] / p2[isys][3];
+ Double_t pst = p2[isys][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] = 1 ;
+ fPyjets->K[1][iGlu] = 21;
+ fPyjets->K[2][iGlu] = fPyjets->K[2][in] + 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] = 1;
+ if (iglu == nGluon[isys] - 1) fPyjets->K[0][iGlu+1] = 1;
+ fPyjets->K[1][iGlu+1] = 21;
+ fPyjets->K[2][iGlu+1] = fPyjets->K[2][in] + 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);
+ }
+/*
+ for (Int_t ig = iGlu; ig < iGlu+ish; ig++) {
+ Double_t px, py, pz;
+ px = fPyjets->P[0][ig];
+ py = fPyjets->P[1][ig];
+ pz = fPyjets->P[2][ig];
+ TVector3 v(px, py, pz);
+ v.RotateZ(-phiq[isys]);
+ v.RotateY(-thetaq[isys]);
+ Double_t pxs = v.X(); Double_t pys = v.Y(); Double_t pzs = v.Z();
+ Double_t r = AliPythiaRndm::GetPythiaRandom()->Rndm();
+ Double_t jtKick = 0.3 * TMath::Sqrt(-TMath::Log(r));
+ if (ish == 2) jtKick = wjtKick[i] * TMath::Sqrt(-TMath::Log(r)) / TMath::Sqrt(2.);
+ Double_t phiKick = 2. * TMath::Pi() * AliPythiaRndm::GetPythiaRandom()->Rndm();
+ pxs += jtKick * TMath::Cos(phiKick);
+ pys += jtKick * TMath::Sin(phiKick);
+ TVector3 w(pxs, pys, pzs);
+ w.RotateY(thetaq[isys]);
+ w.RotateZ(phiq[isys]);
+ fPyjets->P[0][ig] = w.X();
+ fPyjets->P[1][ig] = w.Y();
+ fPyjets->P[2][ig] = w.Z();
+ fPyjets->P[2][ig] = w.Mag();
+ }
+*/
+ } // kGluon
+
+
+ // 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);
+// Fatal("Quench()", "4-Momentum non-conservation");
+ }
+
+ } // end quenching loop (systems)
+// 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;
}
- } // end adding gluons
+ }
+// this->Pylist(1);
} // end quench
-