Optional scoring for background studies in HALL.
[u/mrichter/AliRoot.git] / TOF / AliTOFHitMap.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 /* $Id$ */
17
18 ////////////////////////////////////////////////////////////////////////
19 //
20 // AliTOFHitMap class
21 //
22 // hitmap enables fast check if the pad was already hit
23 // The index of a AliTOFSDigit is saved in the each hitmap "cell"
24 // (there is an offset +1, because the index can be zero and 
25 // zero means empty cell. 
26 // In TOF, number of strips varies according plate type, the highest
27 // number is in plate C. For all plates is used this number, so
28 // the size of the hitmap is a little bit greater than necessary, but
29 // it simplifies the access algorithm. 
30 // 
31 //
32 // Author: Jiri Chudoba (CERN), based on AliMUONHitMap
33 //
34 ////////////////////////////////////////////////////////////////////////
35
36 #include "AliLog.h"
37 #include "AliTOFHitMap.h"
38 #include "AliTOFGeometry.h"
39
40
41 #include <TClonesArray.h>
42
43 ClassImp(AliTOFHitMap)
44
45 AliTOFHitMap::AliTOFHitMap():
46   fNSector(-1),
47   fNplate(-1),
48   fNstrip(-1),
49   fNpx(-1),
50   fNpz(-1),
51   fSDigits(0x0),
52   fMaxIndex(-1),
53   fHitMap(0x0)
54 {
55 //
56 // Default ctor
57 //
58 }
59
60 ////////////////////////////////////////////////////////////////////////
61 AliTOFHitMap::AliTOFHitMap(TClonesArray *dig):
62   fNSector(-1),
63   fNplate(-1),
64   fNstrip(-1),
65   fNpx(-1),
66   fNpz(-1),
67   fSDigits(dig),
68   fMaxIndex(-1),
69   fHitMap(0x0)
70 {
71 //
72 // ctor
73 //
74
75 // of course, these constants must not be hardwired
76 // change later
77
78   fNSector = AliTOFGeometry::NSectors();
79   fNplate = AliTOFGeometry::NPlates();
80   fNstrip = AliTOFGeometry::NStripC();//fTOFGeometry->NMaxNstrip();
81   fNpx  = AliTOFGeometry::NpadX();
82   fNpz  = AliTOFGeometry::NpadZ();
83   fMaxIndex=fNSector*fNplate*fNstrip*fNpx*fNpz;
84   fHitMap = new Int_t[fMaxIndex];
85   Clear();
86 }
87
88 ////////////////////////////////////////////////////////////////////////
89 AliTOFHitMap::AliTOFHitMap(const AliTOFHitMap & hitMap) :
90   TObject(hitMap),
91   fNSector(hitMap.fNSector),
92   fNplate(hitMap.fNplate),
93   fNstrip(hitMap.fNstrip),
94   fNpx(hitMap.fNpx),
95   fNpz(hitMap.fNpz),
96   fSDigits(hitMap.fSDigits),
97   fMaxIndex(hitMap.fMaxIndex),
98   fHitMap(0x0)
99 {
100 //
101 // Dummy copy constructor
102 //
103
104   fMaxIndex=fNSector*fNplate*fNstrip*fNpx*fNpz;
105   fHitMap = new Int_t[fMaxIndex];
106   for (Int_t i=0; i<fMaxIndex; i++)
107     fHitMap[i]=hitMap.fHitMap[i];
108
109 }
110
111  
112 ////////////////////////////////////////////////////////////////////////
113 AliTOFHitMap::~AliTOFHitMap()
114 {
115 //
116 // Destructor
117 //
118   delete[] fHitMap;
119
120 }
121
122 ////////////////////////////////////////////////////////////////////////
123 void AliTOFHitMap::Clear(const char *)
124 {
125 //
126 // Clear hitmap
127 //
128     memset(fHitMap,0,sizeof(int)*fMaxIndex);
129 }
130
131 ////////////////////////////////////////////////////////////////////////
132 Int_t AliTOFHitMap::CheckedIndex(Int_t * const vol) const
133 {
134 //
135 // Return checked indices for vol
136 //
137   Int_t index=
138     vol[0]*fNplate*fNstrip*fNpx*fNpz+             // sector
139     vol[1]*fNstrip*fNpx*fNpz+                     // plate
140     vol[2]*fNpx*fNpz+                             // strip
141     vol[3]*fNpz+                                  // padx
142     vol[4];                                       // padz
143
144     if (index >= fMaxIndex) {
145       AliError("CheckedIndex - input outside bounds");
146         return -1;
147     } else {
148         return index;
149     }
150 }
151
152 ////////////////////////////////////////////////////////////////////////
153 void  AliTOFHitMap::SetHit(Int_t *vol, Int_t idigit)
154 {
155 //
156 // Assign digit to pad vol
157 //
158
159 // 0 means empty pad, we need to shift indeces by 1
160     fHitMap[CheckedIndex(vol)]=idigit+1;
161 }
162
163 ////////////////////////////////////////////////////////////////////////
164 void  AliTOFHitMap::SetHit(Int_t *vol)
165 {
166 //
167 // Assign last digit to pad vol 
168 //
169
170 // 0 means empty pad, we need to shift indeces by 1
171     fHitMap[CheckedIndex(vol)]=fSDigits->GetLast()+1;
172 }
173
174 ////////////////////////////////////////////////////////////////////////
175 Int_t AliTOFHitMap::GetHitIndex(Int_t *vol) const
176 {
177 //
178 // Get contents of pad vol
179 //
180
181 // 0 means empty pad, we need to shift indeces by 1
182     return fHitMap[CheckedIndex(vol)]-1;
183 }
184
185 ////////////////////////////////////////////////////////////////////////
186 TObject* AliTOFHitMap::GetHit(Int_t *vol) const
187 {
188 //
189 // Get pointer to object at vol
190 // return 0 if vol out of bounds
191     Int_t index=GetHitIndex(vol);
192     return (index <0) ? 0 : fSDigits->UncheckedAt(index);
193 }
194
195 ////////////////////////////////////////////////////////////////////////
196 FlagType AliTOFHitMap::TestHit(Int_t *vol) const
197 {
198 //
199 // Check if hit cell is empty, used or unused
200 //
201     Int_t inf=fHitMap[CheckedIndex(vol)];
202     if (inf > 0) {
203         return kUsed;
204     } else if (inf == 0) {
205         return kEmpty;
206     } else {
207         return kUnused;
208     }
209 }
210
211 ////////////////////////////////////////////////////////////////////////
212 AliTOFHitMap & AliTOFHitMap::operator = (const AliTOFHitMap & hitMap) 
213 {
214   // Assignment operator
215
216   if (this == &hitMap)
217     return *this;
218
219   TObject::operator=(hitMap);
220   fNSector=hitMap.fNSector;
221   fNplate=hitMap.fNplate;
222   fNstrip=hitMap.fNstrip;
223   fNpx=hitMap.fNpx;
224   fNpz=hitMap.fNpz;
225
226   fSDigits=hitMap.fSDigits; // coverity: to be solved
227   /*
228   fSDigits = new TClonesArray("AliTOFSDigit");
229   for (Int_t ii=0; ii<hitMap.fSDigits->GetEntriesFast(); ii++)
230     fSDigits->AddLast(hitMap.fSDigits->UncheckedAt(ii));
231   */
232   //fSDigits = TClonesArray(&hitMap.fSDigits);
233
234   fMaxIndex=fNSector*fNplate*fNstrip*fNpx*fNpz;
235   fHitMap = new Int_t[fMaxIndex];
236   for (Int_t i=0; i<fMaxIndex; i++)
237     fHitMap[i]=hitMap.fHitMap[i];
238
239   return *this;
240
241 }