Update of hfe code

[u/mrichter/AliRoot.git] / PWGHF / hfe / AliHFEpidObject.h

diff --git a/PWGHF/hfe/AliHFEpidObject.h b/PWGHF/hfe/AliHFEpidObject.h
index 6eff82ad97dcb85bb36f85aa2b7302d86ccfe2ff..8dbfc14a063a24424031df2aeadb6bd723a1d750 100644 (file)
--- a/PWGHF/hfe/AliHFEpidObject.h
+++ b/PWGHF/hfe/AliHFEpidObject.h
@@ -35,6 +35,7 @@ class AliHFEpidObject{
       fAnalysisType(kESDanalysis),
       fAbInitioPID(-1),
       fCentrality(99),
+      fMultiplicity(0),
       fIsPbPb(kFALSE)         // Default: pp
       {
       }
@@ -43,6 +44,7 @@ class AliHFEpidObject{
       fAnalysisType(ref.fAnalysisType),
       fAbInitioPID(ref.fAbInitioPID),
       fCentrality(ref.fCentrality),
+      fMultiplicity(ref.fMultiplicity),
       fIsPbPb(ref.fIsPbPb)
       {
       }
@@ -54,12 +56,14 @@ class AliHFEpidObject{
     void SetAnalysisType(AnalysisType_t type) { fAnalysisType = type; }
     void SetAbInitioPID(Int_t abInitioPID) { fAbInitioPID = abInitioPID; }
     void SetCentrality(Int_t centrality) { fCentrality = centrality; }
+    void SetMulitplicity(Double_t mult) { fMultiplicity = mult; }
     void SetPbPb() { fIsPbPb = kTRUE; }
     void SetPP() { fIsPbPb = kFALSE; }
 
     const AliVTrack *GetRecTrack() const { return fkRecTrack; }
     Int_t GetAbInitioPID() const { return fAbInitioPID; }
     Int_t GetCentrality() const { return fCentrality; }
+    Double_t GetMultiplicity() const { return fMultiplicity; }
     Bool_t IsAODanalysis() const { return fAnalysisType == static_cast<UChar_t>(kAODanalysis); }
     Bool_t IsESDanalysis() const { return fAnalysisType == static_cast<UChar_t>(kESDanalysis); }
     Bool_t IsPbPb() const { return fIsPbPb; }
@@ -69,6 +73,7 @@ class AliHFEpidObject{
     UChar_t fAnalysisType;              // Analysis Mode (ESD or AOD)
     Int_t fAbInitioPID;                 // AbInitio PID
     Int_t fCentrality;                  // Centrality Information
+    Double_t fMultiplicity;             // Multiplicity information
     Bool_t fIsPbPb;                     // Collision type
 };
 #endif