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

//! Random number generation.
use std::ops::{Bound, RangeBounds};

/// Seed the random number generator with a random seed.
pub fn random_seed() {
    unsafe { lotus_script_sys::rand::random_seed() }
}

/// Seed the random number generator.
pub fn seed(seed: u64) {
    unsafe { lotus_script_sys::rand::seed(seed) }
}

/// Generate a random f64 in the range 0 to 1.
pub fn gen_f64() -> f64 {
    unsafe { lotus_script_sys::rand::f64() }
}

/// Generate a random u64 for the given range.
pub fn gen_u64(range: impl RangeBounds<u64>) -> u64 {
    let min = match range.start_bound() {
        Bound::Included(min) => *min,
        Bound::Excluded(min) => min + 1,
        Bound::Unbounded => 0,
    };

    let max = match range.end_bound() {
        Bound::Included(max) => *max,
        Bound::Excluded(max) => max - 1,
        Bound::Unbounded => u64::MAX,
    };

    assert!(min <= max, "min must be less than or equal to max");

    unsafe { lotus_script_sys::rand::u64(min, max) }
}