1 #include <isl_constraint.h>
2 #include "isl_map_private.h"
4 static unsigned n(struct isl_constraint *c, enum isl_dim_type type)
6 return isl_basic_map_dim(c->bmap, type);
9 static unsigned offset(struct isl_constraint *c, enum isl_dim_type type)
11 struct isl_dim *dim = c->bmap->dim;
13 case isl_dim_param: return 1;
14 case isl_dim_in: return 1 + dim->nparam;
15 case isl_dim_out: return 1 + dim->nparam + dim->n_in;
16 case isl_dim_div: return 1 + dim->nparam + dim->n_in + dim->n_out;
20 static unsigned basic_set_offset(struct isl_basic_set *bset,
21 enum isl_dim_type type)
23 struct isl_dim *dim = bset->dim;
25 case isl_dim_param: return 1;
26 case isl_dim_in: return 1 + dim->nparam;
27 case isl_dim_out: return 1 + dim->nparam + dim->n_in;
28 case isl_dim_div: return 1 + dim->nparam + dim->n_in + dim->n_out;
32 struct isl_constraint *isl_basic_map_constraint(struct isl_basic_map *bmap,
35 struct isl_constraint *constraint;
40 constraint = isl_alloc_type(bmap->ctx, struct isl_constraint);
44 constraint->ctx = bmap->ctx;
45 isl_ctx_ref(constraint->ctx);
47 constraint->bmap = bmap;
48 constraint->line = line;
52 isl_basic_map_free(bmap);
56 struct isl_constraint *isl_basic_set_constraint(struct isl_basic_set *bset,
59 return isl_basic_map_constraint((struct isl_basic_map *)bset, line);
62 struct isl_constraint *isl_equality_alloc(struct isl_dim *dim)
64 struct isl_basic_map *bmap;
69 bmap = isl_basic_map_alloc_dim(dim, 0, 1, 0);
73 isl_basic_map_alloc_equality(bmap);
74 isl_seq_clr(bmap->eq[0], 1 + isl_basic_map_total_dim(bmap));
75 return isl_basic_map_constraint(bmap, &bmap->eq[0]);
78 struct isl_constraint *isl_inequality_alloc(struct isl_dim *dim)
80 struct isl_basic_map *bmap;
85 bmap = isl_basic_map_alloc_dim(dim, 0, 0, 1);
89 isl_basic_map_alloc_inequality(bmap);
90 isl_seq_clr(bmap->ineq[0], 1 + isl_basic_map_total_dim(bmap));
91 return isl_basic_map_constraint(bmap, &bmap->ineq[0]);
94 struct isl_constraint *isl_constraint_dup(struct isl_constraint *c)
99 return isl_basic_map_constraint(isl_basic_map_copy(c->bmap), c->line);
102 struct isl_constraint *isl_constraint_cow(struct isl_constraint *c)
110 return isl_constraint_dup(c);
113 struct isl_constraint *isl_constraint_copy(struct isl_constraint *constraint)
122 struct isl_constraint *isl_constraint_free(struct isl_constraint *c)
130 isl_basic_map_free(c->bmap);
131 isl_ctx_deref(c->ctx);
135 struct isl_constraint *isl_basic_set_first_constraint(
136 struct isl_basic_set *bset)
138 struct isl_constraint *c;
144 return isl_basic_set_constraint(bset, &bset->eq[0]);
146 if (bset->n_ineq > 0)
147 return isl_basic_set_constraint(bset, &bset->ineq[0]);
149 isl_basic_set_free(bset);
153 struct isl_constraint *isl_constraint_next(struct isl_constraint *c)
155 c = isl_constraint_cow(c);
156 if (c->line >= c->bmap->eq) {
158 if (c->line < c->bmap->eq + c->bmap->n_eq)
160 c->line = c->bmap->ineq;
163 if (c->line < c->bmap->ineq + c->bmap->n_ineq)
165 isl_constraint_free(c);
169 int isl_constraint_is_equal(struct isl_constraint *constraint1,
170 struct isl_constraint *constraint2)
172 if (!constraint1 || !constraint2)
174 return constraint1->bmap == constraint2->bmap &&
175 constraint1->line == constraint2->line;
178 struct isl_basic_map *isl_basic_map_add_constraint(
179 struct isl_basic_map *bmap, struct isl_constraint *constraint)
181 if (!bmap || !constraint)
184 isl_assert(constraint->ctx,
185 isl_dim_equal(bmap->dim, constraint->bmap->dim), goto error);
187 bmap = isl_basic_map_intersect(bmap,
188 isl_basic_map_copy(constraint->bmap));
189 isl_constraint_free(constraint);
192 isl_basic_map_free(bmap);
193 isl_constraint_free(constraint);
197 struct isl_basic_set *isl_basic_set_add_constraint(
198 struct isl_basic_set *bset, struct isl_constraint *constraint)
200 return (struct isl_basic_set *)
201 isl_basic_map_add_constraint((struct isl_basic_map *)bset,
205 struct isl_constraint *isl_constraint_add_div(struct isl_constraint *constraint,
206 struct isl_div *div, int *pos)
208 if (!constraint || !div)
211 isl_assert(constraint->ctx,
212 isl_dim_equal(div->bmap->dim, constraint->bmap->dim), goto error);
213 isl_assert(constraint->ctx,
214 constraint->bmap->n_eq + constraint->bmap->n_ineq == 1, goto error);
216 constraint->bmap = isl_basic_map_cow(constraint->bmap);
217 constraint->bmap = isl_basic_map_extend_dim(constraint->bmap,
218 isl_dim_copy(constraint->bmap->dim), 1, 0, 0);
219 if (!constraint->bmap)
221 constraint->line = &constraint->bmap->eq[0];
222 *pos = isl_basic_map_alloc_div(constraint->bmap);
225 isl_seq_cpy(constraint->bmap->div[*pos], div->line[0],
226 1 + 1 + isl_basic_map_total_dim(constraint->bmap));
230 isl_constraint_free(constraint);
235 int isl_constraint_dim(struct isl_constraint *constraint,
236 enum isl_dim_type type)
240 return n(constraint, type);
243 void isl_constraint_get_constant(struct isl_constraint *constraint, isl_int *v)
247 isl_int_set(*v, constraint->line[0][0]);
250 void isl_constraint_get_coefficient(struct isl_constraint *constraint,
251 enum isl_dim_type type, int pos, isl_int *v)
256 isl_assert(constraint->ctx, pos < n(constraint, type), return);
257 isl_int_set(*v, constraint->line[0][offset(constraint, type) + pos]);
260 struct isl_div *isl_constraint_div(struct isl_constraint *constraint, int pos)
265 isl_assert(constraint->ctx, pos < n(constraint, isl_dim_div),
267 return isl_basic_map_div(isl_basic_map_copy(constraint->bmap),
268 constraint->bmap->div+pos);
271 void isl_constraint_set_constant(struct isl_constraint *constraint, isl_int v)
275 isl_int_set(constraint->line[0][0], v);
278 void isl_constraint_set_coefficient(struct isl_constraint *constraint,
279 enum isl_dim_type type, int pos, isl_int v)
284 isl_assert(constraint->ctx, pos < n(constraint, type), return);
285 isl_int_set(constraint->line[0][offset(constraint, type) + pos], v);
288 void isl_constraint_clear(struct isl_constraint *constraint)
290 struct isl_basic_set *bset;
295 total = isl_basic_map_total_dim(constraint->bmap);
296 isl_seq_clr(constraint->line[0], 1 + total);
299 struct isl_constraint *isl_constraint_negate(struct isl_constraint *constraint)
306 isl_assert(constraint->ctx, !isl_constraint_is_equality(constraint),
308 isl_assert(constraint->ctx, constraint->bmap->ref == 1, goto error);
309 total = isl_basic_map_total_dim(constraint->bmap);
310 isl_seq_neg(constraint->line[0], constraint->line[0], 1 + total);
311 isl_int_sub_ui(constraint->line[0][0], constraint->line[0][0], 1);
312 ISL_F_CLR(constraint->bmap, ISL_BASIC_MAP_NORMALIZED);
315 isl_constraint_free(constraint);
319 int isl_constraint_is_equality(struct isl_constraint *constraint)
323 return constraint->line >= constraint->bmap->eq;
327 struct isl_basic_set *isl_basic_set_from_constraint(
328 struct isl_constraint *constraint)
331 struct isl_basic_set *constraint_bset, *bset;
340 isl_assert(constraint->ctx,n(constraint, isl_dim_in) == 0, goto error);
342 constraint_bset = (struct isl_basic_set *)constraint->bmap;
343 bset = isl_basic_set_universe_like(constraint_bset);
344 bset = isl_basic_set_align_divs(bset, constraint_bset);
345 nparam = isl_basic_set_n_param(bset);
346 dim = isl_basic_set_n_dim(bset);
347 bset = isl_basic_set_cow(bset);
348 bset = isl_basic_set_extend(bset, nparam, dim, 0, 1, 1);
349 if (isl_constraint_is_equality(constraint)) {
350 k = isl_basic_set_alloc_equality(bset);
356 k = isl_basic_set_alloc_inequality(bset);
361 total = isl_basic_set_total_dim(bset);
362 isl_seq_cpy(c, constraint->line[0], 1 + total);
363 isl_constraint_free(constraint);
366 isl_constraint_free(constraint);
367 isl_basic_set_free(bset);
371 int isl_basic_set_has_defining_equality(
372 struct isl_basic_set *bset, enum isl_dim_type type, int pos,
373 struct isl_constraint **c)
381 offset = basic_set_offset(bset, type);
382 total = isl_basic_set_total_dim(bset);
383 isl_assert(bset->ctx, pos < isl_basic_set_dim(bset, type), return -1);
384 for (i = 0; i < bset->n_eq; ++i)
385 if (!isl_int_is_zero(bset->eq[i][offset + pos]) &&
386 isl_seq_first_non_zero(bset->eq[i]+offset+pos+1,
387 1+total-offset-pos-1) == -1) {
388 *c= isl_basic_set_constraint(isl_basic_set_copy(bset),
395 int isl_basic_set_has_defining_inequalities(
396 struct isl_basic_set *bset, enum isl_dim_type type, int pos,
397 struct isl_constraint **lower,
398 struct isl_constraint **upper)
404 isl_int **lower_line, **upper_line;
408 offset = basic_set_offset(bset, type);
409 total = isl_basic_set_total_dim(bset);
410 isl_assert(bset->ctx, pos < isl_basic_set_dim(bset, type), return -1);
412 for (i = 0; i < bset->n_ineq; ++i) {
413 if (isl_int_is_zero(bset->ineq[i][offset + pos]))
415 if (isl_int_is_one(bset->ineq[i][offset + pos]))
417 if (isl_int_is_negone(bset->ineq[i][offset + pos]))
419 if (isl_seq_first_non_zero(bset->ineq[i]+offset+pos+1,
420 1+total-offset-pos-1) != -1)
422 for (j = i + 1; j < bset->n_ineq; ++j) {
423 if (!isl_seq_is_neg(bset->ineq[i]+1, bset->ineq[j]+1,
426 isl_int_add(m, bset->ineq[i][0], bset->ineq[j][0]);
427 if (isl_int_abs_ge(m, bset->ineq[i][offset+pos]))
430 if (isl_int_is_pos(bset->ineq[i][offset+pos])) {
431 lower_line = &bset->ineq[i];
432 upper_line = &bset->ineq[j];
434 lower_line = &bset->ineq[j];
435 upper_line = &bset->ineq[i];
437 *lower = isl_basic_set_constraint(
438 isl_basic_set_copy(bset), lower_line);
439 *upper = isl_basic_set_constraint(
440 isl_basic_set_copy(bset), upper_line);