PWG3/vertexingHF/AliAnalysisTaskSEDStar.h

   1 #ifndef ALIANALYSISTASKSEDSTAR_H\r
   2 #define ALIANALYSISTASKSEDSTAR_H\r
   3 \r
   4 /* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *\r
   5  * See cxx source for full Copyright notice                               */\r
   6 \r
   7 //*************************************************************************\r
   8 // Class AliAnalysisTaskSEDStar\r
   9 // AliAnalysisTaskSE for D* candidates invariant mass histogram\r
  10 // and comparison to MC truth (kinematics stored in the AOD)\r
  11 // Authors: Y.wang, yifei@physi.uni-heidelberg.de\r
  12 //*************************************************************************\r
  13 \r
  14 #include <TROOT.h>\r
  15 #include <TSystem.h>\r
  16 #include <TH1F.h>\r
  17 \r
  18 #include "AliAnalysisTaskSE.h"\r
  19 #include "AliAnalysisVertexingHF.h"\r
  20 \r
  21 class AliAnalysisTaskSEDStar : public AliAnalysisTaskSE\r
  22 {\r
  23  public:\r
  24 \r
  25   AliAnalysisTaskSEDStar();\r
  26   AliAnalysisTaskSEDStar(const char *name);\r
  27   virtual ~AliAnalysisTaskSEDStar();\r
  28 \r
  29 \r
  30   // Implementation of interface methods\r
  31   virtual void UserCreateOutputObjects();\r
  32   virtual void Init();\r
  33   virtual void LocalInit() {Init();}\r
  34   virtual void UserExec(Option_t *option);\r
  35   virtual void Terminate(Option_t *option);\r
  36 \r
  37   void         SetReadMC(Bool_t readMC=kFALSE){fReadMC=readMC;}\r
  38   void         PIDon() {fPID=kTRUE;}\r
  39   void         PIDoff() {fPID=kFALSE;}\r
  40   Bool_t       GetPIDStatus() const {return fPID;}\r
  41   void         SetNSigmaTPC(Double_t nsigma) {fNSigma=nsigma;}\r
  42   Double_t     GetNSigmaTPC() const {return fNSigma;}\r
  43  private:\r
  44 \r
  45   AliAnalysisTaskSEDStar(const AliAnalysisTaskSEDStar &source);\r
  46   AliAnalysisTaskSEDStar& operator=(const AliAnalysisTaskSEDStar& source); \r
  47   TList   *fOutput; //! list send on output slot 0\r
  48 \r
  49   TH1F    *fhistMass;  //! output total invariant mass histogram - no MC truth\r
  50   TH1F    *fhistSgn;  //! output signal invariant mass histogram - use cuts\r
  51   TH1F    *fhistBkg;  //! output background invariant mass histogram - use cuts\r
  52   Bool_t   fReadMC;  // flag for MC array: kTRUE = read it, kFALSE = do not read it\r
  53   Bool_t   fPID;  //flag of TPCpid selection\r
  54   Double_t fNSigma;   //nsigma selection for TPCpid\r
  55 \r
  56   AliAnalysisVertexingHF *fVHF;  // Vertexer heavy flavour (used to pass the cuts)\r
  57   Bool_t SelectTPCPID(AliAODTrack *trk, Int_t pid, Double_t nsig);\r
  58 \r
  59   ClassDef(AliAnalysisTaskSEDStar,1); // AliAnalysisTaskSE for the MC association of heavy-flavour decay candidates\r
  60 };\r
  61 \r
  62 #endif\r
  63 \r