Получить значение варианта, который сам может быть другим вариантом - PullRequest
Новая
       101

Получить значение варианта, который сам может быть другим вариантом

1 голос
/ 27 февраля 2020

У меня есть вариант ScalarVar

using ScalarVar = std::variant<int, std::string>;

И вариант Var, который сам может быть ScalarVar или std::vector из ScalarVar s 

using Var = std::variant<ScalarVar, std::vector<ScalarVar>>;

Я хочу сделать функцию template<typename T, typename Variant> T Get(const Variant& var);, чтобы при задании варианта, который не включает внутренние варианты, он действовал так же, как std::get<T>, то есть он будет возвращать значение T, если Variant в настоящее время содержит T или если задан вариант, содержащий другие варианты, он будет рекурсивно получать содержащийся тип, пока не будет найден не вариант, а затем вернет его.

Вот моя лучшая попытка сделать это: 

#include <iostream>
#include <variant>
#include <string>
#include <typeindex>
#include <vector>
#include <map>

template<typename T, typename... T2>
struct is_variant { static inline constexpr bool value = false; };

template<typename... T>
struct is_variant<std::variant<T...>> { static inline constexpr bool value = true; };

template<typename T, typename Variant>
T Get(const Variant& var) {
    static_assert (is_variant<Variant>::value == true, "Template parameter Variant must be a std::variant");
    auto inner = std::visit([](const auto& i){ return i; }, var);
    if constexpr(is_variant<typeof(inner)>::value) {
        return Get<T>(inner);
    }
    else return inner;
}

int main()
{
    using ScalarVar = std::variant<int, std::string>;
    using Var = std::variant<ScalarVar, std::vector<ScalarVar>>;

    ScalarVar s = 5;
    std::cout << Get<int>(s) << std::endl;

    return 0;
}

Это должно просто вернуть T, если T не является std::variant, и вернуть std::get<InnerT>, если T - это std :: вариант, который содержит тип T.

Но я получаю очень сложную ошибку компиляции из g cc для строки:

std::cout << Get<int>(s) << std::endl

/usr/include/c++/9/variant:1005: error: invalid conversion from ‘std::__success_type<std::__cxx11::basic_string<char> >::type (*)(Get(const Variant&) [with T = int; Variant = std::variant<int, std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> > >]::<lambda(const auto:22&)>&&, const std::variant<int, std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> > >&)’ {aka ‘std::__cxx11::basic_string<char> (*)(Get(const Variant&) [with T = int; Variant = std::variant<int, std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> > >]::<lambda(const auto:22&)>&&, const std::variant<int, std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> > >&)’} to ‘int (*)(Get(const Variant&) [with T = int; Variant = std::variant<int, std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> > >]::<lambda(const auto:22&)>&&, const std::variant<int, std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> > >&)’ [-fpermissive]
 1005 |       { return _Array_type{&__visit_invoke}; }
      |                                           ^
      |                                           |
      |                                           std::__success_type<std::__cxx11::basic_string<char> >::type (*)(Get(const Variant&) [with T = int; Variant = std::variant<int, std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> > >]::<lambda(const auto:22&)>&&, const std::variant<int, std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> > >&) {aka std::__cxx11::basic_string<char> (*)(Get(const Variant&) [with T = int; Variant = std::variant<int, std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> > >]::<lambda(const auto:22&)>&&, const std::variant<int, std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> > >&)}

Как мне получить требуемое поведение?

Ответы [ 2 ]

6 голосов
/ 27 февраля 2020

Проблема в этой строке: 

std::visit([](const auto& i){ return i; }, var);

Вызов этого напрямую из main с ScalarVar var = 5; приводит к аналогичному сообщению.

Давайте посмотрим, что cppreference говорит о типе возвращаемого значения std::visit (C ++ 17):

Тип возвращаемого значения выводится из возвращенного выражения, как если бы он был decltype. Вызов некорректен, если приведенный выше вызов не является допустимым выражением того же типа и категории значения для всех комбинаций альтернативных типов всех вариантов.

Это означает, что вы не можете вызвать std::visit, как указано выше, поскольку он должен был возвращать разные типы (int / std::string) в зависимости от значения внутри.

Вы можете использовать перегрузку функций и использовать две различные функции для std::variant и других типов. : 

template <typename T, typename U>
T Get(U const& value)
{
    if constexpr (std::is_same_v<T, U>)
    {
        return value;
    }
    // If you get here, none of the nested variants had a value of type T.
    return T();
}

template <typename T, typename... Args>
T Get(std::variant<Args...> const& var)
{
    return std::visit(
        [](auto const& value) { return Get<T>(value); },
        var
    );
}
1 голос
/ 27 февраля 2020

Вам нужно различать guish между T, std::variant<...> и всем остальным, когда вы звоните std::visit. 

template<typename T>
struct Getter {
    T operator()(const T & t) { return t; }

    template <typename ... Ts>
    T operator()(const std::variant<Ts...> & var) { return std::visit(*this, var); }

    template <typename U>
    T operator()(U) { throw std::bad_variant_access(); }
};

template<typename T, typename Variant>
T Get(const Variant& var) {
    Getter<T> getter;
    return getter(var);
}

/* or 
template <typename T>
inline Getter<T> Get;
*/
...