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 struct isl_dim *dim = c->bmap->dim;
8 case isl_dim_param: return dim->nparam;
9 case isl_dim_in: return dim->n_in;
10 case isl_dim_out: return dim->n_out;
11 case isl_dim_div: return c->bmap->n_div;
12 case isl_dim_all: return isl_basic_map_total_dim(c->bmap);
16 static unsigned offset(struct isl_constraint *c, enum isl_dim_type type)
18 struct isl_dim *dim = c->bmap->dim;
20 case isl_dim_param: return 1;
21 case isl_dim_in: return 1 + dim->nparam;
22 case isl_dim_out: return 1 + dim->nparam + dim->n_in;
23 case isl_dim_div: return 1 + dim->nparam + dim->n_in + dim->n_out;
27 struct isl_constraint *isl_basic_map_constraint(struct isl_basic_map *bmap,
30 struct isl_constraint *constraint;
35 constraint = isl_alloc_type(bmap->ctx, struct isl_constraint);
39 constraint->ctx = bmap->ctx;
40 isl_ctx_ref(constraint->ctx);
42 constraint->bmap = bmap;
43 constraint->line = line;
47 isl_basic_map_free(bmap);
51 struct isl_constraint *isl_basic_set_constraint(struct isl_basic_set *bset,
54 return isl_basic_map_constraint((struct isl_basic_map *)bset, line);
57 struct isl_constraint *isl_equality_alloc(struct isl_dim *dim)
59 struct isl_basic_map *bmap;
64 bmap = isl_basic_map_alloc_dim(dim, 0, 1, 0);
68 isl_basic_map_alloc_equality(bmap);
69 isl_seq_clr(bmap->eq[0], 1 + isl_basic_map_total_dim(bmap));
70 return isl_basic_map_constraint(bmap, &bmap->eq[0]);
73 struct isl_constraint *isl_inequality_alloc(struct isl_dim *dim)
75 struct isl_basic_map *bmap;
80 bmap = isl_basic_map_alloc_dim(dim, 0, 0, 1);
84 isl_basic_map_alloc_inequality(bmap);
85 isl_seq_clr(bmap->ineq[0], 1 + isl_basic_map_total_dim(bmap));
86 return isl_basic_map_constraint(bmap, &bmap->ineq[0]);
89 struct isl_constraint *isl_constraint_dup(struct isl_constraint *c)
94 return isl_basic_map_constraint(isl_basic_map_copy(c->bmap), c->line);
97 struct isl_constraint *isl_constraint_cow(struct isl_constraint *c)
105 return isl_constraint_dup(c);
108 struct isl_constraint *isl_constraint_copy(struct isl_constraint *constraint)
117 struct isl_constraint *isl_constraint_free(struct isl_constraint *c)
125 isl_basic_map_free(c->bmap);
126 isl_ctx_deref(c->ctx);
130 struct isl_constraint *isl_basic_set_first_constraint(
131 struct isl_basic_set *bset)
133 struct isl_constraint *c;
139 return isl_basic_set_constraint(bset, &bset->eq[0]);
141 if (bset->n_ineq > 0)
142 return isl_basic_set_constraint(bset, &bset->ineq[0]);
144 isl_basic_set_free(bset);
148 struct isl_constraint *isl_constraint_next(struct isl_constraint *c)
150 c = isl_constraint_cow(c);
152 if (c->line >= c->bmap->eq + c->bmap->n_eq && c->line < c->bmap->ineq)
153 c->line = c->bmap->ineq;
154 if (c->line < c->bmap->ineq + c->bmap->n_ineq)
156 isl_constraint_free(c);
160 int isl_constraint_is_equal(struct isl_constraint *constraint1,
161 struct isl_constraint *constraint2)
163 if (!constraint1 || !constraint2)
165 return constraint1->bmap == constraint2->bmap &&
166 constraint1->line == constraint2->line;
169 struct isl_basic_set *isl_basic_set_add_constraint(
170 struct isl_basic_set *bset, struct isl_constraint *constraint)
172 if (!bset || !constraint)
175 isl_assert(constraint->ctx,
176 isl_dim_equal(bset->dim, constraint->bmap->dim), goto error);
178 bset = isl_basic_set_intersect(bset,
179 isl_basic_set_copy((struct isl_basic_set *)constraint->bmap));
180 isl_constraint_free(constraint);
183 isl_basic_set_free(bset);
184 isl_constraint_free(constraint);
188 int isl_constraint_dim(struct isl_constraint *constraint,
189 enum isl_dim_type type)
193 return n(constraint, type);
196 void isl_constraint_get_constant(struct isl_constraint *constraint, isl_int *v)
200 isl_int_set(*v, constraint->line[0][0]);
203 void isl_constraint_get_coefficient(struct isl_constraint *constraint,
204 enum isl_dim_type type, int pos, isl_int *v)
209 isl_assert(constraint->ctx, pos < n(constraint, type), return);
210 isl_int_set(*v, constraint->line[0][offset(constraint, type) + pos]);
213 struct isl_div *isl_constraint_div(struct isl_constraint *constraint, int pos)
218 isl_assert(constraint->ctx, pos < n(constraint, isl_dim_div),
220 return isl_basic_map_div(constraint->bmap, constraint->bmap->div+pos);
223 void isl_constraint_set_constant(struct isl_constraint *constraint, isl_int v)
227 isl_int_set(constraint->line[0][0], v);
230 void isl_constraint_set_coefficient(struct isl_constraint *constraint,
231 enum isl_dim_type type, int pos, isl_int v)
236 isl_assert(constraint->ctx, pos < n(constraint, type), return);
237 isl_int_set(constraint->line[0][offset(constraint, type) + pos], v);
240 void isl_constraint_clear(struct isl_constraint *constraint)
242 struct isl_basic_set *bset;
247 total = isl_basic_map_total_dim(constraint->bmap);
248 isl_seq_clr(constraint->line[0], 1 + total);
251 struct isl_constraint *isl_constraint_negate(struct isl_constraint *constraint)
258 isl_assert(constraint->ctx, !isl_constraint_is_equality(constraint),
260 isl_assert(constraint->ctx, constraint->bmap->ref == 1, goto error);
261 total = isl_basic_map_total_dim(constraint->bmap);
262 isl_seq_neg(constraint->line[0], constraint->line[0], 1 + total);
263 isl_int_sub_ui(constraint->line[0][0], constraint->line[0][0], 1);
264 F_CLR(constraint->bmap, ISL_BASIC_MAP_NORMALIZED);
267 isl_constraint_free(constraint);
271 int isl_constraint_is_equality(struct isl_constraint *constraint)
275 return constraint->line < constraint->bmap->eq + constraint->bmap->n_eq;
279 struct isl_basic_set *isl_basic_set_from_constraint(
280 struct isl_constraint *constraint)
283 struct isl_basic_set *constraint_bset, *bset;
292 isl_assert(constraint->ctx,n(constraint, isl_dim_in) == 0, goto error);
294 constraint_bset = (struct isl_basic_set *)constraint->bmap;
295 bset = isl_basic_set_universe_like(constraint_bset);
296 bset = isl_basic_set_align_divs(bset, constraint_bset);
297 nparam = isl_basic_set_n_param(bset);
298 dim = isl_basic_set_n_dim(bset);
299 bset = isl_basic_set_extend(bset, nparam, dim, 0, 1, 1);
300 if (isl_constraint_is_equality(constraint)) {
301 k = isl_basic_set_alloc_equality(bset);
307 k = isl_basic_set_alloc_inequality(bset);
312 total = isl_basic_set_total_dim(bset);
313 isl_seq_cpy(c, constraint->line[0], 1 + total);
314 isl_constraint_free(constraint);
317 isl_constraint_free(constraint);
318 isl_basic_set_free(bset);
322 int isl_basic_set_has_defining_equality(
323 struct isl_basic_set *bset, int pos,
324 struct isl_constraint **c)
327 unsigned dim, nparam;
331 nparam = isl_basic_set_n_param(bset);
332 dim = isl_basic_set_n_dim(bset);
333 isl_assert(bset->ctx, pos < dim, return -1);
334 for (i = 0; i < bset->n_eq; ++i)
335 if (!isl_int_is_zero(bset->eq[i][1 + nparam + pos]) &&
336 isl_seq_first_non_zero(bset->eq[i]+1+nparam+pos+1,
338 *c= isl_basic_set_constraint(isl_basic_set_copy(bset),
345 int isl_basic_set_has_defining_inequalities(
346 struct isl_basic_set *bset, int pos,
347 struct isl_constraint **lower,
348 struct isl_constraint **upper)
355 isl_int **lower_line, **upper_line;
359 nparam = isl_basic_set_n_param(bset);
360 dim = isl_basic_set_n_dim(bset);
361 total = isl_basic_set_total_dim(bset);
362 isl_assert(bset->ctx, pos < dim, return -1);
364 for (i = 0; i < bset->n_ineq; ++i) {
365 if (isl_int_is_zero(bset->ineq[i][1 + nparam + pos]))
367 if (isl_int_is_one(bset->ineq[i][1 + nparam + pos]))
369 if (isl_int_is_negone(bset->ineq[i][1 + nparam + pos]))
371 if (isl_seq_first_non_zero(bset->ineq[i]+1+nparam+pos+1,
374 for (j = i + i; j < bset->n_ineq; ++j) {
375 if (!isl_seq_is_neg(bset->ineq[i]+1, bset->ineq[j]+1,
378 isl_int_add(m, bset->ineq[i][0], bset->ineq[j][0]);
379 if (isl_int_abs_ge(m, bset->ineq[i][1+nparam+pos]))
382 if (isl_int_is_pos(bset->ineq[i][1+nparam+pos])) {
383 lower_line = &bset->ineq[i];
384 upper_line = &bset->ineq[j];
386 lower_line = &bset->ineq[j];
387 upper_line = &bset->ineq[i];
389 *lower = isl_basic_set_constraint(
390 isl_basic_set_copy(bset), lower_line);
391 *upper = isl_basic_set_constraint(
392 isl_basic_set_copy(bset), upper_line);