coverity fix
[u/mrichter/AliRoot.git] / 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