mirror of
https://github.com/bolero-MURAKAMI/Sprout.git
synced 2024-12-23 21:25:49 +00:00
uniform_int_distribution 追加
This commit is contained in:
parent
8e6b3d22cb
commit
ded40d58ae
6 changed files with 640 additions and 12 deletions
|
@ -4,6 +4,7 @@
|
|||
#include <sprout/config.hpp>
|
||||
#include <sprout/random/linear_congruential.hpp>
|
||||
#include <sprout/random/mersenne_twister.hpp>
|
||||
#include <sprout/random/uniform_int_distribution.hpp>
|
||||
#include <sprout/random/uniform_smallint.hpp>
|
||||
#include <sprout/random/uniform_01.hpp>
|
||||
#include <sprout/random/variate_generator.hpp>
|
||||
|
|
|
@ -1,8 +1,8 @@
|
|||
#ifndef SPROUT_RANDOM_CONST_MOD_HPP
|
||||
#define SPROUT_RANDOM_CONST_MOD_HPP
|
||||
#ifndef SPROUT_RANDOM_DETAIL_CONST_MOD_HPP
|
||||
#define SPROUT_RANDOM_DETAIL_CONST_MOD_HPP
|
||||
|
||||
#include <type_traits>
|
||||
#include <limits>
|
||||
#include <type_traits>
|
||||
#include <sprout/config.hpp>
|
||||
|
||||
namespace sprout {
|
||||
|
@ -128,5 +128,5 @@ namespace sprout {
|
|||
} // namespace random
|
||||
} // namespace sprout
|
||||
|
||||
#endif // #ifndef SPROUT_RANDOM_CONST_MOD_HPP
|
||||
#endif // #ifndef SPROUT_RANDOM_DETAIL_CONST_MOD_HPP
|
||||
|
||||
|
|
23
sprout/random/detail/generator_bits.hpp
Normal file
23
sprout/random/detail/generator_bits.hpp
Normal file
|
@ -0,0 +1,23 @@
|
|||
#ifndef SPROUT_RANDOM_DETAIL_GENERATOR_BITS_HPP
|
||||
#define SPROUT_RANDOM_DETAIL_GENERATOR_BITS_HPP
|
||||
|
||||
#include <cstddef>
|
||||
#include <limits>
|
||||
#include <sprout/config.hpp>
|
||||
|
||||
namespace sprout {
|
||||
namespace random {
|
||||
namespace detail {
|
||||
template<typename URNG>
|
||||
struct generator_bits {
|
||||
public:
|
||||
static SPROUT_CONSTEXPR std::size_t value() {
|
||||
return std::numeric_limits<typename URNG::result_type>::digits;
|
||||
}
|
||||
};
|
||||
} // namespace detail
|
||||
} // namespace random
|
||||
} // namespace sprout
|
||||
|
||||
#endif // #ifndef SPROUT_RANDOM_DETAIL_GENERATOR_BITS_HPP
|
||||
|
74
sprout/random/detail/uniform_int_float.hpp
Normal file
74
sprout/random/detail/uniform_int_float.hpp
Normal file
|
@ -0,0 +1,74 @@
|
|||
#ifndef SPROUT_RANDOM_DETAIL_UNIFORM_INT_FLOAT_HPP
|
||||
#define SPROUT_RANDOM_DETAIL_UNIFORM_INT_FLOAT_HPP
|
||||
|
||||
#include <cstddef>
|
||||
#include <cstdint>
|
||||
#include <limits>
|
||||
#include <sprout/config.hpp>
|
||||
#include <sprout/random/random_result.hpp>
|
||||
#include <sprout/random/detail/generator_bits.hpp>
|
||||
#include <sprout/detail/integer.hpp>
|
||||
#include HDR_ALGORITHM_SSCRISK_CEL_OR_SPROUT_DETAIL
|
||||
|
||||
namespace sprout {
|
||||
namespace random {
|
||||
namespace detail {
|
||||
template<class URNG>
|
||||
class uniform_int_float {
|
||||
public:
|
||||
typedef URNG base_type;
|
||||
typedef typename base_type::result_type base_result;
|
||||
typedef typename sprout::detail::uint_t<
|
||||
(std::numeric_limits<std::uintmax_t>::digits < std::numeric_limits<base_result>::digits)
|
||||
? std::numeric_limits<std::uintmax_t>::digits
|
||||
: std::numeric_limits<base_result>::digits
|
||||
>::fast result_type;
|
||||
private:
|
||||
base_type rng_;
|
||||
public:
|
||||
//uniform_int_float = default; // ???
|
||||
SPROUT_CONSTEXPR uniform_int_float()
|
||||
: rng_()
|
||||
{}
|
||||
SPROUT_CONSTEXPR explicit uniform_int_float(base_type const& rng)
|
||||
: rng_(rng)
|
||||
{}
|
||||
SPROUT_CONSTEXPR result_type min() const {
|
||||
return 0;
|
||||
}
|
||||
SPROUT_CONSTEXPR result_type max() const {
|
||||
return (
|
||||
result_type(2) << (
|
||||
NS_SSCRISK_CEL_OR_SPROUT_DETAIL::min(
|
||||
std::numeric_limits<result_type>::digits,
|
||||
sprout::random::detail::generator_bits<base_type>::value()
|
||||
) - 1
|
||||
)
|
||||
) - 1
|
||||
;
|
||||
}
|
||||
base_type& base() {
|
||||
return rng_;
|
||||
}
|
||||
SPROUT_CONSTEXPR base_type const& base() const {
|
||||
return rng_;
|
||||
}
|
||||
SPROUT_CONSTEXPR sprout::random::random_result<uniform_int_float> generate(
|
||||
sprout::random::random_result<base_type> const& rnd
|
||||
) const
|
||||
{
|
||||
return sprout::random::random_result<uniform_int_float>(
|
||||
static_cast<result_type>(rnd.result() * (static_cast<base_result>(max()) + 1)),
|
||||
uniform_int_float(rnd.engine())
|
||||
);
|
||||
}
|
||||
SPROUT_CONSTEXPR sprout::random::random_result<uniform_int_float> operator()() const {
|
||||
return generate(rng_());
|
||||
}
|
||||
};
|
||||
} // namespace detail
|
||||
} // namespace random
|
||||
} // namespace sprout
|
||||
|
||||
#endif // #ifndef SPROUT_RANDOM_DETAIL_UNIFORM_INT_FLOAT_HPP
|
||||
|
530
sprout/random/uniform_int_distribution.hpp
Normal file
530
sprout/random/uniform_int_distribution.hpp
Normal file
|
@ -0,0 +1,530 @@
|
|||
#ifndef SPROUT_RANDOM_UNIFORM_INT_DISTRIBUTION_HPP
|
||||
#define SPROUT_RANDOM_UNIFORM_INT_DISTRIBUTION_HPP
|
||||
|
||||
#include <iosfwd>
|
||||
#include <limits>
|
||||
#include <istream>
|
||||
#include <type_traits>
|
||||
#include <sprout/config.hpp>
|
||||
#include <sprout/random/detail/signed_unsigned_tools.hpp>
|
||||
#include <sprout/random/detail/uniform_int_float.hpp>
|
||||
#include <sprout/random/random_result.hpp>
|
||||
|
||||
namespace sprout {
|
||||
namespace random {
|
||||
namespace detail {
|
||||
//
|
||||
// generate_uniform_int_result
|
||||
//
|
||||
template<typename T, typename Engine>
|
||||
struct generate_uniform_int_result {
|
||||
public:
|
||||
T result;
|
||||
Engine engine;
|
||||
};
|
||||
|
||||
template<typename Engine, typename T>
|
||||
SPROUT_CONSTEXPR sprout::random::detail::generate_uniform_int_result<T, Engine> generate_uniform_int(
|
||||
Engine const& eng,
|
||||
T min_value,
|
||||
T max_value,
|
||||
std::true_type
|
||||
);
|
||||
template<typename Engine, typename T, typename RangeType, typename BaseResult, typename BaseUnsigned>
|
||||
SPROUT_CONSTEXPR sprout::random::detail::generate_uniform_int_result<T, Engine> generate_uniform_int_true_3_1(
|
||||
sprout::random::random_result<Engine> const& rnd,
|
||||
T min_value,
|
||||
RangeType range,
|
||||
BaseResult bmin,
|
||||
BaseUnsigned brange,
|
||||
BaseUnsigned bucket_size
|
||||
);
|
||||
template<typename Engine, typename T, typename RangeType, typename BaseResult, typename BaseUnsigned>
|
||||
SPROUT_CONSTEXPR sprout::random::detail::generate_uniform_int_result<T, Engine> generate_uniform_int_true_3_1_1(
|
||||
Engine const& eng,
|
||||
T min_value,
|
||||
RangeType range,
|
||||
BaseResult bmin,
|
||||
BaseUnsigned brange,
|
||||
BaseUnsigned bucket_size,
|
||||
BaseUnsigned result
|
||||
)
|
||||
{
|
||||
typedef T result_type;
|
||||
return result <= static_cast<BaseUnsigned>(range)
|
||||
? sprout::random::detail::generate_uniform_int_result<T, Engine>{
|
||||
sprout::random::detail::add<BaseUnsigned, result_type>()(result, min_value),
|
||||
eng
|
||||
}
|
||||
: generate_uniform_int_true_3_1(
|
||||
eng(),
|
||||
min_value,
|
||||
range,
|
||||
bmin,
|
||||
brange,
|
||||
bucket_size
|
||||
)
|
||||
;
|
||||
}
|
||||
template<typename Engine, typename T, typename RangeType, typename BaseResult, typename BaseUnsigned>
|
||||
SPROUT_CONSTEXPR sprout::random::detail::generate_uniform_int_result<T, Engine> generate_uniform_int_true_3_1(
|
||||
sprout::random::random_result<Engine> const& rnd,
|
||||
T min_value,
|
||||
RangeType range,
|
||||
BaseResult bmin,
|
||||
BaseUnsigned brange,
|
||||
BaseUnsigned bucket_size
|
||||
)
|
||||
{
|
||||
return sprout::random::detail::generate_uniform_int_true_3_1_1(
|
||||
rnd.engine(),
|
||||
min_value,
|
||||
range,
|
||||
bmin,
|
||||
brange,
|
||||
bucket_size,
|
||||
sprout::random::detail::subtract<BaseResult>()(rnd.result(), bmin) / bucket_size
|
||||
);
|
||||
}
|
||||
template<typename Engine, typename T, typename RangeType, typename BaseResult, typename BaseUnsigned>
|
||||
SPROUT_CONSTEXPR sprout::random::detail::generate_uniform_int_result<T, Engine> generate_uniform_int_true_3(
|
||||
Engine const& eng,
|
||||
T min_value,
|
||||
RangeType range,
|
||||
BaseResult bmin,
|
||||
BaseUnsigned brange
|
||||
)
|
||||
{
|
||||
return sprout::random::detail::generate_uniform_int_true_3_1(
|
||||
eng(),
|
||||
min_value,
|
||||
range,
|
||||
bmin,
|
||||
brange,
|
||||
brange == std::numeric_limits<BaseUnsigned>::max()
|
||||
? brange / (static_cast<BaseUnsigned>(range) + 1) + (
|
||||
brange % (static_cast<BaseUnsigned>(range) + 1) == static_cast<BaseUnsigned>(range) ? 1 : 0
|
||||
)
|
||||
: (brange + 1) / (static_cast<BaseUnsigned>(range) + 1)
|
||||
);
|
||||
}
|
||||
template<typename Engine, typename T, typename RangeType, typename BaseResult, typename BaseUnsigned>
|
||||
SPROUT_CONSTEXPR sprout::random::detail::generate_uniform_int_result<T, Engine> generate_uniform_int_true_2(
|
||||
Engine const& eng,
|
||||
T min_value,
|
||||
RangeType range,
|
||||
BaseResult bmin,
|
||||
BaseUnsigned brange
|
||||
);
|
||||
template<typename Engine, typename T, typename RangeType, typename BaseResult, typename BaseUnsigned>
|
||||
SPROUT_CONSTEXPR sprout::random::detail::generate_uniform_int_result<T, Engine> generate_uniform_int_true_2_4(
|
||||
Engine const& eng,
|
||||
T min_value,
|
||||
RangeType range,
|
||||
BaseResult bmin,
|
||||
BaseUnsigned brange,
|
||||
RangeType result,
|
||||
RangeType result_increment
|
||||
)
|
||||
{
|
||||
typedef T result_type;
|
||||
return result < result_increment || result > range
|
||||
? sprout::random::detail::generate_uniform_int_true_2(eng, min_value, range, bmin, brange)
|
||||
: sprout::random::detail::generate_uniform_int_result<T, Engine>{
|
||||
random::detail::add<RangeType, result_type>()(result, min_value),
|
||||
eng
|
||||
}
|
||||
;
|
||||
}
|
||||
template<typename Engine, typename T, typename RangeType, typename BaseResult, typename BaseUnsigned>
|
||||
SPROUT_CONSTEXPR sprout::random::detail::generate_uniform_int_result<T, Engine> generate_uniform_int_true_2_3(
|
||||
Engine const& eng,
|
||||
T min_value,
|
||||
RangeType range,
|
||||
BaseResult bmin,
|
||||
BaseUnsigned brange,
|
||||
RangeType result,
|
||||
RangeType mult,
|
||||
RangeType result_increment
|
||||
)
|
||||
{
|
||||
return std::numeric_limits<RangeType>::max() / mult < result_increment
|
||||
? sprout::random::detail::generate_uniform_int_true_2(eng, min_value, range, bmin, brange)
|
||||
: generate_uniform_int_true_2_4(
|
||||
eng,
|
||||
min_value,
|
||||
range,
|
||||
bmin,
|
||||
brange,
|
||||
result + (result_increment * mult),
|
||||
result_increment * mult
|
||||
)
|
||||
;
|
||||
}
|
||||
template<typename Engine, typename T, typename RangeType, typename BaseResult, typename BaseUnsigned, typename Result>
|
||||
SPROUT_CONSTEXPR sprout::random::detail::generate_uniform_int_result<T, Engine> generate_uniform_int_true_2_2(
|
||||
Engine const& eng,
|
||||
T min_value,
|
||||
RangeType range,
|
||||
BaseResult bmin,
|
||||
BaseUnsigned brange,
|
||||
RangeType result,
|
||||
RangeType mult,
|
||||
Result const& result_increment_base
|
||||
)
|
||||
{
|
||||
return sprout::random::detail::generate_uniform_int_true_2_3(
|
||||
result_increment_base.engine,
|
||||
min_value,
|
||||
range,
|
||||
bmin,
|
||||
brange,
|
||||
result,
|
||||
mult,
|
||||
result_increment_base.result
|
||||
);
|
||||
}
|
||||
template<typename Engine, typename T, typename RangeType, typename BaseResult, typename BaseUnsigned>
|
||||
SPROUT_CONSTEXPR sprout::random::detail::generate_uniform_int_result<T, Engine> generate_uniform_int_true_2_1(
|
||||
Engine const& eng,
|
||||
T min_value,
|
||||
RangeType range,
|
||||
BaseResult bmin,
|
||||
BaseUnsigned brange,
|
||||
RangeType limit,
|
||||
RangeType result = RangeType(0),
|
||||
RangeType mult = RangeType(1)
|
||||
);
|
||||
template<typename Engine, typename T, typename RangeType, typename BaseResult, typename BaseUnsigned>
|
||||
SPROUT_CONSTEXPR sprout::random::detail::generate_uniform_int_result<T, Engine> generate_uniform_int_true_2_1_1(
|
||||
sprout::random::random_result<Engine> const& rnd,
|
||||
T min_value,
|
||||
RangeType range,
|
||||
BaseResult bmin,
|
||||
BaseUnsigned brange,
|
||||
RangeType limit,
|
||||
RangeType result,
|
||||
RangeType mult
|
||||
)
|
||||
{
|
||||
return mult * RangeType(brange) == range - mult + 1
|
||||
? sprout::random::detail::generate_uniform_int_result<T, Engine>{
|
||||
static_cast<T>(
|
||||
result + static_cast<RangeType>(sprout::random::detail::subtract<BaseResult>()(rnd.result(), bmin) * mult)
|
||||
),
|
||||
rnd.engine()
|
||||
}
|
||||
: sprout::random::detail::generate_uniform_int_true_2_1(
|
||||
rnd.engine(),
|
||||
min_value,
|
||||
range,
|
||||
bmin,
|
||||
brange,
|
||||
limit,
|
||||
result + static_cast<RangeType>(sprout::random::detail::subtract<BaseResult>()(rnd.result(), bmin) * mult),
|
||||
mult * (RangeType(brange) + RangeType(1))
|
||||
)
|
||||
;
|
||||
}
|
||||
template<typename Engine, typename T, typename RangeType, typename BaseResult, typename BaseUnsigned>
|
||||
SPROUT_CONSTEXPR sprout::random::detail::generate_uniform_int_result<T, Engine> generate_uniform_int_true_2_1(
|
||||
Engine const& eng,
|
||||
T min_value,
|
||||
RangeType range,
|
||||
BaseResult bmin,
|
||||
BaseUnsigned brange,
|
||||
RangeType limit,
|
||||
RangeType result,
|
||||
RangeType mult
|
||||
)
|
||||
{
|
||||
return mult <= limit
|
||||
? generate_uniform_int_true_2_1_1(
|
||||
eng(),
|
||||
min_value,
|
||||
range,
|
||||
bmin,
|
||||
brange,
|
||||
limit,
|
||||
result,
|
||||
mult
|
||||
)
|
||||
: sprout::random::detail::generate_uniform_int_true_2_2(
|
||||
eng,
|
||||
min_value,
|
||||
range,
|
||||
bmin,
|
||||
brange,
|
||||
result,
|
||||
mult,
|
||||
sprout::random::detail::generate_uniform_int(
|
||||
eng,
|
||||
static_cast<RangeType>(0),
|
||||
static_cast<RangeType>(range / mult),
|
||||
std::true_type()
|
||||
)
|
||||
)
|
||||
;
|
||||
}
|
||||
template<typename Engine, typename T, typename RangeType, typename BaseResult, typename BaseUnsigned>
|
||||
SPROUT_CONSTEXPR sprout::random::detail::generate_uniform_int_result<T, Engine> generate_uniform_int_true_2(
|
||||
Engine const& eng,
|
||||
T min_value,
|
||||
RangeType range,
|
||||
BaseResult bmin,
|
||||
BaseUnsigned brange
|
||||
)
|
||||
{
|
||||
return sprout::random::detail::generate_uniform_int_true_2_1(
|
||||
eng,
|
||||
min_value,
|
||||
range,
|
||||
bmin,
|
||||
brange,
|
||||
range == std::numeric_limits<RangeType>::max()
|
||||
? range / (RangeType(brange) + 1) + (
|
||||
range % (RangeType(brange) + 1) == RangeType(brange) ? 1 : 0
|
||||
)
|
||||
: (range + 1) / (RangeType(brange) + 1)
|
||||
);
|
||||
}
|
||||
template<typename Engine, typename T, typename BaseResult>
|
||||
SPROUT_CONSTEXPR sprout::random::detail::generate_uniform_int_result<T, Engine> generate_uniform_int_true_1_1(
|
||||
sprout::random::random_result<Engine> const& rnd,
|
||||
T min_value,
|
||||
BaseResult bmin
|
||||
)
|
||||
{
|
||||
typedef T result_type;
|
||||
typedef typename std::make_unsigned<BaseResult>::type base_unsigned;
|
||||
return sprout::random::detail::generate_uniform_int_result<T, Engine>{
|
||||
sprout::random::detail::add<base_unsigned, result_type>()(
|
||||
base_unsigned(sprout::random::detail::subtract<BaseResult>()(rnd.result(), bmin)),
|
||||
min_value
|
||||
),
|
||||
rnd.engine()
|
||||
};
|
||||
}
|
||||
template<typename Engine, typename T, typename RangeType, typename BaseResult, typename BaseUnsigned>
|
||||
SPROUT_CONSTEXPR sprout::random::detail::generate_uniform_int_result<T, Engine> generate_uniform_int_true_1(
|
||||
Engine const& eng,
|
||||
T min_value,
|
||||
T max_value,
|
||||
RangeType range,
|
||||
BaseResult bmin,
|
||||
BaseUnsigned brange
|
||||
)
|
||||
{
|
||||
return range == 0 ? sprout::random::detail::generate_uniform_int_result<T, Engine>{min_value, eng}
|
||||
: brange == range ? sprout::random::detail::generate_uniform_int_true_1_1(eng(), min_value, bmin)
|
||||
: brange < range ? sprout::random::detail::generate_uniform_int_true_2(eng, min_value, range, bmin, brange)
|
||||
: sprout::random::detail::generate_uniform_int_true_3(eng, min_value, range, bmin, brange)
|
||||
;
|
||||
}
|
||||
template<typename Engine, typename T>
|
||||
SPROUT_CONSTEXPR sprout::random::detail::generate_uniform_int_result<T, Engine> generate_uniform_int(
|
||||
Engine const& eng,
|
||||
T min_value,
|
||||
T max_value,
|
||||
std::true_type
|
||||
)
|
||||
{
|
||||
typedef T result_type;
|
||||
typedef typename std::make_unsigned<T>::type range_type;
|
||||
typedef typename Engine::result_type base_result;
|
||||
typedef typename std::make_unsigned<base_result>::type base_unsigned;
|
||||
return sprout::random::detail::generate_uniform_int_true_1(
|
||||
eng,
|
||||
min_value,
|
||||
max_value,
|
||||
range_type(sprout::random::detail::subtract<result_type>()(max_value, min_value)),
|
||||
base_result(eng.min()),
|
||||
base_unsigned(sprout::random::detail::subtract<base_result>()(eng.max(), eng.min()))
|
||||
);
|
||||
};
|
||||
template<typename Engine, typename T, typename Result>
|
||||
SPROUT_CONSTEXPR sprout::random::detail::generate_uniform_int_result<T, Engine> generate_uniform_int_false_1(
|
||||
Result const& rnd
|
||||
)
|
||||
{
|
||||
return sprout::random::detail::generate_uniform_int_result<T, Engine>{
|
||||
rnd.result,
|
||||
rnd.engine.base()
|
||||
};
|
||||
}
|
||||
template<typename Engine, typename T>
|
||||
SPROUT_CONSTEXPR sprout::random::detail::generate_uniform_int_result<T, Engine> generate_uniform_int(
|
||||
Engine const& eng,
|
||||
T min_value,
|
||||
T max_value,
|
||||
std::false_type
|
||||
)
|
||||
{
|
||||
return generate_uniform_int_false_1<Engine, T>(
|
||||
sprout::random::detail::generate_uniform_int(
|
||||
sprout::random::detail::uniform_int_float<Engine>(eng),
|
||||
min_value,
|
||||
max_value,
|
||||
std::true_type()
|
||||
)
|
||||
);
|
||||
}
|
||||
template<typename Engine, typename T>
|
||||
SPROUT_CONSTEXPR sprout::random::detail::generate_uniform_int_result<T, Engine> generate_uniform_int(
|
||||
Engine const& eng,
|
||||
T min_value,
|
||||
T max_value
|
||||
)
|
||||
{
|
||||
return generate_uniform_int(
|
||||
eng,
|
||||
min_value,
|
||||
max_value,
|
||||
std::is_integral<typename Engine::result_type>()
|
||||
);
|
||||
}
|
||||
} // namespace detail
|
||||
//
|
||||
// uniform_int_distribution
|
||||
//
|
||||
template<typename IntType = int>
|
||||
class uniform_int_distribution {
|
||||
public:
|
||||
typedef IntType input_type;
|
||||
typedef IntType result_type;
|
||||
private:
|
||||
static SPROUT_CONSTEXPR IntType arg_check(IntType min_arg, IntType max_arg) {
|
||||
return min_arg <= max_arg
|
||||
? min_arg
|
||||
: throw "assert(min_arg <= max_arg)"
|
||||
;
|
||||
}
|
||||
public:
|
||||
//
|
||||
// param_type
|
||||
//
|
||||
class param_type {
|
||||
public:
|
||||
typedef uniform_int_distribution distribution_type;
|
||||
private:
|
||||
IntType min_;
|
||||
IntType max_;
|
||||
public:
|
||||
SPROUT_CONSTEXPR param_type()
|
||||
: min_(0)
|
||||
, max_(9)
|
||||
{}
|
||||
SPROUT_CONSTEXPR explicit param_type(IntType min_arg, IntType max_arg = 9)
|
||||
: min_(arg_check(min_arg, max_arg))
|
||||
, max_(max_arg)
|
||||
{}
|
||||
SPROUT_CONSTEXPR IntType a() const {
|
||||
return min_;
|
||||
}
|
||||
SPROUT_CONSTEXPR IntType b() const {
|
||||
return max_;
|
||||
}
|
||||
template<typename Elem, typename Traits>
|
||||
friend std::basic_ostream<Elem, Traits>& operator>>(
|
||||
std::basic_istream<Elem, Traits>& lhs,
|
||||
param_type const& rhs
|
||||
)
|
||||
{
|
||||
return lhs >> rhs.min_ >> std::ws >> rhs.max_;
|
||||
}
|
||||
template<typename Elem, typename Traits>
|
||||
friend std::basic_ostream<Elem, Traits>& operator<<(
|
||||
std::basic_ostream<Elem, Traits>& lhs,
|
||||
param_type const& rhs
|
||||
)
|
||||
{
|
||||
return lhs << rhs.min_ << " " << rhs.max_;
|
||||
}
|
||||
SPROUT_CONSTEXPR friend bool operator==(param_type const& lhs, param_type const& rhs) {
|
||||
return lhs.min_ == rhs.min_ && lhs.max_ == rhs.max_;
|
||||
}
|
||||
SPROUT_CONSTEXPR friend bool operator!=(param_type const& lhs, param_type const& rhs) {
|
||||
return !(lhs == rhs);
|
||||
}
|
||||
};
|
||||
private:
|
||||
result_type min_;
|
||||
result_type max_;
|
||||
private:
|
||||
template<typename Engine, typename Result>
|
||||
SPROUT_CONSTEXPR sprout::random::random_result<Engine, uniform_int_distribution> generate(Result const& rnd) const {
|
||||
return sprout::random::random_result<Engine, uniform_int_distribution>(
|
||||
rnd.result,
|
||||
rnd.engine,
|
||||
*this
|
||||
);
|
||||
}
|
||||
public:
|
||||
SPROUT_CONSTEXPR uniform_int_distribution()
|
||||
: min_(0)
|
||||
, max_(9)
|
||||
{}
|
||||
SPROUT_CONSTEXPR explicit uniform_int_distribution(IntType min_arg, IntType max_arg = 9)
|
||||
: min_(arg_check(min_arg, max_arg))
|
||||
, max_(max_arg)
|
||||
{}
|
||||
explicit uniform_int_distribution(param_type const& parm)
|
||||
: min_(parm.a())
|
||||
, max_(parm.b())
|
||||
{}
|
||||
SPROUT_CONSTEXPR result_type a() const {
|
||||
return min_;
|
||||
}
|
||||
SPROUT_CONSTEXPR result_type b() const {
|
||||
return max_;
|
||||
}
|
||||
SPROUT_CONSTEXPR result_type min() const {
|
||||
return min_;
|
||||
}
|
||||
SPROUT_CONSTEXPR result_type max() const {
|
||||
return max_;
|
||||
}
|
||||
SPROUT_CONSTEXPR param_type param() const {
|
||||
return param_type(min_, max_);
|
||||
}
|
||||
void param(param_type const& parm) {
|
||||
min_ = parm.a();
|
||||
max_ = parm.b();
|
||||
}
|
||||
template<typename Engine>
|
||||
SPROUT_CONSTEXPR sprout::random::random_result<Engine, uniform_int_distribution> operator()(Engine const& eng) const {
|
||||
return generate<Engine>(sprout::random::detail::generate_uniform_int(eng, min_, max_));
|
||||
}
|
||||
template<typename Elem, typename Traits>
|
||||
friend std::basic_ostream<Elem, Traits>& operator>>(
|
||||
std::basic_istream<Elem, Traits>& lhs,
|
||||
uniform_int_distribution const& rhs
|
||||
)
|
||||
{
|
||||
param_type parm;
|
||||
return lhs >> parm;
|
||||
param(parm);
|
||||
return lhs;
|
||||
}
|
||||
template<typename Elem, typename Traits>
|
||||
friend std::basic_ostream<Elem, Traits>& operator<<(
|
||||
std::basic_ostream<Elem, Traits>& lhs,
|
||||
uniform_int_distribution const& rhs
|
||||
)
|
||||
{
|
||||
return lhs << param();
|
||||
}
|
||||
SPROUT_CONSTEXPR friend bool operator==(uniform_int_distribution const& lhs, uniform_int_distribution const& rhs) {
|
||||
return lhs.param() == rhs.param();
|
||||
}
|
||||
SPROUT_CONSTEXPR friend bool operator!=(uniform_int_distribution const& lhs, uniform_int_distribution const& rhs) {
|
||||
return !(lhs == rhs);
|
||||
}
|
||||
};
|
||||
} // namespace random
|
||||
|
||||
using sprout::random::uniform_int_distribution;
|
||||
} // namespace sprout
|
||||
|
||||
#endif // #ifndef SPROUT_RANDOM_UNIFORM_INT_DISTRIBUTION_HPP
|
||||
|
|
@ -19,6 +19,13 @@ namespace sprout {
|
|||
public:
|
||||
typedef IntType input_type;
|
||||
typedef IntType result_type;
|
||||
private:
|
||||
static SPROUT_CONSTEXPR IntType arg_check(IntType min_arg, IntType max_arg) {
|
||||
return min_arg <= max_arg
|
||||
? min_arg
|
||||
: throw "assert(min_arg <= max_arg)"
|
||||
;
|
||||
}
|
||||
public:
|
||||
//
|
||||
// param_type
|
||||
|
@ -26,13 +33,6 @@ namespace sprout {
|
|||
class param_type {
|
||||
public:
|
||||
typedef uniform_smallint distribution_type;
|
||||
private:
|
||||
static SPROUT_CONSTEXPR IntType arg_check(IntType min_arg, IntType max_arg) {
|
||||
return min_arg <= max_arg
|
||||
? min_arg
|
||||
: throw "assert(min_arg <= max_arg)"
|
||||
;
|
||||
}
|
||||
private:
|
||||
IntType min_;
|
||||
IntType max_;
|
||||
|
@ -189,7 +189,7 @@ namespace sprout {
|
|||
, max_(9)
|
||||
{}
|
||||
SPROUT_CONSTEXPR explicit uniform_smallint(IntType min_arg, IntType max_arg = 9)
|
||||
: min_(min_arg)
|
||||
: min_(arg_check(min_arg, max_arg))
|
||||
, max_(max_arg)
|
||||
{}
|
||||
explicit uniform_smallint(param_type const& parm)
|
||||
|
|
Loading…
Reference in a new issue