Updates in order to enable the '2D' PID for the TRD developed by Daniel Lohner.
[u/mrichter/AliRoot.git] / STEER / STEERBase / AliTRDPIDParams.cxx
CommitLineData
ce487a7f 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// Container for TRD Threshold parameters stored in the OADB
17//
18// Author: Markus Fasel <M.Fasel@gsi.de>
19//
20#include <TMath.h>
21#include <TSortedList.h>
22
23#include "AliLog.h"
24
25#include "AliTRDPIDParams.h"
26
27ClassImp(AliTRDPIDParams)
28
29const Double_t AliTRDPIDParams::kVerySmall = 1e-5;
30
31//____________________________________________________________
32AliTRDPIDParams::AliTRDPIDParams():
33 TNamed(),
34 fEntries(NULL)
35{
36 //
37 // Dummy constructor
38 //
39}
40
41//____________________________________________________________
42AliTRDPIDParams::AliTRDPIDParams(const char *name) :
43 TNamed(name, ""),
44 fEntries(NULL)
45{
46 //
47 // Default constructor
48 //
49 fEntries = new TSortedList;
50}
51
db0e2c5f 52//____________________________________________________________
53AliTRDPIDParams::AliTRDPIDParams(const AliTRDPIDParams &ref):
54TNamed(ref),
55fEntries(NULL)
56{
57 //
58 // Copy constructor
59 //
60
61 fEntries=(TSortedList*)ref.fEntries->Clone();
62}
63
64//____________________________________________________________
65AliTRDPIDParams::AliTRDPIDParams(const AliTRDPIDParams &ref):
66TNamed(ref),
67fEntries(NULL)
68{
69 //
70 // Copy constructor
71 //
72
73 fEntries=(TSortedList*)ref.fEntries->Clone();
74}
75
ce487a7f 76//____________________________________________________________
77AliTRDPIDParams::~AliTRDPIDParams(){
78 //
79 // Destructor
80 //
81 delete fEntries;
82}
83
84//____________________________________________________________
85Bool_t AliTRDPIDParams::GetThresholdParameters(Int_t ntracklets, Double_t efficiency, Double_t *params) const{
86 //
87 // Retrieve params
88 // Use IsEqual definition
89 //
90 AliTRDPIDThresholds test(ntracklets, efficiency);
91 TObject *result = fEntries->FindObject(&test);
92 if(!result){
93 AliDebug(1, Form("No threshold params found for %d tracklets and an electron efficiency of %f", ntracklets, efficiency));
94 return kFALSE;
95 }
96 AliTRDPIDThresholds *parResult = static_cast<AliTRDPIDThresholds *>(result);
97 AliDebug(1, Form("Threshold params found: NTracklets %d, Electron Efficiency %f", parResult->GetNTracklets(), parResult->GetElectronEfficiency()));
98 memcpy(params, parResult->GetThresholdParams(), sizeof(Double_t) * 4);
99 return kTRUE;
100}
101
102//____________________________________________________________
103void AliTRDPIDParams::SetThresholdParameters(Int_t ntracklets, Double_t effMin, Double_t effMax, Double_t *params){
104 //
105 // Store new Params in the Object
106 //
107 if(effMin > effMax){
108 AliError("Min. efficiency has to be >= max. efficiency");
109 return;
110 }
111 AliDebug(1, Form("Save Parameters for %d tracklets at and electron efficiency of [%f|%f]", ntracklets, effMin, effMax));
112 fEntries->Add(new AliTRDPIDThresholds(ntracklets, effMin, effMax, params));
113}
114
115//____________________________________________________________
116void AliTRDPIDParams::Print(Option_t *) const {
117 printf("Available thresholds:\n");
118 printf("_________________________________________\n");
119 TIter objects(fEntries);
120 AliTRDPIDThresholds *par;
121 while((par = dynamic_cast<AliTRDPIDThresholds *>(objects()))){
122 printf("Number of tracklets %d, Electron efficiency %f\n", par->GetNTracklets(), par->GetElectronEfficiency());
123 }
124}
125
126//____________________________________________________________
127AliTRDPIDParams::AliTRDPIDThresholds::AliTRDPIDThresholds():
128 TObject(),
129 fNTracklets(0)
130{
131 //
132 // Default constructor
133 //
134 memset(fParams, 0, sizeof(Double_t) * 4);
135 memset(fEfficiency, 0, sizeof(Double_t) * 2);
136}
137
138//____________________________________________________________
139AliTRDPIDParams::AliTRDPIDThresholds::AliTRDPIDThresholds(Int_t nTracklets, Double_t effMin, Double_t effMax, Double_t *params) :
140 TObject(),
141 fNTracklets(nTracklets)
142{
143 //
144 // Default Constructor
145 //
146 fEfficiency[0] = effMin;
147 fEfficiency[1] = effMax;
148 if(params) memcpy(fParams, params, sizeof(Double_t) * 4);
149 else memset(fParams, 0, sizeof(Double_t) * 4);
150}
151
152//____________________________________________________________
153AliTRDPIDParams::AliTRDPIDThresholds::AliTRDPIDThresholds(Int_t nTracklets, Double_t eff, Double_t *params) :
154 TObject(),
155 fNTracklets(nTracklets)
156{
157 //
158 // Constructor used to find object in sorted list
159 //
160 fEfficiency[0] = fEfficiency[1] = eff;
161 if(params) memcpy(fParams, params, sizeof(Double_t) * 4);
162 else memset(fParams, 0, sizeof(Double_t) * 4);
163}
164
165//____________________________________________________________
166AliTRDPIDParams::AliTRDPIDThresholds::AliTRDPIDThresholds(const AliTRDPIDThresholds &ref) :
167 TObject(ref),
168 fNTracklets(ref.fNTracklets)
169{
170 //
171 // Copy constructor
172 //
173 memcpy(fParams, ref.fParams, sizeof(Double_t) * 4);
174 memcpy(fEfficiency, ref.fEfficiency, sizeof(Double_t) * 2);
175}
176
177//____________________________________________________________
178AliTRDPIDParams::AliTRDPIDThresholds &AliTRDPIDParams::AliTRDPIDThresholds::operator=(const AliTRDPIDThresholds &ref){
179 //
180 // Assignment operator
181 //
e99fb5c9 182 if(&ref == this) return *this;
183
ce487a7f 184 TObject::operator=(ref);
185
186 fNTracklets = ref.fNTracklets;
187 memcpy(fEfficiency, ref.fEfficiency, sizeof(Double_t) * 2);
188 memcpy(fParams, ref.fParams, sizeof(Double_t) * 4);
189 return *this;
190}
191
192//____________________________________________________________
193Int_t AliTRDPIDParams::AliTRDPIDThresholds::Compare(const TObject *ref) const{
194 //
195 // Compares two objects
196 // Order:
197 // First compare number of tracklets, if they are equal compare electron efficiency
198 //
199 const AliTRDPIDThresholds *refObj = static_cast<const AliTRDPIDThresholds *>(ref);
200 if(fNTracklets < refObj->GetNTracklets()) return -1;
201 else if(fNTracklets > refObj->GetNTracklets()) return 1;
202 else{
203 if(fEfficiency[1] < refObj->GetElectronEfficiency(0)) return -1;
204 else if(fEfficiency[0] > refObj->GetElectronEfficiency(1)) return 1;
205 else return 0;
206 }
207}
208
209//____________________________________________________________
210Bool_t AliTRDPIDParams::AliTRDPIDThresholds::IsEqual(const TObject *ref) const {
211 //
212 // Check for equality
213 // Tracklets and Efficiency are used
214 //
215 const AliTRDPIDThresholds *refObj = dynamic_cast<const AliTRDPIDThresholds *>(ref);
216 if(!refObj) return kFALSE;
217 Bool_t eqNTracklets = fNTracklets == refObj->GetNTracklets();
218 Bool_t eqEff = kFALSE;
219 Bool_t hasRange = TMath::Abs(fEfficiency[1] - fEfficiency[0]) > kVerySmall;
220 Bool_t hasRangeRef = TMath::Abs(refObj->GetElectronEfficiency(1) - refObj->GetElectronEfficiency(0)) > kVerySmall;
221 if(hasRange && hasRangeRef){
222 // Both have ranges, check if they match
223 eqEff = TMath::Abs(fEfficiency[0] - refObj->GetElectronEfficiency(0)) < kVerySmall && TMath::Abs(fEfficiency[1] - refObj->GetElectronEfficiency(1)) < kVerySmall;
224 } else if(hasRange){
225 // this object has ranges, check if the efficiency of ref is inside the range
226 eqEff = refObj->GetElectronEfficiency(0) >= fEfficiency[0] && refObj->GetElectronEfficiency(0) < fEfficiency[1];
227 } else {
228 // ref has ranges, check if this is in range
229 eqEff = fEfficiency[0] >= refObj->GetElectronEfficiency(0) && fEfficiency[0] < refObj->GetElectronEfficiency(1);
230 }
231
232 return eqNTracklets && eqEff;
233}