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 #include "AliHLTCaloConstantsHandler.h"\r
  35 #include "AliHLTCaloMapper.h"\r
  36 #include "AliHLTCaloChannelDataStruct.h"\r
  37 #include "AliHLTCaloChannelDataHeaderStruct.h"\r
  38 #include "AliHLTCaloDigitDataStruct.h"\r
  39 #include "AliHLTCaloCoordinate.h"\r
  40 #include "AliHLTCaloSharedMemoryInterfacev2.h" // added by PTH\r
  41 //#include "AliPHOSEMCAGeometry.h"\r
  42 #include "TH2F.h"\r
  43 #include "AliHLTCaloConstants.h"\r
  44 \r
  45 ClassImp(AliHLTCaloDigitMaker);\r
  46 \r
  47 //using namespace CaloHLTConst;\r
  48 \r
  49 AliHLTCaloDigitMaker::AliHLTCaloDigitMaker(TString det) :\r
  50   AliHLTCaloConstantsHandler(det),\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 void\r
 171 AliHLTCaloDigitMaker::SetGlobalLowGainFactor(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           fLowGainFactors[x][z] = factor;\r
 179         }\r
 180     }\r
 181 }\r
 182 \r
 183 void\r
 184 AliHLTCaloDigitMaker::SetBadChannelMask(TH2F* badChannelHGHist, TH2F* badChannelLGHist, Float_t qCut)\r
 185 {\r
 186  for(int x = 0; x < fCaloConstants->GetNXCOLUMNSMOD(); x++)\r
 187     {\r
 188       for(int z = 0; z < fCaloConstants->GetNZROWSMOD(); z++)\r
 189         {\r
 190           if(badChannelHGHist->GetBinContent(x, z) < qCut && badChannelHGHist->GetBinContent(x, z) > 0)\r
 191             {\r
 192               fBadChannelMask[x][z][fCaloConstants->GetHIGHGAIN()] = true;\r
 193             }\r
 194           else\r
 195             {\r
 196               fBadChannelMask[x][z][fCaloConstants->GetHIGHGAIN()] = false;\r
 197             }\r
 198           if(badChannelLGHist->GetBinContent(x, z) < qCut && badChannelLGHist->GetBinContent(x, z) > 0)\r
 199             {\r
 200               fBadChannelMask[x][z][fCaloConstants->GetLOWGAIN()] = false;\r
 201             }\r
 202           else\r
 203             {\r
 204               fBadChannelMask[x][z][fCaloConstants->GetLOWGAIN()] = false;\r
 205             }\r
 206         }\r
 207     }\r
 208 }\r
 209 \r
 210 void\r
 211 AliHLTCaloDigitMaker::Reset()\r
 212 {\r
 213   fDigitCount = 0;\r
 214   for(int x = 0; x < fCaloConstants->GetNXCOLUMNSMOD(); x++)\r
 215     {\r
 216       for(int z = 0; z < fCaloConstants->GetNZROWSMOD(); z++)\r
 217         {\r
 218           fChannelBook[x][z] = 0;\r
 219         }\r
 220     }     \r
 221 \r
 222 }\r
 223 \r
 224 \r
 225 void AliHLTCaloDigitMaker::AddDigit(AliHLTCaloChannelDataStruct* channelData, AliHLTCaloCoordinate &coord)\r
 226 {\r
 227 \r
 228   AliHLTCaloDigitDataStruct *tmpDigit = fDigitStructPtr + 1;\r
 229 \r
 230   if(fChannelBook[coord.fX][coord.fZ])\r
 231     {\r
 232       tmpDigit = fDigitStructPtr;\r
 233       fDigitStructPtr = fChannelBook[coord.fX][coord.fZ];\r
 234       fDigitCount--;\r
 235       //      printf("Going to overwrite digit: x = %d, z = %d, gain = %d, energy = %f\n", fDigitStructPtr->fX, fDigitStructPtr->fZ, fDigitStructPtr->fGain, fDigitStructPtr->fEnergy);\r
 236     }\r
 237   \r
 238   fChannelBook[coord.fX][coord.fZ] = fDigitStructPtr;\r
 239 \r
 240  \r
 241   fDigitStructPtr->fX = coord.fX;\r
 242   fDigitStructPtr->fZ = coord.fZ;\r
 243   fDigitStructPtr->fGain = coord.fGain;\r
 244   fDigitStructPtr->fOverflow = false;\r
 245   \r
 246   fDigitStructPtr->fID = fDigitStructPtr->fZ * fCaloConstants->GetNXCOLUMNSMOD() + fDigitStructPtr->fX;\r
 247   \r
 248   if(coord.fGain == fCaloConstants->GetHIGHGAIN() )\r
 249     {\r
 250       fDigitStructPtr->fEnergy = channelData->fEnergy*fHighGainFactors[coord.fX][coord.fZ];\r
 251       if(channelData->fEnergy >= fMaxEnergy)\r
 252         {\r
 253           fDigitStructPtr->fOverflow = true;\r
 254         }\r
 255             //printf("HG channel (x = %d, z = %d) with amplitude: %f --> Digit with energy: %f \n", coord.fX, coord.fZ, channelData->fEnergy, fDigitStructPtr->fEnergy);\r
 256     }\r
 257   else\r
 258     {\r
 259       fDigitStructPtr->fEnergy = channelData->fEnergy*fLowGainFactors[coord.fX][coord.fZ];\r
 260       if(channelData->fEnergy >= 1023)\r
 261         {\r
 262           fDigitStructPtr->fOverflow = true;\r
 263         }\r
 264     //        printf("LG channel (x = %d, z = %d) with amplitude: %f --> Digit with energy: %f\n", coord.fX, coord.fZ, channelData->fEnergy, fDigitStructPtr->fEnergy); \r
 265     }\r
 266   fDigitStructPtr->fTime = channelData->fTime * 0.0000001; //TODO\r
 267   fDigitStructPtr->fCrazyness = channelData->fCrazyness;\r
 268   fDigitStructPtr->fModule = coord.fModuleId;\r
 269   fDigitStructPtr = tmpDigit;\r
 270   //  fDigitStructPtr++;\r
 271   fDigitCount++;\r
 272 }\r
 273 \r
 274 bool AliHLTCaloDigitMaker::UseDigit(AliHLTCaloCoordinate &channelCoordinates, AliHLTCaloChannelDataStruct *channel) \r
 275 {\r
 276    \r
 277    if(fBadChannelMask[channelCoordinates.fX][channelCoordinates.fZ][0] == true) return false;\r
 278    if(channel->fTime < fMinTime || channel->fTime > fMaxTime) return false;\r
 279    \r
 280   AliHLTCaloDigitDataStruct *tmpDigit = fChannelBook[channelCoordinates.fX][channelCoordinates.fZ];\r
 281   //printf("UseDigit: Got digit, x: %d, z: %d, gain: %d, amp: %f\n", channelCoordinates.fX, channelCoordinates.fZ, channelCoordinates.fGain, channel->fEnergy);\r
 282   if(tmpDigit)\r
 283     {\r
 284       if(channelCoordinates.fGain == fCaloConstants->GetLOWGAIN())\r
 285         {\r
 286                   //printf("UseDigit: Already have digit with, x: %d, z: %d, with high gain \n", channelCoordinates.fX, channelCoordinates.fZ);\r
 287           if(tmpDigit->fOverflow)\r
 288             {\r
 289                      // printf("But it was in overflow! Let's use this low gain!\n");\r
 290               return true;\r
 291             }\r
 292           return false;\r
 293         }\r
 294       else\r
 295         {\r
 296                   //printf("UseDigit: Already have digit with, x: %d, z: %d, with low gain: %d\n", channelCoordinates.fX, channelCoordinates.fZ);\r
 297           if(channel->fEnergy > fMaxEnergy )\r
 298             {\r
 299               return false;\r
 300             }\r
 301           return true;\r
 302         }\r
 303     }\r
 304   return true;\r
 305 }\r
 306 \r
 307 void AliHLTCaloDigitMaker::SetBadChannel(Int_t x, Int_t z, Bool_t bad)\r
 308 {\r
 309    // See header file for class documentation\r
 310    fBadChannelMask[x][z][0] = bad;\r
 311    fBadChannelMask[x][z][1] = bad;\r
 312 }\r
 313 \r
 314 void AliHLTCaloDigitMaker::SetGain(Int_t x, Int_t z, Float_t ratio, Float_t gain)\r
 315 {\r
 316    // See header file for class documentation\r
 317    \r
 318    fHighGainFactors[x][z] = gain;\r
 319    fLowGainFactors[x][z] = gain * ratio;\r
 320    \r
 321 }\r