solverforge_solver/heuristic/selector/

nearby.rs

//! Nearby selection for distance-based filtering of candidates.
//!
//! Nearby selection improves move quality by preferring destinations that are
//! geographically or otherwise "close" to an origin. This is critical for
//! vehicle routing problems (VRP) where swapping with nearby customers
//! is more likely to improve the solution than swapping with distant ones.
//!
//! # Architecture
//!
//! - [`NearbyDistanceMeter`]: User-defined function to measure distance between elements
//! - [`NearbySelectionConfig`]: Configuration for nearby selection behavior
//! - [`NearbyEntitySelector`]: Selects entities nearby to a reference entity

use std::cmp::Ordering;
use std::fmt::Debug;

use solverforge_core::domain::PlanningSolution;
use solverforge_scoring::ScoreDirector;

use super::entity::{EntityReference, EntitySelector};
use super::mimic::MimicRecorder;

/// Trait for measuring distance between an origin and a destination.
///
/// Implementations should be stateless. The solver may reuse instances.
///
/// # Type Parameters
///
/// - `Origin`: The type of the origin element (usually an entity or value)
/// - `Destination`: The type of the destination element (usually an entity or value)
///
/// # Example
///
/// ```
/// use solverforge_solver::heuristic::selector::NearbyDistanceMeter;
///
/// #[derive(Debug)]
/// struct Location { x: f64, y: f64 }
///
/// #[derive(Debug)]
/// struct EuclideanMeter;
///
/// impl NearbyDistanceMeter<Location, Location> for EuclideanMeter {
///     fn distance(&self, origin: &Location, dest: &Location) -> f64 {
///         let dx = origin.x - dest.x;
///         let dy = origin.y - dest.y;
///         (dx * dx + dy * dy).sqrt()
///     }
/// }
/// ```
pub trait NearbyDistanceMeter<Origin, Destination>: Send + Sync + Debug {
    /// Measures the distance from the origin to the destination.
    ///
    /// The distance can be in any unit (meters, seconds, cost, etc.).
    /// Distances can be asymmetrical: the distance from A to B may differ
    /// from the distance from B to A.
    ///
    /// Returns a value >= 0.0. If origin == destination, typically returns 0.0.
    fn distance(&self, origin: &Origin, destination: &Destination) -> f64;
}

/// Distribution type for nearby selection.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Default)]
pub enum NearbyDistributionType {
    /// Select all candidates sorted by distance (up to a maximum).
    #[default]
    Linear,
    /// Select candidates with probability proportional to distance (closer = more likely).
    Parabolic,
    /// Use a block distribution for k-opt style moves.
    Block,
}

/// Configuration for nearby selection.
#[derive(Debug, Clone)]
pub struct NearbySelectionConfig {
    /// The distribution type to use.
    pub distribution_type: NearbyDistributionType,
    /// Maximum number of nearby candidates to consider.
    /// If None, considers all candidates (sorted by distance).
    pub max_nearby_size: Option<usize>,
    /// Minimum distance to include a candidate (exclusive of origin).
    pub min_distance: f64,
}

impl Default for NearbySelectionConfig {
    fn default() -> Self {
        Self {
            distribution_type: NearbyDistributionType::Linear,
            max_nearby_size: None,
            min_distance: 0.0,
        }
    }
}

impl NearbySelectionConfig {
    /// Creates a new configuration with default values.
    pub fn new() -> Self {
        Self::default()
    }

    /// Sets the distribution type.
    pub fn with_distribution_type(mut self, distribution_type: NearbyDistributionType) -> Self {
        self.distribution_type = distribution_type;
        self
    }

    /// Sets the maximum number of nearby candidates.
    pub fn with_max_nearby_size(mut self, max_size: usize) -> Self {
        self.max_nearby_size = Some(max_size);
        self
    }

    /// Sets the minimum distance threshold.
    pub fn with_min_distance(mut self, min_distance: f64) -> Self {
        self.min_distance = min_distance;
        self
    }
}

/// Type-erased distance meter for dynamic dispatch.
pub trait DynDistanceMeter: Send + Sync + Debug {
    /// Measures distance between two entity references.
    fn distance_between<S: PlanningSolution>(
        &self,
        score_director: &dyn ScoreDirector<S>,
        origin: EntityReference,
        destination: EntityReference,
    ) -> f64;
}

/// An entity selector that returns entities nearby to an origin entity.
///
/// The origin entity is obtained from a mimic recorder, allowing this selector
/// to be synchronized with another selector that picks the "current" entity.
pub struct NearbyEntitySelector<S: PlanningSolution, M: DynDistanceMeter> {
    /// The child selector providing all candidate entities.
    child: Box<dyn EntitySelector<S>>,
    /// The recorder providing the origin entity.
    origin_recorder: MimicRecorder,
    /// The distance meter for measuring nearness.
    distance_meter: M,
    /// Configuration for nearby selection.
    config: NearbySelectionConfig,
}

