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 // QA class of primary vertex study for Heavy Flavor electrons
  17 // this has functionality to reject electrons from primary vertex
  18 // and check primary vertex characteristics
  19 //
  20
  21 #ifndef ALIHFEPRIVTX_H
  22 #define ALIHFEPRIVTX_H
  23
  24 #ifndef ROOT_TObject
  25 //#include <TObject.h>
  26 #endif
  27
  28 class TH1F;
  29 class TH1I;
  30 class TH2F;
  31 class TString;
  32 class AliESDEvent;
  33 class AliESDtrack;
  34 class AliStack;
  35
  36 //________________________________________________________________
  37 class AliHFEpriVtx : public TObject {
  38
  39         public:
  40                 AliHFEpriVtx();
  41                 AliHFEpriVtx(const AliHFEpriVtx &p); // copy constructor
  42                 AliHFEpriVtx &operator=(const AliHFEpriVtx &); // assignment operator
  43                 virtual ~AliHFEpriVtx();
  44
  45                 void CreateHistograms(TString hnopt=""); // create histograms
  46                 void Init();
  47                 void SetEvent(AliESDEvent * const ESD){fESD1=ESD;}; // set ESD pointer
  48                 void SetStack(AliStack * const stack){fStack=stack;} // set stack pointer
  49                 void CountNtracks(Int_t sourcePart, Int_t recpid, Double_t recprob); // count number of tracks passed certain cut
  50                 void FillNtracks(); // fill counted number of tracks
  51                 void CountPriVxtElecContributor(AliESDtrack *ESDelectron, Int_t sourcePart, Int_t recpid, Double_t recprob);
  52                 void GetNPriVxtContributor();
  53                 void FillNprimVtxContributor() const;
  54                 Double_t GetDistanceFromRecalVertexXY(AliESDtrack *ESDelectron);
  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 AliHists{
  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                         AliHists()
  76                         : fNtracks()
  77                         , fNprimVtxContributor()
  78                         , fPtElec()
  79                         , fPtElecContributor()
  80                         , fNtrackCount(0)
  81                         , fNprimVtxContributorCount(0)
  82                         {
  83                           // default constructor
  84                         }
  85
  86                         AliHists(const AliHists & p)
  87                         : fNtracks(p.fNtracks)
  88                         , fNprimVtxContributor(p.fNprimVtxContributor)
  89                         , fPtElec(p.fPtElec)
  90                         , fPtElecContributor(p.fPtElecContributor)
  91                         , fNtrackCount(p.fNtrackCount)
  92                         , fNprimVtxContributorCount(p.fNprimVtxContributorCount)
  93                         {
  94                           // copy constructor
  95                         }
  96                         AliHists &operator=(const AliHists &)
  97                         {
  98                           // assignment operator, not yet implemented
  99                           return *this;
 100                         }
 101                 };
 102                 AliHists fPrimVtx[10]; // define structure of histograms
 103
 104                 Int_t fNtrackswoPid; //  number of track counted
 105                 TH1F *fHNtrackswoPid; // histogram to fill number of track counted
 106                 TH1I *fNESDprimVtxContributor; // histogram to fill number of primary vertex contributor for given event
 107                 TH1I *fNESDprimVtxIndices; // histogram to fill number of primary vertex indices for given event
 108                 TH2F *fDiffDCAvsPt; // histogram to fill DCA difference as a function of pT
 109                 TH2F *fDiffDCAvsNt; // histogram to fill DCA difference as a function of pT
 110
 111
 112         ClassDef(AliHFEpriVtx,0);
 113 };
 114
 115 #endif