doc: fix typo
[platform/upstream/isl.git] / isl_union_map.c
index f97063b..5472261 100644 (file)
@@ -8,14 +8,14 @@
  * 91893 Orsay, France 
  */
 
-#include <isl_ctx.h>
-#include <isl_hash.h>
-#include <isl_map.h>
-#include <isl_set.h>
-#include <isl_dim_private.h>
 #include <isl_map_private.h>
+#include <isl/ctx.h>
+#include <isl/hash.h>
+#include <isl/map.h>
+#include <isl/set.h>
+#include <isl_dim_private.h>
 #include <isl_union_map_private.h>
-#include <isl_union_set.h>
+#include <isl/union_set.h>
 
 static __isl_give isl_union_map *isl_union_map_alloc(__isl_take isl_dim *dim,
        int size)
@@ -337,6 +337,16 @@ static int call_on_copy(void **entry, void *user)
        return data->fn(isl_map_copy(map), data->user);
 }
 
+int isl_union_map_n_map(__isl_keep isl_union_map *umap)
+{
+       return umap ? umap->table.n : 0;
+}
+
+int isl_union_set_n_set(__isl_keep isl_union_set *uset)
+{
+       return uset ? uset->table.n : 0;
+}
+
 int isl_union_map_foreach_map(__isl_keep isl_union_map *umap,
        int (*fn)(__isl_take isl_map *map, void *user), void *user)
 {
@@ -349,6 +359,33 @@ int isl_union_map_foreach_map(__isl_keep isl_union_map *umap,
                                      &call_on_copy, &data);
 }
 
