Charge + VZERO side selection added

[u/mrichter/AliRoot.git] / PWGLF / SPECTRA / PiKaPr / TestAOD / AliAnalysisTaskSpectraAllChAOD.h

#ifndef ALIANALYSISTASKSPECTRAALLCHAOD_H\r
#define ALIANALYSISTASKSPECTRAALLCHAOD_H\r
\r
/*  See cxx source for full Copyright notice */\r
\r
//-------------------------------------------------------------------------\r
//                      AliAnalysisTaskSpectraAllChAOD\r
//\r
//\r
//\r
//\r
// Author: Leonardo Milano, CERN\r
//-------------------------------------------------------------------------\r
\r
class AliAODEvent;\r
class AliSpectraAODTrackCuts;\r
class AliSpectraAODEventCuts;\r
class AliHelperPID;\r
\r
#include "AliAnalysisTaskSE.h"\r
\r
class AliAnalysisTaskSpectraAllChAOD : public AliAnalysisTaskSE\r
{\r
 public:\r
  // constructors\r
 AliAnalysisTaskSpectraAllChAOD() : AliAnalysisTaskSE(),\r
    fAOD(0x0),\r
    fTrackCuts(0x0),\r
    fEventCuts(0x0),\r
    fHelperPID(0x0),\r
    fIsMC(0),\r
    fCharge(0),\r
    fVZEROside(0),\r
    fOutput(0x0),\r
    fnCentBins(20),\r
    fnQvecBins(50)\r
      {}\r
  AliAnalysisTaskSpectraAllChAOD(const char *name);\r
  virtual ~AliAnalysisTaskSpectraAllChAOD() {\r
    Printf("calling detructor of AliAnalysisTaskSpectraAllChAOD - To be implemented");\r
  }\r
  \r
  void SetIsMC(Bool_t isMC = kFALSE)    {fIsMC = isMC; };\r
  Bool_t GetIsMC()           const           { return fIsMC;};\r
 \r
  void SetCharge(Int_t charge = 0)    {fCharge = charge; };\r
  Int_t GetCharge()           const           { return fCharge;};\r
  \r
  virtual void   UserCreateOutputObjects();\r
  virtual void   UserExec(Option_t *option);\r
  virtual void   Terminate(Option_t *);\r
  \r
  AliSpectraAODTrackCuts      * GetTrackCuts()         {  return fTrackCuts; }\r
  AliSpectraAODEventCuts      * GetEventCuts()         {  return fEventCuts; }\r
  AliHelperPID                   * GetHelperPID()          { return fHelperPID; }\r
  TList                          * GetOutputList()         { return fOutput; }\r
  \r
  void SetTrackCuts(AliSpectraAODTrackCuts * tc)       { fTrackCuts = tc; }\r
  void SetEventCuts(AliSpectraAODEventCuts * vc)       { fEventCuts = vc; }\r
  void SetHelperPID(AliHelperPID* pid)                     { fHelperPID = pid; }\r
  void SetnCentBins(Int_t val)                             { fnCentBins = val; }\r
  void SetnQvecBins(Int_t val)                             { fnQvecBins = val; }\r
  \r
 private:\r
  \r
  AliAODEvent                   * fAOD;           //! AOD object\r
  AliSpectraAODTrackCuts      * fTrackCuts;     // Track Cuts\r
  AliSpectraAODEventCuts      * fEventCuts;     // Event Cuts\r
  AliHelperPID                   * fHelperPID;      // points to class for PID\r
  Bool_t                          fIsMC;           // true if processing MC\r
  Int_t                            fCharge;        // charge to be selected\r
  Int_t                            fVZEROside;    // 0: VZERO-A 1: VZERO-C\r
  TList                          * fOutput;        // output list\r
  Int_t                            fnCentBins;        // number of bins for the centrality axis\r
  Int_t                            fnQvecBins;        // number of bins for the q vector axis\r
  AliAnalysisTaskSpectraAllChAOD(const AliAnalysisTaskSpectraAllChAOD&);\r
  AliAnalysisTaskSpectraAllChAOD& operator=(const AliAnalysisTaskSpectraAllChAOD&);\r
  \r
  ClassDef(AliAnalysisTaskSpectraAllChAOD, 4);\r
};\r
\r
#endif\r