1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
mod elitism;
pub use self::elitism::Elitism;
mod greedy;
pub use self::greedy::Greedy;
mod rosomaxa;
pub use self::rosomaxa::Rosomaxa;
pub use self::rosomaxa::RosomaxaConfig;
use crate::construction::heuristics::InsertionContext;
use crate::models::Problem;
use crate::solver::Statistics;
use crate::utils::{compare_floats, Environment};
use std::cmp::Ordering;
use std::fmt::Display;
use std::sync::Arc;
pub type Individual = InsertionContext;
#[derive(Debug, PartialEq, Eq, Hash)]
pub enum SelectionPhase {
Initial,
Exploration,
Exploitation,
}
pub trait Population: Display {
fn add_all(&mut self, individuals: Vec<Individual>) -> bool;
fn add(&mut self, individual: Individual) -> bool;
fn on_generation(&mut self, statistics: &Statistics);
fn cmp(&self, a: &Individual, b: &Individual) -> Ordering;
fn select<'a>(&'a self) -> Box<dyn Iterator<Item = &Individual> + 'a>;
fn ranked<'a>(&'a self) -> Box<dyn Iterator<Item = (&Individual, usize)> + 'a>;
fn size(&self) -> usize;
fn selection_phase(&self) -> SelectionPhase;
}
fn is_same_fitness(a: &Individual, b: &Individual) -> bool {
let fitness_a = a.get_fitness_values();
let fitness_b = b.get_fitness_values();
fitness_a.zip(fitness_b).all(|(a, b)| compare_floats(a, b) == Ordering::Equal)
}
pub fn get_default_selection_size(environment: &Environment) -> usize {
environment.parallelism.available_cpus().min(8)
}
pub fn get_default_population(
problem: Arc<Problem>,
environment: Arc<Environment>,
) -> Box<dyn Population + Send + Sync> {
let selection_size = get_default_selection_size(environment.as_ref());
if selection_size == 1 {
Box::new(Greedy::new(problem, 1, None))
} else {
let config = RosomaxaConfig::new_with_defaults(selection_size);
let population =
Rosomaxa::new(problem, environment, config).expect("cannot create rosomaxa with default configuration");
Box::new(population)
}
}
pub fn create_elitism_population(
problem: Arc<Problem>,
environment: Arc<Environment>,
) -> Box<dyn Population + Sync + Send> {
let selection_size = get_default_selection_size(environment.as_ref());
Box::new(Elitism::new(problem, environment.random.clone(), 4, selection_size))
}