mersenne_twister_engine.h 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534
  1. //===----------------------------------------------------------------------===//
  2. //
  3. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
  4. // See https://llvm.org/LICENSE.txt for license information.
  5. // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
  6. //
  7. //===----------------------------------------------------------------------===//
  8. #ifndef _LIBCPP___RANDOM_MERSENNE_TWISTER_ENGINE_H
  9. #define _LIBCPP___RANDOM_MERSENNE_TWISTER_ENGINE_H
  10. #include <__algorithm/equal.h>
  11. #include <__algorithm/min.h>
  12. #include <__config>
  13. #include <__random/is_seed_sequence.h>
  14. #include <cstddef>
  15. #include <cstdint>
  16. #include <iosfwd>
  17. #include <limits>
  18. #include <type_traits>
  19. #if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
  20. # pragma GCC system_header
  21. #endif
  22. _LIBCPP_PUSH_MACROS
  23. #include <__undef_macros>
  24. _LIBCPP_BEGIN_NAMESPACE_STD
  25. template <class _UIntType, size_t __w, size_t __n, size_t __m, size_t __r,
  26. _UIntType __a, size_t __u, _UIntType __d, size_t __s,
  27. _UIntType __b, size_t __t, _UIntType __c, size_t __l, _UIntType __f>
  28. class _LIBCPP_TEMPLATE_VIS mersenne_twister_engine;
  29. template <class _UInt, size_t _Wp, size_t _Np, size_t _Mp, size_t _Rp,
  30. _UInt _Ap, size_t _Up, _UInt _Dp, size_t _Sp,
  31. _UInt _Bp, size_t _Tp, _UInt _Cp, size_t _Lp, _UInt _Fp>
  32. bool
  33. operator==(const mersenne_twister_engine<_UInt, _Wp, _Np, _Mp, _Rp, _Ap, _Up, _Dp, _Sp,
  34. _Bp, _Tp, _Cp, _Lp, _Fp>& __x,
  35. const mersenne_twister_engine<_UInt, _Wp, _Np, _Mp, _Rp, _Ap, _Up, _Dp, _Sp,
  36. _Bp, _Tp, _Cp, _Lp, _Fp>& __y);
  37. template <class _UInt, size_t _Wp, size_t _Np, size_t _Mp, size_t _Rp,
  38. _UInt _Ap, size_t _Up, _UInt _Dp, size_t _Sp,
  39. _UInt _Bp, size_t _Tp, _UInt _Cp, size_t _Lp, _UInt _Fp>
  40. _LIBCPP_INLINE_VISIBILITY
  41. bool
  42. operator!=(const mersenne_twister_engine<_UInt, _Wp, _Np, _Mp, _Rp, _Ap, _Up, _Dp, _Sp,
  43. _Bp, _Tp, _Cp, _Lp, _Fp>& __x,
  44. const mersenne_twister_engine<_UInt, _Wp, _Np, _Mp, _Rp, _Ap, _Up, _Dp, _Sp,
  45. _Bp, _Tp, _Cp, _Lp, _Fp>& __y);
  46. template <class _CharT, class _Traits,
  47. class _UInt, size_t _Wp, size_t _Np, size_t _Mp, size_t _Rp,
  48. _UInt _Ap, size_t _Up, _UInt _Dp, size_t _Sp,
  49. _UInt _Bp, size_t _Tp, _UInt _Cp, size_t _Lp, _UInt _Fp>
  50. basic_ostream<_CharT, _Traits>&
  51. operator<<(basic_ostream<_CharT, _Traits>& __os,
  52. const mersenne_twister_engine<_UInt, _Wp, _Np, _Mp, _Rp, _Ap, _Up, _Dp, _Sp,
  53. _Bp, _Tp, _Cp, _Lp, _Fp>& __x);
  54. template <class _CharT, class _Traits,
  55. class _UInt, size_t _Wp, size_t _Np, size_t _Mp, size_t _Rp,
  56. _UInt _Ap, size_t _Up, _UInt _Dp, size_t _Sp,
  57. _UInt _Bp, size_t _Tp, _UInt _Cp, size_t _Lp, _UInt _Fp>
  58. basic_istream<_CharT, _Traits>&
  59. operator>>(basic_istream<_CharT, _Traits>& __is,
  60. mersenne_twister_engine<_UInt, _Wp, _Np, _Mp, _Rp, _Ap, _Up, _Dp, _Sp,
  61. _Bp, _Tp, _Cp, _Lp, _Fp>& __x);
  62. template <class _UIntType, size_t __w, size_t __n, size_t __m, size_t __r,
  63. _UIntType __a, size_t __u, _UIntType __d, size_t __s,
  64. _UIntType __b, size_t __t, _UIntType __c, size_t __l, _UIntType __f>
  65. class _LIBCPP_TEMPLATE_VIS mersenne_twister_engine
  66. {
  67. public:
  68. // types
  69. typedef _UIntType result_type;
  70. private:
  71. result_type __x_[__n];
  72. size_t __i_;
  73. static_assert( 0 < __m, "mersenne_twister_engine invalid parameters");
  74. static_assert(__m <= __n, "mersenne_twister_engine invalid parameters");
  75. static _LIBCPP_CONSTEXPR const result_type _Dt = numeric_limits<result_type>::digits;
  76. static_assert(__w <= _Dt, "mersenne_twister_engine invalid parameters");
  77. static_assert( 2 <= __w, "mersenne_twister_engine invalid parameters");
  78. static_assert(__r <= __w, "mersenne_twister_engine invalid parameters");
  79. static_assert(__u <= __w, "mersenne_twister_engine invalid parameters");
  80. static_assert(__s <= __w, "mersenne_twister_engine invalid parameters");
  81. static_assert(__t <= __w, "mersenne_twister_engine invalid parameters");
  82. static_assert(__l <= __w, "mersenne_twister_engine invalid parameters");
  83. public:
  84. static _LIBCPP_CONSTEXPR const result_type _Min = 0;
  85. static _LIBCPP_CONSTEXPR const result_type _Max = __w == _Dt ? result_type(~0) :
  86. (result_type(1) << __w) - result_type(1);
  87. static_assert(_Min < _Max, "mersenne_twister_engine invalid parameters");
  88. static_assert(__a <= _Max, "mersenne_twister_engine invalid parameters");
  89. static_assert(__b <= _Max, "mersenne_twister_engine invalid parameters");
  90. static_assert(__c <= _Max, "mersenne_twister_engine invalid parameters");
  91. static_assert(__d <= _Max, "mersenne_twister_engine invalid parameters");
  92. static_assert(__f <= _Max, "mersenne_twister_engine invalid parameters");
  93. // engine characteristics
  94. static _LIBCPP_CONSTEXPR const size_t word_size = __w;
  95. static _LIBCPP_CONSTEXPR const size_t state_size = __n;
  96. static _LIBCPP_CONSTEXPR const size_t shift_size = __m;
  97. static _LIBCPP_CONSTEXPR const size_t mask_bits = __r;
  98. static _LIBCPP_CONSTEXPR const result_type xor_mask = __a;
  99. static _LIBCPP_CONSTEXPR const size_t tempering_u = __u;
  100. static _LIBCPP_CONSTEXPR const result_type tempering_d = __d;
  101. static _LIBCPP_CONSTEXPR const size_t tempering_s = __s;
  102. static _LIBCPP_CONSTEXPR const result_type tempering_b = __b;
  103. static _LIBCPP_CONSTEXPR const size_t tempering_t = __t;
  104. static _LIBCPP_CONSTEXPR const result_type tempering_c = __c;
  105. static _LIBCPP_CONSTEXPR const size_t tempering_l = __l;
  106. static _LIBCPP_CONSTEXPR const result_type initialization_multiplier = __f;
  107. _LIBCPP_INLINE_VISIBILITY
  108. static _LIBCPP_CONSTEXPR result_type min() { return _Min; }
  109. _LIBCPP_INLINE_VISIBILITY
  110. static _LIBCPP_CONSTEXPR result_type max() { return _Max; }
  111. static _LIBCPP_CONSTEXPR const result_type default_seed = 5489u;
  112. // constructors and seeding functions
  113. #ifndef _LIBCPP_CXX03_LANG
  114. _LIBCPP_INLINE_VISIBILITY
  115. mersenne_twister_engine() : mersenne_twister_engine(default_seed) {}
  116. _LIBCPP_INLINE_VISIBILITY
  117. explicit mersenne_twister_engine(result_type __sd) { seed(__sd); }
  118. #else
  119. _LIBCPP_INLINE_VISIBILITY
  120. explicit mersenne_twister_engine(result_type __sd = default_seed) {
  121. seed(__sd);
  122. }
  123. #endif
  124. template<class _Sseq>
  125. _LIBCPP_INLINE_VISIBILITY
  126. explicit mersenne_twister_engine(_Sseq& __q,
  127. typename enable_if<__is_seed_sequence<_Sseq, mersenne_twister_engine>::value>::type* = 0)
  128. {seed(__q);}
  129. void seed(result_type __sd = default_seed);
  130. template<class _Sseq>
  131. _LIBCPP_INLINE_VISIBILITY
  132. typename enable_if
  133. <
  134. __is_seed_sequence<_Sseq, mersenne_twister_engine>::value,
  135. void
  136. >::type
  137. seed(_Sseq& __q)
  138. {__seed(__q, integral_constant<unsigned, 1 + (__w - 1) / 32>());}
  139. // generating functions
  140. result_type operator()();
  141. _LIBCPP_INLINE_VISIBILITY
  142. void discard(unsigned long long __z) {for (; __z; --__z) operator()();}
  143. template <class _UInt, size_t _Wp, size_t _Np, size_t _Mp, size_t _Rp,
  144. _UInt _Ap, size_t _Up, _UInt _Dp, size_t _Sp,
  145. _UInt _Bp, size_t _Tp, _UInt _Cp, size_t _Lp, _UInt _Fp>
  146. friend
  147. bool
  148. operator==(const mersenne_twister_engine<_UInt, _Wp, _Np, _Mp, _Rp, _Ap, _Up, _Dp, _Sp,
  149. _Bp, _Tp, _Cp, _Lp, _Fp>& __x,
  150. const mersenne_twister_engine<_UInt, _Wp, _Np, _Mp, _Rp, _Ap, _Up, _Dp, _Sp,
  151. _Bp, _Tp, _Cp, _Lp, _Fp>& __y);
  152. template <class _UInt, size_t _Wp, size_t _Np, size_t _Mp, size_t _Rp,
  153. _UInt _Ap, size_t _Up, _UInt _Dp, size_t _Sp,
  154. _UInt _Bp, size_t _Tp, _UInt _Cp, size_t _Lp, _UInt _Fp>
  155. friend
  156. bool
  157. operator!=(const mersenne_twister_engine<_UInt, _Wp, _Np, _Mp, _Rp, _Ap, _Up, _Dp, _Sp,
  158. _Bp, _Tp, _Cp, _Lp, _Fp>& __x,
  159. const mersenne_twister_engine<_UInt, _Wp, _Np, _Mp, _Rp, _Ap, _Up, _Dp, _Sp,
  160. _Bp, _Tp, _Cp, _Lp, _Fp>& __y);
  161. template <class _CharT, class _Traits,
  162. class _UInt, size_t _Wp, size_t _Np, size_t _Mp, size_t _Rp,
  163. _UInt _Ap, size_t _Up, _UInt _Dp, size_t _Sp,
  164. _UInt _Bp, size_t _Tp, _UInt _Cp, size_t _Lp, _UInt _Fp>
  165. friend
  166. basic_ostream<_CharT, _Traits>&
  167. operator<<(basic_ostream<_CharT, _Traits>& __os,
  168. const mersenne_twister_engine<_UInt, _Wp, _Np, _Mp, _Rp, _Ap, _Up, _Dp, _Sp,
  169. _Bp, _Tp, _Cp, _Lp, _Fp>& __x);
  170. template <class _CharT, class _Traits,
  171. class _UInt, size_t _Wp, size_t _Np, size_t _Mp, size_t _Rp,
  172. _UInt _Ap, size_t _Up, _UInt _Dp, size_t _Sp,
  173. _UInt _Bp, size_t _Tp, _UInt _Cp, size_t _Lp, _UInt _Fp>
  174. friend
  175. basic_istream<_CharT, _Traits>&
  176. operator>>(basic_istream<_CharT, _Traits>& __is,
  177. mersenne_twister_engine<_UInt, _Wp, _Np, _Mp, _Rp, _Ap, _Up, _Dp, _Sp,
  178. _Bp, _Tp, _Cp, _Lp, _Fp>& __x);
  179. private:
  180. template<class _Sseq>
  181. void __seed(_Sseq& __q, integral_constant<unsigned, 1>);
  182. template<class _Sseq>
  183. void __seed(_Sseq& __q, integral_constant<unsigned, 2>);
  184. template <size_t __count>
  185. _LIBCPP_INLINE_VISIBILITY
  186. static
  187. typename enable_if
  188. <
  189. __count < __w,
  190. result_type
  191. >::type
  192. __lshift(result_type __x) {return (__x << __count) & _Max;}
  193. template <size_t __count>
  194. _LIBCPP_INLINE_VISIBILITY
  195. static
  196. typename enable_if
  197. <
  198. (__count >= __w),
  199. result_type
  200. >::type
  201. __lshift(result_type) {return result_type(0);}
  202. template <size_t __count>
  203. _LIBCPP_INLINE_VISIBILITY
  204. static
  205. typename enable_if
  206. <
  207. __count < _Dt,
  208. result_type
  209. >::type
  210. __rshift(result_type __x) {return __x >> __count;}
  211. template <size_t __count>
  212. _LIBCPP_INLINE_VISIBILITY
  213. static
  214. typename enable_if
  215. <
  216. (__count >= _Dt),
  217. result_type
  218. >::type
  219. __rshift(result_type) {return result_type(0);}
  220. };
  221. template <class _UIntType, size_t __w, size_t __n, size_t __m, size_t __r,
  222. _UIntType __a, size_t __u, _UIntType __d, size_t __s,
  223. _UIntType __b, size_t __t, _UIntType __c, size_t __l, _UIntType __f>
  224. _LIBCPP_CONSTEXPR const size_t
  225. mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b, __t, __c, __l, __f>::word_size;
  226. template <class _UIntType, size_t __w, size_t __n, size_t __m, size_t __r,
  227. _UIntType __a, size_t __u, _UIntType __d, size_t __s,
  228. _UIntType __b, size_t __t, _UIntType __c, size_t __l, _UIntType __f>
  229. _LIBCPP_CONSTEXPR const size_t
  230. mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b, __t, __c, __l, __f>::state_size;
  231. template <class _UIntType, size_t __w, size_t __n, size_t __m, size_t __r,
  232. _UIntType __a, size_t __u, _UIntType __d, size_t __s,
  233. _UIntType __b, size_t __t, _UIntType __c, size_t __l, _UIntType __f>
  234. _LIBCPP_CONSTEXPR const size_t
  235. mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b, __t, __c, __l, __f>::shift_size;
  236. template <class _UIntType, size_t __w, size_t __n, size_t __m, size_t __r,
  237. _UIntType __a, size_t __u, _UIntType __d, size_t __s,
  238. _UIntType __b, size_t __t, _UIntType __c, size_t __l, _UIntType __f>
  239. _LIBCPP_CONSTEXPR const size_t
  240. mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b, __t, __c, __l, __f>::mask_bits;
  241. template <class _UIntType, size_t __w, size_t __n, size_t __m, size_t __r,
  242. _UIntType __a, size_t __u, _UIntType __d, size_t __s,
  243. _UIntType __b, size_t __t, _UIntType __c, size_t __l, _UIntType __f>
  244. _LIBCPP_CONSTEXPR const typename mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b, __t, __c, __l, __f>::result_type
  245. mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b, __t, __c, __l, __f>::xor_mask;
  246. template <class _UIntType, size_t __w, size_t __n, size_t __m, size_t __r,
  247. _UIntType __a, size_t __u, _UIntType __d, size_t __s,
  248. _UIntType __b, size_t __t, _UIntType __c, size_t __l, _UIntType __f>
  249. _LIBCPP_CONSTEXPR const size_t
  250. mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b, __t, __c, __l, __f>::tempering_u;
  251. template <class _UIntType, size_t __w, size_t __n, size_t __m, size_t __r,
  252. _UIntType __a, size_t __u, _UIntType __d, size_t __s,
  253. _UIntType __b, size_t __t, _UIntType __c, size_t __l, _UIntType __f>
  254. _LIBCPP_CONSTEXPR const typename mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b, __t, __c, __l, __f>::result_type
  255. mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b, __t, __c, __l, __f>::tempering_d;
  256. template <class _UIntType, size_t __w, size_t __n, size_t __m, size_t __r,
  257. _UIntType __a, size_t __u, _UIntType __d, size_t __s,
  258. _UIntType __b, size_t __t, _UIntType __c, size_t __l, _UIntType __f>
  259. _LIBCPP_CONSTEXPR const size_t
  260. mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b, __t, __c, __l, __f>::tempering_s;
  261. template <class _UIntType, size_t __w, size_t __n, size_t __m, size_t __r,
  262. _UIntType __a, size_t __u, _UIntType __d, size_t __s,
  263. _UIntType __b, size_t __t, _UIntType __c, size_t __l, _UIntType __f>
  264. _LIBCPP_CONSTEXPR const typename mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b, __t, __c, __l, __f>::result_type
  265. mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b, __t, __c, __l, __f>::tempering_b;
  266. template <class _UIntType, size_t __w, size_t __n, size_t __m, size_t __r,
  267. _UIntType __a, size_t __u, _UIntType __d, size_t __s,
  268. _UIntType __b, size_t __t, _UIntType __c, size_t __l, _UIntType __f>
  269. _LIBCPP_CONSTEXPR const size_t
  270. mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b, __t, __c, __l, __f>::tempering_t;
  271. template <class _UIntType, size_t __w, size_t __n, size_t __m, size_t __r,
  272. _UIntType __a, size_t __u, _UIntType __d, size_t __s,
  273. _UIntType __b, size_t __t, _UIntType __c, size_t __l, _UIntType __f>
  274. _LIBCPP_CONSTEXPR const typename mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b, __t, __c, __l, __f>::result_type
  275. mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b, __t, __c, __l, __f>::tempering_c;
  276. template <class _UIntType, size_t __w, size_t __n, size_t __m, size_t __r,
  277. _UIntType __a, size_t __u, _UIntType __d, size_t __s,
  278. _UIntType __b, size_t __t, _UIntType __c, size_t __l, _UIntType __f>
  279. _LIBCPP_CONSTEXPR const size_t
  280. mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b, __t, __c, __l, __f>::tempering_l;
  281. template <class _UIntType, size_t __w, size_t __n, size_t __m, size_t __r,
  282. _UIntType __a, size_t __u, _UIntType __d, size_t __s,
  283. _UIntType __b, size_t __t, _UIntType __c, size_t __l, _UIntType __f>
  284. _LIBCPP_CONSTEXPR const typename mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b, __t, __c, __l, __f>::result_type
  285. mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b, __t, __c, __l, __f>::initialization_multiplier;
  286. template <class _UIntType, size_t __w, size_t __n, size_t __m, size_t __r,
  287. _UIntType __a, size_t __u, _UIntType __d, size_t __s,
  288. _UIntType __b, size_t __t, _UIntType __c, size_t __l, _UIntType __f>
  289. _LIBCPP_CONSTEXPR const typename mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b, __t, __c, __l, __f>::result_type
  290. mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b, __t, __c, __l, __f>::default_seed;
  291. template <class _UIntType, size_t __w, size_t __n, size_t __m, size_t __r,
  292. _UIntType __a, size_t __u, _UIntType __d, size_t __s,
  293. _UIntType __b, size_t __t, _UIntType __c, size_t __l, _UIntType __f>
  294. void
  295. mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b,
  296. __t, __c, __l, __f>::seed(result_type __sd)
  297. _LIBCPP_DISABLE_UBSAN_UNSIGNED_INTEGER_CHECK
  298. { // __w >= 2
  299. __x_[0] = __sd & _Max;
  300. for (size_t __i = 1; __i < __n; ++__i)
  301. __x_[__i] = (__f * (__x_[__i-1] ^ __rshift<__w - 2>(__x_[__i-1])) + __i) & _Max;
  302. __i_ = 0;
  303. }
  304. template <class _UIntType, size_t __w, size_t __n, size_t __m, size_t __r,
  305. _UIntType __a, size_t __u, _UIntType __d, size_t __s,
  306. _UIntType __b, size_t __t, _UIntType __c, size_t __l, _UIntType __f>
  307. template<class _Sseq>
  308. void
  309. mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b,
  310. __t, __c, __l, __f>::__seed(_Sseq& __q, integral_constant<unsigned, 1>)
  311. {
  312. const unsigned __k = 1;
  313. uint32_t __ar[__n * __k];
  314. __q.generate(__ar, __ar + __n * __k);
  315. for (size_t __i = 0; __i < __n; ++__i)
  316. __x_[__i] = static_cast<result_type>(__ar[__i] & _Max);
  317. const result_type __mask = __r == _Dt ? result_type(~0) :
  318. (result_type(1) << __r) - result_type(1);
  319. __i_ = 0;
  320. if ((__x_[0] & ~__mask) == 0)
  321. {
  322. for (size_t __i = 1; __i < __n; ++__i)
  323. if (__x_[__i] != 0)
  324. return;
  325. __x_[0] = result_type(1) << (__w - 1);
  326. }
  327. }
  328. template <class _UIntType, size_t __w, size_t __n, size_t __m, size_t __r,
  329. _UIntType __a, size_t __u, _UIntType __d, size_t __s,
  330. _UIntType __b, size_t __t, _UIntType __c, size_t __l, _UIntType __f>
  331. template<class _Sseq>
  332. void
  333. mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b,
  334. __t, __c, __l, __f>::__seed(_Sseq& __q, integral_constant<unsigned, 2>)
  335. {
  336. const unsigned __k = 2;
  337. uint32_t __ar[__n * __k];
  338. __q.generate(__ar, __ar + __n * __k);
  339. for (size_t __i = 0; __i < __n; ++__i)
  340. __x_[__i] = static_cast<result_type>(
  341. (__ar[2 * __i] + ((uint64_t)__ar[2 * __i + 1] << 32)) & _Max);
  342. const result_type __mask = __r == _Dt ? result_type(~0) :
  343. (result_type(1) << __r) - result_type(1);
  344. __i_ = 0;
  345. if ((__x_[0] & ~__mask) == 0)
  346. {
  347. for (size_t __i = 1; __i < __n; ++__i)
  348. if (__x_[__i] != 0)
  349. return;
  350. __x_[0] = result_type(1) << (__w - 1);
  351. }
  352. }
  353. template <class _UIntType, size_t __w, size_t __n, size_t __m, size_t __r,
  354. _UIntType __a, size_t __u, _UIntType __d, size_t __s,
  355. _UIntType __b, size_t __t, _UIntType __c, size_t __l, _UIntType __f>
  356. _UIntType
  357. mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b,
  358. __t, __c, __l, __f>::operator()()
  359. {
  360. const size_t __j = (__i_ + 1) % __n;
  361. const result_type __mask = __r == _Dt ? result_type(~0) :
  362. (result_type(1) << __r) - result_type(1);
  363. const result_type _Yp = (__x_[__i_] & ~__mask) | (__x_[__j] & __mask);
  364. const size_t __k = (__i_ + __m) % __n;
  365. __x_[__i_] = __x_[__k] ^ __rshift<1>(_Yp) ^ (__a * (_Yp & 1));
  366. result_type __z = __x_[__i_] ^ (__rshift<__u>(__x_[__i_]) & __d);
  367. __i_ = __j;
  368. __z ^= __lshift<__s>(__z) & __b;
  369. __z ^= __lshift<__t>(__z) & __c;
  370. return __z ^ __rshift<__l>(__z);
  371. }
  372. template <class _UInt, size_t _Wp, size_t _Np, size_t _Mp, size_t _Rp,
  373. _UInt _Ap, size_t _Up, _UInt _Dp, size_t _Sp,
  374. _UInt _Bp, size_t _Tp, _UInt _Cp, size_t _Lp, _UInt _Fp>
  375. bool
  376. operator==(const mersenne_twister_engine<_UInt, _Wp, _Np, _Mp, _Rp, _Ap, _Up, _Dp, _Sp,
  377. _Bp, _Tp, _Cp, _Lp, _Fp>& __x,
  378. const mersenne_twister_engine<_UInt, _Wp, _Np, _Mp, _Rp, _Ap, _Up, _Dp, _Sp,
  379. _Bp, _Tp, _Cp, _Lp, _Fp>& __y)
  380. {
  381. if (__x.__i_ == __y.__i_)
  382. return _VSTD::equal(__x.__x_, __x.__x_ + _Np, __y.__x_);
  383. if (__x.__i_ == 0 || __y.__i_ == 0)
  384. {
  385. size_t __j = _VSTD::min(_Np - __x.__i_, _Np - __y.__i_);
  386. if (!_VSTD::equal(__x.__x_ + __x.__i_, __x.__x_ + __x.__i_ + __j,
  387. __y.__x_ + __y.__i_))
  388. return false;
  389. if (__x.__i_ == 0)
  390. return _VSTD::equal(__x.__x_ + __j, __x.__x_ + _Np, __y.__x_);
  391. return _VSTD::equal(__x.__x_, __x.__x_ + (_Np - __j), __y.__x_ + __j);
  392. }
  393. if (__x.__i_ < __y.__i_)
  394. {
  395. size_t __j = _Np - __y.__i_;
  396. if (!_VSTD::equal(__x.__x_ + __x.__i_, __x.__x_ + (__x.__i_ + __j),
  397. __y.__x_ + __y.__i_))
  398. return false;
  399. if (!_VSTD::equal(__x.__x_ + (__x.__i_ + __j), __x.__x_ + _Np,
  400. __y.__x_))
  401. return false;
  402. return _VSTD::equal(__x.__x_, __x.__x_ + __x.__i_,
  403. __y.__x_ + (_Np - (__x.__i_ + __j)));
  404. }
  405. size_t __j = _Np - __x.__i_;
  406. if (!_VSTD::equal(__y.__x_ + __y.__i_, __y.__x_ + (__y.__i_ + __j),
  407. __x.__x_ + __x.__i_))
  408. return false;
  409. if (!_VSTD::equal(__y.__x_ + (__y.__i_ + __j), __y.__x_ + _Np,
  410. __x.__x_))
  411. return false;
  412. return _VSTD::equal(__y.__x_, __y.__x_ + __y.__i_,
  413. __x.__x_ + (_Np - (__y.__i_ + __j)));
  414. }
  415. template <class _UInt, size_t _Wp, size_t _Np, size_t _Mp, size_t _Rp,
  416. _UInt _Ap, size_t _Up, _UInt _Dp, size_t _Sp,
  417. _UInt _Bp, size_t _Tp, _UInt _Cp, size_t _Lp, _UInt _Fp>
  418. inline _LIBCPP_INLINE_VISIBILITY
  419. bool
  420. operator!=(const mersenne_twister_engine<_UInt, _Wp, _Np, _Mp, _Rp, _Ap, _Up, _Dp, _Sp,
  421. _Bp, _Tp, _Cp, _Lp, _Fp>& __x,
  422. const mersenne_twister_engine<_UInt, _Wp, _Np, _Mp, _Rp, _Ap, _Up, _Dp, _Sp,
  423. _Bp, _Tp, _Cp, _Lp, _Fp>& __y)
  424. {
  425. return !(__x == __y);
  426. }
  427. template <class _CharT, class _Traits,
  428. class _UInt, size_t _Wp, size_t _Np, size_t _Mp, size_t _Rp,
  429. _UInt _Ap, size_t _Up, _UInt _Dp, size_t _Sp,
  430. _UInt _Bp, size_t _Tp, _UInt _Cp, size_t _Lp, _UInt _Fp>
  431. basic_ostream<_CharT, _Traits>&
  432. operator<<(basic_ostream<_CharT, _Traits>& __os,
  433. const mersenne_twister_engine<_UInt, _Wp, _Np, _Mp, _Rp, _Ap, _Up, _Dp, _Sp,
  434. _Bp, _Tp, _Cp, _Lp, _Fp>& __x)
  435. {
  436. __save_flags<_CharT, _Traits> __lx(__os);
  437. typedef basic_ostream<_CharT, _Traits> _Ostream;
  438. __os.flags(_Ostream::dec | _Ostream::left);
  439. _CharT __sp = __os.widen(' ');
  440. __os.fill(__sp);
  441. __os << __x.__x_[__x.__i_];
  442. for (size_t __j = __x.__i_ + 1; __j < _Np; ++__j)
  443. __os << __sp << __x.__x_[__j];
  444. for (size_t __j = 0; __j < __x.__i_; ++__j)
  445. __os << __sp << __x.__x_[__j];
  446. return __os;
  447. }
  448. template <class _CharT, class _Traits,
  449. class _UInt, size_t _Wp, size_t _Np, size_t _Mp, size_t _Rp,
  450. _UInt _Ap, size_t _Up, _UInt _Dp, size_t _Sp,
  451. _UInt _Bp, size_t _Tp, _UInt _Cp, size_t _Lp, _UInt _Fp>
  452. basic_istream<_CharT, _Traits>&
  453. operator>>(basic_istream<_CharT, _Traits>& __is,
  454. mersenne_twister_engine<_UInt, _Wp, _Np, _Mp, _Rp, _Ap, _Up, _Dp, _Sp,
  455. _Bp, _Tp, _Cp, _Lp, _Fp>& __x)
  456. {
  457. __save_flags<_CharT, _Traits> __lx(__is);
  458. typedef basic_istream<_CharT, _Traits> _Istream;
  459. __is.flags(_Istream::dec | _Istream::skipws);
  460. _UInt __t[_Np];
  461. for (size_t __i = 0; __i < _Np; ++__i)
  462. __is >> __t[__i];
  463. if (!__is.fail())
  464. {
  465. for (size_t __i = 0; __i < _Np; ++__i)
  466. __x.__x_[__i] = __t[__i];
  467. __x.__i_ = 0;
  468. }
  469. return __is;
  470. }
  471. typedef mersenne_twister_engine<uint_fast32_t, 32, 624, 397, 31,
  472. 0x9908b0df, 11, 0xffffffff,
  473. 7, 0x9d2c5680,
  474. 15, 0xefc60000,
  475. 18, 1812433253> mt19937;
  476. typedef mersenne_twister_engine<uint_fast64_t, 64, 312, 156, 31,
  477. 0xb5026f5aa96619e9ULL, 29, 0x5555555555555555ULL,
  478. 17, 0x71d67fffeda60000ULL,
  479. 37, 0xfff7eee000000000ULL,
  480. 43, 6364136223846793005ULL> mt19937_64;
  481. _LIBCPP_END_NAMESPACE_STD
  482. _LIBCPP_POP_MACROS
  483. #endif // _LIBCPP___RANDOM_MERSENNE_TWISTER_ENGINE_H