A modular silicon (vertex) digitizer, acting on pixel-level

CMakeLists.txt

@@ -55,6 +55,7 @@ endfunction()
 add_subdirectory(DCHdigi)
 add_subdirectory(ARCdigi)
 add_subdirectory(VTXdigi)
+add_subdirectory(VTXdigi_Modular)
 add_subdirectory(VTXdigiDetailed)
 set(CMAKE_MODULE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/cmake ${CMAKE_MODULE_PATH})
 add_subdirectory(Tracking)

Makefile

@@ -11,6 +11,25 @@ make:
 	printf "#!/bin/bash\nif [ -n \"\$$KEY4HEP_STACK\" ];\nthen\n  echo '----> Info: Key4hep stack already set up. Skipping...'\nelse\n source /cvmfs/sw-nightlies.hsf.org/key4hep/setup.sh\nfi\nexport LD_LIBRARY_PATH=${CURDIR}/install/lib:${CURDIR}/install/lib64:\$$LD_LIBRARY_PATH\nexport PYTHONPATH=${CURDIR}/install/python:\$$PYTHONPATH\n" > ${CURDIR}/setup.sh ; \
 	chmod +x ${CURDIR}/setup.sh
 
+.PHONY: install
+install:
+	cd build ; \
+	make install -j8 ; \
+	cd .. ; \
+	export LD_LIBRARY_PATH=${CURDIR}/install/lib:${CURDIR}/install/lib64:$$LD_LIBRARY_PATH ; \
+	export PYTHONPATH=${CURDIR}/install/python:$$PYTHONPATH ; \
+	printf "#!/bin/bash\nif [ -n \"\$$KEY4HEP_STACK\" ];\nthen\n  echo '----> Info: Key4hep stack already set up. Skipping...'\nelse\n source /cvmfs/sw-nightlies.hsf.org/key4hep/setup.sh\nfi\nexport LD_LIBRARY_PATH=${CURDIR}/install/lib:${CURDIR}/install/lib64:\$$LD_LIBRARY_PATH\nexport PYTHONPATH=${CURDIR}/install/python:\$$PYTHONPATH\n" > ${CURDIR}/setup.sh ; \
+	chmod +x ${CURDIR}/setup.sh
+
+debug:
+	cd build ; \
+	make install -j8 -d ; \
+	cd .. ; \
+	export LD_LIBRARY_PATH=${CURDIR}/install/lib:${CURDIR}/install/lib64:$$LD_LIBRARY_PATH ; \
+	export PYTHONPATH=${CURDIR}/install/python:$$PYTHONPATH ; \
+	printf "#!/bin/bash\nif [ -n \"\$$KEY4HEP_STACK\" ];\nthen\n  echo '----> Info: Key4hep stack already set up. Skipping...'\nelse\n source /cvmfs/sw-nightlies.hsf.org/key4hep/setup.sh\nfi\nexport LD_LIBRARY_PATH=${CURDIR}/install/lib:${CURDIR}/install/lib64:\$$LD_LIBRARY_PATH\nexport PYTHONPATH=${CURDIR}/install/python:\$$PYTHONPATH\n" > ${CURDIR}/setup.sh ; \
+	chmod +x ${CURDIR}/setup.sh
+
 .PHONY: clean
 clean:
 	@ (rm -r build install && rm setup.sh) || true

README.md

@@ -74,6 +74,7 @@ make get_data
 * `ARCdigi`: ARC digitization (for now, this step produces 'reco' collection)
 * `VTXdigi`: vertex detector digitization (for now, this step produces 'reco' collection)
 * `VTXdigiDetailed`: vertex detector and silicon sensors tracker detector digitization with detailed charge readout (for now, this step produces 'reco' collection)
