e3e6379f918f08676c1481f3b1f63b99156e0405
[u/mrichter/AliRoot.git] / STEER / STEERBase / AliTRDPIDReference.cxx
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 class for the reference distributions for TRD PID
17 // The class contains the reference distributions and the momentum steps
18 // the references are taken at. Mapping is done inside. To derive references,
19 // the functions GetUpperReference and GetLowerReference return the next
20 // reference distribution object and the momentum step above respectively below
21 // the tracklet momentum.
22 //
23 // Authors:
24 //    Markus Fasel <M.Fasel@gsi.de>
25 //
26 #include "TObjArray.h"
27
28 #include "AliLog.h"
29
30 #include "AliTRDPIDReference.h"
31
32 ClassImp(AliTRDPIDReference)
33
34 //____________________________________________________________
35 AliTRDPIDReference::AliTRDPIDReference():
36 TNamed(),
37 fRefContainer(NULL),
38 fMomentumBins()
39 {
40         //
41         // Dummy constructor
42         //
43         SetBit(kIsOwner, kTRUE);
44 }
45
46 //____________________________________________________________
47 AliTRDPIDReference::AliTRDPIDReference(const Char_t *name):
48                   TNamed(name, "TRD PID References"),
49                   fRefContainer(NULL),
50                   fMomentumBins()
51 {
52         //
53         // Default constructor
54         //
55         SetBit(kIsOwner, kTRUE);
56 }
57
58 //____________________________________________________________
59 AliTRDPIDReference::AliTRDPIDReference(const AliTRDPIDReference &ref):
60                   TNamed(ref),
61                   fRefContainer(ref.fRefContainer),
62                   fMomentumBins(ref.fMomentumBins)
63 {
64         //
65         // Copy constructor
66         // Only copies poiters, object is not the owner of the references
67         //
68         SetBit(kIsOwner, kFALSE);
69 }
70
71 //____________________________________________________________
72 AliTRDPIDReference &AliTRDPIDReference::operator=(const AliTRDPIDReference &ref){
73         //
74         // Assginment operator
75         // Only copies poiters, object is not the owner of the references
76         //
77         if(this != &ref){
78                 TNamed::operator=(ref);
79                 if(TestBit(kIsOwner) && fRefContainer) delete fRefContainer;
80                 fRefContainer = ref.fRefContainer;
81                 fMomentumBins = ref.fMomentumBins;
82                 SetBit(kIsOwner, kFALSE);
83         }
84         return *this;
85 }
86
87 //____________________________________________________________
88 AliTRDPIDReference::~AliTRDPIDReference(){
89         //
90         // Destructor
91         // references are deleted if the object is the owner
92         //
93         if(fRefContainer && TestBit(kIsOwner)) delete fRefContainer;
94 }
95
96 //____________________________________________________________
97 void AliTRDPIDReference::SetNumberOfMomentumBins(Int_t nBins, Float_t *momenta){
98         //
99         // Set the momentum binning
100         //
101         if(fRefContainer) fRefContainer->Clear();
102         else{
103                 fRefContainer = new TObjArray;
104                 fRefContainer->SetOwner();
105         }
106         fRefContainer->Expand(nBins * AliPID::kSPECIES);
107         fMomentumBins.Set(nBins,momenta);
108 }
109
110 //____________________________________________________________
111 void AliTRDPIDReference::AddReference(TObject *ref, AliPID::EParticleType spec, Int_t pbin){
112         //
113         // Add a new reference distribution for a given species and a givem
114         // momentum value to the reference container
115         // The reference distribution is associated with the momentum value defined for the
116         // given momentum bin
117         //
118         if(!fRefContainer){
119                 AliError("Reference Container not initialized");
120                 return;
121         }
122         if(pbin > fMomentumBins.GetSize()){
123                 AliError("Pbin overflow");
124                 return;
125         }
126         AliDebug(1, Form("Adding object with address %p to position %d", ref, spec * fMomentumBins.GetSize() + pbin));
127         fRefContainer->AddAt(ref, spec * fMomentumBins.GetSize() + pbin);
128 }
129
130 //____________________________________________________________
131 TObject *AliTRDPIDReference::GetUpperReference(AliPID::EParticleType spec, Float_t p, Float_t &pUpper) const{
132         //
133         // Get the next reference associated with a momentum larger than the requested momentum
134         // In case no next upper reference is found, NULL is returned
135         // The momentum value the reference is associated to is stored in the reference pUpper
136         //
137         Int_t bin = -1;
138         pUpper = 20;
139         for(Int_t ip = 0; ip < fMomentumBins.GetSize(); ip++){
140                 AliDebug(10, Form("Bin %d, p = %.1f", ip, fMomentumBins[ip]));
141                 if(p < fMomentumBins[ip]){
142                         AliDebug(10, "Bin found");
143                         bin = ip;
144                         break;
145                 }
146         }
147         AliDebug(2, Form("p = %.1f, bin = %d\n", p, bin));
148         if(bin >= 0) {
149                 pUpper = fMomentumBins[bin];
150                 return fRefContainer->At(spec * fMomentumBins.GetSize() + bin);
151         }
152         else return NULL;
153 }
154
155 //____________________________________________________________
156 TObject *AliTRDPIDReference::GetLowerReference(AliPID::EParticleType spec, Float_t p, Float_t &pLower) const{
157         //
158         // Get the next reference associated with a momentum smaller than the requested momentum
159         // In case no next lower reference is found, NULL is returned
160         // The momentum value the reference is associated to is stored in the reference pLower
161         //
162         Int_t bin = -1;
163         pLower = 0;
164         for(Int_t ip = fMomentumBins.GetSize() - 1; ip >= 0; ip--){
165                 AliDebug(10, Form("Bin %d, p = %.1f", ip, fMomentumBins[ip]));
166                 if(p > fMomentumBins[ip]){
167                         AliDebug(10, "Bin found");
168                         bin = ip;
169                         break;
170                 }
171         }
172         AliDebug(2, Form("p = %.1f, bin = %d\n", p, bin));
173         if(bin >= 0){
174                 pLower = fMomentumBins[bin];
175                 return fRefContainer->At(spec * fMomentumBins.GetSize() + bin);
176         }
177         else return NULL;
178 }
179
180 //____________________________________________________________
181 void AliTRDPIDReference::Print(const Option_t*) const{
182         //
183         // Print content of the PID reference container
184         //
185         printf("Number of Momentum Bins: %d\n", GetNumberOfMomentumBins());
186         printf("=====================================\n");
187         for(Int_t ip = 0; ip < GetNumberOfMomentumBins(); ip++){
188                 printf("Bin %d: p = %.1f\n", ip, fMomentumBins[ip]);
189         }
190         printf("=====================================\n");
191         if(fRefContainer){
192                 printf("Content of the reference container:\n");
193                 for(Int_t ip = 0; ip < GetNumberOfMomentumBins(); ip++){
194                         printf("[");
195                         for(Int_t is = 0; is < 5; is++) printf("%p|", fRefContainer->At(is * fMomentumBins.GetSize() + ip));
196                         printf("]\n");
197                 }
198         }
199 }