2 * Copyright 2010-2011 INRIA Saclay
3 * Copyright 2013-2014 Ecole Normale Superieure
4 * Copyright 2014 INRIA Rocquencourt
6 * Use of this software is governed by the MIT license
8 * Written by Sven Verdoolaege, INRIA Saclay - Ile-de-France,
9 * Parc Club Orsay Universite, ZAC des vignes, 4 rue Jacques Monod,
11 * and Inria Paris - Rocquencourt, Domaine de Voluceau - Rocquencourt,
12 * B.P. 105 - 78153 Le Chesnay, France
16 #include <isl_map_private.h>
22 #include <isl_space_private.h>
23 #include <isl_union_map_private.h>
24 #include <isl/union_set.h>
25 #include <isl/deprecated/union_map_int.h>
27 /* Return the number of parameters of "umap", where "type"
28 * is required to be set to isl_dim_param.
30 unsigned isl_union_map_dim(__isl_keep isl_union_map
*umap
,
31 enum isl_dim_type type
)
36 if (type
!= isl_dim_param
)
37 isl_die(isl_union_map_get_ctx(umap
), isl_error_invalid
,
38 "can only reference parameters", return 0);
40 return isl_space_dim(umap
->dim
, type
);
43 /* Return the number of parameters of "uset", where "type"
44 * is required to be set to isl_dim_param.
46 unsigned isl_union_set_dim(__isl_keep isl_union_set
*uset
,
47 enum isl_dim_type type
)
49 return isl_union_map_dim(uset
, type
);
52 /* Return the id of the specified dimension.
54 __isl_give isl_id
*isl_union_map_get_dim_id(__isl_keep isl_union_map
*umap
,
55 enum isl_dim_type type
, unsigned pos
)
60 if (type
!= isl_dim_param
)
61 isl_die(isl_union_map_get_ctx(umap
), isl_error_invalid
,
62 "can only reference parameters", return NULL
);
64 return isl_space_get_dim_id(umap
->dim
, type
, pos
);
67 /* Is this union set a parameter domain?
69 int isl_union_set_is_params(__isl_keep isl_union_set
*uset
)
76 if (uset
->table
.n
!= 1)
79 set
= isl_set_from_union_set(isl_union_set_copy(uset
));
80 params
= isl_set_is_params(set
);
85 static __isl_give isl_union_map
*isl_union_map_alloc(
86 __isl_take isl_space
*space
, int size
)
90 space
= isl_space_params(space
);
94 umap
= isl_calloc_type(space
->ctx
, isl_union_map
);
96 isl_space_free(space
);
102 if (isl_hash_table_init(space
->ctx
, &umap
->table
, size
) < 0)
103 return isl_union_map_free(umap
);
108 __isl_give isl_union_map
*isl_union_map_empty(__isl_take isl_space
*dim
)
110 return isl_union_map_alloc(dim
, 16);
113 __isl_give isl_union_set
*isl_union_set_empty(__isl_take isl_space
*dim
)
115 return isl_union_map_empty(dim
);
118 isl_ctx
*isl_union_map_get_ctx(__isl_keep isl_union_map
*umap
)
120 return umap
? umap
->dim
->ctx
: NULL
;
123 isl_ctx
*isl_union_set_get_ctx(__isl_keep isl_union_set
*uset
)
125 return uset
? uset
->dim
->ctx
: NULL
;
128 __isl_give isl_space
*isl_union_map_get_space(__isl_keep isl_union_map
*umap
)
132 return isl_space_copy(umap
->dim
);
135 /* Return the position of the parameter with the given name
137 * Return -1 if no such dimension can be found.
139 int isl_union_map_find_dim_by_name(__isl_keep isl_union_map
*umap
,
140 enum isl_dim_type type
, const char *name
)
144 return isl_space_find_dim_by_name(umap
->dim
, type
, name
);
147 __isl_give isl_space
*isl_union_set_get_space(__isl_keep isl_union_set
*uset
)
149 return isl_union_map_get_space(uset
);
152 static int free_umap_entry(void **entry
, void *user
)
154 isl_map
*map
= *entry
;
159 static int add_map(__isl_take isl_map
*map
, void *user
)
161 isl_union_map
**umap
= (isl_union_map
**)user
;
163 *umap
= isl_union_map_add_map(*umap
, map
);
168 __isl_give isl_union_map
*isl_union_map_dup(__isl_keep isl_union_map
*umap
)
175 dup
= isl_union_map_empty(isl_space_copy(umap
->dim
));
176 if (isl_union_map_foreach_map(umap
, &add_map
, &dup
) < 0)
180 isl_union_map_free(dup
);
184 __isl_give isl_union_map
*isl_union_map_cow(__isl_take isl_union_map
*umap
)
192 return isl_union_map_dup(umap
);
195 struct isl_union_align
{
200 static int align_entry(void **entry
, void *user
)
202 isl_map
*map
= *entry
;
204 struct isl_union_align
*data
= user
;
206 exp
= isl_reordering_extend_space(isl_reordering_copy(data
->exp
),
207 isl_map_get_space(map
));
209 data
->res
= isl_union_map_add_map(data
->res
,
210 isl_map_realign(isl_map_copy(map
), exp
));
215 /* Align the parameters of umap along those of model.
216 * The result has the parameters of model first, in the same order
217 * as they appear in model, followed by any remaining parameters of
218 * umap that do not appear in model.
220 __isl_give isl_union_map
*isl_union_map_align_params(
221 __isl_take isl_union_map
*umap
, __isl_take isl_space
*model
)
223 struct isl_union_align data
= { NULL
, NULL
};
228 if (isl_space_match(umap
->dim
, isl_dim_param
, model
, isl_dim_param
)) {
229 isl_space_free(model
);
233 model
= isl_space_params(model
);
234 data
.exp
= isl_parameter_alignment_reordering(umap
->dim
, model
);
238 data
.res
= isl_union_map_alloc(isl_space_copy(data
.exp
->dim
),
240 if (isl_hash_table_foreach(umap
->dim
->ctx
, &umap
->table
,
241 &align_entry
, &data
) < 0)
244 isl_reordering_free(data
.exp
);
245 isl_union_map_free(umap
);
246 isl_space_free(model
);
249 isl_reordering_free(data
.exp
);
250 isl_union_map_free(umap
);
251 isl_union_map_free(data
.res
);
252 isl_space_free(model
);
256 __isl_give isl_union_set
*isl_union_set_align_params(
257 __isl_take isl_union_set
*uset
, __isl_take isl_space
*model
)
259 return isl_union_map_align_params(uset
, model
);
262 __isl_give isl_union_map
*isl_union_map_union(__isl_take isl_union_map
*umap1
,
263 __isl_take isl_union_map
*umap2
)
265 umap1
= isl_union_map_align_params(umap1
, isl_union_map_get_space(umap2
));
266 umap2
= isl_union_map_align_params(umap2
, isl_union_map_get_space(umap1
));
268 umap1
= isl_union_map_cow(umap1
);
270 if (!umap1
|| !umap2
)
273 if (isl_union_map_foreach_map(umap2
, &add_map
, &umap1
) < 0)
276 isl_union_map_free(umap2
);
280 isl_union_map_free(umap1
);
281 isl_union_map_free(umap2
);
285 __isl_give isl_union_set
*isl_union_set_union(__isl_take isl_union_set
*uset1
,
286 __isl_take isl_union_set
*uset2
)
288 return isl_union_map_union(uset1
, uset2
);
291 __isl_give isl_union_map
*isl_union_map_copy(__isl_keep isl_union_map
*umap
)
300 __isl_give isl_union_set
*isl_union_set_copy(__isl_keep isl_union_set
*uset
)
302 return isl_union_map_copy(uset
);
305 __isl_null isl_union_map
*isl_union_map_free(__isl_take isl_union_map
*umap
)
313 isl_hash_table_foreach(umap
->dim
->ctx
, &umap
->table
,
314 &free_umap_entry
, NULL
);
315 isl_hash_table_clear(&umap
->table
);
316 isl_space_free(umap
->dim
);
321 __isl_null isl_union_set
*isl_union_set_free(__isl_take isl_union_set
*uset
)
323 return isl_union_map_free(uset
);
326 static int has_dim(const void *entry
, const void *val
)
328 isl_map
*map
= (isl_map
*)entry
;
329 isl_space
*dim
= (isl_space
*)val
;
331 return isl_space_is_equal(map
->dim
, dim
);
334 __isl_give isl_union_map
*isl_union_map_add_map(__isl_take isl_union_map
*umap
,
335 __isl_take isl_map
*map
)
338 struct isl_hash_table_entry
*entry
;
343 if (isl_map_plain_is_empty(map
)) {
348 if (!isl_space_match(map
->dim
, isl_dim_param
, umap
->dim
, isl_dim_param
)) {
349 umap
= isl_union_map_align_params(umap
, isl_map_get_space(map
));
350 map
= isl_map_align_params(map
, isl_union_map_get_space(umap
));
353 umap
= isl_union_map_cow(umap
);
358 hash
= isl_space_get_hash(map
->dim
);
359 entry
= isl_hash_table_find(umap
->dim
->ctx
, &umap
->table
, hash
,
360 &has_dim
, map
->dim
, 1);
367 entry
->data
= isl_map_union(entry
->data
, isl_map_copy(map
));
376 isl_union_map_free(umap
);
380 __isl_give isl_union_set
*isl_union_set_add_set(__isl_take isl_union_set
*uset
,
381 __isl_take isl_set
*set
)
383 return isl_union_map_add_map(uset
, (isl_map
*)set
);
386 __isl_give isl_union_map
*isl_union_map_from_map(__isl_take isl_map
*map
)
394 dim
= isl_map_get_space(map
);
395 dim
= isl_space_params(dim
);
396 umap
= isl_union_map_empty(dim
);
397 umap
= isl_union_map_add_map(umap
, map
);
402 __isl_give isl_union_set
*isl_union_set_from_set(__isl_take isl_set
*set
)
404 return isl_union_map_from_map((isl_map
*)set
);
407 __isl_give isl_union_map
*isl_union_map_from_basic_map(
408 __isl_take isl_basic_map
*bmap
)
410 return isl_union_map_from_map(isl_map_from_basic_map(bmap
));
413 __isl_give isl_union_set
*isl_union_set_from_basic_set(
414 __isl_take isl_basic_set
*bset
)
416 return isl_union_map_from_basic_map(bset
);
419 struct isl_union_map_foreach_data
421 int (*fn
)(__isl_take isl_map
*map
, void *user
);
425 static int call_on_copy(void **entry
, void *user
)
427 isl_map
*map
= *entry
;
428 struct isl_union_map_foreach_data
*data
;
429 data
= (struct isl_union_map_foreach_data
*)user
;
431 return data
->fn(isl_map_copy(map
), data
->user
);
434 int isl_union_map_n_map(__isl_keep isl_union_map
*umap
)
436 return umap
? umap
->table
.n
: 0;
439 int isl_union_set_n_set(__isl_keep isl_union_set
*uset
)
441 return uset
? uset
->table
.n
: 0;
444 int isl_union_map_foreach_map(__isl_keep isl_union_map
*umap
,
445 int (*fn
)(__isl_take isl_map
*map
, void *user
), void *user
)
447 struct isl_union_map_foreach_data data
= { fn
, user
};
452 return isl_hash_table_foreach(umap
->dim
->ctx
, &umap
->table
,
453 &call_on_copy
, &data
);
456 static int copy_map(void **entry
, void *user
)
458 isl_map
*map
= *entry
;
459 isl_map
**map_p
= user
;
461 *map_p
= isl_map_copy(map
);
466 __isl_give isl_map
*isl_map_from_union_map(__isl_take isl_union_map
*umap
)
473 ctx
= isl_union_map_get_ctx(umap
);
474 if (umap
->table
.n
!= 1)
475 isl_die(ctx
, isl_error_invalid
,
476 "union map needs to contain elements in exactly "
477 "one space", goto error
);
479 isl_hash_table_foreach(ctx
, &umap
->table
, ©_map
, &map
);
481 isl_union_map_free(umap
);
485 isl_union_map_free(umap
);
489 __isl_give isl_set
*isl_set_from_union_set(__isl_take isl_union_set
*uset
)
491 return isl_map_from_union_map(uset
);
494 /* Extract the map in "umap" that lives in the given space (ignoring
497 __isl_give isl_map
*isl_union_map_extract_map(__isl_keep isl_union_map
*umap
,
498 __isl_take isl_space
*space
)
501 struct isl_hash_table_entry
*entry
;
503 space
= isl_space_drop_dims(space
, isl_dim_param
,
504 0, isl_space_dim(space
, isl_dim_param
));
505 space
= isl_space_align_params(space
, isl_union_map_get_space(umap
));
509 hash
= isl_space_get_hash(space
);
510 entry
= isl_hash_table_find(umap
->dim
->ctx
, &umap
->table
, hash
,
513 return isl_map_empty(space
);
514 isl_space_free(space
);
515 return isl_map_copy(entry
->data
);
517 isl_space_free(space
);
521 __isl_give isl_set
*isl_union_set_extract_set(__isl_keep isl_union_set
*uset
,
522 __isl_take isl_space
*dim
)
524 return (isl_set
*)isl_union_map_extract_map(uset
, dim
);
527 /* Check if umap contains a map in the given space.
529 __isl_give
int isl_union_map_contains(__isl_keep isl_union_map
*umap
,
530 __isl_keep isl_space
*dim
)
533 struct isl_hash_table_entry
*entry
;
538 hash
= isl_space_get_hash(dim
);
539 entry
= isl_hash_table_find(umap
->dim
->ctx
, &umap
->table
, hash
,
544 __isl_give
int isl_union_set_contains(__isl_keep isl_union_set
*uset
,
545 __isl_keep isl_space
*dim
)
547 return isl_union_map_contains(uset
, dim
);
550 int isl_union_set_foreach_set(__isl_keep isl_union_set
*uset
,
551 int (*fn
)(__isl_take isl_set
*set
, void *user
), void *user
)
553 return isl_union_map_foreach_map(uset
,
554 (int(*)(__isl_take isl_map
*, void*))fn
, user
);
557 struct isl_union_set_foreach_point_data
{
558 int (*fn
)(__isl_take isl_point
*pnt
, void *user
);
562 static int foreach_point(__isl_take isl_set
*set
, void *user
)
564 struct isl_union_set_foreach_point_data
*data
= user
;
567 r
= isl_set_foreach_point(set
, data
->fn
, data
->user
);
573 int isl_union_set_foreach_point(__isl_keep isl_union_set
*uset
,
574 int (*fn
)(__isl_take isl_point
*pnt
, void *user
), void *user
)
576 struct isl_union_set_foreach_point_data data
= { fn
, user
};
577 return isl_union_set_foreach_set(uset
, &foreach_point
, &data
);
580 struct isl_union_map_gen_bin_data
{
581 isl_union_map
*umap2
;
585 static int subtract_entry(void **entry
, void *user
)
587 struct isl_union_map_gen_bin_data
*data
= user
;
589 struct isl_hash_table_entry
*entry2
;
590 isl_map
*map
= *entry
;
592 hash
= isl_space_get_hash(map
->dim
);
593 entry2
= isl_hash_table_find(data
->umap2
->dim
->ctx
, &data
->umap2
->table
,
594 hash
, &has_dim
, map
->dim
, 0);
595 map
= isl_map_copy(map
);
598 map
= isl_map_subtract(map
, isl_map_copy(entry2
->data
));
600 empty
= isl_map_is_empty(map
);
610 data
->res
= isl_union_map_add_map(data
->res
, map
);
615 static __isl_give isl_union_map
*gen_bin_op(__isl_take isl_union_map
*umap1
,
616 __isl_take isl_union_map
*umap2
, int (*fn
)(void **, void *))
618 struct isl_union_map_gen_bin_data data
= { NULL
, NULL
};
620 umap1
= isl_union_map_align_params(umap1
, isl_union_map_get_space(umap2
));
621 umap2
= isl_union_map_align_params(umap2
, isl_union_map_get_space(umap1
));
623 if (!umap1
|| !umap2
)
627 data
.res
= isl_union_map_alloc(isl_space_copy(umap1
->dim
),
629 if (isl_hash_table_foreach(umap1
->dim
->ctx
, &umap1
->table
,
633 isl_union_map_free(umap1
);
634 isl_union_map_free(umap2
);
637 isl_union_map_free(umap1
);
638 isl_union_map_free(umap2
);
639 isl_union_map_free(data
.res
);
643 __isl_give isl_union_map
*isl_union_map_subtract(
644 __isl_take isl_union_map
*umap1
, __isl_take isl_union_map
*umap2
)
646 return gen_bin_op(umap1
, umap2
, &subtract_entry
);
649 __isl_give isl_union_set
*isl_union_set_subtract(
650 __isl_take isl_union_set
*uset1
, __isl_take isl_union_set
*uset2
)
652 return isl_union_map_subtract(uset1
, uset2
);
655 struct isl_union_map_gen_bin_set_data
{
660 static int intersect_params_entry(void **entry
, void *user
)
662 struct isl_union_map_gen_bin_set_data
*data
= user
;
663 isl_map
*map
= *entry
;
666 map
= isl_map_copy(map
);
667 map
= isl_map_intersect_params(map
, isl_set_copy(data
->set
));
669 empty
= isl_map_is_empty(map
);
675 data
->res
= isl_union_map_add_map(data
->res
, map
);
680 static __isl_give isl_union_map
*gen_bin_set_op(__isl_take isl_union_map
*umap
,
681 __isl_take isl_set
*set
, int (*fn
)(void **, void *))
683 struct isl_union_map_gen_bin_set_data data
= { NULL
, NULL
};
685 umap
= isl_union_map_align_params(umap
, isl_set_get_space(set
));
686 set
= isl_set_align_params(set
, isl_union_map_get_space(umap
));
692 data
.res
= isl_union_map_alloc(isl_space_copy(umap
->dim
),
694 if (isl_hash_table_foreach(umap
->dim
->ctx
, &umap
->table
,
698 isl_union_map_free(umap
);
702 isl_union_map_free(umap
);
704 isl_union_map_free(data
.res
);
708 __isl_give isl_union_map
*isl_union_map_intersect_params(
709 __isl_take isl_union_map
*umap
, __isl_take isl_set
*set
)
711 return gen_bin_set_op(umap
, set
, &intersect_params_entry
);
714 __isl_give isl_union_set
*isl_union_set_intersect_params(
715 __isl_take isl_union_set
*uset
, __isl_take isl_set
*set
)
717 return isl_union_map_intersect_params(uset
, set
);
720 static __isl_give isl_union_map
*union_map_intersect_params(
721 __isl_take isl_union_map
*umap
, __isl_take isl_union_set
*uset
)
723 return isl_union_map_intersect_params(umap
,
724 isl_set_from_union_set(uset
));
727 static __isl_give isl_union_map
*union_map_gist_params(
728 __isl_take isl_union_map
*umap
, __isl_take isl_union_set
*uset
)
730 return isl_union_map_gist_params(umap
, isl_set_from_union_set(uset
));
733 struct isl_union_map_match_bin_data
{
734 isl_union_map
*umap2
;
736 __isl_give isl_map
*(*fn
)(__isl_take isl_map
*, __isl_take isl_map
*);
739 static int match_bin_entry(void **entry
, void *user
)
741 struct isl_union_map_match_bin_data
*data
= user
;
743 struct isl_hash_table_entry
*entry2
;
744 isl_map
*map
= *entry
;
747 hash
= isl_space_get_hash(map
->dim
);
748 entry2
= isl_hash_table_find(data
->umap2
->dim
->ctx
, &data
->umap2
->table
,
749 hash
, &has_dim
, map
->dim
, 0);
753 map
= isl_map_copy(map
);
754 map
= data
->fn(map
, isl_map_copy(entry2
->data
));
756 empty
= isl_map_is_empty(map
);
766 data
->res
= isl_union_map_add_map(data
->res
, map
);
771 static __isl_give isl_union_map
*match_bin_op(__isl_take isl_union_map
*umap1
,
772 __isl_take isl_union_map
*umap2
,
773 __isl_give isl_map
*(*fn
)(__isl_take isl_map
*, __isl_take isl_map
*))
775 struct isl_union_map_match_bin_data data
= { NULL
, NULL
, fn
};
777 umap1
= isl_union_map_align_params(umap1
, isl_union_map_get_space(umap2
));
778 umap2
= isl_union_map_align_params(umap2
, isl_union_map_get_space(umap1
));
780 if (!umap1
|| !umap2
)
784 data
.res
= isl_union_map_alloc(isl_space_copy(umap1
->dim
),
786 if (isl_hash_table_foreach(umap1
->dim
->ctx
, &umap1
->table
,
787 &match_bin_entry
, &data
) < 0)
790 isl_union_map_free(umap1
);
791 isl_union_map_free(umap2
);
794 isl_union_map_free(umap1
);
795 isl_union_map_free(umap2
);
796 isl_union_map_free(data
.res
);
800 __isl_give isl_union_map
*isl_union_map_intersect(
801 __isl_take isl_union_map
*umap1
, __isl_take isl_union_map
*umap2
)
803 return match_bin_op(umap1
, umap2
, &isl_map_intersect
);
806 /* Compute the intersection of the two union_sets.
807 * As a special case, if exactly one of the two union_sets
808 * is a parameter domain, then intersect the parameter domain
809 * of the other one with this set.
811 __isl_give isl_union_set
*isl_union_set_intersect(
812 __isl_take isl_union_set
*uset1
, __isl_take isl_union_set
*uset2
)
816 p1
= isl_union_set_is_params(uset1
);
817 p2
= isl_union_set_is_params(uset2
);
818 if (p1
< 0 || p2
< 0)
821 return union_map_intersect_params(uset1
, uset2
);
823 return union_map_intersect_params(uset2
, uset1
);
824 return isl_union_map_intersect(uset1
, uset2
);
826 isl_union_set_free(uset1
);
827 isl_union_set_free(uset2
);
831 static int gist_params_entry(void **entry
, void *user
)
833 struct isl_union_map_gen_bin_set_data
*data
= user
;
834 isl_map
*map
= *entry
;
837 map
= isl_map_copy(map
);
838 map
= isl_map_gist_params(map
, isl_set_copy(data
->set
));
840 empty
= isl_map_is_empty(map
);
846 data
->res
= isl_union_map_add_map(data
->res
, map
);
851 __isl_give isl_union_map
*isl_union_map_gist_params(
852 __isl_take isl_union_map
*umap
, __isl_take isl_set
*set
)
854 return gen_bin_set_op(umap
, set
, &gist_params_entry
);
857 __isl_give isl_union_set
*isl_union_set_gist_params(
858 __isl_take isl_union_set
*uset
, __isl_take isl_set
*set
)
860 return isl_union_map_gist_params(uset
, set
);
863 __isl_give isl_union_map
*isl_union_map_gist(__isl_take isl_union_map
*umap
,
864 __isl_take isl_union_map
*context
)
866 return match_bin_op(umap
, context
, &isl_map_gist
);
869 __isl_give isl_union_set
*isl_union_set_gist(__isl_take isl_union_set
*uset
,
870 __isl_take isl_union_set
*context
)
872 if (isl_union_set_is_params(context
))
873 return union_map_gist_params(uset
, context
);
874 return isl_union_map_gist(uset
, context
);
877 static __isl_give isl_map
*lex_le_set(__isl_take isl_map
*set1
,
878 __isl_take isl_map
*set2
)
880 return isl_set_lex_le_set((isl_set
*)set1
, (isl_set
*)set2
);
883 static __isl_give isl_map
*lex_lt_set(__isl_take isl_map
*set1
,
884 __isl_take isl_map
*set2
)
886 return isl_set_lex_lt_set((isl_set
*)set1
, (isl_set
*)set2
);
889 __isl_give isl_union_map
*isl_union_set_lex_lt_union_set(
890 __isl_take isl_union_set
*uset1
, __isl_take isl_union_set
*uset2
)
892 return match_bin_op(uset1
, uset2
, &lex_lt_set
);
895 __isl_give isl_union_map
*isl_union_set_lex_le_union_set(
896 __isl_take isl_union_set
*uset1
, __isl_take isl_union_set
*uset2
)
898 return match_bin_op(uset1
, uset2
, &lex_le_set
);
901 __isl_give isl_union_map
*isl_union_set_lex_gt_union_set(
902 __isl_take isl_union_set
*uset1
, __isl_take isl_union_set
*uset2
)
904 return isl_union_map_reverse(isl_union_set_lex_lt_union_set(uset2
, uset1
));
907 __isl_give isl_union_map
*isl_union_set_lex_ge_union_set(
908 __isl_take isl_union_set
*uset1
, __isl_take isl_union_set
*uset2
)
910 return isl_union_map_reverse(isl_union_set_lex_le_union_set(uset2
, uset1
));
913 __isl_give isl_union_map
*isl_union_map_lex_gt_union_map(
914 __isl_take isl_union_map
*umap1
, __isl_take isl_union_map
*umap2
)
916 return isl_union_map_reverse(isl_union_map_lex_lt_union_map(umap2
, umap1
));
919 __isl_give isl_union_map
*isl_union_map_lex_ge_union_map(
920 __isl_take isl_union_map
*umap1
, __isl_take isl_union_map
*umap2
)
922 return isl_union_map_reverse(isl_union_map_lex_le_union_map(umap2
, umap1
));
925 static int intersect_domain_entry(void **entry
, void *user
)
927 struct isl_union_map_gen_bin_data
*data
= user
;
929 struct isl_hash_table_entry
*entry2
;
931 isl_map
*map
= *entry
;
934 dim
= isl_map_get_space(map
);
935 dim
= isl_space_domain(dim
);
936 hash
= isl_space_get_hash(dim
);
937 entry2
= isl_hash_table_find(data
->umap2
->dim
->ctx
, &data
->umap2
->table
,
938 hash
, &has_dim
, dim
, 0);
943 map
= isl_map_copy(map
);
944 map
= isl_map_intersect_domain(map
, isl_set_copy(entry2
->data
));
946 empty
= isl_map_is_empty(map
);
956 data
->res
= isl_union_map_add_map(data
->res
, map
);
961 /* Intersect the domain of "umap" with "uset".
962 * If "uset" is a parameters domain, then intersect the parameter
963 * domain of "umap" with this set.
965 __isl_give isl_union_map
*isl_union_map_intersect_domain(
966 __isl_take isl_union_map
*umap
, __isl_take isl_union_set
*uset
)
968 if (isl_union_set_is_params(uset
))
969 return union_map_intersect_params(umap
, uset
);
970 return gen_bin_op(umap
, uset
, &intersect_domain_entry
);
973 /* Remove the elements of data->umap2 from the domain of *entry
974 * and add the result to data->res.
976 static int subtract_domain_entry(void **entry
, void *user
)
978 struct isl_union_map_gen_bin_data
*data
= user
;
980 struct isl_hash_table_entry
*entry2
;
982 isl_map
*map
= *entry
;
985 dim
= isl_map_get_space(map
);
986 dim
= isl_space_domain(dim
);
987 hash
= isl_space_get_hash(dim
);
988 entry2
= isl_hash_table_find(data
->umap2
->dim
->ctx
, &data
->umap2
->table
,
989 hash
, &has_dim
, dim
, 0);
992 map
= isl_map_copy(map
);
995 data
->res
= isl_union_map_add_map(data
->res
, map
);
999 map
= isl_map_subtract_domain(map
, isl_set_copy(entry2
->data
));
1001 empty
= isl_map_is_empty(map
);
1011 data
->res
= isl_union_map_add_map(data
->res
, map
);
1016 /* Remove the elements of "uset" from the domain of "umap".
1018 __isl_give isl_union_map
*isl_union_map_subtract_domain(
1019 __isl_take isl_union_map
*umap
, __isl_take isl_union_set
*dom
)
1021 return gen_bin_op(umap
, dom
, &subtract_domain_entry
);
1024 /* Remove the elements of data->umap2 from the range of *entry
1025 * and add the result to data->res.
1027 static int subtract_range_entry(void **entry
, void *user
)
1029 struct isl_union_map_gen_bin_data
*data
= user
;
1031 struct isl_hash_table_entry
*entry2
;
1033 isl_map
*map
= *entry
;
1036 space
= isl_map_get_space(map
);
1037 space
= isl_space_range(space
);
1038 hash
= isl_space_get_hash(space
);
1039 entry2
= isl_hash_table_find(data
->umap2
->dim
->ctx
, &data
->umap2
->table
,
1040 hash
, &has_dim
, space
, 0);
1041 isl_space_free(space
);
1043 map
= isl_map_copy(map
);
1046 data
->res
= isl_union_map_add_map(data
->res
, map
);
1050 map
= isl_map_subtract_range(map
, isl_set_copy(entry2
->data
));
1052 empty
= isl_map_is_empty(map
);
1062 data
->res
= isl_union_map_add_map(data
->res
, map
);
1067 /* Remove the elements of "uset" from the range of "umap".
1069 __isl_give isl_union_map
*isl_union_map_subtract_range(
1070 __isl_take isl_union_map
*umap
, __isl_take isl_union_set
*dom
)
1072 return gen_bin_op(umap
, dom
, &subtract_range_entry
);
1075 static int gist_domain_entry(void **entry
, void *user
)
1077 struct isl_union_map_gen_bin_data
*data
= user
;
1079 struct isl_hash_table_entry
*entry2
;
1081 isl_map
*map
= *entry
;
1084 dim
= isl_map_get_space(map
);
1085 dim
= isl_space_domain(dim
);
1086 hash
= isl_space_get_hash(dim
);
1087 entry2
= isl_hash_table_find(data
->umap2
->dim
->ctx
, &data
->umap2
->table
,
1088 hash
, &has_dim
, dim
, 0);
1089 isl_space_free(dim
);
1093 map
= isl_map_copy(map
);
1094 map
= isl_map_gist_domain(map
, isl_set_copy(entry2
->data
));
1096 empty
= isl_map_is_empty(map
);
1102 data
->res
= isl_union_map_add_map(data
->res
, map
);
1107 /* Compute the gist of "umap" with respect to the domain "uset".
1108 * If "uset" is a parameters domain, then compute the gist
1109 * with respect to this parameter domain.
1111 __isl_give isl_union_map
*isl_union_map_gist_domain(
1112 __isl_take isl_union_map
*umap
, __isl_take isl_union_set
*uset
)
1114 if (isl_union_set_is_params(uset
))
1115 return union_map_gist_params(umap
, uset
);
1116 return gen_bin_op(umap
, uset
, &gist_domain_entry
);
1119 static int gist_range_entry(void **entry
, void *user
)
1121 struct isl_union_map_gen_bin_data
*data
= user
;
1123 struct isl_hash_table_entry
*entry2
;
1125 isl_map
*map
= *entry
;
1128 space
= isl_map_get_space(map
);
1129 space
= isl_space_range(space
);
1130 hash
= isl_space_get_hash(space
);
1131 entry2
= isl_hash_table_find(data
->umap2
->dim
->ctx
, &data
->umap2
->table
,
1132 hash
, &has_dim
, space
, 0);
1133 isl_space_free(space
);
1137 map
= isl_map_copy(map
);
1138 map
= isl_map_gist_range(map
, isl_set_copy(entry2
->data
));
1140 empty
= isl_map_is_empty(map
);
1146 data
->res
= isl_union_map_add_map(data
->res
, map
);
1151 /* Compute the gist of "umap" with respect to the range "uset".
1153 __isl_give isl_union_map
*isl_union_map_gist_range(
1154 __isl_take isl_union_map
*umap
, __isl_take isl_union_set
*uset
)
1156 return gen_bin_op(umap
, uset
, &gist_range_entry
);
1159 static int intersect_range_entry(void **entry
, void *user
)
1161 struct isl_union_map_gen_bin_data
*data
= user
;
1163 struct isl_hash_table_entry
*entry2
;
1165 isl_map
*map
= *entry
;
1168 dim
= isl_map_get_space(map
);
1169 dim
= isl_space_range(dim
);
1170 hash
= isl_space_get_hash(dim
);
1171 entry2
= isl_hash_table_find(data
->umap2
->dim
->ctx
, &data
->umap2
->table
,
1172 hash
, &has_dim
, dim
, 0);
1173 isl_space_free(dim
);
1177 map
= isl_map_copy(map
);
1178 map
= isl_map_intersect_range(map
, isl_set_copy(entry2
->data
));
1180 empty
= isl_map_is_empty(map
);
1190 data
->res
= isl_union_map_add_map(data
->res
, map
);
1195 __isl_give isl_union_map
*isl_union_map_intersect_range(
1196 __isl_take isl_union_map
*umap
, __isl_take isl_union_set
*uset
)
1198 return gen_bin_op(umap
, uset
, &intersect_range_entry
);
1201 struct isl_union_map_bin_data
{
1202 isl_union_map
*umap2
;
1205 int (*fn
)(void **entry
, void *user
);
1208 static int apply_range_entry(void **entry
, void *user
)
1210 struct isl_union_map_bin_data
*data
= user
;
1211 isl_map
*map2
= *entry
;
1214 if (!isl_space_tuple_is_equal(data
->map
->dim
, isl_dim_out
,
1215 map2
->dim
, isl_dim_in
))
1218 map2
= isl_map_apply_range(isl_map_copy(data
->map
), isl_map_copy(map2
));
1220 empty
= isl_map_is_empty(map2
);
1230 data
->res
= isl_union_map_add_map(data
->res
, map2
);
1235 static int bin_entry(void **entry
, void *user
)
1237 struct isl_union_map_bin_data
*data
= user
;
1238 isl_map
*map
= *entry
;
1241 if (isl_hash_table_foreach(data
->umap2
->dim
->ctx
, &data
->umap2
->table
,
1242 data
->fn
, data
) < 0)
1248 static __isl_give isl_union_map
*bin_op(__isl_take isl_union_map
*umap1
,
1249 __isl_take isl_union_map
*umap2
, int (*fn
)(void **entry
, void *user
))
1251 struct isl_union_map_bin_data data
= { NULL
, NULL
, NULL
, fn
};
1253 umap1
= isl_union_map_align_params(umap1
, isl_union_map_get_space(umap2
));
1254 umap2
= isl_union_map_align_params(umap2
, isl_union_map_get_space(umap1
));
1256 if (!umap1
|| !umap2
)
1260 data
.res
= isl_union_map_alloc(isl_space_copy(umap1
->dim
),
1262 if (isl_hash_table_foreach(umap1
->dim
->ctx
, &umap1
->table
,
1263 &bin_entry
, &data
) < 0)
1266 isl_union_map_free(umap1
);
1267 isl_union_map_free(umap2
);
1270 isl_union_map_free(umap1
);
1271 isl_union_map_free(umap2
);
1272 isl_union_map_free(data
.res
);
1276 __isl_give isl_union_map
*isl_union_map_apply_range(
1277 __isl_take isl_union_map
*umap1
, __isl_take isl_union_map
*umap2
)
1279 return bin_op(umap1
, umap2
, &apply_range_entry
);
1282 __isl_give isl_union_map
*isl_union_map_apply_domain(
1283 __isl_take isl_union_map
*umap1
, __isl_take isl_union_map
*umap2
)
1285 umap1
= isl_union_map_reverse(umap1
);
1286 umap1
= isl_union_map_apply_range(umap1
, umap2
);
1287 return isl_union_map_reverse(umap1
);
1290 __isl_give isl_union_set
*isl_union_set_apply(
1291 __isl_take isl_union_set
*uset
, __isl_take isl_union_map
*umap
)
1293 return isl_union_map_apply_range(uset
, umap
);
1296 static int map_lex_lt_entry(void **entry
, void *user
)
1298 struct isl_union_map_bin_data
*data
= user
;
1299 isl_map
*map2
= *entry
;
1301 if (!isl_space_tuple_is_equal(data
->map
->dim
, isl_dim_out
,
1302 map2
->dim
, isl_dim_out
))
1305 map2
= isl_map_lex_lt_map(isl_map_copy(data
->map
), isl_map_copy(map2
));
1307 data
->res
= isl_union_map_add_map(data
->res
, map2
);
1312 __isl_give isl_union_map
*isl_union_map_lex_lt_union_map(
1313 __isl_take isl_union_map
*umap1
, __isl_take isl_union_map
*umap2
)
1315 return bin_op(umap1
, umap2
, &map_lex_lt_entry
);
1318 static int map_lex_le_entry(void **entry
, void *user
)
1320 struct isl_union_map_bin_data
*data
= user
;
1321 isl_map
*map2
= *entry
;
1323 if (!isl_space_tuple_is_equal(data
->map
->dim
, isl_dim_out
,
1324 map2
->dim
, isl_dim_out
))
1327 map2
= isl_map_lex_le_map(isl_map_copy(data
->map
), isl_map_copy(map2
));
1329 data
->res
= isl_union_map_add_map(data
->res
, map2
);
1334 __isl_give isl_union_map
*isl_union_map_lex_le_union_map(
1335 __isl_take isl_union_map
*umap1
, __isl_take isl_union_map
*umap2
)
1337 return bin_op(umap1
, umap2
, &map_lex_le_entry
);
1340 static int product_entry(void **entry
, void *user
)
1342 struct isl_union_map_bin_data
*data
= user
;
1343 isl_map
*map2
= *entry
;
1345 map2
= isl_map_product(isl_map_copy(data
->map
), isl_map_copy(map2
));
1347 data
->res
= isl_union_map_add_map(data
->res
, map2
);
1352 __isl_give isl_union_map
*isl_union_map_product(__isl_take isl_union_map
*umap1
,
1353 __isl_take isl_union_map
*umap2
)
1355 return bin_op(umap1
, umap2
, &product_entry
);
1358 static int set_product_entry(void **entry
, void *user
)
1360 struct isl_union_map_bin_data
*data
= user
;
1361 isl_set
*set2
= *entry
;
1363 set2
= isl_set_product(isl_set_copy(data
->map
), isl_set_copy(set2
));
1365 data
->res
= isl_union_set_add_set(data
->res
, set2
);
1370 __isl_give isl_union_set
*isl_union_set_product(__isl_take isl_union_set
*uset1
,
1371 __isl_take isl_union_set
*uset2
)
1373 return bin_op(uset1
, uset2
, &set_product_entry
);
1376 static int domain_product_entry(void **entry
, void *user
)
1378 struct isl_union_map_bin_data
*data
= user
;
1379 isl_map
*map2
= *entry
;
1381 if (!isl_space_tuple_is_equal(data
->map
->dim
, isl_dim_out
,
1382 map2
->dim
, isl_dim_out
))
1385 map2
= isl_map_domain_product(isl_map_copy(data
->map
),
1386 isl_map_copy(map2
));
1388 data
->res
= isl_union_map_add_map(data
->res
, map2
);
1393 /* Given two maps A -> B and C -> D, construct a map [A -> C] -> (B * D)
1395 __isl_give isl_union_map
*isl_union_map_domain_product(
1396 __isl_take isl_union_map
*umap1
, __isl_take isl_union_map
*umap2
)
1398 return bin_op(umap1
, umap2
, &domain_product_entry
);
1401 static int range_product_entry(void **entry
, void *user
)
1403 struct isl_union_map_bin_data
*data
= user
;
1404 isl_map
*map2
= *entry
;
1406 if (!isl_space_tuple_is_equal(data
->map
->dim
, isl_dim_in
,
1407 map2
->dim
, isl_dim_in
))
1410 map2
= isl_map_range_product(isl_map_copy(data
->map
),
1411 isl_map_copy(map2
));
1413 data
->res
= isl_union_map_add_map(data
->res
, map2
);
1418 __isl_give isl_union_map
*isl_union_map_range_product(
1419 __isl_take isl_union_map
*umap1
, __isl_take isl_union_map
*umap2
)
1421 return bin_op(umap1
, umap2
, &range_product_entry
);
1424 static int flat_range_product_entry(void **entry
, void *user
)
1426 struct isl_union_map_bin_data
*data
= user
;
1427 isl_map
*map2
= *entry
;
1429 if (!isl_space_tuple_is_equal(data
->map
->dim
, isl_dim_in
,
1430 map2
->dim
, isl_dim_in
))
1433 map2
= isl_map_flat_range_product(isl_map_copy(data
->map
),
1434 isl_map_copy(map2
));
1436 data
->res
= isl_union_map_add_map(data
->res
, map2
);
1441 __isl_give isl_union_map
*isl_union_map_flat_range_product(
1442 __isl_take isl_union_map
*umap1
, __isl_take isl_union_map
*umap2
)
1444 return bin_op(umap1
, umap2
, &flat_range_product_entry
);
1447 static __isl_give isl_union_set
*cond_un_op(__isl_take isl_union_map
*umap
,
1448 int (*fn
)(void **, void *))
1455 res
= isl_union_map_alloc(isl_space_copy(umap
->dim
), umap
->table
.n
);
1456 if (isl_hash_table_foreach(umap
->dim
->ctx
, &umap
->table
, fn
, &res
) < 0)
1459 isl_union_map_free(umap
);
1462 isl_union_map_free(umap
);
1463 isl_union_set_free(res
);
1467 static int from_range_entry(void **entry
, void *user
)
1469 isl_map
*set
= *entry
;
1470 isl_union_set
**res
= user
;
1472 *res
= isl_union_map_add_map(*res
,
1473 isl_map_from_range(isl_set_copy(set
)));
1478 __isl_give isl_union_map
*isl_union_map_from_range(
1479 __isl_take isl_union_set
*uset
)
1481 return cond_un_op(uset
, &from_range_entry
);
1484 __isl_give isl_union_map
*isl_union_map_from_domain(
1485 __isl_take isl_union_set
*uset
)
1487 return isl_union_map_reverse(isl_union_map_from_range(uset
));
1490 __isl_give isl_union_map
*isl_union_map_from_domain_and_range(
1491 __isl_take isl_union_set
*domain
, __isl_take isl_union_set
*range
)
1493 return isl_union_map_apply_range(isl_union_map_from_domain(domain
),
1494 isl_union_map_from_range(range
));
1497 static __isl_give isl_union_map
*un_op(__isl_take isl_union_map
*umap
,
1498 int (*fn
)(void **, void *))
1500 umap
= isl_union_map_cow(umap
);
1504 if (isl_hash_table_foreach(umap
->dim
->ctx
, &umap
->table
, fn
, NULL
) < 0)
1509 isl_union_map_free(umap
);
1513 static int affine_entry(void **entry
, void *user
)
1515 isl_map
**map
= (isl_map
**)entry
;
1517 *map
= isl_map_from_basic_map(isl_map_affine_hull(*map
));
1519 return *map
? 0 : -1;
1522 __isl_give isl_union_map
*isl_union_map_affine_hull(
1523 __isl_take isl_union_map
*umap
)
1525 return un_op(umap
, &affine_entry
);
1528 __isl_give isl_union_set
*isl_union_set_affine_hull(
1529 __isl_take isl_union_set
*uset
)
1531 return isl_union_map_affine_hull(uset
);
1534 static int polyhedral_entry(void **entry
, void *user
)
1536 isl_map
**map
= (isl_map
**)entry
;
1538 *map
= isl_map_from_basic_map(isl_map_polyhedral_hull(*map
));
1540 return *map
? 0 : -1;
1543 __isl_give isl_union_map
*isl_union_map_polyhedral_hull(
1544 __isl_take isl_union_map
*umap
)
1546 return un_op(umap
, &polyhedral_entry
);
1549 __isl_give isl_union_set
*isl_union_set_polyhedral_hull(
1550 __isl_take isl_union_set
*uset
)
1552 return isl_union_map_polyhedral_hull(uset
);
1555 static int simple_entry(void **entry
, void *user
)
1557 isl_map
**map
= (isl_map
**)entry
;
1559 *map
= isl_map_from_basic_map(isl_map_simple_hull(*map
));
1561 return *map
? 0 : -1;
1564 __isl_give isl_union_map
*isl_union_map_simple_hull(
1565 __isl_take isl_union_map
*umap
)
1567 return un_op(umap
, &simple_entry
);
1570 __isl_give isl_union_set
*isl_union_set_simple_hull(
1571 __isl_take isl_union_set
*uset
)
1573 return isl_union_map_simple_hull(uset
);
1576 static int inplace_entry(void **entry
, void *user
)
1578 __isl_give isl_map
*(*fn
)(__isl_take isl_map
*);
1579 isl_map
**map
= (isl_map
**)entry
;
1582 fn
= *(__isl_give isl_map
*(**)(__isl_take isl_map
*)) user
;
1583 copy
= fn(isl_map_copy(*map
));
1593 static __isl_give isl_union_map
*inplace(__isl_take isl_union_map
*umap
,
1594 __isl_give isl_map
*(*fn
)(__isl_take isl_map
*))
1599 if (isl_hash_table_foreach(umap
->dim
->ctx
, &umap
->table
,
1600 &inplace_entry
, &fn
) < 0)
1605 isl_union_map_free(umap
);
1609 __isl_give isl_union_map
*isl_union_map_coalesce(
1610 __isl_take isl_union_map
*umap
)
1612 return inplace(umap
, &isl_map_coalesce
);
1615 __isl_give isl_union_set
*isl_union_set_coalesce(
1616 __isl_take isl_union_set
*uset
)
1618 return isl_union_map_coalesce(uset
);
1621 __isl_give isl_union_map
*isl_union_map_detect_equalities(
1622 __isl_take isl_union_map
*umap
)
1624 return inplace(umap
, &isl_map_detect_equalities
);
1627 __isl_give isl_union_set
*isl_union_set_detect_equalities(
1628 __isl_take isl_union_set
*uset
)
1630 return isl_union_map_detect_equalities(uset
);
1633 __isl_give isl_union_map
*isl_union_map_compute_divs(
1634 __isl_take isl_union_map
*umap
)
1636 return inplace(umap
, &isl_map_compute_divs
);
1639 __isl_give isl_union_set
*isl_union_set_compute_divs(
1640 __isl_take isl_union_set
*uset
)
1642 return isl_union_map_compute_divs(uset
);
1645 static int lexmin_entry(void **entry
, void *user
)
1647 isl_map
**map
= (isl_map
**)entry
;
1649 *map
= isl_map_lexmin(*map
);
1651 return *map
? 0 : -1;
1654 __isl_give isl_union_map
*isl_union_map_lexmin(
1655 __isl_take isl_union_map
*umap
)
1657 return un_op(umap
, &lexmin_entry
);
1660 __isl_give isl_union_set
*isl_union_set_lexmin(
1661 __isl_take isl_union_set
*uset
)
1663 return isl_union_map_lexmin(uset
);
1666 static int lexmax_entry(void **entry
, void *user
)
1668 isl_map
**map
= (isl_map
**)entry
;
1670 *map
= isl_map_lexmax(*map
);
1672 return *map
? 0 : -1;
1675 __isl_give isl_union_map
*isl_union_map_lexmax(
1676 __isl_take isl_union_map
*umap
)
1678 return un_op(umap
, &lexmax_entry
);
1681 __isl_give isl_union_set
*isl_union_set_lexmax(
1682 __isl_take isl_union_set
*uset
)
1684 return isl_union_map_lexmax(uset
);
1687 static int universe_entry(void **entry
, void *user
)
1689 isl_map
*map
= *entry
;
1690 isl_union_map
**res
= user
;
1692 map
= isl_map_universe(isl_map_get_space(map
));
1693 *res
= isl_union_map_add_map(*res
, map
);
1698 __isl_give isl_union_map
*isl_union_map_universe(__isl_take isl_union_map
*umap
)
1700 return cond_un_op(umap
, &universe_entry
);
1703 __isl_give isl_union_set
*isl_union_set_universe(__isl_take isl_union_set
*uset
)
1705 return isl_union_map_universe(uset
);
1708 static int reverse_entry(void **entry
, void *user
)
1710 isl_map
*map
= *entry
;
1711 isl_union_map
**res
= user
;
1713 *res
= isl_union_map_add_map(*res
, isl_map_reverse(isl_map_copy(map
)));
1718 __isl_give isl_union_map
*isl_union_map_reverse(__isl_take isl_union_map
*umap
)
1720 return cond_un_op(umap
, &reverse_entry
);
1723 static int params_entry(void **entry
, void *user
)
1725 isl_map
*map
= *entry
;
1726 isl_union_set
**res
= user
;
1728 *res
= isl_union_set_add_set(*res
, isl_map_params(isl_map_copy(map
)));
1733 /* Compute the parameter domain of the given union map.
1735 __isl_give isl_set
*isl_union_map_params(__isl_take isl_union_map
*umap
)
1739 empty
= isl_union_map_is_empty(umap
);
1744 space
= isl_union_map_get_space(umap
);
1745 isl_union_map_free(umap
);
1746 return isl_set_empty(space
);
1748 return isl_set_from_union_set(cond_un_op(umap
, ¶ms_entry
));
1750 isl_union_map_free(umap
);
1754 /* Compute the parameter domain of the given union set.
1756 __isl_give isl_set
*isl_union_set_params(__isl_take isl_union_set
*uset
)
1758 return isl_union_map_params(uset
);
1761 static int domain_entry(void **entry
, void *user
)
1763 isl_map
*map
= *entry
;
1764 isl_union_set
**res
= user
;
1766 *res
= isl_union_set_add_set(*res
, isl_map_domain(isl_map_copy(map
)));
1771 __isl_give isl_union_set
*isl_union_map_domain(__isl_take isl_union_map
*umap
)
1773 return cond_un_op(umap
, &domain_entry
);
1776 static int range_entry(void **entry
, void *user
)
1778 isl_map
*map
= *entry
;
1779 isl_union_set
**res
= user
;
1781 *res
= isl_union_set_add_set(*res
, isl_map_range(isl_map_copy(map
)));
1786 __isl_give isl_union_set
*isl_union_map_range(__isl_take isl_union_map
*umap
)
1788 return cond_un_op(umap
, &range_entry
);
1791 static int domain_map_entry(void **entry
, void *user
)
1793 isl_map
*map
= *entry
;
1794 isl_union_set
**res
= user
;
1796 *res
= isl_union_map_add_map(*res
,
1797 isl_map_domain_map(isl_map_copy(map
)));
1802 __isl_give isl_union_map
*isl_union_map_domain_map(
1803 __isl_take isl_union_map
*umap
)
1805 return cond_un_op(umap
, &domain_map_entry
);
1808 static int range_map_entry(void **entry
, void *user
)
1810 isl_map
*map
= *entry
;
1811 isl_union_set
**res
= user
;
1813 *res
= isl_union_map_add_map(*res
,
1814 isl_map_range_map(isl_map_copy(map
)));
1819 __isl_give isl_union_map
*isl_union_map_range_map(
1820 __isl_take isl_union_map
*umap
)
1822 return cond_un_op(umap
, &range_map_entry
);
1825 /* Check if "set" is of the form A[B -> C].
1826 * If so, add A[B -> C] -> B to "res".
1828 static int wrapped_domain_map_entry(void **entry
, void *user
)
1830 isl_set
*set
= *entry
;
1831 isl_union_set
**res
= user
;
1834 wrapping
= isl_set_is_wrapping(set
);
1840 *res
= isl_union_map_add_map(*res
,
1841 isl_set_wrapped_domain_map(isl_set_copy(set
)));
1846 /* Given a collection of wrapped maps of the form A[B -> C],
1847 * return the collection of maps A[B -> C] -> B.
1849 __isl_give isl_union_map
*isl_union_set_wrapped_domain_map(
1850 __isl_take isl_union_set
*uset
)
1852 return cond_un_op(uset
, &wrapped_domain_map_entry
);
1855 static int deltas_entry(void **entry
, void *user
)
1857 isl_map
*map
= *entry
;
1858 isl_union_set
**res
= user
;
1860 if (!isl_space_tuple_is_equal(map
->dim
, isl_dim_in
,
1861 map
->dim
, isl_dim_out
))
1864 *res
= isl_union_set_add_set(*res
, isl_map_deltas(isl_map_copy(map
)));
1869 __isl_give isl_union_set
*isl_union_map_deltas(__isl_take isl_union_map
*umap
)
1871 return cond_un_op(umap
, &deltas_entry
);
1874 static int deltas_map_entry(void **entry
, void *user
)
1876 isl_map
*map
= *entry
;
1877 isl_union_map
**res
= user
;
1879 if (!isl_space_tuple_is_equal(map
->dim
, isl_dim_in
,
1880 map
->dim
, isl_dim_out
))
1883 *res
= isl_union_map_add_map(*res
,
1884 isl_map_deltas_map(isl_map_copy(map
)));
1889 __isl_give isl_union_map
*isl_union_map_deltas_map(
1890 __isl_take isl_union_map
*umap
)
1892 return cond_un_op(umap
, &deltas_map_entry
);
1895 static int identity_entry(void **entry
, void *user
)
1897 isl_set
*set
= *entry
;
1898 isl_union_map
**res
= user
;
1900 *res
= isl_union_map_add_map(*res
, isl_set_identity(isl_set_copy(set
)));
1905 __isl_give isl_union_map
*isl_union_set_identity(__isl_take isl_union_set
*uset
)
1907 return cond_un_op(uset
, &identity_entry
);
1910 /* If "map" is of the form [A -> B] -> C, then add A -> C to "res".
1912 static int domain_factor_domain_entry(void **entry
, void *user
)
1914 isl_map
*map
= *entry
;
1915 isl_union_map
**res
= user
;
1917 if (!isl_map_domain_is_wrapping(map
))
1920 *res
= isl_union_map_add_map(*res
,
1921 isl_map_domain_factor_domain(isl_map_copy(map
)));
1923 return *res
? 0 : -1;
1926 /* For each map in "umap" of the form [A -> B] -> C,
1927 * construct the map A -> C and collect the results.
1929 __isl_give isl_union_map
*isl_union_map_domain_factor_domain(
1930 __isl_take isl_union_map
*umap
)
1932 return cond_un_op(umap
, &domain_factor_domain_entry
);
1935 /* If "map" is of the form [A -> B] -> C, then add B -> C to "res".
1937 static int domain_factor_range_entry(void **entry
, void *user
)
1939 isl_map
*map
= *entry
;
1940 isl_union_map
**res
= user
;
1942 if (!isl_map_domain_is_wrapping(map
))
1945 *res
= isl_union_map_add_map(*res
,
1946 isl_map_domain_factor_range(isl_map_copy(map
)));
1948 return *res
? 0 : -1;
1951 /* For each map in "umap" of the form [A -> B] -> C,
1952 * construct the map B -> C and collect the results.
1954 __isl_give isl_union_map
*isl_union_map_domain_factor_range(
1955 __isl_take isl_union_map
*umap
)
1957 return cond_un_op(umap
, &domain_factor_range_entry
);
1960 /* If "map" is of the form A -> [B -> C], then add A -> C to "res".
1962 static int range_factor_range_entry(void **entry
, void *user
)
1964 isl_map
*map
= *entry
;
1965 isl_union_map
**res
= user
;
1967 if (!isl_map_range_is_wrapping(map
))
1970 *res
= isl_union_map_add_map(*res
,
1971 isl_map_range_factor_range(isl_map_copy(map
)));
1973 return *res
? 0 : -1;
1976 /* For each map in "umap" of the form A -> [B -> C],
1977 * construct the map A -> C and collect the results.
1979 __isl_give isl_union_map
*isl_union_map_range_factor_range(
1980 __isl_take isl_union_map
*umap
)
1982 return cond_un_op(umap
, &range_factor_range_entry
);
1985 /* If "map" is of the form [A -> B] -> [C -> D], then add A -> C to "res".
1987 static int factor_domain_entry(void **entry
, void *user
)
1989 isl_map
*map
= *entry
;
1990 isl_union_map
**res
= user
;
1992 if (!isl_map_domain_is_wrapping(map
) || !isl_map_range_is_wrapping(map
))
1995 *res
= isl_union_map_add_map(*res
,
1996 isl_map_factor_domain(isl_map_copy(map
)));
1998 return *res
? 0 : -1;
2001 /* For each map in "umap" of the form [A -> B] -> [C -> D],
2002 * construct the map A -> C and collect the results.
2004 __isl_give isl_union_map
*isl_union_map_factor_domain(
2005 __isl_take isl_union_map
*umap
)
2007 return cond_un_op(umap
, &factor_domain_entry
);
2010 /* If "map" is of the form [A -> B] -> [C -> D], then add B -> D to "res".
2012 static int factor_range_entry(void **entry
, void *user
)
2014 isl_map
*map
= *entry
;
2015 isl_union_map
**res
= user
;
2017 if (!isl_map_domain_is_wrapping(map
) || !isl_map_range_is_wrapping(map
))
2020 *res
= isl_union_map_add_map(*res
,
2021 isl_map_factor_range(isl_map_copy(map
)));
2023 return *res
? 0 : -1;
2026 /* For each map in "umap" of the form [A -> B] -> [C -> D],
2027 * construct the map B -> D and collect the results.
2029 __isl_give isl_union_map
*isl_union_map_factor_range(
2030 __isl_take isl_union_map
*umap
)
2032 return cond_un_op(umap
, &factor_range_entry
);
2035 static int unwrap_entry(void **entry
, void *user
)
2037 isl_set
*set
= *entry
;
2038 isl_union_set
**res
= user
;
2040 if (!isl_set_is_wrapping(set
))
2043 *res
= isl_union_map_add_map(*res
, isl_set_unwrap(isl_set_copy(set
)));
2048 __isl_give isl_union_map
*isl_union_set_unwrap(__isl_take isl_union_set
*uset
)
2050 return cond_un_op(uset
, &unwrap_entry
);
2053 static int wrap_entry(void **entry
, void *user
)
2055 isl_map
*map
= *entry
;
2056 isl_union_set
**res
= user
;
2058 *res
= isl_union_set_add_set(*res
, isl_map_wrap(isl_map_copy(map
)));
2063 __isl_give isl_union_set
*isl_union_map_wrap(__isl_take isl_union_map
*umap
)
2065 return cond_un_op(umap
, &wrap_entry
);
2068 struct isl_union_map_is_subset_data
{
2069 isl_union_map
*umap2
;
2073 static int is_subset_entry(void **entry
, void *user
)
2075 struct isl_union_map_is_subset_data
*data
= user
;
2077 struct isl_hash_table_entry
*entry2
;
2078 isl_map
*map
= *entry
;
2080 hash
= isl_space_get_hash(map
->dim
);
2081 entry2
= isl_hash_table_find(data
->umap2
->dim
->ctx
, &data
->umap2
->table
,
2082 hash
, &has_dim
, map
->dim
, 0);
2084 int empty
= isl_map_is_empty(map
);
2089 data
->is_subset
= 0;
2093 data
->is_subset
= isl_map_is_subset(map
, entry2
->data
);
2094 if (data
->is_subset
< 0 || !data
->is_subset
)
2100 int isl_union_map_is_subset(__isl_keep isl_union_map
*umap1
,
2101 __isl_keep isl_union_map
*umap2
)
2103 struct isl_union_map_is_subset_data data
= { NULL
, 1 };
2105 umap1
= isl_union_map_copy(umap1
);
2106 umap2
= isl_union_map_copy(umap2
);
2107 umap1
= isl_union_map_align_params(umap1
, isl_union_map_get_space(umap2
));
2108 umap2
= isl_union_map_align_params(umap2
, isl_union_map_get_space(umap1
));
2110 if (!umap1
|| !umap2
)
2114 if (isl_hash_table_foreach(umap1
->dim
->ctx
, &umap1
->table
,
2115 &is_subset_entry
, &data
) < 0 &&
2119 isl_union_map_free(umap1
);
2120 isl_union_map_free(umap2
);
2122 return data
.is_subset
;
2124 isl_union_map_free(umap1
);
2125 isl_union_map_free(umap2
);
2129 int isl_union_set_is_subset(__isl_keep isl_union_set
*uset1
,
2130 __isl_keep isl_union_set
*uset2
)
2132 return isl_union_map_is_subset(uset1
, uset2
);
2135 int isl_union_map_is_equal(__isl_keep isl_union_map
*umap1
,
2136 __isl_keep isl_union_map
*umap2
)
2140 if (!umap1
|| !umap2
)
2142 is_subset
= isl_union_map_is_subset(umap1
, umap2
);
2145 is_subset
= isl_union_map_is_subset(umap2
, umap1
);
2149 int isl_union_set_is_equal(__isl_keep isl_union_set
*uset1
,
2150 __isl_keep isl_union_set
*uset2
)
2152 return isl_union_map_is_equal(uset1
, uset2
);
2155 int isl_union_map_is_strict_subset(__isl_keep isl_union_map
*umap1
,
2156 __isl_keep isl_union_map
*umap2
)
2160 if (!umap1
|| !umap2
)
2162 is_subset
= isl_union_map_is_subset(umap1
, umap2
);
2165 is_subset
= isl_union_map_is_subset(umap2
, umap1
);
2166 if (is_subset
== -1)
2171 int isl_union_set_is_strict_subset(__isl_keep isl_union_set
*uset1
,
2172 __isl_keep isl_union_set
*uset2
)
2174 return isl_union_map_is_strict_subset(uset1
, uset2
);
2177 static int sample_entry(void **entry
, void *user
)
2179 isl_basic_map
**sample
= (isl_basic_map
**)user
;
2180 isl_map
*map
= *entry
;
2182 *sample
= isl_map_sample(isl_map_copy(map
));
2185 if (!isl_basic_map_plain_is_empty(*sample
))
2190 __isl_give isl_basic_map
*isl_union_map_sample(__isl_take isl_union_map
*umap
)
2192 isl_basic_map
*sample
= NULL
;
2197 if (isl_hash_table_foreach(umap
->dim
->ctx
, &umap
->table
,
2198 &sample_entry
, &sample
) < 0 &&
2203 sample
= isl_basic_map_empty(isl_union_map_get_space(umap
));
2205 isl_union_map_free(umap
);
2209 isl_union_map_free(umap
);
2213 __isl_give isl_basic_set
*isl_union_set_sample(__isl_take isl_union_set
*uset
)
2215 return (isl_basic_set
*)isl_union_map_sample(uset
);
2218 struct isl_forall_data
{
2220 int (*fn
)(__isl_keep isl_map
*map
);
2223 static int forall_entry(void **entry
, void *user
)
2225 struct isl_forall_data
*data
= user
;
2226 isl_map
*map
= *entry
;
2228 data
->res
= data
->fn(map
);
2238 static int union_map_forall(__isl_keep isl_union_map
*umap
,
2239 int (*fn
)(__isl_keep isl_map
*map
))
2241 struct isl_forall_data data
= { 1, fn
};
2246 if (isl_hash_table_foreach(umap
->dim
->ctx
, &umap
->table
,
2247 &forall_entry
, &data
) < 0 && data
.res
)
2253 struct isl_forall_user_data
{
2255 int (*fn
)(__isl_keep isl_map
*map
, void *user
);
2259 static int forall_user_entry(void **entry
, void *user
)
2261 struct isl_forall_user_data
*data
= user
;
2262 isl_map
*map
= *entry
;
2264 data
->res
= data
->fn(map
, data
->user
);
2274 /* Check if fn(map, user) returns true for all maps "map" in umap.
2276 static int union_map_forall_user(__isl_keep isl_union_map
*umap
,
2277 int (*fn
)(__isl_keep isl_map
*map
, void *user
), void *user
)
2279 struct isl_forall_user_data data
= { 1, fn
, user
};
2284 if (isl_hash_table_foreach(umap
->dim
->ctx
, &umap
->table
,
2285 &forall_user_entry
, &data
) < 0 && data
.res
)
2291 int isl_union_map_is_empty(__isl_keep isl_union_map
*umap
)
2293 return union_map_forall(umap
, &isl_map_is_empty
);
2296 int isl_union_set_is_empty(__isl_keep isl_union_set
*uset
)
2298 return isl_union_map_is_empty(uset
);
2301 static int is_subset_of_identity(__isl_keep isl_map
*map
)
2310 if (!isl_space_tuple_is_equal(map
->dim
, isl_dim_in
,
2311 map
->dim
, isl_dim_out
))
2314 dim
= isl_map_get_space(map
);
2315 id
= isl_map_identity(dim
);
2317 is_subset
= isl_map_is_subset(map
, id
);
2324 /* Given an isl_union_map that consists of a single map, check
2325 * if it is single-valued.
2327 static int single_map_is_single_valued(__isl_keep isl_union_map
*umap
)
2332 umap
= isl_union_map_copy(umap
);
2333 map
= isl_map_from_union_map(umap
);
2334 sv
= isl_map_is_single_valued(map
);
2340 /* Internal data structure for single_valued_on_domain.
2342 * "umap" is the union map to be tested.
2343 * "sv" is set to 1 as long as "umap" may still be single-valued.
2345 struct isl_union_map_is_sv_data
{
2346 isl_union_map
*umap
;
2350 /* Check if the data->umap is single-valued on "set".
2352 * If data->umap consists of a single map on "set", then test it
2355 * Otherwise, compute
2359 * check if the result is a subset of the identity mapping and
2360 * store the result in data->sv.
2362 * Terminate as soon as data->umap has been determined not to
2365 static int single_valued_on_domain(__isl_take isl_set
*set
, void *user
)
2367 struct isl_union_map_is_sv_data
*data
= user
;
2368 isl_union_map
*umap
, *test
;
2370 umap
= isl_union_map_copy(data
->umap
);
2371 umap
= isl_union_map_intersect_domain(umap
,
2372 isl_union_set_from_set(set
));
2374 if (isl_union_map_n_map(umap
) == 1) {
2375 data
->sv
= single_map_is_single_valued(umap
);
2376 isl_union_map_free(umap
);
2378 test
= isl_union_map_reverse(isl_union_map_copy(umap
));
2379 test
= isl_union_map_apply_range(test
, umap
);
2381 data
->sv
= union_map_forall(test
, &is_subset_of_identity
);
2383 isl_union_map_free(test
);
2386 if (data
->sv
< 0 || !data
->sv
)
2391 /* Check if the given map is single-valued.
2393 * If the union map consists of a single map, then test it as an isl_map.
2394 * Otherwise, check if the union map is single-valued on each of its
2397 int isl_union_map_is_single_valued(__isl_keep isl_union_map
*umap
)
2399 isl_union_map
*universe
;
2400 isl_union_set
*domain
;
2401 struct isl_union_map_is_sv_data data
;
2403 if (isl_union_map_n_map(umap
) == 1)
2404 return single_map_is_single_valued(umap
);
2406 universe
= isl_union_map_universe(isl_union_map_copy(umap
));
2407 domain
= isl_union_map_domain(universe
);
2411 if (isl_union_set_foreach_set(domain
,
2412 &single_valued_on_domain
, &data
) < 0 && data
.sv
)
2414 isl_union_set_free(domain
);
2419 int isl_union_map_is_injective(__isl_keep isl_union_map
*umap
)
2423 umap
= isl_union_map_copy(umap
);
2424 umap
= isl_union_map_reverse(umap
);
2425 in
= isl_union_map_is_single_valued(umap
);
2426 isl_union_map_free(umap
);
2431 /* Represents a map that has a fixed value (v) for one of its
2433 * The map in this structure is not reference counted, so it
2434 * is only valid while the isl_union_map from which it was
2435 * obtained is still alive.
2437 struct isl_fixed_map
{
2442 static struct isl_fixed_map
*alloc_isl_fixed_map_array(isl_ctx
*ctx
,
2446 struct isl_fixed_map
*v
;
2448 v
= isl_calloc_array(ctx
, struct isl_fixed_map
, n
);
2451 for (i
= 0; i
< n
; ++i
)
2452 isl_int_init(v
[i
].v
);
2456 static void free_isl_fixed_map_array(struct isl_fixed_map
*v
, int n
)
2462 for (i
= 0; i
< n
; ++i
)
2463 isl_int_clear(v
[i
].v
);
2467 /* Compare the "v" field of two isl_fixed_map structs.
2469 static int qsort_fixed_map_cmp(const void *p1
, const void *p2
)
2471 const struct isl_fixed_map
*e1
= (const struct isl_fixed_map
*) p1
;
2472 const struct isl_fixed_map
*e2
= (const struct isl_fixed_map
*) p2
;
2474 return isl_int_cmp(e1
->v
, e2
->v
);
2477 /* Internal data structure used while checking whether all maps
2478 * in a union_map have a fixed value for a given output dimension.
2479 * v is the list of maps, with the fixed value for the dimension
2480 * n is the number of maps considered so far
2481 * pos is the output dimension under investigation
2483 struct isl_fixed_dim_data
{
2484 struct isl_fixed_map
*v
;
2489 static int fixed_at_pos(__isl_keep isl_map
*map
, void *user
)
2491 struct isl_fixed_dim_data
*data
= user
;
2493 data
->v
[data
->n
].map
= map
;
2494 return isl_map_plain_is_fixed(map
, isl_dim_out
, data
->pos
,
2495 &data
->v
[data
->n
++].v
);
2498 static int plain_injective_on_range(__isl_take isl_union_map
*umap
,
2499 int first
, int n_range
);
2501 /* Given a list of the maps, with their fixed values at output dimension "pos",
2502 * check whether the ranges of the maps form an obvious partition.
2504 * We first sort the maps according to their fixed values.
2505 * If all maps have a different value, then we know the ranges form
2507 * Otherwise, we collect the maps with the same fixed value and
2508 * check whether each such collection is obviously injective
2509 * based on later dimensions.
2511 static int separates(struct isl_fixed_map
*v
, int n
,
2512 __isl_take isl_space
*dim
, int pos
, int n_range
)
2519 qsort(v
, n
, sizeof(*v
), &qsort_fixed_map_cmp
);
2521 for (i
= 0; i
+ 1 < n
; ++i
) {
2523 isl_union_map
*part
;
2526 for (j
= i
+ 1; j
< n
; ++j
)
2527 if (isl_int_ne(v
[i
].v
, v
[j
].v
))
2533 part
= isl_union_map_alloc(isl_space_copy(dim
), j
- i
);
2534 for (k
= i
; k
< j
; ++k
)
2535 part
= isl_union_map_add_map(part
,
2536 isl_map_copy(v
[k
].map
));
2538 injective
= plain_injective_on_range(part
, pos
+ 1, n_range
);
2547 isl_space_free(dim
);
2548 free_isl_fixed_map_array(v
, n
);
2551 isl_space_free(dim
);
2552 free_isl_fixed_map_array(v
, n
);
2556 /* Check whether the maps in umap have obviously distinct ranges.
2557 * In particular, check for an output dimension in the range
2558 * [first,n_range) for which all maps have a fixed value
2559 * and then check if these values, possibly along with fixed values
2560 * at later dimensions, entail distinct ranges.
2562 static int plain_injective_on_range(__isl_take isl_union_map
*umap
,
2563 int first
, int n_range
)
2567 struct isl_fixed_dim_data data
= { NULL
};
2569 ctx
= isl_union_map_get_ctx(umap
);
2571 n
= isl_union_map_n_map(umap
);
2576 isl_union_map_free(umap
);
2580 if (first
>= n_range
) {
2581 isl_union_map_free(umap
);
2585 data
.v
= alloc_isl_fixed_map_array(ctx
, n
);
2589 for (data
.pos
= first
; data
.pos
< n_range
; ++data
.pos
) {
2595 fixed
= union_map_forall_user(umap
, &fixed_at_pos
, &data
);
2600 dim
= isl_union_map_get_space(umap
);
2601 injective
= separates(data
.v
, n
, dim
, data
.pos
, n_range
);
2602 isl_union_map_free(umap
);
2606 free_isl_fixed_map_array(data
.v
, n
);
2607 isl_union_map_free(umap
);
2611 free_isl_fixed_map_array(data
.v
, n
);
2612 isl_union_map_free(umap
);
2616 /* Check whether the maps in umap that map to subsets of "ran"
2617 * have obviously distinct ranges.
2619 static int plain_injective_on_range_wrap(__isl_keep isl_set
*ran
, void *user
)
2621 isl_union_map
*umap
= user
;
2623 umap
= isl_union_map_copy(umap
);
2624 umap
= isl_union_map_intersect_range(umap
,
2625 isl_union_set_from_set(isl_set_copy(ran
)));
2626 return plain_injective_on_range(umap
, 0, isl_set_dim(ran
, isl_dim_set
));
2629 /* Check if the given union_map is obviously injective.
2631 * In particular, we first check if all individual maps are obviously
2632 * injective and then check if all the ranges of these maps are
2633 * obviously disjoint.
2635 int isl_union_map_plain_is_injective(__isl_keep isl_union_map
*umap
)
2638 isl_union_map
*univ
;
2641 in
= union_map_forall(umap
, &isl_map_plain_is_injective
);
2647 univ
= isl_union_map_universe(isl_union_map_copy(umap
));
2648 ran
= isl_union_map_range(univ
);
2650 in
= union_map_forall_user(ran
, &plain_injective_on_range_wrap
, umap
);
2652 isl_union_set_free(ran
);
2657 int isl_union_map_is_bijective(__isl_keep isl_union_map
*umap
)
2661 sv
= isl_union_map_is_single_valued(umap
);
2665 return isl_union_map_is_injective(umap
);
2668 static int zip_entry(void **entry
, void *user
)
2670 isl_map
*map
= *entry
;
2671 isl_union_map
**res
= user
;
2673 if (!isl_map_can_zip(map
))
2676 *res
= isl_union_map_add_map(*res
, isl_map_zip(isl_map_copy(map
)));
2681 __isl_give isl_union_map
*isl_union_map_zip(__isl_take isl_union_map
*umap
)
2683 return cond_un_op(umap
, &zip_entry
);
2686 static int uncurry_entry(void **entry
, void *user
)
2688 isl_map
*map
= *entry
;
2689 isl_union_map
**res
= user
;
2691 if (!isl_map_can_uncurry(map
))
2694 *res
= isl_union_map_add_map(*res
, isl_map_uncurry(isl_map_copy(map
)));
2699 /* Given a union map, take the maps of the form A -> (B -> C) and
2700 * return the union of the corresponding maps (A -> B) -> C.
2702 __isl_give isl_union_map
*isl_union_map_uncurry(__isl_take isl_union_map
*umap
)
2704 return cond_un_op(umap
, &uncurry_entry
);
2707 static int curry_entry(void **entry
, void *user
)
2709 isl_map
*map
= *entry
;
2710 isl_union_map
**res
= user
;
2712 if (!isl_map_can_curry(map
))
2715 *res
= isl_union_map_add_map(*res
, isl_map_curry(isl_map_copy(map
)));
2720 /* Given a union map, take the maps of the form (A -> B) -> C and
2721 * return the union of the corresponding maps A -> (B -> C).
2723 __isl_give isl_union_map
*isl_union_map_curry(__isl_take isl_union_map
*umap
)
2725 return cond_un_op(umap
, &curry_entry
);
2728 static int lift_entry(void **entry
, void *user
)
2730 isl_set
*set
= *entry
;
2731 isl_union_set
**res
= user
;
2733 *res
= isl_union_set_add_set(*res
, isl_set_lift(isl_set_copy(set
)));
2738 __isl_give isl_union_set
*isl_union_set_lift(__isl_take isl_union_set
*uset
)
2740 return cond_un_op(uset
, &lift_entry
);
2743 static int coefficients_entry(void **entry
, void *user
)
2745 isl_set
*set
= *entry
;
2746 isl_union_set
**res
= user
;
2748 set
= isl_set_copy(set
);
2749 set
= isl_set_from_basic_set(isl_set_coefficients(set
));
2750 *res
= isl_union_set_add_set(*res
, set
);
2755 __isl_give isl_union_set
*isl_union_set_coefficients(
2756 __isl_take isl_union_set
*uset
)
2765 ctx
= isl_union_set_get_ctx(uset
);
2766 dim
= isl_space_set_alloc(ctx
, 0, 0);
2767 res
= isl_union_map_alloc(dim
, uset
->table
.n
);
2768 if (isl_hash_table_foreach(uset
->dim
->ctx
, &uset
->table
,
2769 &coefficients_entry
, &res
) < 0)
2772 isl_union_set_free(uset
);
2775 isl_union_set_free(uset
);
2776 isl_union_set_free(res
);
2780 static int solutions_entry(void **entry
, void *user
)
2782 isl_set
*set
= *entry
;
2783 isl_union_set
**res
= user
;
2785 set
= isl_set_copy(set
);
2786 set
= isl_set_from_basic_set(isl_set_solutions(set
));
2788 *res
= isl_union_set_from_set(set
);
2790 *res
= isl_union_set_add_set(*res
, set
);
2798 __isl_give isl_union_set
*isl_union_set_solutions(
2799 __isl_take isl_union_set
*uset
)
2801 isl_union_set
*res
= NULL
;
2806 if (uset
->table
.n
== 0) {
2807 res
= isl_union_set_empty(isl_union_set_get_space(uset
));
2808 isl_union_set_free(uset
);
2812 if (isl_hash_table_foreach(uset
->dim
->ctx
, &uset
->table
,
2813 &solutions_entry
, &res
) < 0)
2816 isl_union_set_free(uset
);
2819 isl_union_set_free(uset
);
2820 isl_union_set_free(res
);
2824 /* Is the domain space of "map" equal to "space"?
2826 static int domain_match(__isl_keep isl_map
*map
, __isl_keep isl_space
*space
)
2828 return isl_space_tuple_is_equal(map
->dim
, isl_dim_in
,
2829 space
, isl_dim_out
);
2832 /* Is the range space of "map" equal to "space"?
2834 static int range_match(__isl_keep isl_map
*map
, __isl_keep isl_space
*space
)
2836 return isl_space_tuple_is_equal(map
->dim
, isl_dim_out
,
2837 space
, isl_dim_out
);
2840 /* Is the set space of "map" equal to "space"?
2842 static int set_match(__isl_keep isl_map
*map
, __isl_keep isl_space
*space
)
2844 return isl_space_tuple_is_equal(map
->dim
, isl_dim_set
,
2845 space
, isl_dim_out
);
2848 /* Internal data structure for preimage_pw_multi_aff.
2850 * "pma" is the function under which the preimage should be taken.
2851 * "space" is the space of "pma".
2852 * "res" collects the results.
2853 * "fn" computes the preimage for a given map.
2854 * "match" returns true if "fn" can be called.
2856 struct isl_union_map_preimage_data
{
2858 isl_pw_multi_aff
*pma
;
2860 int (*match
)(__isl_keep isl_map
*map
, __isl_keep isl_space
*space
);
2861 __isl_give isl_map
*(*fn
)(__isl_take isl_map
*map
,
2862 __isl_take isl_pw_multi_aff
*pma
);
2865 /* Call data->fn to compute the preimage of the domain or range of *entry
2866 * under the function represented by data->pma, provided the domain/range
2867 * space of *entry matches the target space of data->pma
2868 * (as given by data->match), and add the result to data->res.
2870 static int preimage_entry(void **entry
, void *user
)
2873 isl_map
*map
= *entry
;
2874 struct isl_union_map_preimage_data
*data
= user
;
2877 m
= data
->match(map
, data
->space
);
2883 map
= isl_map_copy(map
);
2884 map
= data
->fn(map
, isl_pw_multi_aff_copy(data
->pma
));
2886 empty
= isl_map_is_empty(map
);
2887 if (empty
< 0 || empty
) {
2889 return empty
< 0 ? -1 : 0;
2892 data
->res
= isl_union_map_add_map(data
->res
, map
);
2897 /* Compute the preimage of the domain or range of "umap" under the function
2898 * represented by "pma".
2899 * In other words, plug in "pma" in the domain or range of "umap".
2900 * The function "fn" performs the actual preimage computation on a map,
2901 * while "match" determines to which maps the function should be applied.
2903 static __isl_give isl_union_map
*preimage_pw_multi_aff(
2904 __isl_take isl_union_map
*umap
, __isl_take isl_pw_multi_aff
*pma
,
2905 int (*match
)(__isl_keep isl_map
*map
, __isl_keep isl_space
*space
),
2906 __isl_give isl_map
*(*fn
)(__isl_take isl_map
*map
,
2907 __isl_take isl_pw_multi_aff
*pma
))
2911 struct isl_union_map_preimage_data data
;
2913 umap
= isl_union_map_align_params(umap
,
2914 isl_pw_multi_aff_get_space(pma
));
2915 pma
= isl_pw_multi_aff_align_params(pma
, isl_union_map_get_space(umap
));
2920 ctx
= isl_union_map_get_ctx(umap
);
2921 space
= isl_union_map_get_space(umap
);
2922 data
.space
= isl_pw_multi_aff_get_space(pma
);
2924 data
.res
= isl_union_map_alloc(space
, umap
->table
.n
);
2927 if (isl_hash_table_foreach(ctx
, &umap
->table
, &preimage_entry
,
2929 data
.res
= isl_union_map_free(data
.res
);
2931 isl_space_free(data
.space
);
2932 isl_union_map_free(umap
);
2933 isl_pw_multi_aff_free(pma
);
2936 isl_union_map_free(umap
);
2937 isl_pw_multi_aff_free(pma
);
2941 /* Compute the preimage of the domain of "umap" under the function
2942 * represented by "pma".
2943 * In other words, plug in "pma" in the domain of "umap".
2944 * The result contains maps that live in the same spaces as the maps of "umap"
2945 * with domain space equal to the target space of "pma",
2946 * except that the domain has been replaced by the domain space of "pma".
2948 __isl_give isl_union_map
*isl_union_map_preimage_domain_pw_multi_aff(
2949 __isl_take isl_union_map
*umap
, __isl_take isl_pw_multi_aff
*pma
)
2951 return preimage_pw_multi_aff(umap
, pma
, &domain_match
,
2952 &isl_map_preimage_domain_pw_multi_aff
);
2955 /* Compute the preimage of the range of "umap" under the function
2956 * represented by "pma".
2957 * In other words, plug in "pma" in the range of "umap".
2958 * The result contains maps that live in the same spaces as the maps of "umap"
2959 * with range space equal to the target space of "pma",
2960 * except that the range has been replaced by the domain space of "pma".
2962 __isl_give isl_union_map
*isl_union_map_preimage_range_pw_multi_aff(
2963 __isl_take isl_union_map
*umap
, __isl_take isl_pw_multi_aff
*pma
)
2965 return preimage_pw_multi_aff(umap
, pma
, &range_match
,
2966 &isl_map_preimage_range_pw_multi_aff
);
2969 /* Compute the preimage of "uset" under the function represented by "pma".
2970 * In other words, plug in "pma" in "uset".
2971 * The result contains sets that live in the same spaces as the sets of "uset"
2972 * with space equal to the target space of "pma",
2973 * except that the space has been replaced by the domain space of "pma".
2975 __isl_give isl_union_set
*isl_union_set_preimage_pw_multi_aff(
2976 __isl_take isl_union_set
*uset
, __isl_take isl_pw_multi_aff
*pma
)
2978 return preimage_pw_multi_aff(uset
, pma
, &set_match
,
2979 &isl_set_preimage_pw_multi_aff
);
2982 /* Compute the preimage of the domain of "umap" under the function
2983 * represented by "ma".
2984 * In other words, plug in "ma" in the domain of "umap".
2985 * The result contains maps that live in the same spaces as the maps of "umap"
2986 * with domain space equal to the target space of "ma",
2987 * except that the domain has been replaced by the domain space of "ma".
2989 __isl_give isl_union_map
*isl_union_map_preimage_domain_multi_aff(
2990 __isl_take isl_union_map
*umap
, __isl_take isl_multi_aff
*ma
)
2992 return isl_union_map_preimage_domain_pw_multi_aff(umap
,
2993 isl_pw_multi_aff_from_multi_aff(ma
));
2996 /* Compute the preimage of the range of "umap" under the function
2997 * represented by "ma".
2998 * In other words, plug in "ma" in the range of "umap".
2999 * The result contains maps that live in the same spaces as the maps of "umap"
3000 * with range space equal to the target space of "ma",
3001 * except that the range has been replaced by the domain space of "ma".
3003 __isl_give isl_union_map
*isl_union_map_preimage_range_multi_aff(
3004 __isl_take isl_union_map
*umap
, __isl_take isl_multi_aff
*ma
)
3006 return isl_union_map_preimage_range_pw_multi_aff(umap
,
3007 isl_pw_multi_aff_from_multi_aff(ma
));
3010 /* Compute the preimage of "uset" under the function represented by "ma".
3011 * In other words, plug in "ma" in "uset".
3012 * The result contains sets that live in the same spaces as the sets of "uset"
3013 * with space equal to the target space of "ma",
3014 * except that the space has been replaced by the domain space of "ma".
3016 __isl_give isl_union_map
*isl_union_set_preimage_multi_aff(
3017 __isl_take isl_union_set
*uset
, __isl_take isl_multi_aff
*ma
)
3019 return isl_union_set_preimage_pw_multi_aff(uset
,
3020 isl_pw_multi_aff_from_multi_aff(ma
));
3023 /* Internal data structure for preimage_multi_pw_aff.
3025 * "mpa" is the function under which the preimage should be taken.
3026 * "space" is the space of "mpa".
3027 * "res" collects the results.
3028 * "fn" computes the preimage for a given map.
3029 * "match" returns true if "fn" can be called.
3031 struct isl_union_map_preimage_mpa_data
{
3033 isl_multi_pw_aff
*mpa
;
3035 int (*match
)(__isl_keep isl_map
*map
, __isl_keep isl_space
*space
);
3036 __isl_give isl_map
*(*fn
)(__isl_take isl_map
*map
,
3037 __isl_take isl_multi_pw_aff
*mpa
);
3040 /* Call data->fn to compute the preimage of the domain or range of *entry
3041 * under the function represented by data->mpa, provided the domain/range
3042 * space of *entry matches the target space of data->mpa
3043 * (as given by data->match), and add the result to data->res.
3045 static int preimage_mpa_entry(void **entry
, void *user
)
3048 isl_map
*map
= *entry
;
3049 struct isl_union_map_preimage_mpa_data
*data
= user
;
3052 m
= data
->match(map
, data
->space
);
3058 map
= isl_map_copy(map
);
3059 map
= data
->fn(map
, isl_multi_pw_aff_copy(data
->mpa
));
3061 empty
= isl_map_is_empty(map
);
3062 if (empty
< 0 || empty
) {
3064 return empty
< 0 ? -1 : 0;
3067 data
->res
= isl_union_map_add_map(data
->res
, map
);
3072 /* Compute the preimage of the domain or range of "umap" under the function
3073 * represented by "mpa".
3074 * In other words, plug in "mpa" in the domain or range of "umap".
3075 * The function "fn" performs the actual preimage computation on a map,
3076 * while "match" determines to which maps the function should be applied.
3078 static __isl_give isl_union_map
*preimage_multi_pw_aff(
3079 __isl_take isl_union_map
*umap
, __isl_take isl_multi_pw_aff
*mpa
,
3080 int (*match
)(__isl_keep isl_map
*map
, __isl_keep isl_space
*space
),
3081 __isl_give isl_map
*(*fn
)(__isl_take isl_map
*map
,
3082 __isl_take isl_multi_pw_aff
*mpa
))
3086 struct isl_union_map_preimage_mpa_data data
;
3088 umap
= isl_union_map_align_params(umap
,
3089 isl_multi_pw_aff_get_space(mpa
));
3090 mpa
= isl_multi_pw_aff_align_params(mpa
, isl_union_map_get_space(umap
));
3095 ctx
= isl_union_map_get_ctx(umap
);
3096 space
= isl_union_map_get_space(umap
);
3097 data
.space
= isl_multi_pw_aff_get_space(mpa
);
3099 data
.res
= isl_union_map_alloc(space
, umap
->table
.n
);
3102 if (isl_hash_table_foreach(ctx
, &umap
->table
, &preimage_mpa_entry
,
3104 data
.res
= isl_union_map_free(data
.res
);
3106 isl_space_free(data
.space
);
3107 isl_union_map_free(umap
);
3108 isl_multi_pw_aff_free(mpa
);
3111 isl_union_map_free(umap
);
3112 isl_multi_pw_aff_free(mpa
);
3116 /* Compute the preimage of the domain of "umap" under the function
3117 * represented by "mpa".
3118 * In other words, plug in "mpa" in the domain of "umap".
3119 * The result contains maps that live in the same spaces as the maps of "umap"
3120 * with domain space equal to the target space of "mpa",
3121 * except that the domain has been replaced by the domain space of "mpa".
3123 __isl_give isl_union_map
*isl_union_map_preimage_domain_multi_pw_aff(
3124 __isl_take isl_union_map
*umap
, __isl_take isl_multi_pw_aff
*mpa
)
3126 return preimage_multi_pw_aff(umap
, mpa
, &domain_match
,
3127 &isl_map_preimage_domain_multi_pw_aff
);
3130 /* Internal data structure for preimage_upma.
3132 * "umap" is the map of which the preimage should be computed.
3133 * "res" collects the results.
3134 * "fn" computes the preimage for a given piecewise multi-affine function.
3136 struct isl_union_map_preimage_upma_data
{
3137 isl_union_map
*umap
;
3139 __isl_give isl_union_map
*(*fn
)(__isl_take isl_union_map
*umap
,
3140 __isl_take isl_pw_multi_aff
*pma
);
3143 /* Call data->fn to compute the preimage of the domain or range of data->umap
3144 * under the function represented by pma and add the result to data->res.
3146 static int preimage_upma(__isl_take isl_pw_multi_aff
*pma
, void *user
)
3148 struct isl_union_map_preimage_upma_data
*data
= user
;
3149 isl_union_map
*umap
;
3151 umap
= isl_union_map_copy(data
->umap
);
3152 umap
= data
->fn(umap
, pma
);
3153 data
->res
= isl_union_map_union(data
->res
, umap
);
3155 return data
->res
? 0 : -1;
3158 /* Compute the preimage of the domain or range of "umap" under the function
3159 * represented by "upma".
3160 * In other words, plug in "upma" in the domain or range of "umap".
3161 * The function "fn" performs the actual preimage computation
3162 * on a piecewise multi-affine function.
3164 static __isl_give isl_union_map
*preimage_union_pw_multi_aff(
3165 __isl_take isl_union_map
*umap
,
3166 __isl_take isl_union_pw_multi_aff
*upma
,
3167 __isl_give isl_union_map
*(*fn
)(__isl_take isl_union_map
*umap
,
3168 __isl_take isl_pw_multi_aff
*pma
))
3170 struct isl_union_map_preimage_upma_data data
;
3173 data
.res
= isl_union_map_empty(isl_union_map_get_space(umap
));
3175 if (isl_union_pw_multi_aff_foreach_pw_multi_aff(upma
,
3176 &preimage_upma
, &data
) < 0)
3177 data
.res
= isl_union_map_free(data
.res
);
3179 isl_union_map_free(umap
);
3180 isl_union_pw_multi_aff_free(upma
);
3185 /* Compute the preimage of the domain of "umap" under the function
3186 * represented by "upma".
3187 * In other words, plug in "upma" in the domain of "umap".
3188 * The result contains maps that live in the same spaces as the maps of "umap"
3189 * with domain space equal to one of the target spaces of "upma",
3190 * except that the domain has been replaced by one of the the domain spaces that
3191 * corresponds to that target space of "upma".
3193 __isl_give isl_union_map
*isl_union_map_preimage_domain_union_pw_multi_aff(
3194 __isl_take isl_union_map
*umap
,
3195 __isl_take isl_union_pw_multi_aff
*upma
)
3197 return preimage_union_pw_multi_aff(umap
, upma
,
3198 &isl_union_map_preimage_domain_pw_multi_aff
);
3201 /* Compute the preimage of the range of "umap" under the function
3202 * represented by "upma".
3203 * In other words, plug in "upma" in the range of "umap".
3204 * The result contains maps that live in the same spaces as the maps of "umap"
3205 * with range space equal to one of the target spaces of "upma",
3206 * except that the range has been replaced by one of the the domain spaces that
3207 * corresponds to that target space of "upma".
3209 __isl_give isl_union_map
*isl_union_map_preimage_range_union_pw_multi_aff(
3210 __isl_take isl_union_map
*umap
,
3211 __isl_take isl_union_pw_multi_aff
*upma
)
3213 return preimage_union_pw_multi_aff(umap
, upma
,
3214 &isl_union_map_preimage_range_pw_multi_aff
);
3217 /* Compute the preimage of "uset" under the function represented by "upma".
3218 * In other words, plug in "upma" in the range of "uset".
3219 * The result contains sets that live in the same spaces as the sets of "uset"
3220 * with space equal to one of the target spaces of "upma",
3221 * except that the space has been replaced by one of the the domain spaces that
3222 * corresponds to that target space of "upma".
3224 __isl_give isl_union_set
*isl_union_set_preimage_union_pw_multi_aff(
3225 __isl_take isl_union_set
*uset
,
3226 __isl_take isl_union_pw_multi_aff
*upma
)
3228 return preimage_union_pw_multi_aff(uset
, upma
,
3229 &isl_union_set_preimage_pw_multi_aff
);
3232 /* Reset the user pointer on all identifiers of parameters and tuples
3233 * of the space of *entry.
3235 static int reset_user(void **entry
, void *user
)
3237 isl_map
**map
= (isl_map
**)entry
;
3239 *map
= isl_map_reset_user(*map
);
3241 return *map
? 0 : -1;
3244 /* Reset the user pointer on all identifiers of parameters and tuples
3245 * of the spaces of "umap".
3247 __isl_give isl_union_map
*isl_union_map_reset_user(
3248 __isl_take isl_union_map
*umap
)
3250 umap
= isl_union_map_cow(umap
);
3253 umap
->dim
= isl_space_reset_user(umap
->dim
);
3255 return isl_union_map_free(umap
);
3256 umap
= un_op(umap
, &reset_user
);
3261 /* Reset the user pointer on all identifiers of parameters and tuples
3262 * of the spaces of "uset".
3264 __isl_give isl_union_set
*isl_union_set_reset_user(
3265 __isl_take isl_union_set
*uset
)
3267 return isl_union_map_reset_user(uset
);
3270 /* Internal data structure for isl_union_map_project_out.
3271 * "type", "first" and "n" are the arguments for the isl_map_project_out
3273 * "res" collects the results.
3275 struct isl_union_map_project_out_data
{
3276 enum isl_dim_type type
;
3283 /* Turn the data->n dimensions of type data->type, starting at data->first
3284 * into existentially quantified variables and add the result to data->res.
3286 static int project_out(__isl_take isl_map
*map
, void *user
)
3288 struct isl_union_map_project_out_data
*data
= user
;
3290 map
= isl_map_project_out(map
, data
->type
, data
->first
, data
->n
);
3291 data
->res
= isl_union_map_add_map(data
->res
, map
);
3296 /* Turn the "n" dimensions of type "type", starting at "first"
3297 * into existentially quantified variables.
3298 * Since the space of an isl_union_map only contains parameters,
3299 * type is required to be equal to isl_dim_param.
3301 __isl_give isl_union_map
*isl_union_map_project_out(
3302 __isl_take isl_union_map
*umap
,
3303 enum isl_dim_type type
, unsigned first
, unsigned n
)
3306 struct isl_union_map_project_out_data data
= { type
, first
, n
};
3311 if (type
!= isl_dim_param
)
3312 isl_die(isl_union_map_get_ctx(umap
), isl_error_invalid
,
3313 "can only project out parameters",
3314 return isl_union_map_free(umap
));
3316 space
= isl_union_map_get_space(umap
);
3317 space
= isl_space_drop_dims(space
, type
, first
, n
);
3318 data
.res
= isl_union_map_empty(space
);
3319 if (isl_union_map_foreach_map(umap
, &project_out
, &data
) < 0)
3320 data
.res
= isl_union_map_free(data
.res
);
3322 isl_union_map_free(umap
);
3327 /* Internal data structure for isl_union_map_involves_dims.
3328 * "first" and "n" are the arguments for the isl_map_involves_dims calls.
3330 struct isl_union_map_involves_dims_data
{
3335 /* Does "map" _not_ involve the data->n parameters starting at data->first?
3337 static int map_excludes(__isl_keep isl_map
*map
, void *user
)
3339 struct isl_union_map_involves_dims_data
*data
= user
;
3342 involves
= isl_map_involves_dims(map
,
3343 isl_dim_param
, data
->first
, data
->n
);
3349 /* Does "umap" involve any of the n parameters starting at first?
3350 * "type" is required to be set to isl_dim_param.
3352 * "umap" involves any of those parameters if any of its maps
3353 * involve the parameters. In other words, "umap" does not
3354 * involve any of the parameters if all its maps to not
3355 * involve the parameters.
3357 int isl_union_map_involves_dims(__isl_keep isl_union_map
*umap
,
3358 enum isl_dim_type type
, unsigned first
, unsigned n
)
3360 struct isl_union_map_involves_dims_data data
= { first
, n
};
3363 if (type
!= isl_dim_param
)
3364 isl_die(isl_union_map_get_ctx(umap
), isl_error_invalid
,
3365 "can only reference parameters", return 0);
3367 excludes
= union_map_forall_user(umap
, &map_excludes
, &data
);