[ALIROOT-8649] Restore porting the event weight to the pythia header

alisw · Apr 7, 2021 · 71bc53c · 71bc53c
1 parent a048450
commit 71bc53c
Show file tree

Hide file tree

Showing 6 changed files with 30 additions and 1 deletion.
diff --git a/PYTHIA6/AliPythia6/AliGenPythiaPlus.cxx b/PYTHIA6/AliPythia6/AliGenPythiaPlus.cxx
@@ -1322,6 +1322,9 @@ void AliGenPythiaPlus::MakeHeader()
     ((AliGenPythiaEventHeader*) fHeader)->SetPtHard(fPythia->GetPtHard());
     ((AliGenPythiaEventHeader*) fHeader)->SetXsection(fPythia->GetXSection());
 
+// Store Event weight
+    ((AliGenPythiaEventHeader*) fHeader)->SetEventWeight(fPythia->GetEventWeight());
+
 //
 //  Pass header
 //

diff --git a/PYTHIA6/AliPythia6/AliPythia6.cxx b/PYTHIA6/AliPythia6/AliPythia6.cxx
@@ -1660,6 +1660,11 @@ Float_t AliPythia6::GetPtHard()
     return GetVINT(47);
 }
 
+Float_t AliPythia6::GetEventWeight()
+{
+	return GetVINT(97);
+}
+
 Int_t AliPythia6::ProcessCode()
 {
     // Get the subprocess code

diff --git a/PYTHIA6/AliPythia6/AliPythia6.h b/PYTHIA6/AliPythia6/AliPythia6.h
@@ -59,6 +59,12 @@ class AliPythia6 : public TPythia6, public AliPythiaBase
     // Common Getters
     virtual void    GetXandQ(Float_t& x1, Float_t& x2, Float_t& q);
     virtual Float_t GetXSection();
+
+    /**
+     * @brief Get the event weight from the underlying pythia process 
+     * @return Event weight 
+     */
+    virtual Float_t GetEventWeight();
     virtual Int_t   ProcessCode();
     virtual Float_t GetPtHard();
     virtual Int_t   GetNMPI();

diff --git a/PYTHIA6/AliPythia6/AliPythiaBase.h b/PYTHIA6/AliPythia6/AliPythiaBase.h
@@ -71,6 +71,12 @@ class AliPythiaBase : public AliRndm
     virtual Float_t GetPtHard() {return -1.;}
     virtual Int_t   GetNMPI() {return -1;}
     virtual Int_t   GetNSuperpositions() { return 1; }
+
+    /**
+     * @brief Get the event weight from the underlying pythia process 
+     * @return Event weight 
+     */
+    virtual Float_t GetEventWeight() { return 1; }
     //
     //
     virtual void SetDecayTable() {;}

diff --git a/PYTHIA8/AliPythia8/AliPythia8.cxx b/PYTHIA8/AliPythia8/AliPythia8.cxx
@@ -892,7 +892,10 @@ Float_t AliPythia8::GetPtHard()
     return Pythia8()->info.pTHat();
 }
 
-
+Float_t AliPythia8::GetEventWeight()
+{
+    return Pythia8()->info.weight();
+}
 
 
 AliPythia8& AliPythia8::operator=(const  AliPythia8& rhs)

diff --git a/PYTHIA8/AliPythia8/AliPythia8.h b/PYTHIA8/AliPythia8/AliPythia8.h
@@ -84,6 +84,12 @@ class AliPythia8 :public AliTPythia8, public AliPythiaBase
     virtual void    GetXandQ(Float_t& x1, Float_t& x2, Float_t& q);
     virtual Float_t GetXSection();
     virtual Float_t GetPtHard();
+
+    /**
+     * @brief Get the event weight from the underlying pythia process 
+     * @return Event weight 
+     */
+    virtual Float_t GetEventWeight();
     virtual Int_t GetNMPI() { return fLastNMPI; }
     virtual Int_t GetNSuperpositions() { return fLastNSuperposition; }