BoundarySubset.hpp

#pragma once

#include <functional>

#include <amdis/Boundary.hpp>
#include <amdis/BoundaryManager.hpp>
#include <amdis/common/Concepts.hpp>
#include <amdis/common/ConceptsBase.hpp>

namespace AMDiS
{

  template <class IS>
  class BoundarySubset
  {
    using Domain = typename IS::GlobalCoordinate;

  public:
    using Intersection = IS;

    BoundarySubset()
      : predicate_([](Intersection const& is) -> bool { return is.boundary(); })
    {}

    BoundarySubset(BoundaryManagerBase& boundaryManager, BoundaryType id)
      : predicate_([&boundaryManager, id](Intersection const& is) -> bool {
                     return is.boundary() && boundaryManager.boundaryId(is) == id;
                     })
    {}

    template <class Predicate,
      REQUIRES(Concepts::Functor<Predicate, bool(Domain)>)>
    BoundarySubset(Predicate&& predicate)
      : predicate_([predicate](Intersection const& is) -> bool {
                     return predicate(is.geometry().center());
                     })
    {}

    bool operator()(Intersection const& is) const
    {
      return predicate_(is);
    }

  protected:
    std::shared_ptr<BoundaryManagerBase const> boundaryManager_;
    BoundaryType id_{0};
    std::function<bool(Intersection const&)> predicate_;
  };

} // end namespace AMDiS