use rustsat::{
encodings::{card, pb},
instances::{Cnf, ManageVars, SatInstance},
types::{
constraints::{CardConstraint, PbConstraint},
Clause,
},
};
use crate::PreproClauses;
pub trait PreproSat: PreproClauses {
fn new_with_encoders<VM, CardEnc, PBEnc>(
inst: SatInstance<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 (cnf, _) = inst.into_cnf_with_encoders(card_encoder, pb_encoder);
<Self as PreproClauses>::new::<Vec<(Clause, usize)>>(cnf, vec![], inprocessing)
}
fn new<VM>(inst: SatInstance<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) -> SatInstance {
let (cnf, objs) = <Self as PreproClauses>::prepro_instance(self);
debug_assert!(objs.is_empty());
SatInstance::from(cnf)
}
}
impl<PP: PreproClauses> PreproSat for PP {}