Port missing getMatrix method from aliroot

shahor02 · shahor02 · commit bae8e1d1517f · 2021-02-23T00:36:42.000+01:00
diff --git a/Detectors/Base/include/DetectorsBase/GeometryManager.h b/Detectors/Base/include/DetectorsBase/GeometryManager.h
@@ -57,6 +57,7 @@ class GeometryManager : public TObject
   ///< performed before alignment.
   static Bool_t getOriginalMatrix(o2::detectors::DetID detid, int sensid, TGeoHMatrix& m);
   static Bool_t getOriginalMatrix(const char* symname, TGeoHMatrix& m);
+  static TGeoHMatrix* getMatrix(const char* symname);
   static const char* getSymbolicName(o2::detectors::DetID detid, int sensid);
   static TGeoPNEntry* getPNEntry(o2::detectors::DetID detid, Int_t sensid);
   static TGeoHMatrix* getMatrix(o2::detectors::DetID detid, Int_t sensid);
@@ -114,7 +115,7 @@ class GeometryManager : public TObject
  private:
   /// Default constructor
   GeometryManager() = default;
-
+  static TGeoHMatrix* getMatrix(TGeoPNEntry* pne);
   static void accountMaterial(const TGeoMaterial* material, MatBudgetExt& bd);
   static void accountMaterial(const TGeoMaterial* material, o2::base::MatBudget& bd)
   {
diff --git a/Detectors/Base/src/GeometryManager.cxx b/Detectors/Base/src/GeometryManager.cxx
@@ -118,6 +118,47 @@ Bool_t GeometryManager::getOriginalMatrixFromPath(const char* path, TGeoHMatrix&
   return kTRUE;
 }
 
+//______________________________________________________________________
+TGeoHMatrix* GeometryManager::getMatrix(TGeoPNEntry* pne)
+{
+  // Get the global transformation matrix for a given PNEntry
+  // by quering the TGeoManager
+
+  if (!gGeoManager || !gGeoManager->IsClosed()) {
+    LOG(ERROR) << "Can't get the global matrix! gGeoManager doesn't exist or it is still opened!";
+    return nullptr;
+  }
+
+  // if matrix already known --> return it
+  TGeoPhysicalNode* pnode = pne->GetPhysicalNode();
+  if (pnode) {
+    return pnode->GetMatrix();
+  }
+
+  // otherwise calculate it from title and attach via TGeoPhysicalNode
+  pne->SetPhysicalNode(new TGeoPhysicalNode(pne->GetTitle()));
+  return pne->GetPhysicalNode()->GetMatrix();
+}
+
+//______________________________________________________________________
+TGeoHMatrix* GeometryManager::getMatrix(const char* symname)
+{
+  // Get the global transformation matrix for a given alignable volume
+  //  identified by its symbolic name 'symname' by quering the TGeoManager
+
+  if (!gGeoManager || !gGeoManager->IsClosed()) {
+    LOG(ERROR) << "No active geometry or geometry not yet closed!";
+    return nullptr;
+  }
+
+  TGeoPNEntry* pne = gGeoManager->GetAlignableEntry(symname);
+  if (!pne) {
+    return nullptr;
+  }
+
+  return getMatrix(pne);
+}
+
 //______________________________________________________________________
 const char* GeometryManager::getSymbolicName(DetID detid, int sensid)
 {
