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_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
),
267 constraint
->bmap
->div
+pos
);
270 void isl_constraint_set_constant(struct isl_constraint
*constraint
, isl_int v
)
274 isl_int_set(constraint
->line
[0][0], v
);
277 void isl_constraint_set_coefficient(struct isl_constraint
*constraint
,
278 enum isl_dim_type type
, int pos
, isl_int v
)
283 isl_assert(constraint
->ctx
, pos
< n(constraint
, type
), return);
284 isl_int_set(constraint
->line
[0][offset(constraint
, type
) + pos
], v
);
287 void isl_constraint_clear(struct isl_constraint
*constraint
)
289 struct isl_basic_set
*bset
;
294 total
= isl_basic_map_total_dim(constraint
->bmap
);
295 isl_seq_clr(constraint
->line
[0], 1 + total
);
298 struct isl_constraint
*isl_constraint_negate(struct isl_constraint
*constraint
)
305 isl_assert(constraint
->ctx
, !isl_constraint_is_equality(constraint
),
307 isl_assert(constraint
->ctx
, constraint
->bmap
->ref
== 1, goto error
);
308 total
= isl_basic_map_total_dim(constraint
->bmap
);
309 isl_seq_neg(constraint
->line
[0], constraint
->line
[0], 1 + total
);
310 isl_int_sub_ui(constraint
->line
[0][0], constraint
->line
[0][0], 1);
311 ISL_F_CLR(constraint
->bmap
, ISL_BASIC_MAP_NORMALIZED
);
314 isl_constraint_free(constraint
);
318 int isl_constraint_is_equality(struct isl_constraint
*constraint
)
322 return constraint
->line
>= constraint
->bmap
->eq
;
326 struct isl_basic_set
*isl_basic_set_from_constraint(
327 struct isl_constraint
*constraint
)
330 struct isl_basic_set
*constraint_bset
, *bset
;
339 isl_assert(constraint
->ctx
,n(constraint
, isl_dim_in
) == 0, goto error
);
341 constraint_bset
= (struct isl_basic_set
*)constraint
->bmap
;
342 bset
= isl_basic_set_universe_like(constraint_bset
);
343 bset
= isl_basic_set_align_divs(bset
, constraint_bset
);
344 nparam
= isl_basic_set_n_param(bset
);
345 dim
= isl_basic_set_n_dim(bset
);
346 bset
= isl_basic_set_extend(bset
, nparam
, dim
, 0, 1, 1);
347 if (isl_constraint_is_equality(constraint
)) {
348 k
= isl_basic_set_alloc_equality(bset
);
354 k
= isl_basic_set_alloc_inequality(bset
);
359 total
= isl_basic_set_total_dim(bset
);
360 isl_seq_cpy(c
, constraint
->line
[0], 1 + total
);
361 isl_constraint_free(constraint
);
364 isl_constraint_free(constraint
);
365 isl_basic_set_free(bset
);
369 int isl_basic_set_has_defining_equality(
370 struct isl_basic_set
*bset
, enum isl_dim_type type
, int pos
,
371 struct isl_constraint
**c
)
379 offset
= basic_set_offset(bset
, type
);
380 total
= isl_basic_set_total_dim(bset
);
381 isl_assert(bset
->ctx
, pos
< isl_basic_set_dim(bset
, type
), return -1);
382 for (i
= 0; i
< bset
->n_eq
; ++i
)
383 if (!isl_int_is_zero(bset
->eq
[i
][offset
+ pos
]) &&
384 isl_seq_first_non_zero(bset
->eq
[i
]+offset
+pos
+1,
385 1+total
-offset
-pos
-1) == -1) {
386 *c
= isl_basic_set_constraint(isl_basic_set_copy(bset
),
393 int isl_basic_set_has_defining_inequalities(
394 struct isl_basic_set
*bset
, enum isl_dim_type type
, int pos
,
395 struct isl_constraint
**lower
,
396 struct isl_constraint
**upper
)
402 isl_int
**lower_line
, **upper_line
;
406 offset
= basic_set_offset(bset
, type
);
407 total
= isl_basic_set_total_dim(bset
);
408 isl_assert(bset
->ctx
, pos
< isl_basic_set_dim(bset
, type
), return -1);
410 for (i
= 0; i
< bset
->n_ineq
; ++i
) {
411 if (isl_int_is_zero(bset
->ineq
[i
][offset
+ pos
]))
413 if (isl_int_is_one(bset
->ineq
[i
][offset
+ pos
]))
415 if (isl_int_is_negone(bset
->ineq
[i
][offset
+ pos
]))
417 if (isl_seq_first_non_zero(bset
->ineq
[i
]+offset
+pos
+1,
418 1+total
-offset
-pos
-1) != -1)
420 for (j
= i
+ 1; j
< bset
->n_ineq
; ++j
) {
421 if (!isl_seq_is_neg(bset
->ineq
[i
]+1, bset
->ineq
[j
]+1,
424 isl_int_add(m
, bset
->ineq
[i
][0], bset
->ineq
[j
][0]);
425 if (isl_int_abs_ge(m
, bset
->ineq
[i
][offset
+pos
]))
428 if (isl_int_is_pos(bset
->ineq
[i
][offset
+pos
])) {
429 lower_line
= &bset
->ineq
[i
];
430 upper_line
= &bset
->ineq
[j
];
432 lower_line
= &bset
->ineq
[j
];
433 upper_line
= &bset
->ineq
[i
];
435 *lower
= isl_basic_set_constraint(
436 isl_basic_set_copy(bset
), lower_line
);
437 *upper
= isl_basic_set_constraint(
438 isl_basic_set_copy(bset
), upper_line
);