Saklad5/Bone-Market-Solver
3
0
Fork 0
Code Issues Pull Requests Activity

Override CpModel.AddDivisionEquality

Browse Source 
The overridden method supports enforcement literals using indirection.
This commit is contained in:
Jeremy Saklad 2021-09-28 16:34:50 -05:00
parent 06dd19bb45
commit 46edfcb35b
Signed by: Jeremy Saklad
GPG Key ID: 9CA2149583EDBF84
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
bonemarketsolver/objects/bone_market_model.py
View File

@ -22,6 +22,17 @@ Set `upto` to a value that is unlikely to come into play.
Each parameter is interpreted as a BoundedLinearExpression, and a layer of indirection is applied such that each Constraint in the returned tuple can accept an enforcement literal."""
return self.AddAllowedAssignments((target, var), ((int(base**exp), base) for base in range(upto + 1)))
def AddDivisionEquality(self, target, num, denom):
"""Adds `target == num // denom` (integer division rounded towards 0).
Each parameter is interpreted as a BoundedLinearExpression, and a layer of indirection is applied such that each Constraint in the returned tuple can accept an enforcement literal."""
intermediate_target, target_constraint = self.NewIntermediateIntVar(target, f'{repr(target)} == {repr(num)} // {repr(denom)}: target')
intermediate_num, num_constraint = self.NewIntermediateIntVar(num, f'{repr(target)} == {repr(num)} // {repr(denom)}: num', lb = 0)
intermediate_denom, denom_constraint = self.NewIntermediateIntVar(denom, f'{repr(target)} == {repr(num)} // {repr(denom)}: denom', lb = 0)
super().AddDivisionEquality(intermediate_target, intermediate_num, intermediate_denom)
return (target_constraint, num_constraint, denom_constraint)
def AddIf(self, variable, *constraints):
"""Add constraints to the model, only enforced if the specified variable is true.