overloadset.hh

// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
#ifndef DUNE_COMMON_OVERLOADSET_HH
#define DUNE_COMMON_OVERLOADSET_HH
 
#include <utility>
#include <type_traits>
 
#include <dune/common/std/type_traits.hh>
 
 
 
namespace Dune {
 
namespace Impl {
 
  // This overload set derives from
  // all passed functions. Since we
  // cannot do argument pack expansion
  // on using statements this is done recursively.
  template<class F0, class... F>
  class OverloadSet: public OverloadSet<F...>, F0
  {
    using Base = OverloadSet<F...>;
  public:
 
    template<class FF0, class... FF>
    OverloadSet(FF0&& f0, FF&&... ff) :
      Base(std::forward<FF>(ff)...),
      F0(std::forward<FF0>(f0))
    {}
 
    // pull in operator() of F0 and of all F... via the base class
    using F0::operator();
    using Base::operator();
  };
 
  template<class F0>
  class OverloadSet<F0>: public F0
  {
  public:
 
    template<class FF0>
    OverloadSet(FF0&& f0) :
      F0(std::forward<FF0>(f0))
    {}
 
    // pull in operator() of F0
    using F0::operator();
  };
 
} // end namespace Impl
 
 
 
template<class... F>
auto overload(F&&... f)
{
  return Impl::OverloadSet<std::decay_t<F>...>(std::forward<F>(f)...);
}
 
 
 
namespace Impl {
 
  template<class F0, class... F>
  class OrderedOverloadSet: public OrderedOverloadSet<F...>, F0
  {
    using Base = OrderedOverloadSet<F...>;
  public:
 
    template<class FF0, class... FF>
    OrderedOverloadSet(FF0&& f0, FF&&... ff) :
      Base(std::forward<FF>(ff)...),
      F0(std::forward<FF0>(f0))
    {}
 
    // Forward to operator() of F0 if it can be called with the given arguments.
    template<class...  Args,
        std::enable_if_t<Std::is_callable<F0(Args&&...)>::value, int> = 0>
    decltype(auto) operator()(Args&&... args)
    {
      return F0::operator()(std::forward<Args>(args)...);
    }
 
    // Forward to operator() of base class if F0 cannot be called with the given
    // arguments. In this case the base class will successively try operator()
    // of all F... .
    template<class...  Args,
        std::enable_if_t< not Std::is_callable<F0(Args&&...)>::value, int> = 0>
    decltype(auto) operator()(Args&&... args)
    {
      return Base::operator()(std::forward<Args>(args)...);
    }
 
  };
 
  template<class F0>
  class OrderedOverloadSet<F0>: public F0
  {
  public:
 
    template<class FF0>
    OrderedOverloadSet(FF0&& f0) :
      F0(std::forward<FF0>(f0))
    {}
 
    // Forward to operator() of F0. If it cannot be called with
    // the given arguments a static assertion will fail.
    template<class...  Args>
    decltype(auto) operator()(Args&&... args)
    {
      static_assert(Std::is_callable<F0(Args&&...)>::value, "No matching overload found in OrderedOverloadSet");
      return F0::operator()(std::forward<Args>(args)...);
    }
  };
 
} // end namespace Impl
 
 
 
template<class... F>
auto orderedOverload(F&&... f)
{
  return Impl::OrderedOverloadSet<std::decay_t<F>...>(std::forward<F>(f)...);
}
 
 
 
} // end namespace Dune
 
#endif // DUNE_COMMON_OVERLOADSET_HH