- Adding check and flagging for HG present
[u/mrichter/AliRoot.git] / HLT / CALO / AliHLTCaloDigitMaker.cxx
1 // $Id$\r
2 \r
3 /**************************************************************************\r
4  * This file is property of and copyright by the ALICE HLT Project        * \r
5  * All rights reserved.                                                   *\r
6  *                                                                        *\r
7  * Primary Authors: Oystein Djuvsland                                     *\r
8  *                                                                        *\r
9  * Permission to use, copy, modify and distribute this software and its   *\r
10  * documentation strictly for non-commercial purposes is hereby granted   *\r
11  * without fee, provided that the above copyright notice appears in all   *\r
12  * copies and that both the copyright notice and this permission notice   *\r
13  * appear in the supporting documentation. The authors make no claims     *\r
14  * about the suitability of this software for any purpose. It is          * \r
15  * provided "as is" without express or implied warranty.                  *\r
16  **************************************************************************/\r
17  /** \r
18  * @file   AliHLTCALOClusterizer.cxx\r
19  * @author Oystein Djuvsland\r
20  * @date \r
21  * @brief  Digit maker for CALO HLT  \r
22  */\r
23       \r
24 \r
25 // see header file for class documentation\r
26 // or\r
27 // refer to README to build package\r
28 // or\r
29 // visit http://web.ift.uib.no/~kjeks/doc/alice-hlt\r
30 \r
31 #include "AliHLTCaloDigitMaker.h"\r
32 #include "AliHLTCaloConstantsHandler.h"\r
33 #include "AliHLTCaloMapper.h"\r
34 #include "AliHLTCaloChannelDataStruct.h"\r
35 #include "AliHLTCaloChannelDataHeaderStruct.h"\r
36 #include "AliHLTCaloDigitDataStruct.h"\r
37 #include "AliHLTCaloCoordinate.h"\r
38 #include "AliHLTCaloSharedMemoryInterfacev2.h" // added by PTH\r
39 //#include "AliPHOSEMCAGeometry.h"\r
40 #include "TH2F.h"\r
41 #include "AliHLTCaloConstants.h"\r
42 #include "AliHLTLogging.h"\r
43 \r
44 ClassImp(AliHLTCaloDigitMaker);\r
45 \r
46 //using namespace CaloHLTConst;\r
47 \r
48 AliHLTCaloDigitMaker::AliHLTCaloDigitMaker(TString det) :\r
49   AliHLTCaloConstantsHandler(det),\r
50   AliHLTLogging(),\r
51   fShmPtr(0),\r
52   fDigitStructPtr(0),\r
53   fDigitCount(0),\r
54   fMapperPtr(0),\r
55   fHighGainFactors(0),\r
56   fLowGainFactors(0),\r
57   fBadChannelMask(0),\r
58   fChannelBook(0),\r
59   fMaxEnergy(900),\r
60   fMinTime(0.0),\r
61   fMaxTime(1008.0)\r
62 {\r
63   // See header file for documentation\r
64 \r
65   fShmPtr = new AliHLTCaloSharedMemoryInterfacev2(det);\r
66 \r
67   fHighGainFactors = new Float_t*[fCaloConstants->GetNXCOLUMNSMOD()];\r
68   fLowGainFactors = new Float_t*[fCaloConstants->GetNXCOLUMNSMOD()];\r
69   \r
70   fBadChannelMask = new Bool_t**[fCaloConstants->GetNXCOLUMNSMOD()];\r
71   \r
72   fChannelBook= new AliHLTCaloDigitDataStruct**[fCaloConstants->GetNXCOLUMNSMOD()];\r
73   \r
74   for(int x = 0; x < fCaloConstants->GetNXCOLUMNSMOD(); x++)\r
75     {\r
76       fHighGainFactors[x] = new Float_t[fCaloConstants->GetNZROWSMOD()];\r
77       fLowGainFactors[x] = new Float_t[fCaloConstants->GetNZROWSMOD()];\r
78 \r
79       fBadChannelMask[x] = new Bool_t*[fCaloConstants->GetNZROWSMOD()];\r
80 \r
81       fChannelBook[x] = new AliHLTCaloDigitDataStruct*[fCaloConstants->GetNZROWSMOD()];\r
82 \r
83       for(int z = 0; z < fCaloConstants->GetNZROWSMOD(); z++)\r
84         {\r
85 \r
86           fHighGainFactors[x][z] = 0.0153;\r
87           fLowGainFactors[x][z] = 0.245;\r
88          \r
89           fBadChannelMask[x][z] = new Bool_t[fCaloConstants->GetNGAINS()];\r
90           fBadChannelMask[x][z][fCaloConstants->GetHIGHGAIN()] = false;\r
91           fBadChannelMask[x][z][fCaloConstants->GetLOWGAIN()] = false; \r
92           \r
93           fChannelBook[x][z] = 0;\r
94           \r
95         }\r
96     }     \r
97 }\r
98 \r
99 AliHLTCaloDigitMaker::~AliHLTCaloDigitMaker() \r
100 {\r
101   //See header file for documentation\r
102 }\r
103 \r
104 Int_t\r
105 AliHLTCaloDigitMaker::MakeDigits(AliHLTCaloChannelDataHeaderStruct* channelDataHeader, AliHLTUInt32_t availableSize)\r
106 {\r
107   //See header file for documentation\r
108   \r
109   Reset();\r
110 \r
111   UInt_t totSize = sizeof(AliHLTCaloDigitDataStruct);\r
112   \r
113 //   Int_t xMod = -1;\r
114 //   Int_t zMod = -1;\r
115   \r
116 \r
117   AliHLTCaloCoordinate coord;\r
118   \r
119   \r
120   AliHLTCaloChannelDataStruct* currentchannel = 0;\r
121   \r
122   fShmPtr->SetMemory(channelDataHeader);\r
123   currentchannel = fShmPtr->NextChannel();\r
124 \r
125   while(currentchannel != 0)\r
126     {\r
127       if(availableSize < totSize) return -1;\r
128 \r
129       fMapperPtr->ChannelId2Coordinate(currentchannel->fChannelID, coord);\r
130       \r
131       //      fMapperPtr->GetLocalCoord(currentchannel->fChannelID, locCoord);\r
132       if(UseDigit(coord, currentchannel))\r
133       {\r
134         AddDigit(currentchannel, coord);\r
135         //      j++;          \r
136         totSize += sizeof(AliHLTCaloDigitDataStruct);\r
137       }\r
138       currentchannel = fShmPtr->NextChannel(); // Get the next channel\r
139     }\r
140 //       if(currentchannel)\r
141 //      {\r
142 //        fMapperPtr->GetLocalCoord(currentchannel->fChannelID, locCoord);\r
143 //        if(UseDigit(coord1, currentchannel))\r
144 //          {\r
145 //            AddDigit(currentchannel, coord1, locCoord);\r
146 //            j++;            \r
147 //            totSize += sizeof(AliHLTCaloDigitDataStruct);\r
148 //          }\r
149 //        currentchannel = fShmPtr->NextChannel(); // Get the next channel\r
150 //      }\r
151 //     }\r
152    \r
153 //   fDigitCount += j;\r
154    return fDigitCount; \r
155 }\r
156 \r
157 void \r
158 AliHLTCaloDigitMaker::SetGlobalHighGainFactor(Float_t factor)\r
159 {\r
160   //See header file for documentation\r
161   for(int x = 0; x < fCaloConstants->GetNXCOLUMNSMOD(); x++)\r
162     {\r
163       for(int z = 0; z < fCaloConstants->GetNZROWSMOD(); z++)\r
164         {\r
165           fHighGainFactors[x][z] = factor;\r
166         }\r
167     }\r
168 }\r
169 \r
170 \r
171 void\r
172 AliHLTCaloDigitMaker::SetGlobalLowGainFactor(Float_t factor)\r
173 {\r
174   //See header file for documentation\r
175   for(int x = 0; x < fCaloConstants->GetNXCOLUMNSMOD(); x++)\r
176     {\r
177       for(int z = 0; z < fCaloConstants->GetNZROWSMOD(); z++)\r
178         {\r
179           fLowGainFactors[x][z] = factor;\r
180         }\r
181     }\r
182 }\r
183 \r
184 \r
185 void\r
186 AliHLTCaloDigitMaker::SetBadChannelMask(TH2F* badChannelHGHist, TH2F* badChannelLGHist, Float_t qCut)\r
187 {\r
188  for(int x = 0; x < fCaloConstants->GetNXCOLUMNSMOD(); x++)\r
189     {\r
190       for(int z = 0; z < fCaloConstants->GetNZROWSMOD(); z++)\r
191         {\r
192           if(badChannelHGHist->GetBinContent(x, z) < qCut && badChannelHGHist->GetBinContent(x, z) > 0)\r
193             {\r
194               fBadChannelMask[x][z][fCaloConstants->GetHIGHGAIN()] = true;\r
195             }\r
196           else\r
197             {\r
198               fBadChannelMask[x][z][fCaloConstants->GetHIGHGAIN()] = false;\r
199             }\r
200           if(badChannelLGHist->GetBinContent(x, z) < qCut && badChannelLGHist->GetBinContent(x, z) > 0)\r
201             {\r
202               fBadChannelMask[x][z][fCaloConstants->GetLOWGAIN()] = false;\r
203             }\r
204           else\r
205             {\r
206               fBadChannelMask[x][z][fCaloConstants->GetLOWGAIN()] = false;\r
207             }\r
208         }\r
209     }\r
210 }\r
211 \r
212 void\r
213 AliHLTCaloDigitMaker::Reset()\r
214 {\r
215   fDigitCount = 0;\r
216   for(int x = 0; x < fCaloConstants->GetNXCOLUMNSMOD(); x++)\r
217     {\r
218       for(int z = 0; z < fCaloConstants->GetNZROWSMOD(); z++)\r
219         {\r
220           fChannelBook[x][z] = 0;\r
221         }\r
222     }     \r
223 \r
224 }\r
225 \r
226 \r
227 void AliHLTCaloDigitMaker::AddDigit(AliHLTCaloChannelDataStruct* channelData, AliHLTCaloCoordinate &coord)\r
228 {\r
229 \r
230   // Some book keeping of the pointers\r
231   AliHLTCaloDigitDataStruct *tmpDigit = fDigitStructPtr + 1;\r
232 \r
233   // Check if we already have a digit in this position, and correct the book keeping correspondently\r
234   if(fChannelBook[coord.fX][coord.fZ]) \r
235     {\r
236       tmpDigit = fDigitStructPtr;\r
237       fDigitStructPtr = fChannelBook[coord.fX][coord.fZ];\r
238       fDigitCount--;\r
239       //      printf("Going to overwrite digit: x = %d, z = %d, gain = %d, energy = %f\n", fDigitStructPtr->fX, fDigitStructPtr->fZ, fDigitStructPtr->fGain, fDigitStructPtr->fEnergy);\r
240     }\r
241   \r
242   fChannelBook[coord.fX][coord.fZ] = fDigitStructPtr;\r
243  \r
244   fDigitStructPtr->fX = coord.fX;\r
245   fDigitStructPtr->fZ = coord.fZ;\r
246   fDigitStructPtr->fGain = coord.fGain;\r
247   fDigitStructPtr->fOverflow = false;\r
248   fDigitStructPtr->fAssociatedCluster = -1;\r
249   \r
250   fDigitStructPtr->fID = fDigitStructPtr->fZ * fCaloConstants->GetNXCOLUMNSMOD() + fDigitStructPtr->fX;\r
251   \r
252   if(coord.fGain == fCaloConstants->GetHIGHGAIN() )\r
253     {\r
254       fDigitStructPtr->fEnergy = channelData->fEnergy*fHighGainFactors[coord.fX][coord.fZ];\r
255       fDigitStructPtr->fHgPresent = true;\r
256       if(channelData->fEnergy >= fMaxEnergy)\r
257         {\r
258           fDigitStructPtr->fOverflow = true;\r
259         }\r
260       \r
261       HLTDebug("HG channel (x = %d, z = %d) with amplitude: %f --> Digit with energy: %f \n", \r
262                coord.fX, coord.fZ, channelData->fEnergy, fDigitStructPtr->fEnergy);\r
263     }\r
264   else\r
265     {\r
266       fDigitStructPtr->fEnergy = channelData->fEnergy*fLowGainFactors[coord.fX][coord.fZ];\r
267       if(channelData->fEnergy >= fMaxEnergy)\r
268         {\r
269           fDigitStructPtr->fOverflow = true;\r
270         }\r
271       HLTDebug("LG channel (x = %d, z = %d) with amplitude: %f --> Digit with energy: %f\n", coord.fX, coord.fZ, channelData->fEnergy, fDigitStructPtr->fEnergy); \r
272     }\r
273   fDigitStructPtr->fTime = channelData->fTime * 0.0000001; //TODO\r
274   fDigitStructPtr->fCrazyness = channelData->fCrazyness;\r
275   fDigitStructPtr->fModule = coord.fModuleId;\r
276   fDigitStructPtr = tmpDigit;\r
277   //  fDigitStructPtr++;\r
278   fDigitCount++;\r
279 }\r
280 \r
281 bool AliHLTCaloDigitMaker::UseDigit(AliHLTCaloCoordinate &channelCoordinates, AliHLTCaloChannelDataStruct *channel) \r
282 {\r
283    \r
284    if(fBadChannelMask[channelCoordinates.fX][channelCoordinates.fZ][0] == true) return false;\r
285    if(channel->fTime < fMinTime || channel->fTime > fMaxTime) return false;\r
286    \r
287   AliHLTCaloDigitDataStruct *tmpDigit = fChannelBook[channelCoordinates.fX][channelCoordinates.fZ];\r
288   //printf("UseDigit: Got digit, x: %d, z: %d, gain: %d, amp: %f\n", channelCoordinates.fX, channelCoordinates.fZ, channelCoordinates.fGain, channel->fEnergy);\r
289   if(tmpDigit)\r
290     {\r
291       if(channelCoordinates.fGain == fCaloConstants->GetLOWGAIN())\r
292         {\r
293                   //printf("UseDigit: Already have digit with, x: %d, z: %d, with high gain \n", channelCoordinates.fX, channelCoordinates.fZ);\r
294           if(tmpDigit->fOverflow)\r
295             {\r
296                      // printf("But it was in overflow! Let's use this low gain!\n");\r
297               return true;\r
298             }\r
299           return false;\r
300         }\r
301       else\r
302         {\r
303                   //printf("UseDigit: Already have digit with, x: %d, z: %d, with low gain: %d\n", channelCoordinates.fX, channelCoordinates.fZ);\r
304           if(channel->fEnergy > fMaxEnergy )\r
305             {\r
306               tmpDigit->fHgPresent = true;\r
307               return false;\r
308             }\r
309           return true;\r
310         }\r
311     }\r
312   return true;\r
313 }\r
314 \r
315 void AliHLTCaloDigitMaker::SetBadChannel(Int_t x, Int_t z, Bool_t bad)\r
316 {\r
317    // See header file for class documentation\r
318    fBadChannelMask[x][z][0] = bad;\r
319    fBadChannelMask[x][z][1] = bad;\r
320 }\r
321 \r
322 void AliHLTCaloDigitMaker::SetGain(Int_t x, Int_t z, Float_t ratio, Float_t gain)\r
323 {\r
324    // See header file for class documentation\r
325    HLTDebug("Applying gain: %f for channel x: %d, z: %d", gain, x, z); \r
326    fHighGainFactors[x][z] = gain;\r
327    fLowGainFactors[x][z] = gain * ratio;\r
328    \r
329 }\r