isl_dim_map: extract from isl_map.c
[isl.git] / isl_union_map.c
blob84fb6ce9dbabddc9d6fb089241dd6a0dad461ea3
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 __isl_give isl_map *isl_union_map_extract_map(__isl_keep isl_union_map *umap,
363 __isl_take isl_dim *dim)
365 uint32_t hash;
366 struct isl_hash_table_entry *entry;
368 if (!umap || !dim)
369 goto error;
371 hash = isl_dim_get_hash(dim);
372 entry = isl_hash_table_find(umap->dim->ctx, &umap->table, hash,
373 &has_dim, dim, 0);
374 if (!entry)
375 return isl_map_empty(dim);
376 isl_dim_free(dim);
377 return isl_map_copy(entry->data);
378 error:
379 isl_dim_free(dim);
380 return NULL;
383 __isl_give isl_set *isl_union_set_extract_set(__isl_keep isl_union_set *uset,
384 __isl_take isl_dim *dim)
386 return (isl_set *)isl_union_map_extract_map(uset, dim);
389 int isl_union_set_foreach_set(__isl_keep isl_union_set *uset,
390 int (*fn)(__isl_take isl_set *set, void *user), void *user)
392 return isl_union_map_foreach_map(uset,
393 (int(*)(__isl_take isl_map *, void*))fn, user);
396 struct isl_union_set_foreach_point_data {
397 int (*fn)(__isl_take isl_point *pnt, void *user);
398 void *user;
401 static int foreach_point(__isl_take isl_set *set, void *user)
403 struct isl_union_set_foreach_point_data *data = user;
404 int r;
406 r = isl_set_foreach_point(set, data->fn, data->user);
407 isl_set_free(set);
409 return r;
412 int isl_union_set_foreach_point(__isl_keep isl_union_set *uset,
413 int (*fn)(__isl_take isl_point *pnt, void *user), void *user)
415 struct isl_union_set_foreach_point_data data = { fn, user };
416 return isl_union_set_foreach_set(uset, &foreach_point, &data);
419 struct isl_union_map_gen_bin_data {
420 isl_union_map *umap2;
421 isl_union_map *res;
424 static int subtract_entry(void **entry, void *user)
426 struct isl_union_map_gen_bin_data *data = user;
427 uint32_t hash;
428 struct isl_hash_table_entry *entry2;
429 isl_map *map = *entry;
431 hash = isl_dim_get_hash(map->dim);
432 entry2 = isl_hash_table_find(data->umap2->dim->ctx, &data->umap2->table,
433 hash, &has_dim, map->dim, 0);
434 map = isl_map_copy(map);
435 if (entry2) {
436 int empty;
437 map = isl_map_subtract(map, isl_map_copy(entry2->data));
439 empty = isl_map_is_empty(map);
440 if (empty < 0) {
441 isl_map_free(map);
442 return -1;
444 if (empty) {
445 isl_map_free(map);
446 return 0;
449 data->res = isl_union_map_add_map(data->res, map);
451 return 0;
454 static __isl_give isl_union_map *gen_bin_op(__isl_take isl_union_map *umap1,
455 __isl_take isl_union_map *umap2, int (*fn)(void **, void *))
457 struct isl_union_map_gen_bin_data data = { NULL, NULL };
459 umap1 = isl_union_map_align_params(umap1, isl_union_map_get_dim(umap2));
460 umap2 = isl_union_map_align_params(umap2, isl_union_map_get_dim(umap1));
462 if (!umap1 || !umap2)
463 goto error;
465 data.umap2 = umap2;
466 data.res = isl_union_map_alloc(isl_dim_copy(umap1->dim),
467 umap1->table.n);
468 if (isl_hash_table_foreach(umap1->dim->ctx, &umap1->table,
469 fn, &data) < 0)
470 goto error;
472 isl_union_map_free(umap1);
473 isl_union_map_free(umap2);
474 return data.res;
475 error:
476 isl_union_map_free(umap1);
477 isl_union_map_free(umap2);
478 isl_union_map_free(data.res);
479 return NULL;
482 __isl_give isl_union_map *isl_union_map_subtract(
483 __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
485 return gen_bin_op(umap1, umap2, &subtract_entry);
488 __isl_give isl_union_set *isl_union_set_subtract(
489 __isl_take isl_union_set *uset1, __isl_take isl_union_set *uset2)
491 return isl_union_map_subtract(uset1, uset2);
494 struct isl_union_map_match_bin_data {
495 isl_union_map *umap2;
496 isl_union_map *res;
497 __isl_give isl_map *(*fn)(__isl_take isl_map*, __isl_take isl_map*);
500 static int match_bin_entry(void **entry, void *user)
502 struct isl_union_map_match_bin_data *data = user;
503 uint32_t hash;
504 struct isl_hash_table_entry *entry2;
505 isl_map *map = *entry;
506 int empty;
508 hash = isl_dim_get_hash(map->dim);
509 entry2 = isl_hash_table_find(data->umap2->dim->ctx, &data->umap2->table,
510 hash, &has_dim, map->dim, 0);
511 if (!entry2)
512 return 0;
514 map = isl_map_copy(map);
515 map = data->fn(map, isl_map_copy(entry2->data));
517 empty = isl_map_is_empty(map);
518 if (empty < 0) {
519 isl_map_free(map);
520 return -1;
522 if (empty) {
523 isl_map_free(map);
524 return 0;
527 data->res = isl_union_map_add_map(data->res, map);
529 return 0;
532 static __isl_give isl_union_map *match_bin_op(__isl_take isl_union_map *umap1,
533 __isl_take isl_union_map *umap2,
534 __isl_give isl_map *(*fn)(__isl_take isl_map*, __isl_take isl_map*))
536 struct isl_union_map_match_bin_data data = { NULL, NULL, fn };
538 umap1 = isl_union_map_align_params(umap1, isl_union_map_get_dim(umap2));
539 umap2 = isl_union_map_align_params(umap2, isl_union_map_get_dim(umap1));
541 if (!umap1 || !umap2)
542 goto error;
544 data.umap2 = umap2;
545 data.res = isl_union_map_alloc(isl_dim_copy(umap1->dim),
546 umap1->table.n);
547 if (isl_hash_table_foreach(umap1->dim->ctx, &umap1->table,
548 &match_bin_entry, &data) < 0)
549 goto error;
551 isl_union_map_free(umap1);
552 isl_union_map_free(umap2);
553 return data.res;
554 error:
555 isl_union_map_free(umap1);
556 isl_union_map_free(umap2);
557 isl_union_map_free(data.res);
558 return NULL;
561 __isl_give isl_union_map *isl_union_map_intersect(
562 __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
564 return match_bin_op(umap1, umap2, &isl_map_intersect);
567 __isl_give isl_union_set *isl_union_set_intersect(
568 __isl_take isl_union_set *uset1, __isl_take isl_union_set *uset2)
570 return isl_union_map_intersect(uset1, uset2);
573 __isl_give isl_union_map *isl_union_map_gist(__isl_take isl_union_map *umap,
574 __isl_take isl_union_map *context)
576 return match_bin_op(umap, context, &isl_map_gist);
579 __isl_give isl_union_set *isl_union_set_gist(__isl_take isl_union_set *uset,
580 __isl_take isl_union_set *context)
582 return isl_union_map_gist(uset, context);
585 static __isl_give isl_map *lex_le_set(__isl_take isl_map *set1,
586 __isl_take isl_map *set2)
588 return isl_set_lex_le_set((isl_set *)set1, (isl_set *)set2);
591 static __isl_give isl_map *lex_lt_set(__isl_take isl_map *set1,
592 __isl_take isl_map *set2)
594 return isl_set_lex_lt_set((isl_set *)set1, (isl_set *)set2);
597 __isl_give isl_union_map *isl_union_set_lex_lt_union_set(
598 __isl_take isl_union_set *uset1, __isl_take isl_union_set *uset2)
600 return match_bin_op(uset1, uset2, &lex_lt_set);
603 __isl_give isl_union_map *isl_union_set_lex_le_union_set(
604 __isl_take isl_union_set *uset1, __isl_take isl_union_set *uset2)
606 return match_bin_op(uset1, uset2, &lex_le_set);
609 __isl_give isl_union_map *isl_union_set_lex_gt_union_set(
610 __isl_take isl_union_set *uset1, __isl_take isl_union_set *uset2)
612 return isl_union_map_reverse(isl_union_set_lex_lt_union_set(uset2, uset1));
615 __isl_give isl_union_map *isl_union_set_lex_ge_union_set(
616 __isl_take isl_union_set *uset1, __isl_take isl_union_set *uset2)
618 return isl_union_map_reverse(isl_union_set_lex_le_union_set(uset2, uset1));
621 __isl_give isl_union_map *isl_union_map_lex_gt_union_map(
622 __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
624 return isl_union_map_reverse(isl_union_map_lex_lt_union_map(umap2, umap1));
627 __isl_give isl_union_map *isl_union_map_lex_ge_union_map(
628 __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
630 return isl_union_map_reverse(isl_union_map_lex_le_union_map(umap2, umap1));
633 static int intersect_domain_entry(void **entry, void *user)
635 struct isl_union_map_gen_bin_data *data = user;
636 uint32_t hash;
637 struct isl_hash_table_entry *entry2;
638 isl_dim *dim;
639 isl_map *map = *entry;
640 int empty;
642 dim = isl_map_get_dim(map);
643 dim = isl_dim_domain(dim);
644 hash = isl_dim_get_hash(dim);
645 entry2 = isl_hash_table_find(data->umap2->dim->ctx, &data->umap2->table,
646 hash, &has_dim, dim, 0);
647 isl_dim_free(dim);
648 if (!entry2)
649 return 0;
651 map = isl_map_copy(map);
652 map = isl_map_intersect_domain(map, isl_set_copy(entry2->data));
654 empty = isl_map_is_empty(map);
655 if (empty < 0) {
656 isl_map_free(map);
657 return -1;
659 if (empty) {
660 isl_map_free(map);
661 return 0;
664 data->res = isl_union_map_add_map(data->res, map);
666 return 0;
669 __isl_give isl_union_map *isl_union_map_intersect_domain(
670 __isl_take isl_union_map *umap, __isl_take isl_union_set *uset)
672 return gen_bin_op(umap, uset, &intersect_domain_entry);
675 static int intersect_range_entry(void **entry, void *user)
677 struct isl_union_map_gen_bin_data *data = user;
678 uint32_t hash;
679 struct isl_hash_table_entry *entry2;
680 isl_dim *dim;
681 isl_map *map = *entry;
682 int empty;
684 dim = isl_map_get_dim(map);
685 dim = isl_dim_range(dim);
686 hash = isl_dim_get_hash(dim);
687 entry2 = isl_hash_table_find(data->umap2->dim->ctx, &data->umap2->table,
688 hash, &has_dim, dim, 0);
689 isl_dim_free(dim);
690 if (!entry2)
691 return 0;
693 map = isl_map_copy(map);
694 map = isl_map_intersect_range(map, isl_set_copy(entry2->data));
696 empty = isl_map_is_empty(map);
697 if (empty < 0) {
698 isl_map_free(map);
699 return -1;
701 if (empty) {
702 isl_map_free(map);
703 return 0;
706 data->res = isl_union_map_add_map(data->res, map);
708 return 0;
711 __isl_give isl_union_map *isl_union_map_intersect_range(
712 __isl_take isl_union_map *umap, __isl_take isl_union_set *uset)
714 return gen_bin_op(umap, uset, &intersect_range_entry);
717 struct isl_union_map_bin_data {
718 isl_union_map *umap2;
719 isl_union_map *res;
720 isl_map *map;
721 int (*fn)(void **entry, void *user);
724 static int apply_range_entry(void **entry, void *user)
726 struct isl_union_map_bin_data *data = user;
727 isl_map *map2 = *entry;
728 int empty;
730 if (!isl_dim_tuple_match(data->map->dim, isl_dim_out,
731 map2->dim, isl_dim_in))
732 return 0;
734 map2 = isl_map_apply_range(isl_map_copy(data->map), isl_map_copy(map2));
736 empty = isl_map_is_empty(map2);
737 if (empty < 0) {
738 isl_map_free(map2);
739 return -1;
741 if (empty) {
742 isl_map_free(map2);
743 return 0;
746 data->res = isl_union_map_add_map(data->res, map2);
748 return 0;
751 static int bin_entry(void **entry, void *user)
753 struct isl_union_map_bin_data *data = user;
754 isl_map *map = *entry;
756 data->map = map;
757 if (isl_hash_table_foreach(data->umap2->dim->ctx, &data->umap2->table,
758 data->fn, data) < 0)
759 return -1;
761 return 0;
764 static __isl_give isl_union_map *bin_op(__isl_take isl_union_map *umap1,
765 __isl_take isl_union_map *umap2, int (*fn)(void **entry, void *user))
767 struct isl_union_map_bin_data data = { NULL, NULL, NULL, fn };
769 umap1 = isl_union_map_align_params(umap1, isl_union_map_get_dim(umap2));
770 umap2 = isl_union_map_align_params(umap2, isl_union_map_get_dim(umap1));
772 if (!umap1 || !umap2)
773 goto error;
775 data.umap2 = umap2;
776 data.res = isl_union_map_alloc(isl_dim_copy(umap1->dim),
777 umap1->table.n);
778 if (isl_hash_table_foreach(umap1->dim->ctx, &umap1->table,
779 &bin_entry, &data) < 0)
780 goto error;
782 isl_union_map_free(umap1);
783 isl_union_map_free(umap2);
784 return data.res;
785 error:
786 isl_union_map_free(umap1);
787 isl_union_map_free(umap2);
788 isl_union_map_free(data.res);
789 return NULL;
792 __isl_give isl_union_map *isl_union_map_apply_range(
793 __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
795 return bin_op(umap1, umap2, &apply_range_entry);
798 __isl_give isl_union_map *isl_union_map_apply_domain(
799 __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
801 umap1 = isl_union_map_reverse(umap1);
802 umap1 = isl_union_map_apply_range(umap1, umap2);
803 return isl_union_map_reverse(umap1);
806 __isl_give isl_union_set *isl_union_set_apply(
807 __isl_take isl_union_set *uset, __isl_take isl_union_map *umap)
809 return isl_union_map_apply_range(uset, umap);
812 static int map_lex_lt_entry(void **entry, void *user)
814 struct isl_union_map_bin_data *data = user;
815 isl_map *map2 = *entry;
817 if (!isl_dim_tuple_match(data->map->dim, isl_dim_out,
818 map2->dim, isl_dim_out))
819 return 0;
821 map2 = isl_map_lex_lt_map(isl_map_copy(data->map), isl_map_copy(map2));
823 data->res = isl_union_map_add_map(data->res, map2);
825 return 0;
828 __isl_give isl_union_map *isl_union_map_lex_lt_union_map(
829 __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
831 return bin_op(umap1, umap2, &map_lex_lt_entry);
834 static int map_lex_le_entry(void **entry, void *user)
836 struct isl_union_map_bin_data *data = user;
837 isl_map *map2 = *entry;
839 if (!isl_dim_tuple_match(data->map->dim, isl_dim_out,
840 map2->dim, isl_dim_out))
841 return 0;
843 map2 = isl_map_lex_le_map(isl_map_copy(data->map), isl_map_copy(map2));
845 data->res = isl_union_map_add_map(data->res, map2);
847 return 0;
850 __isl_give isl_union_map *isl_union_map_lex_le_union_map(
851 __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
853 return bin_op(umap1, umap2, &map_lex_le_entry);
856 static int product_entry(void **entry, void *user)
858 struct isl_union_map_bin_data *data = user;
859 isl_map *map2 = *entry;
861 map2 = isl_map_product(isl_map_copy(data->map), isl_map_copy(map2));
863 data->res = isl_union_map_add_map(data->res, map2);
865 return 0;
868 __isl_give isl_union_map *isl_union_map_product(__isl_take isl_union_map *umap1,
869 __isl_take isl_union_map *umap2)
871 return bin_op(umap1, umap2, &product_entry);
874 __isl_give isl_union_set *isl_union_set_product(__isl_take isl_union_set *uset1,
875 __isl_take isl_union_set *uset2)
877 return isl_union_map_product(uset1, uset2);
880 static int range_product_entry(void **entry, void *user)
882 struct isl_union_map_bin_data *data = user;
883 isl_map *map2 = *entry;
885 map2 = isl_map_range_product(isl_map_copy(data->map),
886 isl_map_copy(map2));
888 data->res = isl_union_map_add_map(data->res, map2);
890 return 0;
893 __isl_give isl_union_map *isl_union_map_range_product(
894 __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
896 return bin_op(umap1, umap2, &range_product_entry);
899 __isl_give isl_union_map *isl_union_map_from_range(
900 __isl_take isl_union_set *uset)
902 return uset;
905 __isl_give isl_union_map *isl_union_map_from_domain(
906 __isl_take isl_union_set *uset)
908 return isl_union_map_reverse(isl_union_map_from_range(uset));
911 __isl_give isl_union_map *isl_union_map_from_domain_and_range(
912 __isl_take isl_union_set *domain, __isl_take isl_union_set *range)
914 return isl_union_map_apply_range(isl_union_map_from_domain(domain),
915 isl_union_map_from_range(range));
918 static __isl_give isl_union_map *un_op(__isl_take isl_union_map *umap,
919 int (*fn)(void **, void *))
921 umap = isl_union_map_cow(umap);
922 if (!umap)
923 return NULL;
925 if (isl_hash_table_foreach(umap->dim->ctx, &umap->table, fn, NULL) < 0)
926 goto error;
928 return umap;
929 error:
930 isl_union_map_free(umap);
931 return NULL;
934 static int affine_entry(void **entry, void *user)
936 isl_map **map = (isl_map **)entry;
938 *map = isl_map_from_basic_map(isl_map_affine_hull(*map));
940 return *map ? 0 : -1;
943 __isl_give isl_union_map *isl_union_map_affine_hull(
944 __isl_take isl_union_map *umap)
946 return un_op(umap, &affine_entry);
949 __isl_give isl_union_set *isl_union_set_affine_hull(
950 __isl_take isl_union_set *uset)
952 return isl_union_map_affine_hull(uset);
955 static int polyhedral_entry(void **entry, void *user)
957 isl_map **map = (isl_map **)entry;
959 *map = isl_map_from_basic_map(isl_map_polyhedral_hull(*map));
961 return *map ? 0 : -1;
964 __isl_give isl_union_map *isl_union_map_polyhedral_hull(
965 __isl_take isl_union_map *umap)
967 return un_op(umap, &polyhedral_entry);
970 __isl_give isl_union_set *isl_union_set_polyhedral_hull(
971 __isl_take isl_union_set *uset)
973 return isl_union_map_polyhedral_hull(uset);
976 static int simple_entry(void **entry, void *user)
978 isl_map **map = (isl_map **)entry;
980 *map = isl_map_from_basic_map(isl_map_simple_hull(*map));
982 return *map ? 0 : -1;
985 __isl_give isl_union_map *isl_union_map_simple_hull(
986 __isl_take isl_union_map *umap)
988 return un_op(umap, &simple_entry);
991 __isl_give isl_union_set *isl_union_set_simple_hull(
992 __isl_take isl_union_set *uset)
994 return isl_union_map_simple_hull(uset);
997 static int inplace_entry(void **entry, void *user)
999 __isl_give isl_map *(*fn)(__isl_take isl_map *);
1000 isl_map **map = (isl_map **)entry;
1001 isl_map *copy;
1003 fn = *(__isl_give isl_map *(**)(__isl_take isl_map *)) user;
1004 copy = fn(isl_map_copy(*map));
1005 if (!copy)
1006 return -1;
1008 isl_map_free(*map);
1009 *map = copy;
1011 return 0;
1014 static __isl_give isl_union_map *inplace(__isl_take isl_union_map *umap,
1015 __isl_give isl_map *(*fn)(__isl_take isl_map *))
1017 if (!umap)
1018 return NULL;
1020 if (isl_hash_table_foreach(umap->dim->ctx, &umap->table,
1021 &inplace_entry, &fn) < 0)
1022 goto error;
1024 return umap;
1025 error:
1026 isl_union_map_free(umap);
1027 return NULL;
1030 __isl_give isl_union_map *isl_union_map_coalesce(
1031 __isl_take isl_union_map *umap)
1033 return inplace(umap, &isl_map_coalesce);
1036 __isl_give isl_union_set *isl_union_set_coalesce(
1037 __isl_take isl_union_set *uset)
1039 return isl_union_map_coalesce(uset);
1042 __isl_give isl_union_map *isl_union_map_detect_equalities(
1043 __isl_take isl_union_map *umap)
1045 return inplace(umap, &isl_map_detect_equalities);
1048 __isl_give isl_union_set *isl_union_set_detect_equalities(
1049 __isl_take isl_union_set *uset)
1051 return isl_union_map_detect_equalities(uset);
1054 __isl_give isl_union_map *isl_union_map_compute_divs(
1055 __isl_take isl_union_map *umap)
1057 return inplace(umap, &isl_map_compute_divs);
1060 __isl_give isl_union_set *isl_union_set_compute_divs(
1061 __isl_take isl_union_set *uset)
1063 return isl_union_map_compute_divs(uset);
1066 static int lexmin_entry(void **entry, void *user)
1068 isl_map **map = (isl_map **)entry;
1070 *map = isl_map_lexmin(*map);
1072 return *map ? 0 : -1;
1075 __isl_give isl_union_map *isl_union_map_lexmin(
1076 __isl_take isl_union_map *umap)
1078 return un_op(umap, &lexmin_entry);
1081 __isl_give isl_union_set *isl_union_set_lexmin(
1082 __isl_take isl_union_set *uset)
1084 return isl_union_map_lexmin(uset);
1087 static int lexmax_entry(void **entry, void *user)
1089 isl_map **map = (isl_map **)entry;
1091 *map = isl_map_lexmax(*map);
1093 return *map ? 0 : -1;
1096 __isl_give isl_union_map *isl_union_map_lexmax(
1097 __isl_take isl_union_map *umap)
1099 return un_op(umap, &lexmax_entry);
1102 __isl_give isl_union_set *isl_union_set_lexmax(
1103 __isl_take isl_union_set *uset)
1105 return isl_union_map_lexmax(uset);
1108 static __isl_give isl_union_set *cond_un_op(__isl_take isl_union_map *umap,
1109 int (*fn)(void **, void *))
1111 isl_union_set *res;
1113 if (!umap)
1114 return NULL;
1116 res = isl_union_map_alloc(isl_dim_copy(umap->dim), umap->table.n);
1117 if (isl_hash_table_foreach(umap->dim->ctx, &umap->table, fn, &res) < 0)
1118 goto error;
1120 isl_union_map_free(umap);
1121 return res;
1122 error:
1123 isl_union_map_free(umap);
1124 isl_union_set_free(res);
1125 return NULL;
1128 static int reverse_entry(void **entry, void *user)
1130 isl_map *map = *entry;
1131 isl_union_map **res = user;
1133 *res = isl_union_map_add_map(*res, isl_map_reverse(isl_map_copy(map)));
1135 return 0;
1138 __isl_give isl_union_map *isl_union_map_reverse(__isl_take isl_union_map *umap)
1140 return cond_un_op(umap, &reverse_entry);
1143 static int domain_entry(void **entry, void *user)
1145 isl_map *map = *entry;
1146 isl_union_set **res = user;
1148 *res = isl_union_set_add_set(*res, isl_map_domain(isl_map_copy(map)));
1150 return 0;
1153 __isl_give isl_union_set *isl_union_map_domain(__isl_take isl_union_map *umap)
1155 return cond_un_op(umap, &domain_entry);
1158 static int range_entry(void **entry, void *user)
1160 isl_map *map = *entry;
1161 isl_union_set **res = user;
1163 *res = isl_union_set_add_set(*res, isl_map_range(isl_map_copy(map)));
1165 return 0;
1168 __isl_give isl_union_set *isl_union_map_range(__isl_take isl_union_map *umap)
1170 return cond_un_op(umap, &range_entry);
1173 static int domain_map_entry(void **entry, void *user)
1175 isl_map *map = *entry;
1176 isl_union_set **res = user;
1178 *res = isl_union_map_add_map(*res,
1179 isl_map_domain_map(isl_map_copy(map)));
1181 return 0;
1184 __isl_give isl_union_map *isl_union_map_domain_map(
1185 __isl_take isl_union_map *umap)
1187 return cond_un_op(umap, &domain_map_entry);
1190 static int range_map_entry(void **entry, void *user)
1192 isl_map *map = *entry;
1193 isl_union_set **res = user;
1195 *res = isl_union_map_add_map(*res,
1196 isl_map_range_map(isl_map_copy(map)));
1198 return 0;
1201 __isl_give isl_union_map *isl_union_map_range_map(
1202 __isl_take isl_union_map *umap)
1204 return cond_un_op(umap, &range_map_entry);
1207 static int deltas_entry(void **entry, void *user)
1209 isl_map *map = *entry;
1210 isl_union_set **res = user;
1212 if (!isl_dim_tuple_match(map->dim, isl_dim_in, map->dim, isl_dim_out))
1213 return 0;
1215 *res = isl_union_set_add_set(*res, isl_map_deltas(isl_map_copy(map)));
1217 return 0;
1220 __isl_give isl_union_set *isl_union_map_deltas(__isl_take isl_union_map *umap)
1222 return cond_un_op(umap, &deltas_entry);
1225 static int deltas_map_entry(void **entry, void *user)
1227 isl_map *map = *entry;
1228 isl_union_map **res = user;
1230 if (!isl_dim_tuple_match(map->dim, isl_dim_in, map->dim, isl_dim_out))
1231 return 0;
1233 *res = isl_union_map_add_map(*res,
1234 isl_map_deltas_map(isl_map_copy(map)));
1236 return 0;
1239 __isl_give isl_union_map *isl_union_map_deltas_map(
1240 __isl_take isl_union_map *umap)
1242 return cond_un_op(umap, &deltas_map_entry);
1245 static int identity_entry(void **entry, void *user)
1247 isl_set *set = *entry;
1248 isl_union_map **res = user;
1250 *res = isl_union_map_add_map(*res, isl_set_identity(isl_set_copy(set)));
1252 return 0;
1255 __isl_give isl_union_map *isl_union_set_identity(__isl_take isl_union_set *uset)
1257 return cond_un_op(uset, &identity_entry);
1260 static int unwrap_entry(void **entry, void *user)
1262 isl_set *set = *entry;
1263 isl_union_set **res = user;
1265 if (!isl_set_is_wrapping(set))
1266 return 0;
1268 *res = isl_union_map_add_map(*res, isl_set_unwrap(isl_set_copy(set)));
1270 return 0;
1273 __isl_give isl_union_map *isl_union_set_unwrap(__isl_take isl_union_set *uset)
1275 return cond_un_op(uset, &unwrap_entry);
1278 static int wrap_entry(void **entry, void *user)
1280 isl_map *map = *entry;
1281 isl_union_set **res = user;
1283 *res = isl_union_set_add_set(*res, isl_map_wrap(isl_map_copy(map)));
1285 return 0;
1288 __isl_give isl_union_set *isl_union_map_wrap(__isl_take isl_union_map *umap)
1290 return cond_un_op(umap, &wrap_entry);
1293 struct isl_union_map_is_subset_data {
1294 isl_union_map *umap2;
1295 int is_subset;
1298 static int is_subset_entry(void **entry, void *user)
1300 struct isl_union_map_is_subset_data *data = user;
1301 uint32_t hash;
1302 struct isl_hash_table_entry *entry2;
1303 isl_map *map = *entry;
1305 hash = isl_dim_get_hash(map->dim);
1306 entry2 = isl_hash_table_find(data->umap2->dim->ctx, &data->umap2->table,
1307 hash, &has_dim, map->dim, 0);
1308 if (!entry2) {
1309 data->is_subset = 0;
1310 return -1;
1313 data->is_subset = isl_map_is_subset(map, entry2->data);
1314 if (data->is_subset < 0 || !data->is_subset)
1315 return -1;
1317 return 0;
1320 int isl_union_map_is_subset(__isl_keep isl_union_map *umap1,
1321 __isl_keep isl_union_map *umap2)
1323 struct isl_union_map_is_subset_data data = { NULL, 1 };
1325 umap1 = isl_union_map_copy(umap1);
1326 umap2 = isl_union_map_copy(umap2);
1327 umap1 = isl_union_map_align_params(umap1, isl_union_map_get_dim(umap2));
1328 umap2 = isl_union_map_align_params(umap2, isl_union_map_get_dim(umap1));
1330 if (!umap1 || !umap2)
1331 goto error;
1333 data.umap2 = umap2;
1334 if (isl_hash_table_foreach(umap1->dim->ctx, &umap1->table,
1335 &is_subset_entry, &data) < 0 &&
1336 data.is_subset)
1337 goto error;
1339 isl_union_map_free(umap1);
1340 isl_union_map_free(umap2);
1342 return data.is_subset;
1343 error:
1344 isl_union_map_free(umap1);
1345 isl_union_map_free(umap2);
1346 return -1;
1349 int isl_union_set_is_subset(__isl_keep isl_union_set *uset1,
1350 __isl_keep isl_union_set *uset2)
1352 return isl_union_map_is_subset(uset1, uset2);
1355 int isl_union_map_is_equal(__isl_keep isl_union_map *umap1,
1356 __isl_keep isl_union_map *umap2)
1358 int is_subset;
1360 if (!umap1 || !umap2)
1361 return -1;
1362 is_subset = isl_union_map_is_subset(umap1, umap2);
1363 if (is_subset != 1)
1364 return is_subset;
1365 is_subset = isl_union_map_is_subset(umap2, umap1);
1366 return is_subset;
1369 int isl_union_set_is_equal(__isl_keep isl_union_set *uset1,
1370 __isl_keep isl_union_set *uset2)
1372 return isl_union_map_is_equal(uset1, uset2);
1375 int isl_union_map_is_strict_subset(__isl_keep isl_union_map *umap1,
1376 __isl_keep isl_union_map *umap2)
1378 int is_subset;
1380 if (!umap1 || !umap2)
1381 return -1;
1382 is_subset = isl_union_map_is_subset(umap1, umap2);
1383 if (is_subset != 1)
1384 return is_subset;
1385 is_subset = isl_union_map_is_subset(umap2, umap1);
1386 if (is_subset == -1)
1387 return is_subset;
1388 return !is_subset;
1391 int isl_union_set_is_strict_subset(__isl_keep isl_union_set *uset1,
1392 __isl_keep isl_union_set *uset2)
1394 return isl_union_map_is_strict_subset(uset1, uset2);
1397 static int sample_entry(void **entry, void *user)
1399 isl_basic_map **sample = (isl_basic_map **)user;
1400 isl_map *map = *entry;
1402 *sample = isl_map_sample(isl_map_copy(map));
1403 if (!*sample)
1404 return -1;
1405 if (!isl_basic_map_fast_is_empty(*sample))
1406 return -1;
1407 return 0;
1410 __isl_give isl_basic_map *isl_union_map_sample(__isl_take isl_union_map *umap)
1412 isl_basic_map *sample = NULL;
1414 if (!umap)
1415 return NULL;
1417 if (isl_hash_table_foreach(umap->dim->ctx, &umap->table,
1418 &sample_entry, &sample) < 0 &&
1419 !sample)
1420 goto error;
1422 if (!sample)
1423 sample = isl_basic_map_empty(isl_union_map_get_dim(umap));
1425 isl_union_map_free(umap);
1427 return sample;
1428 error:
1429 isl_union_map_free(umap);
1430 return NULL;
1433 __isl_give isl_basic_set *isl_union_set_sample(__isl_take isl_union_set *uset)
1435 return (isl_basic_set *)isl_union_map_sample(uset);
1438 static int empty_entry(void **entry, void *user)
1440 int *empty = user;
1441 isl_map *map = *entry;
1443 if (isl_map_is_empty(map))
1444 return 0;
1446 *empty = 0;
1448 return -1;
1451 __isl_give int isl_union_map_is_empty(__isl_keep isl_union_map *umap)
1453 int empty = 1;
1455 if (!umap)
1456 return -1;
1458 if (isl_hash_table_foreach(umap->dim->ctx, &umap->table,
1459 &empty_entry, &empty) < 0 && empty)
1460 return -1;
1462 return empty;
1465 int isl_union_set_is_empty(__isl_keep isl_union_set *uset)
1467 return isl_union_map_is_empty(uset);
1470 static int zip_entry(void **entry, void *user)
1472 isl_map *map = *entry;
1473 isl_union_map **res = user;
1475 if (!isl_map_can_zip(map))
1476 return 0;
1478 *res = isl_union_map_add_map(*res, isl_map_zip(isl_map_copy(map)));
1480 return 0;
1483 __isl_give isl_union_map *isl_union_map_zip(__isl_take isl_union_map *umap)
1485 return cond_un_op(umap, &zip_entry);
1488 static int lift_entry(void **entry, void *user)
1490 isl_set *set = *entry;
1491 isl_union_set **res = user;
1493 *res = isl_union_set_add_set(*res, isl_set_lift(isl_set_copy(set)));
1495 return 0;
1498 __isl_give isl_union_set *isl_union_set_lift(__isl_take isl_union_set *uset)
1500 return cond_un_op(uset, &lift_entry);
1503 static int coefficients_entry(void **entry, void *user)
1505 isl_set *set = *entry;
1506 isl_union_set **res = user;
1508 set = isl_set_copy(set);
1509 set = isl_set_from_basic_set(isl_set_coefficients(set));
1510 *res = isl_union_set_add_set(*res, set);
1512 return 0;
1515 __isl_give isl_union_set *isl_union_set_coefficients(
1516 __isl_take isl_union_set *uset)
1518 isl_ctx *ctx;
1519 isl_dim *dim;
1520 isl_union_set *res;
1522 if (!uset)
1523 return NULL;
1525 ctx = isl_union_set_get_ctx(uset);
1526 dim = isl_dim_set_alloc(ctx, 0, 0);
1527 res = isl_union_map_alloc(dim, uset->table.n);
1528 if (isl_hash_table_foreach(uset->dim->ctx, &uset->table,
1529 &coefficients_entry, &res) < 0)
1530 goto error;
1532 isl_union_set_free(uset);
1533 return res;
1534 error:
1535 isl_union_set_free(uset);
1536 isl_union_set_free(res);
1537 return NULL;
1540 static int solutions_entry(void **entry, void *user)
1542 isl_set *set = *entry;
1543 isl_union_set **res = user;
1545 set = isl_set_copy(set);
1546 set = isl_set_from_basic_set(isl_set_solutions(set));
1547 if (!*res)
1548 *res = isl_union_set_from_set(set);
1549 else
1550 *res = isl_union_set_add_set(*res, set);
1552 if (!*res)
1553 return -1;
1555 return 0;
1558 __isl_give isl_union_set *isl_union_set_solutions(
1559 __isl_take isl_union_set *uset)
1561 isl_ctx *ctx;
1562 isl_dim *dim;
1563 isl_union_set *res = NULL;
1565 if (!uset)
1566 return NULL;
1568 if (uset->table.n == 0) {
1569 res = isl_union_set_empty(isl_union_set_get_dim(uset));
1570 isl_union_set_free(uset);
1571 return res;
1574 if (isl_hash_table_foreach(uset->dim->ctx, &uset->table,
1575 &solutions_entry, &res) < 0)
1576 goto error;
1578 isl_union_set_free(uset);
1579 return res;
1580 error:
1581 isl_union_set_free(uset);
1582 isl_union_set_free(res);
1583 return NULL;