feat: Add elemental fractions to the material recording step

Core/include/Acts/Material/MaterialInteraction.hpp

@@ -81,6 +81,10 @@ struct MaterialInteraction {
   double pathCorrection = 1.;
   /// The effective, passed material properties including the path correction.
   MaterialSlab materialSlab = MaterialSlab::Nothing();
+  /// Vectors of the individual elements present
+  std::vector<unsigned int> elementZ = {};
+  /// How much of each element are present
+  std::vector<float> elementFrac = {};
 };
 
 /// Simple result struct to be returned

Examples/Algorithms/Geant4/include/ActsExamples/Geant4/Geant4Simulation.hpp

@@ -191,6 +191,7 @@ class Geant4MaterialRecording final : public Geant4SimulationBase {
 
     /// Materials to exclude from the recording.
     std::vector<std::string> excludeMaterials = {"Air", "Vacuum"};
+    bool recordElementFractions = false;
   };
 
   /// Material recording constructor

Examples/Algorithms/Geant4/include/ActsExamples/Geant4/MaterialSteppingAction.hpp

@@ -35,6 +35,7 @@ class MaterialSteppingAction final : public G4UserSteppingAction {
     std::shared_ptr<EventStore> eventStore;
 
     std::vector<std::string> excludeMaterials = {};
+    bool recordElementFractions = false;
   };
 
   /// Construct the action

Examples/Algorithms/Geant4/src/Geant4Simulation.cpp

