use selection_rate::SelectionRates::*;
use slope_params::SlopeParams;
pub trait SelectionRate: Send + Sync {
fn rate(
&self,
generation: u64,
progress: f64,
n_solutions: usize,
population_fitness: &[f64],
) -> usize;
}
pub enum SelectionRates {
Constant(usize),
Linear(SlopeParams),
Quadratic(SlopeParams),
}
impl SelectionRate for SelectionRates {
fn rate(
&self,
generation: u64,
_progress: f64,
_n_solutions: usize,
_population_fitness: &[f64],
) -> usize {
match self {
Constant(c) => *c,
Linear(sp) => sp
.check_bound(sp.coefficient * generation as f64 + sp.start)
.ceil() as usize,
Quadratic(sp) => sp
.check_bound(sp.coefficient * generation as f64 * generation as f64 + sp.start)
.ceil() as usize,
}
}
}