TPC: add possibility to load Krypton Gain map via Params for residual calibration

wiechula · davidrohr · commit 41cf2696b014 · 2021-12-14T09:34:57.000+01:00
diff --git a/Detectors/TPC/reconstruction/include/TPCReconstruction/KrBoxClusterFinderParam.h b/Detectors/TPC/reconstruction/include/TPCReconstruction/KrBoxClusterFinderParam.h
@@ -17,6 +17,8 @@
 #ifndef ALICEO2_TPC_KrBoxClusterFinderParam_H_
 #define ALICEO2_TPC_KrBoxClusterFinderParam_H_
 
+#include <string>
+
 #include "CommonUtils/ConfigurableParam.h"
 #include "CommonUtils/ConfigurableParamHelper.h"
 
@@ -56,6 +58,9 @@ struct KrBoxClusterFinderParam : public o2::conf::ConfigurableParamHelper<KrBoxC
   unsigned char CutMaxSize{255}; ///< Max cluster size in number of digits
   bool ApplyCuts{false};         ///< if to apply cluster cuts above
 
+  std::string GainMapFile{};          ///< gain map file to apply during reconstruction
+  std::string GainMapName{"GainMap"}; ///< gain map file to apply during reconstruction
+
   O2ParamDef(KrBoxClusterFinderParam, "TPCKrBoxClusterFinder");
 };
 } // namespace tpc
diff --git a/Detectors/TPC/reconstruction/src/KrBoxClusterFinder.cxx b/Detectors/TPC/reconstruction/src/KrBoxClusterFinder.cxx
@@ -72,7 +72,7 @@ void KrBoxClusterFinder::fillADCValueInLastSlice(int cru, int rowInSector, int p
   int padNum = mMapperInstance.globalPadNumber(PadPos(rowInSector, padInRow));
   float correctionFactor = correctionFactorCalDet->getValue(mSector, padNum);
 
-  if (correctionFactor == 0) {
+  if (correctionFactor <= 0) {
     LOGP(warning, "Encountered correction factor which is zero.");
     LOGP(warning, "Digit will be set to 0!");
     adcValue = 0;
@@ -158,6 +158,11 @@ void KrBoxClusterFinder::init()
   mCutQtotSizeSlope = param.CutQtotSizeSlope;
   mCutMaxSize = param.CutMaxSize;
   mApplyCuts = param.ApplyCuts;
+
+  if (param.GainMapFile.size()) {
+    LOGP(info, "loading gain map '{}' from file {}", param.GainMapName, param.GainMapFile);
+    loadGainMapFromFile(param.GainMapFile, param.GainMapName);
+  }
 }
 
 //#################################################
