PWG3/vertexingHF/AliCFTaskForDStarAnalysis.h

   1 #ifndef ALICFTASKFORDSTARANALYSIS_H
   2 #define ALICFTASKFORDSTARANALYSIS_H
   3 /**************************************************************************
   4  * Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
   5  *                                                                        *
   6  * Author: The ALICE Off-line Project.                                    *
   7  * Contributors are mentioned in the code where appropriate.              *
   8  *                                                                        *
   9  * Permission to use, copy, modify and distribute this software and its   *
  10  * documentation strictly for non-commercial purposes is hereby granted   *
  11  * without fee, provided that the above copyright notice appears in all   *
  12  * copies and that both the copyright notice and this permission notice   *
  13  * appear in the supporting documentation. The authors make no claims     *
  14  * about the suitability of this software for any purpose. It is          *
  15  * provided "as is" without express or implied warranty.                  *
  16  **************************************************************************/
  17
  18 //-----------------------------------------------------------------------
  19 // Class for D* corrections --
  20
  21 #include "AliAnalysisTaskSE.h"
  22
  23 class TH1I;
  24 class TParticle ;
  25 class TFile ;
  26 class TClonesArray ;
  27 class AliCFManager;
  28 class AliAODRecoDecay;
  29 class AliAODRecoDecayHF2Prong;
  30 class AliAODMCParticle;
  31 class THnSparse;
  32
  33 class AliCFTaskForDStarAnalysis : public AliAnalysisTaskSE {
  34   public:
  35
  36   enum {
  37     kStepGenerated       = 0,
  38     kStepAcceptance      = 1,
  39     kStepVertex          = 2,
  40     kStepRefit           = 3,
  41     kStepReconstructed   = 4,
  42     kStepRecoAcceptance  = 5,
  43     kStepRecoITSClusters = 6,
  44     kStepRecoCuts        = 7
  45   };
  46
  47   AliCFTaskForDStarAnalysis();
  48   AliCFTaskForDStarAnalysis(const Char_t* name);
  49   AliCFTaskForDStarAnalysis& operator= (const AliCFTaskForDStarAnalysis& c);
  50   AliCFTaskForDStarAnalysis(const AliCFTaskForDStarAnalysis& c);
  51   virtual ~AliCFTaskForDStarAnalysis();
  52
  53   // ANALYSIS FRAMEWORK STUFF to loop on data and fill output objects
  54   void     UserCreateOutputObjects();
  55   void     UserExec(Option_t *option);
  56   void     Terminate(Option_t *);
  57
  58  // UNFOLDING
  59   void     SetCorrelationMatrix(THnSparse* h) {fCorrelation=h;}
  60   void     SetAcceptanceUnf(Bool_t AcceptanceUnf) {fAcceptanceUnf = AcceptanceUnf;}
  61   Bool_t   GetAcceptanceUnf() const {return fAcceptanceUnf;}
  62
  63   // CORRECTION FRAMEWORK
  64   void           SetCFManager(AliCFManager* io) {fCFManager = io;}   // global correction manager
  65   AliCFManager * GetCFManager()                 {return fCFManager;} // get corr manager
  66
  67   Bool_t   GetDStarMCParticle(AliAODMCParticle* mcPart, TClonesArray* mcArray, Double_t* vectorMC)const;
  68   Bool_t   EvaluateIfD0toKpi(AliAODMCParticle* neutralDaugh, TClonesArray* mcArray, Double_t* VectorD0)const;
  69   // for the D0
  70   void     SetMinITSClusters(Int_t minITSClusters) {fMinITSClusters = minITSClusters;}
  71   Int_t    GetMinITSClusters() const {return fMinITSClusters;}
  72   // for the soft pion
  73   void     SetMinITSClustersSoft(Int_t minITSClustersSoft) {fMinITSClustersSoft = minITSClustersSoft;}
  74   Int_t    GetMinITSClustersSoft() const {return fMinITSClustersSoft;}
  75
  76  protected:
  77
  78   AliCFManager   *fCFManager;   //  pointer to the CF manager
  79   TH1I *fHistEventsProcessed;   //! simple histo for monitoring the number of events processed
  80   THnSparse* fCorrelation;      //  response matrix for unfolding
  81   Int_t fCountRecoDStarSel;     //  Reco particle found that satisfy cuts in D* selection
  82   Int_t fEvents;                //  n. of events
  83   Int_t fMinITSClusters;        //  min n. of ITS clusters for RecoDecay
  84   Int_t fMinITSClustersSoft;    //  min n. of ITS clusters for RecoDecay soft pion
  85   Bool_t fAcceptanceUnf;        //  flag for unfolding before or after cuts.
  86
  87   ClassDef(AliCFTaskForDStarAnalysis,3); // class for D* corrections
  88 };
  89
  90 #endif