tuple 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479
  1. #ifndef __GBLIBCPP_TUPLE__
  2. #define __GBLIBCPP_TUPLE__
  3. #include <cstddef>
  4. #include <type_traits>
  5. #include <utility>
  6. #include <functional>
  7. #include <bits/tuple_tools>
  8. #include <bits/sequence>
  9. namespace std {
  10. template <std::size_t I, typename T>
  11. struct tuple_element<I, const T> {
  12. using type = std::add_const_t<tuple_element_t<I, T>>;
  13. };
  14. template <std::size_t I, typename T>
  15. struct tuple_element<I, volatile T> {
  16. using type = std::add_volatile_t<tuple_element_t<I, T>>;
  17. };
  18. template <std::size_t I, typename T>
  19. struct tuple_element<I, const volatile T> {
  20. using type = std::add_cv_t<tuple_element_t<I, T>>;
  21. };
  22. namespace __helpers {
  23. template <typename T, typename = void>
  24. struct __const_tuple_size {};
  25. template <typename T>
  26. struct __const_tuple_size<T, void_t<decltype(std::tuple_size<T>::value)>>
  27. : std::integral_constant<std::size_t, std::tuple_size<T>::value> {};
  28. } // namespace __helpers
  29. template <typename T>
  30. struct tuple_size<const T> : __helpers::__const_tuple_size<T> {};
  31. template <typename... Types>
  32. class tuple;
  33. template <std::size_t I, typename Type, typename... Types>
  34. struct tuple_element<I, std::tuple<Type, Types...>>
  35. : tuple_element<I-1, std::tuple<Types...>> {};
  36. template <typename Type, typename... Types>
  37. struct tuple_element<0, std::tuple<Type, Types...>> { using type = Type; };
  38. template <typename... Types>
  39. struct tuple_size<std::tuple<Types...>>
  40. : std::integral_constant<std::size_t, sizeof...(Types)> {};
  41. namespace __helpers {
  42. template <std::size_t I, typename... Types>
  43. class tuple_impl;
  44. template <>
  45. class tuple_impl<0> {
  46. template <std::size_t, typename...>
  47. friend class tuple_impl;
  48. template <std::size_t J, typename... UTypes>
  49. friend constexpr auto std::get(std::tuple<UTypes...>& tpl) noexcept
  50. -> tuple_element_t<J, std::tuple<UTypes...>>&;
  51. template <std::size_t J, typename... UTypes>
  52. friend constexpr auto std::get(std::tuple<UTypes...>&& tpl) noexcept
  53. -> tuple_element_t<J, std::tuple<UTypes...>>&&;
  54. template <std::size_t J, typename... UTypes>
  55. friend constexpr auto std::get(const std::tuple<UTypes...>& tpl) noexcept
  56. -> tuple_element_t<J, std::tuple<UTypes...>> const&;
  57. template <std::size_t J, typename... UTypes>
  58. friend constexpr auto std::get(const std::tuple<UTypes...>&& tpl) noexcept
  59. -> tuple_element_t<J, std::tuple<UTypes...>> const&&;
  60. template <std::size_t> constexpr void _getl(void) const = delete;
  61. template <std::size_t> constexpr void _getr(void) const = delete;
  62. public:
  63. constexpr tuple_impl() noexcept = default;
  64. constexpr tuple_impl(const tuple_impl& arg) noexcept = default;
  65. constexpr tuple_impl(tuple_impl&& arg) noexcept = default;
  66. template <typename... UTypes>
  67. constexpr tuple_impl(const std::tuple<UTypes...>& other) = delete;
  68. template <typename... UTypes>
  69. constexpr tuple_impl(std::tuple<UTypes...>&& other) = delete;
  70. constexpr tuple_impl& operator=(const tuple_impl& other) noexcept = default;
  71. constexpr tuple_impl& operator=(tuple_impl&& other) noexcept = default;
  72. };
  73. template <std::size_t I, typename Type, typename... Types>
  74. class tuple_impl<I, Type, Types...> {
  75. template <std::size_t, typename...>
  76. friend class tuple_impl;
  77. template <std::size_t J, typename... UTypes>
  78. friend constexpr auto std::get(std::tuple<UTypes...>& tpl) noexcept
  79. -> tuple_element_t<J, std::tuple<UTypes...>>&;
  80. template <std::size_t J, typename... UTypes>
  81. friend constexpr auto std::get(std::tuple<UTypes...>&& tpl) noexcept
  82. -> tuple_element_t<J, std::tuple<UTypes...>>&&;
  83. template <std::size_t J, typename... UTypes>
  84. friend constexpr auto std::get(const std::tuple<UTypes...>& tpl) noexcept
  85. -> tuple_element_t<J, std::tuple<UTypes...>> const&;
  86. template <std::size_t J, typename... UTypes>
  87. friend constexpr auto std::get(const std::tuple<UTypes...>&& tpl) noexcept
  88. -> tuple_element_t<J, std::tuple<UTypes...>> const&&;
  89. Type val;
  90. tuple_impl<I+1, Types...> next;
  91. template <std::size_t J>
  92. constexpr auto& _getl(void)
  93. {
  94. if constexpr (I == J)
  95. return val;
  96. else
  97. return next.template _getl<J>();
  98. }
  99. template <std::size_t J>
  100. constexpr const auto& _getl(void) const
  101. {
  102. if constexpr (I == J)
  103. return val;
  104. else
  105. return next.template _getl<J>();
  106. }
  107. template <std::size_t J>
  108. constexpr auto&& _getr(void)
  109. {
  110. if constexpr (I == J)
  111. return std::move(val);
  112. else
  113. return std::move(next.template _getr<J>());
  114. }
  115. template <std::size_t J>
  116. constexpr const auto&& _getr(void) const
  117. {
  118. if constexpr (I == J)
  119. return std::move(val);
  120. else
  121. return std::move(next.template _getr<J>());
  122. }
  123. public:
  124. constexpr tuple_impl()
  125. : val(), next() {}
  126. constexpr tuple_impl(const tuple_impl& arg) = default;
  127. constexpr tuple_impl(tuple_impl&& arg) = default;
  128. constexpr tuple_impl(const Type& val, const Types&... vals)
  129. : val(val), next(vals...) {}
  130. template <typename UType, typename... UTypes>
  131. constexpr tuple_impl(UType&& val, UTypes&&... vals)
  132. : val(std::forward<UType>(val)), next(std::forward<UTypes>(vals)...) {}
  133. template <typename... UTypes>
  134. constexpr tuple_impl(const std::tuple<UTypes...>& other)
  135. : val(std::get<I>(other)), next(other) {}
  136. template <typename... UTypes>
  137. constexpr tuple_impl(std::tuple<UTypes...>&& other)
  138. : val(std::get<I>(other)), next(std::move(other)) {}
  139. constexpr tuple_impl& operator=(const tuple_impl& other) = default;
  140. constexpr tuple_impl& operator=(tuple_impl&& other) = default;
  141. };
  142. template <std::size_t I, typename Type>
  143. class tuple_impl<I, Type> {
  144. template <std::size_t, typename...>
  145. friend class tuple_impl;
  146. template <std::size_t J, typename... UTypes>
  147. friend constexpr auto std::get(std::tuple<UTypes...>& tpl) noexcept
  148. -> tuple_element_t<J, std::tuple<UTypes...>>&;
  149. template <std::size_t J, typename... UTypes>
  150. friend constexpr auto std::get(std::tuple<UTypes...>&& tpl) noexcept
  151. -> tuple_element_t<J, std::tuple<UTypes...>>&&;
  152. template <std::size_t J, typename... UTypes>
  153. friend constexpr auto std::get(const std::tuple<UTypes...>& tpl) noexcept
  154. -> tuple_element_t<J, std::tuple<UTypes...>> const&;
  155. template <std::size_t J, typename... UTypes>
  156. friend constexpr auto std::get(const std::tuple<UTypes...>&& tpl) noexcept
  157. -> tuple_element_t<J, std::tuple<UTypes...>> const&&;
  158. Type val;
  159. template <std::size_t J>
  160. constexpr Type& _getl(void)
  161. {
  162. static_assert(J == I);
  163. return val;
  164. }
  165. template <std::size_t J>
  166. constexpr Type const& _getl(void) const
  167. {
  168. static_assert(J == I);
  169. return val;
  170. }
  171. template <std::size_t J>
  172. constexpr Type&& _getr(void)
  173. {
  174. static_assert(J == I);
  175. return std::move(val);
  176. }
  177. template <std::size_t J>
  178. constexpr Type const&& _getr(void) const
  179. {
  180. static_assert(J == I);
  181. return std::move(val);
  182. }
  183. public:
  184. constexpr tuple_impl()
  185. : val() {}
  186. constexpr tuple_impl(const tuple_impl& arg) = default;
  187. constexpr tuple_impl(tuple_impl&& arg) = default;
  188. constexpr tuple_impl(const Type& val) : val(val) {}
  189. template <typename UType>
  190. constexpr tuple_impl(UType&& val) : val(std::forward<UType>(val)) {}
  191. template <typename... UTypes>
  192. constexpr tuple_impl(const std::tuple<UTypes...>& other)
  193. : val(std::get<I>(other)) {}
  194. template <typename... UTypes>
  195. constexpr tuple_impl(std::tuple<UTypes...>&& other)
  196. : val(std::get<I>(other)) {}
  197. constexpr tuple_impl& operator=(const tuple_impl& other) = default;
  198. constexpr tuple_impl& operator=(tuple_impl&& other) = default;
  199. };
  200. template <typename... Types>
  201. struct tuple_constraints {
  202. static constexpr std::size_t Tsize = sizeof...(Types);
  203. template <std::size_t... I>
  204. static constexpr auto idx_seq(std::index_sequence<I...> seq)
  205. { return seq; }
  206. template <typename UTuple, typename Seq, typename... UTypes>
  207. struct tuple_copy_move_constructible_impl;
  208. template <typename UTuple, typename... UTypes, std::size_t... I>
  209. struct tuple_copy_move_constructible_impl<UTuple,
  210. std::index_sequence<I...>, UTypes...> : std::bool_constant<
  211. Tsize == sizeof...(UTypes)
  212. && (... && std::is_constructible_v<Types, decltype(
  213. std::get<I>(std::declval<UTuple>()))>)
  214. && ((Tsize != 1) || (!(... && std::is_convertible_v<UTuple, Types>)
  215. && !(... && std::is_constructible_v<Types, UTuple>)
  216. && !(... && std::is_same_v<Types, UTypes>)))
  217. > {};
  218. template <typename UTuple, typename... UTypes>
  219. struct tuple_copy_move_constructible : tuple_copy_move_constructible_impl<
  220. UTuple, decltype(idx_seq(std::index_sequence_for<UTypes...>{})), UTypes...> {};
  221. template <typename Pair, typename = std::index_sequence<0, 1>>
  222. struct tuple_pair_constructible;
  223. template <typename Pair, std::size_t... I>
  224. struct tuple_pair_constructible<Pair, std::index_sequence<I...>>
  225. : std::bool_constant<sizeof...(Types) == 2
  226. && (... && std::is_constructible_v<Types,
  227. decltype(std::get<I>(std::declval<Pair>()))>)
  228. > {};
  229. template <typename... Us>
  230. struct tuple_pair_copy_assignable : std::bool_constant<
  231. sizeof...(Types) == 2
  232. && (... && std::is_assignable_v<Types&, const Us&>)
  233. > {};
  234. template <typename... Us>
  235. struct tuple_pair_move_assignable : std::bool_constant<
  236. sizeof...(Types) == 2
  237. && (... && std::is_assignable_v<Types&, Us>)
  238. > {};
  239. };
  240. } // namespace __helpers
  241. template <typename... Types>
  242. class tuple : public __helpers::tuple_impl<0, Types...> {
  243. private:
  244. using base = __helpers::tuple_impl<0, Types...>;
  245. template <typename... UTypes, std::size_t... I>
  246. constexpr void __copy_assignment(
  247. const tuple<UTypes...>& other, std::index_sequence<I...>)
  248. { ((std::get<I>(*this) = std::get<I>(other)), ...); }
  249. template <typename... UTypes, std::size_t... I>
  250. constexpr void __move_assignment(
  251. tuple<UTypes...>&& other, std::index_sequence<I...>)
  252. {
  253. ((std::get<I>(*this) = std::forward<UTypes>(std::get<I>(other))), ...);
  254. }
  255. public:
  256. template <typename _ = void, std::enable_if_t<
  257. (... && std::is_default_constructible_v<Types>)
  258. && __helpers::template_true_type<_>::value
  259. , bool> = true>
  260. constexpr tuple() : base() {}
  261. template <typename _ = void, std::enable_if_t<
  262. sizeof...(Types) >= 1
  263. && (... && std::is_copy_constructible_v<Types>)
  264. && __helpers::template_true_type<_>::value
  265. , bool> = true>
  266. constexpr tuple(const Types&... args)
  267. : base(args...) {}
  268. template <typename... UTypes, std::enable_if_t<
  269. sizeof...(UTypes) == sizeof...(Types)
  270. && sizeof...(Types) >= 1
  271. && (... && std::is_constructible_v<Types, UTypes>)
  272. && !(sizeof...(Types) == 1
  273. && (... && std::is_same_v<tuple, std::remove_cvref_t<UTypes>>))
  274. , bool> = true>
  275. constexpr tuple(UTypes&&... args)
  276. : base(std::forward<UTypes>(args)...) {}
  277. template <typename... UTypes, std::enable_if_t<
  278. __helpers::tuple_constraints<Types...>::template
  279. tuple_copy_move_constructible<const tuple<UTypes...>&, UTypes...>::value
  280. , bool> = true>
  281. constexpr tuple(const tuple<UTypes...>& other)
  282. : base(other) {}
  283. template <typename... UTypes, std::enable_if_t<
  284. __helpers::tuple_constraints<Types...>::template
  285. tuple_copy_move_constructible<tuple<UTypes...>&&, UTypes...>::value
  286. , bool> = true>
  287. constexpr tuple(tuple<UTypes...>&& other)
  288. : base(std::move(other)) {}
  289. template <typename U1, typename U2, std::enable_if_t<
  290. __helpers::tuple_constraints<Types...>::template
  291. tuple_pair_constructible<const std::pair<U1, U2>&>::value
  292. , bool> = true>
  293. constexpr tuple(const std::pair<U1, U2>& p)
  294. : base(std::get<0>(p), std::get<1>(p)) {}
  295. template <typename U1, typename U2, std::enable_if_t<
  296. __helpers::tuple_constraints<Types...>::template
  297. tuple_pair_constructible<std::pair<U1, U2>&&>::value
  298. , bool> = true>
  299. constexpr tuple(std::pair<U1, U2>&& p)
  300. : base(std::get<0>(std::move(p)), std::get<1>(std::move(p))) {}
  301. constexpr tuple(const tuple&) = default;
  302. constexpr tuple(tuple&&) = default;
  303. constexpr tuple& operator=(const tuple& other) = default;
  304. constexpr tuple& operator=(tuple&& other) = default;
  305. template <typename... UTypes, std::enable_if_t<
  306. sizeof...(Types) == sizeof...(UTypes)
  307. && (... && std::is_assignable_v<Types&, const UTypes&>)
  308. , bool> = true>
  309. constexpr tuple& operator=(const tuple<UTypes...>& other)
  310. {
  311. __copy_assignment(other, index_sequence_for<Types...> {});
  312. return *this;
  313. }
  314. template <typename... UTypes, std::enable_if_t<
  315. sizeof...(Types) == sizeof...(UTypes)
  316. && (... && std::is_assignable_v<Types&, UTypes>)
  317. , bool> = true>
  318. constexpr tuple& operator=(tuple<UTypes...>&& other)
  319. {
  320. __move_assignment(std::move(other), index_sequence_for<Types...> {});
  321. return *this;
  322. }
  323. template <typename U1, typename U2, std::enable_if_t<
  324. __helpers::tuple_constraints<Types...>::template
  325. tuple_pair_copy_assignable<U1, U2>::value
  326. , bool> = true>
  327. constexpr tuple& operator=(const std::pair<U1, U2>& p)
  328. {
  329. std::get<0>(*this) = p.first;
  330. std::get<1>(*this) = p.second;
  331. return *this;
  332. }
  333. template <typename U1, typename U2, std::enable_if_t<
  334. __helpers::tuple_constraints<Types...>::template
  335. tuple_pair_move_assignable<U1, U2>::value
  336. , bool> = true>
  337. constexpr tuple& operator=(std::pair<U1, U2>&& p)
  338. {
  339. std::get<0>(*this) = std::forward<U1>(p.first);
  340. std::get<1>(*this) = std::forward<U2>(p.second);
  341. return *this;
  342. }
  343. constexpr void swap(tuple& other) { std::swap(*this, other); }
  344. };
  345. template <typename... Types>
  346. tuple(Types...) -> tuple<Types...>;
  347. namespace __helpers {
  348. struct ignore_t {
  349. template <typename T>
  350. constexpr void operator=(T&&) const noexcept {}
  351. };
  352. template <typename Func, typename Tuple, std::size_t... I>
  353. constexpr auto apply_impl(Func&& func, Tuple&& tpl, std::index_sequence<I...>)
  354. {
  355. return std::invoke(
  356. std::forward<Func>(func),
  357. std::get<I>(std::forward<Tuple>(tpl))...);
  358. }
  359. } // namespace __helpers
  360. inline constexpr __helpers::ignore_t ignore;
  361. template <typename... Types>
  362. constexpr void swap(std::tuple<Types...>& lhs, std::tuple<Types...>& rhs)
  363. noexcept(noexcept(lhs.swap(rhs)))
  364. {
  365. lhs.swap(rhs);
  366. }
  367. template <typename... Types>
  368. constexpr auto make_tuple(Types&&... args)
  369. -> std::tuple<__helpers::to_tuple_type_t<Types>...>
  370. {
  371. return std::tuple<__helpers::to_tuple_type_t<Types>...>
  372. { std::forward<Types>(args)... };
  373. }
  374. template <typename... Types>
  375. constexpr std::tuple<Types&...> tie(Types&... args) noexcept
  376. {
  377. return { args... };
  378. }
  379. template <typename... Types>
  380. constexpr std::tuple<Types&&...> forward_as_tuple(Types&&... args) noexcept
  381. {
  382. return { std::forward<Types>(args)... };
  383. }
  384. template <typename Func, typename Tuple>
  385. constexpr decltype(auto) apply(Func&& func, Tuple&& tpl)
  386. {
  387. return __helpers::apply_impl(
  388. std::forward<Func>(func), std::forward<Tuple>(tpl),
  389. std::make_index_sequence<
  390. std::tuple_size_v<std::remove_cvref_t<Tuple>>> {}
  391. );
  392. }
  393. } // namespace std
  394. #endif