+__isl_give isl_map *isl_union_map_extract_map(__isl_keep isl_union_map *umap,
+       __isl_take isl_dim *dim)
+{
+       uint32_t hash;
+       struct isl_hash_table_entry *entry;
+
+       if (!umap || !dim)
+               goto error;
+
+       hash = isl_dim_get_hash(dim);
+       entry = isl_hash_table_find(umap->dim->ctx, &umap->table, hash,
+                                   &has_dim, dim, 0);
+       if (!entry)
+               return isl_map_empty(dim);
+       isl_dim_free(dim);
+       return isl_map_copy(entry->data);
+error:
+       isl_dim_free(dim);
+       return NULL;
+}
+
+__isl_give isl_set *isl_union_set_extract_set(__isl_keep isl_union_set *uset,
+       __isl_take isl_dim *dim)
+{
+       return (isl_set *)isl_union_map_extract_map(uset, dim);
+}
+
 int isl_union_set_foreach_set(__isl_keep isl_union_set *uset,
        int (*fn)(__isl_take isl_set *set, void *user), void *user)
 {
@@ -545,6 +582,54 @@ __isl_give isl_union_set *isl_union_set_gist(__isl_take isl_union_set *uset,
        return isl_union_map_gist(uset, context);
 }
 
+static __isl_give isl_map *lex_le_set(__isl_take isl_map *set1,
+       __isl_take isl_map *set2)
+{
+       return isl_set_lex_le_set((isl_set *)set1, (isl_set *)set2);
+}
+
+static __isl_give isl_map *lex_lt_set(__isl_take isl_map *set1,
+       __isl_take isl_map *set2)
+{
+       return isl_set_lex_lt_set((isl_set *)set1, (isl_set *)set2);
+}
+
+__isl_give isl_union_map *isl_union_set_lex_lt_union_set(
+       __isl_take isl_union_set *uset1, __isl_take isl_union_set *uset2)
+{
+       return match_bin_op(uset1, uset2, &lex_lt_set);
+}
+
+__isl_give isl_union_map *isl_union_set_lex_le_union_set(
+       __isl_take isl_union_set *uset1, __isl_take isl_union_set *uset2)
+{
+       return match_bin_op(uset1, uset2, &lex_le_set);
+}
+
+__isl_give isl_union_map *isl_union_set_lex_gt_union_set(
+       __isl_take isl_union_set *uset1, __isl_take isl_union_set *uset2)
+{
+       return isl_union_map_reverse(isl_union_set_lex_lt_union_set(uset2, uset1));
+}
+
+__isl_give isl_union_map *isl_union_set_lex_ge_union_set(
+       __isl_take isl_union_set *uset1, __isl_take isl_union_set *uset2)
+{
+       return isl_union_map_reverse(isl_union_set_lex_le_union_set(uset2, uset1));
+}
+
+__isl_give isl_union_map *isl_union_map_lex_gt_union_map(
+       __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
+{
+       return isl_union_map_reverse(isl_union_map_lex_lt_union_map(umap2, umap1));
+}
+
+__isl_give isl_union_map *isl_union_map_lex_ge_union_map(
+       __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
+{
+       return isl_union_map_reverse(isl_union_map_lex_le_union_map(umap2, umap1));
+}
+
 static int intersect_domain_entry(void **entry, void *user)
 {
        struct isl_union_map_gen_bin_data *data = user;
@@ -587,6 +672,48 @@ __isl_give isl_union_map *isl_union_map_intersect_domain(
        return gen_bin_op(umap, uset, &intersect_domain_entry);
 }
 
+static int intersect_range_entry(void **entry, void *user)
+{
+       struct isl_union_map_gen_bin_data *data = user;
+       uint32_t hash;
+       struct isl_hash_table_entry *entry2;
+       isl_dim *dim;
+       isl_map *map = *entry;
+       int empty;
+
+       dim = isl_map_get_dim(map);
+       dim = isl_dim_range(dim);
+       hash = isl_dim_get_hash(dim);
+       entry2 = isl_hash_table_find(data->umap2->dim->ctx, &data->umap2->table,
+                                    hash, &has_dim, dim, 0);
+       isl_dim_free(dim);
+       if (!entry2)
+               return 0;
+
+       map = isl_map_copy(map);
+       map = isl_map_intersect_range(map, isl_set_copy(entry2->data));
+
+       empty = isl_map_is_empty(map);
+       if (empty < 0) {
+               isl_map_free(map);
+               return -1;
+       }
+       if (empty) {
+               isl_map_free(map);
+               return 0;
+       }
+
+       data->res = isl_union_map_add_map(data->res, map);
+
+       return 0;
+}
+
+__isl_give isl_union_map *isl_union_map_intersect_range(
+       __isl_take isl_union_map *umap, __isl_take isl_union_set *uset)
+{
+       return gen_bin_op(umap, uset, &intersect_range_entry);
+}
+
 struct isl_union_map_bin_data {
        isl_union_map *umap2;
        isl_union_map *res;
@@ -668,12 +795,64 @@ __isl_give isl_union_map *isl_union_map_apply_range(
        return bin_op(umap1, umap2, &apply_range_entry);
 }
 
+__isl_give isl_union_map *isl_union_map_apply_domain(
+       __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
+{
+       umap1 = isl_union_map_reverse(umap1);
+       umap1 = isl_union_map_apply_range(umap1, umap2);
+       return isl_union_map_reverse(umap1);
+}
+
 __isl_give isl_union_set *isl_union_set_apply(
        __isl_take isl_union_set *uset, __isl_take isl_union_map *umap)
 {
        return isl_union_map_apply_range(uset, umap);
 }
 
+static int map_lex_lt_entry(void **entry, void *user)
+{
+       struct isl_union_map_bin_data *data = user;
+       isl_map *map2 = *entry;
+
+       if (!isl_dim_tuple_match(data->map->dim, isl_dim_out,
+                                map2->dim, isl_dim_out))
+               return 0;
+
+       map2 = isl_map_lex_lt_map(isl_map_copy(data->map), isl_map_copy(map2));
+
+       data->res = isl_union_map_add_map(data->res, map2);
+
+       return 0;
+}
+
+__isl_give isl_union_map *isl_union_map_lex_lt_union_map(
+       __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
+{
+       return bin_op(umap1, umap2, &map_lex_lt_entry);
+}
+
+static int map_lex_le_entry(void **entry, void *user)
+{
+       struct isl_union_map_bin_data *data = user;
+       isl_map *map2 = *entry;
+
+       if (!isl_dim_tuple_match(data->map->dim, isl_dim_out,
+                                map2->dim, isl_dim_out))
+               return 0;
+
+       map2 = isl_map_lex_le_map(isl_map_copy(data->map), isl_map_copy(map2));
+
+       data->res = isl_union_map_add_map(data->res, map2);
+
+       return 0;
+}
+
+__isl_give isl_union_map *isl_union_map_lex_le_union_map(
+       __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
+{
+       return bin_op(umap1, umap2, &map_lex_le_entry);
+}
+
 static int product_entry(void **entry, void *user)
 {
        struct isl_union_map_bin_data *data = user;
@@ -698,6 +877,25 @@ __isl_give isl_union_set *isl_union_set_product(__isl_take isl_union_set *uset1,
        return isl_union_map_product(uset1, uset2);
 }
 
+static int range_product_entry(void **entry, void *user)
+{
+       struct isl_union_map_bin_data *data = user;
+       isl_map *map2 = *entry;
+
+       map2 = isl_map_range_product(isl_map_copy(data->map),
+                                    isl_map_copy(map2));
+
+       data->res = isl_union_map_add_map(data->res, map2);
+
+       return 0;
+}
+
+__isl_give isl_union_map *isl_union_map_range_product(
+       __isl_take isl_union_map *umap1, __isl_take isl_union_map *umap2)
+{
+       return bin_op(umap1, umap2, &range_product_entry);
+}
+
 __isl_give isl_union_map *isl_union_map_from_range(
        __isl_take isl_union_set *uset)
 {
@@ -718,10 +916,9 @@ __isl_give isl_union_map *isl_union_map_from_domain_and_range(
 }
 
 static __isl_give isl_union_map *un_op(__isl_take isl_union_map *umap,
-       int (*fn)(void **, void *), int cow)
+       int (*fn)(void **, void *))
 {
-       if (cow)
-               umap = isl_union_map_cow(umap);
+       umap = isl_union_map_cow(umap);
        if (!umap)
                return NULL;
 
@@ -746,7 +943,7 @@ static int affine_entry(void **entry, void *user)
 __isl_give isl_union_map *isl_union_map_affine_hull(
        __isl_take isl_union_map *umap)
 {
-       return un_op(umap, &affine_entry, 1);
+       return un_op(umap, &affine_entry);
 }
 
 __isl_give isl_union_set *isl_union_set_affine_hull(
@@ -755,40 +952,109 @@ __isl_give isl_union_set *isl_union_set_affine_hull(
        return isl_union_map_affine_hull(uset);
 }
 
-static int coalesce_entry(void **entry, void *user)
+static int polyhedral_entry(void **entry, void *user)
 {
        isl_map **map = (isl_map **)entry;
 
-       *map = isl_map_coalesce(*map);
+       *map = isl_map_from_basic_map(isl_map_polyhedral_hull(*map));
 
        return *map ? 0 : -1;
 }
 
-__isl_give isl_union_map *isl_union_map_coalesce(
+__isl_give isl_union_map *isl_union_map_polyhedral_hull(
        __isl_take isl_union_map *umap)
 {
-       return un_op(umap, &coalesce_entry, 0);
+       return un_op(umap, &polyhedral_entry);
 }
 
-__isl_give isl_union_set *isl_union_set_coalesce(
+__isl_give isl_union_set *isl_union_set_polyhedral_hull(
        __isl_take isl_union_set *uset)
 {
-       return isl_union_map_coalesce(uset);
+       return isl_union_map_polyhedral_hull(uset);
 }
 
-static int compute_divs_entry(void **entry, void *user)
+static int simple_entry(void **entry, void *user)
 {
        isl_map **map = (isl_map **)entry;
 
-       *map = isl_map_compute_divs(*map);
+       *map = isl_map_from_basic_map(isl_map_simple_hull(*map));
 
        return *map ? 0 : -1;
 }
 
+__isl_give isl_union_map *isl_union_map_simple_hull(
+       __isl_take isl_union_map *umap)
+{
+       return un_op(umap, &simple_entry);
+}
+
+__isl_give isl_union_set *isl_union_set_simple_hull(
+       __isl_take isl_union_set *uset)
+{
+       return isl_union_map_simple_hull(uset);
+}
+
+static int inplace_entry(void **entry, void *user)
+{
+       __isl_give isl_map *(*fn)(__isl_take isl_map *);
+       isl_map **map = (isl_map **)entry;
+       isl_map *copy;
+
+       fn = *(__isl_give isl_map *(**)(__isl_take isl_map *)) user;
+       copy = fn(isl_map_copy(*map));
+       if (!copy)
+               return -1;
+
+       isl_map_free(*map);
+       *map = copy;
+
+       return 0;
+}
+
+static __isl_give isl_union_map *inplace(__isl_take isl_union_map *umap,
+       __isl_give isl_map *(*fn)(__isl_take isl_map *))
+{
+       if (!umap)
+               return NULL;
+
+       if (isl_hash_table_foreach(umap->dim->ctx, &umap->table,
+                                   &inplace_entry, &fn) < 0)
+               goto error;
+
+       return umap;
+error:
+       isl_union_map_free(umap);
+       return NULL;
+}
+
+__isl_give isl_union_map *isl_union_map_coalesce(
+       __isl_take isl_union_map *umap)
+{
+       return inplace(umap, &isl_map_coalesce);
+}
+
+__isl_give isl_union_set *isl_union_set_coalesce(
+       __isl_take isl_union_set *uset)
+{
+       return isl_union_map_coalesce(uset);
+}
+
+__isl_give isl_union_map *isl_union_map_detect_equalities(
+       __isl_take isl_union_map *umap)
+{
+       return inplace(umap, &isl_map_detect_equalities);
+}
+
+__isl_give isl_union_set *isl_union_set_detect_equalities(
+       __isl_take isl_union_set *uset)
+{
+       return isl_union_map_detect_equalities(uset);
+}
+
 __isl_give isl_union_map *isl_union_map_compute_divs(
        __isl_take isl_union_map *umap)
 {
-       return un_op(umap, &compute_divs_entry, 0);
+       return inplace(umap, &isl_map_compute_divs);
 }
 
 __isl_give isl_union_set *isl_union_set_compute_divs(
@@ -809,7 +1075,7 @@ static int lexmin_entry(void **entry, void *user)
 __isl_give isl_union_map *isl_union_map_lexmin(
        __isl_take isl_union_map *umap)
 {
-       return un_op(umap, &lexmin_entry, 1);
+       return un_op(umap, &lexmin_entry);
 }
 
 __isl_give isl_union_set *isl_union_set_lexmin(
@@ -830,7 +1096,7 @@ static int lexmax_entry(void **entry, void *user)
 __isl_give isl_union_map *isl_union_map_lexmax(
        __isl_take isl_union_map *umap)
 {
-       return un_op(umap, &lexmax_entry, 1);
+       return un_op(umap, &lexmax_entry);
 }
 
 __isl_give isl_union_set *isl_union_set_lexmax(
@@ -904,6 +1170,40 @@ __isl_give isl_union_set *isl_union_map_range(__isl_take isl_union_map *umap)
        return cond_un_op(umap, &range_entry);
 }
 
+static int domain_map_entry(void **entry, void *user)
+{
+       isl_map *map = *entry;
+       isl_union_set **res = user;
+
+       *res = isl_union_map_add_map(*res,
+                                       isl_map_domain_map(isl_map_copy(map)));
+
+       return 0;
+}
+
+__isl_give isl_union_map *isl_union_map_domain_map(
+       __isl_take isl_union_map *umap)
+{
+       return cond_un_op(umap, &domain_map_entry);
+}
+
+static int range_map_entry(void **entry, void *user)
+{
+       isl_map *map = *entry;
+       isl_union_set **res = user;
+
+       *res = isl_union_map_add_map(*res,
+                                       isl_map_range_map(isl_map_copy(map)));
+
+       return 0;
+}
+
+__isl_give isl_union_map *isl_union_map_range_map(
+       __isl_take isl_union_map *umap)
+{
+       return cond_un_op(umap, &range_map_entry);
+}
+
 static int deltas_entry(void **entry, void *user)
 {
        isl_map *map = *entry;
@@ -922,6 +1222,21 @@ __isl_give isl_union_set *isl_union_map_deltas(__isl_take isl_union_map *umap)
        return cond_un_op(umap, &deltas_entry);
 }
 
+static int identity_entry(void **entry, void *user)
+{
+       isl_set *set = *entry;
+       isl_union_map **res = user;
+
+       *res = isl_union_map_add_map(*res, isl_set_identity(isl_set_copy(set)));
+
+       return 0;
+}
+
+__isl_give isl_union_map *isl_union_set_identity(__isl_take isl_union_set *uset)
+{
+       return cond_un_op(uset, &identity_entry);
+}
+
 static int unwrap_entry(void **entry, void *user)
 {
        isl_set *set = *entry;
@@ -1011,6 +1326,12 @@ error:
        return -1;
 }
 
+int isl_union_set_is_subset(__isl_keep isl_union_set *uset1,
+       __isl_keep isl_union_set *uset2)
+{
+       return isl_union_map_is_subset(uset1, uset2);
+}
+
 int isl_union_map_is_equal(__isl_keep isl_union_map *umap1,
        __isl_keep isl_union_map *umap2)
 {
@@ -1025,6 +1346,12 @@ int isl_union_map_is_equal(__isl_keep isl_union_map *umap1,
        return is_subset;
 }
 
+int isl_union_set_is_equal(__isl_keep isl_union_set *uset1,
+       __isl_keep isl_union_set *uset2)
+{
+       return isl_union_map_is_equal(uset1, uset2);
+}
+
 int isl_union_map_is_strict_subset(__isl_keep isl_union_map *umap1,
        __isl_keep isl_union_map *umap2)
 {
@@ -1041,6 +1368,12 @@ int isl_union_map_is_strict_subset(__isl_keep isl_union_map *umap1,
        return !is_subset;
 }
 
+int isl_union_set_is_strict_subset(__isl_keep isl_union_set *uset1,
+       __isl_keep isl_union_set *uset2)
+{
+       return isl_union_map_is_strict_subset(uset1, uset2);
+}
+
 static int sample_entry(void **entry, void *user)
 {
        isl_basic_map **sample = (isl_basic_map **)user;
@@ -1066,6 +1399,9 @@ __isl_give isl_basic_map *isl_union_map_sample(__isl_take isl_union_map *umap)
            !sample)
                goto error;
 
+       if (!sample)
+               sample = isl_basic_map_empty(isl_union_map_get_dim(umap));
+
        isl_union_map_free(umap);
 
        return sample;