celeritas/user/DetectorSteps.hh

0001 //------------------------------- -*- C++ -*- -------------------------------//
0002 // Copyright Celeritas contributors: see top-level COPYRIGHT file for details
0003 // SPDX-License-Identifier: (Apache-2.0 OR MIT)
0004 //---------------------------------------------------------------------------//
0005 //! \file celeritas/user/DetectorSteps.hh
0006 //---------------------------------------------------------------------------//
0007 #pragma once
0008
0009 #include <vector>
0010
0011 #include "corecel/Assert.hh"
0012 #include "corecel/Macros.hh"
0013 #include "corecel/cont/EnumArray.hh"
0014 #include "corecel/data/PinnedAllocator.hh"
0015 #include "celeritas/Quantities.hh"
0016 #include "celeritas/Types.hh"
0017
0018 namespace celeritas
0019 {
0020 //---------------------------------------------------------------------------//
0021 template<Ownership W, MemSpace M>
0022 struct StepStateData;
0023
0024 //---------------------------------------------------------------------------//
0025 /*!
0026  * CPU results for detector stepping at the beginning or end of a step.
0027  *
0028  * Since the volume has a one-to-one mapping to a DetectorId, we omit it. Since
0029  * multiple "touchable" (multi-level geometry instance) volumes can point to
0030  * the same detector, and the location in that hierarchy can be important, a
0031  * separate "volume instance IDs" multi-D level is available.
0032  */
0033 struct DetectorStepPointOutput
0034 {
0035     //// TYPES ////
0036
0037     using Energy = units::MevEnergy;
0038     template<class T>
0039     using PinnedVec = std::vector<T, PinnedAllocator<T>>;
0040
0041     //// DATA ////
0042
0043     PinnedVec<real_type> time;
0044     PinnedVec<Real3> pos;
0045     PinnedVec<Real3> dir;
0046     PinnedVec<Energy> energy;
0047
0048     PinnedVec<VolumeInstanceId> volume_instance_ids;
0049 };
0050
0051 //---------------------------------------------------------------------------//
0052 /*!
0053  * CPU results for many in-detector tracks at a single step iteration.
0054  *
0055  * This convenience class can be used to postprocess the results from sensitive
0056  * detectors on CPU. The data members will be available based on the \c
0057  * selection of the \c StepInterface class that gathered the data.
0058  *
0059  * Unlike \c StepStateData, which leaves gaps for inactive or filtered
0060  * tracks, every entry of these vectors will be valid and correspond to a
0061  * single DetectorId.
0062  */
0063 struct DetectorStepOutput
0064 {
0065     //// TYPES ////
0066
0067     using Energy = units::MevEnergy;
0068     template<class T>
0069     using PinnedVec = std::vector<T, PinnedAllocator<T>>;
0070
0071     //// DATA ////
0072
0073     // Pre- and post-step data
0074     EnumArray<StepPoint, DetectorStepPointOutput> points;
0075
0076     // Detector ID and track ID are always set
0077     PinnedVec<DetectorId> detector;
0078     PinnedVec<TrackId> track_id;
0079
0080     // Additional optional data
0081     PinnedVec<EventId> event_id;
0082     PinnedVec<TrackId> parent_id;
0083     PinnedVec<PrimaryId> primary_id;
0084     PinnedVec<size_type> track_step_count;
0085     PinnedVec<real_type> step_length;
0086     PinnedVec<real_type> weight;
0087     PinnedVec<ParticleId> particle;
0088     PinnedVec<Energy> energy_deposition;
0089
0090     // 2D size for volume instances
0091     size_type volume_instance_depth{0};
0092
0093     //// METHODS ////
0094
0095     //! Number of elements in the detector output.
0096     size_type size() const { return detector.size(); }
0097     //! Whether the size is nonzero
0098     explicit operator bool() const { return !detector.empty(); }
0099 };
0100
0101 //---------------------------------------------------------------------------//
0102 // Copy state data for all steps inside detectors to the output.
0103 template<MemSpace M>
0104 void copy_steps(DetectorStepOutput* output,
0105                 StepStateData<Ownership::reference, M> const& state);
0106
0107 template<>
0108 void copy_steps<MemSpace::host>(
0109     DetectorStepOutput*,
0110     StepStateData<Ownership::reference, MemSpace::host> const&);
0111 template<>
0112 void copy_steps<MemSpace::device>(
0113     DetectorStepOutput*,
0114     StepStateData<Ownership::reference, MemSpace::device> const&);
0115
0116 //---------------------------------------------------------------------------//
0117 #if !CELER_USE_DEVICE
0118 template<>
0119 inline void copy_steps<MemSpace::device>(
0120     DetectorStepOutput*,
0121     StepStateData<Ownership::reference, MemSpace::device> const&)
0122 {
0123     CELER_NOT_CONFIGURED("CUDA or HIP");
0124 }
0125
0126 #endif
0127 //---------------------------------------------------------------------------//
0128 }  // namespace celeritas