make the map loading funciton generic

Author: GiorgioAlbertoLucia

Detectors/Upgrades/ALICE3/IOTOF/simulation/include/IOTOFSimulation/Digitizer.h

@@ -92,8 +92,8 @@ class Digitizer : public TObject
   int energyToCharge(float energyLoss) const;
 
   /// Load the efficiency map from a file
-  void loadEfficiencyMap(const std::string& filePath);
-  void loadEfficiencyMapFromCCDB(const std::string& ccdbPath);
+  void loadMap(TH2* map, const std::string& filePath, const char* mapName);
+  void loadMapFromCCDB(TH2* map, const std::string& ccdbPath, const char* mapName);
 
   /// Check if the hit passes efficiency cut
   /// \param x Detector local coordinate x in cm with respect to the center of the sensitive volume.
@@ -119,6 +119,8 @@ class Digitizer : public TObject
 
   const o2::iotof::GeometryTGeo* mGeometry = nullptr; ///< IOTOF geometry
   TH2D* mEfficiencyMap = nullptr;                     ///< Efficiency map for the detector
+  TH2D* mResolutionMap = nullptr;                     ///< Resolution map for the detector
+  TH2D* mTimeOfArrivalMap = nullptr;                  ///< Time of arrival map for the detector
 
   std::vector<o2::iotof::Chip> mChips;                                               //! Chips in the detector, indexed by chip ID
   std::deque<std::unique_ptr<std::vector<o2::iotof::McLabelRef>>> mExtraLabelBuffer; //! buffer for multiple mc labels to the same pixel

Detectors/Upgrades/ALICE3/IOTOF/simulation/src/Digitizer.cxx

@@ -65,10 +65,10 @@ void Digitizer::init()
 
   if (!digitizerParams.efficiencyCcdbPath.empty()) {
     LOG(info) << "Loading efficiency map from CCDB: " << digitizerParams.efficiencyCcdbPath;
-    loadEfficiencyMapFromCCDB(digitizerParams.efficiencyCcdbPath);
+    loadMapFromCCDB(mEfficiencyMap, digitizerParams.efficiencyCcdbPath, "mEfficiencyMap");
   } else if (!digitizerParams.efficiencyFilePath.empty()) {
     LOG(info) << "Loading efficiency map from file: " << digitizerParams.efficiencyFilePath;
-    loadEfficiencyMap(digitizerParams.efficiencyFilePath);
+    loadMap(mEfficiencyMap, digitizerParams.efficiencyFilePath, "mEfficiencyMap");
   } else {
     LOG(info) << "No efficiency map provided, using uniform efficiency: " << digitizerParams.efficiency * 100 << " %";
   }
@@ -346,7 +346,7 @@ int Digitizer::energyToCharge(float energyLoss) const
 
 //_______________________________________________________________________
 
-void Digitizer::loadEfficiencyMap(const std::string& filePath)
+void Digitizer::loadMap(TH1D* map, const std::string& filePath, const char* mapName)
 {
   // Load the efficiency map from a file
   TFile* file = TFile::Open(filePath.c_str());
@@ -367,13 +367,13 @@ void Digitizer::loadEfficiencyMap(const std::string& filePath)
     file->Close();
     return;
   }
-  mEfficiencyMap = dynamic_cast<TH2D*>(rawMap->Clone("mEfficiencyMap"));
-  mEfficiencyMap->SetDirectory(nullptr); // Detach from file to avoid deletion when file is closed
+  map = dynamic_cast<TH2D*>(rawMap->Clone(mapName));
+  map->SetDirectory(nullptr); // Detach from file to avoid deletion when file is closed
 
   file->Close();
 }
 
-void Digitizer::loadEfficiencyMapFromCCDB(const std::string& ccdbPath)
+void Digitizer::loadMapFromCCDB(TH1D* map, const std::string& ccdbPath, const char* mapName)
 {
   // Load the efficiency map from CCDB
   auto rawMap = o2::ccdb::BasicCCDBManager::instance().get<TH2D>(ccdbPath);
@@ -386,8 +386,8 @@ void Digitizer::loadEfficiencyMapFromCCDB(const std::string& ccdbPath)
     LOG(info) << "Efficiency map dimensions: " << rawMap->GetNbinsX() << " x " << rawMap->GetNbinsY();
   }
 
-  mEfficiencyMap = dynamic_cast<TH2D*>(rawMap->Clone("mEfficiencyMap"));
-  mEfficiencyMap->SetDirectory(nullptr); // Detach from file to avoid deletion when file is closed
+  map = dynamic_cast<TH2D*>(rawMap->Clone(mapName));
+  map->SetDirectory(nullptr); // Detach from file to avoid deletion when file is closed
 
   LOG(info) << "Successfully loaded efficiency map from CCDB path: " << ccdbPath;
 }