use artificial_bee_colony::{bee::Bee, food::FoodSource, give_output, Input};
fn main() {
    println!("Hello, world!");

    let input = Input::get();
    let mut function_results = vec![];

    let mut food_sources = FoodSource::create_food_sources(
        input.food_source_number,
        input.decision_variable_count,
        input.upper_bound,
        input.lower_bound,
    );

    for run_counter in 0..input.run {
        let mut best_food_source = FoodSource::get(vec![]);
        for food_source in &food_sources {
            if best_food_source.fitness < food_source.fitness {
                best_food_source
                    .coordinates
                    .clone_from(&food_source.coordinates);
                best_food_source.fitness = food_source.fitness;
            }
        }
        for _ in 0..input.iteration {
            for index in 0..input.food_source_number as usize {
                Bee::worker_bee(
                    &mut food_sources,
                    index,
                    input.decision_variable_count,
                    input.upper_bound,
                    input.lower_bound,
                );
                let mut total_fitness = 0.0;
                for food_source in &food_sources {
                    total_fitness += food_source.fitness;
                }
                Bee::onlooker_bee(
                    &mut food_sources,
                    index,
                    total_fitness,
                    input.decision_variable_count,
                    input.upper_bound,
                    input.lower_bound,
                );
                let mut most_tried_index = 0;
                for i in 0..food_sources.len() {
                    if food_sources[most_tried_index].try_counter < food_sources[i].try_counter {
                        most_tried_index = i;
                    }
                }
                best_food_source = food_sources[most_tried_index].clone();
                Bee::scout_bee(
                    &mut food_sources,
                    most_tried_index,
                    input.food_source_try_limit,
                    input.lower_bound,
                    input.upper_bound,
                    input.decision_variable_count,
                );
            }
        }
        function_results.push(best_food_source.function_calculation);
        give_output(
            best_food_source,
            &function_results[..],
            input.run,
            run_counter,
        )
    }
}