Marshall Clow | 40a097d | 2016-01-28 04:14:56 +0000 | [diff] [blame] | 1 | //===----------------------------------------------------------------------===// |
| 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
| 5 | // This file is dual licensed under the MIT and the University of Illinois Open |
| 6 | // Source Licenses. See LICENSE.TXT for details. |
| 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | |
| 10 | // UNSUPPORTED: c++98, c++03, c++11 |
| 11 | |
| 12 | // <experimental/iterator> |
| 13 | // |
| 14 | // template <class _Delim, class _CharT = char, class _Traits = char_traits<_CharT>> |
| 15 | // class ostream_joiner; |
| 16 | // |
| 17 | // template<typename T> |
| 18 | // ostream_joiner & operator=(const T&) |
| 19 | // |
| 20 | |
| 21 | #include <experimental/iterator> |
| 22 | #include <iostream> |
| 23 | #include <sstream> |
| 24 | #include <cassert> |
| 25 | |
| 26 | #include "test_macros.h" |
| 27 | #include "test_iterators.h" |
| 28 | |
| 29 | struct mutating_delimiter { |
| 30 | mutating_delimiter(char c = ' ') : c_(c) {} |
| 31 | char get () const { return c_++; } |
| 32 | mutable char c_; |
| 33 | }; |
| 34 | |
| 35 | template<class _CharT, class _Traits> |
| 36 | std::basic_ostream<_CharT, _Traits>& |
| 37 | operator<<(std::basic_ostream<_CharT, _Traits>& os, const mutating_delimiter &d) |
| 38 | { return os << d.get(); } |
| 39 | |
| 40 | |
| 41 | struct mutating_delimiter2 { // same as above, w/o the const and the mutable |
| 42 | mutating_delimiter2(char c = ' ') : c_(c) {} |
| 43 | char get () { return c_++; } |
| 44 | char c_; |
| 45 | }; |
| 46 | |
| 47 | template<class _CharT, class _Traits> |
| 48 | std::basic_ostream<_CharT, _Traits>& |
| 49 | operator<<(std::basic_ostream<_CharT, _Traits>& os, mutating_delimiter2 &d) |
| 50 | { return os << d.get(); } |
| 51 | |
| 52 | |
| 53 | namespace exp = std::experimental; |
| 54 | |
| 55 | template <class Delim, class Iter, class CharT = char, class Traits = std::char_traits<CharT>> |
| 56 | void test (Delim &&d, Iter first, Iter last, const CharT *expected ) { |
| 57 | typedef exp::ostream_joiner<typename std::decay<Delim>::type, CharT, Traits> Joiner; |
| 58 | |
| 59 | static_assert((std::is_copy_constructible<Joiner>::value == std::is_copy_constructible<typename std::decay<Delim>::type>::value), "" ); |
| 60 | static_assert((std::is_move_constructible<Joiner>::value == std::is_move_constructible<typename std::decay<Delim>::type>::value), "" ); |
| 61 | static_assert((std::is_copy_assignable<Joiner> ::value == std::is_copy_assignable<typename std::decay<Delim>::type> ::value), "" ); |
| 62 | static_assert((std::is_move_assignable<Joiner> ::value == std::is_move_assignable<typename std::decay<Delim>::type> ::value), "" ); |
| 63 | |
| 64 | std::basic_stringstream<CharT, Traits> sstream; |
| 65 | Joiner joiner(sstream, d); |
| 66 | while (first != last) |
| 67 | *joiner++ = *first++; |
| 68 | assert(sstream.str() == expected); |
| 69 | } |
| 70 | |
| 71 | int main () { |
| 72 | { |
| 73 | const char chars[] = "0123456789"; |
| 74 | const int ints [] = { 10, 11, 12, 13, 14, 15, 16, 17, 18, 19 }; |
| 75 | |
| 76 | test('X', chars, chars+10, "0X1X2X3X4X5X6X7X8X9"); |
| 77 | test('x', ints, ints+10, "10x11x12x13x14x15x16x17x18x19"); |
| 78 | test('X', input_iterator<const char*>(chars), input_iterator<const char*>(chars+10), "0X1X2X3X4X5X6X7X8X9"); |
| 79 | test('x', input_iterator<const int*>(ints), input_iterator<const int*>(ints+10), "10x11x12x13x14x15x16x17x18x19"); |
| 80 | test('X', forward_iterator<const char*>(chars), forward_iterator<const char*>(chars+10), "0X1X2X3X4X5X6X7X8X9"); |
| 81 | test('x', forward_iterator<const int*>(ints), forward_iterator<const int*>(ints+10), "10x11x12x13x14x15x16x17x18x19"); |
| 82 | test('X', random_access_iterator<const char*>(chars), random_access_iterator<const char*>(chars+10), "0X1X2X3X4X5X6X7X8X9"); |
| 83 | test('x', random_access_iterator<const int*>(ints), random_access_iterator<const int*>(ints+10), "10x11x12x13x14x15x16x17x18x19"); |
| 84 | |
| 85 | test("Z", chars, chars+10, "0Z1Z2Z3Z4Z5Z6Z7Z8Z9"); |
| 86 | test("z", ints, ints+10, "10z11z12z13z14z15z16z17z18z19"); |
| 87 | |
| 88 | test<char, const char *, wchar_t> ('X', chars, chars+10, L"0X1X2X3X4X5X6X7X8X9"); |
| 89 | test<char, const int *, wchar_t> ('x', ints, ints+10, L"10x11x12x13x14x15x16x17x18x19"); |
| 90 | // test<char, const char *, char16_t>('X', chars, chars+10, u"0X1X2X3X4X5X6X7X8X9"); |
| 91 | // test<char, const int *, char16_t>('x', ints, ints+10, u"10x11x12x13x14x15x16x17x18x19"); |
| 92 | // test<char, const char *, char32_t>('X', chars, chars+10, U"0X1X2X3X4X5X6X7X8X9"); |
| 93 | // test<char, const int *, char32_t>('x', ints, ints+10, U"10x11x12x13x14x15x16x17x18x19"); |
| 94 | |
| 95 | test(mutating_delimiter(), chars, chars+10, "0 1!2\"3#4$5%6&7'8(9"); |
| 96 | test(mutating_delimiter2(), chars, chars+10, "0 1!2\"3#4$5%6&7'8(9"); |
| 97 | } |
| 98 | |
| 99 | { |
| 100 | const wchar_t chars[] = L"0123456789"; |
| 101 | const int ints [] = { 10, 11, 12, 13, 14, 15, 16, 17, 18, 19 }; |
| 102 | test(L'X', chars, chars+10, L"0X1X2X3X4X5X6X7X8X9"); |
| 103 | test(L'x', ints, ints+10, L"10x11x12x13x14x15x16x17x18x19"); |
| 104 | test(L'X', input_iterator<const wchar_t*>(chars), input_iterator<const wchar_t*>(chars+10), L"0X1X2X3X4X5X6X7X8X9"); |
| 105 | test(L'x', input_iterator<const int*>(ints), input_iterator<const int*>(ints+10), L"10x11x12x13x14x15x16x17x18x19"); |
| 106 | test(L'X', forward_iterator<const wchar_t*>(chars), forward_iterator<const wchar_t*>(chars+10), L"0X1X2X3X4X5X6X7X8X9"); |
| 107 | test(L'x', forward_iterator<const int*>(ints), forward_iterator<const int*>(ints+10), L"10x11x12x13x14x15x16x17x18x19"); |
| 108 | test(L'X', random_access_iterator<const wchar_t*>(chars), random_access_iterator<const wchar_t*>(chars+10), L"0X1X2X3X4X5X6X7X8X9"); |
| 109 | test(L'x', random_access_iterator<const int*>(ints), random_access_iterator<const int*>(ints+10), L"10x11x12x13x14x15x16x17x18x19"); |
| 110 | |
| 111 | test(L"Z", chars, chars+10, L"0Z1Z2Z3Z4Z5Z6Z7Z8Z9"); |
| 112 | test(L"z", ints, ints+10, L"10z11z12z13z14z15z16z17z18z19"); |
| 113 | |
| 114 | test<char, const wchar_t *, wchar_t> ('X', chars, chars+10, L"0X1X2X3X4X5X6X7X8X9"); |
| 115 | test<char, const int *, wchar_t> ('x', ints, ints+10, L"10x11x12x13x14x15x16x17x18x19"); |
| 116 | |
| 117 | test(mutating_delimiter(), chars, chars+10, L"0 1!2\"3#4$5%6&7'8(9"); |
| 118 | } |
| 119 | |
| 120 | } |