Merge branch 'master' of https://git.cern.ch/reps/AliRoot

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

/**************************************************************************\r
* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *\r
*                                                                        *\r
* Author: The ALICE Off-line Project.                                    *\r
* Contributors are mentioned in the code where appropriate.              *\r
*                                                                        *\r
* Permission to use, copy, modify and distribute this software and its   *\r
* documentation strictly for non-commercial purposes is hereby granted   *\r
* without fee, provided that the above copyright notice appears in all   *\r
* copies and that both the copyright notice and this permission notice   *\r
* appear in the supporting documentation. The authors make no claims     *\r
* about the suitability of this software for any purpose. It is          *\r
* provided "as is" without express or implied warranty.                  *\r
**************************************************************************/\r
//\r
// Flow task class for the ALICE HFE group\r
//\r
//\r
#ifndef ALIANALYSISTASKHFEQA_H\r
#define ALIANALYSISTASKHFEQA_H\r
\r
\r
#ifndef ALIANALYSISTASKSE_H\r
#include "AliAnalysisTaskSE.h"\r
#endif\r
\r
#ifndef ROOT_TString\r
#include <TString.h>\r
#endif\r
\r
#ifndef ROOT_TBits\r
#include <TBits.h>\r
#endif\r
\r
\r
class TList;\r
class AliHFEcuts;\r
class AliHFEpid;\r
class AliHFEpidQAmanager;\r
class AliAODMCHeader;\r
class TClonesArray;\r
class TH1F;\r
\r
\r
class AliAnalysisTaskHFEQA: public AliAnalysisTaskSE {\r
public:\r
\r
  typedef enum{\r
    kpp = 0,\r
    kpPb = 1,\r
    kPbPb = 2\r
  } ESystem_t;\r
  \r
\r
  AliAnalysisTaskHFEQA();\r
  AliAnalysisTaskHFEQA(const char *name);\r
  AliAnalysisTaskHFEQA(const AliAnalysisTaskHFEQA &ref);\r
  AliAnalysisTaskHFEQA& operator=(const AliAnalysisTaskHFEQA &ref);\r
  virtual void Copy(TObject &o) const;\r
  virtual ~AliAnalysisTaskHFEQA();\r
  \r
  virtual void  UserExec(Option_t */*option*/);\r
  virtual void  UserCreateOutputObjects();\r
\r
  void SetDoTPConly(Bool_t tpconlydo)   { fTPConlydo = tpconlydo; };\r
  void SetDoTRDonly(Bool_t trdonlydo)   { fTRDonlydo = trdonlydo; };\r
  void SetDoTOFTPC(Bool_t toftpcdo)   { fTOFTPCdo = toftpcdo; };\r
  void SetDoTPCTRD(Bool_t tpctrddo)   { fTPCTRDdo = tpctrddo; };\r
  void SetDoTPCEMCal(Bool_t tpcemcaldo)   { fTPCEMCaldo = tpcemcaldo; };\r
  void SetAODAnalysis(Bool_t aodAnalysis)   { fAODAnalysis = aodAnalysis; };\r
  void SetCentralityEstimator(const char *estimator) { fCentralityEstimator = estimator; }\r
  void SetppAnalysis(){\r
    fCollisionSystem.SetBitNumber(kpPb, kFALSE); \r
    fCollisionSystem.SetBitNumber(kPbPb, kFALSE); \r
    fCollisionSystem.SetBitNumber(kpp, kTRUE); \r
  }\r
  void SetpPbAnalysis() {\r
    fCollisionSystem.SetBitNumber(kpp, kFALSE); \r
    fCollisionSystem.SetBitNumber(kPbPb, kFALSE); \r
    fCollisionSystem.SetBitNumber(kpPb, kTRUE); \r
  }\r
  void SetPbPbAnalysis() { \r
    fCollisionSystem.SetBitNumber(kpp, kFALSE); \r
    fCollisionSystem.SetBitNumber(kpPb, kFALSE); \r
    fCollisionSystem.SetBitNumber(kPbPb, kTRUE); \r
  };\r
  Bool_t Ispp() const { return fCollisionSystem.TestBitNumber(kpp); }\r
  Bool_t IsPbPb() const { return fCollisionSystem.TestBitNumber(kPbPb); }\r
  Bool_t IspPb() const { return fCollisionSystem.TestBitNumber(kpPb); }\r
  \r
  AliHFEpid *GetPIDTPConly() const { return fPIDTPConly; }\r
  AliHFEpid *GetPIDTRDonly() const { return fPIDTRDonly; }\r
  AliHFEpid *GetPIDTOFTPC() const { return fPIDTOFTPC; }\r
  AliHFEpid *GetPIDTPCTRD() const { return fPIDTPCTRD; }\r
  AliHFEpid *GetPIDTPCEMCal() const { return fPIDTPCEMCal; }\r
  AliHFEpidQAmanager *GetPIDQAManagerTRDonly() const { return fPIDqaTRDonly; }\r
  AliHFEpidQAmanager *GetPIDQAManagerTOFTPC() const { return fPIDqaTOFTPC; }\r
  AliHFEpidQAmanager *GetPIDQAManagerTPCTRD() const { return fPIDqaTPCTRD; }\r
  AliHFEpidQAmanager *GetPIDQAManagerTPCEMCal() const { return fPIDqaTPCEMCal; }\r
 \r
\r
  void SetHFECuts(AliHFEcuts * const cuts) { fHFECuts = cuts; };\r
 \r
 private:\r
  TList     *fListHist;         //! TH list\r
  Bool_t    fAODAnalysis;       // AOD analysis\r
  AliAODMCHeader *fAODMCHeader;         // ! MC info AOD\r
  TClonesArray *fAODArrayMCInfo;        // ! MC info particle AOD\r
 \r
  // Cuts for HFE\r
  AliHFEcuts *fHFECuts;                   // HFE cuts\r
  AliHFEpid  *fPIDTPConly;                // PID cuts \r
  AliHFEpid  *fPIDTRDonly;                // PID cuts \r
  AliHFEpid  *fPIDTOFTPC;                 // PID cuts TOF-TPC only\r
  AliHFEpid  *fPIDTPCTRD;                 // PID cuts TPC-TRD \r
  AliHFEpid  *fPIDTPCEMCal;                 // PID cuts TPC-EMCal \r
  AliHFEpidQAmanager *fPIDqaTRDonly;       // QA Manager TOF TPC\r
  AliHFEpidQAmanager *fPIDqaTOFTPC;       // QA Manager TOF TPC\r
  AliHFEpidQAmanager *fPIDqaTPCTRD;       // QA Manager TPC TRD\r
  AliHFEpidQAmanager *fPIDqaTPCEMCal;       // QA Manager TPC EMCal\r
  TString fCentralityEstimator;         // Centrality Estimator\r
  TBits fCollisionSystem;              // Collision System;\r
\r
  // Histo yields\r
  TH1F *fNbEvent;                      // Number of events\r
  TH1F *fTPConly;                      // TPC only electron yield\r
  TH1F *fTOFTPC;                      // TOF TPC  electron yield\r
  TH1F *fTPCTRD;                      // TPC TRD  electron yield\r
  TH1F *fTPCEMCal;                      // TPC EMCal electron yield\r
\r
  // Do PID or not\r
  Bool_t fTPConlydo;                   // Do TPC only PID\r
  Bool_t fTRDonlydo;                   // Do TRD only PID\r
  Bool_t fTOFTPCdo;                    // Do TOF TPC \r
  Bool_t fTPCTRDdo;                    // Do TPC TRD \r
  Bool_t fTPCEMCaldo;                    // Do TPC EMCal \r
\r
  \r
  // Debuging Cuts step by step all centrality together: pt, step (6)\r
  //THnSparseF *fTrackingCuts; //! Tracking Cuts\r
\r
  \r
  \r
  ClassDef(AliAnalysisTaskHFEQA, 3); // analysisclass\r
};\r
\r
#endif\r