solverforge_solver/

solver.rs

//! Solver implementation.

use std::fmt::Debug;
use std::marker::PhantomData;
use std::sync::atomic::AtomicBool;
use std::time::Duration;

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

use crate::phase::Phase;
use crate::scope::BestSolutionCallback;
use crate::scope::SolverScope;
use crate::stats::SolverStats;
use crate::termination::Termination;

/// Result of a solve operation containing solution and telemetry.
///
/// This is the canonical return type for `Solver::solve()`. It provides
/// both the optimized solution and comprehensive statistics about the
/// solving process.
#[derive(Debug)]
pub struct SolveResult<S: PlanningSolution> {
    /// The best solution found during solving.
    pub solution: S,
    /// Solver statistics including steps, moves evaluated, and acceptance rates.
    pub stats: SolverStats,
}

impl<S: PlanningSolution> SolveResult<S> {
    /// Returns a reference to the solution.
    pub fn solution(&self) -> &S {
        &self.solution
    }

    /// Returns the solution, consuming this result.
    pub fn into_solution(self) -> S {
        self.solution
    }

    /// Returns a reference to the solver statistics.
    pub fn stats(&self) -> &SolverStats {
        &self.stats
    }

    /// Returns the total number of steps taken across all phases.
    pub fn step_count(&self) -> u64 {
        self.stats.step_count
    }

    /// Returns the total number of moves evaluated across all phases.
    pub fn moves_evaluated(&self) -> u64 {
        self.stats.moves_evaluated
    }

    /// Returns the total number of moves accepted across all phases.
    pub fn moves_accepted(&self) -> u64 {
        self.stats.moves_accepted
    }
}

/// The main solver that optimizes planning solutions.
///
/// Uses macro-generated tuple implementations for phases, preserving
/// concrete types through the entire pipeline (zero-erasure architecture).
///
/// # Type Parameters
/// * `'t` - Lifetime of the termination flag reference
/// * `P` - Tuple of phases to execute
/// * `T` - Termination condition (use `Option<ConcreteTermination>`)
/// * `S` - Solution type
/// * `D` - Score director type
/// * `BestCb` - Best-solution callback type (default `()`)
pub struct Solver<'t, P, T, S, D, BestCb = ()> {
    phases: P,
    termination: T,
    terminate: Option<&'t AtomicBool>,
    config: Option<SolverConfig>,
    time_limit: Option<Duration>,
    /// Callback invoked when a better solution is found during solving.
    best_solution_callback: BestCb,
    _phantom: PhantomData<fn(S, D)>,
}

impl<P: Debug, T: Debug, S, D, BestCb> Debug for Solver<'_, P, T, S, D, BestCb> {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("Solver")
            .field("phases", &self.phases)
            .field("termination", &self.termination)
            .finish()
    }
}

impl<P, S, D> Solver<'static, P, NoTermination, S, D, ()>
where
    S: PlanningSolution,
{
    /// Creates a new solver with the given phases tuple and no termination.
    pub fn new(phases: P) -> Self {
        Solver {
            phases,
            termination: NoTermination,
            terminate: None,
            config: None,
            time_limit: None,
            best_solution_callback: (),
            _phantom: PhantomData,
        }
    }

    /// Sets the termination condition.
    pub fn with_termination<T>(self, termination: T) -> Solver<'static, P, Option<T>, S, D, ()> {
        Solver {
            phases: self.phases,
            termination: Some(termination),
            terminate: self.terminate,
            config: self.config,
            time_limit: self.time_limit,
            best_solution_callback: self.best_solution_callback,
            _phantom: PhantomData,
        }
    }
}

