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 struct isl_constraint
*isl_basic_set_first_constraint(
137 struct isl_basic_set
*bset
)
143 return isl_basic_set_constraint(bset
, &bset
->eq
[0]);
145 if (bset
->n_ineq
> 0)
146 return isl_basic_set_constraint(bset
, &bset
->ineq
[0]);
148 isl_basic_set_free(bset
);
152 struct isl_constraint
*isl_constraint_next(struct isl_constraint
*c
)
154 c
= isl_constraint_cow(c
);
155 if (c
->line
>= c
->bmap
->eq
) {
157 if (c
->line
< c
->bmap
->eq
+ c
->bmap
->n_eq
)
159 c
->line
= c
->bmap
->ineq
;
162 if (c
->line
< c
->bmap
->ineq
+ c
->bmap
->n_ineq
)
164 isl_constraint_free(c
);
168 int isl_constraint_is_equal(struct isl_constraint
*constraint1
,
169 struct isl_constraint
*constraint2
)
171 if (!constraint1
|| !constraint2
)
173 return constraint1
->bmap
== constraint2
->bmap
&&
174 constraint1
->line
== constraint2
->line
;
177 struct isl_basic_map
*isl_basic_map_add_constraint(
178 struct isl_basic_map
*bmap
, struct isl_constraint
*constraint
)
180 if (!bmap
|| !constraint
)
183 isl_assert(constraint
->ctx
,
184 isl_dim_equal(bmap
->dim
, constraint
->bmap
->dim
), goto error
);
186 bmap
= isl_basic_map_intersect(bmap
,
187 isl_basic_map_copy(constraint
->bmap
));
188 isl_constraint_free(constraint
);
191 isl_basic_map_free(bmap
);
192 isl_constraint_free(constraint
);
196 struct isl_basic_set
*isl_basic_set_add_constraint(
197 struct isl_basic_set
*bset
, struct isl_constraint
*constraint
)
199 return (struct isl_basic_set
*)
200 isl_basic_map_add_constraint((struct isl_basic_map
*)bset
,
204 struct isl_constraint
*isl_constraint_add_div(struct isl_constraint
*constraint
,
205 struct isl_div
*div
, int *pos
)
207 if (!constraint
|| !div
)
210 isl_assert(constraint
->ctx
,
211 isl_dim_equal(div
->bmap
->dim
, constraint
->bmap
->dim
), goto error
);
212 isl_assert(constraint
->ctx
,
213 constraint
->bmap
->n_eq
+ constraint
->bmap
->n_ineq
== 1, goto error
);
215 constraint
->bmap
= isl_basic_map_cow(constraint
->bmap
);
216 constraint
->bmap
= isl_basic_map_extend_dim(constraint
->bmap
,
217 isl_dim_copy(constraint
->bmap
->dim
), 1, 0, 0);
218 if (!constraint
->bmap
)
220 constraint
->line
= &constraint
->bmap
->eq
[0];
221 *pos
= isl_basic_map_alloc_div(constraint
->bmap
);
224 isl_seq_cpy(constraint
->bmap
->div
[*pos
], div
->line
[0],
225 1 + 1 + isl_basic_map_total_dim(constraint
->bmap
));
229 isl_constraint_free(constraint
);
234 int isl_constraint_dim(struct isl_constraint
*constraint
,
235 enum isl_dim_type type
)
239 return n(constraint
, type
);
242 void isl_constraint_get_constant(struct isl_constraint
*constraint
, isl_int
*v
)
246 isl_int_set(*v
, constraint
->line
[0][0]);
249 void isl_constraint_get_coefficient(struct isl_constraint
*constraint
,
250 enum isl_dim_type type
, int pos
, isl_int
*v
)
255 isl_assert(constraint
->ctx
, pos
< n(constraint
, type
), return);
256 isl_int_set(*v
, constraint
->line
[0][offset(constraint
, type
) + pos
]);
259 struct isl_div
*isl_constraint_div(struct isl_constraint
*constraint
, int pos
)
264 isl_assert(constraint
->ctx
, pos
< n(constraint
, isl_dim_div
),
266 return isl_basic_map_div(isl_basic_map_copy(constraint
->bmap
), pos
);
269 void isl_constraint_set_constant(struct isl_constraint
*constraint
, isl_int v
)
273 isl_int_set(constraint
->line
[0][0], v
);
276 void isl_constraint_set_coefficient(struct isl_constraint
*constraint
,
277 enum isl_dim_type type
, int pos
, isl_int v
)
282 isl_assert(constraint
->ctx
, pos
< n(constraint
, type
), return);
283 isl_int_set(constraint
->line
[0][offset(constraint
, type
) + pos
], v
);
286 void isl_constraint_clear(struct isl_constraint
*constraint
)
292 total
= isl_basic_map_total_dim(constraint
->bmap
);
293 isl_seq_clr(constraint
->line
[0], 1 + total
);
296 struct isl_constraint
*isl_constraint_negate(struct isl_constraint
*constraint
)
303 isl_assert(constraint
->ctx
, !isl_constraint_is_equality(constraint
),
305 isl_assert(constraint
->ctx
, constraint
->bmap
->ref
== 1, goto error
);
306 total
= isl_basic_map_total_dim(constraint
->bmap
);
307 isl_seq_neg(constraint
->line
[0], constraint
->line
[0], 1 + total
);
308 isl_int_sub_ui(constraint
->line
[0][0], constraint
->line
[0][0], 1);
309 ISL_F_CLR(constraint
->bmap
, ISL_BASIC_MAP_NORMALIZED
);
312 isl_constraint_free(constraint
);
316 int isl_constraint_is_equality(struct isl_constraint
*constraint
)
320 return constraint
->line
>= constraint
->bmap
->eq
;
324 struct isl_basic_set
*isl_basic_set_from_constraint(
325 struct isl_constraint
*constraint
)
328 struct isl_basic_set
*constraint_bset
, *bset
;
337 isl_assert(constraint
->ctx
,n(constraint
, isl_dim_in
) == 0, goto error
);
339 constraint_bset
= (struct isl_basic_set
*)constraint
->bmap
;
340 bset
= isl_basic_set_universe_like(constraint_bset
);
341 bset
= isl_basic_set_align_divs(bset
, constraint_bset
);
342 nparam
= isl_basic_set_n_param(bset
);
343 dim
= isl_basic_set_n_dim(bset
);
344 bset
= isl_basic_set_cow(bset
);
345 bset
= isl_basic_set_extend(bset
, nparam
, dim
, 0, 1, 1);
346 if (isl_constraint_is_equality(constraint
)) {
347 k
= isl_basic_set_alloc_equality(bset
);
353 k
= isl_basic_set_alloc_inequality(bset
);
358 total
= isl_basic_set_total_dim(bset
);
359 isl_seq_cpy(c
, constraint
->line
[0], 1 + total
);
360 isl_constraint_free(constraint
);
363 isl_constraint_free(constraint
);
364 isl_basic_set_free(bset
);
368 int isl_basic_set_has_defining_equality(
369 struct isl_basic_set
*bset
, enum isl_dim_type type
, int pos
,
370 struct isl_constraint
**c
)
378 offset
= basic_set_offset(bset
, type
);
379 total
= isl_basic_set_total_dim(bset
);
380 isl_assert(bset
->ctx
, pos
< isl_basic_set_dim(bset
, type
), return -1);
381 for (i
= 0; i
< bset
->n_eq
; ++i
)
382 if (!isl_int_is_zero(bset
->eq
[i
][offset
+ pos
]) &&
383 isl_seq_first_non_zero(bset
->eq
[i
]+offset
+pos
+1,
384 1+total
-offset
-pos
-1) == -1) {
385 *c
= isl_basic_set_constraint(isl_basic_set_copy(bset
),
392 int isl_basic_set_has_defining_inequalities(
393 struct isl_basic_set
*bset
, enum isl_dim_type type
, int pos
,
394 struct isl_constraint
**lower
,
395 struct isl_constraint
**upper
)
401 isl_int
**lower_line
, **upper_line
;
405 offset
= basic_set_offset(bset
, type
);
406 total
= isl_basic_set_total_dim(bset
);
407 isl_assert(bset
->ctx
, pos
< isl_basic_set_dim(bset
, type
), return -1);
409 for (i
= 0; i
< bset
->n_ineq
; ++i
) {
410 if (isl_int_is_zero(bset
->ineq
[i
][offset
+ pos
]))
412 if (isl_int_is_one(bset
->ineq
[i
][offset
+ pos
]))
414 if (isl_int_is_negone(bset
->ineq
[i
][offset
+ pos
]))
416 if (isl_seq_first_non_zero(bset
->ineq
[i
]+offset
+pos
+1,
417 1+total
-offset
-pos
-1) != -1)
419 for (j
= i
+ 1; j
< bset
->n_ineq
; ++j
) {
420 if (!isl_seq_is_neg(bset
->ineq
[i
]+1, bset
->ineq
[j
]+1,
423 isl_int_add(m
, bset
->ineq
[i
][0], bset
->ineq
[j
][0]);
424 if (isl_int_abs_ge(m
, bset
->ineq
[i
][offset
+pos
]))
427 if (isl_int_is_pos(bset
->ineq
[i
][offset
+pos
])) {
428 lower_line
= &bset
->ineq
[i
];
429 upper_line
= &bset
->ineq
[j
];
431 lower_line
= &bset
->ineq
[j
];
432 upper_line
= &bset
->ineq
[i
];
434 *lower
= isl_basic_set_constraint(
435 isl_basic_set_copy(bset
), lower_line
);
436 *upper
= isl_basic_set_constraint(
437 isl_basic_set_copy(bset
), upper_line
);