@@ -392,6 +392,7 @@ Geant4MaterialRecording::Geant4MaterialRecording(
     Geant4::MaterialSteppingAction::Config steppingCfg;
     steppingCfg.eventStore = m_eventStore;
     steppingCfg.excludeMaterials = m_cfg.excludeMaterials;
+    steppingCfg.recordElementFractions = m_cfg.recordElementFractions;
     // G4RunManager will take care of deletion
     auto steppingAction = new Geant4::MaterialSteppingAction(
         steppingCfg, this->logger().cloneWithSuffix("MaterialSteppingAction"));

Examples/Algorithms/Geant4/src/MaterialSteppingAction.cpp

@@ -90,6 +90,20 @@ void MaterialSteppingAction::UserSteppingAction(const G4Step* stepPtr) {
   mInteraction.materialSlab = slab;
   mInteraction.pathCorrection = step.GetStepLength() * convertLengthToActs;
 
+  if (m_cfg.recordElementFractions) {
+    if (nElements == 1) {
+      mInteraction.elementZ.push_back(
+          static_cast<unsigned int>(material.GetZ()));
+      mInteraction.elementFrac.push_back(1.0f);
+    } else {
+      for (std::size_t i = 0; i < nElements; i++) {
+        mInteraction.elementZ.push_back(
+            static_cast<unsigned int>(elements->at(i)->GetZ()));
+        mInteraction.elementFrac.push_back(static_cast<float>(fraction[i]));
+      }
+    }
+  }
+
   assert(step.GetTrack() != nullptr);
   const G4Track& track = *step.GetTrack();
   const std::size_t trackId = track.GetTrackID();

Examples/Scripts/Python/material_recording.py

@@ -75,6 +75,7 @@ def runMaterialRecording(
         randomNumbers=rnd,
         inputParticles=hepmc3Converter.config.outputParticles,
         outputMaterialTracks=materialTrackCollectionName,
+        recordElementFractions=False,
     )
 
     s.addAlgorithm(g4Alg)

Plugins/Root/include/ActsPlugins/Root/RootMaterialTrackIo.hpp

@@ -157,6 +157,12 @@ class RootMaterialTrackIo {
     /// step material rho
     std::vector<float> stepMatRho;
     std::vector<float>* stepMatRhoPtr = &stepMatRho;
+    /// step material element
+    std::vector<std::vector<unsigned int>> stepElementZ;
+    std::vector<std::vector<unsigned int>>* stepElementZPtr = &stepElementZ;
+    /// step material fraction of elements
+    std::vector<std::vector<float>> stepFraction;
+    std::vector<std::vector<float>>* stepFractionPtr = &stepFraction;
   };
 
   struct MaterialSurfacePayload {

Plugins/Root/src/RootMaterialTrackIo.cpp

@@ -48,6 +48,8 @@ void RootMaterialTrackIo::connectForRead(TTree& materialChain) {
   materialChain.SetBranchAddress("mat_A", &m_stepPayload.stepMatAPtr);
   materialChain.SetBranchAddress("mat_Z", &m_stepPayload.stepMatZPtr);
   materialChain.SetBranchAddress("mat_rho", &m_stepPayload.stepMatRhoPtr);
+  materialChain.SetBranchAddress("elements", &m_stepPayload.stepElementZPtr);
+  materialChain.SetBranchAddress("fraction", &m_stepPayload.stepFractionPtr);
   if (m_cfg.surfaceInfo) {
     materialChain.SetBranchAddress("sur_id", &m_surfacePayload.surfaceIdPtr);
     materialChain.SetBranchAddress("sur_x", &m_surfacePayload.surfaceXPtr);
@@ -85,6 +87,8 @@ void RootMaterialTrackIo::connectForWrite(TTree& materialTree) {
   materialTree.Branch("mat_A", &m_stepPayload.stepMatA);
   materialTree.Branch("mat_Z", &m_stepPayload.stepMatZ);
   materialTree.Branch("mat_rho", &m_stepPayload.stepMatRho);
+  materialTree.Branch("elements", &m_stepPayload.stepElementZ);
+  materialTree.Branch("fraction", &m_stepPayload.stepFraction);
 
   if (m_cfg.prePostStepInfo) {
     materialTree.Branch("mat_sx", &m_stepPayload.stepXs);
@@ -133,6 +137,8 @@ void RootMaterialTrackIo::write(const GeometryContext& gctx,
   m_stepPayload.stepMatA.clear();
   m_stepPayload.stepMatZ.clear();
   m_stepPayload.stepMatRho.clear();
+  m_stepPayload.stepElementZ.clear();
+  m_stepPayload.stepFraction.clear();
 
   m_surfacePayload.surfaceId.clear();
   m_surfacePayload.surfaceX.clear();
@@ -167,6 +173,8 @@ void RootMaterialTrackIo::write(const GeometryContext& gctx,
   m_stepPayload.stepMatA.reserve(mints);
   m_stepPayload.stepMatZ.reserve(mints);
   m_stepPayload.stepMatRho.reserve(mints);
+  m_stepPayload.stepElementZ.reserve(mints);
+  m_stepPayload.stepFraction.reserve(mints);
 
   m_surfacePayload.surfaceId.reserve(mints);
   m_surfacePayload.surfaceX.reserve(mints);
@@ -279,6 +287,8 @@ void RootMaterialTrackIo::write(const GeometryContext& gctx,
     m_stepPayload.stepMatA.push_back(mprops.material().Ar());
     m_stepPayload.stepMatZ.push_back(mprops.material().Z());
     m_stepPayload.stepMatRho.push_back(mprops.material().massDensity());
+    m_stepPayload.stepElementZ.push_back(mint.elementZ);
+    m_stepPayload.stepFraction.push_back(mint.elementFrac);
     // re-calculate if defined to do so
     if (m_cfg.recalculateTotals) {
       m_summaryPayload.tX0 += mprops.thicknessInX0();
@@ -325,6 +335,11 @@ RecordedMaterialTrack RootMaterialTrackIo::read() const {
                                   m_stepPayload.stepMatZ[is],
                                   m_stepPayload.stepMatRho[is]),
         s);
+    /// adding the element vectors and fractions
+    if (is < m_stepPayload.stepElementZ.size()) {
+      mInteraction.elementZ = m_stepPayload.stepElementZ[is];
+      mInteraction.elementFrac = m_stepPayload.stepFraction[is];
+    }
     if (m_cfg.surfaceInfo) {
       // add the surface information to the interaction this allows the
       // mapping to be speed up

Python/Examples/src/plugins/Geant4.cpp

@@ -170,7 +170,8 @@ PYBIND11_MODULE(ActsExamplesPythonBindingsGeant4, mod) {
     auto c = py::class_<Algorithm::Config, Geant4SimulationBase::Config,
                         std::shared_ptr<Algorithm::Config>>(alg, "Config")
                  .def(py::init<>());
-    ACTS_PYTHON_STRUCT(c, outputMaterialTracks, excludeMaterials);
+    ACTS_PYTHON_STRUCT(c, outputMaterialTracks, excludeMaterials,
+                       recordElementFractions);
   }
 
   {

Python/Examples/tests/root_file_hashes.txt

@@ -17,11 +17,11 @@ test_itk_seeding__performance_seeding.root: 78ebda54cd0f026ba4b7f316724ffd946de5
 test_itk_seeding__particles.root: e8eff478087acd920d4e1567ae3f9805e4e6ad108e2ff342c1f96b8860fa8197
 test_itk_seeding__particles_simulation.root: 0c135b78e7e127b4b7373b40c3a55cb86f3cddf3c880d95b01dfa95a47e832bb
 test_propagation__propagation_summary.root: 243028b59d166969dcdd7ef7283a87f429af805b9570b34b68d29f43db0276be
-test_material_recording__geant4_material_tracks.root: c5771bf1821617dcb327976e26de10716bb8035a2ef6bf691c14d0eece46bf7e
+test_material_recording__geant4_material_tracks.root: 28b077e1745805332cfae4d629624f882d9f73e9fff848fa46caeeb3e5e6feda
 test_material_mapping__material_mapping_map.root: bcc7bcfbfed6e576926c24cc6b5808d5f7c890ef1d700e732d5f75cf7119e93f
 test_material_mapping__material_mapping_mapped.root: e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855
 test_material_mapping__material_mapping_unmapped.root: e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855
-test_material_mapping__material_validation.root: 63208746c50177723c5efc5d989686dac48c304fe8810680bf34c6a9639605ad
+test_material_mapping__material_validation.root: 9a5693dad6ff9cfbd93350fcb8c65e9b53c5a2dad34052401f0eecc7ff496c85
 test_truth_tracking_gsf[generic]__trackstates_gsf.root: 71c14f1053a7974ceb49ad7b5cf8d33ad2dafa1c0f7fe404a6dfe7ff77ca2106
 test_truth_tracking_gsf[generic]__tracksummary_gsf.root: ca82b2c4735b790adcc57f2bfb10c4689fcd6eb478be960d876bd58bcda36850
 test_truth_tracking_gsf[odd]__trackstates_gsf.root: b22a2263de05cb6af1cbf84d1e6952470aad5a37a34d323008bda492b304e08d