impl<S: PlanningSolution, M: DynDistanceMeter> NearbyEntitySelector<S, M> {
    /// Creates a new nearby entity selector.
    pub fn new(
        child: Box<dyn EntitySelector<S>>,
        origin_recorder: MimicRecorder,
        distance_meter: M,
        config: NearbySelectionConfig,
    ) -> Self {
        Self {
            child,
            origin_recorder,
            distance_meter,
            config,
        }
    }
}

impl<S: PlanningSolution, M: DynDistanceMeter> Debug for NearbyEntitySelector<S, M> {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("NearbyEntitySelector")
            .field("child", &self.child)
            .field("origin_recorder_id", &self.origin_recorder.id())
            .field("distance_meter", &self.distance_meter)
            .field("config", &self.config)
            .finish()
    }
}

impl<S: PlanningSolution, M: DynDistanceMeter + 'static> EntitySelector<S>
    for NearbyEntitySelector<S, M>
{
    fn iter<'a>(
        &'a self,
        score_director: &'a dyn ScoreDirector<S>,
    ) -> Box<dyn Iterator<Item = EntityReference> + 'a> {
        // Get the origin entity from the recorder
        let origin = match self.origin_recorder.get_recorded_entity() {
            Some(e) => e,
            None => {
                // No origin recorded yet, return empty
                return Box::new(std::iter::empty());
            }
        };

        // Collect all candidate entities with their distances
        let mut candidates: Vec<(EntityReference, f64)> = self
            .child
            .iter(score_director)
            .filter(|&dest| dest != origin) // Exclude the origin itself
            .map(|dest| {
                let dist = self
                    .distance_meter
                    .distance_between(score_director, origin, dest);
                (dest, dist)
            })
            .filter(|(_, dist)| *dist >= self.config.min_distance)
            .collect();

        // Sort by distance (closest first)
        candidates.sort_by(|a, b| a.1.partial_cmp(&b.1).unwrap_or(Ordering::Equal));

        // Apply max size limit
        if let Some(max_size) = self.config.max_nearby_size {
            candidates.truncate(max_size);
        }

        Box::new(candidates.into_iter().map(|(entity, _)| entity))
    }

    fn size(&self, score_director: &dyn ScoreDirector<S>) -> usize {
        // This is an estimate; the actual size depends on the origin
        let child_size = self.child.size(score_director);
        match self.config.max_nearby_size {
            Some(max) => child_size.min(max),
            None => child_size,
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::heuristic::selector::entity::FromSolutionEntitySelector;
    use crate::heuristic::selector::mimic::{MimicRecorder, MimicRecordingEntitySelector};
    use solverforge_core::domain::{EntityDescriptor, SolutionDescriptor, TypedEntityExtractor};
    use solverforge_core::score::SimpleScore;
    use solverforge_scoring::SimpleScoreDirector;
    use std::any::TypeId;

    #[allow(dead_code)]
    #[derive(Clone, Debug)]
    struct Location {
        id: i64,
        x: f64,
        y: f64,
        assigned_to: Option<i64>,
    }

    #[derive(Clone, Debug)]
    struct RoutingSolution {
        locations: Vec<Location>,
        score: Option<SimpleScore>,
    }

    impl PlanningSolution for RoutingSolution {
        type Score = SimpleScore;

        fn score(&self) -> Option<Self::Score> {
            self.score
        }

        fn set_score(&mut self, score: Option<Self::Score>) {
            self.score = score;
        }
    }

    fn get_locations(s: &RoutingSolution) -> &Vec<Location> {
        &s.locations
    }

    fn get_locations_mut(s: &mut RoutingSolution) -> &mut Vec<Location> {
        &mut s.locations
    }

    /// Distance meter that uses Euclidean distance.
    #[derive(Debug)]
    struct EuclideanDistanceMeter {
        /// Cached locations for quick lookup.
        locations: Vec<(f64, f64)>,
    }

    impl EuclideanDistanceMeter {
        fn new(locations: &[Location]) -> Self {
            Self {
                locations: locations.iter().map(|l| (l.x, l.y)).collect(),
            }
        }
    }

    impl DynDistanceMeter for EuclideanDistanceMeter {
        fn distance_between<S: PlanningSolution>(
            &self,
            _score_director: &dyn ScoreDirector<S>,
            origin: EntityReference,
            destination: EntityReference,
        ) -> f64 {
            let (ox, oy) = self.locations[origin.entity_index];
            let (dx, dy) = self.locations[destination.entity_index];
            let delta_x = ox - dx;
            let delta_y = oy - dy;
            (delta_x * delta_x + delta_y * delta_y).sqrt()
        }
    }

    fn create_test_director(
    ) -> SimpleScoreDirector<RoutingSolution, impl Fn(&RoutingSolution) -> SimpleScore> {
        // Create a grid of locations: (0,0), (1,0), (2,0), (0,1), (1,1), (2,1)
        let locations = vec![
            Location {
                id: 0,
                x: 0.0,
                y: 0.0,
                assigned_to: None,
            },
            Location {
                id: 1,
                x: 1.0,
                y: 0.0,
                assigned_to: None,
            },
            Location {
                id: 2,
                x: 2.0,
                y: 0.0,
                assigned_to: None,
            },
            Location {
                id: 3,
                x: 0.0,
                y: 1.0,
                assigned_to: None,
            },
            Location {
                id: 4,
                x: 1.0,
                y: 1.0,
                assigned_to: None,
            },
            Location {
                id: 5,
                x: 2.0,
                y: 1.0,
                assigned_to: None,
            },
        ];

        let solution = RoutingSolution {
            locations,
            score: None,
        };

        let extractor = Box::new(TypedEntityExtractor::new(
            "Location",
            "locations",
            get_locations,
            get_locations_mut,
        ));
        let entity_desc = EntityDescriptor::new("Location", TypeId::of::<Location>(), "locations")
            .with_extractor(extractor);

        let descriptor =
            SolutionDescriptor::new("RoutingSolution", TypeId::of::<RoutingSolution>())
                .with_entity(entity_desc);

        SimpleScoreDirector::with_calculator(solution, descriptor, |_| SimpleScore::of(0))
    }

    #[test]
    fn test_nearby_selector_sorts_by_distance() {
        let director = create_test_director();

        // Verify entity IDs match indices
        let solution = director.working_solution();
        for (i, loc) in solution.locations.iter().enumerate() {
            assert_eq!(loc.id, i as i64);
        }

        // Create mimic recorder and recording selector for origin
        let recorder = MimicRecorder::new("origin");
        let origin_child = Box::new(FromSolutionEntitySelector::new(0));
        let origin_selector = MimicRecordingEntitySelector::new(origin_child, recorder.clone());

        // Create nearby selector for destinations
        let dest_child = Box::new(FromSolutionEntitySelector::new(0));
        let distance_meter = EuclideanDistanceMeter::new(&director.working_solution().locations);
        let nearby_config = NearbySelectionConfig::default();
        let nearby_selector =
            NearbyEntitySelector::new(dest_child, recorder.clone(), distance_meter, nearby_config);

        // Select origin entity (location 0 at 0,0)
        let mut origin_iter = origin_selector.iter(&director);
        let _origin = origin_iter.next().unwrap();

        // Get nearby entities (should be sorted by distance from 0,0)
        let nearby: Vec<_> = nearby_selector.iter(&director).collect();

        // Expected order: 1 (dist 1), 3 (dist 1), 2 (dist 2), 4 (dist √2 ≈ 1.41), 5 (dist √5 ≈ 2.24)
        // Actually: 1 at (1,0) dist 1, 3 at (0,1) dist 1, 4 at (1,1) dist √2, 2 at (2,0) dist 2, 5 at (2,1) dist √5
        assert_eq!(nearby.len(), 5); // 6 locations - 1 (origin) = 5

        // First two should be at distance 1 (locations 1 and 3)
        assert!(
            nearby[0].entity_index == 1 || nearby[0].entity_index == 3,
            "Expected location 1 or 3, got {}",
            nearby[0].entity_index
        );
    }

    #[test]
    fn test_nearby_selector_with_max_size() {
        let director = create_test_director();

        let recorder = MimicRecorder::new("origin");
        let origin_child = Box::new(FromSolutionEntitySelector::new(0));
        let origin_selector = MimicRecordingEntitySelector::new(origin_child, recorder.clone());

        let dest_child = Box::new(FromSolutionEntitySelector::new(0));
        let distance_meter = EuclideanDistanceMeter::new(&director.working_solution().locations);
        let nearby_config = NearbySelectionConfig::default().with_max_nearby_size(2);
        let nearby_selector =
            NearbyEntitySelector::new(dest_child, recorder.clone(), distance_meter, nearby_config);

        // Select origin
        let mut origin_iter = origin_selector.iter(&director);
        origin_iter.next();

        // Should only get 2 nearest
        let nearby: Vec<_> = nearby_selector.iter(&director).collect();
        assert_eq!(nearby.len(), 2);
    }

    #[test]
    fn test_nearby_selector_excludes_origin() {
        let director = create_test_director();

        let recorder = MimicRecorder::new("origin");
        let origin_child = Box::new(FromSolutionEntitySelector::new(0));
        let origin_selector = MimicRecordingEntitySelector::new(origin_child, recorder.clone());

        let dest_child = Box::new(FromSolutionEntitySelector::new(0));
        let distance_meter = EuclideanDistanceMeter::new(&director.working_solution().locations);
        let nearby_config = NearbySelectionConfig::default();
        let nearby_selector =
            NearbyEntitySelector::new(dest_child, recorder.clone(), distance_meter, nearby_config);

        // Select origin (entity 0)
        let mut origin_iter = origin_selector.iter(&director);
        let origin = origin_iter.next().unwrap();

        // Nearby should not include the origin
        let nearby: Vec<_> = nearby_selector.iter(&director).collect();
        assert!(!nearby.contains(&origin));
    }
}