#[cfg(feature = "telemetry")]
extern crate open;

extern crate rustneat;

#[cfg(feature = "telemetry")]
extern crate rusty_dashed;

extern crate rand;

use rustneat::Environment;
use rustneat::Organism;
use rustneat::Population;

#[cfg(feature = "telemetry")]
use rusty_dashed::Dashboard;

struct XORClassification;

impl Environment for XORClassification {
    fn test(&self, organism: &mut Organism) -> f64 {
        let mut output = vec![0f64];
        let mut distance: f64;
        organism.activate(&vec![0f64, 0f64], &mut output);
        distance = (0f64 - output[0]).abs();
        organism.activate(&vec![0f64, 1f64], &mut output);
        distance += (1f64 - output[0]).abs();
        organism.activate(&vec![1f64, 0f64], &mut output);
        distance += (1f64 - output[0]).abs();
        organism.activate(&vec![1f64, 1f64], &mut output);
        distance += (0f64 - output[0]).abs();

        let fitness = (4f64 - distance).powi(2);

        fitness
    }
}

#[cfg(feature = "telemetry")]
fn enable_telemetry() {
    let mut dashboard = Dashboard::new();
    dashboard.add_graph("fitness1", "fitness", 0, 0, 4, 4);
    dashboard.add_graph("network1", "network", 4, 0, 4, 4);

    rusty_dashed::Server::serve_dashboard(dashboard);

    match open::that("http://localhost:3000") {
        Err(_) => println!(
            "\nOpen browser and go to http://localhost:3000 to see how neural network evolves\n"
        ),
        _ => println!("Openning browser..."),
    }
}

fn main() {
    #[cfg(feature = "telemetry")]
    enable_telemetry();

    let mut population = Population::create_population(150);
    let mut environment = XORClassification;
    let mut champion: Option<Organism> = None;
    while champion.is_none() {
        population.evolve();
        population.evaluate_in(&mut environment);
        for organism in &population.get_organisms() {
            if organism.fitness > 15.9f64 {
                champion = Some(organism.clone());
            }
        }
    }
    println!("{:?}", champion.unwrap().genome);
}