Add quenching information to header.

[u/mrichter/AliRoot.git] / PYTHIA6 / AliGenPythiaEventHeader.cxx

diff --git a/PYTHIA6/AliGenPythiaEventHeader.cxx b/PYTHIA6/AliGenPythiaEventHeader.cxx
index 27e140d17d14012d006b63905b2733cc3b8c5ef6..8ad8a6515d468cc7eed755bff1f6b33eed10aa39 100644 (file)
--- a/PYTHIA6/AliGenPythiaEventHeader.cxx
+++ b/PYTHIA6/AliGenPythiaEventHeader.cxx
@@ -23,13 +23,22 @@ ClassImp(AliGenPythiaEventHeader);
 AliGenPythiaEventHeader::AliGenPythiaEventHeader()
 {
 // Default Constructor
-    fNJets = 0;
+    fNJets   = 0;
+    fNUQJets = 0;
+    for (Int_t i = 0; i < 4; i++) fZquench[i] = 0.;
+    fXJet = -1.;
+    fYJet = -1.;
 }
 
 AliGenPythiaEventHeader::AliGenPythiaEventHeader(const char* name):AliGenEventHeader(name)
 {
 // Constructor
-    fNJets = 0;
+    fNJets   = 0;
+    fNUQJets = 0;
+    for (Int_t i = 0; i < 4; i++) fZquench[i] = 0.;
+    fXJet = -1.;
+    fYJet = -1.;
+
 }
 
 void AliGenPythiaEventHeader::AddJet(Float_t px, Float_t py, Float_t pz, Float_t e)
@@ -48,6 +57,38 @@ void AliGenPythiaEventHeader::AddJet(Float_t px, Float_t py, Float_t pz, Float_t
     }
 }
 
+void AliGenPythiaEventHeader::AddUQJet(Float_t px, Float_t py, Float_t pz, Float_t e)
+{
+//
+//  Add a jet 
+//
+    if (fNUQJets < 10) {
+       fUQJets[0][fNUQJets] = px;
+       fUQJets[1][fNUQJets] = py;
+       fUQJets[2][fNUQJets] = pz;
+       fUQJets[3][fNUQJets] = e;
+       fNUQJets++;
+    } else {
+       printf("\nWarning: More than 10 jets triggered !!\n");
+    }
+}
+
+void AliGenPythiaEventHeader::SetZQuench(Double_t z[4])
+{
+    //
+    // Set quenching fraction
+    //
+    for (Int_t i = 0; i < 4; i++) fZquench[i] = z[i];
+}
+
+void AliGenPythiaEventHeader::GetZQuench(Double_t z[4])
+{
+    //
+    // Get quenching fraction
+    //
+    for (Int_t i = 0; i < 4; i++) z[i] = fZquench[i];
+}
+
 void AliGenPythiaEventHeader::TriggerJet(Int_t i, Float_t p[4])
 {
 //
@@ -62,3 +103,28 @@ void AliGenPythiaEventHeader::TriggerJet(Int_t i, Float_t p[4])
        p[3] = fJets[3][i];
     }
 }
+
+void AliGenPythiaEventHeader::UQJet(Int_t i, Float_t p[4])
+{
+//
+// Give back jet #i
+//
+    if (i >= fNUQJets) {
+       printf("\nWarning: Unquenched Jets, index out of Range!!\n");
+    } else {
+       p[0] = fUQJets[0][i];
+       p[1] = fUQJets[1][i];
+       p[2] = fUQJets[2][i];
+       p[3] = fUQJets[3][i];
+    }
+}
+
+void AliGenPythiaEventHeader::SetXYJet(Double_t x, Double_t y)
+{
+
+//
+//  Add jet production point
+//
+    fXJet = x; 
+    fYJet = y; 
+}