use rustsat::{
encodings::{card, pb},
instances::{Cnf, ManageVars, Objective, OptInstance, SatInstance},
types::constraints::{CardConstraint, PbConstraint},
};
use crate::PreproClauses;
pub trait PreproOpt: PreproClauses {
fn new_with_encoders<VM, CardEnc, PBEnc>(
inst: OptInstance<VM>,
card_encoder: CardEnc,
pb_encoder: PBEnc,
inprocessing: bool,
) -> Self
where
VM: ManageVars,
CardEnc: FnMut(CardConstraint, &mut Cnf, &mut dyn ManageVars),
PBEnc: FnMut(PbConstraint, &mut Cnf, &mut dyn ManageVars),
Self: Sized,
{
let (constrs, obj) = inst.decompose();
let (cnf, _) = constrs.into_cnf_with_encoders(card_encoder, pb_encoder);
let softs = obj.into_soft_cls();
<Self as PreproClauses>::new(cnf, vec![softs], inprocessing)
}
fn new<VM>(inst: OptInstance<VM>, inprocessing: bool) -> Self
where
VM: ManageVars,
Self: Sized,
{
Self::new_with_encoders(
inst,
|constr, cnf, vm| {
card::default_encode_cardinality_constraint(constr, cnf, vm)
.expect("cardinality encoding ran out of memory")
},
|constr, cnf, vm| {
pb::default_encode_pb_constraint(constr, cnf, vm)
.expect("pb encoding ran out of memory")
},
inprocessing,
)
}
fn prepro_instance(&mut self) -> OptInstance {
let (cnf, objs) = <Self as PreproClauses>::prepro_instance(self);
debug_assert_eq!(objs.len(), 1);
let constrs = SatInstance::from(cnf);
let obj = if let Some((softs, offset)) = objs.into_iter().last() {
let mut obj = Objective::from_iter(softs);
obj.set_offset(offset);
obj
} else {
panic!()
};
OptInstance::compose(constrs, obj)
}
}
impl<PP: PreproClauses> PreproOpt for PP {}