2 * Copyright 2008-2009 Katholieke Universiteit Leuven
4 * Use of this software is governed by the GNU LGPLv2.1 license
6 * Written by Sven Verdoolaege, K.U.Leuven, Departement
7 * Computerwetenschappen, Celestijnenlaan 200A, B-3001 Leuven, Belgium
10 #include <isl_constraint.h>
12 #include "isl_map_private.h"
14 static unsigned n(struct isl_constraint
*c
, enum isl_dim_type type
)
16 return isl_basic_map_dim(c
->bmap
, type
);
19 static unsigned offset(struct isl_constraint
*c
, enum isl_dim_type type
)
21 struct isl_dim
*dim
= c
->bmap
->dim
;
23 case isl_dim_param
: return 1;
24 case isl_dim_in
: return 1 + dim
->nparam
;
25 case isl_dim_out
: return 1 + dim
->nparam
+ dim
->n_in
;
26 case isl_dim_div
: return 1 + dim
->nparam
+ dim
->n_in
+ dim
->n_out
;
30 static unsigned basic_set_offset(struct isl_basic_set
*bset
,
31 enum isl_dim_type type
)
33 struct isl_dim
*dim
= bset
->dim
;
35 case isl_dim_param
: return 1;
36 case isl_dim_in
: return 1 + dim
->nparam
;
37 case isl_dim_out
: return 1 + dim
->nparam
+ dim
->n_in
;
38 case isl_dim_div
: return 1 + dim
->nparam
+ dim
->n_in
+ dim
->n_out
;
42 struct isl_constraint
*isl_basic_map_constraint(struct isl_basic_map
*bmap
,
45 struct isl_constraint
*constraint
;
50 constraint
= isl_alloc_type(bmap
->ctx
, struct isl_constraint
);
54 constraint
->ctx
= bmap
->ctx
;
55 isl_ctx_ref(constraint
->ctx
);
57 constraint
->bmap
= bmap
;
58 constraint
->line
= line
;
62 isl_basic_map_free(bmap
);
66 struct isl_constraint
*isl_basic_set_constraint(struct isl_basic_set
*bset
,
69 return isl_basic_map_constraint((struct isl_basic_map
*)bset
, line
);
72 struct isl_constraint
*isl_equality_alloc(struct isl_dim
*dim
)
74 struct isl_basic_map
*bmap
;
79 bmap
= isl_basic_map_alloc_dim(dim
, 0, 1, 0);
83 isl_basic_map_alloc_equality(bmap
);
84 isl_seq_clr(bmap
->eq
[0], 1 + isl_basic_map_total_dim(bmap
));
85 return isl_basic_map_constraint(bmap
, &bmap
->eq
[0]);
88 struct isl_constraint
*isl_inequality_alloc(struct isl_dim
*dim
)
90 struct isl_basic_map
*bmap
;
95 bmap
= isl_basic_map_alloc_dim(dim
, 0, 0, 1);
99 isl_basic_map_alloc_inequality(bmap
);
100 isl_seq_clr(bmap
->ineq
[0], 1 + isl_basic_map_total_dim(bmap
));
101 return isl_basic_map_constraint(bmap
, &bmap
->ineq
[0]);
104 struct isl_constraint
*isl_constraint_dup(struct isl_constraint
*c
)
109 return isl_basic_map_constraint(isl_basic_map_copy(c
->bmap
), c
->line
);
112 struct isl_constraint
*isl_constraint_cow(struct isl_constraint
*c
)
120 return isl_constraint_dup(c
);
123 struct isl_constraint
*isl_constraint_copy(struct isl_constraint
*constraint
)
132 void isl_constraint_free(struct isl_constraint
*c
)
140 isl_basic_map_free(c
->bmap
);
141 isl_ctx_deref(c
->ctx
);
145 __isl_give isl_constraint
*isl_basic_map_first_constraint(
146 __isl_take isl_basic_map
*bmap
)
152 return isl_basic_map_constraint(bmap
, &bmap
->eq
[0]);
154 if (bmap
->n_ineq
> 0)
155 return isl_basic_map_constraint(bmap
, &bmap
->ineq
[0]);
157 isl_basic_map_free(bmap
);
161 __isl_give isl_constraint
*isl_basic_set_first_constraint(
162 __isl_take isl_basic_set
*bset
)
164 return isl_basic_map_first_constraint((struct isl_basic_map
*)bset
);
167 struct isl_constraint
*isl_constraint_next(struct isl_constraint
*c
)
169 c
= isl_constraint_cow(c
);
170 if (c
->line
>= c
->bmap
->eq
) {
172 if (c
->line
< c
->bmap
->eq
+ c
->bmap
->n_eq
)
174 c
->line
= c
->bmap
->ineq
;
177 if (c
->line
< c
->bmap
->ineq
+ c
->bmap
->n_ineq
)
179 isl_constraint_free(c
);
183 int isl_constraint_is_equal(struct isl_constraint
*constraint1
,
184 struct isl_constraint
*constraint2
)
186 if (!constraint1
|| !constraint2
)
188 return constraint1
->bmap
== constraint2
->bmap
&&
189 constraint1
->line
== constraint2
->line
;
192 struct isl_basic_map
*isl_basic_map_add_constraint(
193 struct isl_basic_map
*bmap
, struct isl_constraint
*constraint
)
195 if (!bmap
|| !constraint
)
198 isl_assert(constraint
->ctx
,
199 isl_dim_equal(bmap
->dim
, constraint
->bmap
->dim
), goto error
);
201 bmap
= isl_basic_map_intersect(bmap
,
202 isl_basic_map_from_constraint(constraint
));
205 isl_basic_map_free(bmap
);
206 isl_constraint_free(constraint
);
210 struct isl_basic_set
*isl_basic_set_add_constraint(
211 struct isl_basic_set
*bset
, struct isl_constraint
*constraint
)
213 return (struct isl_basic_set
*)
214 isl_basic_map_add_constraint((struct isl_basic_map
*)bset
,
218 struct isl_constraint
*isl_constraint_add_div(struct isl_constraint
*constraint
,
219 struct isl_div
*div
, int *pos
)
221 if (!constraint
|| !div
)
224 isl_assert(constraint
->ctx
,
225 isl_dim_equal(div
->bmap
->dim
, constraint
->bmap
->dim
), goto error
);
226 isl_assert(constraint
->ctx
,
227 constraint
->bmap
->n_eq
+ constraint
->bmap
->n_ineq
== 1, goto error
);
229 constraint
->bmap
= isl_basic_map_cow(constraint
->bmap
);
230 constraint
->bmap
= isl_basic_map_extend_dim(constraint
->bmap
,
231 isl_dim_copy(constraint
->bmap
->dim
), 1, 0, 0);
232 if (!constraint
->bmap
)
234 constraint
->line
= &constraint
->bmap
->eq
[0];
235 *pos
= isl_basic_map_alloc_div(constraint
->bmap
);
238 isl_seq_cpy(constraint
->bmap
->div
[*pos
], div
->line
[0],
239 1 + 1 + isl_basic_map_total_dim(constraint
->bmap
));
243 isl_constraint_free(constraint
);
248 int isl_constraint_dim(struct isl_constraint
*constraint
,
249 enum isl_dim_type type
)
253 return n(constraint
, type
);
256 void isl_constraint_get_constant(struct isl_constraint
*constraint
, isl_int
*v
)
260 isl_int_set(*v
, constraint
->line
[0][0]);
263 void isl_constraint_get_coefficient(struct isl_constraint
*constraint
,
264 enum isl_dim_type type
, int pos
, isl_int
*v
)
269 isl_assert(constraint
->ctx
, pos
< n(constraint
, type
), return);
270 isl_int_set(*v
, constraint
->line
[0][offset(constraint
, type
) + pos
]);
273 struct isl_div
*isl_constraint_div(struct isl_constraint
*constraint
, int pos
)
278 isl_assert(constraint
->ctx
, pos
< n(constraint
, isl_dim_div
),
280 return isl_basic_map_div(isl_basic_map_copy(constraint
->bmap
), pos
);
283 void isl_constraint_set_constant(struct isl_constraint
*constraint
, isl_int v
)
287 isl_int_set(constraint
->line
[0][0], v
);
290 void isl_constraint_set_coefficient(struct isl_constraint
*constraint
,
291 enum isl_dim_type type
, int pos
, isl_int v
)
296 isl_assert(constraint
->ctx
, pos
< n(constraint
, type
), return);
297 isl_int_set(constraint
->line
[0][offset(constraint
, type
) + pos
], v
);
300 void isl_constraint_clear(struct isl_constraint
*constraint
)
306 total
= isl_basic_map_total_dim(constraint
->bmap
);
307 isl_seq_clr(constraint
->line
[0], 1 + total
);
310 struct isl_constraint
*isl_constraint_negate(struct isl_constraint
*constraint
)
317 isl_assert(constraint
->ctx
, !isl_constraint_is_equality(constraint
),
319 isl_assert(constraint
->ctx
, constraint
->bmap
->ref
== 1, goto error
);
320 total
= isl_basic_map_total_dim(constraint
->bmap
);
321 isl_seq_neg(constraint
->line
[0], constraint
->line
[0], 1 + total
);
322 isl_int_sub_ui(constraint
->line
[0][0], constraint
->line
[0][0], 1);
323 ISL_F_CLR(constraint
->bmap
, ISL_BASIC_MAP_NORMALIZED
);
326 isl_constraint_free(constraint
);
330 int isl_constraint_is_equality(struct isl_constraint
*constraint
)
334 return constraint
->line
>= constraint
->bmap
->eq
;
337 __isl_give isl_basic_map
*isl_basic_map_from_constraint(
338 __isl_take isl_constraint
*constraint
)
341 struct isl_basic_map
*bmap
;
348 if (constraint
->bmap
->n_eq
+ constraint
->bmap
->n_ineq
== 1) {
349 bmap
= isl_basic_map_copy(constraint
->bmap
);
350 isl_constraint_free(constraint
);
354 bmap
= isl_basic_map_universe_like(constraint
->bmap
);
355 bmap
= isl_basic_map_align_divs(bmap
, constraint
->bmap
);
356 bmap
= isl_basic_map_cow(bmap
);
357 bmap
= isl_basic_map_extend_constraints(bmap
, 1, 1);
358 if (isl_constraint_is_equality(constraint
)) {
359 k
= isl_basic_map_alloc_equality(bmap
);
365 k
= isl_basic_map_alloc_inequality(bmap
);
370 total
= isl_basic_map_total_dim(bmap
);
371 isl_seq_cpy(c
, constraint
->line
[0], 1 + total
);
372 isl_constraint_free(constraint
);
375 isl_constraint_free(constraint
);
376 isl_basic_map_free(bmap
);
380 struct isl_basic_set
*isl_basic_set_from_constraint(
381 struct isl_constraint
*constraint
)
386 isl_assert(constraint
->ctx
,n(constraint
, isl_dim_in
) == 0, goto error
);
387 return (isl_basic_set
*)isl_basic_map_from_constraint(constraint
);
389 isl_constraint_free(constraint
);
393 int isl_basic_set_has_defining_equality(
394 struct isl_basic_set
*bset
, enum isl_dim_type type
, int pos
,
395 struct isl_constraint
**c
)
403 offset
= basic_set_offset(bset
, type
);
404 total
= isl_basic_set_total_dim(bset
);
405 isl_assert(bset
->ctx
, pos
< isl_basic_set_dim(bset
, type
), return -1);
406 for (i
= 0; i
< bset
->n_eq
; ++i
)
407 if (!isl_int_is_zero(bset
->eq
[i
][offset
+ pos
]) &&
408 isl_seq_first_non_zero(bset
->eq
[i
]+offset
+pos
+1,
409 1+total
-offset
-pos
-1) == -1) {
410 *c
= isl_basic_set_constraint(isl_basic_set_copy(bset
),
417 int isl_basic_set_has_defining_inequalities(
418 struct isl_basic_set
*bset
, enum isl_dim_type type
, int pos
,
419 struct isl_constraint
**lower
,
420 struct isl_constraint
**upper
)
426 isl_int
**lower_line
, **upper_line
;
430 offset
= basic_set_offset(bset
, type
);
431 total
= isl_basic_set_total_dim(bset
);
432 isl_assert(bset
->ctx
, pos
< isl_basic_set_dim(bset
, type
), return -1);
434 for (i
= 0; i
< bset
->n_ineq
; ++i
) {
435 if (isl_int_is_zero(bset
->ineq
[i
][offset
+ pos
]))
437 if (isl_int_is_one(bset
->ineq
[i
][offset
+ pos
]))
439 if (isl_int_is_negone(bset
->ineq
[i
][offset
+ pos
]))
441 if (isl_seq_first_non_zero(bset
->ineq
[i
]+offset
+pos
+1,
442 1+total
-offset
-pos
-1) != -1)
444 for (j
= i
+ 1; j
< bset
->n_ineq
; ++j
) {
445 if (!isl_seq_is_neg(bset
->ineq
[i
]+1, bset
->ineq
[j
]+1,
448 isl_int_add(m
, bset
->ineq
[i
][0], bset
->ineq
[j
][0]);
449 if (isl_int_abs_ge(m
, bset
->ineq
[i
][offset
+pos
]))
452 if (isl_int_is_pos(bset
->ineq
[i
][offset
+pos
])) {
453 lower_line
= &bset
->ineq
[i
];
454 upper_line
= &bset
->ineq
[j
];
456 lower_line
= &bset
->ineq
[j
];
457 upper_line
= &bset
->ineq
[i
];
459 *lower
= isl_basic_set_constraint(
460 isl_basic_set_copy(bset
), lower_line
);
461 *upper
= isl_basic_set_constraint(
462 isl_basic_set_copy(bset
), upper_line
);