Updates in order to enable the '2D' PID for the TRD developed by Daniel Lohner.
[u/mrichter/AliRoot.git] / STEER / STEERBase / AliTRDPIDResponse.h
CommitLineData
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
31class TObjArray;
32class AliVTrack;
db0e2c5f 33class AliTRDPIDResponseObject;
34
ffb1ee30 35class 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 90AliTRDPIDResponse::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