EIC code displayed by LXR master/​include/​celeritas/​ext/​GeantTrackReconstruction.hh	Source navigation Diff markup Identifier search General search
	Version: master test Revert   Change

File indexing completed on 2026-10-27 09:25:10

 //------------------------------- -*- C++ -*- -------------------------------//
 // Copyright Celeritas contributors: see top-level COPYRIGHT file for details
 // SPDX-License-Identifier: (Apache-2.0 OR MIT)
 //---------------------------------------------------------------------------//
 //! \file celeritas/ext/GeantTrackReconstruction.hh
 //---------------------------------------------------------------------------//
 #pragma once
 
 #include <memory>
 #include <vector>
 
 #include "corecel/Macros.hh"
 #include "celeritas/Types.hh"
 
 class G4ParticleDefinition;
 class G4Step;
 class G4Track;
 class G4VProcess;
 class G4VUserTrackInformation;
 
 namespace celeritas
 {
 //---------------------------------------------------------------------------//
 /*!
  * Manage track information for reconstruction.
  *
  * This class handles the bookkeeping of Geant4 track information needed
  * to reconstruct tracks during hit processing. It maintains mappings between
  * Celeritas PrimaryID and Geant4 track data.
  *
  * \par Usage
  * - \c init_event
  * - \c acquire (multiple times)
  * - \c view (may be interleaved with acquire)
  * - \c clear (once all active tracks are used up)
  * - then it can be initialized with a new event, or new primaries can be
  *   added to the current event.
  */
 class GeantTrackReconstruction
 {
   public:
     //!@{
     //! \name Type aliases
     using VecParticle = std::vector<G4ParticleDefinition const*>;
     using SPStep = std::shared_ptr<G4Step>;
     using EventIdGetter = int (*)();
     //!@}
 
   public:
     // Create a G4Step object with cleared data
     static SPStep make_g4step();
 
     // Construct with particle definitions for track reconstruction
     GeantTrackReconstruction(VecParticle const&, SPStep);
 
     ~GeantTrackReconstruction();
     CELER_DEFAULT_MOVE_DELETE_COPY(GeantTrackReconstruction);
 
     // Clear G4Track reconstruction data
     void clear();
 
     // Register mapping from Celeritas PrimaryID to Geant4 track ID
     [[nodiscard]] PrimaryId acquire(G4Track&);
 
     // Reset primary ID at each event start
     void init_event();
 
     // Restore track information for given primary and particle IDs
     [[nodiscard]] G4Track& view(ParticleId, PrimaryId) const;
 
     // View a track with the given particle ID
     [[nodiscard]] G4Track& view(ParticleId) const;
 
     // Event ID function pointer for unit testing (only used in
     // CELERITAS_DEBUG)
     static EventIdGetter get_current_event_id;
 
   private:
     //! Data needed to reconstruct a G4Track from Celeritas transport
     class AcquiredData
     {
       public:
         //! Save the G4Track reconstruction data
         explicit AcquiredData(G4Track&);
         //! Whether the data is valid
         explicit operator bool() const { return track_id_ >= 0; }
         //! Restore the G4Track from the reconstruction data
         void restore(G4Track&) const;
 
       private:
         //! Original Geant4 track ID
         int track_id_{-1};
         //! Original Geant4 parent ID
         int parent_id_{0};
         //! User track information
         std::unique_ptr<G4VUserTrackInformation> user_info_;
         //! Process that created the track
         G4VProcess const* creator_process_{nullptr};
     };
 
     //! G4Track reconstruction data indexed by Celeritas PrimaryID
     std::vector<AcquiredData> g4_track_data_;
     //! Tracks for each particle type
     std::vector<std::unique_ptr<G4Track>> tracks_;
     //! Shared step object
     SPStep step_;
     //! Starting primary id
     PrimaryId start_{0};
     //! Last G4 event ID for error checking
     int g4_event_id_{-1};
 };
 
 //---------------------------------------------------------------------------//
 }  // namespace celeritas

This page was automatically generated by the 2.3.7 LXR engine. The LXR team