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