mop-blocks 0.5.0

use crate::{gp::Gp, Obj, ObjDirection};
use cl_traits::Storage;
use core::{
  iter::Sum,
  ops::{Add, Div},
  slice::Iter,
};
use mop_common::TraitCfg;
use num_traits::One;

#[derive(Debug)]
pub struct MinWeightedSum<OI, WI> {
  objs: OI,
  weights: WI,
}

impl<OI, WI> MinWeightedSum<OI, WI> {
  pub fn new<CS, D, OR, S>(objs: OI, weights: WI) -> Self {
    Self { objs, weights }
  }
}

impl<'a, O, WI> MinWeightedSum<Iter<'a, O>, WI> {
  pub fn from_gp<D, HCRS, HCS, ORS, OS, SCRS, SCS, SS>(
    mp: &'a Gp<D, HCRS, HCS, ORS, OS, SCRS, SCS, SS>,
    weights: WI,
  ) -> Self
  where
    OS: AsRef<[O]> + Storage<Item = O>,
  {
    Self { objs: mp.defs().objs().as_ref().iter(), weights }
  }
}

impl<O, OR, OI, S, WI> Obj<OR, S> for MinWeightedSum<OI, WI>
where
  O: Obj<OR, S>,
  OR: Add<Output = OR> + Div<Output = OR> + One + Sum,
  OI: Clone + Iterator<Item = O> + TraitCfg,
  WI: Clone + Iterator<Item = OR> + TraitCfg,
{
  fn obj_direction(&self) -> ObjDirection {
    ObjDirection::Min
  }

  fn result(&self, s: &S) -> OR {
    self
      .objs
      .clone()
      .zip(self.weights.clone())
      .map(|(o, w)| {
        let result = o.result(s);
        let transformed = match o.obj_direction() {
          ObjDirection::Max => OR::one() / (OR::one() + result),
          ObjDirection::Min => result,
        };
        transformed * w
      })
      .sum()
  }
}

impl<'a, O, OR, OI, S, WI> From<&'a MinWeightedSum<OI, WI>> for &'a dyn Obj<OR, S>
where
  O: Obj<OR, S>,
  OR: Add<Output = OR> + Div<Output = OR> + One + Sum,
  OI: Clone + Iterator<Item = O> + TraitCfg,
  WI: Clone + Iterator<Item = OR> + TraitCfg,
{
  fn from(f: &'a MinWeightedSum<OI, WI>) -> Self {
    f
  }
}