Tahinli/optimization_algorithms
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

pre commit

Browse source
This commit is contained in:
Ahmet Kaan GÜMÜŞ 2024-11-20 00:20:40 +03:00
parent b0fb8f40d8
commit 48a0497ead
2 changed files with 50 additions and 36 deletions
Download patch file Download diff file Expand all files Collapse all files

49
artificial_bee_colony/src/bee.rs
View file

@ -1,10 +1,53 @@
use rand::Rng; use rand::Rng;
use crate::food::FoodSource; use crate::{food::FoodSource, utils::Input};
pub struct Bee {} pub struct Bee {}
impl Bee { impl Bee {
pub fn employed_bee( pub fn send_all_employed_bees(
food_sources: &mut [FoodSource],
input:&Input,
) {
for food_source_index in 0..input.food_source_number {
Bee::employed_bee(
food_sources,
food_source_index,
input.decision_variable_count,
input.upper_bound,
input.lower_bound,
);
}
}
pub fn send_all_onlooker_bees(food_sources: &mut [FoodSource],
input:&Input,) {
let total_fitness = food_sources
.iter()
.map(|food_source| food_source.fitness_calculation)
.sum();
let onlooker_bee_count = input.food_source_number;
let mut last_looked = 0;
for _ in 0..onlooker_bee_count {
loop {
if last_looked >= input.food_source_number {
last_looked = 0;
}
if Bee::onlooker_bee(
food_sources,
last_looked,
total_fitness,
input.decision_variable_count,
input.upper_bound,
input.lower_bound,
) {
break;
}
last_looked += 1;
}
}
}
fn employed_bee(
food_sources: &mut [FoodSource], food_sources: &mut [FoodSource],
food_source_index: usize, food_source_index: usize,
decision_variable_count: usize, decision_variable_count: usize,
@ -20,7 +63,7 @@ impl Bee {
); );
} }
pub fn onlooker_bee( fn onlooker_bee(
food_sources: &mut [FoodSource], food_sources: &mut [FoodSource],
food_source_index: usize, food_source_index: usize,
total_fitness: f64, total_fitness: f64,

39
artificial_bee_colony/src/main.rs
View file

@ -19,41 +19,12 @@ fn main() {
); );
let mut best_food_source = FoodSource::find_best_food_source(&food_sources); let mut best_food_source = FoodSource::find_best_food_source(&food_sources);
for _ in 0..input.iteration {
for food_source_index in 0..input.food_source_number {
Bee::employed_bee(
&mut food_sources,
food_source_index,
input.decision_variable_count,
input.upper_bound,
input.lower_bound,
);
}
let total_fitness = food_sources for _ in 0..input.iteration {
.iter() Bee::send_all_employed_bees(&mut food_sources, &input);
.map(|food_source| food_source.fitness_calculation)
.sum(); Bee::send_all_onlooker_bees(&mut food_sources, &input);
let onlooker_bee_count = input.food_source_number;
let mut last_looked = 0;
for _ in 0..onlooker_bee_count {
loop {
if last_looked >= input.food_source_number {
last_looked = 0;
}
if Bee::onlooker_bee(
&mut food_sources,
last_looked,
total_fitness,
input.decision_variable_count,
input.upper_bound,
input.lower_bound,
) {
break;
}
last_looked += 1;
}
}
best_food_source = FoodSource::find_best_food_source(&food_sources); best_food_source = FoodSource::find_best_food_source(&food_sources);
let most_tried_food_source_index = let most_tried_food_source_index =