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
52
53
54
55
56
57
58
59
60
61
|
use crate::stats::bivariate::Data;
use crate::stats::float::Float;
use crate::stats::rand_util::{new_rng, Rng};
pub struct Resamples<'a, X, Y>
where
X: 'a + Float,
Y: 'a + Float,
{
rng: Rng,
data: (&'a [X], &'a [Y]),
stage: Option<(Vec<X>, Vec<Y>)>,
}
#[cfg_attr(feature = "cargo-clippy", allow(clippy::should_implement_trait))]
impl<'a, X, Y> Resamples<'a, X, Y>
where
X: 'a + Float,
Y: 'a + Float,
{
pub fn new(data: Data<'a, X, Y>) -> Resamples<'a, X, Y> {
Resamples {
rng: new_rng(),
data: (data.x(), data.y()),
stage: None,
}
}
pub fn next(&mut self) -> Data<'_, X, Y> {
let n = self.data.0.len();
match self.stage {
None => {
let mut stage = (Vec::with_capacity(n), Vec::with_capacity(n));
for _ in 0..n {
let i = self.rng.rand_range(0u64..(self.data.0.len() as u64)) as usize;
stage.0.push(self.data.0[i]);
stage.1.push(self.data.1[i]);
}
self.stage = Some(stage);
}
Some(ref mut stage) => {
for i in 0..n {
let j = self.rng.rand_range(0u64..(self.data.0.len() as u64)) as usize;
stage.0[i] = self.data.0[j];
stage.1[i] = self.data.1[j];
}
}
}
if let Some((ref x, ref y)) = self.stage {
Data(x, y)
} else {
unreachable!();
}
}
}
|
