optical/action/InitializeTracksAction.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/optical/action/InitializeTracksAction.hh
0006 //---------------------------------------------------------------------------//
0007 #pragma once
0008
0009 #include "ActionInterface.hh"
0010
0011 namespace celeritas
0012 {
0013 namespace optical
0014 {
0015 //---------------------------------------------------------------------------//
0016 /*!
0017  * Initialize optical track states.
0018  */
0019 class InitializeTracksAction final : public OpticalStepActionInterface,
0020                                      public ConcreteAction
0021 {
0022   public:
0023     //! Construct with ID
0024     explicit InitializeTracksAction(ActionId);
0025
0026     //! Execute the action with host data
0027     void step(CoreParams const&, CoreStateHost&) const final;
0028
0029     //! Execute the action with device data
0030     void step(CoreParams const&, CoreStateDevice&) const final;
0031
0032     //! Dependency ordering of the action
0033     StepActionOrder order() const final { return StepActionOrder::start; }
0034
0035   private:
0036     template<MemSpace M>
0037     void step_impl(CoreParams const&, CoreState<M>&) const;
0038
0039     void step_impl(CoreParams const&, CoreStateHost&, size_type) const;
0040     void step_impl(CoreParams const&, CoreStateDevice&, size_type) const;
0041 };
0042
0043 //---------------------------------------------------------------------------//
0044 }  // namespace optical
0045 }  // namespace celeritas