use std::fmt::Debug;
use std::marker::PhantomData;
use solverforge_core::domain::PlanningSolution;
use solverforge_scoring::Director;
use super::k_opt_reconnection::KOptReconnection;
use super::list_kernel::{
k_opt_do_move, k_opt_is_doable, k_opt_tabu_signature, k_opt_undo_move, StaticListWindowAccess,
};
use super::metadata::hash_str;
use super::{Move, MoveTabuSignature};
#[derive(Clone, Copy, Debug, Default, PartialEq, Eq)]
pub struct CutPoint {
entity_index: usize,
position: usize,
}
impl CutPoint {
#[inline]
pub const fn new(entity_index: usize, position: usize) -> Self {
Self {
entity_index,
position,
}
}
#[inline]
pub const fn entity_index(&self) -> usize {
self.entity_index
}
#[inline]
pub const fn position(&self) -> usize {
self.position
}
}
pub struct KOptMove<S, V> {
cuts: [CutPoint; 5],
cut_count: u8,
reconnection: KOptReconnection,
list_len: fn(&S, usize) -> usize,
list_get: fn(&S, usize, usize) -> Option<V>,
sublist_remove: fn(&mut S, usize, usize, usize) -> Vec<V>,
sublist_insert: fn(&mut S, usize, usize, Vec<V>),
variable_name: &'static str,
variable_id: u64,
descriptor_index: usize,
entity_index: usize,
_phantom: PhantomData<fn() -> V>,
}
impl<S, V> Clone for KOptMove<S, V> {
fn clone(&self) -> Self {
Self {
cuts: self.cuts,
cut_count: self.cut_count,
reconnection: self.reconnection,
list_len: self.list_len,
list_get: self.list_get,
sublist_remove: self.sublist_remove,
sublist_insert: self.sublist_insert,
variable_name: self.variable_name,
variable_id: self.variable_id,
descriptor_index: self.descriptor_index,
entity_index: self.entity_index,
_phantom: PhantomData,
}
}
}
impl<S, V: Debug> Debug for KOptMove<S, V> {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
let cuts: Vec<_> = self.cuts[..self.cut_count as usize]
.iter()
.map(|c| c.position)
.collect();
f.debug_struct("KOptMove")
.field("k", &self.cut_count)
.field("entity", &self.entity_index)
.field("cuts", &cuts)
.field("reconnection", &self.reconnection)
.field("variable_name", &self.variable_name)
.finish()
}
}
impl<S, V> KOptMove<S, V> {
#[allow(clippy::too_many_arguments)]
pub fn new(
cuts: &[CutPoint],
reconnection: &KOptReconnection,
list_len: fn(&S, usize) -> usize,
list_get: fn(&S, usize, usize) -> Option<V>,
sublist_remove: fn(&mut S, usize, usize, usize) -> Vec<V>,
sublist_insert: fn(&mut S, usize, usize, Vec<V>),
variable_name: &'static str,
descriptor_index: usize,
) -> Self {
Self::new_with_variable_id(
cuts,
reconnection,
list_len,
list_get,
sublist_remove,
sublist_insert,
variable_name,
hash_str(variable_name),
descriptor_index,
)
}
#[allow(clippy::too_many_arguments)]
pub(crate) fn new_with_variable_id(
cuts: &[CutPoint],
reconnection: &KOptReconnection,
list_len: fn(&S, usize) -> usize,
list_get: fn(&S, usize, usize) -> Option<V>,
sublist_remove: fn(&mut S, usize, usize, usize) -> Vec<V>,
sublist_insert: fn(&mut S, usize, usize, Vec<V>),
variable_name: &'static str,
variable_id: u64,
descriptor_index: usize,
) -> Self {
assert!(!cuts.is_empty() && cuts.len() <= 5, "k must be 1-5");
let mut cut_array = [CutPoint::default(); 5];
for (i, cut) in cuts.iter().enumerate() {
cut_array[i] = *cut;
}
let entity_index = cuts[0].entity_index;
Self {
cuts: cut_array,
cut_count: cuts.len() as u8,
reconnection: *reconnection,
list_len,
list_get,
sublist_remove,
sublist_insert,
variable_name,
variable_id,
descriptor_index,
entity_index,
_phantom: PhantomData,
}
}
#[inline]
pub fn k(&self) -> usize {
self.cut_count as usize
}
#[inline]
pub fn cuts(&self) -> &[CutPoint] {
&self.cuts[..self.cut_count as usize]
}
pub fn is_intra_route(&self) -> bool {
let first = self.cuts[0].entity_index;
self.cuts[..self.cut_count as usize]
.iter()
.all(|c| c.entity_index == first)
}
fn access(&self) -> StaticListWindowAccess<S, V> {
StaticListWindowAccess {
list_len: self.list_len,
list_get: self.list_get,
sublist_remove: self.sublist_remove,
sublist_insert: self.sublist_insert,
variable_name: self.variable_name,
descriptor_index: self.descriptor_index,
}
}
}
impl<S, V> Move<S> for KOptMove<S, V>
where
S: PlanningSolution,
V: Clone + Send + Sync + Debug + 'static,
{
type Undo = Vec<V>;
fn is_doable<D: Director<S>>(&self, score_director: &D) -> bool {
k_opt_is_doable(
&self.access(),
self.cuts(),
&self.reconnection,
self.entity_index,
score_director,
)
}
fn do_move<D: Director<S>>(&self, score_director: &mut D) -> Self::Undo {
k_opt_do_move(
&self.access(),
self.cuts(),
&self.reconnection,
self.entity_index,
score_director,
)
}
fn undo_move<D: Director<S>>(&self, score_director: &mut D, undo: Self::Undo) {
k_opt_undo_move(&self.access(), self.entity_index, undo, score_director);
}
fn descriptor_index(&self) -> usize {
self.descriptor_index
}
fn entity_indices(&self) -> &[usize] {
std::slice::from_ref(&self.entity_index)
}
fn variable_name(&self) -> &str {
self.variable_name
}
fn telemetry_label(&self) -> &'static str {
"k_opt"
}
fn tabu_signature<D: Director<S>>(&self, score_director: &D) -> MoveTabuSignature {
k_opt_tabu_signature(
&self.access(),
self.cuts(),
&self.reconnection,
self.variable_id,
self.entity_index,
score_director,
)
}
}