1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
/* -*- mode: C++; c-basic-offset: 2; indent-tabs-mode: nil -*- */
/*
* Main authors:
* Guido Tack <guido.tack@monash.edu>
*/
/* 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 <minizinc/flat_exp.hh>
namespace MiniZinc {
EE flatten_setlit(EnvI& env, const Ctx& ctx, Expression* e, VarDecl* r, VarDecl* b) {
CallStackItem _csi(env, e);
EE ret;
auto* sl = Expression::cast<SetLit>(e);
assert(sl->isv() == nullptr && sl->fsv() == nullptr);
std::vector<EE> elems_ee(sl->v().size());
for (unsigned int i = sl->v().size(); (i--) != 0U;) {
elems_ee[i] = flat_exp(env, ctx, sl->v()[i], nullptr, ctx.partialityVar(env));
}
std::vector<Expression*> elems(elems_ee.size());
bool allPar = true;
bool hadOpt = false;
for (auto i = static_cast<unsigned int>(elems.size()); (i--) != 0U;) {
elems[i] = elems_ee[i].r();
allPar = allPar && Expression::type(elems[i]).isPar();
hadOpt = hadOpt || Expression::type(elems[i]).isOpt();
}
ret.b = conj(env, b, Ctx(), elems_ee);
if (allPar) {
GCLock lock;
auto* nsl = new SetLit(Location().introduce(), elems);
Type nsl_t(Expression::type(e));
nsl_t.ti(Type::TI_PAR);
nsl->type(nsl_t);
Expression* ee = eval_set_lit(env, nsl);
ret.r = bind(env, Ctx(), r, ee);
} else {
GCLock lock;
auto* al = new ArrayLit(Expression::loc(sl), elems);
Type al_t = Type::varint(1);
if (hadOpt) {
al_t.ot(Type::OT_OPTIONAL);
}
al->type(al_t);
std::vector<Expression*> args(1);
args[0] = al;
Call* cc = Call::a(Expression::loc(sl).introduce(), "array2set", args);
cc->type(Type::varsetint());
FunctionI* fi = env.model->matchFn(env, cc->id(), args, false);
if (fi == nullptr) {
throw FlatteningError(env, Expression::loc(cc), "cannot find matching declaration");
}
assert(fi);
assert(env.isSubtype(fi->rtype(env, args, nullptr, false), cc->type(), false));
cc->decl(fi);
EE ee = flat_exp(env, Ctx(), cc, nullptr, env.constants.varTrue);
ret.r = bind(env, Ctx(), r, ee.r());
}
return ret;
}
} // namespace MiniZinc
|
