add isl_union_map_is_injective
[isl.git] / isl_union_map.c
blobf384fd5f72e3bd26abb208b40efa497f1da0e331
1 /*
2 * Copyright 2010 INRIA Saclay
4 * Use of this software is governed by the GNU LGPLv2.1 license
6 * Written by Sven Verdoolaege, INRIA Saclay - Ile-de-France,
7 * Parc Club Orsay Universite, ZAC des vignes, 4 rue Jacques Monod,
8 * 91893 Orsay, France
9 */
11 #include <isl_map_private.h>
12 #include <isl/ctx.h>
13 #include <isl/hash.h>
14 #include <isl/map.h>
15 #include <isl/set.h>
16 #include <isl_dim_private.h>
17 #include <isl_union_map_private.h>
18 #include <isl/union_set.h>
20 static __isl_give isl_union_map *isl_union_map_alloc(__isl_take isl_dim *dim,
21 int size)
23 isl_union_map *umap;
25 if (!dim)
26 return NULL;
28 umap = isl_calloc_type(dim->ctx, isl_union_map);
29 if (!umap)
30 return NULL;
32 umap->ref = 1;
33 umap->dim = dim;
34 if (isl_hash_table_init(dim->ctx, &umap->table, size) < 0)
35 goto error;
37 return umap;
38 error:
39 isl_dim_free(dim);
40 isl_union_map_free(umap);
41 return NULL;
44 __isl_give isl_union_map *isl_union_map_empty(__isl_take isl_dim *dim)
46 return isl_union_map_alloc(dim, 16);
49 __isl_give isl_union_set *isl_union_set_empty(__isl_take isl_dim *dim)
51 return isl_union_map_empty(dim);
54 isl_ctx *isl_union_map_get_ctx(__isl_keep isl_union_map *umap)
56 return umap ? umap->dim->ctx : NULL;
59 isl_ctx *isl_union_set_get_ctx(__isl_keep isl_union_set *uset)
61 return uset ? uset->dim->ctx : NULL;
64 __isl_give isl_dim *isl_union_map_get_dim(__isl_keep isl_union_map *umap)
66 if (!umap)
67 return NULL;
68 return isl_dim_copy(umap->dim);
71 __isl_give isl_dim *isl_union_set_get_dim(__isl_keep isl_union_set *uset)
73 return isl_union_map_get_dim(uset);
76 static int free_umap_entry(void **entry, void *user)
78 isl_map *map = *entry;
79 isl_map_free(map);
80 return 0;
83 static int add_map(__isl_take isl_map *map, void *user)
85 isl_union_map **umap = (isl_union_map **)user;
87 *umap = isl_union_map_add_map(*umap, map);
89 return 0;
92 __isl_give isl_union_map *isl_union_map_dup(__isl_keep isl_union_map *umap)
94 isl_union_map *dup;
96 if (!umap)
97 return NULL;
99 dup = isl_union_map_empty(isl_dim_copy(umap->dim));
100 if (isl_union_map_foreach_map(umap, &add_map, &dup) < 0)
101 goto error;
102 return dup;
103 error:
104 isl_union_map_free(dup);
105 return NULL;
108 __isl_give isl_union_map *isl_union_map_cow(__isl_take isl_union_map *umap)
110 if (!umap)
111 return NULL;
113 if (umap->ref == 1)
114 return umap;
115 umap->ref--;
116 return isl_union_map_dup(umap);
119 struct isl_union_align {
120 isl_reordering *exp;
121 isl_union_map *res;
124 static int align_entry(void **entry, void *user)
126 isl_map *map = *entry;
127 isl_reordering *exp;
128 struct isl_union_align *data = user;
130 exp = isl_reordering_extend_dim(isl_reordering_copy(data->exp),
131 isl_map_get_dim(map));
133 data->res = isl_union_map_add_map(data->res,
134 isl_map_realign(isl_map_copy(map), exp));
136 return 0;
139 /* Align the parameters of umap along those of model.
140 * The result has the parameters of model first, in the same order
141 * as they appear in model, followed by any remaining parameters of
142 * umap that do not appear in model.
144 __isl_give isl_union_map *isl_union_map_align_params(
145 __isl_take isl_union_map *umap, __isl_take isl_dim *model)
147 struct isl_union_align data = { NULL, NULL };
149 if (!umap || !model)
150 goto error;
152 if (isl_dim_match(umap->dim, isl_dim_param, model, isl_dim_param)) {
153 isl_dim_free(model);
154 return umap;
157 data.exp = isl_parameter_alignment_reordering(umap->dim, model);
158 if (!data.exp)
159 goto error;
161 data.res = isl_union_map_alloc(isl_dim_copy(data.exp->dim),
162 umap->table.n);
163 if (isl_hash_table_foreach(umap->dim->ctx, &umap->table,
164 &align_entry, &data) < 0)
165 goto error;
167 isl_reordering_free(data.exp);
168 isl_union_map_free(umap);
169 isl_dim_free(model);
170 return data.res;
171 error:
172 isl_reordering_free(data.exp);
173 isl_union_map_free(umap);
174 isl_union_map_free(data.res);
175 isl_dim_free(model);
176 return NULL;
179 __isl_give isl_union_set *isl_union_set_align_params(
180 __isl_take isl_union_set *uset, __isl_take isl_dim *model)
182 return isl_union_map_align_params(uset, model);
185 __isl_give isl_union_map *isl_union_map_union(__isl_take isl_union_map *umap1,
186 __isl_take isl_union_map *umap2)
188 umap1 = isl_union_map_align_params(umap1, isl_union_map_get_dim(umap2));
189 umap2 = isl_union_map_align_params(umap2, isl_union_map_get_dim(umap1));
191 umap1 = isl_union_map_cow(umap1);
193 if (!umap1 || !umap2)
194 goto error;
196 if (isl_union_map_foreach_map(umap2, &add_map, &umap1) < 0)
197 goto error;
199 isl_union_map_free(umap2);
201 return umap1;
202 error:
203 isl_union_map_free(umap1);
204 isl_union_map_free(umap2);
205 return NULL;
208 __isl_give isl_union_set *isl_union_set_union(__isl_take isl_union_set *uset1,
209 __isl_take isl_union_set *uset2)
211 return isl_union_map_union(uset1, uset2);
214 __isl_give isl_union_map *isl_union_map_copy(__isl_keep isl_union_map *umap)
216 if (!umap)
217 return NULL;
219 umap->ref++;
220 return umap;
223 __isl_give isl_union_set *isl_union_set_copy(__isl_keep isl_union_set *uset)
225 return isl_union_map_copy(uset);
228 void isl_union_map_free(__isl_take isl_union_map *umap)
230 if (!umap)
231 return;
233 if (--umap->ref > 0)
234 return;
236 isl_hash_table_foreach(umap->dim->ctx, &umap->table,
237 &free_umap_entry, NULL);
238 isl_hash_table_clear(&umap->table);
239 isl_dim_free(umap->dim);
240 free(umap);
243 void isl_union_set_free(__isl_take isl_union_set *uset)
245 isl_union_map_free(uset);
248 static int has_dim(const void *entry, const void *val)
250 isl_map *map = (isl_map *)entry;
251 isl_dim *dim = (isl_dim *)val;
253 return isl_dim_equal(map->dim, dim);
256 __isl_give isl_union_map *isl_union_map_add_map(__isl_take isl_union_map *umap,
257 __isl_take isl_map *map)
259 uint32_t hash;
260 struct isl_hash_table_entry *entry;
262 if (isl_map_fast_is_empty(map)) {
263 isl_map_free(map);
264 return umap;
267 umap = isl_union_map_cow(umap);
269 if (!map || !umap)
270 goto error;
272 isl_assert(map->ctx, isl_dim_match(map->dim, isl_dim_param, umap->dim,
273 isl_dim_param), goto error);
275 hash = isl_dim_get_hash(map->dim);
276 entry = isl_hash_table_find(umap->dim->ctx, &umap->table, hash,
277 &has_dim, map->dim, 1);
278 if (!entry)
279 goto error;
281 if (!entry->data)
282 entry->data = map;
283 else {
284 entry->data = isl_map_union(entry->data, isl_map_copy(map));
285 if (!entry->data)
286 goto error;
287 isl_map_free(map);
290 return umap;
291 error:
292 isl_map_free(map);
293 isl_union_map_free(umap);
294 return NULL;
297 __isl_give isl_union_set *isl_union_set_add_set(__isl_take isl_union_set *uset,
298 __isl_take isl_set *set)
300 return isl_union_map_add_map(uset, (isl_map *)set);
303 __isl_give isl_union_map *isl_union_map_from_map(__isl_take isl_map *map)
305 isl_dim *dim;
306 isl_union_map *umap;
308 if (!map)
309 return NULL;
311 dim = isl_map_get_dim(map);
312 dim = isl_dim_drop(dim, isl_dim_in, 0, isl_dim_size(dim, isl_dim_in));
313 dim = isl_dim_drop(dim, isl_dim_out, 0, isl_dim_size(dim, isl_dim_out));
314 umap = isl_union_map_empty(dim);
315 umap = isl_union_map_add_map(umap, map);
317 return umap;
320 __isl_give isl_union_set *isl_union_set_from_set(__isl_take isl_set *set)
322 return isl_union_map_from_map((isl_map *)set);
325 struct isl_union_map_foreach_data
327 int (*fn)(__isl_take isl_map *map, void *user);
328 void *user;
331 static int call_on_copy(void **entry, void *user)
333 isl_map *map = *entry;
334 struct isl_union_map_foreach_data *data;
335 data = (struct isl_union_map_foreach_data *)user;
337 return data->fn(isl_map_copy(map), data->user);
340 int isl_union_map_n_map(__isl_keep isl_union_map *umap)
342 return umap ? umap->table.n : 0;
345 int isl_union_set_n_set(__isl_keep isl_union_set *uset)
347 return uset ? uset->table.n : 0;
350 int isl_union_map_foreach_map(__isl_keep isl_union_map *umap,
351 int (*fn)(__isl_take isl_map *map, void *user), void *user)
353 struct isl_union_map_foreach_data data = { fn, user };
355 if (!umap)
356 return -1;
358 return isl_hash_table_foreach(umap->dim->ctx, &umap->table,
359 &call_on_copy, &data);
362 static int copy_map(void **entry, void *user)
364 isl_map *map = *entry;
365 isl_map **map_p = user;
367 *map_p = isl_map_copy(map);
369 return -1;
372 __isl_give isl_map *isl_union_map_copy_map(__isl_keep isl_union_map *umap)
374 isl_map *map = NULL;
376 if (!umap || umap->table.n == 0)
377 return NULL;
379 isl_hash_table_foreach(umap->dim->ctx, &umap->table, &copy_map, &map);
381 return map;
384 __isl_give isl_set *isl_union_set_copy_set(__isl_keep isl_union_set *uset)
386 return isl_union_map_copy_map(uset);
389 __isl_give isl_map *isl_union_map_extract_map(__isl_keep isl_union_map *umap,
390 __isl_take isl_dim *dim)
392 uint32_t hash;
393 struct isl_hash_table_entry *entry;
395 if (!umap || !dim)
396 goto error;
398 hash = isl_dim_get_hash(dim);
399 entry = isl_hash_table_find(umap->dim->ctx, &umap->table, hash,
400 &has_dim, dim, 0);
401 if (!entry)
402 return isl_map_empty(dim);
403 isl_dim_free(dim);
404 return isl_map_copy(entry->data);
405 error:
406 isl_dim_free(dim);
407 return NULL;
410 __isl_give isl_set *isl_union_set_extract_set(__isl_keep isl_union_set *uset,
411 __isl_take isl_dim *dim)
413 return (isl_set *)isl_union_map_extract_map(uset, dim);
416 /* Check if umap contains a map in the given space.
418 __isl_give int isl_union_map_contains(__isl_keep isl_union_map *umap,
419 __isl_keep isl_dim *dim)
421 uint32_t hash;
422 struct isl_hash_table_entry *entry;
424 if (!umap || !dim)
425 return -1;
427 hash = isl_dim_get_hash(dim);
428 entry = isl_hash_table_find(umap->dim->ctx, &umap->table, hash,
429 &has_dim, dim, 0);
430 return !!entry;
433 __isl_give int isl_union_set_contains(__isl_keep isl_union_set *uset,
434 __isl_keep isl_dim *dim)
436 return isl_union_map_contains(uset, dim);
439 int isl_union_set_foreach_set(__isl_keep isl_union_set *uset,
440 int (*fn)(__isl_take isl_set *set, void *user), void *user)
442 return isl_union_map_foreach_map(uset,
443 (int(*)(__isl_take isl_map *, void*))fn, user);
446 struct isl_union_set_foreach_point_data {
447 int (*fn)(__isl_take isl_point *pnt, void *user);
448 void *user;
451 static int foreach_point(__isl_take isl_set *set, void *user)
453 struct isl_union_set_foreach_point_data *data = user;
454 int r;
456 r = isl_set_foreach_point(set, data->fn, data->user);
457 isl_set_free(set);
459 return r;
462 int isl_union_set_foreach_point(__isl_keep isl_union_set *uset,
463 int (*fn)(__isl_take isl_point *pnt, void *user), void *user)
465 struct isl_union_set_foreach_point_data data = { fn, user };
466 return isl_union_set_foreach_set(uset, &foreach_point, &data);
469 struct isl_union_map_gen_bin_data {
470 isl_union_map *umap2;
471 isl_union_map *res;
474 static int subtract_entry(void **entry, void *user)
476 struct isl_union_map_gen_bin_data *data = user;
477 uint32_t hash;
478 struct isl_hash_table_entry *entry2;
479 isl_map *map = *entry;
481 hash = isl_dim_get_hash(map->dim);
482 entry2 = isl_hash_table_find(data->umap2->dim->ctx, &data->umap2->table,
483 hash, &has_dim, map->dim, 0);
484 map = isl_map_copy(map);
485 if (entry2) {
486 int empty;
487 map = isl_map_subtract(map, isl_map_copy(entry2->data));
489 empty = isl_map_is_empty(map);
490 if (empty < 0) {
491 isl_map_free(map);
492 return -1;
494 if (empty) {
495 isl_map_free(map);
496 return 0;
499 data->res = isl_union_map_add_map(data->res, map);
501 return 0;
504 static __isl_give isl_union_map *gen_bin_op(__isl_take isl_union_map *umap1,
505 __isl_take isl_union_map *umap2, int (*fn)(void **, void *))
507 struct isl_union_map_gen_bin_data data = { NULL, NULL };
509 umap1 = isl_union_map_align_params(umap1, isl_union_map_get_dim(umap2));
510 umap2 = isl_union_map_align_params(umap2, isl_union_map_get_dim(umap1));
512 if (!umap1 || !umap2)
513 goto error;
515 data.umap2 = umap2;
516 data.res = isl_union_map_alloc(isl_dim_copy(umap1->dim),
517 umap1->table.n);
518 if (isl_hash_table_foreach(umap1->dim->ctx, &umap1->table,
519 fn, &data) < 0)
520 goto error;
522 isl_union_map_free(umap1);
523 isl_union_map_free(umap2);
524 return data.res;
525 error:
526 isl_union_map_free(umap1);
527 isl_union_map_free(umap2);
528 isl_union_map_free(data.res);
529 return NULL;
532 __isl_give isl_union_map *isl_union_map_subtract(
533 __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
535 return gen_bin_op(umap1, umap2, &subtract_entry);
538 __isl_give isl_union_set *isl_union_set_subtract(
539 __isl_take isl_union_set *uset1, __isl_take isl_union_set *uset2)
541 return isl_union_map_subtract(uset1, uset2);
544 struct isl_union_map_match_bin_data {
545 isl_union_map *umap2;
546 isl_union_map *res;
547 __isl_give isl_map *(*fn)(__isl_take isl_map*, __isl_take isl_map*);
550 static int match_bin_entry(void **entry, void *user)
552 struct isl_union_map_match_bin_data *data = user;
553 uint32_t hash;
554 struct isl_hash_table_entry *entry2;
555 isl_map *map = *entry;
556 int empty;
558 hash = isl_dim_get_hash(map->dim);
559 entry2 = isl_hash_table_find(data->umap2->dim->ctx, &data->umap2->table,
560 hash, &has_dim, map->dim, 0);
561 if (!entry2)
562 return 0;
564 map = isl_map_copy(map);
565 map = data->fn(map, isl_map_copy(entry2->data));
567 empty = isl_map_is_empty(map);
568 if (empty < 0) {
569 isl_map_free(map);
570 return -1;
572 if (empty) {
573 isl_map_free(map);
574 return 0;
577 data->res = isl_union_map_add_map(data->res, map);
579 return 0;
582 static __isl_give isl_union_map *match_bin_op(__isl_take isl_union_map *umap1,
583 __isl_take isl_union_map *umap2,
584 __isl_give isl_map *(*fn)(__isl_take isl_map*, __isl_take isl_map*))
586 struct isl_union_map_match_bin_data data = { NULL, NULL, fn };
588 umap1 = isl_union_map_align_params(umap1, isl_union_map_get_dim(umap2));
589 umap2 = isl_union_map_align_params(umap2, isl_union_map_get_dim(umap1));
591 if (!umap1 || !umap2)
592 goto error;
594 data.umap2 = umap2;
595 data.res = isl_union_map_alloc(isl_dim_copy(umap1->dim),
596 umap1->table.n);
597 if (isl_hash_table_foreach(umap1->dim->ctx, &umap1->table,
598 &match_bin_entry, &data) < 0)
599 goto error;
601 isl_union_map_free(umap1);
602 isl_union_map_free(umap2);
603 return data.res;
604 error:
605 isl_union_map_free(umap1);
606 isl_union_map_free(umap2);
607 isl_union_map_free(data.res);
608 return NULL;
611 __isl_give isl_union_map *isl_union_map_intersect(
612 __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
614 return match_bin_op(umap1, umap2, &isl_map_intersect);
617 __isl_give isl_union_set *isl_union_set_intersect(
618 __isl_take isl_union_set *uset1, __isl_take isl_union_set *uset2)
620 return isl_union_map_intersect(uset1, uset2);
623 __isl_give isl_union_map *isl_union_map_gist(__isl_take isl_union_map *umap,
624 __isl_take isl_union_map *context)
626 return match_bin_op(umap, context, &isl_map_gist);
629 __isl_give isl_union_set *isl_union_set_gist(__isl_take isl_union_set *uset,
630 __isl_take isl_union_set *context)
632 return isl_union_map_gist(uset, context);
635 static __isl_give isl_map *lex_le_set(__isl_take isl_map *set1,
636 __isl_take isl_map *set2)
638 return isl_set_lex_le_set((isl_set *)set1, (isl_set *)set2);
641 static __isl_give isl_map *lex_lt_set(__isl_take isl_map *set1,
642 __isl_take isl_map *set2)
644 return isl_set_lex_lt_set((isl_set *)set1, (isl_set *)set2);
647 __isl_give isl_union_map *isl_union_set_lex_lt_union_set(
648 __isl_take isl_union_set *uset1, __isl_take isl_union_set *uset2)
650 return match_bin_op(uset1, uset2, &lex_lt_set);
653 __isl_give isl_union_map *isl_union_set_lex_le_union_set(
654 __isl_take isl_union_set *uset1, __isl_take isl_union_set *uset2)
656 return match_bin_op(uset1, uset2, &lex_le_set);
659 __isl_give isl_union_map *isl_union_set_lex_gt_union_set(
660 __isl_take isl_union_set *uset1, __isl_take isl_union_set *uset2)
662 return isl_union_map_reverse(isl_union_set_lex_lt_union_set(uset2, uset1));
665 __isl_give isl_union_map *isl_union_set_lex_ge_union_set(
666 __isl_take isl_union_set *uset1, __isl_take isl_union_set *uset2)
668 return isl_union_map_reverse(isl_union_set_lex_le_union_set(uset2, uset1));
671 __isl_give isl_union_map *isl_union_map_lex_gt_union_map(
672 __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
674 return isl_union_map_reverse(isl_union_map_lex_lt_union_map(umap2, umap1));
677 __isl_give isl_union_map *isl_union_map_lex_ge_union_map(
678 __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
680 return isl_union_map_reverse(isl_union_map_lex_le_union_map(umap2, umap1));
683 static int intersect_domain_entry(void **entry, void *user)
685 struct isl_union_map_gen_bin_data *data = user;
686 uint32_t hash;
687 struct isl_hash_table_entry *entry2;
688 isl_dim *dim;
689 isl_map *map = *entry;
690 int empty;
692 dim = isl_map_get_dim(map);
693 dim = isl_dim_domain(dim);
694 hash = isl_dim_get_hash(dim);
695 entry2 = isl_hash_table_find(data->umap2->dim->ctx, &data->umap2->table,
696 hash, &has_dim, dim, 0);
697 isl_dim_free(dim);
698 if (!entry2)
699 return 0;
701 map = isl_map_copy(map);
702 map = isl_map_intersect_domain(map, isl_set_copy(entry2->data));
704 empty = isl_map_is_empty(map);
705 if (empty < 0) {
706 isl_map_free(map);
707 return -1;
709 if (empty) {
710 isl_map_free(map);
711 return 0;
714 data->res = isl_union_map_add_map(data->res, map);
716 return 0;
719 __isl_give isl_union_map *isl_union_map_intersect_domain(
720 __isl_take isl_union_map *umap, __isl_take isl_union_set *uset)
722 return gen_bin_op(umap, uset, &intersect_domain_entry);
725 static int intersect_range_entry(void **entry, void *user)
727 struct isl_union_map_gen_bin_data *data = user;
728 uint32_t hash;
729 struct isl_hash_table_entry *entry2;
730 isl_dim *dim;
731 isl_map *map = *entry;
732 int empty;
734 dim = isl_map_get_dim(map);
735 dim = isl_dim_range(dim);
736 hash = isl_dim_get_hash(dim);
737 entry2 = isl_hash_table_find(data->umap2->dim->ctx, &data->umap2->table,
738 hash, &has_dim, dim, 0);
739 isl_dim_free(dim);
740 if (!entry2)
741 return 0;
743 map = isl_map_copy(map);
744 map = isl_map_intersect_range(map, isl_set_copy(entry2->data));
746 empty = isl_map_is_empty(map);
747 if (empty < 0) {
748 isl_map_free(map);
749 return -1;
751 if (empty) {
752 isl_map_free(map);
753 return 0;
756 data->res = isl_union_map_add_map(data->res, map);
758 return 0;
761 __isl_give isl_union_map *isl_union_map_intersect_range(
762 __isl_take isl_union_map *umap, __isl_take isl_union_set *uset)
764 return gen_bin_op(umap, uset, &intersect_range_entry);
767 struct isl_union_map_bin_data {
768 isl_union_map *umap2;
769 isl_union_map *res;
770 isl_map *map;
771 int (*fn)(void **entry, void *user);
774 static int apply_range_entry(void **entry, void *user)
776 struct isl_union_map_bin_data *data = user;
777 isl_map *map2 = *entry;
778 int empty;
780 if (!isl_dim_tuple_match(data->map->dim, isl_dim_out,
781 map2->dim, isl_dim_in))
782 return 0;
784 map2 = isl_map_apply_range(isl_map_copy(data->map), isl_map_copy(map2));
786 empty = isl_map_is_empty(map2);
787 if (empty < 0) {
788 isl_map_free(map2);
789 return -1;
791 if (empty) {
792 isl_map_free(map2);
793 return 0;
796 data->res = isl_union_map_add_map(data->res, map2);
798 return 0;
801 static int bin_entry(void **entry, void *user)
803 struct isl_union_map_bin_data *data = user;
804 isl_map *map = *entry;
806 data->map = map;
807 if (isl_hash_table_foreach(data->umap2->dim->ctx, &data->umap2->table,
808 data->fn, data) < 0)
809 return -1;
811 return 0;
814 static __isl_give isl_union_map *bin_op(__isl_take isl_union_map *umap1,
815 __isl_take isl_union_map *umap2, int (*fn)(void **entry, void *user))
817 struct isl_union_map_bin_data data = { NULL, NULL, NULL, fn };
819 umap1 = isl_union_map_align_params(umap1, isl_union_map_get_dim(umap2));
820 umap2 = isl_union_map_align_params(umap2, isl_union_map_get_dim(umap1));
822 if (!umap1 || !umap2)
823 goto error;
825 data.umap2 = umap2;
826 data.res = isl_union_map_alloc(isl_dim_copy(umap1->dim),
827 umap1->table.n);
828 if (isl_hash_table_foreach(umap1->dim->ctx, &umap1->table,
829 &bin_entry, &data) < 0)
830 goto error;
832 isl_union_map_free(umap1);
833 isl_union_map_free(umap2);
834 return data.res;
835 error:
836 isl_union_map_free(umap1);
837 isl_union_map_free(umap2);
838 isl_union_map_free(data.res);
839 return NULL;
842 __isl_give isl_union_map *isl_union_map_apply_range(
843 __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
845 return bin_op(umap1, umap2, &apply_range_entry);
848 __isl_give isl_union_map *isl_union_map_apply_domain(
849 __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
851 umap1 = isl_union_map_reverse(umap1);
852 umap1 = isl_union_map_apply_range(umap1, umap2);
853 return isl_union_map_reverse(umap1);
856 __isl_give isl_union_set *isl_union_set_apply(
857 __isl_take isl_union_set *uset, __isl_take isl_union_map *umap)
859 return isl_union_map_apply_range(uset, umap);
862 static int map_lex_lt_entry(void **entry, void *user)
864 struct isl_union_map_bin_data *data = user;
865 isl_map *map2 = *entry;
867 if (!isl_dim_tuple_match(data->map->dim, isl_dim_out,
868 map2->dim, isl_dim_out))
869 return 0;
871 map2 = isl_map_lex_lt_map(isl_map_copy(data->map), isl_map_copy(map2));
873 data->res = isl_union_map_add_map(data->res, map2);
875 return 0;
878 __isl_give isl_union_map *isl_union_map_lex_lt_union_map(
879 __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
881 return bin_op(umap1, umap2, &map_lex_lt_entry);
884 static int map_lex_le_entry(void **entry, void *user)
886 struct isl_union_map_bin_data *data = user;
887 isl_map *map2 = *entry;
889 if (!isl_dim_tuple_match(data->map->dim, isl_dim_out,
890 map2->dim, isl_dim_out))
891 return 0;
893 map2 = isl_map_lex_le_map(isl_map_copy(data->map), isl_map_copy(map2));
895 data->res = isl_union_map_add_map(data->res, map2);
897 return 0;
900 __isl_give isl_union_map *isl_union_map_lex_le_union_map(
901 __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
903 return bin_op(umap1, umap2, &map_lex_le_entry);
906 static int product_entry(void **entry, void *user)
908 struct isl_union_map_bin_data *data = user;
909 isl_map *map2 = *entry;
911 map2 = isl_map_product(isl_map_copy(data->map), isl_map_copy(map2));
913 data->res = isl_union_map_add_map(data->res, map2);
915 return 0;
918 __isl_give isl_union_map *isl_union_map_product(__isl_take isl_union_map *umap1,
919 __isl_take isl_union_map *umap2)
921 return bin_op(umap1, umap2, &product_entry);
924 __isl_give isl_union_set *isl_union_set_product(__isl_take isl_union_set *uset1,
925 __isl_take isl_union_set *uset2)
927 return isl_union_map_product(uset1, uset2);
930 static int range_product_entry(void **entry, void *user)
932 struct isl_union_map_bin_data *data = user;
933 isl_map *map2 = *entry;
935 map2 = isl_map_range_product(isl_map_copy(data->map),
936 isl_map_copy(map2));
938 data->res = isl_union_map_add_map(data->res, map2);
940 return 0;
943 __isl_give isl_union_map *isl_union_map_range_product(
944 __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
946 return bin_op(umap1, umap2, &range_product_entry);
949 __isl_give isl_union_map *isl_union_map_from_range(
950 __isl_take isl_union_set *uset)
952 return uset;
955 __isl_give isl_union_map *isl_union_map_from_domain(
956 __isl_take isl_union_set *uset)
958 return isl_union_map_reverse(isl_union_map_from_range(uset));
961 __isl_give isl_union_map *isl_union_map_from_domain_and_range(
962 __isl_take isl_union_set *domain, __isl_take isl_union_set *range)
964 return isl_union_map_apply_range(isl_union_map_from_domain(domain),
965 isl_union_map_from_range(range));
968 static __isl_give isl_union_map *un_op(__isl_take isl_union_map *umap,
969 int (*fn)(void **, void *))
971 umap = isl_union_map_cow(umap);
972 if (!umap)
973 return NULL;
975 if (isl_hash_table_foreach(umap->dim->ctx, &umap->table, fn, NULL) < 0)
976 goto error;
978 return umap;
979 error:
980 isl_union_map_free(umap);
981 return NULL;
984 static int affine_entry(void **entry, void *user)
986 isl_map **map = (isl_map **)entry;
988 *map = isl_map_from_basic_map(isl_map_affine_hull(*map));
990 return *map ? 0 : -1;
993 __isl_give isl_union_map *isl_union_map_affine_hull(
994 __isl_take isl_union_map *umap)
996 return un_op(umap, &affine_entry);
999 __isl_give isl_union_set *isl_union_set_affine_hull(
1000 __isl_take isl_union_set *uset)
1002 return isl_union_map_affine_hull(uset);
1005 static int polyhedral_entry(void **entry, void *user)
1007 isl_map **map = (isl_map **)entry;
1009 *map = isl_map_from_basic_map(isl_map_polyhedral_hull(*map));
1011 return *map ? 0 : -1;
1014 __isl_give isl_union_map *isl_union_map_polyhedral_hull(
1015 __isl_take isl_union_map *umap)
1017 return un_op(umap, &polyhedral_entry);
1020 __isl_give isl_union_set *isl_union_set_polyhedral_hull(
1021 __isl_take isl_union_set *uset)
1023 return isl_union_map_polyhedral_hull(uset);
1026 static int simple_entry(void **entry, void *user)
1028 isl_map **map = (isl_map **)entry;
1030 *map = isl_map_from_basic_map(isl_map_simple_hull(*map));
1032 return *map ? 0 : -1;
1035 __isl_give isl_union_map *isl_union_map_simple_hull(
1036 __isl_take isl_union_map *umap)
1038 return un_op(umap, &simple_entry);
1041 __isl_give isl_union_set *isl_union_set_simple_hull(
1042 __isl_take isl_union_set *uset)
1044 return isl_union_map_simple_hull(uset);
1047 static int inplace_entry(void **entry, void *user)
1049 __isl_give isl_map *(*fn)(__isl_take isl_map *);
1050 isl_map **map = (isl_map **)entry;
1051 isl_map *copy;
1053 fn = *(__isl_give isl_map *(**)(__isl_take isl_map *)) user;
1054 copy = fn(isl_map_copy(*map));
1055 if (!copy)
1056 return -1;
1058 isl_map_free(*map);
1059 *map = copy;
1061 return 0;
1064 static __isl_give isl_union_map *inplace(__isl_take isl_union_map *umap,
1065 __isl_give isl_map *(*fn)(__isl_take isl_map *))
1067 if (!umap)
1068 return NULL;
1070 if (isl_hash_table_foreach(umap->dim->ctx, &umap->table,
1071 &inplace_entry, &fn) < 0)
1072 goto error;
1074 return umap;
1075 error:
1076 isl_union_map_free(umap);
1077 return NULL;
1080 __isl_give isl_union_map *isl_union_map_coalesce(
1081 __isl_take isl_union_map *umap)
1083 return inplace(umap, &isl_map_coalesce);
1086 __isl_give isl_union_set *isl_union_set_coalesce(
1087 __isl_take isl_union_set *uset)
1089 return isl_union_map_coalesce(uset);
1092 __isl_give isl_union_map *isl_union_map_detect_equalities(
1093 __isl_take isl_union_map *umap)
1095 return inplace(umap, &isl_map_detect_equalities);
1098 __isl_give isl_union_set *isl_union_set_detect_equalities(
1099 __isl_take isl_union_set *uset)
1101 return isl_union_map_detect_equalities(uset);
1104 __isl_give isl_union_map *isl_union_map_compute_divs(
1105 __isl_take isl_union_map *umap)
1107 return inplace(umap, &isl_map_compute_divs);
1110 __isl_give isl_union_set *isl_union_set_compute_divs(
1111 __isl_take isl_union_set *uset)
1113 return isl_union_map_compute_divs(uset);
1116 static int lexmin_entry(void **entry, void *user)
1118 isl_map **map = (isl_map **)entry;
1120 *map = isl_map_lexmin(*map);
1122 return *map ? 0 : -1;
1125 __isl_give isl_union_map *isl_union_map_lexmin(
1126 __isl_take isl_union_map *umap)
1128 return un_op(umap, &lexmin_entry);
1131 __isl_give isl_union_set *isl_union_set_lexmin(
1132 __isl_take isl_union_set *uset)
1134 return isl_union_map_lexmin(uset);
1137 static int lexmax_entry(void **entry, void *user)
1139 isl_map **map = (isl_map **)entry;
1141 *map = isl_map_lexmax(*map);
1143 return *map ? 0 : -1;
1146 __isl_give isl_union_map *isl_union_map_lexmax(
1147 __isl_take isl_union_map *umap)
1149 return un_op(umap, &lexmax_entry);
1152 __isl_give isl_union_set *isl_union_set_lexmax(
1153 __isl_take isl_union_set *uset)
1155 return isl_union_map_lexmax(uset);
1158 static __isl_give isl_union_set *cond_un_op(__isl_take isl_union_map *umap,
1159 int (*fn)(void **, void *))
1161 isl_union_set *res;
1163 if (!umap)
1164 return NULL;
1166 res = isl_union_map_alloc(isl_dim_copy(umap->dim), umap->table.n);
1167 if (isl_hash_table_foreach(umap->dim->ctx, &umap->table, fn, &res) < 0)
1168 goto error;
1170 isl_union_map_free(umap);
1171 return res;
1172 error:
1173 isl_union_map_free(umap);
1174 isl_union_set_free(res);
1175 return NULL;
1178 static int reverse_entry(void **entry, void *user)
1180 isl_map *map = *entry;
1181 isl_union_map **res = user;
1183 *res = isl_union_map_add_map(*res, isl_map_reverse(isl_map_copy(map)));
1185 return 0;
1188 __isl_give isl_union_map *isl_union_map_reverse(__isl_take isl_union_map *umap)
1190 return cond_un_op(umap, &reverse_entry);
1193 static int domain_entry(void **entry, void *user)
1195 isl_map *map = *entry;
1196 isl_union_set **res = user;
1198 *res = isl_union_set_add_set(*res, isl_map_domain(isl_map_copy(map)));
1200 return 0;
1203 __isl_give isl_union_set *isl_union_map_domain(__isl_take isl_union_map *umap)
1205 return cond_un_op(umap, &domain_entry);
1208 static int range_entry(void **entry, void *user)
1210 isl_map *map = *entry;
1211 isl_union_set **res = user;
1213 *res = isl_union_set_add_set(*res, isl_map_range(isl_map_copy(map)));
1215 return 0;
1218 __isl_give isl_union_set *isl_union_map_range(__isl_take isl_union_map *umap)
1220 return cond_un_op(umap, &range_entry);
1223 static int domain_map_entry(void **entry, void *user)
1225 isl_map *map = *entry;
1226 isl_union_set **res = user;
1228 *res = isl_union_map_add_map(*res,
1229 isl_map_domain_map(isl_map_copy(map)));
1231 return 0;
1234 __isl_give isl_union_map *isl_union_map_domain_map(
1235 __isl_take isl_union_map *umap)
1237 return cond_un_op(umap, &domain_map_entry);
1240 static int range_map_entry(void **entry, void *user)
1242 isl_map *map = *entry;
1243 isl_union_set **res = user;
1245 *res = isl_union_map_add_map(*res,
1246 isl_map_range_map(isl_map_copy(map)));
1248 return 0;
1251 __isl_give isl_union_map *isl_union_map_range_map(
1252 __isl_take isl_union_map *umap)
1254 return cond_un_op(umap, &range_map_entry);
1257 static int deltas_entry(void **entry, void *user)
1259 isl_map *map = *entry;
1260 isl_union_set **res = user;
1262 if (!isl_dim_tuple_match(map->dim, isl_dim_in, map->dim, isl_dim_out))
1263 return 0;
1265 *res = isl_union_set_add_set(*res, isl_map_deltas(isl_map_copy(map)));
1267 return 0;
1270 __isl_give isl_union_set *isl_union_map_deltas(__isl_take isl_union_map *umap)
1272 return cond_un_op(umap, &deltas_entry);
1275 static int deltas_map_entry(void **entry, void *user)
1277 isl_map *map = *entry;
1278 isl_union_map **res = user;
1280 if (!isl_dim_tuple_match(map->dim, isl_dim_in, map->dim, isl_dim_out))
1281 return 0;
1283 *res = isl_union_map_add_map(*res,
1284 isl_map_deltas_map(isl_map_copy(map)));
1286 return 0;
1289 __isl_give isl_union_map *isl_union_map_deltas_map(
1290 __isl_take isl_union_map *umap)
1292 return cond_un_op(umap, &deltas_map_entry);
1295 static int identity_entry(void **entry, void *user)
1297 isl_set *set = *entry;
1298 isl_union_map **res = user;
1300 *res = isl_union_map_add_map(*res, isl_set_identity(isl_set_copy(set)));
1302 return 0;
1305 __isl_give isl_union_map *isl_union_set_identity(__isl_take isl_union_set *uset)
1307 return cond_un_op(uset, &identity_entry);
1310 static int unwrap_entry(void **entry, void *user)
1312 isl_set *set = *entry;
1313 isl_union_set **res = user;
1315 if (!isl_set_is_wrapping(set))
1316 return 0;
1318 *res = isl_union_map_add_map(*res, isl_set_unwrap(isl_set_copy(set)));
1320 return 0;
1323 __isl_give isl_union_map *isl_union_set_unwrap(__isl_take isl_union_set *uset)
1325 return cond_un_op(uset, &unwrap_entry);
1328 static int wrap_entry(void **entry, void *user)
1330 isl_map *map = *entry;
1331 isl_union_set **res = user;
1333 *res = isl_union_set_add_set(*res, isl_map_wrap(isl_map_copy(map)));
1335 return 0;
1338 __isl_give isl_union_set *isl_union_map_wrap(__isl_take isl_union_map *umap)
1340 return cond_un_op(umap, &wrap_entry);
1343 struct isl_union_map_is_subset_data {
1344 isl_union_map *umap2;
1345 int is_subset;
1348 static int is_subset_entry(void **entry, void *user)
1350 struct isl_union_map_is_subset_data *data = user;
1351 uint32_t hash;
1352 struct isl_hash_table_entry *entry2;
1353 isl_map *map = *entry;
1355 hash = isl_dim_get_hash(map->dim);
1356 entry2 = isl_hash_table_find(data->umap2->dim->ctx, &data->umap2->table,
1357 hash, &has_dim, map->dim, 0);
1358 if (!entry2) {
1359 data->is_subset = 0;
1360 return -1;
1363 data->is_subset = isl_map_is_subset(map, entry2->data);
1364 if (data->is_subset < 0 || !data->is_subset)
1365 return -1;
1367 return 0;
1370 int isl_union_map_is_subset(__isl_keep isl_union_map *umap1,
1371 __isl_keep isl_union_map *umap2)
1373 struct isl_union_map_is_subset_data data = { NULL, 1 };
1375 umap1 = isl_union_map_copy(umap1);
1376 umap2 = isl_union_map_copy(umap2);
1377 umap1 = isl_union_map_align_params(umap1, isl_union_map_get_dim(umap2));
1378 umap2 = isl_union_map_align_params(umap2, isl_union_map_get_dim(umap1));
1380 if (!umap1 || !umap2)
1381 goto error;
1383 data.umap2 = umap2;
1384 if (isl_hash_table_foreach(umap1->dim->ctx, &umap1->table,
1385 &is_subset_entry, &data) < 0 &&
1386 data.is_subset)
1387 goto error;
1389 isl_union_map_free(umap1);
1390 isl_union_map_free(umap2);
1392 return data.is_subset;
1393 error:
1394 isl_union_map_free(umap1);
1395 isl_union_map_free(umap2);
1396 return -1;
1399 int isl_union_set_is_subset(__isl_keep isl_union_set *uset1,
1400 __isl_keep isl_union_set *uset2)
1402 return isl_union_map_is_subset(uset1, uset2);
1405 int isl_union_map_is_equal(__isl_keep isl_union_map *umap1,
1406 __isl_keep isl_union_map *umap2)
1408 int is_subset;
1410 if (!umap1 || !umap2)
1411 return -1;
1412 is_subset = isl_union_map_is_subset(umap1, umap2);
1413 if (is_subset != 1)
1414 return is_subset;
1415 is_subset = isl_union_map_is_subset(umap2, umap1);
1416 return is_subset;
1419 int isl_union_set_is_equal(__isl_keep isl_union_set *uset1,
1420 __isl_keep isl_union_set *uset2)
1422 return isl_union_map_is_equal(uset1, uset2);
1425 int isl_union_map_is_strict_subset(__isl_keep isl_union_map *umap1,
1426 __isl_keep isl_union_map *umap2)
1428 int is_subset;
1430 if (!umap1 || !umap2)
1431 return -1;
1432 is_subset = isl_union_map_is_subset(umap1, umap2);
1433 if (is_subset != 1)
1434 return is_subset;
1435 is_subset = isl_union_map_is_subset(umap2, umap1);
1436 if (is_subset == -1)
1437 return is_subset;
1438 return !is_subset;
1441 int isl_union_set_is_strict_subset(__isl_keep isl_union_set *uset1,
1442 __isl_keep isl_union_set *uset2)
1444 return isl_union_map_is_strict_subset(uset1, uset2);
1447 static int sample_entry(void **entry, void *user)
1449 isl_basic_map **sample = (isl_basic_map **)user;
1450 isl_map *map = *entry;
1452 *sample = isl_map_sample(isl_map_copy(map));
1453 if (!*sample)
1454 return -1;
1455 if (!isl_basic_map_fast_is_empty(*sample))
1456 return -1;
1457 return 0;
1460 __isl_give isl_basic_map *isl_union_map_sample(__isl_take isl_union_map *umap)
1462 isl_basic_map *sample = NULL;
1464 if (!umap)
1465 return NULL;
1467 if (isl_hash_table_foreach(umap->dim->ctx, &umap->table,
1468 &sample_entry, &sample) < 0 &&
1469 !sample)
1470 goto error;
1472 if (!sample)
1473 sample = isl_basic_map_empty(isl_union_map_get_dim(umap));
1475 isl_union_map_free(umap);
1477 return sample;
1478 error:
1479 isl_union_map_free(umap);
1480 return NULL;
1483 __isl_give isl_basic_set *isl_union_set_sample(__isl_take isl_union_set *uset)
1485 return (isl_basic_set *)isl_union_map_sample(uset);
1488 struct isl_forall_data {
1489 int res;
1490 int (*fn)(__isl_keep isl_map *map);
1493 static int forall_entry(void **entry, void *user)
1495 struct isl_forall_data *data = user;
1496 isl_map *map = *entry;
1498 data->res = data->fn(map);
1499 if (data->res < 0)
1500 return -1;
1502 if (!data->res)
1503 return -1;
1505 return 0;
1508 static int union_map_forall(__isl_keep isl_union_map *umap,
1509 int (*fn)(__isl_keep isl_map *map))
1511 struct isl_forall_data data = { 1, fn };
1513 if (!umap)
1514 return -1;
1516 if (isl_hash_table_foreach(umap->dim->ctx, &umap->table,
1517 &forall_entry, &data) < 0 && data.res)
1518 return -1;
1520 return data.res;
1523 int isl_union_map_is_empty(__isl_keep isl_union_map *umap)
1525 return union_map_forall(umap, &isl_map_is_empty);
1528 int isl_union_set_is_empty(__isl_keep isl_union_set *uset)
1530 return isl_union_map_is_empty(uset);
1533 static int is_subset_of_identity(__isl_keep isl_map *map)
1535 int is_subset;
1536 isl_dim *dim;
1537 isl_map *id;
1539 if (!map)
1540 return -1;
1542 if (!isl_dim_tuple_match(map->dim, isl_dim_in, map->dim, isl_dim_out))
1543 return 0;
1545 dim = isl_map_get_dim(map);
1546 id = isl_map_identity(dim);
1548 is_subset = isl_map_is_subset(map, id);
1550 isl_map_free(id);
1552 return is_subset;
1555 /* Check if the given map is single-valued.
1556 * We simply compute
1558 * M \circ M^-1
1560 * and check if the result is a subset of the identity mapping.
1562 int isl_union_map_is_single_valued(__isl_keep isl_union_map *umap)
1564 isl_union_map *test;
1565 int sv;
1567 if (isl_union_map_n_map(umap) == 1) {
1568 isl_map *map = isl_union_map_copy_map(umap);
1569 sv = isl_map_is_single_valued(map);
1570 isl_map_free(map);
1571 return sv;
1574 test = isl_union_map_reverse(isl_union_map_copy(umap));
1575 test = isl_union_map_apply_range(test, isl_union_map_copy(umap));
1577 sv = union_map_forall(test, &is_subset_of_identity);
1579 isl_union_map_free(test);
1581 return sv;
1584 int isl_union_map_is_injective(__isl_keep isl_union_map *umap)
1586 int in;
1588 umap = isl_union_map_copy(umap);
1589 umap = isl_union_map_reverse(umap);
1590 in = isl_union_map_is_single_valued(umap);
1591 isl_union_map_free(umap);
1593 return in;
1596 int isl_union_map_is_bijective(__isl_keep isl_union_map *umap)
1598 int sv;
1600 sv = isl_union_map_is_single_valued(umap);
1601 if (sv < 0 || !sv)
1602 return sv;
1604 return isl_union_map_is_injective(umap);
1607 static int zip_entry(void **entry, void *user)
1609 isl_map *map = *entry;
1610 isl_union_map **res = user;
1612 if (!isl_map_can_zip(map))
1613 return 0;
1615 *res = isl_union_map_add_map(*res, isl_map_zip(isl_map_copy(map)));
1617 return 0;
1620 __isl_give isl_union_map *isl_union_map_zip(__isl_take isl_union_map *umap)
1622 return cond_un_op(umap, &zip_entry);
1625 static int lift_entry(void **entry, void *user)
1627 isl_set *set = *entry;
1628 isl_union_set **res = user;
1630 *res = isl_union_set_add_set(*res, isl_set_lift(isl_set_copy(set)));
1632 return 0;
1635 __isl_give isl_union_set *isl_union_set_lift(__isl_take isl_union_set *uset)
1637 return cond_un_op(uset, &lift_entry);
1640 static int coefficients_entry(void **entry, void *user)
1642 isl_set *set = *entry;
1643 isl_union_set **res = user;
1645 set = isl_set_copy(set);
1646 set = isl_set_from_basic_set(isl_set_coefficients(set));
1647 *res = isl_union_set_add_set(*res, set);
1649 return 0;
1652 __isl_give isl_union_set *isl_union_set_coefficients(
1653 __isl_take isl_union_set *uset)
1655 isl_ctx *ctx;
1656 isl_dim *dim;
1657 isl_union_set *res;
1659 if (!uset)
1660 return NULL;
1662 ctx = isl_union_set_get_ctx(uset);
1663 dim = isl_dim_set_alloc(ctx, 0, 0);
1664 res = isl_union_map_alloc(dim, uset->table.n);
1665 if (isl_hash_table_foreach(uset->dim->ctx, &uset->table,
1666 &coefficients_entry, &res) < 0)
1667 goto error;
1669 isl_union_set_free(uset);
1670 return res;
1671 error:
1672 isl_union_set_free(uset);
1673 isl_union_set_free(res);
1674 return NULL;
1677 static int solutions_entry(void **entry, void *user)
1679 isl_set *set = *entry;
1680 isl_union_set **res = user;
1682 set = isl_set_copy(set);
1683 set = isl_set_from_basic_set(isl_set_solutions(set));
1684 if (!*res)
1685 *res = isl_union_set_from_set(set);
1686 else
1687 *res = isl_union_set_add_set(*res, set);
1689 if (!*res)
1690 return -1;
1692 return 0;
1695 __isl_give isl_union_set *isl_union_set_solutions(
1696 __isl_take isl_union_set *uset)
1698 isl_ctx *ctx;
1699 isl_dim *dim;
1700 isl_union_set *res = NULL;
1702 if (!uset)
1703 return NULL;
1705 if (uset->table.n == 0) {
1706 res = isl_union_set_empty(isl_union_set_get_dim(uset));
1707 isl_union_set_free(uset);
1708 return res;
1711 if (isl_hash_table_foreach(uset->dim->ctx, &uset->table,
1712 &solutions_entry, &res) < 0)
1713 goto error;
1715 isl_union_set_free(uset);
1716 return res;
1717 error:
1718 isl_union_set_free(uset);
1719 isl_union_set_free(res);
1720 return NULL;