ALICE3-TRK: adding getter for sensitive volumes indices in GeometryTGeo

Author: atriolo

Detectors/Upgrades/ALICE3/TRK/base/include/TRKBase/GeometryTGeo.h

@@ -71,13 +71,32 @@ class GeometryTGeo : public o2::detectors::DetMatrixCache
   void setOwner(bool v) { mOwner = v; }
 
   void Print(Option_t* opt = "") const;
+  void PrintChipID(int index, int subDetID, int petalcase, int disk, int lay, int stave, int indexRetrieved) const;
 
   int getLayer(int index) const;
   int getStave(int index) const;
   int getSubDetID(int index) const;
   int getPetalCase(int index) const;
   int getDisk(int index) const;
 
+  /// This routine computes the chip index number from the subDetID, petal, disk, layer, stave /// TODO: retrieve also from chip when chips will be available
+  /// in substave
+  /// \param int subDetID The subdetector ID, 0 for VD, 1 for MLOT
+  /// \param int petalcase The petal case number for VD, from 0 to 3
+  /// \param int disk The disk number for VD, from 0 to 5
+  /// \param int lay The layer number. Starting from 0 both for VD and MLOT
+  /// \param int stave The stave number for MLOT. Starting from 0
+  int getChipIndex(int subDetID, int petalcase, int disk, int lay, int stave) const;
+
+  /// This routine computes subDetID, petal, disk, layer, stave given the chip index number /// TODO: copute also from chip when chips will be available
+  /// \param int index The chip index number, starting from 0
+  /// \param int subDetID The subdetector ID, 0 for VD, 1 for MLOT
+  /// \param int petalcase The petal case number for VD, from 0 to 3
+  /// \param int disk The disk number for VD, from 0 to 5
+  /// \param int lay The layer number. Starting from 0 both for VD and MLOT
+  /// \param int stave The stave number for MLOT. Starting from 0
+  bool getChipID(int index, int& subDetID, int& petalcase, int& disk, int& lay, int& stave) const;
+
   int getLastChipIndex(int lay) const { return mLastChipIndex[lay]; }
   int getFirstChipIndex(int lay, int petalcase, int subDetID) const
   {

Detectors/Upgrades/ALICE3/TRK/base/src/GeometryTGeo.cxx

@@ -150,8 +150,7 @@ int GeometryTGeo::getLayer(int index) const
     if (index % mNumberOfChipsPerPetalVD[petalcase] >= mNumberOfLayersVD) {
       return -1; /// disks
     }
-    lay = index % mNumberOfChipsPerPetalVD[petalcase];
-    // return index % mNumberOfChipsPerPetalVD[petalcase];
+    return index % mNumberOfChipsPerPetalVD[petalcase];
   } else if (subDetID == 1) { /// MLOT
     while (index > mLastChipIndex[lay]) {
       lay++;
@@ -194,24 +193,43 @@ int GeometryTGeo::getDisk(int index) const
 }
 
 //__________________________________________________________________________
-TString GeometryTGeo::getMatrixPath(int index) const
+int GeometryTGeo::getChipIndex(int subDetID, int petalcase, int disk, int lay, int stave) const
 {
+  if (subDetID == 0) { // VD
+    if (lay == -1) {   // disk
+      return getFirstChipIndex(lay, petalcase, subDetID) + mNumberOfLayersVD + disk;
+    } else { // layer
+      return getFirstChipIndex(lay, petalcase, subDetID) + lay;
+    }
+  } else if (subDetID == 1) { // MLOT
+    return getFirstChipIndex(lay, petalcase, subDetID) + stave;
+  }
+  return -1; // not found
+}
 
-  // int lay, hba, stav, sstav, mod, chipInMod;
-  int subDetID, petalcase, disk, lay, stave; /// TODO: add chips in a second step
+//__________________________________________________________________________
+bool GeometryTGeo::getChipID(int index, int& subDetID, int& petalcase, int& disk, int& lay, int& stave) const
+{
   subDetID = getSubDetID(index);
   petalcase = getPetalCase(index);
-  lay = getLayer(index);
   disk = getDisk(index);
+  lay = getLayer(index);
   stave = getStave(index);
 
-  // std::cout << "\nindex = " << index << std::endl;
-  // std::cout << "subDetID = " << subDetID << std::endl;
-  // std::cout << "petalcase = " << petalcase << std::endl;
-  // std::cout << "layer = " << lay << std::endl;
-  // std::cout << "disk = " << disk << std::endl;
-  // std::cout << "first chip index = " << getFirstChipIndex(lay, petalcase, subDetID) << std::endl;
-  // std::cout << "stave = " << stave << std::endl;
+  return kTRUE;
+}
+
+//__________________________________________________________________________
+TString GeometryTGeo::getMatrixPath(int index) const
+{
+
+  // int lay, hba, stav, sstav, mod, chipInMod;
+  int subDetID, petalcase, disk, lay, stave; //// TODO: add chips in a second step
+  getChipID(index, subDetID, petalcase, disk, lay, stave);
+
+  int indexRetrieved = getChipIndex(subDetID, petalcase, disk, lay, stave);
+
+  PrintChipID(index, subDetID, petalcase, disk, lay, stave, indexRetrieved);
 
   // TString path = Form("/cave_1/barrel_1/%s_2/", GeometryTGeo::getTRKVolPattern());
   TString path = "/cave_1/barrel_1/TRKV_2/TRKLayer0_1/TRKStave0_1/TRKChip0_1/TRKSensor0_1/"; /// dummy path, to be replaced
@@ -591,6 +609,20 @@ int GeometryTGeo::extractNumberOfStavesMLOT(int lay) const
   return numberOfStaves;
 }
 
+//__________________________________________________________________________
+void GeometryTGeo::PrintChipID(int index, int subDetID, int petalcase, int disk, int lay, int stave, int indexRetrieved) const
+{
+  std::cout << "\nindex = " << index << std::endl;
+  std::cout << "subDetID = " << subDetID << std::endl;
+  std::cout << "petalcase = " << petalcase << std::endl;
+  std::cout << "layer = " << lay << std::endl;
+  std::cout << "disk = " << disk << std::endl;
+  std::cout << "first chip index = " << getFirstChipIndex(lay, petalcase, subDetID) << std::endl;
+  std::cout << "stave = " << stave << std::endl;
+  std::cout << "chck index Retrieved = " << indexRetrieved << std::endl;
+}
+
+//__________________________________________________________________________
 void GeometryTGeo::Print(Option_t*) const
 {
   if (!isBuilt()) {