TRD nSigma OADB related new codes and modifications and OADB root file -- Xianguo Lu
[u/mrichter/AliRoot.git] / ITS / AliITSMapA1.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 #include <TObjArray.h>
16 #include "AliITSMapA1.h"
17 #include "AliITSsegmentation.h"
18 #include "AliITSdigit.h"
19
20 ////////////////////////////////////////////////////////////////////////
21 //  Map Class for ITS. Implementation A1. In this implementation, the //
22 // 2 dimensional (iz,ix) map is filled with integers values. For each //
23 // cell a corresponding TObject, a hit, can also be stored.           //
24 //  The detector geometry is accessed via the that detectors          //
25 // segmentation class and stored here for conveniance.                //
26 ////////////////////////////////////////////////////////////////////////
27
28 ClassImp(AliITSMapA1)
29 //______________________________________________________________________
30 AliITSMapA1::AliITSMapA1():
31 fSegmentation(0),
32 fNpx(0),
33 fNpz(0),
34 fObjects(0),
35 fNobjects(0),
36 fMaxIndex(0),
37 fMapThresholdArr(0),
38 fHitMap(0),
39 fMapThreshold(0){
40     // default constructor
41
42 }
43 //______________________________________________________________________
44 AliITSMapA1::AliITSMapA1(AliITSsegmentation *seg):
45 fSegmentation(seg),
46 fNpx(0),
47 fNpz(0),
48 fObjects(0),
49 fNobjects(0),
50 fMaxIndex(0),
51 fMapThresholdArr(0),
52 fHitMap(0),
53 fMapThreshold(0){
54     //constructor
55
56     fNpz          = fSegmentation->Npz();
57     fNpx          = fSegmentation->Npx();
58     fMaxIndex     = fNpz*fNpx+fNpx;             // 2 halves of detector
59     fHitMap       = new Int_t[fMaxIndex];
60     ClearMap();
61 }
62 //______________________________________________________________________
63 AliITSMapA1::AliITSMapA1(AliITSsegmentation *seg, TObjArray *obj):
64 fSegmentation(seg),
65 fNpx(0),
66 fNpz(0),
67 fObjects(obj),
68 fNobjects(0),
69 fMaxIndex(0),
70 fMapThresholdArr(0),
71 fHitMap(0),
72 fMapThreshold(0){
73     //constructor
74
75   fNpz          = fSegmentation->Npz();
76   fNpx          = fSegmentation->Npx();
77   fMaxIndex     = fNpz*fNpx+fNpx;             // 2 halves of detector
78   fHitMap       = new Int_t[fMaxIndex];
79   if(fObjects) fNobjects = fObjects->GetEntriesFast();
80   ClearMap();
81 }
82 //______________________________________________________________________
83 AliITSMapA1::AliITSMapA1(AliITSsegmentation *seg, TObjArray *obj, Int_t thr):
84 fSegmentation(seg),
85 fNpx(0),
86 fNpz(0),
87 fObjects(obj),
88 fNobjects(0),
89 fMaxIndex(0),
90 fMapThresholdArr(0),
91 fHitMap(0),
92 fMapThreshold(thr){
93     //constructor
94
95   fNpz          = fSegmentation->Npz();
96   fNpx          = fSegmentation->Npx();
97   fMaxIndex     = fNpz*fNpx+fNpx;             // 2 halves of detector
98   fHitMap       = new Int_t[fMaxIndex];
99   if(fObjects) fNobjects = fObjects->GetEntriesFast();
100   ClearMap();
101 }
102 //______________________________________________________________________
103 AliITSMapA1::AliITSMapA1(AliITSsegmentation *seg, TObjArray *obj, TArrayI thr):
104 fSegmentation(seg),
105 fNpx(0),
106 fNpz(0),
107 fObjects(obj),
108 fNobjects(0),
109 fMaxIndex(0),
110 fMapThresholdArr(thr),
111 fHitMap(0),
112 fMapThreshold(0){
113     //constructor
114
115   fNpz          = fSegmentation->Npz();
116   fNpx          = fSegmentation->Npx();
117   fMaxIndex     = fNpz*fNpx+fNpx;             // 2 halves of detector
118   fHitMap       = new Int_t[fMaxIndex];
119   if(fObjects) fNobjects = fObjects->GetEntriesFast();
120   ClearMap();
121 }
122
123 //______________________________________________________________________
124 AliITSMapA1::~AliITSMapA1(){
125     //destructor
126
127     if(fHitMap) delete[] fHitMap;
128 }
129
130 //______________________________________________________________________
131 void AliITSMapA1::ClearMap(){
132     //clear array
133
134     memset(fHitMap,0,sizeof(int)*fMaxIndex);
135 }
136 //______________________________________________________________________
137 void AliITSMapA1::SetArray(TObjArray *obj){
138     // set array of objects
139
140     fObjects =  obj;
141     if (fObjects) fNobjects = fObjects->GetEntriesFast();
142 }
143 //______________________________________________________________________
144 Int_t AliITSMapA1::CheckedIndex(Int_t iz, Int_t ix) const {
145     //check boundaries and return an index in array
146     Int_t index=fNpx*iz+ix;
147
148     //if (index > fMaxIndex) {
149     if (index > fMaxIndex || index < 0) {
150         printf("\n \n \n Try to read/write outside array !!!!"
151                " \n \n %d %d %d %d %d %d \n",iz,ix,fMaxIndex,index,fNpz,fNpx);
152         // force crash
153         return  -1;
154     } else {
155         return index;
156     } // end if index>max or < 0
157 }
158 //______________________________________________________________________
159 void  AliITSMapA1::FillMap(){
160     // fill array with digits indices
161
162     Int_t ndigits = fObjects->GetEntriesFast();
163     if (!ndigits) return;
164
165     AliITSdigit *dig;
166     for (Int_t ndig=0; ndig<ndigits; ndig++) {
167         dig = (AliITSdigit*)fObjects->UncheckedAt(ndig);
168         if(dig->GetSignal() > fMapThreshold) {
169             SetHit(dig->GetCoord1(),dig->GetCoord2(),ndig);
170         } // end if fSignal > fMapthreshold
171     } // end for ndig
172 }
173 //______________________________________________________________________
174 void  AliITSMapA1::FillMap2(){
175     // fill array with digits indices
176
177     Int_t ndigits = fObjects->GetEntriesFast();
178     if (!ndigits) return;
179
180     AliITSdigit *dig;
181     for (Int_t ndig=0; ndig<ndigits; ndig++) {
182         dig = (AliITSdigit*)fObjects->UncheckedAt(ndig);
183         if(dig->GetSignal() > fMapThresholdArr[dig->GetCoord1()]) {
184             SetHit(dig->GetCoord1(),dig->GetCoord2(),ndig);
185         } // end if fSignal > fMapthreshold
186     } // end for ndig
187 }
188
189 //______________________________________________________________________
190 void  AliITSMapA1::SetHit(Int_t iz, Int_t ix, Int_t idigit){
191     // set the digit index at a certain position in array
192
193     fHitMap[CheckedIndex(iz, ix)]=idigit+1;
194 }
195 //______________________________________________________________________
196 void AliITSMapA1::DeleteHit(Int_t iz, Int_t ix){
197     // delete an entry in array
198
199     fHitMap[CheckedIndex(iz, ix)]=0;
200 }
201 //______________________________________________________________________
202 void AliITSMapA1::FlagHit(Int_t iz, Int_t ix){
203     // flag an entry in array
204
205     fHitMap[CheckedIndex(iz, ix)] = -TMath::Abs(fHitMap[CheckedIndex(iz, ix)]);
206 }
207 //______________________________________________________________________
208 Int_t AliITSMapA1::GetHitIndex(Int_t iz, Int_t ix) const {
209     // return the digit index from a specific entry in array
210
211     return TMath::Abs(fHitMap[CheckedIndex(iz, ix)])-1;
212 }
213 //______________________________________________________________________
214 TObject* AliITSMapA1::GetHit(Int_t iz, Int_t ix) const {
215     // return the pointer to the digit 
216
217     Int_t index=GetHitIndex(iz,ix);
218     // Force crash if index does not exist ! 
219     return (index <0) ? 0 : fObjects->UncheckedAt(GetHitIndex(iz,ix));
220 }
221 //______________________________________________________________________
222 Double_t AliITSMapA1::GetSignal(Int_t iz, Int_t ix) const{
223     // get a pad signal
224     Double_t signal;
225
226     AliITSdigit *dig = (AliITSdigit*)GetHit(iz,ix);
227     if(dig) signal=(Double_t)dig->GetSignal();
228     else signal=0.;
229     return signal;
230 }
231 //______________________________________________________________________
232 FlagTypeITS AliITSMapA1::TestHit(Int_t iz, Int_t ix)  {
233     // check whether the digit has already been flagged
234
235     if (CheckedIndex(iz, ix) < 0) return kEmptyITS;
236     Int_t inf=fHitMap[CheckedIndex(iz, ix)]; 
237     if (inf < 0) {
238         return kUsedITS;
239     } else if (inf == 0) {
240         return kEmptyITS;
241     } else {
242         return kUnusedITS;
243     } // end if inf
244 }
245