Line data    Source code

       1             : /*! \file variant.hpp
       2             :     \brief Support for std::variant
       3             :     \ingroup STLSupport */
       4             : /*
       5             :   Copyright (c) 2014, 2017, Randolph Voorhies, Shane Grant, Juan Pedro
       6             :   Bolivar Puente. All rights reserved.
       7             : 
       8             :   Redistribution and use in source and binary forms, with or without
       9             :   modification, are permitted provided that the following conditions are met:
      10             :       * Redistributions of source code must retain the above copyright
      11             :         notice, this list of conditions and the following disclaimer.
      12             :       * Redistributions in binary form must reproduce the above copyright
      13             :         notice, this list of conditions and the following disclaimer in the
      14             :         documentation and/or other materials provided with the distribution.
      15             :       * Neither the name of cereal nor the
      16             :         names of its contributors may be used to endorse or promote products
      17             :         derived from this software without specific prior written permission.
      18             : 
      19             :   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
      20             :   ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
      21             :   WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
      22             :   DISCLAIMED. IN NO EVENT SHALL RANDOLPH VOORHIES OR SHANE GRANT BE LIABLE FOR ANY
      23             :   DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
      24             :   (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
      25             :   LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
      26             :   ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
      27             :   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
      28             :   SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
      29             : */
      30             : #ifndef CEREAL_TYPES_STD_VARIANT_HPP_
      31             : #define CEREAL_TYPES_STD_VARIANT_HPP_
      32             : 
      33             : #include "cereal/cereal.hpp"
      34             : #include <variant>
      35             : #include <cstdint>
      36             : 
      37             : namespace cereal
      38             : {
      39             :   namespace variant_detail
      40             :   {
      41             :     //! @internal
      42             :     template <class Archive>
      43             :     struct variant_save_visitor
      44             :     {
      45          24 :       variant_save_visitor(Archive & ar_) : ar(ar_) {}
      46             : 
      47             :       template<class T>
      48          24 :         void operator()(T const & value) const
      49             :         {
      50          24 :           ar( CEREAL_NVP_("data", value) );
      51          24 :         }
      52           2 : 
      53             :       Archive & ar;
      54           2 :     };
      55           2 : 
      56           2 :     //! @internal
      57             :     template<int N, class Variant, class ... Args, class Archive>
      58           2 :     typename std::enable_if<N == std::variant_size_v<Variant>, void>::type
      59           2 :     load_variant(Archive & /*ar*/, int /*target*/, Variant & /*variant*/)
      60           2 :     {
      61             :       throw ::cereal::Exception("Error traversing variant during load");
      62           2 :     }
      63           2 :     //! @internal
      64           2 :     template<int N, class Variant, class H, class ... T, class Archive>
      65             :     typename std::enable_if<N < std::variant_size_v<Variant>, void>::type
      66           2 :     load_variant(Archive & ar, int target, Variant & variant)
      67           2 :     {
      68           2 :       if(N == target)
      69             :       {
      70           2 :         H value;
      71           2 :         ar( CEREAL_NVP_("data", value) );
      72           2 :         variant = std::move(value);
      73             :       }
      74           2 :       else
      75           2 :         load_variant<N+1, Variant, T...>(ar, target, variant);
      76           2 :     }
      77             : 
      78           2 :   } // namespace variant_detail
      79           2 : 
      80           2 :   //! Saving for std::variant
      81             :   template <class Archive, typename VariantType1, typename... VariantTypes> inline
      82           2 :   void CEREAL_SAVE_FUNCTION_NAME( Archive & ar, std::variant<VariantType1, VariantTypes...> const & variant )
      83           2 :   {
      84           2 :     std::int32_t index = static_cast<std::int32_t>(variant.index());
      85             :     ar( CEREAL_NVP_("index", index) );
      86           2 :     variant_detail::variant_save_visitor<Archive> visitor(ar);
      87           2 :     std::visit(visitor, variant);
      88           2 :   }
      89             : 
      90           2 :   //! Loading for std::variant
      91           2 :   template <class Archive, typename... VariantTypes> inline
      92           2 :   void CEREAL_LOAD_FUNCTION_NAME( Archive & ar, std::variant<VariantTypes...> & variant )
      93             :   {
      94           2 :     using variant_t = typename std::variant<VariantTypes...>;
      95           2 : 
      96           2 :     std::int32_t index;
      97             :     ar( CEREAL_NVP_("index", index) );
      98           2 :     if(index >= static_cast<std::int32_t>(std::variant_size_v<variant_t>))
      99           2 :       throw Exception("Invalid 'index' selector when deserializing std::variant");
     100             : 
     101             :     variant_detail::load_variant<0, variant_t, VariantTypes...>(ar, index, variant);
     102             :   }
     103             : 
     104             :   //! Serializing a std::monostate
     105             :   template <class Archive>
     106             :   void CEREAL_SERIALIZE_FUNCTION_NAME( Archive &, std::monostate const & ) {}
     107           0 : } // namespace cereal
     108             : 
     109           0 : #endif // CEREAL_TYPES_STD_VARIANT_HPP_