complex.hpp 2.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091
  1. //
  2. // MessagePack for C++ static resolution routine
  3. //
  4. // Copyright (C) 2020 KONDO Takatoshi
  5. //
  6. // Distributed under the Boost Software License, Version 1.0.
  7. // (See accompanying file LICENSE_1_0.txt or copy at
  8. // http://www.boost.org/LICENSE_1_0.txt)
  9. //
  10. #ifndef MSGPACK_V1_TYPE_COMPLEX_HPP
  11. #define MSGPACK_V1_TYPE_COMPLEX_HPP
  12. #include <complex>
  13. #include "msgpack/versioning.hpp"
  14. #include "msgpack/adaptor/adaptor_base.hpp"
  15. #include "msgpack/meta.hpp"
  16. namespace msgpack {
  17. /// @cond
  18. MSGPACK_API_VERSION_NAMESPACE(v1) {
  19. /// @endcond
  20. namespace adaptor {
  21. #if !defined(MSGPACK_USE_CPP03)
  22. template <typename T>
  23. struct as<std::complex<T>, typename std::enable_if<msgpack::has_as<T>::value>::type> {
  24. std::complex<T> operator()(msgpack::object const& o) const {
  25. if (o.type != msgpack::type::ARRAY)
  26. throw msgpack::type_error();
  27. if (o.via.array.size != 2)
  28. throw msgpack::type_error();
  29. return std::complex<T>(o.via.array.ptr[0].as<T>(), o.via.array.ptr[1].as<T>());
  30. }
  31. };
  32. #endif // !defined(MSGPACK_USE_CPP03)
  33. template <typename T>
  34. struct convert<std::complex<T> > {
  35. msgpack::object const& operator()(msgpack::object const& o, std::complex<T>& v) const {
  36. if(o.type != msgpack::type::ARRAY)
  37. throw msgpack::type_error();
  38. if(o.via.array.size != 2)
  39. throw msgpack::type_error();
  40. T real;
  41. T imag;
  42. o.via.array.ptr[0].convert(real);
  43. o.via.array.ptr[1].convert(imag);
  44. v.real(real);
  45. v.imag(imag);
  46. return o;
  47. }
  48. };
  49. template <typename T>
  50. struct pack<std::complex<T> > {
  51. template <typename Stream>
  52. msgpack::packer<Stream>& operator()(msgpack::packer<Stream>& o, std::complex<T> const& v) const {
  53. o.pack_array(2);
  54. o.pack(v.real());
  55. o.pack(v.imag());
  56. return o;
  57. }
  58. };
  59. template <typename T>
  60. struct object_with_zone<std::complex<T> > {
  61. void operator()(msgpack::object::with_zone& o, std::complex<T> const& v) const {
  62. o.type = msgpack::type::ARRAY;
  63. msgpack::object* p = static_cast<msgpack::object*>(o.zone.allocate_align(sizeof(msgpack::object)*2, MSGPACK_ZONE_ALIGNOF(msgpack::object)));
  64. o.via.array.ptr = p;
  65. o.via.array.size = 2;
  66. p[0] = msgpack::object(v.real(), o.zone);
  67. p[1] = msgpack::object(v.imag(), o.zone);
  68. }
  69. };
  70. } // namespace adaptor
  71. /// @cond
  72. } // MSGPACK_API_VERSION_NAMESPACE(v1)
  73. /// @endcond
  74. } // namespace msgpack
  75. #endif // MSGPACK_V1_TYPE_COMPLEX_HPP