Search code examples
c++streamc++17variant

How to stream std::variant<...,...>


My std::variant contains streamable types:

std::variant<int, std::string> a, b;
a = 1;
b = "hi";
std::cout << a << b << std::endl;

Compiling with g++7 with -std=c++1z returns compilation time errors.

An excerpt:

test.cpp: In function 'int main(int, char**)':
test.cpp:10:13: error: no match for 'operator<<' (operand types are 'std::ostream {aka std::basic_ostream<char>}' and 'std::variant<int, std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> > >')
   std::cout << a << b << std::endl;
   ~~~~~~~~~~^~~~

Seemingly a std::variant<int, std::string> is not able to stream. How can I achieve that I can directly stream the variant to an output stream?

Expected output:

1hi

Solution

  • This streams nested variants too.

    template<class T>
    struct streamer {
        const T& val;
    };
    template<class T> streamer(T) -> streamer<T>;
    
    template<class T>
    std::ostream& operator<<(std::ostream& os, streamer<T> s) {
        os << s.val;
        return os;
    }
    
    template<class... Ts>
    std::ostream& operator<<(std::ostream& os, streamer<std::variant<Ts...>> sv) {
       std::visit([&os](const auto& v) { os << streamer{v}; }, sv.val);
       return os;
    }
    

    Use as:

    std::cout << streamer{a} << streamer{b} << '\n';