Place the config and root file at the right place

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

diff --git a/PWGHF/hfe/AliHFEpidObject.h b/PWGHF/hfe/AliHFEpidObject.h
index a0c91ed4f7a921d54e8f54e0c400d5a603d7a699..65dc3025fea5b6cccb1c332e81f2b3a1b0413eab 100644 (file)
--- a/PWGHF/hfe/AliHFEpidObject.h
+++ b/PWGHF/hfe/AliHFEpidObject.h
@@ -21,11 +21,12 @@
 
 #include <Rtypes.h>
 
+class AliVTrack;
 class AliVParticle;
 
 class AliHFEpidObject{
   public:
-    enum AnalysisType_t { 
+    enum AnalysisType_t{ 
       kESDanalysis,
       kAODanalysis
     };
@@ -34,7 +35,9 @@ class AliHFEpidObject{
       fAnalysisType(kESDanalysis),
       fAbInitioPID(-1),
       fCentrality(99),
-      fIsPbPb(kFALSE)         // Default: pp
+      fMultiplicity(0),
+      fIsPbPb(kFALSE),         // Default: pp
+      fIspPb(kFALSE)         // Default: pp
       {
       }
     AliHFEpidObject(const AliHFEpidObject &ref):
@@ -42,33 +45,41 @@ class AliHFEpidObject{
       fAnalysisType(ref.fAnalysisType),
       fAbInitioPID(ref.fAbInitioPID),
       fCentrality(ref.fCentrality),
-      fIsPbPb(ref.fIsPbPb)
+      fMultiplicity(ref.fMultiplicity),
+      fIsPbPb(ref.fIsPbPb),
+      fIspPb(ref.fIspPb)
       {
       }
     AliHFEpidObject &operator=(const AliHFEpidObject &ref);
     ~AliHFEpidObject(){};
 
-    void SetRecTrack(const AliVParticle * recTrack) {fkRecTrack = recTrack; }
+    void SetRecTrack(const AliVTrack * recTrack) {fkRecTrack = recTrack; }
     void SetMCTrack(const AliVParticle * mcTrack);
     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 SetpPb() { fIsPbPb = kFALSE; fIspPb=kTRUE; }
     void SetPP() { fIsPbPb = kFALSE; }
 
-    const AliVParticle *GetRecTrack() const { return fkRecTrack; }
+    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; }
+    Bool_t IspPb() const { return fIspPb; }
 
   private:
-    const AliVParticle *fkRecTrack;     // Reconstructed track
+    const AliVTrack *fkRecTrack;        // Reconstructed track
     UChar_t fAnalysisType;              // Analysis Mode (ESD or AOD)
     Int_t fAbInitioPID;                 // AbInitio PID
     Int_t fCentrality;                  // Centrality Information
-    Bool_t fIsPbPb;                     // Collision type
+    Double_t fMultiplicity;             // Multiplicity information
+    Bool_t fIsPbPb;                     // Collision type PbPb
+    Bool_t fIspPb;                      // Collision type pPb
 };
 #endif