0
0
Fork 0
mirror of https://github.com/matrix-construct/construct synced 2024-07-07 19:18:35 +02:00
construct/include/ircd/rand.h
2020-10-07 03:08:28 -07:00

83 lines
1.9 KiB
C++

// Matrix Construct
//
// Copyright (C) Matrix Construct Developers, Authors & Contributors
// Copyright (C) 2016-2018 Jason Volk <jason@zemos.net>
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice is present in all copies. The
// full license for this software is available in the LICENSE file.
#pragma once
#define HAVE_IRCD_RAND_H
/// Some character set dictionaries
namespace ircd::rand::dict
{
extern const std::string alnum;
extern const std::string alpha;
extern const std::string upper;
extern const std::string lower;
extern const std::string numeric;
}
/// Tools for randomization
namespace ircd::rand
{
// Interface state.
extern std::random_device device;
extern std::mt19937_64 mt;
// Random integer
uint64_t integer();
uint64_t integer(const uint64_t &min, const uint64_t &max); // inclusive
// Random vector
template<class T> T vector() = delete;
template<> u128x1 vector();
template<> u256x1 vector();
template<> u512x1 vector();
// Random character from dictionary
char character(const std::string &dict = dict::alnum);
// Random string from dictionary, fills buffer
string_view string(const mutable_buffer &out, const std::string &dict);
// Random fill of buffer
const_buffer fill(const mutable_buffer &out);
}
/// Random character from dictionary
inline char
ircd::rand::character(const std::string &dict)
{
assert(!dict.empty());
const auto pos
{
integer(0, dict.size() - 1)
};
return dict.at(pos);
}
/// Random integer in range (inclusive)
inline uint64_t
ircd::rand::integer(const uint64_t &min,
const uint64_t &max)
{
std::uniform_int_distribution<uint64_t> dist
{
min, max
};
return dist(mt);
}
/// Random 64-bits
inline uint64_t
ircd::rand::integer()
{
return mt();
}