add isl_union_map_is_bijective
[platform/upstream/isl.git] / isl_union_map.c
1 /*
2  * Copyright 2010      INRIA Saclay
3  *
4  * Use of this software is governed by the GNU LGPLv2.1 license
5  *
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  */
10
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>
19
20 static __isl_give isl_union_map *isl_union_map_alloc(__isl_take isl_dim *dim,
21         int size)
22 {
23         isl_union_map *umap;
24
25         if (!dim)
26                 return NULL;
27
28         umap = isl_calloc_type(dim->ctx, isl_union_map);
29         if (!umap)
30                 return NULL;
31
32         umap->ref = 1;
33         umap->dim = dim;
34         if (isl_hash_table_init(dim->ctx, &umap->table, size) < 0)
35                 goto error;
36
37         return umap;
38 error:
39         isl_dim_free(dim);
40         isl_union_map_free(umap);
41         return NULL;
42 }
43
44 __isl_give isl_union_map *isl_union_map_empty(__isl_take isl_dim *dim)
45 {
46         return isl_union_map_alloc(dim, 16);
47 }
48
49 __isl_give isl_union_set *isl_union_set_empty(__isl_take isl_dim *dim)
50 {
51         return isl_union_map_empty(dim);
52 }
53
54 isl_ctx *isl_union_map_get_ctx(__isl_keep isl_union_map *umap)
55 {
56         return umap ? umap->dim->ctx : NULL;
57 }
58
59 isl_ctx *isl_union_set_get_ctx(__isl_keep isl_union_set *uset)
60 {
61         return uset ? uset->dim->ctx : NULL;
62 }
63
64 __isl_give isl_dim *isl_union_map_get_dim(__isl_keep isl_union_map *umap)
65 {
66         if (!umap)
67                 return NULL;
68         return isl_dim_copy(umap->dim);
69 }
70
71 __isl_give isl_dim *isl_union_set_get_dim(__isl_keep isl_union_set *uset)
72 {
73         return isl_union_map_get_dim(uset);
74 }
75
76 static int free_umap_entry(void **entry, void *user)
77 {
78         isl_map *map = *entry;
79         isl_map_free(map);
80         return 0;
81 }
82
83 static int add_map(__isl_take isl_map *map, void *user)
84 {
85         isl_union_map **umap = (isl_union_map **)user;
86
87         *umap = isl_union_map_add_map(*umap, map);
88
89         return 0;
90 }
91
92 __isl_give isl_union_map *isl_union_map_dup(__isl_keep isl_union_map *umap)
93 {
94         isl_union_map *dup;
95
96         if (!umap)
97                 return NULL;
98
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;
106 }
107
108 __isl_give isl_union_map *isl_union_map_cow(__isl_take isl_union_map *umap)
109 {
110         if (!umap)
111                 return NULL;
112
113         if (umap->ref == 1)
114                 return umap;
115         umap->ref--;
116         return isl_union_map_dup(umap);
117 }
118
119 struct isl_union_align {
120         isl_reordering *exp;
121         isl_union_map *res;
122 };
123
124 static int align_entry(void **entry, void *user)
125 {
126         isl_map *map = *entry;
127         isl_reordering *exp;
128         struct isl_union_align *data = user;
129
130         exp = isl_reordering_extend_dim(isl_reordering_copy(data->exp),
131                                     isl_map_get_dim(map));
132
133         data->res = isl_union_map_add_map(data->res,
134                                         isl_map_realign(isl_map_copy(map), exp));
135
136         return 0;
137 }
138
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.
143  */
144 __isl_give isl_union_map *isl_union_map_align_params(
145         __isl_take isl_union_map *umap, __isl_take isl_dim *model)
146 {
147         struct isl_union_align data = { NULL, NULL };
148
149         if (!umap || !model)
150                 goto error;
151
152         if (isl_dim_match(umap->dim, isl_dim_param, model, isl_dim_param)) {
153                 isl_dim_free(model);
154                 return umap;
155         }
156
157         data.exp = isl_parameter_alignment_reordering(umap->dim, model);
158         if (!data.exp)
159                 goto error;
160
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;
166
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;
177 }
178
179 __isl_give isl_union_set *isl_union_set_align_params(
180         __isl_take isl_union_set *uset, __isl_take isl_dim *model)
181 {
182         return isl_union_map_align_params(uset, model);
183 }
184
185 __isl_give isl_union_map *isl_union_map_union(__isl_take isl_union_map *umap1,
186         __isl_take isl_union_map *umap2)
187 {
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));
190
191         umap1 = isl_union_map_cow(umap1);
192
193         if (!umap1 || !umap2)
194                 goto error;
195
196         if (isl_union_map_foreach_map(umap2, &add_map, &umap1) < 0)
197                 goto error;
198
199         isl_union_map_free(umap2);
200
201         return umap1;
202 error:
203         isl_union_map_free(umap1);
204         isl_union_map_free(umap2);
205         return NULL;
206 }
207
208 __isl_give isl_union_set *isl_union_set_union(__isl_take isl_union_set *uset1,
209         __isl_take isl_union_set *uset2)
210 {
211         return isl_union_map_union(uset1, uset2);
212 }
213
214 __isl_give isl_union_map *isl_union_map_copy(__isl_keep isl_union_map *umap)
215 {
216         if (!umap)
217                 return NULL;
218
219         umap->ref++;
220         return umap;
221 }
222
223 __isl_give isl_union_set *isl_union_set_copy(__isl_keep isl_union_set *uset)
224 {
225         return isl_union_map_copy(uset);
226 }
227
228 void isl_union_map_free(__isl_take isl_union_map *umap)
229 {
230         if (!umap)
231                 return;
232
233         if (--umap->ref > 0)
234                 return;
235
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);
241 }
242
243 void isl_union_set_free(__isl_take isl_union_set *uset)
244 {
245         isl_union_map_free(uset);
246 }
247
248 static int has_dim(const void *entry, const void *val)
249 {
250         isl_map *map = (isl_map *)entry;
251         isl_dim *dim = (isl_dim *)val;
252
253         return isl_dim_equal(map->dim, dim);
254 }
255
256 __isl_give isl_union_map *isl_union_map_add_map(__isl_take isl_union_map *umap,
257         __isl_take isl_map *map)
258 {
259         uint32_t hash;
260         struct isl_hash_table_entry *entry;
261
262         if (isl_map_fast_is_empty(map)) {
263                 isl_map_free(map);
264                 return umap;
265         }
266
267         umap = isl_union_map_cow(umap);
268
269         if (!map || !umap)
270                 goto error;
271
272         isl_assert(map->ctx, isl_dim_match(map->dim, isl_dim_param, umap->dim,
273                                            isl_dim_param), goto error);
274
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;
280
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);
288         }
289
290         return umap;
291 error:
292         isl_map_free(map);
293         isl_union_map_free(umap);
294         return NULL;
295 }
296
297 __isl_give isl_union_set *isl_union_set_add_set(__isl_take isl_union_set *uset,
298         __isl_take isl_set *set)
299 {
300         return isl_union_map_add_map(uset, (isl_map *)set);
301 }
302
303 __isl_give isl_union_map *isl_union_map_from_map(__isl_take isl_map *map)
304 {
305         isl_dim *dim;
306         isl_union_map *umap;
307
308         if (!map)
309                 return NULL;
310
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);
316
317         return umap;
318 }
319
320 __isl_give isl_union_set *isl_union_set_from_set(__isl_take isl_set *set)
321 {
322         return isl_union_map_from_map((isl_map *)set);
323 }
324
325 struct isl_union_map_foreach_data
326 {
327         int (*fn)(__isl_take isl_map *map, void *user);
328         void *user;
329 };
330
331 static int call_on_copy(void **entry, void *user)
332 {
333         isl_map *map = *entry;
334         struct isl_union_map_foreach_data *data;
335         data = (struct isl_union_map_foreach_data *)user;
336
337         return data->fn(isl_map_copy(map), data->user);
338 }
339
340 int isl_union_map_n_map(__isl_keep isl_union_map *umap)
341 {
342         return umap ? umap->table.n : 0;
343 }
344
345 int isl_union_set_n_set(__isl_keep isl_union_set *uset)
346 {
347         return uset ? uset->table.n : 0;
348 }
349
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)
352 {
353         struct isl_union_map_foreach_data data = { fn, user };
354
355         if (!umap)
356                 return -1;
357
358         return isl_hash_table_foreach(umap->dim->ctx, &umap->table,
359                                       &call_on_copy, &data);
360 }
361
362 static int copy_map(void **entry, void *user)
363 {
364         isl_map *map = *entry;
365         isl_map **map_p = user;
366
367         *map_p = isl_map_copy(map);
368
369         return -1;
370 }
371
372 __isl_give isl_map *isl_union_map_copy_map(__isl_keep isl_union_map *umap)
373 {
374         isl_map *map = NULL;
375
376         if (!umap || umap->table.n == 0)
377                 return NULL;
378
379         isl_hash_table_foreach(umap->dim->ctx, &umap->table, &copy_map, &map);
380
381         return map;
382 }
383
384 __isl_give isl_set *isl_union_set_copy_set(__isl_keep isl_union_set *uset)
385 {
386         return isl_union_map_copy_map(uset);
387 }
388
389 __isl_give isl_map *isl_union_map_extract_map(__isl_keep isl_union_map *umap,
390         __isl_take isl_dim *dim)
391 {
392         uint32_t hash;
393         struct isl_hash_table_entry *entry;
394
395         if (!umap || !dim)
396                 goto error;
397
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;
408 }
409
410 __isl_give isl_set *isl_union_set_extract_set(__isl_keep isl_union_set *uset,
411         __isl_take isl_dim *dim)
412 {
413         return (isl_set *)isl_union_map_extract_map(uset, dim);
414 }
415
416 /* Check if umap contains a map in the given space.
417  */
418 __isl_give int isl_union_map_contains(__isl_keep isl_union_map *umap,
419         __isl_keep isl_dim *dim)
420 {
421         uint32_t hash;
422         struct isl_hash_table_entry *entry;
423
424         if (!umap || !dim)
425                 return -1;
426
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;
431 }
432
433 __isl_give int isl_union_set_contains(__isl_keep isl_union_set *uset,
434         __isl_keep isl_dim *dim)
435 {
436         return isl_union_map_contains(uset, dim);
437 }
438
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)
441 {
442         return isl_union_map_foreach_map(uset,
443                 (int(*)(__isl_take isl_map *, void*))fn, user);
444 }
445
446 struct isl_union_set_foreach_point_data {
447         int (*fn)(__isl_take isl_point *pnt, void *user);
448         void *user;
449 };
450
451 static int foreach_point(__isl_take isl_set *set, void *user)
452 {
453         struct isl_union_set_foreach_point_data *data = user;
454         int r;
455
456         r = isl_set_foreach_point(set, data->fn, data->user);
457         isl_set_free(set);
458
459         return r;
460 }
461
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)
464 {
465         struct isl_union_set_foreach_point_data data = { fn, user };
466         return isl_union_set_foreach_set(uset, &foreach_point, &data);
467 }
468
469 struct isl_union_map_gen_bin_data {
470         isl_union_map *umap2;
471         isl_union_map *res;
472 };
473
474 static int subtract_entry(void **entry, void *user)
475 {
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;
480
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));
488
489                 empty = isl_map_is_empty(map);
490                 if (empty < 0) {
491                         isl_map_free(map);
492                         return -1;
493                 }
494                 if (empty) {
495                         isl_map_free(map);
496                         return 0;
497                 }
498         }
499         data->res = isl_union_map_add_map(data->res, map);
500
501         return 0;
502 }
503
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 *))
506 {
507         struct isl_union_map_gen_bin_data data = { NULL, NULL };
508
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));
511
512         if (!umap1 || !umap2)
513                 goto error;
514
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;
521
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;
530 }
531
532 __isl_give isl_union_map *isl_union_map_subtract(
533         __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
534 {
535         return gen_bin_op(umap1, umap2, &subtract_entry);
536 }
537
538 __isl_give isl_union_set *isl_union_set_subtract(
539         __isl_take isl_union_set *uset1, __isl_take isl_union_set *uset2)
540 {
541         return isl_union_map_subtract(uset1, uset2);
542 }
543
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*);
548 };
549
550 static int match_bin_entry(void **entry, void *user)
551 {
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;
557
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;
563
564         map = isl_map_copy(map);
565         map = data->fn(map, isl_map_copy(entry2->data));
566
567         empty = isl_map_is_empty(map);
568         if (empty < 0) {
569                 isl_map_free(map);
570                 return -1;
571         }
572         if (empty) {
573                 isl_map_free(map);
574                 return 0;
575         }
576
577         data->res = isl_union_map_add_map(data->res, map);
578
579         return 0;
580 }
581
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*))
585 {
586         struct isl_union_map_match_bin_data data = { NULL, NULL, fn };
587
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));
590
591         if (!umap1 || !umap2)
592                 goto error;
593
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;
600
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;
609 }
610
611 __isl_give isl_union_map *isl_union_map_intersect(
612         __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
613 {
614         return match_bin_op(umap1, umap2, &isl_map_intersect);
615 }
616
617 __isl_give isl_union_set *isl_union_set_intersect(
618         __isl_take isl_union_set *uset1, __isl_take isl_union_set *uset2)
619 {
620         return isl_union_map_intersect(uset1, uset2);
621 }
622
623 __isl_give isl_union_map *isl_union_map_gist(__isl_take isl_union_map *umap,
624         __isl_take isl_union_map *context)
625 {
626         return match_bin_op(umap, context, &isl_map_gist);
627 }
628
629 __isl_give isl_union_set *isl_union_set_gist(__isl_take isl_union_set *uset,
630         __isl_take isl_union_set *context)
631 {
632         return isl_union_map_gist(uset, context);
633 }
634
635 static __isl_give isl_map *lex_le_set(__isl_take isl_map *set1,
636         __isl_take isl_map *set2)
637 {
638         return isl_set_lex_le_set((isl_set *)set1, (isl_set *)set2);
639 }
640
641 static __isl_give isl_map *lex_lt_set(__isl_take isl_map *set1,
642         __isl_take isl_map *set2)
643 {
644         return isl_set_lex_lt_set((isl_set *)set1, (isl_set *)set2);
645 }
646
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)
649 {
650         return match_bin_op(uset1, uset2, &lex_lt_set);
651 }
652
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)
655 {
656         return match_bin_op(uset1, uset2, &lex_le_set);
657 }
658
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)
661 {
662         return isl_union_map_reverse(isl_union_set_lex_lt_union_set(uset2, uset1));
663 }
664
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)
667 {
668         return isl_union_map_reverse(isl_union_set_lex_le_union_set(uset2, uset1));
669 }
670
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)
673 {
674         return isl_union_map_reverse(isl_union_map_lex_lt_union_map(umap2, umap1));
675 }
676
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)
679 {
680         return isl_union_map_reverse(isl_union_map_lex_le_union_map(umap2, umap1));
681 }
682
683 static int intersect_domain_entry(void **entry, void *user)
684 {
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;
691
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;
700
701         map = isl_map_copy(map);
702         map = isl_map_intersect_domain(map, isl_set_copy(entry2->data));
703
704         empty = isl_map_is_empty(map);
705         if (empty < 0) {
706                 isl_map_free(map);
707                 return -1;
708         }
709         if (empty) {
710                 isl_map_free(map);
711                 return 0;
712         }
713
714         data->res = isl_union_map_add_map(data->res, map);
715
716         return 0;
717 }
718
719 __isl_give isl_union_map *isl_union_map_intersect_domain(
720         __isl_take isl_union_map *umap, __isl_take isl_union_set *uset)
721 {
722         return gen_bin_op(umap, uset, &intersect_domain_entry);
723 }
724
725 static int intersect_range_entry(void **entry, void *user)
726 {
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;
733
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;
742
743         map = isl_map_copy(map);
744         map = isl_map_intersect_range(map, isl_set_copy(entry2->data));
745
746         empty = isl_map_is_empty(map);
747         if (empty < 0) {
748                 isl_map_free(map);
749                 return -1;
750         }
751         if (empty) {
752                 isl_map_free(map);
753                 return 0;
754         }
755
756         data->res = isl_union_map_add_map(data->res, map);
757
758         return 0;
759 }
760
761 __isl_give isl_union_map *isl_union_map_intersect_range(
762         __isl_take isl_union_map *umap, __isl_take isl_union_set *uset)
763 {
764         return gen_bin_op(umap, uset, &intersect_range_entry);
765 }
766
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);
772 };
773
774 static int apply_range_entry(void **entry, void *user)
775 {
776         struct isl_union_map_bin_data *data = user;
777         isl_map *map2 = *entry;
778         int empty;
779
780         if (!isl_dim_tuple_match(data->map->dim, isl_dim_out,
781                                  map2->dim, isl_dim_in))
782                 return 0;
783
784         map2 = isl_map_apply_range(isl_map_copy(data->map), isl_map_copy(map2));
785
786         empty = isl_map_is_empty(map2);
787         if (empty < 0) {
788                 isl_map_free(map2);
789                 return -1;
790         }
791         if (empty) {
792                 isl_map_free(map2);
793                 return 0;
794         }
795
796         data->res = isl_union_map_add_map(data->res, map2);
797
798         return 0;
799 }
800
801 static int bin_entry(void **entry, void *user)
802 {
803         struct isl_union_map_bin_data *data = user;
804         isl_map *map = *entry;
805
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;
810
811         return 0;
812 }
813
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))
816 {
817         struct isl_union_map_bin_data data = { NULL, NULL, NULL, fn };
818
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));
821
822         if (!umap1 || !umap2)
823                 goto error;
824
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;
831
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;
840 }
841
842 __isl_give isl_union_map *isl_union_map_apply_range(
843         __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
844 {
845         return bin_op(umap1, umap2, &apply_range_entry);
846 }
847
848 __isl_give isl_union_map *isl_union_map_apply_domain(
849         __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
850 {
851         umap1 = isl_union_map_reverse(umap1);
852         umap1 = isl_union_map_apply_range(umap1, umap2);
853         return isl_union_map_reverse(umap1);
854 }
855
856 __isl_give isl_union_set *isl_union_set_apply(
857         __isl_take isl_union_set *uset, __isl_take isl_union_map *umap)
858 {
859         return isl_union_map_apply_range(uset, umap);
860 }
861
862 static int map_lex_lt_entry(void **entry, void *user)
863 {
864         struct isl_union_map_bin_data *data = user;
865         isl_map *map2 = *entry;
866
867         if (!isl_dim_tuple_match(data->map->dim, isl_dim_out,
868                                  map2->dim, isl_dim_out))
869                 return 0;
870
871         map2 = isl_map_lex_lt_map(isl_map_copy(data->map), isl_map_copy(map2));
872
873         data->res = isl_union_map_add_map(data->res, map2);
874
875         return 0;
876 }
877
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)
880 {
881         return bin_op(umap1, umap2, &map_lex_lt_entry);
882 }
883
884 static int map_lex_le_entry(void **entry, void *user)
885 {
886         struct isl_union_map_bin_data *data = user;
887         isl_map *map2 = *entry;
888
889         if (!isl_dim_tuple_match(data->map->dim, isl_dim_out,
890                                  map2->dim, isl_dim_out))
891                 return 0;
892
893         map2 = isl_map_lex_le_map(isl_map_copy(data->map), isl_map_copy(map2));
894
895         data->res = isl_union_map_add_map(data->res, map2);
896
897         return 0;
898 }
899
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)
902 {
903         return bin_op(umap1, umap2, &map_lex_le_entry);
904 }
905
906 static int product_entry(void **entry, void *user)
907 {
908         struct isl_union_map_bin_data *data = user;
909         isl_map *map2 = *entry;
910
911         map2 = isl_map_product(isl_map_copy(data->map), isl_map_copy(map2));
912
913         data->res = isl_union_map_add_map(data->res, map2);
914
915         return 0;
916 }
917
918 __isl_give isl_union_map *isl_union_map_product(__isl_take isl_union_map *umap1,
919         __isl_take isl_union_map *umap2)
920 {
921         return bin_op(umap1, umap2, &product_entry);
922 }
923
924 __isl_give isl_union_set *isl_union_set_product(__isl_take isl_union_set *uset1,
925         __isl_take isl_union_set *uset2)
926 {
927         return isl_union_map_product(uset1, uset2);
928 }
929
930 static int range_product_entry(void **entry, void *user)
931 {
932         struct isl_union_map_bin_data *data = user;
933         isl_map *map2 = *entry;
934
935         map2 = isl_map_range_product(isl_map_copy(data->map),
936                                      isl_map_copy(map2));
937
938         data->res = isl_union_map_add_map(data->res, map2);
939
940         return 0;
941 }
942
943 __isl_give isl_union_map *isl_union_map_range_product(
944         __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
945 {
946         return bin_op(umap1, umap2, &range_product_entry);
947 }
948
949 __isl_give isl_union_map *isl_union_map_from_range(
950         __isl_take isl_union_set *uset)
951 {
952         return uset;
953 }
954
955 __isl_give isl_union_map *isl_union_map_from_domain(
956         __isl_take isl_union_set *uset)
957 {
958         return isl_union_map_reverse(isl_union_map_from_range(uset));
959 }
960
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)
963 {
964         return isl_union_map_apply_range(isl_union_map_from_domain(domain),
965                                          isl_union_map_from_range(range));
966 }
967
968 static __isl_give isl_union_map *un_op(__isl_take isl_union_map *umap,
969         int (*fn)(void **, void *))
970 {
971         umap = isl_union_map_cow(umap);
972         if (!umap)
973                 return NULL;
974
975         if (isl_hash_table_foreach(umap->dim->ctx, &umap->table, fn, NULL) < 0)
976                 goto error;
977
978         return umap;
979 error:
980         isl_union_map_free(umap);
981         return NULL;
982 }
983
984 static int affine_entry(void **entry, void *user)
985 {
986         isl_map **map = (isl_map **)entry;
987
988         *map = isl_map_from_basic_map(isl_map_affine_hull(*map));
989
990         return *map ? 0 : -1;
991 }
992
993 __isl_give isl_union_map *isl_union_map_affine_hull(
994         __isl_take isl_union_map *umap)
995 {
996         return un_op(umap, &affine_entry);
997 }
998
999 __isl_give isl_union_set *isl_union_set_affine_hull(
1000         __isl_take isl_union_set *uset)
1001 {
1002         return isl_union_map_affine_hull(uset);
1003 }
1004
1005 static int polyhedral_entry(void **entry, void *user)
1006 {
1007         isl_map **map = (isl_map **)entry;
1008
1009         *map = isl_map_from_basic_map(isl_map_polyhedral_hull(*map));
1010
1011         return *map ? 0 : -1;
1012 }
1013
1014 __isl_give isl_union_map *isl_union_map_polyhedral_hull(
1015         __isl_take isl_union_map *umap)
1016 {
1017         return un_op(umap, &polyhedral_entry);
1018 }
1019
1020 __isl_give isl_union_set *isl_union_set_polyhedral_hull(
1021         __isl_take isl_union_set *uset)
1022 {
1023         return isl_union_map_polyhedral_hull(uset);
1024 }
1025
1026 static int simple_entry(void **entry, void *user)
1027 {
1028         isl_map **map = (isl_map **)entry;
1029
1030         *map = isl_map_from_basic_map(isl_map_simple_hull(*map));
1031
1032         return *map ? 0 : -1;
1033 }
1034
1035 __isl_give isl_union_map *isl_union_map_simple_hull(
1036         __isl_take isl_union_map *umap)
1037 {
1038         return un_op(umap, &simple_entry);
1039 }
1040
1041 __isl_give isl_union_set *isl_union_set_simple_hull(
1042         __isl_take isl_union_set *uset)
1043 {
1044         return isl_union_map_simple_hull(uset);
1045 }
1046
1047 static int inplace_entry(void **entry, void *user)
1048 {
1049         __isl_give isl_map *(*fn)(__isl_take isl_map *);
1050         isl_map **map = (isl_map **)entry;
1051         isl_map *copy;
1052
1053         fn = *(__isl_give isl_map *(**)(__isl_take isl_map *)) user;
1054         copy = fn(isl_map_copy(*map));
1055         if (!copy)
1056                 return -1;
1057
1058         isl_map_free(*map);
1059         *map = copy;
1060
1061         return 0;
1062 }
1063
1064 static __isl_give isl_union_map *inplace(__isl_take isl_union_map *umap,
1065         __isl_give isl_map *(*fn)(__isl_take isl_map *))
1066 {
1067         if (!umap)
1068                 return NULL;
1069
1070         if (isl_hash_table_foreach(umap->dim->ctx, &umap->table,
1071                                     &inplace_entry, &fn) < 0)
1072                 goto error;
1073
1074         return umap;
1075 error:
1076         isl_union_map_free(umap);
1077         return NULL;
1078 }
1079
1080 __isl_give isl_union_map *isl_union_map_coalesce(
1081         __isl_take isl_union_map *umap)
1082 {
1083         return inplace(umap, &isl_map_coalesce);
1084 }
1085
1086 __isl_give isl_union_set *isl_union_set_coalesce(
1087         __isl_take isl_union_set *uset)
1088 {
1089         return isl_union_map_coalesce(uset);
1090 }
1091
1092 __isl_give isl_union_map *isl_union_map_detect_equalities(
1093         __isl_take isl_union_map *umap)
1094 {
1095         return inplace(umap, &isl_map_detect_equalities);
1096 }
1097
1098 __isl_give isl_union_set *isl_union_set_detect_equalities(
1099         __isl_take isl_union_set *uset)
1100 {
1101         return isl_union_map_detect_equalities(uset);
1102 }
1103
1104 __isl_give isl_union_map *isl_union_map_compute_divs(
1105         __isl_take isl_union_map *umap)
1106 {
1107         return inplace(umap, &isl_map_compute_divs);
1108 }
1109
1110 __isl_give isl_union_set *isl_union_set_compute_divs(
1111         __isl_take isl_union_set *uset)
1112 {
1113         return isl_union_map_compute_divs(uset);
1114 }
1115
1116 static int lexmin_entry(void **entry, void *user)
1117 {
1118         isl_map **map = (isl_map **)entry;
1119
1120         *map = isl_map_lexmin(*map);
1121
1122         return *map ? 0 : -1;
1123 }
1124
1125 __isl_give isl_union_map *isl_union_map_lexmin(
1126         __isl_take isl_union_map *umap)
1127 {
1128         return un_op(umap, &lexmin_entry);
1129 }
1130
1131 __isl_give isl_union_set *isl_union_set_lexmin(
1132         __isl_take isl_union_set *uset)
1133 {
1134         return isl_union_map_lexmin(uset);
1135 }
1136
1137 static int lexmax_entry(void **entry, void *user)
1138 {
1139         isl_map **map = (isl_map **)entry;
1140
1141         *map = isl_map_lexmax(*map);
1142
1143         return *map ? 0 : -1;
1144 }
1145
1146 __isl_give isl_union_map *isl_union_map_lexmax(
1147         __isl_take isl_union_map *umap)
1148 {
1149         return un_op(umap, &lexmax_entry);
1150 }
1151
1152 __isl_give isl_union_set *isl_union_set_lexmax(
1153         __isl_take isl_union_set *uset)
1154 {
1155         return isl_union_map_lexmax(uset);
1156 }
1157
1158 static __isl_give isl_union_set *cond_un_op(__isl_take isl_union_map *umap,
1159         int (*fn)(void **, void *))
1160 {
1161         isl_union_set *res;
1162
1163         if (!umap)
1164                 return NULL;
1165
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;
1169
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;
1176 }
1177
1178 static int reverse_entry(void **entry, void *user)
1179 {
1180         isl_map *map = *entry;
1181         isl_union_map **res = user;
1182
1183         *res = isl_union_map_add_map(*res, isl_map_reverse(isl_map_copy(map)));
1184
1185         return 0;
1186 }
1187
1188 __isl_give isl_union_map *isl_union_map_reverse(__isl_take isl_union_map *umap)
1189 {
1190         return cond_un_op(umap, &reverse_entry);
1191 }
1192
1193 static int domain_entry(void **entry, void *user)
1194 {
1195         isl_map *map = *entry;
1196         isl_union_set **res = user;
1197
1198         *res = isl_union_set_add_set(*res, isl_map_domain(isl_map_copy(map)));
1199
1200         return 0;
1201 }
1202
1203 __isl_give isl_union_set *isl_union_map_domain(__isl_take isl_union_map *umap)
1204 {
1205         return cond_un_op(umap, &domain_entry);
1206 }
1207
1208 static int range_entry(void **entry, void *user)
1209 {
1210         isl_map *map = *entry;
1211         isl_union_set **res = user;
1212
1213         *res = isl_union_set_add_set(*res, isl_map_range(isl_map_copy(map)));
1214
1215         return 0;
1216 }
1217
1218 __isl_give isl_union_set *isl_union_map_range(__isl_take isl_union_map *umap)
1219 {
1220         return cond_un_op(umap, &range_entry);
1221 }
1222
1223 static int domain_map_entry(void **entry, void *user)
1224 {
1225         isl_map *map = *entry;
1226         isl_union_set **res = user;
1227
1228         *res = isl_union_map_add_map(*res,
1229                                         isl_map_domain_map(isl_map_copy(map)));
1230
1231         return 0;
1232 }
1233
1234 __isl_give isl_union_map *isl_union_map_domain_map(
1235         __isl_take isl_union_map *umap)
1236 {
1237         return cond_un_op(umap, &domain_map_entry);
1238 }
1239
1240 static int range_map_entry(void **entry, void *user)
1241 {
1242         isl_map *map = *entry;
1243         isl_union_set **res = user;
1244
1245         *res = isl_union_map_add_map(*res,
1246                                         isl_map_range_map(isl_map_copy(map)));
1247
1248         return 0;
1249 }
1250
1251 __isl_give isl_union_map *isl_union_map_range_map(
1252         __isl_take isl_union_map *umap)
1253 {
1254         return cond_un_op(umap, &range_map_entry);
1255 }
1256
1257 static int deltas_entry(void **entry, void *user)
1258 {
1259         isl_map *map = *entry;
1260         isl_union_set **res = user;
1261
1262         if (!isl_dim_tuple_match(map->dim, isl_dim_in, map->dim, isl_dim_out))
1263                 return 0;
1264
1265         *res = isl_union_set_add_set(*res, isl_map_deltas(isl_map_copy(map)));
1266
1267         return 0;
1268 }
1269
1270 __isl_give isl_union_set *isl_union_map_deltas(__isl_take isl_union_map *umap)
1271 {
1272         return cond_un_op(umap, &deltas_entry);
1273 }
1274
1275 static int deltas_map_entry(void **entry, void *user)
1276 {
1277         isl_map *map = *entry;
1278         isl_union_map **res = user;
1279
1280         if (!isl_dim_tuple_match(map->dim, isl_dim_in, map->dim, isl_dim_out))
1281                 return 0;
1282
1283         *res = isl_union_map_add_map(*res,
1284                                      isl_map_deltas_map(isl_map_copy(map)));
1285
1286         return 0;
1287 }
1288
1289 __isl_give isl_union_map *isl_union_map_deltas_map(
1290         __isl_take isl_union_map *umap)
1291 {
1292         return cond_un_op(umap, &deltas_map_entry);
1293 }
1294
1295 static int identity_entry(void **entry, void *user)
1296 {
1297         isl_set *set = *entry;
1298         isl_union_map **res = user;
1299
1300         *res = isl_union_map_add_map(*res, isl_set_identity(isl_set_copy(set)));
1301
1302         return 0;
1303 }
1304
1305 __isl_give isl_union_map *isl_union_set_identity(__isl_take isl_union_set *uset)
1306 {
1307         return cond_un_op(uset, &identity_entry);
1308 }
1309
1310 static int unwrap_entry(void **entry, void *user)
1311 {
1312         isl_set *set = *entry;
1313         isl_union_set **res = user;
1314
1315         if (!isl_set_is_wrapping(set))
1316                 return 0;
1317
1318         *res = isl_union_map_add_map(*res, isl_set_unwrap(isl_set_copy(set)));
1319
1320         return 0;
1321 }
1322
1323 __isl_give isl_union_map *isl_union_set_unwrap(__isl_take isl_union_set *uset)
1324 {
1325         return cond_un_op(uset, &unwrap_entry);
1326 }
1327
1328 static int wrap_entry(void **entry, void *user)
1329 {
1330         isl_map *map = *entry;
1331         isl_union_set **res = user;
1332
1333         *res = isl_union_set_add_set(*res, isl_map_wrap(isl_map_copy(map)));
1334
1335         return 0;
1336 }
1337
1338 __isl_give isl_union_set *isl_union_map_wrap(__isl_take isl_union_map *umap)
1339 {
1340         return cond_un_op(umap, &wrap_entry);
1341 }
1342
1343 struct isl_union_map_is_subset_data {
1344         isl_union_map *umap2;
1345         int is_subset;
1346 };
1347
1348 static int is_subset_entry(void **entry, void *user)
1349 {
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;
1354
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;
1361         }
1362
1363         data->is_subset = isl_map_is_subset(map, entry2->data);
1364         if (data->is_subset < 0 || !data->is_subset)
1365                 return -1;
1366
1367         return 0;
1368 }
1369
1370 int isl_union_map_is_subset(__isl_keep isl_union_map *umap1,
1371         __isl_keep isl_union_map *umap2)
1372 {
1373         struct isl_union_map_is_subset_data data = { NULL, 1 };
1374
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));
1379
1380         if (!umap1 || !umap2)
1381                 goto error;
1382
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;
1388
1389         isl_union_map_free(umap1);
1390         isl_union_map_free(umap2);
1391
1392         return data.is_subset;
1393 error:
1394         isl_union_map_free(umap1);
1395         isl_union_map_free(umap2);
1396         return -1;
1397 }
1398
1399 int isl_union_set_is_subset(__isl_keep isl_union_set *uset1,
1400         __isl_keep isl_union_set *uset2)
1401 {
1402         return isl_union_map_is_subset(uset1, uset2);
1403 }
1404
1405 int isl_union_map_is_equal(__isl_keep isl_union_map *umap1,
1406         __isl_keep isl_union_map *umap2)
1407 {
1408         int is_subset;
1409
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;
1417 }
1418
1419 int isl_union_set_is_equal(__isl_keep isl_union_set *uset1,
1420         __isl_keep isl_union_set *uset2)
1421 {
1422         return isl_union_map_is_equal(uset1, uset2);
1423 }
1424
1425 int isl_union_map_is_strict_subset(__isl_keep isl_union_map *umap1,
1426         __isl_keep isl_union_map *umap2)
1427 {
1428         int is_subset;
1429
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;
1439 }
1440
1441 int isl_union_set_is_strict_subset(__isl_keep isl_union_set *uset1,
1442         __isl_keep isl_union_set *uset2)
1443 {
1444         return isl_union_map_is_strict_subset(uset1, uset2);
1445 }
1446
1447 static int sample_entry(void **entry, void *user)
1448 {
1449         isl_basic_map **sample = (isl_basic_map **)user;
1450         isl_map *map = *entry;
1451
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;
1458 }
1459
1460 __isl_give isl_basic_map *isl_union_map_sample(__isl_take isl_union_map *umap)
1461 {
1462         isl_basic_map *sample = NULL;
1463
1464         if (!umap)
1465                 return NULL;
1466
1467         if (isl_hash_table_foreach(umap->dim->ctx, &umap->table,
1468                                    &sample_entry, &sample) < 0 &&
1469             !sample)
1470                 goto error;
1471
1472         if (!sample)
1473                 sample = isl_basic_map_empty(isl_union_map_get_dim(umap));
1474
1475         isl_union_map_free(umap);
1476
1477         return sample;
1478 error:
1479         isl_union_map_free(umap);
1480         return NULL;
1481 }
1482
1483 __isl_give isl_basic_set *isl_union_set_sample(__isl_take isl_union_set *uset)
1484 {
1485         return (isl_basic_set *)isl_union_map_sample(uset);
1486 }
1487
1488 struct isl_forall_data {
1489         int res;
1490         int (*fn)(__isl_keep isl_map *map);
1491 };
1492
1493 static int forall_entry(void **entry, void *user)
1494 {
1495         struct isl_forall_data *data = user;
1496         isl_map *map = *entry;
1497
1498         data->res = data->fn(map);
1499         if (data->res < 0)
1500                 return -1;
1501
1502         if (!data->res)
1503                 return -1;
1504
1505         return 0;
1506 }
1507
1508 static int union_map_forall(__isl_keep isl_union_map *umap,
1509         int (*fn)(__isl_keep isl_map *map))
1510 {
1511         struct isl_forall_data data = { 1, fn };
1512
1513         if (!umap)
1514                 return -1;
1515
1516         if (isl_hash_table_foreach(umap->dim->ctx, &umap->table,
1517                                    &forall_entry, &data) < 0 && data.res)
1518                 return -1;
1519
1520         return data.res;
1521 }
1522
1523 int isl_union_map_is_empty(__isl_keep isl_union_map *umap)
1524 {
1525         return union_map_forall(umap, &isl_map_is_empty);
1526 }
1527
1528 int isl_union_set_is_empty(__isl_keep isl_union_set *uset)
1529 {
1530         return isl_union_map_is_empty(uset);
1531 }
1532
1533 static int is_subset_of_identity(__isl_keep isl_map *map)
1534 {
1535         int is_subset;
1536         isl_dim *dim;
1537         isl_map *id;
1538
1539         if (!map)
1540                 return -1;
1541
1542         if (!isl_dim_tuple_match(map->dim, isl_dim_in, map->dim, isl_dim_out))
1543                 return 0;
1544
1545         dim = isl_map_get_dim(map);
1546         id = isl_map_identity(dim);
1547
1548         is_subset = isl_map_is_subset(map, id);
1549
1550         isl_map_free(id);
1551
1552         return is_subset;
1553 }
1554
1555 /* Check if the given map is single-valued.
1556  * We simply compute
1557  *
1558  *      M \circ M^-1
1559  *
1560  * and check if the result is a subset of the identity mapping.
1561  */
1562 int isl_union_map_is_single_valued(__isl_keep isl_union_map *umap)
1563 {
1564         isl_union_map *test;
1565         int sv;
1566
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;
1572         }
1573
1574         test = isl_union_map_reverse(isl_union_map_copy(umap));
1575         test = isl_union_map_apply_range(test, isl_union_map_copy(umap));
1576
1577         sv = union_map_forall(test, &is_subset_of_identity);
1578
1579         isl_union_map_free(test);
1580
1581         return sv;
1582 }
1583
1584 int isl_union_map_is_bijective(__isl_keep isl_union_map *umap)
1585 {
1586         int sv;
1587
1588         sv = isl_union_map_is_single_valued(umap);
1589         if (sv < 0 || !sv)
1590                 return sv;
1591
1592         umap = isl_union_map_copy(umap);
1593         umap = isl_union_map_reverse(umap);
1594         sv = isl_union_map_is_single_valued(umap);
1595         isl_union_map_free(umap);
1596
1597         return sv;
1598 }
1599
1600 static int zip_entry(void **entry, void *user)
1601 {
1602         isl_map *map = *entry;
1603         isl_union_map **res = user;
1604
1605         if (!isl_map_can_zip(map))
1606                 return 0;
1607
1608         *res = isl_union_map_add_map(*res, isl_map_zip(isl_map_copy(map)));
1609
1610         return 0;
1611 }
1612
1613 __isl_give isl_union_map *isl_union_map_zip(__isl_take isl_union_map *umap)
1614 {
1615         return cond_un_op(umap, &zip_entry);
1616 }
1617
1618 static int lift_entry(void **entry, void *user)
1619 {
1620         isl_set *set = *entry;
1621         isl_union_set **res = user;
1622
1623         *res = isl_union_set_add_set(*res, isl_set_lift(isl_set_copy(set)));
1624
1625         return 0;
1626 }
1627
1628 __isl_give isl_union_set *isl_union_set_lift(__isl_take isl_union_set *uset)
1629 {
1630         return cond_un_op(uset, &lift_entry);
1631 }
1632
1633 static int coefficients_entry(void **entry, void *user)
1634 {
1635         isl_set *set = *entry;
1636         isl_union_set **res = user;
1637
1638         set = isl_set_copy(set);
1639         set = isl_set_from_basic_set(isl_set_coefficients(set));
1640         *res = isl_union_set_add_set(*res, set);
1641
1642         return 0;
1643 }
1644
1645 __isl_give isl_union_set *isl_union_set_coefficients(
1646         __isl_take isl_union_set *uset)
1647 {
1648         isl_ctx *ctx;
1649         isl_dim *dim;
1650         isl_union_set *res;
1651
1652         if (!uset)
1653                 return NULL;
1654
1655         ctx = isl_union_set_get_ctx(uset);
1656         dim = isl_dim_set_alloc(ctx, 0, 0);
1657         res = isl_union_map_alloc(dim, uset->table.n);
1658         if (isl_hash_table_foreach(uset->dim->ctx, &uset->table,
1659                                    &coefficients_entry, &res) < 0)
1660                 goto error;
1661
1662         isl_union_set_free(uset);
1663         return res;
1664 error:
1665         isl_union_set_free(uset);
1666         isl_union_set_free(res);
1667         return NULL;
1668 }
1669
1670 static int solutions_entry(void **entry, void *user)
1671 {
1672         isl_set *set = *entry;
1673         isl_union_set **res = user;
1674
1675         set = isl_set_copy(set);
1676         set = isl_set_from_basic_set(isl_set_solutions(set));
1677         if (!*res)
1678                 *res = isl_union_set_from_set(set);
1679         else
1680                 *res = isl_union_set_add_set(*res, set);
1681
1682         if (!*res)
1683                 return -1;
1684
1685         return 0;
1686 }
1687
1688 __isl_give isl_union_set *isl_union_set_solutions(
1689         __isl_take isl_union_set *uset)
1690 {
1691         isl_ctx *ctx;
1692         isl_dim *dim;
1693         isl_union_set *res = NULL;
1694
1695         if (!uset)
1696                 return NULL;
1697
1698         if (uset->table.n == 0) {
1699                 res = isl_union_set_empty(isl_union_set_get_dim(uset));
1700                 isl_union_set_free(uset);
1701                 return res;
1702         }
1703
1704         if (isl_hash_table_foreach(uset->dim->ctx, &uset->table,
1705                                    &solutions_entry, &res) < 0)
1706                 goto error;
1707
1708         isl_union_set_free(uset);
1709         return res;
1710 error:
1711         isl_union_set_free(uset);
1712         isl_union_set_free(res);
1713         return NULL;
1714 }