PWG3/hfe/AliHFEpriVtx.h

   1 /**************************************************************************
   2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  * Author: The ALICE Off-line Project.                                    *
   5  * Contributors are mentioned in the code where appropriate.              *
   6  *                                                                        *
   7  * Permission to use, copy, modify and distribute this software and its   *
   8  * documentation strictly for non-commercial purposes is hereby granted   *
   9  * without fee, provided that the above copyright notice appears in all   *
  10  * copies and that both the copyright notice and this permission notice   *
  11  * appear in the supporting documentation. The authors make no claims     *
  12  * about the suitability of this software for any purpose. It is          *
  13  * provided "as is" without express or implied warranty.                  *
  14  **************************************************************************/
  15
  16 /**************************************************************************
  17  *                                                                        *
  18  * QA class of primary vertex study for Heavy Flavor electrons            *
  19  *                                                                        *
  20  **************************************************************************/
  21
  22 #ifndef _ALIHFEPRIVTX_H_
  23 #define _ALIHFEPRIVTX_H_
  24
  25 #ifndef ROOT_TObject
  26 #include <TObject.h>
  27 #endif
  28
  29 class TH1F;
  30 class TH1I;
  31 class TH2F;
  32 class TString;
  33 class AliESDEvent;
  34 class AliESDtrack;
  35 class AliStack;
  36
  37 //________________________________________________________________
  38 class AliHFEpriVtx : public TObject {
  39
  40         public:
  41                 AliHFEpriVtx();
  42                 AliHFEpriVtx(const AliHFEpriVtx &p); // copy constructor
  43                 AliHFEpriVtx &operator=(const AliHFEpriVtx &); // assignment operator
  44                 virtual ~AliHFEpriVtx();
  45
  46                 void CreateHistograms(TString hnopt=""); // create histograms
  47                 void Init();
  48                 void SetEvent(AliESDEvent* ESD){fESD1=ESD;}; // set ESD pointer
  49                 void SetStack(AliStack* stack){fStack=stack;} // set stack pointer
  50                 void CountNtracks(Int_t sourcePart, Int_t recpid, Double_t recprob); // count number of tracks passed certain cut
  51                 void FillNtracks(); // fill counted number of tracks
  52                 void CountPriVxtElecContributor(AliESDtrack *ESDelectron, Int_t sourcePart, Int_t recpid, Double_t recprob);
  53                 void GetNPriVxtContributor();
  54                 void FillNprimVtxContributor();
  55
  56                 Int_t GetMCPID(AliESDtrack *track); // return mc pid
  57
  58         private:
  59
  60                 AliESDEvent* fESD1; // ESD event
  61                 AliStack* fStack; // MC Stack
  62
  63                 TString fkSourceLabel[10]; // storing source label
  64
  65                 enum kSources {kAll, kDirectCharm, kDirectBeauty, kBeautyCharm, kGamma, kPi0, kElse, kBeautyGamma, kBeautyPi0, kBeautyElse};
  66
  67                 struct hists{
  68                         TH1F *fNtracks; // histogram to fill number of counted tracks for different sources
  69                         TH1F *fNprimVtxContributor; // histogram to fill number of tracks contributing primary vertex
  70                         TH1F *fPtElec; // histogram to fill pt of electron tracks
  71                         TH1F *fPtElecContributor; // histogram to fill pt of electron tracks contributing primary vertex
  72                         Int_t fNtrackCount; // number of counted track
  73                         Int_t fNprimVtxContributorCount; // number of tracks contributing primary vertex
  74                 };
  75
  76                 hists fPrimVtx[10]; //
  77
  78                 Int_t fNtrackswoPid; //  number of track counted
  79                 TH1F *fHNtrackswoPid; // histogram to fill number of track counted
  80                 TH1I *fNESDprimVtxContributor; // histogram to fill number of primary vertex contributor for given event
  81                 TH1I *fNESDprimVtxIndices; // histogram to fill number of primary vertex indices for given event
  82                 TH2F *fDiffDCAvsPt; // histogram to fill DCA difference as a function of pT
  83                 TH2F *fDiffDCAvsNt; // histogram to fill DCA difference as a function of pT
  84
  85
  86         ClassDef(AliHFEpriVtx,0);
  87 };
  88
  89 #endif