argmin/core/math/
random_vec.rs

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
// Copyright 2018-2020 argmin developers
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://apache.org/licenses/LICENSE-2.0> or the MIT license <LICENSE-MIT or
// http://opensource.org/licenses/MIT>, at your option. This file may not be
// copied, modified, or distributed except according to those terms.

use crate::core::math::ArgminRandom;
use rand::distributions::uniform::SampleUniform;
use rand::Rng;

impl<T> ArgminRandom for Vec<T>
where
    T: SampleUniform + std::cmp::PartialOrd + Clone,
{
    fn rand_from_range(min: &Self, max: &Self) -> Vec<T> {
        assert!(!min.is_empty());
        assert_eq!(min.len(), max.len());

        let mut rng = rand::thread_rng();

        min.iter()
            .zip(max.iter())
            .map(|(a, b)| {
                // Do not require a < b:

                if a == b {
                    a.clone()
                } else if a < b {
                    rng.gen_range(a, b)
                } else {
                    rng.gen_range(b, a)
                }
            })
            .collect()
    }
}