Use GeoSvc to get dd4hep geo

key4hep · Nov 28, 2023 · 1de96da · 1de96da
1 parent a627e58
commit 1de96da
Show file tree

Hide file tree

Showing 4 changed files with 30 additions and 57 deletions.
diff --git a/k4ActsTracking/src/components/ActsGeoSvc.cpp b/k4ActsTracking/src/components/ActsGeoSvc.cpp
@@ -28,38 +28,22 @@
 #include "DD4hep/Printout.h"
 #include "GaudiKernel/Service.h"
 #include "TGeoManager.h"
+#include "k4Interface/IGeoSvc.h"
 
 using namespace Gaudi;
 
 DECLARE_COMPONENT(ActsGeoSvc)
 
 ActsGeoSvc::ActsGeoSvc(const std::string& name, ISvcLocator* svc) : base_class(name, svc), m_log(msgSvc(), name) {}
 
-ActsGeoSvc::~ActsGeoSvc() {
-  if (m_dd4hepGeo != nullptr) {
-    try {
-      m_dd4hepGeo->destroyInstance();
-      m_dd4hepGeo = nullptr;
-    } catch (...) {
-    }
-  }
-}
+ActsGeoSvc::~ActsGeoSvc() {};
 
 StatusCode ActsGeoSvc::initialize() {
-  if (m_xmlFileNames.size() > 0) {
-    m_log << MSG::INFO << "Loading xml files from:  '" << m_xmlFileNames << "'" << endmsg;
-  } else {
-    m_log << MSG::ERROR << "No xml file!" << endmsg;
-    return StatusCode::FAILURE;
-  }
 
-  /// Check if the DD4Hep Geometry is built successfully
-  if (buildDD4HepGeo().isFailure()) {
-    m_log << MSG::ERROR << "Could not build DD4Hep geometry" << endmsg;
-    return StatusCode::FAILURE;
-  } else {
-    m_log << MSG::INFO << "DD4Hep geometry SUCCESSFULLY built" << endmsg;
-  }
+  // TODO: get dd4hep geo
+  m_dd4hepGeo = svcLocator()->service<IGeoSvc>(m_geoSvcName)->getDetector();
+  // necessary?
+  // m_dd4hepGeo->addExtension<IActsGeoSvc>(this);
 
   Acts::BinningType bTypePhi              = Acts::equidistant;
   Acts::BinningType bTypeR                = Acts::equidistant;
@@ -96,21 +80,6 @@ StatusCode ActsGeoSvc::execute() { return StatusCode::SUCCESS; }
 
 StatusCode ActsGeoSvc::finalize() { return StatusCode::SUCCESS; }
 
-StatusCode ActsGeoSvc::buildDD4HepGeo() {
-  m_dd4hepGeo = &(dd4hep::Detector::getInstance());
-  m_dd4hepGeo->addExtension<IActsGeoSvc>(this);
-
-  /// Load geometry
-  for (auto& filename : m_xmlFileNames) {
-    m_log << MSG::INFO << "Loading geometry from file:  '" << filename << "'" << endmsg;
-    m_dd4hepGeo->fromCompact(filename);
-  }
-  m_dd4hepGeo->volumeManager();
-  m_dd4hepGeo->apply("DD4hepVolumeManager", 0, nullptr);
-
-  return StatusCode::SUCCESS;
-}
-
 /// Create a geometry OBJ file
 StatusCode ActsGeoSvc::createGeoObj() {
   // Convert DD4Hep geometry to acts

diff --git a/k4ActsTracking/src/components/ActsGeoSvc.h b/k4ActsTracking/src/components/ActsGeoSvc.h
@@ -17,8 +17,8 @@
  * limitations under the License.
  */
 
-#ifndef GEOSVC_H
-#define GEOSVC_H
+#ifndef ACTSGEOSVC_H
+#define ACTSGEOSVC_H
 
 #include <Acts/Material/IMaterialDecorator.hpp>
 #include "Acts/Definitions/Common.hpp"
@@ -62,8 +62,7 @@ class ActsGeoSvc : public extends<Service, IActsGeoSvc> {
   /// ACTS surface lookup container for hit surfaces that generate smeared hits
   VolumeSurfaceMap m_surfaces;
 
-  /// XML-files with the detector description
-  Gaudi::Property<std::vector<std::string>> m_xmlFileNames{this, "detectors", {}, "Detector descriptions XML-files"};
+  Gaudi::Property<std::string> m_geoSvcName{this, "GeoSvcName", "GeoSvc", "The name of the GeoSvc instance"};
 
   /// Option for the Debug Geometry
   Gaudi::Property<bool> m_debugGeometry{this, "debugGeometry", false, "Option for geometry debugging"};
@@ -84,8 +83,6 @@ class ActsGeoSvc : public extends<Service, IActsGeoSvc> {
 
   virtual StatusCode finalize() final;
 
-  StatusCode buildDD4HepGeo();
-
   StatusCode createGeoObj();
 
   virtual const Acts::TrackingGeometry& trackingGeometry() const;

diff --git a/k4ActsTracking/src/components/IActsGeoSvc.h b/k4ActsTracking/src/components/IActsGeoSvc.h
@@ -17,8 +17,8 @@
  * limitations under the License.
  */
 
-#ifndef IGEOSVC_H
-#define IGEOSVC_H
+#ifndef IACTSGEOSVC_H
+#define IACTSGEOSVC_H
 
 #include <GaudiKernel/IService.h>
 #include <unordered_map>
@@ -46,4 +46,4 @@ class GAUDI_API IActsGeoSvc : virtual public IService {
   virtual ~IActsGeoSvc() {}
 };
 
-#endif  // IGEOSVC_H
+#endif  // IACTSGEOSVC_H
diff --git a/test/options/geosvc.py b/test/options/geosvc.py
@@ -17,19 +17,26 @@
 # limitations under the License.
 #
 import os
-from pprint import pprint
-from Gaudi.Configuration import *
+from Gaudi.Configuration import INFO
 
-from Configurables import ActsGeoSvc
+from Configurables import ActsGeoSvc, ApplicationMgr, GeoSvc
 
 algList = []
 
-a = ActsGeoSvc("ActsGeoSvc")
-a.detectors = [f"{os.environ['OPENDATADETECTOR']}/xml/OpenDataDetector.xml"]
-a.debugGeometry = True
-a.outputFileName = "MyObjFile"
-from Configurables import ApplicationMgr
+dd4hep_geo = GeoSvc("GeoSvc")
+dd4hep_geo.detectors = [f"{os.environ['OPENDATADETECTOR']}/xml/OpenDataDetector.xml"]
+dd4hep_geo.EnableGeant4Geo = False
 
-ApplicationMgr(TopAlg=algList, EvtSel="NONE", EvtMax=2, ExtSvc=[
-a
-], OutputLevel=INFO)
+acts_geo = ActsGeoSvc("ActsGeoSvc")
+acts_geo.GeoSvcName = dd4hep_geo.name()
+acts_geo.debugGeometry = True
+acts_geo.outputFileName = "MyObjFile"
+
+ApplicationMgr(
+    TopAlg=algList,
+    EvtSel="NONE",
+    EvtMax=2,
+    # order dependent...
+    ExtSvc=[dd4hep_geo, acts_geo],
+    OutputLevel=INFO,
+)