em/model/BetheHeitlerModel.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/em/model/BetheHeitlerModel.hh
0006 //---------------------------------------------------------------------------//
0007 #pragma once
0008
0009 #include <memory>
0010
0011 #include "celeritas/em/data/BetheHeitlerData.hh"
0012 #include "celeritas/phys/ImportedModelAdapter.hh"
0013 #include "celeritas/phys/ImportedProcessAdapter.hh"
0014 #include "celeritas/phys/Model.hh"
0015 #include "celeritas/phys/ParticleParams.hh"
0016
0017 namespace celeritas
0018 {
0019 //---------------------------------------------------------------------------//
0020 /*!
0021  * Set up and launch the Bethe-Heitler model interaction.
0022  */
0023 class BetheHeitlerModel final : public Model, public StaticConcreteAction
0024 {
0025   public:
0026     //!@{
0027     //! \name Type aliases
0028     using SPConstImported = std::shared_ptr<ImportedProcesses const>;
0029     //!@}
0030
0031   public:
0032     // Construct from model ID and other necessary data
0033     BetheHeitlerModel(ActionId id,
0034                       ParticleParams const& particles,
0035                       SPConstImported data,
0036                       bool enable_lpm);
0037
0038     // Particle types and energy ranges that this model applies to
0039     SetApplicability applicability() const final;
0040
0041     // Get the microscopic cross sections for the given particle and material
0042     MicroXsBuilders micro_xs(Applicability) const final;
0043
0044     // Apply the interaction kernel on host
0045     void step(CoreParams const&, CoreStateHost&) const final;
0046
0047     // Apply the interaction kernel on device
0048     void step(CoreParams const&, CoreStateDevice&) const final;
0049
0050     //!@{
0051     //! Access model data
0052     BetheHeitlerData const& host_ref() const { return data_; }
0053     BetheHeitlerData const& device_ref() const { return data_; }
0054     //!@}
0055
0056   private:
0057     BetheHeitlerData data_;
0058     ImportedModelAdapter imported_;
0059 };
0060
0061 //---------------------------------------------------------------------------//
0062 }  // namespace celeritas