use std::fmt::Debug;
use std::marker::PhantomData;
use solverforge_core::domain::PlanningSolution;
use solverforge_scoring::Director;
use crate::heuristic::r#move::{ListMoveImpl, ListReverseMove};
use super::entity::EntitySelector;
use super::move_selector::MoveSelector;
pub struct ListReverseMoveSelector<S, V, ES> {
entity_selector: ES,
list_len: fn(&S, usize) -> usize,
list_reverse: fn(&mut S, usize, usize, usize),
variable_name: &'static str,
descriptor_index: usize,
_phantom: PhantomData<(fn() -> S, fn() -> V)>,
}
impl<S, V: Debug, ES: Debug> Debug for ListReverseMoveSelector<S, V, ES> {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
f.debug_struct("ListReverseMoveSelector")
.field("entity_selector", &self.entity_selector)
.field("variable_name", &self.variable_name)
.field("descriptor_index", &self.descriptor_index)
.finish()
}
}
impl<S, V, ES> ListReverseMoveSelector<S, V, ES> {
pub fn new(
entity_selector: ES,
list_len: fn(&S, usize) -> usize,
list_reverse: fn(&mut S, usize, usize, usize),
variable_name: &'static str,
descriptor_index: usize,
) -> Self {
Self {
entity_selector,
list_len,
list_reverse,
variable_name,
descriptor_index,
_phantom: PhantomData,
}
}
}
impl<S, V, ES> MoveSelector<S, ListReverseMove<S, V>> for ListReverseMoveSelector<S, V, ES>
where
S: PlanningSolution,
V: Clone + Send + Sync + Debug + 'static,
ES: EntitySelector<S>,
{
fn iter_moves<'a, D: Director<S>>(
&'a self,
score_director: &'a D,
) -> impl Iterator<Item = ListReverseMove<S, V>> + 'a {
let solution = score_director.working_solution();
let list_len = self.list_len;
let list_reverse = self.list_reverse;
let variable_name = self.variable_name;
let descriptor_index = self.descriptor_index;
let entities: Vec<usize> = self
.entity_selector
.iter(score_director)
.map(|r| r.entity_index)
.collect();
let mut moves = Vec::new();
for &entity in &entities {
let len = list_len(solution, entity);
if len < 2 {
continue;
}
for start in 0..len {
for end in (start + 2)..=len {
moves.push(ListReverseMove::new(
entity,
start,
end,
list_len,
list_reverse,
variable_name,
descriptor_index,
));
}
}
}
moves.into_iter()
}
fn size<D: Director<S>>(&self, score_director: &D) -> usize {
let solution = score_director.working_solution();
let list_len = self.list_len;
self.entity_selector
.iter(score_director)
.map(|r| {
let m = list_len(solution, r.entity_index);
if m >= 2 {
m * (m - 1) / 2
} else {
0
}
})
.sum()
}
}
pub struct ListMoveListReverseSelector<S, V, ES> {
inner: ListReverseMoveSelector<S, V, ES>,
}
impl<S, V: Debug, ES: Debug> Debug for ListMoveListReverseSelector<S, V, ES> {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
f.debug_struct("ListMoveListReverseSelector")
.field("inner", &self.inner)
.finish()
}
}
impl<S, V, ES> ListMoveListReverseSelector<S, V, ES> {
pub fn new(inner: ListReverseMoveSelector<S, V, ES>) -> Self {
Self { inner }
}
}
impl<S, V, ES> MoveSelector<S, ListMoveImpl<S, V>> for ListMoveListReverseSelector<S, V, ES>
where
S: PlanningSolution,
V: Clone + PartialEq + Send + Sync + Debug + 'static,
ES: EntitySelector<S>,
{
fn iter_moves<'a, D: Director<S>>(
&'a self,
score_director: &'a D,
) -> impl Iterator<Item = ListMoveImpl<S, V>> + 'a {
self.inner
.iter_moves(score_director)
.map(ListMoveImpl::ListReverse)
}
fn size<D: Director<S>>(&self, score_director: &D) -> usize {
self.inner.size(score_director)
}
}