impl<'t, P, T, S, D, BestCb> Solver<'t, P, T, S, D, BestCb>
where
    S: PlanningSolution,
{
    /// Sets the external termination flag.
    ///
    /// The solver will check this flag periodically and terminate early if set.
    pub fn with_terminate(self, terminate: &'t AtomicBool) -> Solver<'t, P, T, S, D, BestCb> {
        Solver {
            phases: self.phases,
            termination: self.termination,
            terminate: Some(terminate),
            config: self.config,
            time_limit: self.time_limit,
            best_solution_callback: self.best_solution_callback,
            _phantom: PhantomData,
        }
    }

    /// Sets the time limit for solving.
    pub fn with_time_limit(mut self, limit: Duration) -> Self {
        self.time_limit = Some(limit);
        self
    }

    /// Sets configuration.
    pub fn with_config(mut self, config: SolverConfig) -> Self {
        self.config = Some(config);
        self
    }

    /// Sets a callback to be invoked when a better solution is found during solving.
    ///
    /// Transitions the `BestCb` type parameter to the concrete closure type.
    pub fn with_best_solution_callback<F: Fn(&S) + Send + Sync>(
        self,
        callback: F,
    ) -> Solver<'t, P, T, S, D, F> {
        Solver {
            phases: self.phases,
            termination: self.termination,
            terminate: self.terminate,
            config: self.config,
            time_limit: self.time_limit,
            best_solution_callback: callback,
            _phantom: PhantomData,
        }
    }

    /// Returns the configuration if set.
    pub fn config(&self) -> Option<&SolverConfig> {
        self.config.as_ref()
    }
}

/// Marker type indicating no termination.
#[derive(Debug, Clone, Copy, Default)]
pub struct NoTermination;

/// Marker trait for termination types that can be used in Solver.
pub trait MaybeTermination<
    S: PlanningSolution,
    D: ScoreDirector<S>,
    BestCb: BestSolutionCallback<S> = (),
