1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
16 // Container for TRD Threshold parameters stored in the OADB
18 // Author: Markus Fasel <M.Fasel@gsi.de>
22 #include <TSortedList.h>
26 #include "AliTRDPIDParams.h"
28 ClassImp(AliTRDPIDParams)
29 //ClassImp(AliTRDPIDParams::AliTRDPIDThresholds)
30 //ClassImp(AliTRDPIDParams::AliTRDPIDCentrality)
32 const Double_t AliTRDPIDParams::kVerySmall = 1e-5;
34 //____________________________________________________________
35 AliTRDPIDParams::AliTRDPIDParams():
44 //____________________________________________________________
45 AliTRDPIDParams::AliTRDPIDParams(const char *name) :
50 // Default constructor
55 //____________________________________________________________
56 AliTRDPIDParams::AliTRDPIDParams(const AliTRDPIDParams &ref):
64 fEntries=(TList*)ref.fEntries->Clone();
67 //____________________________________________________________
68 AliTRDPIDParams::~AliTRDPIDParams(){
75 //____________________________________________________________
76 void AliTRDPIDParams::AddCentralityClass(Double_t minCentrality, Double_t maxCentrality){
78 // Add new centrality class
81 // check whether centrality class already exists
82 AliTRDPIDCentrality *checklow = FindCentrality(minCentrality + 0.01),
83 *checkhigh = FindCentrality(maxCentrality - 0.01);
85 if(!checklow && ! checkhigh)
86 fEntries->Add(new AliTRDPIDCentrality(minCentrality, maxCentrality));
89 //____________________________________________________________
90 AliTRDPIDParams::AliTRDPIDCentrality *AliTRDPIDParams::FindCentrality(Double_t val) const {
92 // Find centrality bin
94 TIter centralities(fEntries);
95 AliTRDPIDCentrality *obj(NULL), *tmp(NULL);
96 while((obj = dynamic_cast<AliTRDPIDCentrality *>(centralities()))){
97 if(val >= obj->GetMinCentrality() && val <= obj->GetMaxCentrality()){
103 AliDebug(10,"Using default centrality class");
104 return FindCentrality(-1);
109 //____________________________________________________________
110 Bool_t AliTRDPIDParams::GetThresholdParameters(Int_t ntracklets, Double_t efficiency, Double_t *params, Double_t centrality) const{
113 // Use IsEqual definition
115 AliTRDPIDCentrality *cent = FindCentrality(centrality);
117 AliDebug(1, "Centrality class not available");
120 cent->GetThresholdParameters(ntracklets, efficiency, params);
124 //____________________________________________________________
125 void AliTRDPIDParams::SetThresholdParameters(Int_t ntracklets, Double_t effMin, Double_t effMax, Double_t *params, Double_t centrality){
127 // Set new threshold parameters
129 AliTRDPIDCentrality *cent = FindCentrality(centrality);
130 if(cent) cent->SetThresholdParameters(ntracklets, effMin, effMax, params);
131 else AliDebug(1, "Centrality class not available");
134 //____________________________________________________________
135 void AliTRDPIDParams::Print(Option_t *) const {
136 TIter centIter(fEntries);
137 AliTRDPIDCentrality *cent;
138 while((cent = dynamic_cast<AliTRDPIDCentrality *>(centIter()))) cent->Print(NULL);
141 //____________________________________________________________
142 AliTRDPIDParams::AliTRDPIDThresholds::AliTRDPIDThresholds():
147 // Default constructor
149 memset(fParams, 0, sizeof(Double_t) * 4);
150 memset(fEfficiency, 0, sizeof(Double_t) * 2);
153 //____________________________________________________________
154 AliTRDPIDParams::AliTRDPIDThresholds::AliTRDPIDThresholds(Int_t nTracklets, Double_t effMin, Double_t effMax, Double_t *params) :
156 fNTracklets(nTracklets)
159 // Default Constructor
161 fEfficiency[0] = effMin;
162 fEfficiency[1] = effMax;
163 if(params) memcpy(fParams, params, sizeof(Double_t) * 4);
164 else memset(fParams, 0, sizeof(Double_t) * 4);
167 //____________________________________________________________
168 AliTRDPIDParams::AliTRDPIDThresholds::AliTRDPIDThresholds(Int_t nTracklets, Double_t eff, Double_t *params) :
170 fNTracklets(nTracklets)
173 // Constructor used to find object in sorted list
175 fEfficiency[0] = fEfficiency[1] = eff;
176 if(params) memcpy(fParams, params, sizeof(Double_t) * 4);
177 else memset(fParams, 0, sizeof(Double_t) * 4);
180 //____________________________________________________________
181 AliTRDPIDParams::AliTRDPIDThresholds::AliTRDPIDThresholds(const AliTRDPIDThresholds &ref) :
183 fNTracklets(ref.fNTracklets)
188 memcpy(fParams, ref.fParams, sizeof(Double_t) * 4);
189 memcpy(fEfficiency, ref.fEfficiency, sizeof(Double_t) * 2);
192 //____________________________________________________________
193 AliTRDPIDParams::AliTRDPIDThresholds &AliTRDPIDParams::AliTRDPIDThresholds::operator=(const AliTRDPIDThresholds &ref){
195 // Assignment operator
197 if(&ref == this) return *this;
199 TObject::operator=(ref);
201 fNTracklets = ref.fNTracklets;
202 memcpy(fEfficiency, ref.fEfficiency, sizeof(Double_t) * 2);
203 memcpy(fParams, ref.fParams, sizeof(Double_t) * 4);
207 //____________________________________________________________
208 Int_t AliTRDPIDParams::AliTRDPIDThresholds::Compare(const TObject *ref) const{
210 // Compares two objects
212 // First compare number of tracklets, if they are equal compare electron efficiency
214 const AliTRDPIDThresholds *refObj = static_cast<const AliTRDPIDThresholds *>(ref);
215 if(fNTracklets < refObj->GetNTracklets()) return -1;
216 else if(fNTracklets > refObj->GetNTracklets()) return 1;
218 if(fEfficiency[1] < refObj->GetElectronEfficiency(0)) return -1;
219 else if(fEfficiency[0] > refObj->GetElectronEfficiency(1)) return 1;
224 //____________________________________________________________
225 Bool_t AliTRDPIDParams::AliTRDPIDThresholds::IsEqual(const TObject *ref) const {
227 // Check for equality
228 // Tracklets and Efficiency are used
230 const AliTRDPIDThresholds *refObj = dynamic_cast<const AliTRDPIDThresholds *>(ref);
231 if(!refObj) return kFALSE;
232 Bool_t eqNTracklets = fNTracklets == refObj->GetNTracklets();
233 Bool_t eqEff = kFALSE;
234 Bool_t hasRange = TMath::Abs(fEfficiency[1] - fEfficiency[0]) > kVerySmall;
235 Bool_t hasRangeRef = TMath::Abs(refObj->GetElectronEfficiency(1) - refObj->GetElectronEfficiency(0)) > kVerySmall;
236 if(hasRange && hasRangeRef){
237 // Both have ranges, check if they match
238 eqEff = TMath::Abs(fEfficiency[0] - refObj->GetElectronEfficiency(0)) < kVerySmall && TMath::Abs(fEfficiency[1] - refObj->GetElectronEfficiency(1)) < kVerySmall;
240 // this object has ranges, check if the efficiency of ref is inside the range
241 eqEff = refObj->GetElectronEfficiency(0) >= fEfficiency[0] && refObj->GetElectronEfficiency(0) < fEfficiency[1];
243 // ref has ranges, check if this is in range
244 eqEff = fEfficiency[0] >= refObj->GetElectronEfficiency(0) && fEfficiency[0] < refObj->GetElectronEfficiency(1);
247 return eqNTracklets && eqEff;
250 //____________________________________________________________
251 AliTRDPIDParams::AliTRDPIDCentrality::AliTRDPIDCentrality():
261 //____________________________________________________________
262 AliTRDPIDParams::AliTRDPIDCentrality::AliTRDPIDCentrality(Double_t minCentrality, Double_t maxCentrality):
264 fMinCentrality(minCentrality),
265 fMaxCentrality(maxCentrality)
268 // Default constructor
270 fEntries = new TSortedList;
271 fEntries->SetOwner();
274 //____________________________________________________________
275 AliTRDPIDParams::AliTRDPIDCentrality::AliTRDPIDCentrality(const AliTRDPIDParams::AliTRDPIDCentrality &ref):
278 fMinCentrality(ref.fMinCentrality),
279 fMaxCentrality(ref.fMaxCentrality)
284 fEntries = new TSortedList;
285 // Coply entries to the new list
286 TIter entries(ref.fEntries);
288 while((o = entries())) fEntries->Add(o);
291 //____________________________________________________________
292 AliTRDPIDParams::AliTRDPIDCentrality &AliTRDPIDParams::AliTRDPIDCentrality::operator=(const AliTRDPIDCentrality &ref){
294 // Assignment operator
297 if(fEntries) delete fEntries;
298 fEntries = new TSortedList;
299 TIter entries(ref.fEntries);
301 while((o = entries())) fEntries->Add(o);
302 fMinCentrality = ref.fMinCentrality;
303 fMaxCentrality = ref.fMaxCentrality;
308 //____________________________________________________________
309 AliTRDPIDParams::AliTRDPIDCentrality::~AliTRDPIDCentrality(){
313 if(fEntries) delete fEntries;
316 //____________________________________________________________
317 Bool_t AliTRDPIDParams::AliTRDPIDCentrality::GetThresholdParameters(Int_t ntracklets, Double_t efficiency, Double_t *params) const{
319 // Get the threshold parameters
321 AliTRDPIDThresholds test(ntracklets, efficiency);
322 TObject *result = fEntries->FindObject(&test);
324 AliDebug(1, Form("No threshold params found for %d tracklets and an electron efficiency of %f", ntracklets, efficiency));
327 AliTRDPIDThresholds *parResult = static_cast<AliTRDPIDThresholds *>(result);
328 AliDebug(1, Form("Threshold params found: NTracklets %d, Electron Efficiency %f", parResult->GetNTracklets(), parResult->GetElectronEfficiency()));
329 memcpy(params, parResult->GetThresholdParams(), sizeof(Double_t) * 4);
333 //____________________________________________________________
334 void AliTRDPIDParams::AliTRDPIDCentrality::SetThresholdParameters(Int_t ntracklets, Double_t effMin, Double_t effMax, Double_t *params){
336 // Store new Params in the Object
339 AliError("Min. efficiency has to be >= max. efficiency");
342 AliDebug(1, Form("Save Parameters for %d tracklets at and electron efficiency of [%f|%f]", ntracklets, effMin, effMax));
343 fEntries->Add(new AliTRDPIDThresholds(ntracklets, effMin, effMax, params));
346 //____________________________________________________________
347 void AliTRDPIDParams::AliTRDPIDCentrality::Print(Option_t *) const {
348 printf("Min. Centrality: %f, Max. Centrality: %f\n", fMinCentrality, fMaxCentrality);
349 printf("Available thresholds:\n");
350 printf("_________________________________________\n");
351 TIter objects(fEntries);
352 AliTRDPIDThresholds *par;
353 while((par = dynamic_cast<AliTRDPIDThresholds *>(objects()))){
354 printf("Number of tracklets %d, Electron efficiency %f\n", par->GetNTracklets(), 0.5*(par->GetElectronEfficiency(0)+par->GetElectronEfficiency(1)));