variant_visit.cpp 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126
  1. // Copyright 2017 Peter Dimov.
  2. //
  3. // Distributed under the Boost Software License, Version 1.0.
  4. //
  5. // See accompanying file LICENSE_1_0.txt or copy at
  6. // http://www.boost.org/LICENSE_1_0.txt
  7. #if defined(_MSC_VER)
  8. # pragma warning( disable: 4244 ) // conversion from float to int, possible loss of data
  9. #endif
  10. #include <boost/variant2/variant.hpp>
  11. #include <boost/mp11.hpp>
  12. #include <boost/core/lightweight_test.hpp>
  13. #include <boost/core/lightweight_test_trait.hpp>
  14. #include <boost/config.hpp>
  15. #include <type_traits>
  16. #include <utility>
  17. #include <string>
  18. #include <cstdio>
  19. using namespace boost::variant2;
  20. using boost::mp11::mp_size_t;
  21. struct X
  22. {
  23. };
  24. struct F
  25. {
  26. mp_size_t<1> operator()( X& ) const;
  27. mp_size_t<2> operator()( X const& ) const;
  28. mp_size_t<3> operator()( X&& ) const;
  29. mp_size_t<4> operator()( X const&& ) const;
  30. };
  31. int main()
  32. {
  33. {
  34. variant<int> v( 1 );
  35. BOOST_TEST_EQ( (visit( []( int x ){ return x; }, v )), 1 );
  36. visit( []( int x ){ BOOST_TEST_EQ( x, 1 ); }, v );
  37. visit( []( int x ){ BOOST_TEST_EQ( x, 1 ); }, std::move(v) );
  38. }
  39. {
  40. variant<int> const v( 2 );
  41. BOOST_TEST_EQ( (visit( []( int x ){ return x; }, v )), 2 );
  42. visit( []( int x ){ BOOST_TEST_EQ( x, 2 ); }, v );
  43. visit( []( int x ){ BOOST_TEST_EQ( x, 2 ); }, std::move(v) );
  44. }
  45. {
  46. variant<int const> v( 3 );
  47. BOOST_TEST_EQ( (visit( []( int x ){ return x; }, v )), 3 );
  48. visit( []( int x ){ BOOST_TEST_EQ( x, 3 ); }, v );
  49. visit( []( int x ){ BOOST_TEST_EQ( x, 3 ); }, std::move(v) );
  50. }
  51. {
  52. variant<int const> const v( 4 );
  53. BOOST_TEST_EQ( (visit( []( int x ){ return x; }, v )), 4 );
  54. visit( []( int x ){ BOOST_TEST_EQ( x, 4 ); }, v );
  55. visit( []( int x ){ BOOST_TEST_EQ( x, 4 ); }, std::move(v) );
  56. }
  57. {
  58. variant<int, float> v1( 1 );
  59. variant<int, float> const v2( 3.14f );
  60. BOOST_TEST_EQ( (visit( []( int x1, float x2 ){ return (int)(x1 * 1000) + (int)(x2 * 100); }, v1, v2 )), 1314 );
  61. visit( []( int x1, float x2 ){ BOOST_TEST_EQ( x1, 1 ); BOOST_TEST_EQ( x2, 3.14f ); }, v1, v2 );
  62. visit( []( int x1, float x2 ){ BOOST_TEST_EQ( x1, 1 ); BOOST_TEST_EQ( x2, 3.14f ); }, std::move(v1), std::move(v2) );
  63. }
  64. {
  65. variant<int, float, double> v1( 1 );
  66. variant<int, float, double> const v2( 3.14f );
  67. variant<int, float, double> v3( 6.28 );
  68. BOOST_TEST_EQ( (visit( []( int x1, float x2, double x3 ){ return (int)(x1 * 100) * 1000000 + (int)(x2 * 100) * 1000 + (int)(x3 * 100); }, v1, v2, v3 )), 100314628 );
  69. visit( []( int x1, float x2, double x3 ){ BOOST_TEST_EQ( x1, 1 ); BOOST_TEST_EQ( x2, 3.14f ); BOOST_TEST_EQ( x3, 6.28 ); }, v1, v2, v3 );
  70. visit( []( int x1, float x2, double x3 ){ BOOST_TEST_EQ( x1, 1 ); BOOST_TEST_EQ( x2, 3.14f ); BOOST_TEST_EQ( x3, 6.28 ); }, std::move(v1), std::move(v2), std::move(v3) );
  71. }
  72. {
  73. variant<int, float, double, char> v1( 1 );
  74. variant<int, float, double, char> const v2( 3.14f );
  75. variant<int, float, double, char> v3( 6.28 );
  76. variant<int, float, double, char> const v4( 'A' );
  77. BOOST_TEST_EQ( (visit( []( int x1, float x2, double x3, char x4 ){ return (long long)(x1 * 100) * 100000000 + (long long)(x2 * 100) * 100000 + (long long)(x3 * 10000) + (int)x4; }, v1, v2, v3, v4 )), 10031462800 + 'A' );
  78. visit( []( int x1, float x2, double x3, char x4 ){ BOOST_TEST_EQ( x1, 1 ); BOOST_TEST_EQ( x2, 3.14f ); BOOST_TEST_EQ( x3, 6.28 ); BOOST_TEST_EQ( x4, 'A' ); }, v1, v2, v3, v4 );
  79. visit( []( int x1, float x2, double x3, char x4 ){ BOOST_TEST_EQ( x1, 1 ); BOOST_TEST_EQ( x2, 3.14f ); BOOST_TEST_EQ( x3, 6.28 ); BOOST_TEST_EQ( x4, 'A' ); }, std::move(v1), std::move(v2), std::move(v3), std::move(v4) );
  80. }
  81. {
  82. variant<X> v;
  83. variant<X> const cv;
  84. BOOST_TEST_EQ( decltype(visit(F{}, v))::value, 1 );
  85. BOOST_TEST_EQ( decltype(visit(F{}, cv))::value, 2 );
  86. BOOST_TEST_EQ( decltype(visit(F{}, std::move(v)))::value, 3 );
  87. #if !BOOST_WORKAROUND(BOOST_GCC, < 40900)
  88. // g++ 4.8 doesn't handle const&& particularly well
  89. BOOST_TEST_EQ( decltype(visit(F{}, std::move(cv)))::value, 4 );
  90. #endif
  91. }
  92. return boost::report_errors();
  93. }