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
use crate::Dataset;
pub struct ComposedDataset<I> {
datasets: Vec<Box<dyn Dataset<I>>>,
}
impl<I> ComposedDataset<I> {
pub fn new(datasets: Vec<Box<dyn Dataset<I>>>) -> Self {
Self { datasets }
}
}
impl<I> Dataset<I> for ComposedDataset<I>
where
I: Clone,
{
fn get(&self, index: usize) -> Option<I> {
let mut current_index = 0;
for dataset in self.datasets.iter() {
if index < dataset.len() + current_index {
return dataset.get(index - current_index);
}
current_index += dataset.len();
}
None
}
fn len(&self) -> usize {
let mut total = 0;
for dataset in self.datasets.iter() {
total += dataset.len();
}
total
}
}