spdkit 0.1.0

spdkit: Structure Predication Development Kit
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

// [[file:../spdkit.note::*imports][imports:1]]
use std::sync::Mutex;

use crate::common::*;
// imports:1 ends here

// [[file:../spdkit.note::6594d8a9][6594d8a9]]
pub use rand::prelude::*;

// To not cause trouble, make sure one RNG per thread
fn random_seed_u64() -> u64 {
    rand::random::<u64>()
}

fn get_rng_with_seed(seed: Option<u64>) -> StdRng {
    let seed = seed.unwrap_or_else(|| rand::random::<u64>());
    info!("Initialize rng with seed {}", seed);
    println!("spdkit: the random process can be repeated with seed {seed}");
    let line = format!("export SPDKIT_RANDOM_SEED={seed}");
    println!("spdkit: use shell cmd {line:?} to set it");

    StdRng::seed_from_u64(seed)
}

lazy_static! {
    pub static ref RNG: Mutex<StdRng> = {
        let seed = crate::vars::Vars::from_env().random_seed;
        let mut r = get_rng_with_seed(seed);
        Mutex::new(r)
    };
}

#[macro_export]
macro_rules! get_rng {
    () => {
        RNG.lock().unwrap()
    };
}

#[cfg(test)]
mod test {
    #[test]
    fn test_rng() {
        use crate::random::*;

        let mut rng = get_rng!();
        rng.gen::<i32>();
        assert_eq!(rng.gen_range(0..1), 0);
    }
}
// 6594d8a9 ends here