refactor: ♻️ bee, food

artificial_bee_colony/src/bee.rs

@@ -0,0 +1,65 @@
+use rand::Rng;
+
+use crate::food::FoodSource;
+
+pub enum Bee {}
+impl Bee {
+    pub fn worker_bee(
+        food_sources: &mut [FoodSource],
+        index: usize,
+        decision_variable_count: usize,
+    ) {
+        let mut different_food_source_index = rand::thread_rng().gen_range(0..food_sources.len());
+        while different_food_source_index == index {
+            different_food_source_index = rand::thread_rng().gen_range(0..food_sources.len());
+        }
+        let selected_coordinate_index = rand::thread_rng().gen_range(0..decision_variable_count);
+        let randomness = rand::thread_rng().gen_range(-1.0..=1.0);
+
+        let candidate_one_index = food_sources[index].coordinates[selected_coordinate_index]
+            + randomness
+                * (food_sources[index].coordinates[selected_coordinate_index]
+                    - food_sources[different_food_source_index].coordinates
+                        [selected_coordinate_index]);
+        let mut candidate_coordinates = food_sources[index].coordinates.clone();
+        candidate_coordinates[selected_coordinate_index] = candidate_one_index;
+        let candidate = FoodSource::get(candidate_coordinates);
+        food_sources[index].try_counter += 1;
+        if candidate.fitness > food_sources[index].fitness {
+            food_sources[index] = candidate;
+            food_sources[index].try_counter = 0;
+        }
+    }
+
+    pub fn onlooker_bee(
+        food_sources: &mut [FoodSource],
+        index: usize,
+        total_fitness: f64,
+        decision_variable_count: usize,
+    ) {
+        let fitness_for_index = food_sources[index].fitness;
+        if fitness_for_index / total_fitness <= rand::thread_rng().gen_range(0.0..=1.0) {
+            Self::worker_bee(food_sources, index, decision_variable_count);
+        }
+    }
+
+    pub fn scout_bee(
+        food_sources: &mut [FoodSource],
+        most_tried_index: usize,
+        limit: u128,
+        lower_bound: f64,
+        upper_bound: f64,
+        decision_variable_count: usize,
+    ) {
+        if food_sources[most_tried_index].try_counter > limit {
+            let mut coordinates_for_new = vec![];
+            for _ in 0..decision_variable_count {
+                let random = lower_bound
+                    + rand::thread_rng().gen_range(0.0..=1.0) * (upper_bound - lower_bound);
+                coordinates_for_new.push(random);
+            }
+            let new_food_source = FoodSource::get(coordinates_for_new);
+            food_sources[most_tried_index] = new_food_source;
+        }
+    }
+}