PWGLF/SPECTRA/MultEvShape/AliAnalysisCentralCutESD.h

   1 #ifndef ALIANALYSISCENTRALCUTESD_H
   2 #define ALIANALYSISCENTRALCUTESD_H
   3
   4
   5 /*
   6  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved.
   7  * See cxx source for full Copyright notice
   8  * $Id$
   9  */
  10
  11 // ---------------------------------------------------
  12 //  ESD particle level cuts for azimuthal isotropic
  13 //  expansion in highly central collisions analysis
  14 //  author: Cristian Andrei
  15 //           acristian@niham.nipne.ro
  16 // ---------------------------------------------------
  17
  18
  19 #include <TString.h>
  20 #include <TPDGCode.h>
  21
  22 #include "AliAnalysisCuts.h"
  23
  24
  25 class TObject;
  26 class TList;
  27 class TF1;
  28
  29 class AliESDtrack;
  30
  31
  32 class AliAnalysisCentralCutESD: public AliAnalysisCuts {
  33 public:
  34     AliAnalysisCentralCutESD(const char *name="AliAnalysisCentralCutESD", const char *title="ESD_cuts");
  35     virtual ~AliAnalysisCentralCutESD();
  36
  37     Bool_t  IsSelected(TObject* obj);
  38     Bool_t  IsSelected(TList* /*list*/) {return kTRUE;}
  39
  40     void SetPartType(PDG_t type) {fReqPID = kTRUE; fPartType = type;}
  41     void SetPIDtype(TString type) {fPIDtype = type;}
  42     void SetPriorFunctions(Bool_t pfunc) {fPriorsFunc = pfunc;}
  43
  44     void SetReqIsCharged() {fReqCharge = kTRUE;}
  45
  46 private:
  47   AliAnalysisCentralCutESD(const AliAnalysisCentralCutESD &ref);
  48   AliAnalysisCentralCutESD& operator=(const AliAnalysisCentralCutESD &ref);
  49
  50     Bool_t fReqPID;  //kTRUE -> run the PID
  51
  52     Bool_t fReqCharge; //kTRUE -> only charged particles selected
  53
  54     PDG_t fPartType; //can be kPi*, kK* or kProton
  55
  56     TString fPIDtype; //PID method -> can be Custom or Bayesian
  57     Bool_t fPriorsFunc; // if kTRUE -> use function priors
  58
  59     Double_t fPartPriors[10]; //prior probabilities
  60     TF1  *fElectronFunction; //momentum dependence of the prior probs
  61     TF1  *fMuonFunction; //momentum dependence of the prior probs
  62     TF1  *fPionFunction; //momentum dependence of the prior probs
  63     TF1  *fKaonFunction; //momentum dependence of the prior probs
  64     TF1  *fProtonFunction; //momentum dependence of the prior probs
  65
  66     Bool_t IsA(AliESDtrack *track, PDG_t type);
  67
  68     Bool_t IsCharged  (AliESDtrack* const track) const;
  69
  70     Double_t GetPriors(Int_t i, Double_t P);
  71
  72     ClassDef(AliAnalysisCentralCutESD, 1);
  73 };
  74
  75 #endif