ecrs 0.1.0-beta.3

use ecrs::ga;
mod util;

#[allow(clippy::ptr_arg)]
pub fn rastrigin_fitness(chromosome: &Vec<f64>) -> f64 {
  1000.0 * f64::exp(-rastrigin(chromosome))
}

fn rastrigin(chromosome: &[f64]) -> f64 {
  10.0 * chromosome.len() as f64
    + chromosome.iter().fold(0.0, |sum, x| {
      sum + x.powi(2) - 10.0 * (2.0 * std::f64::consts::PI * x).cos()
    })
}

fn main() {
  let _ = util::init_logging();

  let best_individual = ga::Builder::with_rvc()
    .set_max_duration(std::time::Duration::from_secs(10))
    .fitness_fn(rastrigin_fitness)
    .dim(5)
    .build()
    .run()
    .unwrap();

  println!("5D Rastrigin function zero approximation {:#?}", best_individual)
}