Add fast merging option (Diego)

[u/mrichter/AliRoot.git] / PWG3 / hfe / AliHFEtpcPIDqa.h

diff --git a/PWG3/hfe/AliHFEtpcPIDqa.h b/PWG3/hfe/AliHFEtpcPIDqa.h
index dc3c909536e69b94b9410f404c0b3ee1aeabc8dc..7e631a143966e3b63845b311740b973adeee32cb 100644 (file)
--- a/PWG3/hfe/AliHFEtpcPIDqa.h
+++ b/PWG3/hfe/AliHFEtpcPIDqa.h
@@ -24,11 +24,14 @@
 #include "AliHFEdetPIDqa.h"
 #endif
 
+#ifndef ALIHFEPIDBASE_H
+#include "AliHFEpidBase.h"
+#endif
+
+class TBrowser;
 class TH2;
 class AliHFEcollection;
-class AliHFEpidObject;
-class AliESDtrack;
-class AliAODTrack;
+class AliVParticle;
 
 class AliHFEtpcPIDqa : public AliHFEdetPIDqa{
   public:
@@ -39,19 +42,24 @@ class AliHFEtpcPIDqa : public AliHFEdetPIDqa{
     ~AliHFEtpcPIDqa();
     void Copy(TObject &o) const;
     virtual Long64_t Merge(TCollection *col);
+    virtual void Browse(TBrowser *b);
+    virtual Bool_t IsFolder() const { return kTRUE; };
   
     virtual void Initialize();
-    virtual void ProcessTrack(AliHFEpidObject *track, AliHFEdetPIDqa::EStep_t step);
+    virtual void ProcessTrack(const AliHFEpidObject *track, AliHFEdetPIDqa::EStep_t step);
+
+    void SetBrowseCentrality(Int_t browseCentrality) { browseCentrality < 11  && browseCentrality >= -1 ? fBrowseCentrality = browseCentrality : -1;} // *MENU*
 
     AliHFEcollection *GetHistograms() const { return fHistos; }
-    TH2 *MakeSpectrumdEdx(AliHFEdetPIDqa::EStep_t step, Int_t species = -1);
-    TH2 *MakeSpectrumNSigma(AliHFEdetPIDqa::EStep_t step, Int_t species = -1);
+    TH2 *MakeSpectrumdEdx(AliHFEdetPIDqa::EStep_t step, Int_t species = -1, Int_t centralityClass = -1);
+    TH2 *MakeSpectrumNSigma(AliHFEdetPIDqa::EStep_t step, Int_t species = -1, Int_t centralityClass = -1);
 
   protected:
-    void ProcessESDtrack(const AliESDtrack *track, AliHFEdetPIDqa::EStep_t step, Int_t species, Float_t centrality);
-    void ProcessAODtrack(const AliAODTrack *track, AliHFEdetPIDqa::EStep_t step, Int_t species, Float_t centrality);
+    Double_t GetTPCsignal(const AliVParticle *track, AliHFEpidObject::AnalysisType_t anatype);
+
   private:
     AliHFEcollection *fHistos;        // Container for Histograms
+    Int_t fBrowseCentrality;          // Centrality Class for Browser
 
     ClassDef(AliHFEtpcPIDqa, 1);
 };