-
Notifications
You must be signed in to change notification settings - Fork 35
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add action to generate optical primaries from scintillation distribut…
…ions (#1414)
- Loading branch information
1 parent
a7dc59d
commit c4806e3
Showing
10 changed files
with
461 additions
and
38 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,161 @@ | ||
| //----------------------------------*-C++-*----------------------------------// | ||
| // Copyright 2024 UT-Battelle, LLC, and other Celeritas developers. | ||
| // See the top-level COPYRIGHT file for details. | ||
| // SPDX-License-Identifier: (Apache-2.0 OR MIT) | ||
| //---------------------------------------------------------------------------// | ||
| //! \file celeritas/optical/detail/ScintGeneratorAction.cc | ||
| //---------------------------------------------------------------------------// | ||
| #include "ScintGeneratorAction.hh" | ||
|
|
||
| #include <algorithm> | ||
|
|
||
| #include "corecel/Assert.hh" | ||
| #include "corecel/data/AuxStateVec.hh" | ||
| #include "celeritas/global/ActionLauncher.hh" | ||
| #include "celeritas/global/CoreParams.hh" | ||
| #include "celeritas/global/CoreState.hh" | ||
| #include "celeritas/global/TrackExecutor.hh" | ||
| #include "celeritas/optical/CoreParams.hh" | ||
| #include "celeritas/optical/CoreState.hh" | ||
| #include "celeritas/optical/ScintillationParams.hh" | ||
| #include "celeritas/optical/TrackData.hh" | ||
|
|
||
| #include "OffloadParams.hh" | ||
| #include "OpticalGenAlgorithms.hh" | ||
| #include "ScintGeneratorExecutor.hh" | ||
|
|
||
| namespace celeritas | ||
| { | ||
| namespace detail | ||
| { | ||
| //---------------------------------------------------------------------------// | ||
| /*! | ||
| * Construct with action ID, data IDs, and optical properties. | ||
| */ | ||
| ScintGeneratorAction::ScintGeneratorAction(ActionId id, | ||
| AuxId offload_id, | ||
| AuxId optical_id, | ||
| SPConstScintillation scintillation, | ||
| size_type auto_flush) | ||
| : id_(id) | ||
| , offload_id_{offload_id} | ||
| , optical_id_{optical_id} | ||
| , scintillation_(std::move(scintillation)) | ||
| , auto_flush_(auto_flush) | ||
| { | ||
| CELER_EXPECT(id_); | ||
| CELER_EXPECT(offload_id_); | ||
| CELER_EXPECT(optical_id_); | ||
| CELER_EXPECT(scintillation_); | ||
| CELER_EXPECT(auto_flush_ > 0); | ||
| } | ||
|
|
||
| //---------------------------------------------------------------------------// | ||
| /*! | ||
| * Descriptive name of the action. | ||
| */ | ||
| std::string_view ScintGeneratorAction::description() const | ||
| { | ||
| return "generate scintillation photons from optical distribution data"; | ||
| } | ||
|
|
||
| //---------------------------------------------------------------------------// | ||
| /*! | ||
| * Execute the action with host data. | ||
| */ | ||
| void ScintGeneratorAction::step(CoreParams const& params, | ||
| CoreStateHost& state) const | ||
| { | ||
| this->step_impl(params, state); | ||
| } | ||
|
|
||
| //---------------------------------------------------------------------------// | ||
| /*! | ||
| * Execute the action with device data. | ||
| */ | ||
| void ScintGeneratorAction::step(CoreParams const& params, | ||
| CoreStateDevice& state) const | ||
| { | ||
| this->step_impl(params, state); | ||
| } | ||
|
|
||
| //---------------------------------------------------------------------------// | ||
| /*! | ||
| * Generate optical track initializers from scintillation distribution data. | ||
| */ | ||
| template<MemSpace M> | ||
| void ScintGeneratorAction::step_impl(CoreParams const& core_params, | ||
| CoreState<M>& core_state) const | ||
| { | ||
| auto& offload_state | ||
| = get<OpticalOffloadState<M>>(core_state.aux(), offload_id_); | ||
| auto& optical_state | ||
| = get<optical::CoreState<M>>(core_state.aux(), optical_id_); | ||
|
|
||
| auto& num_photons = optical_state.counters().num_initializers; | ||
| auto& num_new_photons = offload_state.buffer_size.num_photons; | ||
|
|
||
| if (num_photons + num_new_photons < auto_flush_) | ||
| return; | ||
|
|
||
| auto initializers_size = optical_state.ref().init.initializers.size(); | ||
| CELER_VALIDATE(num_photons + num_new_photons <= initializers_size, | ||
| << "insufficient capacity (" << initializers_size | ||
| << ") for optical photon initializers (total capacity " | ||
| "requirement of " | ||
| << num_photons + num_new_photons << ")"); | ||
|
|
||
| auto& offload = offload_state.store.ref(); | ||
| auto& buffer_size = offload_state.buffer_size.scintillation; | ||
| CELER_ASSERT(buffer_size > 0); | ||
|
|
||
| // Calculate the cumulative sum of the number of photons in the buffered | ||
| // distributions. These values are used to determine which thread will | ||
| // generate initializers from which distribution | ||
| auto count = inclusive_scan_photons(offload.scintillation, | ||
| offload.offsets, | ||
| buffer_size, | ||
| core_state.stream_id()); | ||
|
|
||
| // Generate the optical photon initializers from the distribution data | ||
| this->generate(core_params, core_state); | ||
|
|
||
| num_photons += count; | ||
| num_new_photons -= count; | ||
| buffer_size = 0; | ||
| } | ||
|
|
||
| //---------------------------------------------------------------------------// | ||
| /*! | ||
| * Launch a (host) kernel to generate optical photon initializers. | ||
| */ | ||
| void ScintGeneratorAction::generate(CoreParams const& core_params, | ||
| CoreStateHost& core_state) const | ||
| { | ||
| auto& offload_state = get<OpticalOffloadState<MemSpace::native>>( | ||
| core_state.aux(), offload_id_); | ||
| auto& optical_state = get<optical::CoreState<MemSpace::native>>( | ||
| core_state.aux(), optical_id_); | ||
|
|
||
| TrackExecutor execute{ | ||
| core_params.ptr<MemSpace::native>(), | ||
| core_state.ptr(), | ||
| detail::ScintGeneratorExecutor{core_state.ptr(), | ||
| scintillation_->host_ref(), | ||
| offload_state.store.ref(), | ||
| optical_state.ptr(), | ||
| offload_state.buffer_size}}; | ||
| launch_action(*this, core_params, core_state, execute); | ||
| } | ||
|
|
||
| //---------------------------------------------------------------------------// | ||
| #if !CELER_USE_DEVICE | ||
| void ScintGeneratorAction::generate(CoreParams const&, CoreStateDevice&) const | ||
| { | ||
| CELER_NOT_CONFIGURED("CUDA OR HIP"); | ||
| } | ||
| #endif | ||
|
|
||
| //---------------------------------------------------------------------------// | ||
| } // namespace detail | ||
| } // namespace celeritas |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,52 @@ | ||
| //---------------------------------*-CUDA-*----------------------------------// | ||
| // Copyright 2024 UT-Battelle, LLC, and other Celeritas developers. | ||
| // See the top-level COPYRIGHT file for details. | ||
| // SPDX-License-Identifier: (Apache-2.0 OR MIT) | ||
| //---------------------------------------------------------------------------// | ||
| //! \file celeritas/optical/detail/ScintGeneratorAction.cu | ||
| //---------------------------------------------------------------------------// | ||
| #include "ScintGeneratorAction.hh" | ||
|
|
||
| #include "corecel/Assert.hh" | ||
| #include "corecel/sys/ScopedProfiling.hh" | ||
| #include "celeritas/global/ActionLauncher.device.hh" | ||
| #include "celeritas/global/TrackExecutor.hh" | ||
| #include "celeritas/optical/CoreParams.hh" | ||
| #include "celeritas/optical/CoreState.hh" | ||
| #include "celeritas/optical/ScintillationParams.hh" | ||
|
|
||
| #include "OffloadParams.hh" | ||
| #include "OpticalGenAlgorithms.hh" | ||
| #include "ScintGeneratorExecutor.hh" | ||
|
|
||
| namespace celeritas | ||
| { | ||
| namespace detail | ||
| { | ||
| //---------------------------------------------------------------------------// | ||
| /*! | ||
| * Launch a kernel to generate optical photon initializers. | ||
| */ | ||
| void ScintGeneratorAction::generate(CoreParams const& core_params, | ||
| CoreStateDevice& core_state) const | ||
| { | ||
| auto& offload_state = get<OpticalOffloadState<MemSpace::native>>( | ||
| core_state.aux(), offload_id_); | ||
| auto& optical_state = get<optical::CoreState<MemSpace::native>>( | ||
| core_state.aux(), optical_id_); | ||
|
|
||
| TrackExecutor execute{ | ||
| core_params.ptr<MemSpace::native>(), | ||
| core_state.ptr(), | ||
| detail::ScintGeneratorExecutor{core_state.ptr(), | ||
| scintillation_->device_ref(), | ||
| offload_state.store.ref(), | ||
| optical_state.ptr(), | ||
| offload_state.buffer_size}}; | ||
| static ActionLauncher<decltype(execute)> const launch_kernel(*this); | ||
| launch_kernel(core_state, execute); | ||
| } | ||
|
|
||
| //---------------------------------------------------------------------------// | ||
| } // namespace detail | ||
| } // namespace celeritas |
Oops, something went wrong.