Sprout/sprout/variant/as_visitor.hpp
2013-06-17 23:44:18 +09:00

82 lines
2.6 KiB
C++

#ifndef SPROUT_VARIANT_AS_VISITOR_HPP
#define SPROUT_VARIANT_AS_VISITOR_HPP
#include <utility>
#include <type_traits>
#include <sprout/config.hpp>
#include <sprout/utility/forward.hpp>
#include <sprout/utility/lvalue_forward.hpp>
#include <sprout/utility/value_holder/value_holder.hpp>
#include <sprout/type_traits/lvalue_reference.hpp>
#include <sprout/variant/static_visitor.hpp>
#include <sprout/functional/type_traits/has_type.hpp>
#include <sprout/functional/type_traits/weak_result_type.hpp>
namespace sprout {
namespace detail {
template<typename T, typename = void>
struct result_type_or_void;
template<typename T>
struct result_type_or_void<
T,
typename std::enable_if<sprout::has_result_type<sprout::weak_result_type<T> >::value>::type
> {
public:
typedef typename sprout::weak_result_type<T>::result_type type;
};
template<typename T>
struct result_type_or_void<
T,
typename std::enable_if<!sprout::has_result_type<sprout::weak_result_type<T> >::value>::type
> {
public:
typedef void type;
};
} // namespace detail
//
// visitor_wrapper
//
template<
typename Visitor,
typename R = typename sprout::detail::result_type_or_void<Visitor>::type
>
class visitor_wrapper
: public sprout::static_visitor<R>
{
public:
typedef Visitor visitor_type;
typedef sprout::value_holder<visitor_type&> holder_type;
private:
holder_type v_;
public:
SPROUT_CONSTEXPR explicit visitor_wrapper(typename holder_type::param_type v)
: v_(v)
{}
template<typename T>
SPROUT_CONSTEXPR auto operator()(T&& t) const
-> decltype(std::declval<holder_type const>().get()(sprout::forward<T>(t)))
{
return v_.get()(sprout::forward<T>(t));
}
};
//
// as_visitor
//
template<typename Visitor>
inline SPROUT_CONSTEXPR sprout::visitor_wrapper<typename std::remove_reference<typename sprout::lvalue_reference<Visitor>::type>::type>
as_visitor(Visitor&& visitor) {
typedef sprout::visitor_wrapper<typename std::remove_reference<typename sprout::lvalue_reference<Visitor>::type>::type> type;
return type(sprout::lvalue_forward<Visitor>(visitor));
}
template<typename R, typename Visitor>
inline SPROUT_CONSTEXPR sprout::visitor_wrapper<typename std::remove_reference<typename sprout::lvalue_reference<Visitor>::type>::type, R>
as_visitor(Visitor&& visitor) {
typedef sprout::visitor_wrapper<typename std::remove_reference<typename sprout::lvalue_reference<Visitor>::type>::type, R> type;
return type(sprout::lvalue_forward<Visitor>(visitor));
}
} // namespace sprout
#endif // #ifndef SPROUT_VARIANT_AS_VISITOR_HPP