Add builtin surface roughness models

src/celeritas/CMakeLists.txt

@@ -363,6 +363,9 @@ celeritas_polysource(optical/gen/PrimaryGeneratorAction)
 celeritas_polysource(optical/gen/detail/GeneratorAlgorithms)
 celeritas_polysource(optical/gen/detail/OffloadAlgorithms)
 celeritas_polysource(optical/surface/BoundaryAction)
+celeritas_polysource(optical/surface/model/GaussianRoughnessModel)
+celeritas_polysource(optical/surface/model/PolishedRoughnessModel)
+celeritas_polysource(optical/surface/model/SmearRoughnessModel)
 celeritas_polysource(phys/detail/DiscreteSelectAction)
 celeritas_polysource(phys/detail/PreStepAction)
 celeritas_polysource(phys/detail/TrackingCutAction)

src/celeritas/optical/CoreTrackView.hh

@@ -141,6 +141,9 @@ CoreTrackView::operator=(TrackInitializer const& init)
     // Initialize the physics state
     this->physics() = PhysicsTrackView::Initializer{};
 
+    // Initialize the surface state
+    this->surface_physics().reset();
+
     return *this;
 }
 

src/celeritas/optical/action/ActionLauncher.device.hh

@@ -52,16 +52,17 @@ class ActionLauncher : public KernelLauncher<F>
          || CELER_COMPILER == CELER_COMPILER_CLANG)
             && !std::is_pointer_v<F> && !std::is_reference_v<F>,
         R"(Launched action must be a trivially copyable function object)");
-    using StepActionT = OpticalStepActionInterface;
 
   public:
     // Create a launcher from a string
     using KernelLauncher<F>::KernelLauncher;
 
     // Create a launcher from an action
+    template<class StepActionT>
     explicit ActionLauncher(StepActionT const& action);
 
     // Create a launcher with a string extension
+    template<class StepActionT>
     ActionLauncher(StepActionT const& action, std::string_view ext);
 
     // Launch a kernel for a thread range or number of threads
@@ -77,8 +78,9 @@ class ActionLauncher : public KernelLauncher<F>
  * Create a launcher from an action.
  */
 template<class F>
+template<class StepActionT>
 ActionLauncher<F>::ActionLauncher(StepActionT const& action)
-    : ActionLauncher{action.label()}
+    : KernelLauncher<F>{action.label()}
 {
 }
 
@@ -87,9 +89,10 @@ ActionLauncher<F>::ActionLauncher(StepActionT const& action)
  * Create a launcher with a string extension.
  */
 template<class F>
+template<class StepActionT>
 ActionLauncher<F>::ActionLauncher(StepActionT const& action,
                                   std::string_view ext)
-    : ActionLauncher{std::string(action.label()) + "-" + std::string(ext)}
+    : KernelLauncher<F>{std::string(action.label()) + "-" + std::string(ext)}
 {
 }
 

src/celeritas/optical/action/detail/AlongStepExecutor.hh

@@ -34,6 +34,11 @@ struct AlongStepExecutor
 //---------------------------------------------------------------------------//
 CELER_FUNCTION void AlongStepExecutor::operator()(CoreTrackView& track)
 {
+    if (track.surface_physics().is_crossing_boundary())
+    {
+        return;
+    }
+
     auto sim = track.sim();
 
     CELER_ASSERT(sim.status() == TrackStatus::alive);

src/celeritas/optical/action/detail/PreStepExecutor.hh

@@ -55,6 +55,11 @@ CELER_FUNCTION void PreStepExecutor::operator()(CoreTrackView const& track)
                  || sim.status() == TrackStatus::alive);
     sim.status(TrackStatus::alive);
 
