/* * Main authors: * Kevin Leo * Andrea Rendl * Guido Tack */ /* This Source Code Form is subject to the terms of the Mozilla Public * License, v. 2.0. If a copy of the MPL was not distributed with this * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ #include #include #include #include using namespace Gecode; namespace MiniZinc { namespace GecodeConstraints { void p_distinct(SolverInstanceBase& s, const Call* call) { auto& gi = static_cast(s); IntVarArgs va = gi.arg2intvarargs(call->arg(0)); MZ_IntConLevel icl = GecodeSolverInstance::ann2icl(call->ann()); unshare(*gi.currentSpace, va); distinct(*gi.currentSpace, va, icl == MZ_ICL_DEF ? MZ_ICL_DOM : icl); } void p_distinct_offset(SolverInstanceBase& s, const Call* call) { auto& gi = static_cast(s); IntVarArgs va = gi.arg2intvarargs(call->arg(1)); unshare(*gi.currentSpace, va); IntArgs oa = GecodeSolverInstance::arg2intargs(call->arg(0)); MZ_IntConLevel icl = GecodeSolverInstance::ann2icl(call->ann()); distinct(*gi.currentSpace, oa, va, icl == MZ_ICL_DEF ? MZ_ICL_DOM : icl); } void p_all_equal(SolverInstanceBase& s, const Call* call) { auto& gi = static_cast(s); IntVarArgs va = gi.arg2intvarargs(call->arg(0)); rel(*gi.currentSpace, va, IRT_EQ, GecodeSolverInstance::ann2icl(call->ann())); } void p_int_cmp(GecodeSolverInstance& s, IntRelType irt, const Call* ce) { const Annotation& ann = ce->ann(); Expression* lhs = ce->arg(0); Expression* rhs = ce->arg(1); if (lhs->type().isvarint()) { if (rhs->type().isvarint()) { rel(*s.currentSpace, s.arg2intvar(lhs), irt, s.arg2intvar(rhs), GecodeSolverInstance::ann2icl(ann)); } else { rel(*s.currentSpace, s.arg2intvar(lhs), irt, static_cast(rhs->cast()->v().toInt()), GecodeSolverInstance::ann2icl(ann)); } } else { rel(*s.currentSpace, s.arg2intvar(rhs), swap(irt), static_cast(lhs->cast()->v().toInt()), GecodeSolverInstance::ann2icl(ann)); } } void p_int_eq(SolverInstanceBase& s, const Call* call) { p_int_cmp(static_cast(s), IRT_EQ, call); } void p_int_ne(SolverInstanceBase& s, const Call* call) { p_int_cmp(static_cast(s), IRT_NQ, call); } void p_int_ge(SolverInstanceBase& s, const Call* call) { p_int_cmp(static_cast(s), IRT_GQ, call); } void p_int_gt(SolverInstanceBase& s, const Call* call) { p_int_cmp(static_cast(s), IRT_GR, call); } void p_int_le(SolverInstanceBase& s, const Call* call) { p_int_cmp(static_cast(s), IRT_LQ, call); } void p_int_lt(SolverInstanceBase& s, const Call* call) { p_int_cmp(static_cast(s), IRT_LE, call); } void p_int_cmp_reif(GecodeSolverInstance& s, IntRelType irt, ReifyMode rm, const Call* call) { const Annotation& ann = call->ann(); if (rm == RM_EQV && !call->arg(2)->type().isvar()) { if (call->arg(2)->cast()->v()) { p_int_cmp(s, irt, call); } else { p_int_cmp(s, neg(irt), call); } return; } if (call->arg(0)->type().isvarint()) { if (call->arg(1)->type().isvarint()) { rel(*s.currentSpace, s.arg2intvar(call->arg(0)), irt, s.arg2intvar(call->arg(1)), Reify(s.arg2boolvar(call->arg(2)), rm), GecodeSolverInstance::ann2icl(ann)); } else { rel(*s.currentSpace, s.arg2intvar(call->arg(0)), irt, static_cast(call->arg(1)->cast()->v().toInt()), Reify(s.arg2boolvar(call->arg(2)), rm), GecodeSolverInstance::ann2icl(ann)); } } else { rel(*s.currentSpace, s.arg2intvar(call->arg(1)), swap(irt), static_cast(call->arg(0)->cast()->v().toInt()), Reify(s.arg2boolvar(call->arg(2)), rm), GecodeSolverInstance::ann2icl(ann)); } } ///* Comparisons */ void p_int_eq_reif(SolverInstanceBase& s, const Call* call) { p_int_cmp_reif(static_cast(s), IRT_EQ, RM_EQV, call); } void p_int_ne_reif(SolverInstanceBase& s, const Call* call) { p_int_cmp_reif(static_cast(s), IRT_NQ, RM_EQV, call); } void p_int_ge_reif(SolverInstanceBase& s, const Call* call) { p_int_cmp_reif(static_cast(s), IRT_GQ, RM_EQV, call); } void p_int_gt_reif(SolverInstanceBase& s, const Call* call) { p_int_cmp_reif(static_cast(s), IRT_GR, RM_EQV, call); } void p_int_le_reif(SolverInstanceBase& s, const Call* call) { p_int_cmp_reif(static_cast(s), IRT_LQ, RM_EQV, call); } void p_int_lt_reif(SolverInstanceBase& s, const Call* call) { p_int_cmp_reif(static_cast(s), IRT_LE, RM_EQV, call); } void p_int_eq_imp(SolverInstanceBase& s, const Call* call) { p_int_cmp_reif(static_cast(s), IRT_EQ, RM_IMP, call); } void p_int_ne_imp(SolverInstanceBase& s, const Call* call) { p_int_cmp_reif(static_cast(s), IRT_NQ, RM_IMP, call); } void p_int_ge_imp(SolverInstanceBase& s, const Call* call) { p_int_cmp_reif(static_cast(s), IRT_GQ, RM_IMP, call); } void p_int_gt_imp(SolverInstanceBase& s, const Call* call) { p_int_cmp_reif(static_cast(s), IRT_GR, RM_IMP, call); } void p_int_le_imp(SolverInstanceBase& s, const Call* call) { p_int_cmp_reif(static_cast(s), IRT_LQ, RM_IMP, call); } void p_int_lt_imp(SolverInstanceBase& s, const Call* call) { p_int_cmp_reif(static_cast(s), IRT_LE, RM_IMP, call); } void p_int_lin_cmp(GecodeSolverInstance& s, IntRelType irt, const Call* call) { const Annotation& ann = call->ann(); IntArgs ia = GecodeSolverInstance::arg2intargs(call->arg(0)); ArrayLit* vars = s.arg2arraylit(call->arg(1)); int singleIntVar; if (s.isBoolArray(vars, singleIntVar)) { if (singleIntVar != -1) { if (std::abs(ia[singleIntVar]) == 1 && call->arg(2)->cast()->v().toInt() == 0) { IntVar siv = s.arg2intvar((*vars)[singleIntVar]); BoolVarArgs iv = s.arg2boolvarargs(vars, 0, singleIntVar); IntArgs ia_tmp(ia.size() - 1); int count = 0; for (int i = 0; i < ia.size(); i++) { if (i != singleIntVar) { ia_tmp[count++] = ia[singleIntVar] == -1 ? ia[i] : -ia[i]; } } IntRelType t = (ia[singleIntVar] == -1 ? irt : swap(irt)); linear(*s.currentSpace, ia_tmp, iv, t, siv, GecodeSolverInstance::ann2icl(ann)); } else { IntVarArgs iv = s.arg2intvarargs(vars); linear(*s.currentSpace, ia, iv, irt, static_cast(call->arg(2)->cast()->v().toInt()), GecodeSolverInstance::ann2icl(ann)); } } else { BoolVarArgs iv = s.arg2boolvarargs(vars); linear(*s.currentSpace, ia, iv, irt, static_cast(call->arg(2)->cast()->v().toInt()), GecodeSolverInstance::ann2icl(ann)); } } else { IntVarArgs iv = s.arg2intvarargs(vars); linear(*s.currentSpace, ia, iv, irt, static_cast(call->arg(2)->cast()->v().toInt()), GecodeSolverInstance::ann2icl(ann)); } } void p_int_lin_cmp_reif(GecodeSolverInstance& s, IntRelType irt, ReifyMode rm, const Call* call) { const Annotation& ann = call->ann(); if (rm == RM_EQV && call->arg(2)->type().isbool()) { if (call->arg(2)->cast()->v()) { p_int_lin_cmp(s, irt, call); } else { p_int_lin_cmp(s, neg(irt), call); } return; } IntArgs ia = GecodeSolverInstance::arg2intargs(call->arg(0)); ArrayLit* vars = s.arg2arraylit(call->arg(1)); int singleIntVar; if (s.isBoolArray(vars, singleIntVar)) { if (singleIntVar != -1) { if (std::abs(ia[singleIntVar]) == 1 && call->arg(2)->cast()->v().toInt() == 0) { IntVar siv = s.arg2intvar((*vars)[singleIntVar]); BoolVarArgs iv = s.arg2boolvarargs(vars, 0, singleIntVar); IntArgs ia_tmp(ia.size() - 1); int count = 0; for (int i = 0; i < ia.size(); i++) { if (i != singleIntVar) { ia_tmp[count++] = ia[singleIntVar] == -1 ? ia[i] : -ia[i]; } } IntRelType t = (ia[singleIntVar] == -1 ? irt : swap(irt)); linear(*s.currentSpace, ia_tmp, iv, t, siv, Reify(s.arg2boolvar(call->arg(3)), rm), GecodeSolverInstance::ann2icl(ann)); } else { IntVarArgs iv = s.arg2intvarargs(vars); linear(*s.currentSpace, ia, iv, irt, static_cast(call->arg(2)->cast()->v().toInt()), Reify(s.arg2boolvar(call->arg(3)), rm), GecodeSolverInstance::ann2icl(ann)); } } else { BoolVarArgs iv = s.arg2boolvarargs(vars); linear(*s.currentSpace, ia, iv, irt, static_cast(call->arg(2)->cast()->v().toInt()), Reify(s.arg2boolvar(call->arg(3)), rm), GecodeSolverInstance::ann2icl(ann)); } } else { IntVarArgs iv = s.arg2intvarargs(vars); linear(*s.currentSpace, ia, iv, irt, static_cast(call->arg(2)->cast()->v().toInt()), Reify(s.arg2boolvar(call->arg(3)), rm), GecodeSolverInstance::ann2icl(ann)); } } void p_int_lin_eq(SolverInstanceBase& s, const Call* call) { p_int_lin_cmp(static_cast(s), IRT_EQ, call); } void p_int_lin_eq_reif(SolverInstanceBase& s, const Call* call) { p_int_lin_cmp_reif(static_cast(s), IRT_EQ, RM_EQV, call); } void p_int_lin_eq_imp(SolverInstanceBase& s, const Call* call) { p_int_lin_cmp_reif(static_cast(s), IRT_EQ, RM_IMP, call); } void p_int_lin_ne(SolverInstanceBase& s, const Call* call) { p_int_lin_cmp(static_cast(s), IRT_NQ, call); } void p_int_lin_ne_reif(SolverInstanceBase& s, const Call* call) { p_int_lin_cmp_reif(static_cast(s), IRT_NQ, RM_EQV, call); } void p_int_lin_ne_imp(SolverInstanceBase& s, const Call* call) { p_int_lin_cmp_reif(static_cast(s), IRT_NQ, RM_IMP, call); } void p_int_lin_le(SolverInstanceBase& s, const Call* call) { p_int_lin_cmp(static_cast(s), IRT_LQ, call); } void p_int_lin_le_reif(SolverInstanceBase& s, const Call* call) { p_int_lin_cmp_reif(static_cast(s), IRT_LQ, RM_EQV, call); } void p_int_lin_le_imp(SolverInstanceBase& s, const Call* call) { p_int_lin_cmp_reif(static_cast(s), IRT_LQ, RM_IMP, call); } void p_int_lin_lt(SolverInstanceBase& s, const Call* call) { p_int_lin_cmp(static_cast(s), IRT_LE, call); } void p_int_lin_lt_reif(SolverInstanceBase& s, const Call* call) { p_int_lin_cmp_reif(static_cast(s), IRT_LE, RM_EQV, call); } void p_int_lin_lt_imp(SolverInstanceBase& s, const Call* call) { p_int_lin_cmp_reif(static_cast(s), IRT_LE, RM_IMP, call); } void p_int_lin_ge(SolverInstanceBase& s, const Call* call) { p_int_lin_cmp(static_cast(s), IRT_GQ, call); } void p_int_lin_ge_reif(SolverInstanceBase& s, const Call* call) { p_int_lin_cmp_reif(static_cast(s), IRT_GQ, RM_EQV, call); } void p_int_lin_ge_imp(SolverInstanceBase& s, const Call* call) { p_int_lin_cmp_reif(static_cast(s), IRT_GQ, RM_IMP, call); } void p_int_lin_gt(SolverInstanceBase& s, const Call* call) { p_int_lin_cmp(static_cast(s), IRT_GR, call); } void p_int_lin_gt_reif(SolverInstanceBase& s, const Call* call) { p_int_lin_cmp_reif(static_cast(s), IRT_GR, RM_EQV, call); } void p_int_lin_gt_imp(SolverInstanceBase& s, const Call* call) { p_int_lin_cmp_reif(static_cast(s), IRT_GR, RM_IMP, call); } void p_bool_lin_cmp(GecodeSolverInstance& s, IntRelType irt, const Call* call) { const Annotation& ann = call->ann(); IntArgs ia = GecodeSolverInstance::arg2intargs(call->arg(0)); BoolVarArgs iv = s.arg2boolvarargs(call->arg(1)); if (call->arg(2)->type().isvarint()) { linear(*s.currentSpace, ia, iv, irt, s.resolveVar(call->arg(2)->cast()->decl()).intVar(s.currentSpace), GecodeSolverInstance::ann2icl(ann)); } else { linear(*s.currentSpace, ia, iv, irt, static_cast(call->arg(2)->cast()->v().toInt()), GecodeSolverInstance::ann2icl(ann)); } } void p_bool_lin_cmp_reif(GecodeSolverInstance& s, IntRelType irt, ReifyMode rm, const Call* call) { const Annotation& ann = call->ann(); if (rm == RM_EQV && call->arg(2)->type().isbool()) { if (call->arg(2)->cast()->v()) { p_bool_lin_cmp(s, irt, call); } else { p_bool_lin_cmp(s, neg(irt), call); } return; } IntArgs ia = GecodeSolverInstance::arg2intargs(call->arg(0)); BoolVarArgs iv = s.arg2boolvarargs(call->arg(1)); if (call->arg(2)->type().isvarint()) { linear(*s.currentSpace, ia, iv, irt, s.resolveVar(call->arg(2)->cast()->decl()).intVar(s.currentSpace), Reify(s.arg2boolvar(call->arg(3)), rm), GecodeSolverInstance::ann2icl(ann)); } else { linear(*s.currentSpace, ia, iv, irt, static_cast(call->arg(2)->cast()->v().toInt()), Reify(s.arg2boolvar(call->arg(3)), rm), GecodeSolverInstance::ann2icl(ann)); } } void p_bool_lin_eq(SolverInstanceBase& s, const Call* call) { p_bool_lin_cmp(static_cast(s), IRT_EQ, call); } void p_bool_lin_eq_reif(SolverInstanceBase& s, const Call* call) { p_bool_lin_cmp_reif(static_cast(s), IRT_EQ, RM_EQV, call); } void p_bool_lin_eq_imp(SolverInstanceBase& s, const Call* call) { p_bool_lin_cmp_reif(static_cast(s), IRT_EQ, RM_IMP, call); } void p_bool_lin_ne(SolverInstanceBase& s, const Call* call) { p_bool_lin_cmp(static_cast(s), IRT_NQ, call); } void p_bool_lin_ne_reif(SolverInstanceBase& s, const Call* call) { p_bool_lin_cmp_reif(static_cast(s), IRT_NQ, RM_EQV, call); } void p_bool_lin_ne_imp(SolverInstanceBase& s, const Call* call) { p_bool_lin_cmp_reif(static_cast(s), IRT_NQ, RM_IMP, call); } void p_bool_lin_le(SolverInstanceBase& s, const Call* call) { p_bool_lin_cmp(static_cast(s), IRT_LQ, call); } void p_bool_lin_le_reif(SolverInstanceBase& s, const Call* call) { p_bool_lin_cmp_reif(static_cast(s), IRT_LQ, RM_EQV, call); } void p_bool_lin_le_imp(SolverInstanceBase& s, const Call* call) { p_bool_lin_cmp_reif(static_cast(s), IRT_LQ, RM_IMP, call); } void p_bool_lin_lt(SolverInstanceBase& s, const Call* call) { p_bool_lin_cmp(static_cast(s), IRT_LE, call); } void p_bool_lin_lt_reif(SolverInstanceBase& s, const Call* call) { p_bool_lin_cmp_reif(static_cast(s), IRT_LE, RM_EQV, call); } void p_bool_lin_lt_imp(SolverInstanceBase& s, const Call* call) { p_bool_lin_cmp_reif(static_cast(s), IRT_LE, RM_IMP, call); } void p_bool_lin_ge(SolverInstanceBase& s, const Call* call) { p_bool_lin_cmp(static_cast(s), IRT_GQ, call); } void p_bool_lin_ge_reif(SolverInstanceBase& s, const Call* call) { p_bool_lin_cmp_reif(static_cast(s), IRT_GQ, RM_EQV, call); } void p_bool_lin_ge_imp(SolverInstanceBase& s, const Call* call) { p_bool_lin_cmp_reif(static_cast(s), IRT_GQ, RM_IMP, call); } void p_bool_lin_gt(SolverInstanceBase& s, const Call* call) { p_bool_lin_cmp(static_cast(s), IRT_GR, call); } void p_bool_lin_gt_reif(SolverInstanceBase& s, const Call* call) { p_bool_lin_cmp_reif(static_cast(s), IRT_GR, RM_EQV, call); } void p_bool_lin_gt_imp(SolverInstanceBase& s, const Call* call) { p_bool_lin_cmp_reif(static_cast(s), IRT_GR, RM_IMP, call); } ///* arithmetic constraints */ void p_int_plus(SolverInstanceBase& s, const Call* call) { auto& gi = static_cast(s); if (!call->arg(0)->type().isvarint()) { rel(*gi.currentSpace, static_cast(call->arg(0)->cast()->v().toInt()) + gi.arg2intvar(call->arg(1)) == gi.arg2intvar(call->arg(2)), GecodeSolverInstance::ann2icl(call->ann())); } else if (!call->arg(1)->type().isvarint()) { rel(*gi.currentSpace, gi.arg2intvar(call->arg(0)) + static_cast(call->arg(1)->cast()->v().toInt()) == gi.arg2intvar(call->arg(2)), GecodeSolverInstance::ann2icl(call->ann())); } else if (!call->arg(2)->type().isvarint()) { rel(*gi.currentSpace, gi.arg2intvar(call->arg(0)) + gi.arg2intvar(call->arg(1)) == static_cast(call->arg(2)->cast()->v().toInt()), GecodeSolverInstance::ann2icl(call->ann())); } else { rel(*gi.currentSpace, gi.arg2intvar(call->arg(0)) + gi.arg2intvar(call->arg(1)) == gi.arg2intvar(call->arg(2)), GecodeSolverInstance::ann2icl(call->ann())); } } void p_int_minus(SolverInstanceBase& s, const Call* call) { auto& gi = static_cast(s); if (!call->arg(0)->type().isvarint()) { rel(*gi.currentSpace, static_cast(call->arg(0)->cast()->v().toInt()) - gi.arg2intvar(call->arg(1)) == gi.arg2intvar(call->arg(2)), GecodeSolverInstance::ann2icl(call->ann())); } else if (!call->arg(1)->type().isvarint()) { rel(*gi.currentSpace, gi.arg2intvar(call->arg(0)) - static_cast(call->arg(1)->cast()->v().toInt()) == gi.arg2intvar(call->arg(2)), GecodeSolverInstance::ann2icl(call->ann())); } else if (!call->arg(2)->type().isvarint()) { rel(*gi.currentSpace, gi.arg2intvar(call->arg(0)) - gi.arg2intvar(call->arg(1)) == static_cast(call->arg(2)->cast()->v().toInt()), GecodeSolverInstance::ann2icl(call->ann())); } else { rel(*gi.currentSpace, gi.arg2intvar(call->arg(0)) - gi.arg2intvar(call->arg(1)) == gi.arg2intvar(call->arg(2)), GecodeSolverInstance::ann2icl(call->ann())); } } void p_int_times(SolverInstanceBase& s, const Call* call) { auto& gi = static_cast(s); IntVar x0 = gi.arg2intvar(call->arg(0)); IntVar x1 = gi.arg2intvar(call->arg(1)); IntVar x2 = gi.arg2intvar(call->arg(2)); mult(*gi.currentSpace, x0, x1, x2, GecodeSolverInstance::ann2icl(call->ann())); } void p_int_div(SolverInstanceBase& s, const Call* call) { auto& gi = static_cast(s); IntVar x0 = gi.arg2intvar(call->arg(0)); IntVar x1 = gi.arg2intvar(call->arg(1)); IntVar x2 = gi.arg2intvar(call->arg(2)); div(*gi.currentSpace, x0, x1, x2, GecodeSolverInstance::ann2icl(call->ann())); } void p_int_mod(SolverInstanceBase& s, const Call* call) { auto& gi = static_cast(s); IntVar x0 = gi.arg2intvar(call->arg(0)); IntVar x1 = gi.arg2intvar(call->arg(1)); IntVar x2 = gi.arg2intvar(call->arg(2)); mod(*gi.currentSpace, x0, x1, x2, GecodeSolverInstance::ann2icl(call->ann())); } void p_int_min(SolverInstanceBase& s, const Call* call) { auto& gi = static_cast(s); IntVar x0 = gi.arg2intvar(call->arg(0)); IntVar x1 = gi.arg2intvar(call->arg(1)); IntVar x2 = gi.arg2intvar(call->arg(2)); min(*gi.currentSpace, x0, x1, x2, GecodeSolverInstance::ann2icl(call->ann())); } void p_int_max(SolverInstanceBase& s, const Call* call) { auto& gi = static_cast(s); IntVar x0 = gi.arg2intvar(call->arg(0)); IntVar x1 = gi.arg2intvar(call->arg(1)); IntVar x2 = gi.arg2intvar(call->arg(2)); max(*gi.currentSpace, x0, x1, x2, GecodeSolverInstance::ann2icl(call->ann())); } void p_int_negate(SolverInstanceBase& s, const Call* call) { auto& gi = static_cast(s); IntVar x0 = gi.arg2intvar(call->arg(0)); IntVar x1 = gi.arg2intvar(call->arg(1)); rel(*gi.currentSpace, x0 == -x1, GecodeSolverInstance::ann2icl(call->ann())); } ///* Boolean constraints */ void p_bool_cmp(GecodeSolverInstance& s, IntRelType irt, const Call* call) { const Annotation& ann = call->ann(); rel(*s.currentSpace, s.arg2boolvar(call->arg(0)), irt, s.arg2boolvar(call->arg(1)), GecodeSolverInstance::ann2icl(ann)); } void p_bool_cmp_reif(GecodeSolverInstance& s, IntRelType irt, ReifyMode rm, const Call* call) { const Annotation& ann = call->ann(); rel(*s.currentSpace, s.arg2boolvar(call->arg(0)), irt, s.arg2boolvar(call->arg(1)), Reify(s.arg2boolvar(call->arg(2)), rm), GecodeSolverInstance::ann2icl(ann)); } void p_bool_eq(SolverInstanceBase& s, const Call* call) { p_bool_cmp(static_cast(s), IRT_EQ, call); } void p_bool_eq_reif(SolverInstanceBase& s, const Call* call) { p_bool_cmp_reif(static_cast(s), IRT_EQ, RM_EQV, call); } void p_bool_eq_imp(SolverInstanceBase& s, const Call* call) { p_bool_cmp_reif(static_cast(s), IRT_EQ, RM_IMP, call); } void p_bool_ne(SolverInstanceBase& s, const Call* call) { p_bool_cmp(static_cast(s), IRT_NQ, call); } void p_bool_ne_reif(SolverInstanceBase& s, const Call* call) { p_bool_cmp_reif(static_cast(s), IRT_NQ, RM_EQV, call); } void p_bool_ne_imp(SolverInstanceBase& s, const Call* call) { p_bool_cmp_reif(static_cast(s), IRT_NQ, RM_IMP, call); } void p_bool_ge(SolverInstanceBase& s, const Call* call) { p_bool_cmp(static_cast(s), IRT_GQ, call); } void p_bool_ge_reif(SolverInstanceBase& s, const Call* call) { p_bool_cmp_reif(static_cast(s), IRT_GQ, RM_EQV, call); } void p_bool_ge_imp(SolverInstanceBase& s, const Call* call) { p_bool_cmp_reif(static_cast(s), IRT_GQ, RM_IMP, call); } void p_bool_le(SolverInstanceBase& s, const Call* call) { p_bool_cmp(static_cast(s), IRT_LQ, call); } void p_bool_le_reif(SolverInstanceBase& s, const Call* call) { p_bool_cmp_reif(static_cast(s), IRT_LQ, RM_EQV, call); } void p_bool_le_imp(SolverInstanceBase& s, const Call* call) { p_bool_cmp_reif(static_cast(s), IRT_LQ, RM_IMP, call); } void p_bool_gt(SolverInstanceBase& s, const Call* call) { p_bool_cmp(static_cast(s), IRT_GR, call); } void p_bool_gt_reif(SolverInstanceBase& s, const Call* call) { p_bool_cmp_reif(static_cast(s), IRT_GR, RM_EQV, call); } void p_bool_gt_imp(SolverInstanceBase& s, const Call* call) { p_bool_cmp_reif(static_cast(s), IRT_GR, RM_IMP, call); } void p_bool_lt(SolverInstanceBase& s, const Call* call) { p_bool_cmp(static_cast(s), IRT_LE, call); } void p_bool_lt_reif(SolverInstanceBase& s, const Call* call) { p_bool_cmp_reif(static_cast(s), IRT_LE, RM_EQV, call); } void p_bool_lt_imp(SolverInstanceBase& s, const Call* call) { p_bool_cmp_reif(static_cast(s), IRT_LE, RM_IMP, call); } #define BOOL_OP(op) \ BoolVar b0 = gi.arg2boolvar(call->arg(0)); \ BoolVar b1 = gi.arg2boolvar(call->arg(1)); \ if (!call->arg(2)->type().isvar() && call->arg(2)->type().isbool()) { \ rel(*gi.currentSpace, b0, op, b1, call->arg(2)->cast()->v(), gi.ann2icl(ann)); \ } else { \ rel(*gi.currentSpace, b0, op, b1, \ gi.resolveVar(gi.getVarDecl(call->arg(2))).boolVar(gi.currentSpace), gi.ann2icl(ann)); \ } #define BOOL_ARRAY_OP(op) \ BoolVarArgs bv = gi.arg2boolvarargs(call->arg(0)); \ if (call->argCount() == 1) { \ rel(*gi.currentSpace, op, bv, 1, gi.ann2icl(ann)); \ } else if (!call->arg(1)->type().isvar() && call->arg(1)->type().isbool()) { \ rel(*gi.currentSpace, op, bv, call->arg(1)->cast()->v(), gi.ann2icl(ann)); \ } else { \ rel(*gi.currentSpace, op, bv, \ gi.resolveVar(gi.getVarDecl(call->arg(1))).boolVar(gi.currentSpace), gi.ann2icl(ann)); \ } void p_bool_or(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BOOL_OP(BoolOpType::BOT_OR); } void p_bool_or_imp(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BoolVar b0 = gi.arg2boolvar(call->arg(0)); BoolVar b1 = gi.arg2boolvar(call->arg(1)); BoolVar b2 = gi.arg2boolvar(call->arg(2)); clause(*gi.currentSpace, BoolOpType::BOT_OR, BoolVarArgs() << b0 << b1, BoolVarArgs() << b2, 1, GecodeSolverInstance::ann2icl(ann)); } void p_bool_and(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BOOL_OP(BoolOpType::BOT_AND); } void p_bool_and_imp(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BoolVar b0 = gi.arg2boolvar(call->arg(0)); BoolVar b1 = gi.arg2boolvar(call->arg(1)); BoolVar b2 = gi.arg2boolvar(call->arg(2)); rel(*gi.currentSpace, b2, BoolOpType::BOT_IMP, b0, 1, GecodeSolverInstance::ann2icl(ann)); rel(*gi.currentSpace, b2, BoolOpType::BOT_IMP, b1, 1, GecodeSolverInstance::ann2icl(ann)); } void p_array_bool_and(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BOOL_ARRAY_OP(Gecode::BoolOpType::BOT_AND); } void p_array_bool_and_imp(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BoolVarArgs bv = gi.arg2boolvarargs(call->arg(0)); BoolVar b1 = gi.arg2boolvar(call->arg(1)); for (int i = static_cast(bv.size()); (i--) != 0;) { rel(*gi.currentSpace, b1, Gecode::BoolOpType::BOT_IMP, bv[i], 1, GecodeSolverInstance::ann2icl(ann)); } } void p_array_bool_or(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BOOL_ARRAY_OP(BoolOpType::BOT_OR); } void p_array_bool_or_imp(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BoolVarArgs bv = gi.arg2boolvarargs(call->arg(0)); BoolVar b1 = gi.arg2boolvar(call->arg(1)); clause(*gi.currentSpace, BoolOpType::BOT_OR, bv, BoolVarArgs() << b1, 1, GecodeSolverInstance::ann2icl(ann)); } void p_array_bool_xor(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BOOL_ARRAY_OP(BoolOpType::BOT_XOR); } void p_array_bool_xor_imp(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BoolVarArgs bv = gi.arg2boolvarargs(call->arg(0)); BoolVar tmp(*gi.currentSpace, 0, 1); rel(*gi.currentSpace, BoolOpType::BOT_XOR, bv, tmp, GecodeSolverInstance::ann2icl(ann)); rel(*gi.currentSpace, gi.arg2boolvar(call->arg(1)), BoolOpType::BOT_IMP, tmp, 1); } void p_array_bool_clause(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BoolVarArgs bvp = gi.arg2boolvarargs(call->arg(0)); BoolVarArgs bvn = gi.arg2boolvarargs(call->arg(1)); clause(*gi.currentSpace, BoolOpType::BOT_OR, bvp, bvn, 1, GecodeSolverInstance::ann2icl(ann)); } void p_array_bool_clause_reif(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BoolVarArgs bvp = gi.arg2boolvarargs(call->arg(0)); BoolVarArgs bvn = gi.arg2boolvarargs(call->arg(1)); BoolVar b0 = gi.arg2boolvar(call->arg(2)); clause(*gi.currentSpace, BoolOpType::BOT_OR, bvp, bvn, b0, GecodeSolverInstance::ann2icl(ann)); } void p_array_bool_clause_imp(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BoolVarArgs bvp = gi.arg2boolvarargs(call->arg(0)); BoolVarArgs bvn = gi.arg2boolvarargs(call->arg(1)); BoolVar b0 = gi.arg2boolvar(call->arg(2)); clause(*gi.currentSpace, BoolOpType::BOT_OR, bvp, bvn, b0, GecodeSolverInstance::ann2icl(ann)); } void p_bool_xor(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BOOL_OP(BoolOpType::BOT_XOR); } void p_bool_xor_imp(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BoolVar b0 = gi.arg2boolvar(call->arg(0)); BoolVar b1 = gi.arg2boolvar(call->arg(1)); BoolVar b2 = gi.arg2boolvar(call->arg(2)); clause(*gi.currentSpace, BoolOpType::BOT_OR, BoolVarArgs() << b0 << b1, BoolVarArgs() << b2, 1, GecodeSolverInstance::ann2icl(ann)); clause(*gi.currentSpace, BoolOpType::BOT_OR, BoolVarArgs(), BoolVarArgs() << b0 << b1 << b2, 1, GecodeSolverInstance::ann2icl(ann)); } void p_bool_l_imp(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BoolVar b0 = gi.arg2boolvar(call->arg(0)); BoolVar b1 = gi.arg2boolvar(call->arg(1)); if (call->arg(2)->type().isbool()) { rel(*gi.currentSpace, b1, BoolOpType::BOT_IMP, b0, static_cast(call->arg(2)->cast()->v()), GecodeSolverInstance::ann2icl(ann)); } else { rel(*gi.currentSpace, b1, BoolOpType::BOT_IMP, b0, gi.resolveVar(call->arg(2)->cast()->decl()).boolVar(gi.currentSpace), GecodeSolverInstance::ann2icl(ann)); } } void p_bool_r_imp(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BOOL_OP(BoolOpType::BOT_IMP); } void p_bool_not(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BoolVar x0 = gi.arg2boolvar(call->arg(0)); BoolVar x1 = gi.arg2boolvar(call->arg(1)); rel(*gi.currentSpace, x0, BoolOpType::BOT_XOR, x1, 1, GecodeSolverInstance::ann2icl(ann)); } ///* element constraints */ void p_array_int_element(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVar selector = gi.arg2intvar(call->arg(0)); rel(*gi.currentSpace, selector > 0); if (call->arg(1)->type().isvar()) { IntVarArgs iv = gi.arg2intvarargs(call->arg(1), 1); element(*gi.currentSpace, iv, selector, gi.arg2intvar(call->arg(2)), GecodeSolverInstance::ann2icl(ann)); } else { IntArgs ia = GecodeSolverInstance::arg2intargs(call->arg(1), 1); element(*gi.currentSpace, ia, selector, gi.arg2intvar(call->arg(2)), GecodeSolverInstance::ann2icl(ann)); } } void p_array_bool_element(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVar selector = gi.arg2intvar(call->arg(0)); rel(*gi.currentSpace, selector > 0); if (call->arg(1)->type().isvar()) { BoolVarArgs iv = gi.arg2boolvarargs(call->arg(1), 1); element(*gi.currentSpace, iv, selector, gi.arg2boolvar(call->arg(2)), GecodeSolverInstance::ann2icl(ann)); } else { IntArgs ia = GecodeSolverInstance::arg2boolargs(call->arg(1), 1); element(*gi.currentSpace, ia, selector, gi.arg2boolvar(call->arg(2)), GecodeSolverInstance::ann2icl(ann)); } } ///* coercion constraints */ void p_bool2int(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BoolVar x0 = gi.arg2boolvar(call->arg(0)); IntVar x1 = gi.arg2intvar(call->arg(1)); if (call->arg(0)->type().isvarbool() && call->arg(1)->type().isvarint()) { unsigned int index = gi.resolveVar(call->arg(0)->cast()->decl()).index(); gi.resolveVar(call->arg(1)->cast()->decl()).setBoolAliasIndex(index); } channel(*gi.currentSpace, x0, x1, GecodeSolverInstance::ann2icl(ann)); } void p_int_in(SolverInstanceBase& s, const Call* call) { auto& gi = static_cast(s); IntSet d = gi.arg2intset(s.env().envi(), call->arg(1)); if (call->arg(0)->type().isvarbool()) { Gecode::IntSetRanges dr(d); Iter::Ranges::Singleton sr(0, 1); Iter::Ranges::Inter i(dr, sr); IntSet d01(i); if (d01.size() == 0) { gi.currentSpace->fail(); } else { rel(*gi.currentSpace, gi.arg2boolvar(call->arg(0)), IRT_GQ, d01.min()); rel(*gi.currentSpace, gi.arg2boolvar(call->arg(0)), IRT_LQ, d01.max()); } } else { dom(*gi.currentSpace, gi.arg2intvar(call->arg(0)), d); } } void p_int_in_reif(SolverInstanceBase& s, const Call* call) { auto& gi = static_cast(s); IntSet d = gi.arg2intset(s.env().envi(), call->arg(1)); if (call->arg(0)->type().isvarbool()) { Gecode::IntSetRanges dr(d); Iter::Ranges::Singleton sr(0, 1); Iter::Ranges::Inter i(dr, sr); IntSet d01(i); if (d01.size() == 0) { rel(*gi.currentSpace, gi.arg2boolvar(call->arg(2)) == 0); } else if (d01.max() == 0) { rel(*gi.currentSpace, gi.arg2boolvar(call->arg(2)) == !gi.arg2boolvar(call->arg(0))); } else if (d01.min() == 1) { rel(*gi.currentSpace, gi.arg2boolvar(call->arg(2)) == gi.arg2boolvar(call->arg(0))); } else { rel(*gi.currentSpace, gi.arg2boolvar(call->arg(2)) == 1); } } else { dom(*gi.currentSpace, gi.arg2intvar(call->arg(0)), d, gi.arg2boolvar(call->arg(2))); } } void p_int_in_imp(SolverInstanceBase& s, const Call* call) { auto& gi = static_cast(s); IntSet d = gi.arg2intset(s.env().envi(), call->arg(1)); if (call->arg(0)->type().isvarbool()) { Gecode::IntSetRanges dr(d); Iter::Ranges::Singleton sr(0, 1); Iter::Ranges::Inter i(dr, sr); IntSet d01(i); if (d01.size() == 0) { rel(*gi.currentSpace, gi.arg2boolvar(call->arg(2)) == 0); } else if (d01.max() == 0) { rel(*gi.currentSpace, gi.arg2boolvar(call->arg(2)) >> !gi.arg2boolvar(call->arg(0))); } else if (d01.min() == 1) { rel(*gi.currentSpace, gi.arg2boolvar(call->arg(2)) >> gi.arg2boolvar(call->arg(0))); } } else { dom(*gi.currentSpace, gi.arg2intvar(call->arg(0)), d, Reify(gi.arg2boolvar(call->arg(2)), RM_IMP)); } } ///* constraints from the standard library */ void p_abs(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVar x0 = gi.arg2intvar(call->arg(0)); IntVar x1 = gi.arg2intvar(call->arg(1)); abs(*gi.currentSpace, x0, x1, GecodeSolverInstance::ann2icl(ann)); } void p_array_int_lt(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs iv0 = gi.arg2intvarargs(call->arg(0)); IntVarArgs iv1 = gi.arg2intvarargs(call->arg(1)); rel(*gi.currentSpace, iv0, IRT_LE, iv1, GecodeSolverInstance::ann2icl(ann)); } void p_array_int_lq(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs iv0 = gi.arg2intvarargs(call->arg(0)); IntVarArgs iv1 = gi.arg2intvarargs(call->arg(1)); rel(*gi.currentSpace, iv0, IRT_LQ, iv1, GecodeSolverInstance::ann2icl(ann)); } void p_array_bool_lt(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BoolVarArgs bv0 = gi.arg2boolvarargs(call->arg(0)); BoolVarArgs bv1 = gi.arg2boolvarargs(call->arg(1)); rel(*gi.currentSpace, bv0, IRT_LE, bv1, GecodeSolverInstance::ann2icl(ann)); } void p_array_bool_lq(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BoolVarArgs bv0 = gi.arg2boolvarargs(call->arg(0)); BoolVarArgs bv1 = gi.arg2boolvarargs(call->arg(1)); rel(*gi.currentSpace, bv0, IRT_LQ, bv1, GecodeSolverInstance::ann2icl(ann)); } void p_count(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs iv = gi.arg2intvarargs(call->arg(0)); if (!call->arg(1)->type().isvarint()) { if (!call->arg(2)->type().isvarint()) { count(*gi.currentSpace, iv, static_cast(call->arg(1)->cast()->v().toInt()), IRT_EQ, static_cast(call->arg(2)->cast()->v().toInt()), GecodeSolverInstance::ann2icl(ann)); } else { count(*gi.currentSpace, iv, static_cast(call->arg(1)->cast()->v().toInt()), IRT_EQ, gi.arg2intvar(call->arg(2)), GecodeSolverInstance::ann2icl(ann)); } } else if (!call->arg(2)->type().isvarint()) { count(*gi.currentSpace, iv, gi.arg2intvar(call->arg(1)), IRT_EQ, static_cast(call->arg(2)->cast()->v().toInt()), GecodeSolverInstance::ann2icl(ann)); } else { count(*gi.currentSpace, iv, gi.arg2intvar(call->arg(1)), IRT_EQ, gi.arg2intvar(call->arg(2)), GecodeSolverInstance::ann2icl(ann)); } } void p_count_reif(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs iv = gi.arg2intvarargs(call->arg(0)); IntVar x = gi.arg2intvar(call->arg(1)); IntVar y = gi.arg2intvar(call->arg(2)); BoolVar b = gi.arg2boolvar(call->arg(3)); IntVar c(*gi.currentSpace, 0, Int::Limits::max); count(*gi.currentSpace, iv, x, IRT_EQ, c, GecodeSolverInstance::ann2icl(ann)); rel(*gi.currentSpace, b == (c == y)); } void p_count_imp(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs iv = gi.arg2intvarargs(call->arg(0)); IntVar x = gi.arg2intvar(call->arg(1)); IntVar y = gi.arg2intvar(call->arg(2)); BoolVar b = gi.arg2boolvar(call->arg(3)); IntVar c(*gi.currentSpace, 0, Int::Limits::max); count(*gi.currentSpace, iv, x, IRT_EQ, c, GecodeSolverInstance::ann2icl(ann)); rel(*gi.currentSpace, b >> (c == y)); } void count_rel(IntRelType irt, SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs iv = gi.arg2intvarargs(call->arg(1)); count(*gi.currentSpace, iv, static_cast(call->arg(2)->cast()->v().toInt()), irt, static_cast(call->arg(0)->cast()->v().toInt()), GecodeSolverInstance::ann2icl(ann)); } void p_at_most(SolverInstanceBase& s, const Call* call) { count_rel(IRT_LQ, s, call); } void p_at_least(SolverInstanceBase& s, const Call* call) { count_rel(IRT_GQ, s, call); } void p_bin_packing_load(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); int minIdx = static_cast(call->arg(3)->cast()->v().toInt()); IntVarArgs load = gi.arg2intvarargs(call->arg(0)); IntVarArgs l; IntVarArgs bin = gi.arg2intvarargs(call->arg(1)); for (int i = bin.size(); (i--) != 0;) { rel(*gi.currentSpace, bin[i] >= minIdx); } if (minIdx > 0) { for (int i = minIdx; (i--) != 0;) { l << IntVar(*gi.currentSpace, 0, 0); } } else if (minIdx < 0) { IntVarArgs bin2(bin.size()); for (int i = bin.size(); (i--) != 0;) { bin2[i] = expr(*gi.currentSpace, bin[i] - minIdx, GecodeSolverInstance::ann2icl(ann)); } bin = bin2; } l << load; IntArgs sizes = GecodeSolverInstance::arg2intargs(call->arg(2)); IntVarArgs allvars = l + bin; unshare(*gi.currentSpace, allvars); binpacking(*gi.currentSpace, allvars.slice(0, 1, l.size()), allvars.slice(l.size(), 1, bin.size()), sizes, GecodeSolverInstance::ann2icl(ann)); } void p_global_cardinality(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs iv0 = gi.arg2intvarargs(call->arg(0)); IntArgs cover = GecodeSolverInstance::arg2intargs(call->arg(1)); IntVarArgs iv1 = gi.arg2intvarargs(call->arg(2)); Region re; IntSet cover_s(cover); Gecode::IntSetRanges cover_r(cover_s); auto* iv0_ri = re.alloc(iv0.size()); for (int i = iv0.size(); (i--) != 0;) { iv0_ri[i] = Gecode::IntVarRanges(iv0[i]); } Iter::Ranges::NaryUnion iv0_r(re, iv0_ri, iv0.size()); Iter::Ranges::Diff extra_r(iv0_r, cover_r); Iter::Ranges::ToValues > extra( extra_r); for (; extra(); ++extra) { cover << extra.val(); iv1 << IntVar(*gi.currentSpace, 0, iv0.size()); } MZ_IntConLevel icl = GecodeSolverInstance::ann2icl(ann); if (icl == MZ_ICL_DOM) { IntVarArgs allvars = iv0 + iv1; unshare(*gi.currentSpace, allvars); count(*gi.currentSpace, allvars.slice(0, 1, iv0.size()), allvars.slice(iv0.size(), 1, iv1.size()), cover, GecodeSolverInstance::ann2icl(ann)); } else { unshare(*gi.currentSpace, iv0); count(*gi.currentSpace, iv0, iv1, cover, GecodeSolverInstance::ann2icl(ann)); } } void p_global_cardinality_closed(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs iv0 = gi.arg2intvarargs(call->arg(0)); IntArgs cover = GecodeSolverInstance::arg2intargs(call->arg(1)); IntVarArgs iv1 = gi.arg2intvarargs(call->arg(2)); unshare(*gi.currentSpace, iv0); count(*gi.currentSpace, iv0, iv1, cover, GecodeSolverInstance::ann2icl(ann)); } void p_global_cardinality_low_up(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs x = gi.arg2intvarargs(call->arg(0)); IntArgs cover = GecodeSolverInstance::arg2intargs(call->arg(1)); IntArgs lbound = GecodeSolverInstance::arg2intargs(call->arg(2)); IntArgs ubound = GecodeSolverInstance::arg2intargs(call->arg(3)); IntSetArgs y(cover.size()); for (int i = cover.size(); (i--) != 0;) { y[i] = IntSet(lbound[i], ubound[i]); } IntSet cover_s(cover); Region re; auto* xrs = re.alloc(x.size()); for (int i = x.size(); (i--) != 0;) { xrs[i].init(x[i]); } Iter::Ranges::NaryUnion u(re, xrs, x.size()); Iter::Ranges::ToValues uv(u); for (; uv(); ++uv) { if (!cover_s.in(uv.val())) { cover << uv.val(); y << IntSet(0, x.size()); } } unshare(*gi.currentSpace, x); count(*gi.currentSpace, x, y, cover, GecodeSolverInstance::ann2icl(ann)); } void p_global_cardinality_low_up_closed(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs x = gi.arg2intvarargs(call->arg(0)); IntArgs cover = GecodeSolverInstance::arg2intargs(call->arg(1)); IntArgs lbound = GecodeSolverInstance::arg2intargs(call->arg(2)); IntArgs ubound = GecodeSolverInstance::arg2intargs(call->arg(3)); IntSetArgs y(cover.size()); for (int i = cover.size(); (i--) != 0;) { y[i] = IntSet(lbound[i], ubound[i]); } unshare(*gi.currentSpace, x); count(*gi.currentSpace, x, y, cover, GecodeSolverInstance::ann2icl(ann)); } void p_minimum(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs iv = gi.arg2intvarargs(call->arg(1)); min(*gi.currentSpace, iv, gi.arg2intvar(call->arg(0)), GecodeSolverInstance::ann2icl(ann)); } void p_maximum(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs iv = gi.arg2intvarargs(call->arg(1)); max(*gi.currentSpace, iv, gi.arg2intvar(call->arg(0)), GecodeSolverInstance::ann2icl(ann)); } void p_maximum_arg(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs iv = gi.arg2intvarargs(call->arg(0)); argmax(*gi.currentSpace, iv, gi.arg2intvar(call->arg(1)), true, GecodeSolverInstance::ann2icl(ann)); } void p_minimum_arg(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs iv = gi.arg2intvarargs(call->arg(0)); argmin(*gi.currentSpace, iv, gi.arg2intvar(call->arg(1)), true, GecodeSolverInstance::ann2icl(ann)); } void p_regular(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs iv = gi.arg2intvarargs(call->arg(0)); int q = static_cast(call->arg(1)->cast()->v().toInt()); int symbols = static_cast(call->arg(2)->cast()->v().toInt()); IntArgs d = GecodeSolverInstance::arg2intargs(call->arg(3)); int q0 = static_cast(call->arg(4)->cast()->v().toInt()); int noOfTrans = 0; for (int i = 1; i <= q; i++) { for (int j = 1; j <= symbols; j++) { if (d[(i - 1) * symbols + (j - 1)] > 0) { noOfTrans++; } } } Region re; auto* t = re.alloc(noOfTrans + 1); noOfTrans = 0; for (int i = 1; i <= q; i++) { for (int j = 1; j <= symbols; j++) { if (d[(i - 1) * symbols + (j - 1)] > 0) { t[noOfTrans].i_state = i; t[noOfTrans].symbol = j; t[noOfTrans].o_state = d[(i - 1) * symbols + (j - 1)]; noOfTrans++; } } } t[noOfTrans].i_state = -1; // Final states IntSetVal* isv = eval_intset(s.env().envi(), call->arg(5)); IntSetRanges isr(isv); int* f = static_cast(malloc(sizeof(int) * (isv->card().toInt()) + 1)); int i = 0; for (Ranges::ToValues val_iter(isr); val_iter(); ++val_iter, ++i) { f[i] = static_cast(val_iter.val().toInt()); } f[i] = -1; DFA dfa(q0, t, f); free(f); unshare(*gi.currentSpace, iv); extensional(*gi.currentSpace, iv, dfa, GecodeSolverInstance::ann2icl(ann)); } void p_sort(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs x = gi.arg2intvarargs(call->arg(0)); IntVarArgs y = gi.arg2intvarargs(call->arg(1)); IntVarArgs xy(x.size() + y.size()); for (int i = x.size(); (i--) != 0;) { xy[i] = x[i]; } for (int i = y.size(); (i--) != 0;) { xy[i + x.size()] = y[i]; } unshare(*gi.currentSpace, xy); for (int i = x.size(); (i--) != 0;) { x[i] = xy[i]; } for (int i = y.size(); (i--) != 0;) { y[i] = xy[i + x.size()]; } sorted(*gi.currentSpace, x, y, GecodeSolverInstance::ann2icl(ann)); } void p_inverse_offsets(SolverInstanceBase& s, const Call* call) { auto& gi = static_cast(s); IntVarArgs x = gi.arg2intvarargs(call->arg(0)); unshare(*gi.currentSpace, x); int xoff = static_cast(call->arg(1)->cast()->v().toInt()); IntVarArgs y = gi.arg2intvarargs(call->arg(2)); unshare(*gi.currentSpace, y); int yoff = static_cast(call->arg(3)->cast()->v().toInt()); MZ_IntConLevel icl = GecodeSolverInstance::ann2icl(call->ann()); channel(*gi.currentSpace, x, xoff, y, yoff, icl == MZ_ICL_DEF ? MZ_ICL_DOM : icl); } void p_increasing_int(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs x = gi.arg2intvarargs(call->arg(0)); rel(*gi.currentSpace, x, IRT_LQ, GecodeSolverInstance::ann2icl(ann)); } void p_increasing_bool(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BoolVarArgs x = gi.arg2boolvarargs(call->arg(0)); rel(*gi.currentSpace, x, IRT_LQ, GecodeSolverInstance::ann2icl(ann)); } void p_decreasing_int(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs x = gi.arg2intvarargs(call->arg(0)); rel(*gi.currentSpace, x, IRT_GQ, GecodeSolverInstance::ann2icl(ann)); } void p_decreasing_bool(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BoolVarArgs x = gi.arg2boolvarargs(call->arg(0)); rel(*gi.currentSpace, x, IRT_GQ, GecodeSolverInstance::ann2icl(ann)); } void p_table_int(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs x = gi.arg2intvarargs(call->arg(0)); IntArgs tuples = GecodeSolverInstance::arg2intargs(call->arg(1)); int noOfVars = x.size(); int noOfTuples = tuples.size() == 0 ? 0 : (tuples.size() / noOfVars); TupleSet ts(noOfVars); for (int i = 0; i < noOfTuples; i++) { IntArgs t(noOfVars); for (int j = 0; j < x.size(); j++) { t[j] = tuples[i * noOfVars + j]; } ts.add(t); } ts.finalize(); extensional(*gi.currentSpace, x, ts, GecodeSolverInstance::ann2icl(ann)); } void p_table_bool(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BoolVarArgs x = gi.arg2boolvarargs(call->arg(0)); IntArgs tuples = GecodeSolverInstance::arg2boolargs(call->arg(1)); int noOfVars = x.size(); int noOfTuples = tuples.size() == 0 ? 0 : (tuples.size() / noOfVars); TupleSet ts(noOfVars); for (int i = 0; i < noOfTuples; i++) { IntArgs t(noOfVars); for (int j = 0; j < x.size(); j++) { t[j] = tuples[i * noOfVars + j]; } ts.add(t); } ts.finalize(); extensional(*gi.currentSpace, x, ts, GecodeSolverInstance::ann2icl(ann)); } void p_cumulatives(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs start = gi.arg2intvarargs(call->arg(0)); IntVarArgs duration = gi.arg2intvarargs(call->arg(1)); IntVarArgs height = gi.arg2intvarargs(call->arg(2)); int n = start.size(); IntVar bound = gi.arg2intvar(call->arg(3)); int minHeight = INT_MAX; int minHeight2 = INT_MAX; for (int i = n; (i--) != 0;) { if (height[i].min() < minHeight) { minHeight = height[i].min(); } else if (height[i].min() < minHeight2) { minHeight2 = height[i].min(); } } bool disjunctive = (minHeight > bound.max() / 2) || (minHeight2 > bound.max() / 2 && minHeight + minHeight2 > bound.max()); if (disjunctive) { rel(*gi.currentSpace, bound >= max(height)); // Unary if (duration.assigned()) { IntArgs durationI(n); for (int i = n; (i--) != 0;) { durationI[i] = duration[i].val(); } unshare(*gi.currentSpace, start); unary(*gi.currentSpace, start, durationI); } else { IntVarArgs end(n); for (int i = n; (i--) != 0;) { end[i] = expr(*gi.currentSpace, start[i] + duration[i]); } unshare(*gi.currentSpace, start); unary(*gi.currentSpace, start, duration, end); } } else if (height.assigned()) { IntArgs heightI(n); for (int i = n; (i--) != 0;) { heightI[i] = height[i].val(); } if (duration.assigned()) { IntArgs durationI(n); for (int i = n; (i--) != 0;) { durationI[i] = duration[i].val(); } cumulative(*gi.currentSpace, bound, start, durationI, heightI); } else { IntVarArgs end(n); for (int i = n; (i--) != 0;) { end[i] = expr(*gi.currentSpace, start[i] + duration[i]); } cumulative(*gi.currentSpace, bound, start, duration, end, heightI); } } else if (bound.assigned()) { IntArgs machine = IntArgs::create(n, 0, 0); IntArgs limit({bound.val()}); IntVarArgs end(n); for (int i = n; (i--) != 0;) { end[i] = expr(*gi.currentSpace, start[i] + duration[i]); } cumulatives(*gi.currentSpace, machine, start, duration, end, height, limit, true, GecodeSolverInstance::ann2icl(ann)); } else { int min = Gecode::Int::Limits::max; int max = Gecode::Int::Limits::min; IntVarArgs end(start.size()); for (int i = start.size(); (i--) != 0;) { min = std::min(min, start[i].min()); max = std::max(max, start[i].max() + duration[i].max()); end[i] = expr(*gi.currentSpace, start[i] + duration[i]); } for (int time = min; time < max; ++time) { IntVarArgs x(start.size()); for (int i = start.size(); (i--) != 0;) { IntVar overlaps = channel(*gi.currentSpace, expr(*gi.currentSpace, (start[i] <= time) && (time < end[i]))); x[i] = expr(*gi.currentSpace, overlaps * height[i]); } linear(*gi.currentSpace, x, IRT_LQ, bound); } } } void p_among_seq_int(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); Gecode::IntVarArgs x = gi.arg2intvarargs(call->arg(0)); IntSet S = gi.arg2intset(s.env().envi(), call->arg(1)); int q = static_cast(call->arg(2)->cast()->v().toInt()); int l = static_cast(call->arg(3)->cast()->v().toInt()); int u = static_cast(call->arg(4)->cast()->v().toInt()); unshare(*gi.currentSpace, x); sequence(*gi.currentSpace, x, S, q, l, u, GecodeSolverInstance::ann2icl(ann)); } void p_among_seq_bool(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BoolVarArgs x = gi.arg2boolvarargs(call->arg(0)); bool val = call->arg(1)->cast()->v(); int q = static_cast(call->arg(2)->cast()->v().toInt()); int l = static_cast(call->arg(3)->cast()->v().toInt()); int u = static_cast(call->arg(4)->cast()->v().toInt()); IntSet S(static_cast(val), static_cast(val)); unshare(*gi.currentSpace, x); sequence(*gi.currentSpace, x, S, q, l, u, GecodeSolverInstance::ann2icl(ann)); } void p_schedule_unary(SolverInstanceBase& s, const Call* call) { auto& gi = static_cast(s); IntVarArgs x = gi.arg2intvarargs(call->arg(0)); IntArgs p = GecodeSolverInstance::arg2intargs(call->arg(1)); unshare(*gi.currentSpace, x); unary(*gi.currentSpace, x, p); } void p_schedule_unary_optional(SolverInstanceBase& s, const Call* call) { auto& gi = static_cast(s); IntVarArgs x = gi.arg2intvarargs(call->arg(0)); IntArgs p = GecodeSolverInstance::arg2intargs(call->arg(1)); BoolVarArgs m = gi.arg2boolvarargs(call->arg(2)); unshare(*gi.currentSpace, x); unary(*gi.currentSpace, x, p, m); } void p_cumulative_opt(SolverInstanceBase& s, const Call* ce) { const Annotation& ann = ce->ann(); auto& gi = static_cast(s); IntVarArgs start = gi.arg2intvarargs(ce->arg(0)); IntArgs duration = GecodeSolverInstance::arg2intargs(ce->arg(1)); IntArgs height = GecodeSolverInstance::arg2intargs(ce->arg(2)); BoolVarArgs opt = gi.arg2boolvarargs(ce->arg(3)); int bound = static_cast(ce->arg(4)->cast()->v().toInt()); unshare(*gi.currentSpace, start); cumulative(*gi.currentSpace, bound, start, duration, height, opt, GecodeSolverInstance::ann2icl(ann)); } void p_circuit(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); int off = static_cast(call->arg(0)->cast()->v().toInt()); IntVarArgs xv = gi.arg2intvarargs(call->arg(1)); unshare(*gi.currentSpace, xv); circuit(*gi.currentSpace, off, xv, GecodeSolverInstance::ann2icl(ann)); } void p_circuit_cost_array(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntArgs c = GecodeSolverInstance::arg2intargs(call->arg(0)); IntVarArgs xv = gi.arg2intvarargs(call->arg(1)); IntVarArgs yv = gi.arg2intvarargs(call->arg(2)); IntVar z = gi.arg2intvar(call->arg(3)); unshare(*gi.currentSpace, xv); circuit(*gi.currentSpace, c, xv, yv, z, GecodeSolverInstance::ann2icl(ann)); } void p_circuit_cost(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntArgs c = GecodeSolverInstance::arg2intargs(call->arg(0)); IntVarArgs xv = gi.arg2intvarargs(call->arg(1)); IntVar z = gi.arg2intvar(call->arg(2)); unshare(*gi.currentSpace, xv); circuit(*gi.currentSpace, c, xv, z, GecodeSolverInstance::ann2icl(ann)); } void p_nooverlap(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs x0 = gi.arg2intvarargs(call->arg(0)); IntVarArgs w = gi.arg2intvarargs(call->arg(1)); IntVarArgs y0 = gi.arg2intvarargs(call->arg(2)); IntVarArgs h = gi.arg2intvarargs(call->arg(3)); if (w.assigned() && h.assigned()) { IntArgs iw(w.size()); for (int i = w.size(); (i--) != 0;) { iw[i] = w[i].val(); } IntArgs ih(h.size()); for (int i = h.size(); (i--) != 0;) { ih[i] = h[i].val(); } nooverlap(*gi.currentSpace, x0, iw, y0, ih, GecodeSolverInstance::ann2icl(ann)); } else { IntVarArgs x1(x0.size()); IntVarArgs y1(y0.size()); for (int i = x0.size(); (i--) != 0;) { x1[i] = expr(*gi.currentSpace, x0[i] + w[i]); } for (int i = y0.size(); (i--) != 0;) { y1[i] = expr(*gi.currentSpace, y0[i] + h[i]); } nooverlap(*gi.currentSpace, x0, w, x1, y0, h, y1, GecodeSolverInstance::ann2icl(ann)); } } void p_precede(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs x = gi.arg2intvarargs(call->arg(0)); int p_s = static_cast(call->arg(1)->cast()->v().toInt()); int p_t = static_cast(call->arg(2)->cast()->v().toInt()); precede(*gi.currentSpace, x, p_s, p_t, GecodeSolverInstance::ann2icl(ann)); } void p_nvalue(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs x = gi.arg2intvarargs(call->arg(1)); if (call->arg(0)->type().isvarint()) { IntVar y = gi.arg2intvar(call->arg(0)); nvalues(*gi.currentSpace, x, IRT_EQ, y, GecodeSolverInstance::ann2icl(ann)); } else { nvalues(*gi.currentSpace, x, IRT_EQ, static_cast(call->arg(0)->cast()->v().toInt()), GecodeSolverInstance::ann2icl(ann)); } } void p_among(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs x = gi.arg2intvarargs(call->arg(1)); IntSet v = gi.arg2intset(s.env().envi(), call->arg(2)); if (call->arg(0)->type().isvarint()) { IntVar n = gi.arg2intvar(call->arg(0)); unshare(*gi.currentSpace, x); count(*gi.currentSpace, x, v, IRT_EQ, n, GecodeSolverInstance::ann2icl(ann)); } else { unshare(*gi.currentSpace, x); count(*gi.currentSpace, x, v, IRT_EQ, static_cast(call->arg(0)->cast()->v().toInt()), GecodeSolverInstance::ann2icl(ann)); } } void p_member_int(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs x = gi.arg2intvarargs(call->arg(0)); IntVar y = gi.arg2intvar(call->arg(1)); member(*gi.currentSpace, x, y, GecodeSolverInstance::ann2icl(ann)); } void p_member_int_reif(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); IntVarArgs x = gi.arg2intvarargs(call->arg(0)); IntVar y = gi.arg2intvar(call->arg(1)); BoolVar b = gi.arg2boolvar(call->arg(2)); member(*gi.currentSpace, x, y, b, GecodeSolverInstance::ann2icl(ann)); } void p_member_bool(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BoolVarArgs x = gi.arg2boolvarargs(call->arg(0)); BoolVar y = gi.arg2boolvar(call->arg(1)); member(*gi.currentSpace, x, y, GecodeSolverInstance::ann2icl(ann)); } void p_member_bool_reif(SolverInstanceBase& s, const Call* call) { const Annotation& ann = call->ann(); auto& gi = static_cast(s); BoolVarArgs x = gi.arg2boolvarargs(call->arg(0)); BoolVar y = gi.arg2boolvar(call->arg(1)); member(*gi.currentSpace, x, y, gi.arg2boolvar(call->arg(2)), GecodeSolverInstance::ann2icl(ann)); } // FLOAT CONSTRAINTS #ifdef GECODE_HAS_FLOAT_VARS void p_int2float(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); IntVar x0 = gi.arg2intvar(ce->arg(0)); FloatVar x1 = gi.arg2floatvar(ce->arg(1)); channel(*gi.currentSpace, x0, x1); } void p_float_lin_cmp(GecodeSolverInstance& s, FloatRelType frt, const Call* ce) { FloatValArgs fa = GecodeSolverInstance::arg2floatargs(ce->arg(0)); FloatVarArgs fv = s.arg2floatvarargs(ce->arg(1)); linear(*s.currentSpace, fa, fv, frt, ce->arg(2)->cast()->v().toDouble()); } void p_float_lin_cmp_reif(GecodeSolverInstance& s, FloatRelType frt, const Call* ce) { FloatValArgs fa = GecodeSolverInstance::arg2floatargs(ce->arg(0)); FloatVarArgs fv = s.arg2floatvarargs(ce->arg(1)); linear(*s.currentSpace, fa, fv, frt, ce->arg(2)->cast()->v().toDouble(), s.arg2boolvar(ce->arg(3))); } void p_float_lin_eq(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); p_float_lin_cmp(gi, FRT_EQ, ce); } void p_float_lin_eq_reif(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); p_float_lin_cmp_reif(gi, FRT_EQ, ce); } void p_float_lin_le(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); p_float_lin_cmp(gi, FRT_LQ, ce); } void p_float_lin_le_reif(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); p_float_lin_cmp_reif(gi, FRT_LQ, ce); } void p_float_times(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); FloatVar x = gi.arg2floatvar(ce->arg(0)); FloatVar y = gi.arg2floatvar(ce->arg(1)); FloatVar z = gi.arg2floatvar(ce->arg(2)); mult(*gi.currentSpace, x, y, z); } void p_float_div(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); FloatVar x = gi.arg2floatvar(ce->arg(0)); FloatVar y = gi.arg2floatvar(ce->arg(1)); FloatVar z = gi.arg2floatvar(ce->arg(2)); div(*gi.currentSpace, x, y, z); } void p_float_plus(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); FloatVar x = gi.arg2floatvar(ce->arg(0)); FloatVar y = gi.arg2floatvar(ce->arg(1)); FloatVar z = gi.arg2floatvar(ce->arg(2)); rel(*gi.currentSpace, x + y == z); } void p_float_sqrt(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); FloatVar x = gi.arg2floatvar(ce->arg(0)); FloatVar y = gi.arg2floatvar(ce->arg(1)); sqrt(*gi.currentSpace, x, y); } void p_float_abs(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); FloatVar x = gi.arg2floatvar(ce->arg(0)); FloatVar y = gi.arg2floatvar(ce->arg(1)); abs(*gi.currentSpace, x, y); } void p_float_eq(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); FloatVar x = gi.arg2floatvar(ce->arg(0)); FloatVar y = gi.arg2floatvar(ce->arg(1)); rel(*gi.currentSpace, x, FRT_EQ, y); } void p_float_eq_reif(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); FloatVar x = gi.arg2floatvar(ce->arg(0)); FloatVar y = gi.arg2floatvar(ce->arg(1)); BoolVar b = gi.arg2boolvar(ce->arg(2)); rel(*gi.currentSpace, x, FRT_EQ, y, b); } void p_float_le(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); FloatVar x = gi.arg2floatvar(ce->arg(0)); FloatVar y = gi.arg2floatvar(ce->arg(1)); rel(*gi.currentSpace, x, FRT_LQ, y); } void p_float_le_reif(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); FloatVar x = gi.arg2floatvar(ce->arg(0)); FloatVar y = gi.arg2floatvar(ce->arg(1)); BoolVar b = gi.arg2boolvar(ce->arg(2)); rel(*gi.currentSpace, x, FRT_LQ, y, b); } void p_float_max(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); FloatVar x = gi.arg2floatvar(ce->arg(0)); FloatVar y = gi.arg2floatvar(ce->arg(1)); FloatVar z = gi.arg2floatvar(ce->arg(2)); max(*gi.currentSpace, x, y, z); } void p_float_min(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); FloatVar x = gi.arg2floatvar(ce->arg(0)); FloatVar y = gi.arg2floatvar(ce->arg(1)); FloatVar z = gi.arg2floatvar(ce->arg(2)); min(*gi.currentSpace, x, y, z); } void p_float_lt(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); FloatVar x = gi.arg2floatvar(ce->arg(0)); FloatVar y = gi.arg2floatvar(ce->arg(1)); rel(*gi.currentSpace, x, FRT_LQ, y); rel(*gi.currentSpace, x, FRT_EQ, y, BoolVar(*gi.currentSpace, 0, 0)); } void p_float_lt_reif(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); FloatVar x = gi.arg2floatvar(ce->arg(0)); FloatVar y = gi.arg2floatvar(ce->arg(1)); BoolVar b = gi.arg2boolvar(ce->arg(2)); BoolVar b0(*gi.currentSpace, 0, 1); BoolVar b1(*gi.currentSpace, 0, 1); rel(*gi.currentSpace, b == (b0 && !b1)); rel(*gi.currentSpace, x, FRT_LQ, y, b0); rel(*gi.currentSpace, x, FRT_EQ, y, b1); } void p_float_ne(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); FloatVar x = gi.arg2floatvar(ce->arg(0)); FloatVar y = gi.arg2floatvar(ce->arg(1)); rel(*gi.currentSpace, x, FRT_EQ, y, BoolVar(*gi.currentSpace, 0, 0)); } #ifdef GECODE_HAS_MPFR #define P_FLOAT_OP(Op) \ void p_float_##Op(SolverInstanceBase& s, const Call* ce) { \ GecodeSolverInstance& gi = static_cast(s); \ FloatVar x = gi.arg2floatvar(ce->arg(0)); \ FloatVar y = gi.arg2floatvar(ce->arg(1)); \ Op(*gi.currentSpace, x, y); \ } P_FLOAT_OP(acos) P_FLOAT_OP(asin) P_FLOAT_OP(atan) P_FLOAT_OP(cos) P_FLOAT_OP(exp) P_FLOAT_OP(sin) P_FLOAT_OP(tan) // P_FLOAT_OP(sinh) // P_FLOAT_OP(tanh) // P_FLOAT_OP(cosh) #undef P_FLOAT_OP void p_float_ln(SolverInstanceBase& s, const Call* ce) { GecodeSolverInstance& gi = static_cast(s); FloatVar x = gi.arg2floatvar(ce->arg(0)); FloatVar y = gi.arg2floatvar(ce->arg(1)); log(*gi.currentSpace, x, y); } void p_float_log10(SolverInstanceBase& s, const Call* ce) { GecodeSolverInstance& gi = static_cast(s); FloatVar x = gi.arg2floatvar(ce->arg(0)); FloatVar y = gi.arg2floatvar(ce->arg(1)); log(*gi.currentSpace, 10.0, x, y); } void p_float_log2(SolverInstanceBase& s, const Call* ce) { GecodeSolverInstance& gi = static_cast(s); FloatVar x = gi.arg2floatvar(ce->arg(0)); FloatVar y = gi.arg2floatvar(ce->arg(1)); log(*gi.currentSpace, 2.0, x, y); } #endif #endif #ifdef GECODE_HAS_SET_VARS void p_set_op(SolverInstanceBase& s, SetOpType op, const Call* ce) { auto& gi = static_cast(s); rel(*gi.currentSpace, gi.arg2setvar(ce->arg(0)), op, gi.arg2setvar(ce->arg(1)), SRT_EQ, gi.arg2setvar(ce->arg(2))); } void p_set_union(SolverInstanceBase& s, const Call* ce) { p_set_op(s, SOT_UNION, ce); } void p_set_intersect(SolverInstanceBase& s, const Call* ce) { p_set_op(s, SOT_INTER, ce); } void p_set_diff(SolverInstanceBase& s, const Call* ce) { p_set_op(s, SOT_MINUS, ce); } void p_set_symdiff(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); SetVar x = gi.arg2setvar(ce->arg(0)); SetVar y = gi.arg2setvar(ce->arg(1)); SetVarLubRanges xub(x); IntSet xubs(xub); SetVar x_y(*gi.currentSpace, IntSet::empty, xubs); rel(*gi.currentSpace, x, SOT_MINUS, y, SRT_EQ, x_y); SetVarLubRanges yub(y); IntSet yubs(yub); SetVar y_x(*gi.currentSpace, IntSet::empty, yubs); rel(*gi.currentSpace, y, SOT_MINUS, x, SRT_EQ, y_x); rel(*gi.currentSpace, x_y, SOT_UNION, y_x, SRT_EQ, gi.arg2setvar(ce->arg(2))); } void p_array_set_op(SolverInstanceBase& s, SetOpType op, const Call* ce) { auto& gi = static_cast(s); SetVarArgs xs = gi.arg2setvarargs(ce->arg(0)); rel(*gi.currentSpace, op, xs, gi.arg2setvar(ce->arg(1))); } void p_array_set_union(SolverInstanceBase& s, const Call* ce) { p_array_set_op(s, SOT_UNION, ce); } void p_array_set_partition(SolverInstanceBase& s, const Call* ce) { p_array_set_op(s, SOT_DUNION, ce); } void p_set_rel(SolverInstanceBase& s, SetRelType srt, const Call* ce) { auto& gi = static_cast(s); rel(*gi.currentSpace, gi.arg2setvar(ce->arg(0)), srt, gi.arg2setvar(ce->arg(1))); } void p_set_eq(SolverInstanceBase& s, const Call* ce) { p_set_rel(s, SRT_EQ, ce); } void p_set_ne(SolverInstanceBase& s, const Call* ce) { p_set_rel(s, SRT_NQ, ce); } void p_set_subset(SolverInstanceBase& s, const Call* ce) { p_set_rel(s, SRT_SUB, ce); } void p_set_superset(SolverInstanceBase& s, const Call* ce) { p_set_rel(s, SRT_SUP, ce); } void p_set_le(SolverInstanceBase& s, const Call* ce) { p_set_rel(s, SRT_LQ, ce); } void p_set_lt(SolverInstanceBase& s, const Call* ce) { p_set_rel(s, SRT_LE, ce); } void p_set_card(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); if (!ce->arg(1)->type().isvar()) { IntVal i = ce->arg(1)->cast()->v().toInt(); assert(i >= 0 && i <= Gecode::Int::Limits::max); auto ui = static_cast(i.toInt()); cardinality(*gi.currentSpace, gi.arg2setvar(ce->arg(0)), ui, ui); } else { cardinality(*gi.currentSpace, gi.arg2setvar(ce->arg(0)), gi.arg2intvar(ce->arg(1))); } } void p_set_in(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); if (!ce->arg(1)->type().isvar()) { IntSet d = gi.arg2intset(s.env().envi(), ce->arg(1)); if (ce->arg(0)->type().isvar()) { Gecode::IntSetRanges dr(d); Iter::Ranges::Singleton sr(0, 1); Iter::Ranges::Inter i(dr, sr); IntSet d01(i); if (d01.size() == 0) { gi.currentSpace->fail(); } else { rel(*gi.currentSpace, gi.arg2boolvar(ce->arg(0)), IRT_GQ, d01.min()); rel(*gi.currentSpace, gi.arg2boolvar(ce->arg(0)), IRT_LQ, d01.max()); } } else { dom(*gi.currentSpace, gi.arg2intvar(ce->arg(0)), d); } } else { if (!ce->arg(0)->type().isvar()) { dom(*gi.currentSpace, gi.arg2setvar(ce->arg(1)), SRT_SUP, static_cast(ce->arg(0)->cast()->v().toInt())); } else { rel(*gi.currentSpace, gi.arg2setvar(ce->arg(1)), SRT_SUP, gi.arg2intvar(ce->arg(0))); } } } void p_set_rel_reif(SolverInstanceBase& s, SetRelType srt, const Call* ce) { auto& gi = static_cast(s); rel(*gi.currentSpace, gi.arg2setvar(ce->arg(0)), srt, gi.arg2setvar(ce->arg(1)), gi.arg2boolvar(ce->arg(2))); } void p_set_eq_reif(SolverInstanceBase& s, const Call* ce) { p_set_rel_reif(s, SRT_EQ, ce); } void p_set_le_reif(SolverInstanceBase& s, const Call* ce) { p_set_rel_reif(s, SRT_LQ, ce); } void p_set_lt_reif(SolverInstanceBase& s, const Call* ce) { p_set_rel_reif(s, SRT_LE, ce); } void p_set_ne_reif(SolverInstanceBase& s, const Call* ce) { p_set_rel_reif(s, SRT_NQ, ce); } void p_set_subset_reif(SolverInstanceBase& s, const Call* ce) { p_set_rel_reif(s, SRT_SUB, ce); } void p_set_superset_reif(SolverInstanceBase& s, const Call* ce) { p_set_rel_reif(s, SRT_SUP, ce); } void p_set_in_reif(SolverInstanceBase& s, const Call* ce, ReifyMode rm) { auto& gi = static_cast(s); if (!ce->arg(1)->type().isvar()) { if (rm == RM_EQV) { p_int_in_reif(s, ce); } else { assert(rm == RM_IMP); p_int_in_imp(s, ce); } } else { if (!ce->arg(0)->type().isvar()) { dom(*gi.currentSpace, gi.arg2setvar(ce->arg(1)), SRT_SUP, static_cast(ce->arg(0)->cast()->v().toInt()), Reify(gi.arg2boolvar(ce->arg(2)), rm)); } else { rel(*gi.currentSpace, gi.arg2setvar(ce->arg(1)), SRT_SUP, gi.arg2intvar(ce->arg(0)), Reify(gi.arg2boolvar(ce->arg(2)), rm)); } } } void p_set_in_reif(SolverInstanceBase& s, const Call* ce) { p_set_in_reif(s, ce, RM_EQV); } void p_set_in_imp(SolverInstanceBase& s, const Call* ce) { p_set_in_reif(s, ce, RM_IMP); } void p_set_disjoint(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); rel(*gi.currentSpace, gi.arg2setvar(ce->arg(0)), SRT_DISJ, gi.arg2setvar(ce->arg(1))); } void p_link_set_to_booleans(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); SetVar x = gi.arg2setvar(ce->arg(0)); int idx = static_cast(ce->arg(2)->cast()->v().toInt()); assert(idx >= 0); rel(*gi.currentSpace, x || IntSet(Set::Limits::min, idx - 1)); BoolVarArgs y = gi.arg2boolvarargs(ce->arg(1), idx); unshare(*gi.currentSpace, y); channel(*gi.currentSpace, y, x); } void p_array_set_element(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); bool isConstant = true; ArrayLit* a = gi.arg2arraylit(ce->arg(1)); for (unsigned int i = a->size(); (i--) != 0U;) { if ((*a)[i]->type().isvar()) { isConstant = false; break; } } IntVar selector = gi.arg2intvar(ce->arg(0)); rel(*gi.currentSpace, selector > 0); if (isConstant) { IntSetArgs sv = gi.arg2intsetargs(gi.env().envi(), ce->arg(1), 1); element(*gi.currentSpace, sv, selector, gi.arg2setvar(ce->arg(2))); } else { SetVarArgs sv = gi.arg2setvarargs(ce->arg(1), 1); element(*gi.currentSpace, sv, selector, gi.arg2setvar(ce->arg(2))); } } void p_array_set_element_op(SolverInstanceBase& s, const Call* ce, SetOpType op, const IntSet& universe = IntSet(Set::Limits::min, Set::Limits::max)) { auto& gi = static_cast(s); bool isConstant = true; ArrayLit* a = gi.arg2arraylit(ce->arg(1)); for (unsigned int i = a->size(); (i--) != 0;) { if ((*a)[i]->type().isvar()) { isConstant = false; break; } } SetVar selector = gi.arg2setvar(ce->arg(0)); dom(*gi.currentSpace, selector, SRT_DISJ, 0); if (isConstant) { IntSetArgs sv = gi.arg2intsetargs(gi.env().envi(), ce->arg(1), 1); element(*gi.currentSpace, op, sv, selector, gi.arg2setvar(ce->arg(2)), universe); } else { SetVarArgs sv = gi.arg2setvarargs(ce->arg(1), 1); element(*gi.currentSpace, op, sv, selector, gi.arg2setvar(ce->arg(2)), universe); } } void p_array_set_element_union(SolverInstanceBase& s, const Call* ce) { p_array_set_element_op(s, ce, SOT_UNION); } void p_array_set_element_intersect(SolverInstanceBase& s, const Call* ce) { p_array_set_element_op(s, ce, SOT_INTER); } void p_array_set_element_intersect_in(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); IntSet d = gi.arg2intset(gi.env().envi(), ce->arg(3)); p_array_set_element_op(s, ce, SOT_INTER, d); } void p_array_set_element_partition(SolverInstanceBase& s, const Call* ce) { p_array_set_element_op(s, ce, SOT_DUNION); } void p_set_convex(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); convex(*gi.currentSpace, gi.arg2setvar(ce->arg(0))); } void p_array_set_seq(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); SetVarArgs sv = gi.arg2setvarargs(ce->arg(0)); sequence(*gi.currentSpace, sv); } void p_array_set_seq_union(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); SetVarArgs sv = gi.arg2setvarargs(ce->arg(0)); sequence(*gi.currentSpace, sv, gi.arg2setvar(ce->arg(1))); } void p_int_set_channel(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); int xoff = static_cast(ce->arg(1)->cast()->v().toInt()); assert(xoff >= 0); int yoff = static_cast(ce->arg(3)->cast()->v().toInt()); assert(yoff >= 0); IntVarArgs xv = gi.arg2intvarargs(ce->arg(0), xoff); SetVarArgs yv = gi.arg2setvarargs(ce->arg(2), yoff, 1, IntSet(0, xoff - 1)); IntSet xd(yoff, yv.size() - 1); for (int i = xoff; i < xv.size(); i++) { dom(*gi.currentSpace, xv[i], xd); } IntSet yd(xoff, xv.size() - 1); for (int i = yoff; i < yv.size(); i++) { dom(*gi.currentSpace, yv[i], SRT_SUB, yd); } channel(*gi.currentSpace, xv, yv); } void p_range(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); int xoff = static_cast(ce->arg(1)->cast()->v().toInt()); assert(xoff >= 0); IntVarArgs xv = gi.arg2intvarargs(ce->arg(0), xoff); element(*gi.currentSpace, SOT_UNION, xv, gi.arg2setvar(ce->arg(2)), gi.arg2setvar(ce->arg(3))); } void p_weights(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); IntArgs e = GecodeSolverInstance::arg2intargs(ce->arg(0)); IntArgs w = GecodeSolverInstance::arg2intargs(ce->arg(1)); SetVar x = gi.arg2setvar(ce->arg(2)); IntVar y = gi.arg2intvar(ce->arg(3)); weights(*gi.currentSpace, e, w, x, y); } void p_inverse_set(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); int xoff = static_cast(ce->arg(2)->cast()->v().toInt()); int yoff = static_cast(ce->arg(3)->cast()->v().toInt()); SetVarArgs x = gi.arg2setvarargs(ce->arg(0), xoff); SetVarArgs y = gi.arg2setvarargs(ce->arg(1), yoff); channel(*gi.currentSpace, x, y); } void p_precede_set(SolverInstanceBase& s, const Call* ce) { auto& gi = static_cast(s); SetVarArgs x = gi.arg2setvarargs(ce->arg(0)); int p_s = static_cast(ce->arg(1)->cast()->v().toInt()); int p_t = static_cast(ce->arg(2)->cast()->v().toInt()); precede(*gi.currentSpace, x, p_s, p_t); } #endif } // namespace GecodeConstraints } // namespace MiniZinc