Implement Dummy k4ActsAlgorithm that instantiates a DD4hepPlugin looping over surfaces

Author: andresailer

CMakeLists.txt

@@ -6,6 +6,7 @@ find_package(ROOT COMPONENTS RIO Tree)
 find_package(EDM4HEP)
 find_package(k4FWCore)
 find_package(Acts)
+find_package(DD4hep)
 
 #---------------------------------------------------------------
 # Load macros and functions for Gaudi-based projects

README.md

@@ -18,15 +18,15 @@ This repository contains the necessary tools to use ACTS functionality in Key4he
 
 * k4FWCore
 
-## Installation
+## Compilation and testing
 
 
 ```
+source /cvmfs/sw-nightlies.hsf.org/key4hep/setup.sh
 mkdir build install
 cd build;
-cmake .. -DCMAKE_INSTALL_PREFIX=../install
-make install
-
-
+cmake .. -DCMAKE_INSTALL_PREFIX=../install -G Ninja
+ninja
+ctest
 ```
 

k4ActsTracking/CMakeLists.txt

@@ -1,16 +1,23 @@
 file(GLOB _plugin_sources src/components/*.cpp)
 gaudi_add_module(k4ActsTrackingPlugins
   SOURCES ${_plugin_sources}
-  LINK Gaudi::GaudiKernel Gaudi::GaudiAlgLib k4FWCore::k4FWCore EDM4HEP::edm4hep)
+  LINK Gaudi::GaudiKernel Gaudi::GaudiAlgLib DD4hep::DDCore)
 
-set(GAUDI_GENCONF_DIR "genConfDir")
+
+file(GLOB _dd4hep_plugin_sources src/dd4hepplugin/*.cpp)
+add_dd4hep_plugin(k4ActsDD4hepPlugin SHARED ${_dd4hep_plugin_sources})
+target_link_libraries(k4ActsDD4hepPlugin DD4hep::DDCore DD4hep::DDRec  DD4hep::DDParsers)
 
 function(set_test_env _testname)
   set_property(TEST ${_testname} APPEND PROPERTY ENVIRONMENT
-    LD_LIBRARY_PATH=${CMAKE_BINARY_DIR}:$<TARGET_FILE_DIR:k4ActsTrackingPlugins>:$<TARGET_FILE_DIR:ROOT::Core>:$<TARGET_FILE_DIR:k4FWCore::k4FWCore>:$<TARGET_FILE_DIR:EDM4HEP::edm4hep>:$<TARGET_FILE_DIR:podio::podio>:$ENV{LD_LIBRARY_PATH}
-    PYTHONPATH=${CMAKE_BINARY_DIR}/${CMAKE_PROJECT_NAME}/${GAUDI_GENCONF_DIR}:$<TARGET_FILE_DIR:k4FWCore::k4FWCore>/../python:$ENV{PYTHONPATH}
+    LD_LIBRARY_PATH=${CMAKE_BINARY_DIR}:$<TARGET_FILE_DIR:k4ActsTrackingPlugins>:$<TARGET_FILE_DIR:k4ActsDD4hepPlugin>:$<TARGET_FILE_DIR:ROOT::Core>:$<TARGET_FILE_DIR:k4FWCore::k4FWCore>:$<TARGET_FILE_DIR:EDM4HEP::edm4hep>:$<TARGET_FILE_DIR:podio::podio>:$ENV{LD_LIBRARY_PATH}
+    PYTHONPATH=${CMAKE_BINARY_DIR}/${CMAKE_PROJECT_NAME}/genConf:$<TARGET_FILE_DIR:k4FWCore::k4FWCore>/../python:$ENV{PYTHONPATH}
     PATH=$<TARGET_FILE_DIR:k4FWCore::k4FWCore>/../bin:$ENV{PATH}
     K4ACTSTRACKING=${CMAKE_CURRENT_LIST_DIR}/
     )
 endfunction()
 
+add_test(NAME ActsAlgTest
+  WORKING_DIRECTORY ${PROJECT_SOURCE_DIR}
+  COMMAND k4run k4ActsTracking/options/run_acts_plugin.py)
+set_test_env(ActsAlgTest)

k4ActsTracking/options/run_acts_plugin.py

@@ -0,0 +1,21 @@
+import os
+from pprint import pprint
+from Gaudi.Configuration import *
+
+from Configurables import ActsAlg, GeoSvc
+algList = []
+
+geosvc = GeoSvc("GeoSvc")
+# using this instead of CLIC_o3_v14 because it is much faster to instantiate
+geosvc.detectors = [os.environ["LCGEO"]+"/CLIC/compact/CLIC_o2_v04/CLIC_o2_v04.xml"]
+
+a = ActsAlg("MyActsAlg")
+algList.append(a)
+
+from Configurables import ApplicationMgr
+ApplicationMgr( TopAlg = algList,
+                EvtSel = 'NONE',
+                EvtMax = 2,
+                ExtSvc = [geosvc],
+                OutputLevel=INFO
+              )

k4ActsTracking/src/components/ActsAlg.cpp

@@ -0,0 +1,30 @@
+#include "ActsAlg.h"
+
+#include <DD4hep/DetFactoryHelper.h>
+#include <DD4hep/Factories.h>
+
+DECLARE_COMPONENT(ActsAlg)
+
+ActsAlg::ActsAlg(const std::string& aName, ISvcLocator* aSvcLoc) : GaudiAlgorithm(aName, aSvcLoc) {}
+
+ActsAlg::~ActsAlg() {}
+
+StatusCode ActsAlg::initialize() {
+  info() << "Initializing the ACTS algorithm" << endmsg;
+  int                 argc = 0;
+  char*               argv = {};
+  dd4hep::PluginDebug dbg;
+  std::string         name   = "k4acts_addActsExtensions";
+  long                result = dd4hep::PluginService::Create<long>(name, &dd4hep::Detector::getInstance(), argc, &argv);
+  if (0 == result) {
+    error() << "Failed to locate plugin " << name << "\n" << dbg.missingFactory(name) << endmsg;
+    return StatusCode::FAILURE;
+  }
+
+  info() << "Successfully loaded the plugin!" << endmsg;
+  return StatusCode::SUCCESS;
+}
+
+StatusCode ActsAlg::execute() { return StatusCode::SUCCESS; }
+
+StatusCode ActsAlg::finalize() { return StatusCode::SUCCESS; }

k4ActsTracking/src/components/EmptyAlg.h renamed to k4ActsTracking/src/components/ActsAlg.h

@@ -4,10 +4,10 @@
 #include "Gaudi/Property.h"
 #include "GaudiAlg/GaudiAlgorithm.h"
 
-class EmptyAlg : public GaudiAlgorithm {
+class ActsAlg : public GaudiAlgorithm {
 public:
-  explicit EmptyAlg(const std::string&, ISvcLocator*);
-  virtual ~EmptyAlg();
+  explicit ActsAlg(const std::string&, ISvcLocator*);
+  virtual ~ActsAlg();
   /**  Initialize.
    *   @return status code
    */

k4ActsTracking/src/components/EmptyAlg.cpp

@@ -0,0 +1,57 @@
+#include <DD4hep/DetElement.h>
+#include <DD4hep/Detector.h>
+#include <DD4hep/Factories.h>
+#include <DD4hep/Printout.h>
+
+#include <DDRec/DetectorData.h>
+#include <DDRec/SurfaceHelper.h>
+
+#include <string>
+
+using dd4hep::DetElement;
+using dd4hep::PrintLevel;
+
+namespace {
+
+  /** Plugin for adding the ACTS Extensions to surface, or DetElements, or something
+   *
+   */
+
+  static long addActsExtensions(dd4hep::Detector& description, int argc, char** argv) {
+    const std::string LOG_SOURCE("AddActsExtensions");
+    printout(PrintLevel::INFO, LOG_SOURCE, "Running extension");
+
+    dd4hep::rec::SurfaceHelper      ds(description.world());
+    dd4hep::rec::SurfaceList const& detSL = ds.surfaceList();
+
+    for (dd4hep::rec::ISurface* surf : detSL) {
+      dd4hep::Volume volume     = ((dd4hep::rec::Surface*)surf)->volume();
+      auto*          ddsurf     = ((dd4hep::rec::Surface*)surf);
+      auto const     volumeName = std::string(volume->GetName());
+      if (not ddsurf->detElement().isValid())
+        continue;
+      std::string path = ddsurf->detElement().path();
+
+      //FIXME: replace this with the ACTS type to be used as an extension
+      dd4hep::rec::DoubleParameters* para = nullptr;
+      try {  // use existing map, or create a new one
+        para                                    = ddsurf->detElement().extension<dd4hep::rec::DoubleParameters>();
+        para->doubleParameters["SortingPolicy"] = 123;
+      } catch (...) {
+        para                                    = new dd4hep::rec::DoubleParameters;
+        para->doubleParameters["SortingPolicy"] = 321;
+        ddsurf->detElement().addExtension<dd4hep::rec::DoubleParameters>(para);
+      }
+      printout(PrintLevel::INFO, LOG_SOURCE,
+               "Added extension to %s"
+               " path %s, type %s, value %3.5f",
+               volumeName.c_str(), ddsurf->detElement().path().c_str(), ddsurf->detElement().type().c_str(),
+               para->doubleParameters["SortingPolicy"]);
+    }  // for all surfaces
+
+    return 0;  // this return value isn't really returned from the plugin service
+  }
+
+}  // namespace
+
+DECLARE_APPLY(k4acts_addActsExtensions, ::addActsExtensions)