+* `VTXdigi_Modular`: silicon pixel digitizer, shares charge deposition across neighboring pixels. Supports different charge-sharing implementations, mainly lookup table-based implementation for now. Envisioned to be merged with the drift-bases implementation from `VTXdigiDetailed` in the future. (for now, this step produces 'reco' collection) 
 * `Tracking`: tracking algorithms orchestrating [GenFit](https://github.com/GenFit/GenFit)
 
 ## Execute Examples

VTXdigi_Modular/CMakeLists.txt

@@ -0,0 +1,56 @@
+set(PackageName VTXdigi_Modular)
+
+project(${PackageName})
+
+file(GLOB sources
+    ${PROJECT_SOURCE_DIR}/src/*.cpp
+)
+
+file(GLOB headers
+  ${PROJECT_SOURCE_DIR}/include/*.h
+)
+
+gaudi_add_module(${PackageName}
+  SOURCES ${sources}
+  LINK
+  Gaudi::GaudiKernel
+  EDM4HEP::edm4hep
+  k4FWCore::k4FWCore
+  k4FWCore::k4Interface
+  DD4hep::DDRec
+)
+
+target_include_directories(${PackageName} PUBLIC
+  $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/include>
+  $<INSTALL_INTERFACE:include>
+)
+
+set_target_properties(${PackageName} PROPERTIES PUBLIC_HEADER "${headers}")
+
+file(GLOB scripts
+  ${PROJECT_SOURCE_DIR}/test/*.py
+)
+
+file(COPY ${scripts} DESTINATION ${CMAKE_CURRENT_BINARY_DIR}/test)
+
+install(TARGETS ${PackageName}
+  EXPORT ${CMAKE_PROJECT_NAME}Targets
+  RUNTIME DESTINATION "${CMAKE_INSTALL_BINDIR}" COMPONENT bin
+  LIBRARY DESTINATION "${CMAKE_INSTALL_LIBDIR}" COMPONENT shlib
+  PUBLIC_HEADER DESTINATION "${CMAKE_INSTALL_INCLUDEDIR}/@{CMAKE_PROJECT_NAME}" COMPONENT dev
+)
+
+install(FILES ${scripts} DESTINATION test)
+
+set(test_name "test_runVTXdigi_Modular")
+set(test_script "${CMAKE_CURRENT_SOURCE_DIR}/test/test.sh")
+add_test(NAME ${test_name}
+         COMMAND bash ${test_script}
+         WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}
+)
+set_tests_properties(${test_name} PROPERTIES
+    ENVIRONMENT "SOURCE_DIR_TEST=${CMAKE_CURRENT_SOURCE_DIR}/test;K4RECTRACKER=${CMAKE_INSTALL_PREFIX}/share/k4RecTracker;PATH=${CMAKE_INSTALL_PREFIX}/bin:$ENV{PATH};PYTHONPATH=${CMAKE_INSTALL_PREFIX}/python:${CMAKE_CURRENT_SOURCE_DIR}/test:$ENV{PYTHONPATH};LD_LIBRARY_PATH=${CMAKE_INSTALL_PREFIX}/lib:${CMAKE_INSTALL_PREFIX}/lib64:$ENV{LD_LIBRARY_PATH}"
+)
+if(COMMAND set_test_env)
+  set_test_env(${test_name})
+endif()

VTXdigi_Modular/include/IChargeCollector.h

@@ -0,0 +1,28 @@
+// VTXdigi_Modular/include/IChargeCollector.h
+#pragma once
+
+#include "TGeoMatrix.h"
+
+struct VTXdigi_Modular;
+
+namespace VTXdigi_tools {
+
+  class SimHitWrapper; // forward-declare things in include/VTXdigi_tools.h
+  class HitMap;
+
+class IChargeCollector {
+public:
+  virtual ~IChargeCollector() = default;
+  virtual void FillHit(const SimHitWrapper& simHit, HitMap& hitMap, const TGeoHMatrix& trafoMatrix) const = 0;
+  float GetChargeCollectionDepthCenter() const { return m_chargeCollectionDepthCenter; }
+
+protected:
+  explicit IChargeCollector(const VTXdigi_Modular& digitizer) : m_digitizer(digitizer) {}
+
+  const VTXdigi_Modular& m_digitizer;
+  float m_chargeCollectionDepthCenter=0; // Defines the vertical center of the charge collection region in the sensitive volume. Needed for correct digiHit positions (and residual plots) with LUTs where the charge collection varies along their depth (like TPSCo 65nm CIS). in mm, wrt. the sensor local w coordinate (w=0 at center of sensitive volume)
+};
+
+std::unique_ptr<IChargeCollector> CreateChargeCollector(const VTXdigi_Modular& digitizer, const std::string& algorithm);
+
+} // namespace VTXdigi_tools