ffb1ee30 |
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 | // PID Response class for the TRD detector |
17 | // Based on 1D Likelihood approach |
18 | // For further information see implementation file |
19 | // |
20 | #ifndef ALITRDPIDRESPONSE_H |
21 | #define ALITRDPIDRESPONSE_H |
22 | |
23 | #ifndef ROOT_TObject |
24 | #include <TObject.h> |
25 | #endif |
26 | |
27 | #ifndef ALIPID_H |
28 | #include "AliPID.h" |
29 | #endif |
30 | |
31 | class TObjArray; |
32 | class AliVTrack; |
db0e2c5f |
33 | class AliTRDPIDResponseObject; |
34 | |
ffb1ee30 |
35 | class AliTRDPIDResponse : public TObject { |
e0de37e9 |
36 | public: |
37 | enum ETRDPIDResponseStatus { |
9006fe9c |
38 | kIsOwner = BIT(14) |
e0de37e9 |
39 | }; |
40 | enum ETRDPIDResponseDef { |
db0e2c5f |
41 | kNlayer = 6 |
42 | ,kNPBins = 6 |
e0de37e9 |
43 | }; |
44 | enum ETRDPIDMethod { |
45 | kNN = 0, |
46 | kLQ2D = 1, |
47 | kLQ1D = 2 |
48 | }; |
db0e2c5f |
49 | enum ETRDPIDNMethod { |
50 | kNMethod=3 |
51 | }; |
e0de37e9 |
52 | enum ETRDNslices { |
53 | kNslicesLQ1D = 1, |
54 | kNslicesLQ2D = 2, |
55 | kNslicesNN = 7 |
56 | }; |
57 | AliTRDPIDResponse(); |
58 | AliTRDPIDResponse(const AliTRDPIDResponse &ref); |
51a0ce25 |
59 | AliTRDPIDResponse& operator=(const AliTRDPIDResponse &ref); |
e0de37e9 |
60 | ~AliTRDPIDResponse(); |
61 | |
62 | ETRDPIDMethod GetPIDmethod() const { return fPIDmethod;} |
51a0ce25 |
63 | Bool_t GetResponse(Int_t n, const Double_t * const dedx, const Float_t * const p, Double_t prob[AliPID::kSPECIES], Bool_t kNorm=kTRUE) const; |
e0de37e9 |
64 | inline ETRDNslices GetNumberOfSlices() const; |
65 | |
66 | Bool_t IsOwner() const {return TestBit(kIsOwner);} |
67 | |
68 | void SetOwner(); |
69 | void SetPIDmethod(ETRDPIDMethod m) {fPIDmethod=m;} |
70 | void SetGainNormalisationFactor(Double_t gainFactor) { fGainNormalisationFactor = gainFactor; } |
ffb1ee30 |
71 | |
db0e2c5f |
72 | Bool_t SetPIDResponseObject(const AliTRDPIDResponseObject * obj) { fkPIDResponseObject = obj; return kTRUE;} |
73 | |
74 | Bool_t Load(const Char_t *filename = NULL); |
e0de37e9 |
75 | |
ea235c90 |
76 | Bool_t IdentifiedAsElectron(Int_t nTracklets, const Double_t *like, Double_t p, Double_t level) const; |
77 | |
e0de37e9 |
78 | private: |
51a0ce25 |
79 | Bool_t CookdEdx(Int_t nSlice, const Double_t * const in, Double_t *out) const; |
db0e2c5f |
80 | Double_t GetProbabilitySingleLayer(Int_t species, Double_t plocal, Double_t *dEdx) const; |
e0de37e9 |
81 | |
db0e2c5f |
82 | const AliTRDPIDResponseObject *fkPIDResponseObject; // PID References and Params |
ea235c90 |
83 | Double_t fGainNormalisationFactor; // Gain normalisation factor |
84 | ETRDPIDMethod fPIDmethod; // PID method selector |
85 | |
e0de37e9 |
86 | |
87 | ClassDef(AliTRDPIDResponse, 3) // Tool for TRD PID |
ffb1ee30 |
88 | }; |
ffb1ee30 |
89 | |
e0de37e9 |
90 | AliTRDPIDResponse::ETRDNslices AliTRDPIDResponse::GetNumberOfSlices() const { |
91 | // Get the current number of slices |
92 | ETRDNslices slices = kNslicesLQ1D; |
93 | switch(fPIDmethod){ |
94 | case kLQ1D: slices = kNslicesLQ1D; break; |
95 | case kLQ2D: slices = kNslicesLQ2D; break; |
96 | case kNN: slices = kNslicesNN; break; |
97 | }; |
98 | return slices; |
99 | } |
100 | #endif |
db0e2c5f |
101 | |