Updates on TRD PID Response
[u/mrichter/AliRoot.git] / STEER / STEERBase / AliTRDPIDResponse.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 // 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;
33 class AliTRDPIDParams;
34 class AliTRDPIDReference;
35 class AliTRDPIDResponse : public TObject {
36   public:
37     enum ETRDPIDResponseStatus {
38       kIsOwner = BIT(14)
39     };
40     enum ETRDPIDResponseDef {
41       kNlayer = 6
42       ,kNPBins = 6
43     };
44     enum ETRDPIDMethod {
45       kNN   = 0,
46       kLQ2D = 1,
47       kLQ1D = 2
48     };
49     enum ETRDNslices {
50       kNslicesLQ1D = 1,
51       kNslicesLQ2D = 2,
52       kNslicesNN = 7
53     };
54     AliTRDPIDResponse();
55     AliTRDPIDResponse(const AliTRDPIDResponse &ref);
56     AliTRDPIDResponse& operator=(const AliTRDPIDResponse &ref);
57     ~AliTRDPIDResponse();
58     
59     ETRDPIDMethod     GetPIDmethod() const { return fPIDmethod;}
60     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;
61     inline ETRDNslices  GetNumberOfSlices() const;
62     
63     Bool_t    IsOwner() const {return TestBit(kIsOwner);}
64     
65     void      SetOwner();
66     void      SetPIDmethod(ETRDPIDMethod m) {fPIDmethod=m;}
67     void      SetGainNormalisationFactor(Double_t gainFactor) { fGainNormalisationFactor = gainFactor; }
68     void      SetPIDParams(const AliTRDPIDParams * params) { fkPIDParams = params; }
69
70     Bool_t    Load(const Char_t *filename = NULL, const Char_t *refName = "RefTRDLQ1D");
71     Bool_t    Load(const AliTRDPIDReference *ref) { fkPIDReference = ref; return kTRUE; }
72   
73     Bool_t    IdentifiedAsElectron(Int_t nTracklets, const Double_t *like, Double_t p, Double_t level) const;
74   
75   private:
76     Bool_t    CookdEdx(Int_t nSlice, const Double_t * const in, Double_t *out) const;
77     Double_t  GetProbabilitySingleLayer(Int_t species, Double_t plocal, Double_t dEdx) const;
78     
79     const AliTRDPIDReference *fkPIDReference;   // PID References
80     const AliTRDPIDParams *fkPIDParams;         // PID Params
81     Double_t  fGainNormalisationFactor;         // Gain normalisation factor
82     ETRDPIDMethod   fPIDmethod;                 // PID method selector
83       
84   
85   ClassDef(AliTRDPIDResponse, 3)    // Tool for TRD PID
86 };
87
88 AliTRDPIDResponse::ETRDNslices AliTRDPIDResponse::GetNumberOfSlices() const {
89   // Get the current number of slices
90   ETRDNslices slices = kNslicesLQ1D;
91   switch(fPIDmethod){
92     case kLQ1D: slices = kNslicesLQ1D; break;
93     case kLQ2D: slices = kNslicesLQ2D; break;
94     case kNN:   slices = kNslicesNN; break;
95   };
96   return slices;
97 }
98 #endif