+    if (track.surface_physics().is_crossing_boundary())
+    {
+        return;
+    }
+
     auto phys = track.physics();
     if (!phys.has_interaction_mfp())
     {

src/celeritas/optical/action/detail/PropagateExecutor.hh

@@ -32,6 +32,11 @@ struct PropagateExecutor
 //---------------------------------------------------------------------------//
 CELER_FUNCTION void PropagateExecutor::operator()(CoreTrackView& track)
 {
+    if (track.surface_physics().is_crossing_boundary())
+    {
+        return;
+    }
+
     auto&& sim = track.sim();
     CELER_ASSERT(sim.status() == TrackStatus::alive);
 

src/celeritas/optical/action/detail/TrackingCutExecutor.hh

@@ -31,6 +31,11 @@ struct TrackingCutExecutor
 //---------------------------------------------------------------------------//
 CELER_FUNCTION void TrackingCutExecutor::operator()(CoreTrackView& track)
 {
+    if (track.surface_physics().is_crossing_boundary())
+    {
+        return;
+    }
+
     using Energy = ParticleTrackView::Energy;
 
     auto deposited = track.particle().energy().value();

src/celeritas/optical/surface/SurfaceModelView.hh

@@ -0,0 +1,110 @@
+//------------------------------- -*- C++ -*- -------------------------------//
+// Copyright Celeritas contributors: see top-level COPYRIGHT file for details
+// SPDX-License-Identifier: (Apache-2.0 OR MIT)
+//---------------------------------------------------------------------------//
+//! \file celeritas/optical/surface/SurfaceModelView.hh
+//---------------------------------------------------------------------------//
+#pragma once
+
+#include "celeritas/optical/Types.hh"
+#include "celeritas/phys/SurfacePhysicsMapView.hh"
+
+namespace celeritas
+{
+namespace optical
+{
+//---------------------------------------------------------------------------//
+/*!
+ * Optical surface data for a model.
+ *
+ * Wraps common behavior for querying the surface model data for a given
+ * physics surface interface.
+ */
+class SurfaceModelView
+{
+  public:
+    //!@{
+    //! \name Type aliases
+    using InternalSurfaceId = SurfacePhysicsMapView::InternalSurfaceId;
+    //!@}
+
+  public:
+    // Construct from map view and materials
+    inline CELER_FUNCTION SurfaceModelView(SurfacePhysicsMapView,
+                                           OptMatId pre_mat,
+                                           OptMatId post_mat);
+
+    // Get surface model ID
+    inline CELER_FUNCTION SurfaceModelId model_id() const;
+
+    // Get internal surface ID for the model
+    inline CELER_FUNCTION InternalSurfaceId internal_surface_id() const;
+
+    // Get pre-volume optical material
+    inline CELER_FUNCTION OptMatId pre_material() const;
+
+    // Get post-volume optical material
+    inline CELER_FUNCTION OptMatId post_material() const;
+
+  private:
+    SurfacePhysicsMapView physics_map_;
+    OptMatId pre_material_;
+    OptMatId post_material_;
+};
+
+//---------------------------------------------------------------------------//
+// INLINE DEFINITIONS
+//---------------------------------------------------------------------------//
+/*!
+ * Construct from physics map view and materials.
+ */
+CELER_FUNCTION
+SurfaceModelView::SurfaceModelView(SurfacePhysicsMapView physics_map,
+                                   OptMatId pre_material,
+                                   OptMatId post_material)
+    : physics_map_(physics_map)
+    , pre_material_(pre_material)
+    , post_material_(post_material)
+{
+}
+
+//---------------------------------------------------------------------------//
+/*!
+ * Get the surface model for this physics surface.
+ */
+CELER_FUNCTION SurfaceModelId SurfaceModelView::model_id() const
+{
+    return physics_map_.surface_model_id();
+}
+
+//---------------------------------------------------------------------------//
+/*!
+ * Get the internal surface ID for the physics surface in this model.
+ */
+CELER_FUNCTION auto SurfaceModelView::internal_surface_id() const
+    -> InternalSurfaceId
+{
+    return physics_map_.internal_surface_id();
+}
+
+//---------------------------------------------------------------------------//
+/*!
+ * Get the optical material before the interface.
+ */
+CELER_FUNCTION OptMatId SurfaceModelView::pre_material() const
+{
+    return pre_material_;
+}
+
+//---------------------------------------------------------------------------//
+/*!
+ * Get the optical material after the interface.
+ */
+CELER_FUNCTION OptMatId SurfaceModelView::post_material() const
+{
+    return post_material_;
+}
+
+//---------------------------------------------------------------------------//
+}  // namespace optical
+}  // namespace celeritas

src/celeritas/optical/surface/SurfacePhysicsData.hh

@@ -148,13 +148,15 @@ struct SurfacePhysicsStateData
     StateItems<OptMatId> post_volume_material;
 
     StateItems<SurfaceTrackPosition> surface_position;
+    StateItems<SubsurfaceDirection> track_direction;
     StateItems<Real3> facet_normal;
 
     //! Whether data is assigned
     explicit CELER_FUNCTION operator bool() const
     {
         return !surface.empty() && surface.size() == surface_orientation.size()
                && surface.size() == surface_position.size()
+               && surface.size() == track_direction.size()
                && surface.size() == pre_volume_material.size()
                && surface.size() == post_volume_material.size()
                && surface.size() == global_normal.size()
@@ -174,6 +176,7 @@ struct SurfacePhysicsStateData
         surface_orientation = other.surface_orientation;
         global_normal = other.global_normal;
         surface_position = other.surface_position;
+        track_direction = other.track_direction;
         pre_volume_material = other.pre_volume_material;
         post_volume_material = other.post_volume_material;
         facet_normal = other.facet_normal;
@@ -196,6 +199,7 @@ resize(SurfacePhysicsStateData<Ownership::value, M>* state, size_type size)
     resize(&state->surface_orientation, size);
     resize(&state->global_normal, size);
     resize(&state->surface_position, size);
+    resize(&state->track_direction, size);
     resize(&state->pre_volume_material, size);
     resize(&state->post_volume_material, size);
     resize(&state->facet_normal, size);

src/celeritas/optical/surface/SurfacePhysicsParams.cc

@@ -11,80 +11,18 @@
 #include "celeritas/inp/SurfacePhysics.hh"
 #include "celeritas/phys/SurfacePhysicsMapBuilder.hh"
 
+#include "model/GaussianRoughnessModel.hh"
+#include "model/PolishedRoughnessModel.hh"
+#include "model/SmearRoughnessModel.hh"
+
+#include "detail/BuiltinSurfaceModelBuilder.hh"
+
 namespace celeritas
 {
 namespace optical
 {
 namespace
 {
-//---------------------------------------------------------------------------//
-/*!
- * A fake model used to mock actual surface physics models.
- *
- * TODO: Remove and replace with actual built-in surface models.
- */
-template<class T>
-class FakeModel : public SurfaceModel
-{
-  public:
-    FakeModel(SurfaceModelId model_id,
-              std::string_view label,
-              std::map<PhysSurfaceId, T> const& layer_map)
-        : SurfaceModel(model_id, label), layers_(layer_map)
-    {
-    }
-
-    VecSurfaceLayer get_surfaces() const final
-    {
-        VecSurfaceLayer result;
-        for ([[maybe_unused]] auto const& [layer, data] : layers_)
-        {
-            result.push_back(PhysSurfaceId(layer.get()));
-        }
-        return result;
-    }
-
-    void step(CoreParams const&, CoreStateHost&) const final {}
-    void step(CoreParams const&, CoreStateDevice&) const final {}
-
-  private:
-    std::map<PhysSurfaceId, T> layers_;
-};
-
-//---------------------------------------------------------------------------//
-/*!
- * Helper to build fake models.
- *
- * Doesn't insert empty models and keeps track of number of physics surfaces
- * for validation purposes.
- */
-class FakeModelBuilder
-{
-  public:
-    FakeModelBuilder(std::vector<std::shared_ptr<SurfaceModel>>& models)
-        : models_(models)
-    {
-    }
-
-    template<class T>
-    void operator()(std::string_view label,
-                    std::map<PhysSurfaceId, T> const& layer_map)
-    {
-        if (!layer_map.empty())
-        {
-            models_.push_back(std::make_shared<FakeModel<T>>(
-                SurfaceModelId(models_.size()), label, layer_map));
-            num_surf_ += layer_map.size();
-        }
-    }
-
-    size_type num_surfaces() const { return num_surf_; }
-
-  private:
-    std::vector<std::shared_ptr<SurfaceModel>>& models_;
-    size_type num_surf_{0};
-};
-
 //---------------------------------------------------------------------------//
 /*!
  * Calculate number of physics surfaces as defined by interstitial materials.
@@ -196,24 +134,25 @@ auto SurfacePhysicsParams::build_models(
     for (auto step : range(SurfacePhysicsOrder::size_))
     {
         // Build fake models
-        FakeModelBuilder build_model{step_models[step]};
+        detail::BuiltinSurfaceModelBuilder build_model{step_models[step]};
         switch (step)
         {
             case SurfacePhysicsOrder::roughness:
-                build_model("polished", input.roughness.polished);
-                build_model("smear", input.roughness.smear);
-                build_model("gaussian", input.roughness.gaussian);
+                build_model.build<PolishedRoughnessModel>(
+                    input.roughness.polished);
+                build_model.build<SmearRoughnessModel>(input.roughness.smear);
+                build_model.build<GaussianRoughnessModel>(
+                    input.roughness.gaussian);
                 break;
-
             case SurfacePhysicsOrder::reflectivity:
-                build_model("grid", input.reflectivity.grid);
-                build_model("fresnel", input.reflectivity.fresnel);
+                build_model.build_fake("grid", input.reflectivity.grid);
+                build_model.build_fake("fresnel", input.reflectivity.fresnel);
                 break;
             case SurfacePhysicsOrder::interaction:
-                build_model("dielectric-dielectric",
-                            input.interaction.dielectric_dielectric);
-                build_model("dielectric-metal",
-                            input.interaction.dielectric_metal);
+                build_model.build_fake("dielectric-dielectric",
+                                       input.interaction.dielectric_dielectric);
+                build_model.build_fake("dielectric-metal",
+                                       input.interaction.dielectric_metal);
                 break;
             default:
                 CELER_ASSERT_UNREACHABLE();