1 #include <isl_constraint.h>
3 #include "isl_map_private.h"
5 static unsigned n(struct isl_constraint
*c
, enum isl_dim_type type
)
7 return isl_basic_map_dim(c
->bmap
, type
);
10 static unsigned offset(struct isl_constraint
*c
, enum isl_dim_type type
)
12 struct isl_dim
*dim
= c
->bmap
->dim
;
14 case isl_dim_param
: return 1;
15 case isl_dim_in
: return 1 + dim
->nparam
;
16 case isl_dim_out
: return 1 + dim
->nparam
+ dim
->n_in
;
17 case isl_dim_div
: return 1 + dim
->nparam
+ dim
->n_in
+ dim
->n_out
;
21 static unsigned basic_set_offset(struct isl_basic_set
*bset
,
22 enum isl_dim_type type
)
24 struct isl_dim
*dim
= bset
->dim
;
26 case isl_dim_param
: return 1;
27 case isl_dim_in
: return 1 + dim
->nparam
;
28 case isl_dim_out
: return 1 + dim
->nparam
+ dim
->n_in
;
29 case isl_dim_div
: return 1 + dim
->nparam
+ dim
->n_in
+ dim
->n_out
;
33 struct isl_constraint
*isl_basic_map_constraint(struct isl_basic_map
*bmap
,
36 struct isl_constraint
*constraint
;
41 constraint
= isl_alloc_type(bmap
->ctx
, struct isl_constraint
);
45 constraint
->ctx
= bmap
->ctx
;
46 isl_ctx_ref(constraint
->ctx
);
48 constraint
->bmap
= bmap
;
49 constraint
->line
= line
;
53 isl_basic_map_free(bmap
);
57 struct isl_constraint
*isl_basic_set_constraint(struct isl_basic_set
*bset
,
60 return isl_basic_map_constraint((struct isl_basic_map
*)bset
, line
);
63 struct isl_constraint
*isl_equality_alloc(struct isl_dim
*dim
)
65 struct isl_basic_map
*bmap
;
70 bmap
= isl_basic_map_alloc_dim(dim
, 0, 1, 0);
74 isl_basic_map_alloc_equality(bmap
);
75 isl_seq_clr(bmap
->eq
[0], 1 + isl_basic_map_total_dim(bmap
));
76 return isl_basic_map_constraint(bmap
, &bmap
->eq
[0]);
79 struct isl_constraint
*isl_inequality_alloc(struct isl_dim
*dim
)
81 struct isl_basic_map
*bmap
;
86 bmap
= isl_basic_map_alloc_dim(dim
, 0, 0, 1);
90 isl_basic_map_alloc_inequality(bmap
);
91 isl_seq_clr(bmap
->ineq
[0], 1 + isl_basic_map_total_dim(bmap
));
92 return isl_basic_map_constraint(bmap
, &bmap
->ineq
[0]);
95 struct isl_constraint
*isl_constraint_dup(struct isl_constraint
*c
)
100 return isl_basic_map_constraint(isl_basic_map_copy(c
->bmap
), c
->line
);
103 struct isl_constraint
*isl_constraint_cow(struct isl_constraint
*c
)
111 return isl_constraint_dup(c
);
114 struct isl_constraint
*isl_constraint_copy(struct isl_constraint
*constraint
)
123 void isl_constraint_free(struct isl_constraint
*c
)
131 isl_basic_map_free(c
->bmap
);
132 isl_ctx_deref(c
->ctx
);
136 __isl_give isl_constraint
*isl_basic_map_first_constraint(
137 __isl_take isl_basic_map
*bmap
)
143 return isl_basic_map_constraint(bmap
, &bmap
->eq
[0]);
145 if (bmap
->n_ineq
> 0)
146 return isl_basic_map_constraint(bmap
, &bmap
->ineq
[0]);
148 isl_basic_map_free(bmap
);
152 __isl_give isl_constraint
*isl_basic_set_first_constraint(
153 __isl_take isl_basic_set
*bset
)
155 return isl_basic_map_first_constraint((struct isl_basic_map
*)bset
);
158 struct isl_constraint
*isl_constraint_next(struct isl_constraint
*c
)
160 c
= isl_constraint_cow(c
);
161 if (c
->line
>= c
->bmap
->eq
) {
163 if (c
->line
< c
->bmap
->eq
+ c
->bmap
->n_eq
)
165 c
->line
= c
->bmap
->ineq
;
168 if (c
->line
< c
->bmap
->ineq
+ c
->bmap
->n_ineq
)
170 isl_constraint_free(c
);
174 int isl_constraint_is_equal(struct isl_constraint
*constraint1
,
175 struct isl_constraint
*constraint2
)
177 if (!constraint1
|| !constraint2
)
179 return constraint1
->bmap
== constraint2
->bmap
&&
180 constraint1
->line
== constraint2
->line
;
183 struct isl_basic_map
*isl_basic_map_add_constraint(
184 struct isl_basic_map
*bmap
, struct isl_constraint
*constraint
)
186 if (!bmap
|| !constraint
)
189 isl_assert(constraint
->ctx
,
190 isl_dim_equal(bmap
->dim
, constraint
->bmap
->dim
), goto error
);
192 bmap
= isl_basic_map_intersect(bmap
,
193 isl_basic_map_from_constraint(constraint
));
196 isl_basic_map_free(bmap
);
197 isl_constraint_free(constraint
);
201 struct isl_basic_set
*isl_basic_set_add_constraint(
202 struct isl_basic_set
*bset
, struct isl_constraint
*constraint
)
204 return (struct isl_basic_set
*)
205 isl_basic_map_add_constraint((struct isl_basic_map
*)bset
,
209 struct isl_constraint
*isl_constraint_add_div(struct isl_constraint
*constraint
,
210 struct isl_div
*div
, int *pos
)
212 if (!constraint
|| !div
)
215 isl_assert(constraint
->ctx
,
216 isl_dim_equal(div
->bmap
->dim
, constraint
->bmap
->dim
), goto error
);
217 isl_assert(constraint
->ctx
,
218 constraint
->bmap
->n_eq
+ constraint
->bmap
->n_ineq
== 1, goto error
);
220 constraint
->bmap
= isl_basic_map_cow(constraint
->bmap
);
221 constraint
->bmap
= isl_basic_map_extend_dim(constraint
->bmap
,
222 isl_dim_copy(constraint
->bmap
->dim
), 1, 0, 0);
223 if (!constraint
->bmap
)
225 constraint
->line
= &constraint
->bmap
->eq
[0];
226 *pos
= isl_basic_map_alloc_div(constraint
->bmap
);
229 isl_seq_cpy(constraint
->bmap
->div
[*pos
], div
->line
[0],
230 1 + 1 + isl_basic_map_total_dim(constraint
->bmap
));
234 isl_constraint_free(constraint
);
239 int isl_constraint_dim(struct isl_constraint
*constraint
,
240 enum isl_dim_type type
)
244 return n(constraint
, type
);
247 void isl_constraint_get_constant(struct isl_constraint
*constraint
, isl_int
*v
)
251 isl_int_set(*v
, constraint
->line
[0][0]);
254 void isl_constraint_get_coefficient(struct isl_constraint
*constraint
,
255 enum isl_dim_type type
, int pos
, isl_int
*v
)
260 isl_assert(constraint
->ctx
, pos
< n(constraint
, type
), return);
261 isl_int_set(*v
, constraint
->line
[0][offset(constraint
, type
) + pos
]);
264 struct isl_div
*isl_constraint_div(struct isl_constraint
*constraint
, int pos
)
269 isl_assert(constraint
->ctx
, pos
< n(constraint
, isl_dim_div
),
271 return isl_basic_map_div(isl_basic_map_copy(constraint
->bmap
), pos
);
274 void isl_constraint_set_constant(struct isl_constraint
*constraint
, isl_int v
)
278 isl_int_set(constraint
->line
[0][0], v
);
281 void isl_constraint_set_coefficient(struct isl_constraint
*constraint
,
282 enum isl_dim_type type
, int pos
, isl_int v
)
287 isl_assert(constraint
->ctx
, pos
< n(constraint
, type
), return);
288 isl_int_set(constraint
->line
[0][offset(constraint
, type
) + pos
], v
);
291 void isl_constraint_clear(struct isl_constraint
*constraint
)
297 total
= isl_basic_map_total_dim(constraint
->bmap
);
298 isl_seq_clr(constraint
->line
[0], 1 + total
);
301 struct isl_constraint
*isl_constraint_negate(struct isl_constraint
*constraint
)
308 isl_assert(constraint
->ctx
, !isl_constraint_is_equality(constraint
),
310 isl_assert(constraint
->ctx
, constraint
->bmap
->ref
== 1, goto error
);
311 total
= isl_basic_map_total_dim(constraint
->bmap
);
312 isl_seq_neg(constraint
->line
[0], constraint
->line
[0], 1 + total
);
313 isl_int_sub_ui(constraint
->line
[0][0], constraint
->line
[0][0], 1);
314 ISL_F_CLR(constraint
->bmap
, ISL_BASIC_MAP_NORMALIZED
);
317 isl_constraint_free(constraint
);
321 int isl_constraint_is_equality(struct isl_constraint
*constraint
)
325 return constraint
->line
>= constraint
->bmap
->eq
;
328 __isl_give isl_basic_map
*isl_basic_map_from_constraint(
329 __isl_take isl_constraint
*constraint
)
332 struct isl_basic_map
*bmap
;
339 if (constraint
->bmap
->n_eq
+ constraint
->bmap
->n_ineq
== 1) {
340 bmap
= isl_basic_map_copy(constraint
->bmap
);
341 isl_constraint_free(constraint
);
345 bmap
= isl_basic_map_universe_like(constraint
->bmap
);
346 bmap
= isl_basic_map_align_divs(bmap
, constraint
->bmap
);
347 bmap
= isl_basic_map_cow(bmap
);
348 bmap
= isl_basic_map_extend_constraints(bmap
, 1, 1);
349 if (isl_constraint_is_equality(constraint
)) {
350 k
= isl_basic_map_alloc_equality(bmap
);
356 k
= isl_basic_map_alloc_inequality(bmap
);
361 total
= isl_basic_map_total_dim(bmap
);
362 isl_seq_cpy(c
, constraint
->line
[0], 1 + total
);
363 isl_constraint_free(constraint
);
366 isl_constraint_free(constraint
);
367 isl_basic_map_free(bmap
);
371 struct isl_basic_set
*isl_basic_set_from_constraint(
372 struct isl_constraint
*constraint
)
377 isl_assert(constraint
->ctx
,n(constraint
, isl_dim_in
) == 0, goto error
);
378 return (isl_basic_set
*)isl_basic_map_from_constraint(constraint
);
380 isl_constraint_free(constraint
);
384 int isl_basic_set_has_defining_equality(
385 struct isl_basic_set
*bset
, enum isl_dim_type type
, int pos
,
386 struct isl_constraint
**c
)
394 offset
= basic_set_offset(bset
, type
);
395 total
= isl_basic_set_total_dim(bset
);
396 isl_assert(bset
->ctx
, pos
< isl_basic_set_dim(bset
, type
), return -1);
397 for (i
= 0; i
< bset
->n_eq
; ++i
)
398 if (!isl_int_is_zero(bset
->eq
[i
][offset
+ pos
]) &&
399 isl_seq_first_non_zero(bset
->eq
[i
]+offset
+pos
+1,
400 1+total
-offset
-pos
-1) == -1) {
401 *c
= isl_basic_set_constraint(isl_basic_set_copy(bset
),
408 int isl_basic_set_has_defining_inequalities(
409 struct isl_basic_set
*bset
, enum isl_dim_type type
, int pos
,
410 struct isl_constraint
**lower
,
411 struct isl_constraint
**upper
)
417 isl_int
**lower_line
, **upper_line
;
421 offset
= basic_set_offset(bset
, type
);
422 total
= isl_basic_set_total_dim(bset
);
423 isl_assert(bset
->ctx
, pos
< isl_basic_set_dim(bset
, type
), return -1);
425 for (i
= 0; i
< bset
->n_ineq
; ++i
) {
426 if (isl_int_is_zero(bset
->ineq
[i
][offset
+ pos
]))
428 if (isl_int_is_one(bset
->ineq
[i
][offset
+ pos
]))
430 if (isl_int_is_negone(bset
->ineq
[i
][offset
+ pos
]))
432 if (isl_seq_first_non_zero(bset
->ineq
[i
]+offset
+pos
+1,
433 1+total
-offset
-pos
-1) != -1)
435 for (j
= i
+ 1; j
< bset
->n_ineq
; ++j
) {
436 if (!isl_seq_is_neg(bset
->ineq
[i
]+1, bset
->ineq
[j
]+1,
439 isl_int_add(m
, bset
->ineq
[i
][0], bset
->ineq
[j
][0]);
440 if (isl_int_abs_ge(m
, bset
->ineq
[i
][offset
+pos
]))
443 if (isl_int_is_pos(bset
->ineq
[i
][offset
+pos
])) {
444 lower_line
= &bset
->ineq
[i
];
445 upper_line
= &bset
->ineq
[j
];
447 lower_line
= &bset
->ineq
[j
];
448 upper_line
= &bset
->ineq
[i
];
450 *lower
= isl_basic_set_constraint(
451 isl_basic_set_copy(bset
), lower_line
);
452 *upper
= isl_basic_set_constraint(
453 isl_basic_set_copy(bset
), upper_line
);