New quenching algorithm.
[u/mrichter/AliRoot.git] / PYTHIA6 / AliPythia.cxx
index 49cc5e9..0a8d044 100644 (file)
  * provided "as is" without express or implied warranty.                  *
  **************************************************************************/
 
-/*
-$Log$
-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 &);
 
 //_____________________________________________________________________________
 
@@ -128,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;
@@ -215,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:
 //
@@ -249,13 +195,13 @@ void AliPythia::ProcInit(Process_t process, Float_t energy, StrucFunc_t strucfun
     case kPyDirectGamma:
        SetMSEL(10);
        break;
-    case kPyCharmPbMNR:
-    case kPyD0PbMNR:
+    case kPyCharmPbPbMNR:
+    case kPyD0PbPbMNR:
       // 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 kCTEQ_4L PDFs.
+      // (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.
 
@@ -278,7 +224,7 @@ void AliPythia::ProcInit(Process_t process, Float_t energy, StrucFunc_t strucfun
       SetMSTP(32,2);
       SetPARP(34,1.0);
 
-      // Intrinsic <kT^2>
+      // Intrinsic <kT>
       SetMSTP(91,1);
       SetPARP(91,1.304);
       SetPARP(93,6.52);
@@ -287,7 +233,83 @@ void AliPythia::ProcInit(Process_t process, Float_t energy, StrucFunc_t strucfun
       SetPMAS(4,1,1.2);
 
       break;
-    case kPyBeautyPbMNR:
+    case kPyCharmpPbMNR:
+    case kPyD0pPbMNR:
+      // 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:
+      // 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
       // between with the NLO calculation by Mangano, Nason, Ridolfi for the
       // b-bbar single inclusive and double differential distributions.
@@ -317,7 +339,7 @@ void AliPythia::ProcInit(Process_t process, Float_t energy, StrucFunc_t strucfun
       SetPARP(67,1.0);
       SetPARP(71,1.0);
 
-      // Intrinsic <kT^2>
+      // Intrinsic <kT>
       SetMSTP(91,1);
       SetPARP(91,2.035);
       SetPARP(93,10.17);
@@ -326,6 +348,84 @@ void AliPythia::ProcInit(Process_t process, Float_t energy, StrucFunc_t strucfun
       SetPMAS(5,1,4.75);
 
       break;
+    case kPyBeautypPbMNR:
+      // Tuning of Pythia parameters aimed to get a resonable agreement
+      // between with the NLO calculation by Mangano, Nason, Ridolfi for the
+      // b-bbar 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.75GeV. Example in ConfigBeautyPPR.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);
+      SetPARP(67,1.0);
+      SetPARP(71,1.0);
+
+      // Intrinsic <kT>
+      SetMSTP(91,1);
+      SetPARP(91,1.60);
+      SetPARP(93,8.00);
+
+      // Set b-quark mass
+      SetPMAS(5,1,4.75);
+
+      break;
+    case kPyBeautyppMNR:
+      // Tuning of Pythia parameters aimed to get a resonable agreement
+      // between with the NLO calculation by Mangano, Nason, Ridolfi for the
+      // b-bbar 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.75GeV. Example in ConfigBeautyPPR.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);
+      SetPARP(67,1.0);
+      SetPARP(71,1.0);
+
+      // Intrinsic <kT>
+      SetMSTP(91,1);
+      SetPARP(91,1.);
+      SetPARP(93,5.);
+
+      // Set b-quark mass
+      SetPMAS(5,1,4.75);
+
+      break;
     }
 //
 //  Initialize PYTHIA
@@ -375,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++;
@@ -429,34 +527,547 @@ 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);
+    
+    for (Int_t iglu = 0; iglu < kGluons; iglu++) {
+       for (Int_t k = 0; k < 4; k++)
+       {
+           p0[0][k] = 0.; p0[1][k] = 0.;
+           p1[0][k] = 0.; p1[1][k] = 0.;
+           p2[0][k] = 0.; p2[1][k] = 0.;
+       }
+       
+       Int_t nq[2] = {0, 0};
+       
+       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 != 7 && imo != 8 && imo != 1007 && imo != 1008) 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 == 1007 || imo == 1008) {
+               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 (!quenched[index]) continue;
+           //
+           //
+           //      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 mt  = TMath::Sqrt(mt2);
+           Double_t zmin = 0.;
+           Double_t zmax = 1.;     
+           //
+           // z**2 mt**2 - pt**2 + pt'**2 > 0
+            // z**2 mt**2 + mt'**2 + m**2 - pt**2
+            // z**2 mt**2 + (1-z)**2 mt**2 + m**2 - pt**2
+            // mt**2(z**2 + 1 + z**2  - 2z) + m**2 - pt**2
+            // mt**2(2z**2 + 1 - 2z) + m**2 - pt**2 > 0
+            // mt**2(2z**2 + 1 - 2z) + 2 m**2 - mt**2 > 0
+            // mt**2(2z**2 - 2z) + 2 m**2 > 0
+            // z mt**2 (1 - z) -  m**2 < 0
+           // z**2 - z + 1/4 > 1/4 - m**2/mt**2
+            // (z-1/2)**2 > 1/4 - m**2/mt**2
+            // |z-1/2| > sqrt(1/4 - m**2/mt**2)
+            //
+            // m/mt < 1/2
+            // mt   > 2m
+           //
+           if (mt < 2. * m) {
+               printf("No phase space for quenching !: mt (%e) < 2 m (%e) \n", mt, m);
+               p1[index][0] += px;
+               p1[index][1] += py;
+               p1[index][2] += pz;
+               p1[index][3] += e;
+               continue;
+           } else {
+               zmin = 0.5 - TMath::Sqrt(0.25 - m * m / mt2);
+               if (z < zmin) {
+                   printf("No phase space for quenching ??: z (%e) < zmin (%e) \n", z, zmin);
+//                 z = zmin * 1.01;
+
+                   p1[index][0] += px;
+                   p1[index][1] += py;
+                   p1[index][2] += pz;
+                   p1[index][3] += e;
+                   continue;
+
+               }
+           }
+           //
+           // 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 eNew0    = 0.5 * (eppzNew + empzNew);
+           Double_t pzNew0   = 0.5 * (eppzNew - empzNew);
+           
+           Double_t ptNew;
+           //
+           // 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);
+               ptNew = 0;
+           } else {
+               ptNew    = TMath::Sqrt(mt2New - m * m);
+           }
+
+           
+           //
+           //     Calculate new px, py
+           //
+           Double_t pxNew0   = ptNew / jt * pxs;
+           Double_t pyNew0   = ptNew / jt * pys;       
+/*
+           Double_t dpx = pxs - pxNew0;
+           Double_t dpy = pys - pyNew0;
+           Double_t dpz = pl  - pzNew0;
+           Double_t de  = e   - eNew0;
+           Double_t dmass2 = de * de  - dpx * dpx - dpy * dpy - dpz * dpz;
+*/
+           //
+           //      Rotate back
+           //  
+           TVector3 w(pxNew0, pyNew0, pzNew0);
+           w.RotateY(thetaq[index]);
+           w.RotateZ(phiq[index]);
+           pxNew0 = w.X(); pyNew0 = w.Y(); pzNew0 = w.Z();
+           
+           
+           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;
+           nq[index]++;
+           
+       }
+       
+       //
+       // Gluons
+       // 
+       
+       for (Int_t k = 0; k < 2; k++) 
+       {
+           //
+           // Check if there was phase-space for quenching
+           //
+           if (nq[k] == 0) quenched[k] = kFALSE;
+           
+           if (!quenched[k]) continue;
+           
+           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.) {
+               p2[k][4] = TMath::Sqrt(p2[k][4]);
+           } else {
+               printf("Warning negative mass squared in system %d %f ! \n", k, zInitial[k]);
+               printf("Kinematics %10.3e %10.3e %10.3e %10.3e %10.3e \n", p2[k][0], p2[k][1], p2[k][2], p2[k][3], p2[k][4]);
+               if (p2[k][4] < -0.1) Fatal("Boost", "Negative mass squared !");
+               p2[k][4] = 0.;
+           }
+           //
+           // 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]);
+           */
+       }
+       
+       //
+       // 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);
+           }
+       } // end adding gluons
+       //
+       // 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");
+       }
 
+    } // end quenchin loop
+    // Clean-up
+    for (Int_t i = 0; i < numpart; i++)
+    {
+       imo =  fPyjets->K[2][i];
+       if (imo > 1000) fPyjets->K[2][i] -= 1000;
+    }
+       
+} // end quench