work/smartcore
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix std_rand feature

Browse Source
This commit is contained in:
Lorenzo (Mec-iS)
2022-11-08 12:04:39 +00:00
parent c1af60cafb
commit 3c4a807be8
1 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/rand_custom.rs
+8 -7
View File
@@ -1,19 +1,20 @@
#[cfg(not(feature = "std"))]
pub(crate) use rand::rngs::SmallRng as RngImpl;
#[cfg(feature = "std")]
pub(crate) use rand::rngs::StdRng as RngImpl;
#[cfg(not(feature = "std_rand"))]
pub use rand::rngs::SmallRng as RngImpl;
#[cfg(feature = "std_rand")]
pub use rand::rngs::StdRng as RngImpl;
use rand::SeedableRng;
pub(crate) fn get_rng_impl(seed: Option<u64>) -> RngImpl {
/// Custom switch for random fuctions
pub fn get_rng_impl(seed: Option<u64>) -> RngImpl {
match seed {
Some(seed) => RngImpl::seed_from_u64(seed),
None => {
cfg_if::cfg_if! {
if #[cfg(feature = "std")] {
if #[cfg(feature = "std_rand")] {
use rand::RngCore;
RngImpl::seed_from_u64(rand::thread_rng().next_u64())
} else {
// non-std build, use getrandom
// no std_random feature build, use getrandom
let mut buf = [0u8; 64];
getrandom::getrandom(&mut buf).unwrap();
RngImpl::seed_from_u64(buf[0] as u64)