PWG3/hfe/AliHFEpidTRD.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 #ifndef ALIHFEPIDTRD_H
  16 #define ALIHFEPIDTRD_H
  17
  18  #ifndef ALIHFEPIDBASE_H
  19  #include "AliHFEpidBase.h"
  20  #endif
  21
  22 class AliESDtrack;
  23 class AliVParticle;
  24 class TList;
  25 class TH2F;
  26
  27 class AliHFEpidTRD : public AliHFEpidBase{
  28   public:
  29     typedef enum{
  30       kLQ = 0,
  31       kNN = 1
  32     } PIDMethodTRD_t;
  33     enum{
  34       kThreshParams = 24
  35     };
  36     enum{
  37       kHistTRDSigV1 = 0,
  38       kHistTRDSigV2 = 1,
  39       kHistOverallSpecies = 2
  40     };
  41     AliHFEpidTRD(const Char_t *name);
  42     AliHFEpidTRD(const AliHFEpidTRD &ref);
  43     AliHFEpidTRD& operator=(const AliHFEpidTRD &ref);
  44     virtual ~AliHFEpidTRD();
  45
  46     virtual Bool_t InitializePID();
  47     virtual Int_t IsSelected(AliVParticle *track);
  48     virtual Bool_t HasQAhistos() const { return kTRUE; };
  49
  50     Double_t GetTRDSignalV1(AliESDtrack *track);
  51     Double_t GetTRDSignalV2(AliESDtrack *track);
  52
  53     void SetPIDMethod(PIDMethodTRD_t method) { fPIDMethod = method; };
  54   protected:
  55     void Copy(TObject &ref) const;
  56     Double_t GetTRDthresholds(Double_t electronEff, Double_t p);
  57     Int_t GetMCpid(AliESDtrack *track);
  58     void InitParameters();
  59     virtual void AddQAhistograms(TList *l);
  60     void GetParameters(Double_t electronEff, Double_t *parameters);
  61
  62     void FillHistogramsTRDSignalV1(Double_t signal, Double_t p, Int_t species);
  63     void FillHistogramsTRDSignalV2(Double_t signal, Double_t p, Int_t species);
  64   private:
  65     static const Double_t fgkVerySmall;                       // Check for 0
  66     PIDMethodTRD_t fPIDMethod;                              // PID Method: 2D Likelihood or Neural Network
  67     Double_t fThreshParams[kThreshParams];                  // Threshold parametrisation
  68     TList *fContainer;                                      // QA  Histogram Container
  69   ClassDef(AliHFEpidTRD, 1)     // TRD electron ID class
  70 };
  71
  72 #endif