69 lines
1.7 KiB
Rust
69 lines
1.7 KiB
Rust
use super::{Domain, Stage};
|
|
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
/// A map for data stored in memory for processing and saving to disk.
|
|
// derived Ord impl *should* use self.domain.start to sort? Not sure.
|
|
// Use `sort_by_key()` to be safe.
|
|
#[derive(Clone, Copy, Debug, Serialize, Deserialize, PartialEq, Eq, PartialOrd, Ord)]
|
|
pub struct Cluster {
|
|
pub domain: Domain,
|
|
pub stage: Stage,
|
|
}
|
|
|
|
impl Default for Cluster {
|
|
fn default() -> Self {
|
|
Cluster {
|
|
domain: Domain::default(),
|
|
stage: Stage::default(),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Cluster {
|
|
/// Breaks apart into a vec of clusters,
|
|
/// each of cluster_size, excepting last.
|
|
#[allow(dead_code)]
|
|
pub fn subdivide(&mut self, cluster_len: usize) -> Vec<Cluster> {
|
|
let domain_len = self.domain.len();
|
|
let mut start = self.domain.start;
|
|
let mut clusters: Vec<Cluster> = vec![];
|
|
|
|
for _ in 0..(domain_len / cluster_len) {
|
|
clusters.push(Cluster {
|
|
domain: Domain {
|
|
start,
|
|
end: start + cluster_len,
|
|
},
|
|
stage: self.stage,
|
|
});
|
|
|
|
start += cluster_len;
|
|
}
|
|
|
|
clusters.push(Cluster {
|
|
domain: Domain {
|
|
start,
|
|
end: self.domain.end,
|
|
},
|
|
stage: self.stage,
|
|
});
|
|
|
|
clusters
|
|
}
|
|
|
|
// This is used in unit tests at present. Ideally it probably shouldn't exist.
|
|
#[allow(dead_code)]
|
|
pub fn set_stage(&mut self, stage: Stage) -> &mut Self {
|
|
self.stage = stage;
|
|
self
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
|
|
// Test for Cluster::subdivide()
|
|
}
|