Adding a flag for the shuffling + kinematic cuts for the MC analysis

[u/mrichter/AliRoot.git] / PWG2 / EBYE / AliAnalysisTaskBF.h

#ifndef ALIANALYSISTASKBF_CXX\r
#define ALIANALYSISTASKBF_CXX\r
\r
// Analysis task for the BF code\r
// Authors: Panos Cristakoglou@cern.ch\r
\r
class TList;\r
class TH1F;\r
class TH2F;\r
\r
class AliBalance;\r
class AliESDtrackCuts;\r
\r
#include "AliAnalysisTaskSE.h"\r
#include "AliBalance.h"\r
\r
\r
class AliAnalysisTaskBF : public AliAnalysisTaskSE {\r
 public:\r
  AliAnalysisTaskBF(const char *name = "AliAnalysisTaskBF");\r
  virtual ~AliAnalysisTaskBF(); \r
  \r
  \r
  virtual void   UserCreateOutputObjects();\r
  virtual void   UserExec(Option_t *option);\r
  virtual void   FinishTaskOutput();\r
  virtual void   Terminate(Option_t *);\r
\r
  void SetAnalysisObject(AliBalance *const analysis) {\r
    fBalance         = analysis;\r
    }\r
  void SetShufflingObject(AliBalance *const analysisShuffled) {\r
    fRunShuffling = kTRUE;\r
    fShuffledBalance = analysisShuffled;\r
  }\r
  void SetAnalysisCutObject(AliESDtrackCuts *const trackCuts) {\r
    fESDtrackCuts = trackCuts;}\r
  void SetVertexDiamond(Double_t vx, Double_t vy, Double_t vz) {\r
    fVxMax = vx;\r
    fVyMax = vy;\r
    fVzMax = vz;\r
  }\r
  void SetAODtrackCutBit(Int_t bit){\r
    nAODtrackCutBit = bit;\r
  }\r
  void SetKinematicsCutsAOD(Double_t ptmin, Double_t ptmax, Double_t etamin, Double_t etamax){\r
    fPtMin  = ptmin;\r
    fPtMax  = ptmax;\r
    fEtaMin = etamin;\r
    fEtaMax = etamax;\r
\r
  }\r
  void SetExtraDCACutsAOD(Double_t DCAxy, Double_t DCAz){\r
    fDCAxyCut  = DCAxy;\r
    fDCAzCut = DCAz;\r
  }\r
  void SetKinematicsCutsMC(Double_t ptmin, Double_t ptmax,\r
                           Double_t etamin, Double_t etamax){\r
    fPtMin  = ptmin; fPtMax  = ptmax;\r
    fEtaMin = etamin; fEtaMax = etamax;\r
  }\r
\r
  //Centrality\r
  void SetCentralityEstimator(const char* centralityEstimator) {fCentralityEstimator = centralityEstimator;}\r
  const char* GetCentralityEstimator(void)                     {return fCentralityEstimator;}\r
  void SetCentralityPercentileRange(Double_t min, Double_t max) { \r
    fCentralityPercentileMin=min;\r
    fCentralityPercentileMax=max;\r
  }\r
\r
  void UseOfflineTrigger() {fUseOfflineTrigger = kTRUE;}\r
\r
 private:\r
  AliBalance *fBalance; //BF object\r
  Bool_t fRunShuffling;//run shuffling or not\r
  AliBalance *fShuffledBalance; //BF object (shuffled)\r
  TList *fList; //fList object\r
  TList *fListBF; //fList object\r
  TList *fListBFS; //fList object\r
\r
  TH1F *fHistEventStats; //event stats\r
  TH1F *fHistTrackStats; //Track filter bit stats\r
  TH1F *fHistVx; //x coordinate of the primary vertex\r
  TH1F *fHistVy; //y coordinate of the primary vertex\r
  TH1F *fHistVz; //z coordinate of the primary vertex\r
\r
  TH2F *fHistClus;\r
  TH2F *fHistDCA;\r
  TH1F *fHistChi2;\r
  TH1F *fHistPt;\r
  TH1F *fHistEta;\r
  TH1F *fHistPhi;\r
  TH2F *fHistV0M;\r
\r
  AliESDtrackCuts *fESDtrackCuts; //ESD track cuts\r
\r
  TString fCentralityEstimator;      //"V0M","TRK","TKL","ZDC","FMD"\r
  Double_t fCentralityPercentileMin;\r
  Double_t fCentralityPercentileMax;\r
\r
  Bool_t fUseOfflineTrigger;//Usage of the offline trigger selection\r
\r
  Double_t fVxMax;//vxmax\r
  Double_t fVyMax;//vymax\r
  Double_t fVzMax;//vzmax\r
\r
  Int_t nAODtrackCutBit;//track cut bit from track selection (only used for AODs)\r
\r
  Double_t fPtMin;//only used for AODs\r
  Double_t fPtMax;//only used for AODs\r
  Double_t fEtaMin;//only used for AODs\r
  Double_t fEtaMax;//only used for AODs\r
\r
  Double_t fDCAxyCut;//only used for AODs\r
  Double_t fDCAzCut;//only used for AODs\r
\r
  AliAnalysisTaskBF(const AliAnalysisTaskBF&); // not implemented\r
  AliAnalysisTaskBF& operator=(const AliAnalysisTaskBF&); // not implemented\r
  \r
  ClassDef(AliAnalysisTaskBF, 5); // example of analysis\r
};\r
\r
#endif\r