IsInSymmetricSlab.hpp

Go to the documentation of this file.

// Copyright 2024 Sebastian Eibl
// Copyright 2026 Julian Friedrich Hille
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
 
#pragma once
 
#include "datatypes.hpp"
 
namespace mrmd
{
namespace util
{
class IsInSymmetricSlab
{
private:
    const real_t center_;
    const real_t slabMin_;
    const real_t slabMax_;
    const AXIS axis_;
    const real_t tolerance_;
 
public:
    KOKKOS_INLINE_FUNCTION
    bool operator()(const real_t& x, const real_t& y, const real_t& z) const
    {
        auto dx = Point3D{x, y, z}[to_underlying(axis_)] - center_;
        auto absDx = std::abs(dx);
 
        return (absDx >= slabMin_ - tolerance_ && absDx <= slabMax_ + tolerance_);
    }
 
    KOKKOS_INLINE_FUNCTION
    bool operator()(const real_t& coord) const
    {
        auto dx = coord - center_;
        auto absDx = std::abs(dx);
 
        return (absDx >= slabMin_ - tolerance_ && absDx <= slabMax_ + tolerance_);
    }
 
    IsInSymmetricSlab(const Point3D& center,
                      const real_t slabMin,
                      const real_t slabMax,
                      const AXIS axis = AXIS::X,
                      const real_t tolerance = 0_r)
        : center_(center[to_underlying(axis)]),
          slabMin_(slabMin),
          slabMax_(slabMax),
          axis_(axis),
          tolerance_(tolerance)
    {
    }
};
}  // namespace util
}  // namespace mrmd