use std::fmt::Debug;
use solverforge_core::domain::PlanningSolution;
use solverforge_scoring::Director;
use super::{
ChangeMove, CompoundScalarMove, ConflictRepairMove, Move, MoveTabuSignature, PillarChangeMove,
PillarSwapMove, RuinRecreateMove, SwapMove,
};
#[allow(clippy::large_enum_variant)]
pub enum ScalarMoveUnion<S, V> {
Change(ChangeMove<S, V>),
Swap(SwapMove<S, V>),
PillarChange(PillarChangeMove<S, V>),
PillarSwap(PillarSwapMove<S, V>),
RuinRecreate(RuinRecreateMove<S>),
CompoundScalar(CompoundScalarMove<S>),
ConflictRepair(ConflictRepairMove<S>),
}
pub enum ScalarMoveUnionUndo<S, V>
where
S: PlanningSolution,
V: Clone + PartialEq + Send + Sync + Debug + 'static,
{
Change(<ChangeMove<S, V> as Move<S>>::Undo),
Swap(<SwapMove<S, V> as Move<S>>::Undo),
PillarChange(<PillarChangeMove<S, V> as Move<S>>::Undo),
PillarSwap(<PillarSwapMove<S, V> as Move<S>>::Undo),
RuinRecreate(<RuinRecreateMove<S> as Move<S>>::Undo),
CompoundScalar(<CompoundScalarMove<S> as Move<S>>::Undo),
ConflictRepair(<ConflictRepairMove<S> as Move<S>>::Undo),
}
impl<S, V> Clone for ScalarMoveUnion<S, V>
where
S: PlanningSolution,
V: Clone + PartialEq + Send + Sync + Debug + 'static,
{
fn clone(&self) -> Self {
match self {
Self::Change(m) => Self::Change(m.clone()),
Self::Swap(m) => Self::Swap(*m),
Self::PillarChange(m) => Self::PillarChange(m.clone()),
Self::PillarSwap(m) => Self::PillarSwap(m.clone()),
Self::RuinRecreate(m) => Self::RuinRecreate(m.clone()),
Self::CompoundScalar(m) => Self::CompoundScalar(m.clone()),
Self::ConflictRepair(m) => Self::ConflictRepair(m.clone()),
}
}
}
impl<S, V> Debug for ScalarMoveUnion<S, V>
where
S: PlanningSolution,
V: Clone + PartialEq + Send + Sync + Debug + 'static,
{
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
match self {
Self::Change(m) => m.fmt(f),
Self::Swap(m) => m.fmt(f),
Self::PillarChange(m) => m.fmt(f),
Self::PillarSwap(m) => m.fmt(f),
Self::RuinRecreate(m) => m.fmt(f),
Self::CompoundScalar(m) => m.fmt(f),
Self::ConflictRepair(m) => m.fmt(f),
}
}
}
impl<S, V> Move<S> for ScalarMoveUnion<S, V>
where
S: PlanningSolution,
V: Clone + PartialEq + Send + Sync + Debug + 'static,
{
type Undo = ScalarMoveUnionUndo<S, V>;
fn is_doable<D: Director<S>>(&self, score_director: &D) -> bool {
match self {
Self::Change(m) => m.is_doable(score_director),
Self::Swap(m) => m.is_doable(score_director),
Self::PillarChange(m) => m.is_doable(score_director),
Self::PillarSwap(m) => m.is_doable(score_director),
Self::RuinRecreate(m) => m.is_doable(score_director),
Self::CompoundScalar(m) => m.is_doable(score_director),
Self::ConflictRepair(m) => m.is_doable(score_director),
}
}
fn do_move<D: Director<S>>(&self, score_director: &mut D) -> Self::Undo {
match self {
Self::Change(m) => ScalarMoveUnionUndo::Change(m.do_move(score_director)),
Self::Swap(m) => ScalarMoveUnionUndo::Swap(m.do_move(score_director)),
Self::PillarChange(m) => ScalarMoveUnionUndo::PillarChange(m.do_move(score_director)),
Self::PillarSwap(m) => ScalarMoveUnionUndo::PillarSwap(m.do_move(score_director)),
Self::RuinRecreate(m) => ScalarMoveUnionUndo::RuinRecreate(m.do_move(score_director)),
Self::CompoundScalar(m) => {
ScalarMoveUnionUndo::CompoundScalar(m.do_move(score_director))
}
Self::ConflictRepair(m) => {
ScalarMoveUnionUndo::ConflictRepair(m.do_move(score_director))
}
}
}
fn undo_move<D: Director<S>>(&self, score_director: &mut D, undo: Self::Undo) {
match (self, undo) {
(Self::Change(m), ScalarMoveUnionUndo::Change(undo)) => {
m.undo_move(score_director, undo)
}
(Self::Swap(m), ScalarMoveUnionUndo::Swap(undo)) => m.undo_move(score_director, undo),
(Self::PillarChange(m), ScalarMoveUnionUndo::PillarChange(undo)) => {
m.undo_move(score_director, undo)
}
(Self::PillarSwap(m), ScalarMoveUnionUndo::PillarSwap(undo)) => {
m.undo_move(score_director, undo)
}
(Self::RuinRecreate(m), ScalarMoveUnionUndo::RuinRecreate(undo)) => {
m.undo_move(score_director, undo)
}
(Self::CompoundScalar(m), ScalarMoveUnionUndo::CompoundScalar(undo)) => {
m.undo_move(score_director, undo)
}
(Self::ConflictRepair(m), ScalarMoveUnionUndo::ConflictRepair(undo)) => {
m.undo_move(score_director, undo)
}
_ => panic!("scalar move undo shape must match move shape"),
}
}
fn descriptor_index(&self) -> usize {
match self {
Self::Change(m) => m.descriptor_index(),
Self::Swap(m) => m.descriptor_index(),
Self::PillarChange(m) => m.descriptor_index(),
Self::PillarSwap(m) => m.descriptor_index(),
Self::RuinRecreate(m) => m.descriptor_index(),
Self::CompoundScalar(m) => m.descriptor_index(),
Self::ConflictRepair(m) => m.descriptor_index(),
}
}
fn entity_indices(&self) -> &[usize] {
match self {
Self::Change(m) => m.entity_indices(),
Self::Swap(m) => m.entity_indices(),
Self::PillarChange(m) => m.entity_indices(),
Self::PillarSwap(m) => m.entity_indices(),
Self::RuinRecreate(m) => m.entity_indices(),
Self::CompoundScalar(m) => m.entity_indices(),
Self::ConflictRepair(m) => m.entity_indices(),
}
}
fn variable_name(&self) -> &str {
match self {
Self::Change(m) => m.variable_name(),
Self::Swap(m) => m.variable_name(),
Self::PillarChange(m) => m.variable_name(),
Self::PillarSwap(m) => m.variable_name(),
Self::RuinRecreate(m) => m.variable_name(),
Self::CompoundScalar(m) => m.variable_name(),
Self::ConflictRepair(m) => m.variable_name(),
}
}
fn telemetry_label(&self) -> &'static str {
match self {
Self::Change(m) => m.telemetry_label(),
Self::Swap(m) => m.telemetry_label(),
Self::PillarChange(m) => m.telemetry_label(),
Self::PillarSwap(m) => m.telemetry_label(),
Self::RuinRecreate(m) => m.telemetry_label(),
Self::CompoundScalar(m) => m.telemetry_label(),
Self::ConflictRepair(m) => m.telemetry_label(),
}
}
fn requires_hard_improvement(&self) -> bool {
match self {
Self::Change(m) => m.requires_hard_improvement(),
Self::Swap(m) => m.requires_hard_improvement(),
Self::PillarChange(m) => m.requires_hard_improvement(),
Self::PillarSwap(m) => m.requires_hard_improvement(),
Self::RuinRecreate(m) => m.requires_hard_improvement(),
Self::CompoundScalar(m) => m.requires_hard_improvement(),
Self::ConflictRepair(m) => m.requires_hard_improvement(),
}
}
fn tabu_signature<D: Director<S>>(&self, score_director: &D) -> MoveTabuSignature {
match self {
Self::Change(m) => m.tabu_signature(score_director),
Self::Swap(m) => m.tabu_signature(score_director),
Self::PillarChange(m) => m.tabu_signature(score_director),
Self::PillarSwap(m) => m.tabu_signature(score_director),
Self::RuinRecreate(m) => m.tabu_signature(score_director),
Self::CompoundScalar(m) => m.tabu_signature(score_director),
Self::ConflictRepair(m) => m.tabu_signature(score_director),
}
}
fn for_each_affected_entity(&self, visitor: &mut dyn FnMut(super::MoveAffectedEntity<'_>)) {
match self {
Self::Change(m) => m.for_each_affected_entity(visitor),
Self::Swap(m) => m.for_each_affected_entity(visitor),
Self::PillarChange(m) => m.for_each_affected_entity(visitor),
Self::PillarSwap(m) => m.for_each_affected_entity(visitor),
Self::RuinRecreate(m) => m.for_each_affected_entity(visitor),
Self::CompoundScalar(m) => m.for_each_affected_entity(visitor),
Self::ConflictRepair(m) => m.for_each_affected_entity(visitor),
}
}
}