use crate::FitnessEvaluator;
pub struct Parameters<T: FitnessEvaluator> {
pub population_size: usize,
pub genetic_code_length: usize,
pub keep_threshold: f64,
pub mutation_chance_percent: f64,
pub emit_result_every: usize,
pub fitness_evaluator: T,
}
impl<T> Parameters<T>
where
T: FitnessEvaluator,
{
pub fn new(evaluator: T) -> Parameters<T> {
Parameters {
population_size: 100,
genetic_code_length: 10,
keep_threshold: 0.5,
mutation_chance_percent: 0.01,
emit_result_every: 1000,
fitness_evaluator: evaluator,
}
}
pub fn with_population_size(mut self, population_size: usize) -> Self {
self.population_size = population_size;
self
}
pub fn with_genetic_code_length(mut self, length: usize) -> Self {
self.genetic_code_length = length;
self
}
pub fn with_keep_threshold(mut self, threshold: f64) -> Self {
self.keep_threshold = threshold;
self
}
pub fn with_mutation_chance_percent(mut self, mutation_chance: f64) -> Self {
self.mutation_chance_percent = mutation_chance;
self
}
}
#[cfg(test)]
mod tests {
use super::{FitnessEvaluator, Parameters};
struct MockEvaluator;
impl FitnessEvaluator for MockEvaluator {
fn evaluate(&self, _code: &Vec<u8>) -> f64 {
0.0
}
}
#[test]
fn parameters_init() {
let params = Parameters::new(MockEvaluator {});
assert_eq!(params.genetic_code_length, 10);
assert_eq!(params.population_size, 100);
assert_eq!(params.keep_threshold, 0.5);
assert_eq!(params.mutation_chance_percent, 0.01);
assert_eq!(params.emit_result_every, 1000);
}
}