>: Send
{
    /// Checks if the solver should terminate.
    fn should_terminate(&self, solver_scope: &SolverScope<'_, S, D, BestCb>) -> bool;

    /// Installs in-phase termination limits on the solver scope.
    ///
    /// This allows `Termination` conditions (step count, move count, etc.) to fire
    /// inside the phase step loop, not only between phases (T1 fix).
    ///
    /// The default implementation is a no-op. Override for terminations that
    /// express a concrete limit via a scope field.
    fn install_inphase_limits(&self, _solver_scope: &mut SolverScope<'_, S, D, BestCb>) {}
}

impl<S, D, BestCb, T> MaybeTermination<S, D, BestCb> for Option<T>
where
    S: PlanningSolution,
    D: ScoreDirector<S>,
    BestCb: BestSolutionCallback<S>,
    T: Termination<S, D, BestCb>,
{
    fn should_terminate(&self, solver_scope: &SolverScope<'_, S, D, BestCb>) -> bool {
        match self {
            Some(t) => t.is_terminated(solver_scope),
            None => false,
        }
    }

    fn install_inphase_limits(&self, solver_scope: &mut SolverScope<'_, S, D, BestCb>) {
        if let Some(t) = self {
            t.install_inphase_limits(solver_scope);
        }
    }
}

impl<S, D, BestCb> MaybeTermination<S, D, BestCb> for NoTermination
where
    S: PlanningSolution,
    D: ScoreDirector<S>,
    BestCb: BestSolutionCallback<S>,
{
    fn should_terminate(&self, _solver_scope: &SolverScope<'_, S, D, BestCb>) -> bool {
        false
    }

    // install_inphase_limits: no-op (default)
}

impl<S, D, BestCb> Termination<S, D, BestCb> for NoTermination
where
    S: PlanningSolution,
    D: ScoreDirector<S>,
    BestCb: BestSolutionCallback<S>,
{
    fn is_terminated(&self, _solver_scope: &SolverScope<'_, S, D, BestCb>) -> bool {
        false
    }
}

macro_rules! impl_solver {
    ($($idx:tt: $P:ident),+) => {
        impl<'t, S, D, T, BestCb, $($P),+> Solver<'t, ($($P,)+), T, S, D, BestCb>
        where
            S: PlanningSolution,
            D: ScoreDirector<S>,
            T: MaybeTermination<S, D, BestCb>,
            BestCb: BestSolutionCallback<S>,
            $($P: Phase<S, D, BestCb>,)+
        {
            /// Solves using the provided score director.
            ///
            /// Returns a `SolveResult` containing the best solution found
            /// and comprehensive solver statistics.
            pub fn solve(self, score_director: D) -> SolveResult<S> {
                let Solver {
                    mut phases,
                    termination,
                    terminate,
                    time_limit,
                    best_solution_callback,
                    ..
                } = self;

                let mut solver_scope = SolverScope::new_with_callback(
                    score_director,
                    best_solution_callback,
                    terminate,
                );
                if let Some(limit) = time_limit {
                    solver_scope.set_time_limit(limit);
                }
                solver_scope.start_solving();

                // Install in-phase termination limits so phases can check them
                // inside their step loops (T1: StepCountTermination, MoveCountTermination, etc.)
                termination.install_inphase_limits(&mut solver_scope);

                // Execute phases with termination checking
                $(
                    if !check_termination(&termination, &solver_scope) {
                        tracing::debug!(
                            "Starting phase {} ({})",
                            $idx,
                            phases.$idx.phase_type_name()
                        );
                        phases.$idx.solve(&mut solver_scope);
                        tracing::debug!(
                            "Finished phase {} ({}) with score {:?}",
                            $idx,
                            phases.$idx.phase_type_name(),
                            solver_scope.best_score()
                        );
                    }
                )+

                // Extract solution and stats before consuming scope
                let (solution, stats) = solver_scope.take_solution_and_stats();
                SolveResult { solution, stats }
            }
        }
    };
}

fn check_termination<S, D, BestCb, T>(
    termination: &T,
    solver_scope: &SolverScope<'_, S, D, BestCb>,
) -> bool
where
    S: PlanningSolution,
    D: ScoreDirector<S>,
    BestCb: BestSolutionCallback<S>,
    T: MaybeTermination<S, D, BestCb>,
{
    if solver_scope.is_terminate_early() {
        return true;
    }
    termination.should_terminate(solver_scope)
}

macro_rules! impl_solver_with_director {
    ($($idx:tt: $P:ident),+) => {
        impl<'t, S, T, BestCb, $($P),+> Solver<'t, ($($P,)+), T, S, (), BestCb>
        where
            S: PlanningSolution,
            T: Send,
            BestCb: Send + Sync,
        {
            /// Solves using a provided score director.
            pub fn solve_with_director<D>(self, director: D) -> SolveResult<S>
            where
                D: ScoreDirector<S>,
                BestCb: BestSolutionCallback<S>,
                T: MaybeTermination<S, D, BestCb>,
                $($P: Phase<S, D, BestCb>,)+
            {
                let solver: Solver<'t, ($($P,)+), T, S, D, BestCb> = Solver {
                    phases: self.phases,
                    termination: self.termination,
                    terminate: self.terminate,
                    config: self.config,
                    time_limit: self.time_limit,
                    best_solution_callback: self.best_solution_callback,
                    _phantom: PhantomData,
                };
                solver.solve(director)
            }
        }
    };
}

impl_solver_with_director!(0: P0);
impl_solver_with_director!(0: P0, 1: P1);
impl_solver_with_director!(0: P0, 1: P1, 2: P2);
impl_solver_with_director!(0: P0, 1: P1, 2: P2, 3: P3);
impl_solver_with_director!(0: P0, 1: P1, 2: P2, 3: P3, 4: P4);
impl_solver_with_director!(0: P0, 1: P1, 2: P2, 3: P3, 4: P4, 5: P5);
impl_solver_with_director!(0: P0, 1: P1, 2: P2, 3: P3, 4: P4, 5: P5, 6: P6);
impl_solver_with_director!(0: P0, 1: P1, 2: P2, 3: P3, 4: P4, 5: P5, 6: P6, 7: P7);

impl_solver!(0: P0);
impl_solver!(0: P0, 1: P1);
impl_solver!(0: P0, 1: P1, 2: P2);
impl_solver!(0: P0, 1: P1, 2: P2, 3: P3);
impl_solver!(0: P0, 1: P1, 2: P2, 3: P3, 4: P4);
impl_solver!(0: P0, 1: P1, 2: P2, 3: P3, 4: P4, 5: P5);
impl_solver!(0: P0, 1: P1, 2: P2, 3: P3, 4: P4, 5: P5, 6: P6);
impl_solver!(0: P0, 1: P1, 2: P2, 3: P3, 4: P4, 5: P5, 6: P6, 7: P7);