Update (Chiara)
[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliAnalysisTaskSEHFQA.h
index 938edae..746ad68 100644 (file)
@@ -4,6 +4,8 @@
 /* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
  * See cxx source for full Copyright notice                               */
 
+/* $Id$ */ 
+
 //*************************************************************************
 // Class AliAnalysisTaskSEHFQA
 // AliAnalysisTaskSE for HF quality assurance
@@ -38,10 +40,19 @@ class AliAnalysisTaskSEHFQA : public AliAnalysisTaskSE
 
   //setters
   void SetReadMC(Bool_t mcflag){fReadMC=mcflag;}
+  void SetSimpleMode(Bool_t flag){fSimpleMode=flag;}
+  void SetTrackOn(Bool_t trackon=kTRUE){fOnOff[0]=trackon;}
+  void SetPIDOn(Bool_t pidon=kTRUE){fOnOff[1]=pidon;}
+  void SetCentralityOn(Bool_t centron=kTRUE){fOnOff[2]=centron;}
+  void SetEvSelectionOn(Bool_t evselon=kTRUE){fOnOff[3]=evselon;}
 
   //getters
   AliRDHFCuts* GetCutObject() const {return fCuts;}
   DecChannel GetDecayChannel()const {return fDecayChannel;}
+  Bool_t GetTrackStatus() const {return fOnOff[0];}
+  Bool_t GetPIDStatus() const {return fOnOff[1];}
+  Bool_t GetCentralityStatus() const {return fOnOff[2];}
+  Bool_t GetEvSelStatus() const {return fOnOff[3];}
 
  private:
   AliAnalysisTaskSEHFQA(const AliAnalysisTaskSEHFQA &source);
@@ -52,13 +63,16 @@ class AliAnalysisTaskSEHFQA : public AliAnalysisTaskSE
  TList* fOutputTrack;      //! list sent on output slot 3
  TList* fOutputCounters;   //! list sent on output slot 5
  TList* fOutputCheckCentrality;   //! list sent on output slot 6
+ TList* fOutputEvSelection; //! list sent on output slot 7
  DecChannel fDecayChannel; //identify the decay channel
  AliRDHFCuts* fCuts;       // object containing cuts
  AliRDHFCuts::ECentrality fEstimator; //2nd estimator for centrality
  Bool_t fReadMC;           // flag to read MC
-
- ClassDef(AliAnalysisTaskSEHFQA,3); //AnalysisTaskSE for the quality assurance of HF in hadrons
+ Bool_t fSimpleMode;       // if true, don't do candidates (much faster in PbPb)
+ Bool_t fOnOff[4];         // on-off the QA on tracks (0), PID (1), centrality (2), event selection -- default is {kTRUE,kTRUE,kTRUE,kTRUE}
+ ClassDef(AliAnalysisTaskSEHFQA,6); //AnalysisTaskSE for the quality assurance of HF in hadrons
 
 };
 
 #endif
+