Added some more component and structs that were needed, compiles and load beautifully
[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     \r
26 \r
27 // see header file for class documentation\r
28 // or\r
29 // refer to README to build package\r
30 // or\r
31 // visit http://web.ift.uib.no/~kjeks/doc/alice-hlt\r
32 \r
33 #include "AliHLTCaloDigitMaker.h"\r
34 \r
35 #include "AliHLTCaloConstants.h"\r
36 #include "AliHLTCaloMapper.h"\r
37 \r
38 #include "AliHLTCaloChannelDataStruct.h"\r
39 #include "AliHLTCaloChannelDataHeaderStruct.h"\r
40 #include "AliHLTCaloDigitDataStruct.h"\r
41 #include "AliHLTCaloSharedMemoryInterfacev2.h" // added by PTH\r
42 //#include "AliPHOSEMCAGeometry.h"\r
43 #include "TH2F.h"\r
44 #include "AliHLTCaloConstantsHandler.h"\r
45 #include "AliHLTCaloConstants.h"\r
46 \r
47 ClassImp(AliHLTCaloDigitMaker);\r
48 \r
49 //using namespace CaloHLTConst;\r
50 \r
51 AliHLTCaloDigitMaker::AliHLTCaloDigitMaker(TString det) :\r
52   AliHLTCaloConstantsHandler(det),\r
53   fShmPtr(0),\r
54   fDigitStructPtr(0),\r
55   fDigitCount(0),\r
56   fOrdered(true),\r
57   fMapperPtr(0),\r
58   fHighGainFactors(0),\r
59   fLowGainFactors(0),\r
60   fBadChannelMask(0),\r
61   fChannelBook(0)\r
62 {\r
63   //BALLE BALLE\r
64   \r
65   //Must set this in the child instance\r
66   \r
67   // See header file for documentation\r
68 \r
69   fShmPtr = new AliHLTCaloSharedMemoryInterfacev2();\r
70 \r
71   fHighGainFactors = new Float_t*[fCaloConstants->GetNXCOLUMNSMOD()];\r
72   fLowGainFactors = new Float_t*[fCaloConstants->GetNXCOLUMNSMOD()];\r
73 \r
74   fBadChannelMask = new Float_t**[fCaloConstants->GetNXCOLUMNSMOD()];\r
75 \r
76   fChannelBook= new AliHLTCaloDigitDataStruct**[fCaloConstants->GetNXCOLUMNSMOD()];\r
77 \r
78   for(int x = 0; x < fCaloConstants->GetNXCOLUMNSMOD(); x++)\r
79     {\r
80       fHighGainFactors[x] = new Float_t[fCaloConstants->GetNZROWSMOD()];\r
81       fLowGainFactors[x] = new Float_t[fCaloConstants->GetNZROWSMOD()];\r
82 \r
83       fBadChannelMask[x] = new Float_t*[fCaloConstants->GetNZROWSMOD()];\r
84 \r
85       fChannelBook[x] = new AliHLTCaloDigitDataStruct*[fCaloConstants->GetNZROWSMOD()];\r
86 \r
87       for(int z = 0; z < fCaloConstants->GetNZROWSMOD(); z++)\r
88         {\r
89 \r
90           fHighGainFactors[x][z] = 0.005;\r
91           fLowGainFactors[x][z] = 0.08;\r
92          \r
93           fBadChannelMask[x][z] = new Float_t[fCaloConstants->GetNGAINS()];\r
94           fBadChannelMask[x][z][fCaloConstants->GetHIGHGAIN()] = 1;\r
95           fBadChannelMask[x][z][fCaloConstants->GetLOWGAIN()] = 1; \r
96           \r
97           fChannelBook[x][z] = 0;\r
98           \r
99         }\r
100     }     \r
101   \r
102   //Must be set in child instance\r
103 //fMapperPtr = new AliHLTCaloMapper(det);\r
104 }\r
105    \r
106 AliHLTCaloDigitMaker::~AliHLTCaloDigitMaker() \r
107 {\r
108   //See header file for documentation\r
109 }\r
110 \r
111 Int_t\r
112 AliHLTCaloDigitMaker::MakeDigits(AliHLTCaloChannelDataHeaderStruct* channelDataHeader, AliHLTUInt32_t availableSize)\r
113 {\r
114   //See header file for documentation\r
115   \r
116   Int_t j = 0;\r
117   UInt_t totSize = sizeof(AliHLTCaloDigitDataStruct);\r
118   \r
119 //   Int_t xMod = -1;\r
120 //   Int_t zMod = -1;\r
121   \r
122   UShort_t coord1[4];\r
123   UShort_t coord2[4];\r
124   Float_t locCoord[3];\r
125   \r
126   \r
127   AliHLTCaloChannelDataStruct* currentchannel = 0;\r
128   AliHLTCaloChannelDataStruct* currentchannelLG = 0;  \r
129   AliHLTCaloChannelDataStruct* tmpchannel = 0;\r
130   \r
131   fShmPtr->SetMemory(channelDataHeader);\r
132   currentchannel = fShmPtr->NextChannel();\r
133 \r
134   while(currentchannel != 0)\r
135     {\r
136       if(availableSize < totSize) return -1;\r
137 \r
138       fMapperPtr->GetChannelCoord(currentchannel->fChannelID, coord1);\r
139       \r
140       tmpchannel = currentchannel;\r
141           \r
142       if(coord1[2] == fCaloConstants->GetHIGHGAIN()) // We got a completely new crystal\r
143         {\r
144           fMapperPtr->GetLocalCoord(currentchannel->fChannelID, locCoord);\r
145           if(UseDigit(coord1, currentchannel))\r
146             {\r
147               AddDigit(currentchannel, coord1, locCoord);\r
148               j++;            \r
149               totSize += sizeof(AliHLTCaloDigitDataStruct);\r
150             }\r
151           currentchannel = fShmPtr->NextChannel(); // Get the next channel\r
152         }\r
153       else if(coord1[2] == fCaloConstants->GetLOWGAIN())\r
154         {\r
155           fMapperPtr->GetLocalCoord(currentchannel->fChannelID, locCoord);\r
156           if(UseDigit(coord1, currentchannel))\r
157             {\r
158               AddDigit(currentchannel, coord1, locCoord);\r
159               j++;            \r
160               totSize += sizeof(AliHLTCaloDigitDataStruct);\r
161             }\r
162           currentchannel = fShmPtr->NextChannel(); // Get the next channel\r
163         }\r
164     }\r
165 \r
166   fDigitCount += j;\r
167   return fDigitCount; \r
168 }\r
169 \r
170 void \r
171 AliHLTCaloDigitMaker::SetGlobalHighGainFactor(Float_t factor)\r
172 {\r
173   //See header file for documentation\r
174   for(int x = 0; x < fCaloConstants->GetNXCOLUMNSMOD(); x++)\r
175     {\r
176       for(int z = 0; z < fCaloConstants->GetNZROWSMOD(); z++)\r
177         {\r
178           fHighGainFactors[x][z] = factor;\r
179         }\r
180     }\r
181 }\r
182 \r
183 void\r
184 AliHLTCaloDigitMaker::SetGlobalLowGainFactor(Float_t factor)\r
185 {\r
186   //See header file for documentation\r
187   for(int x = 0; x < fCaloConstants->GetNXCOLUMNSMOD(); x++)\r
188     {\r
189       for(int z = 0; z < fCaloConstants->GetNZROWSMOD(); z++)\r
190         {\r
191           fLowGainFactors[x][z] = factor;\r
192         }\r
193     }\r
194 }\r
195 \r
196 void\r
197 AliHLTCaloDigitMaker::SetBadChannelMask(TH2F* badChannelHGHist, TH2F* badChannelLGHist, Float_t qCut)\r
198 {\r
199  for(int x = 0; x < fCaloConstants->GetNXCOLUMNSMOD(); x++)\r
200     {\r
201       for(int z = 0; z < fCaloConstants->GetNZROWSMOD(); z++)\r
202         {\r
203           if(badChannelHGHist->GetBinContent(x, z) < qCut && badChannelHGHist->GetBinContent(x, z) > 0)\r
204             {\r
205               fBadChannelMask[x][z][fCaloConstants->GetHIGHGAIN()] = 1;\r
206             }\r
207           else\r
208             {\r
209               fBadChannelMask[x][z][fCaloConstants->GetHIGHGAIN()] = 0;\r
210             }\r
211           if(badChannelLGHist->GetBinContent(x, z) < qCut && badChannelLGHist->GetBinContent(x, z) > 0)\r
212             {\r
213               fBadChannelMask[x][z][fCaloConstants->GetLOWGAIN()] = 0;\r
214             }\r
215           else\r
216             {\r
217               fBadChannelMask[x][z][fCaloConstants->GetLOWGAIN()] = 0;\r
218             }\r
219         }\r
220     }\r
221 }\r
222 \r
223 void\r
224 AliHLTCaloDigitMaker::Reset()\r
225 {\r
226   fDigitCount = 0;\r
227   for(int x = 0; x < fCaloConstants->GetNXCOLUMNSMOD(); x++)\r
228     {\r
229       for(int z = 0; z < fCaloConstants->GetNZROWSMOD(); z++)\r
230         {\r
231           fChannelBook[x][z] = 0;\r
232         }\r
233     }     \r
234 \r
235 }\r
236 \r
237 \r
238 void AliHLTCaloDigitMaker::AddDigit(AliHLTCaloChannelDataStruct* channelData, UShort_t* channelCoordinates, Float_t* localCoordinates)\r
239 {\r
240 \r
241   fChannelBook[channelCoordinates[0]][channelCoordinates[0]] = fDigitStructPtr;\r
242 \r
243   fDigitStructPtr->fX = channelCoordinates[0];\r
244   fDigitStructPtr->fZ = channelCoordinates[1];\r
245 \r
246   fDigitStructPtr->fLocX = localCoordinates[0];\r
247   fDigitStructPtr->fLocZ = localCoordinates[1];\r
248 \r
249   if(channelCoordinates[2] == fCaloConstants->GetHIGHGAIN() )\r
250     {\r
251       fDigitStructPtr->fEnergy = channelData->fEnergy*fHighGainFactors[channelCoordinates[0]][channelCoordinates[1]];\r
252       if(channelData->fEnergy >= 1023)\r
253         {\r
254           fDigitStructPtr->fOverflow = true;\r
255         }\r
256       //        printf("HG channel (x = %d, z = %d) with amplitude: %f --> Digit with energy: %f \n", channelCoordinates[0], channelCoordinates[1], channelData->fEnergy, fDigitStructPtr->fEnergy);\r
257     }\r
258   else\r
259     {\r
260       fDigitStructPtr->fEnergy = channelData->fEnergy*fLowGainFactors[channelCoordinates[0]][channelCoordinates[1]];\r
261       if(channelData->fEnergy >= 1023)\r
262         {\r
263           fDigitStructPtr->fOverflow = true;\r
264         }\r
265       //        printf("LG channel (x = %d, z = %d) with amplitude: %f --> Digit with energy: %f\n", channelCoordinates[0], channelCoordinates[1], channelData->fEnergy, fDigitStructPtr->fEnergy); \r
266     }\r
267   fDigitStructPtr->fTime = channelData->fTime * 0.0000001; //TODO\r
268   fDigitStructPtr->fCrazyness = channelData->fCrazyness;\r
269   fDigitStructPtr->fModule = channelCoordinates[3];\r
270   fDigitStructPtr++;\r
271 }\r
272 \r
273 bool AliHLTCaloDigitMaker::UseDigit(UShort_t *channelCoordinates, AliHLTCaloChannelDataStruct *channel) \r
274 {\r
275   AliHLTCaloDigitDataStruct *tmpDigit = fChannelBook[channelCoordinates[0]][channelCoordinates[1]];\r
276   if(tmpDigit)\r
277     {\r
278       if(channelCoordinates[2] == fCaloConstants->GetLOWGAIN())\r
279         {\r
280           if(tmpDigit->fOverflow)\r
281             {\r
282               return true;\r
283             }\r
284           return false;\r
285         }\r
286       else\r
287         {\r
288           if(channel->fEnergy >= fCaloConstants->GetMAXBINVALUE() )\r
289             {\r
290               return false;\r
291             }\r
292           return true;\r
293         }\r
294     }\r
295   return true;\r
296 }\r