feat(pareto): add pareto_front and best_candidate

`pareto_front` returns all candidates not dominated by any other in
input order (O(N²·M), acceptable for v1 per spec §9.3).

`best_candidate` is the single-objective "best" finder: returns None
unless there is exactly one objective; ignores infeasibles; returns None
if every candidate is infeasible (spec §9.4).

Both re-exported from the prelude.
This commit is contained in:
2026-05-04 19:19:13 -06:00
parent 95e797647b
commit da9baf0325
3 changed files with 155 additions and 1 deletions
+152
View File
@@ -0,0 +1,152 @@
//! Compute the non-dominated front of a population, and the single-objective best.
use crate::core::candidate::Candidate;
use crate::core::objective::ObjectiveSpace;
use crate::pareto::dominance::{Dominance, pareto_compare};
/// Return all candidates that are not dominated by any other candidate.
///
/// O(N²·M) in v1 (spec §9.3). Input order is preserved among returned
/// candidates.
pub fn pareto_front<D: Clone>(
population: &[Candidate<D>],
objectives: &ObjectiveSpace,
) -> Vec<Candidate<D>> {
let mut out = Vec::new();
'outer: for (i, a) in population.iter().enumerate() {
for (j, b) in population.iter().enumerate() {
if i == j {
continue;
}
if matches!(
pareto_compare(&a.evaluation, &b.evaluation, objectives),
Dominance::DominatedBy
) {
continue 'outer;
}
}
out.push(a.clone());
}
out
}
/// Return the best candidate for a single-objective problem.
///
/// Returns `None` if there is not exactly one objective, if the population is
/// empty, or if every candidate is infeasible (spec §9.4).
pub fn best_candidate<D: Clone>(
population: &[Candidate<D>],
objectives: &ObjectiveSpace,
) -> Option<Candidate<D>> {
if !objectives.is_single_objective() {
return None;
}
let mut best: Option<&Candidate<D>> = None;
let mut best_min: f64 = f64::INFINITY;
for c in population {
if !c.evaluation.is_feasible() {
continue;
}
let m = objectives.as_minimization(&c.evaluation.objectives);
let v = m.first().copied().unwrap_or(f64::INFINITY);
if best.is_none() || v < best_min {
best = Some(c);
best_min = v;
}
}
best.cloned()
}
#[cfg(test)]
mod tests {
use super::*;
use crate::core::evaluation::Evaluation;
use crate::core::objective::Objective;
fn cand(decision: u32, obj: Vec<f64>) -> Candidate<u32> {
Candidate::new(decision, Evaluation::new(obj))
}
fn space_min2() -> ObjectiveSpace {
ObjectiveSpace::new(vec![
Objective::minimize("f1"),
Objective::minimize("f2"),
])
}
#[test]
fn empty_population_returns_empty_front() {
let s = space_min2();
let front = pareto_front::<u32>(&[], &s);
assert!(front.is_empty());
}
#[test]
fn single_candidate_is_its_own_front() {
let s = space_min2();
let pop = [cand(1, vec![1.0, 2.0])];
let front = pareto_front(&pop, &s);
assert_eq!(front.len(), 1);
assert_eq!(front[0].decision, 1);
}
#[test]
fn dominated_points_removed_non_dominated_kept() {
let s = space_min2();
let pop = [
cand(1, vec![1.0, 4.0]), // non-dominated
cand(2, vec![3.0, 2.0]), // non-dominated
cand(3, vec![5.0, 5.0]), // dominated by 1 and 2
cand(4, vec![2.0, 3.0]), // non-dominated
];
let front = pareto_front(&pop, &s);
let kept: Vec<u32> = front.iter().map(|c| c.decision).collect();
assert_eq!(kept, vec![1, 2, 4]);
}
#[test]
fn best_candidate_none_when_multi_objective() {
let s = space_min2();
let pop = [cand(1, vec![1.0, 1.0])];
assert!(best_candidate(&pop, &s).is_none());
}
#[test]
fn best_candidate_returns_min_for_minimize() {
let s = ObjectiveSpace::new(vec![Objective::minimize("f")]);
let pop = [cand(1, vec![3.0]), cand(2, vec![1.0]), cand(3, vec![2.0])];
let best = best_candidate(&pop, &s).unwrap();
assert_eq!(best.decision, 2);
}
#[test]
fn best_candidate_returns_max_for_maximize() {
let s = ObjectiveSpace::new(vec![Objective::maximize("score")]);
let pop = [cand(1, vec![3.0]), cand(2, vec![5.0]), cand(3, vec![2.0])];
let best = best_candidate(&pop, &s).unwrap();
assert_eq!(best.decision, 2);
}
#[test]
fn best_candidate_skips_infeasible() {
let s = ObjectiveSpace::new(vec![Objective::minimize("f")]);
let pop = [
Candidate::new(1u32, Evaluation::constrained(vec![0.0], 5.0)), // infeasible
Candidate::new(2u32, Evaluation::new(vec![10.0])),
Candidate::new(3u32, Evaluation::new(vec![3.0])),
];
let best = best_candidate(&pop, &s).unwrap();
assert_eq!(best.decision, 3);
}
#[test]
fn best_candidate_none_when_all_infeasible() {
let s = ObjectiveSpace::new(vec![Objective::minimize("f")]);
let pop = [
Candidate::new(1u32, Evaluation::constrained(vec![0.0], 1.0)),
Candidate::new(2u32, Evaluation::constrained(vec![0.0], 2.0)),
];
assert!(best_candidate(&pop, &s).is_none());
}
}
+2
View File
@@ -1,5 +1,7 @@
//! Pareto utilities: dominance, fronts, sorting, crowding, and an archive.
pub mod dominance;
pub mod front;
pub use dominance::*;
pub use front::*;
+1 -1
View File
@@ -11,4 +11,4 @@ pub use crate::core::{
pub use crate::traits::{Initializer, Optimizer, Variation};
pub use crate::pareto::{Dominance, pareto_compare};
pub use crate::pareto::{Dominance, best_candidate, pareto_compare, pareto_front};