#include <isl_map.h>
#include <isl_set.h>
#include <isl_dim_private.h>
+#include <isl_map_private.h>
#include <isl_union_map_private.h>
#include <isl_union_set.h>
return umap ? umap->dim->ctx : NULL;
}
+isl_ctx *isl_union_set_get_ctx(__isl_keep isl_union_set *uset)
+{
+ return uset ? uset->dim->ctx : NULL;
+}
+
__isl_give isl_dim *isl_union_map_get_dim(__isl_keep isl_union_map *umap)
{
if (!umap)
return isl_union_map_dup(umap);
}
+struct isl_union_align {
+ isl_reordering *exp;
+ isl_union_map *res;
+};
+
+static int align_entry(void **entry, void *user)
+{
+ isl_map *map = *entry;
+ isl_reordering *exp;
+ struct isl_union_align *data = user;
+
+ exp = isl_reordering_extend_dim(isl_reordering_copy(data->exp),
+ isl_map_get_dim(map));
+
+ data->res = isl_union_map_add_map(data->res,
+ isl_map_realign(isl_map_copy(map), exp));
+
+ return 0;
+}
+
+/* Align the parameters of umap along those of model.
+ * The result has the parameters of model first, in the same order
+ * as they appear in model, followed by any remaining parameters of
+ * umap that do not appear in model.
+ */
+__isl_give isl_union_map *isl_union_map_align_params(
+ __isl_take isl_union_map *umap, __isl_take isl_dim *model)
+{
+ struct isl_union_align data = { NULL, NULL };
+
+ if (!umap || !model)
+ goto error;
+
+ if (isl_dim_match(umap->dim, isl_dim_param, model, isl_dim_param)) {
+ isl_dim_free(model);
+ return umap;
+ }
+
+ data.exp = isl_parameter_alignment_reordering(umap->dim, model);
+ if (!data.exp)
+ goto error;
+
+ data.res = isl_union_map_alloc(isl_dim_copy(data.exp->dim),
+ umap->table.n);
+ if (isl_hash_table_foreach(umap->dim->ctx, &umap->table,
+ &align_entry, &data) < 0)
+ goto error;
+
+ isl_reordering_free(data.exp);
+ isl_union_map_free(umap);
+ isl_dim_free(model);
+ return data.res;
+error:
+ isl_reordering_free(data.exp);
+ isl_union_map_free(umap);
+ isl_union_map_free(data.res);
+ isl_dim_free(model);
+ return NULL;
+}
+
+__isl_give isl_union_set *isl_union_set_align_params(
+ __isl_take isl_union_set *uset, __isl_take isl_dim *model)
+{
+ return isl_union_map_align_params(uset, model);
+}
+
__isl_give isl_union_map *isl_union_map_union(__isl_take isl_union_map *umap1,
__isl_take isl_union_map *umap2)
{
+ umap1 = isl_union_map_align_params(umap1, isl_union_map_get_dim(umap2));
+ umap2 = isl_union_map_align_params(umap2, isl_union_map_get_dim(umap1));
+
umap1 = isl_union_map_cow(umap1);
if (!umap1 || !umap2)
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)
{
&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)
{
static __isl_give isl_union_map *gen_bin_op(__isl_take isl_union_map *umap1,
__isl_take isl_union_map *umap2, int (*fn)(void **, void *))
{
- struct isl_union_map_gen_bin_data data = { umap2, NULL };
+ struct isl_union_map_gen_bin_data data = { NULL, NULL };
+
+ umap1 = isl_union_map_align_params(umap1, isl_union_map_get_dim(umap2));
+ umap2 = isl_union_map_align_params(umap2, isl_union_map_get_dim(umap1));
if (!umap1 || !umap2)
goto error;
+ data.umap2 = umap2;
data.res = isl_union_map_alloc(isl_dim_copy(umap1->dim),
umap1->table.n);
if (isl_hash_table_foreach(umap1->dim->ctx, &umap1->table,
__isl_take isl_union_map *umap2,
__isl_give isl_map *(*fn)(__isl_take isl_map*, __isl_take isl_map*))
{
- struct isl_union_map_match_bin_data data = { umap2, NULL, fn };
+ struct isl_union_map_match_bin_data data = { NULL, NULL, fn };
+
+ umap1 = isl_union_map_align_params(umap1, isl_union_map_get_dim(umap2));
+ umap2 = isl_union_map_align_params(umap2, isl_union_map_get_dim(umap1));
if (!umap1 || !umap2)
goto error;
+ data.umap2 = umap2;
data.res = isl_union_map_alloc(isl_dim_copy(umap1->dim),
umap1->table.n);
if (isl_hash_table_foreach(umap1->dim->ctx, &umap1->table,
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;
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;
static __isl_give isl_union_map *bin_op(__isl_take isl_union_map *umap1,
__isl_take isl_union_map *umap2, int (*fn)(void **entry, void *user))
{
- struct isl_union_map_bin_data data = { umap2, NULL, NULL, fn };
+ struct isl_union_map_bin_data data = { NULL, NULL, NULL, fn };
+
+ umap1 = isl_union_map_align_params(umap1, isl_union_map_get_dim(umap2));
+ umap2 = isl_union_map_align_params(umap2, isl_union_map_get_dim(umap1));
if (!umap1 || !umap2)
goto error;
+ data.umap2 = umap2;
data.res = isl_union_map_alloc(isl_dim_copy(umap1->dim),
umap1->table.n);
if (isl_hash_table_foreach(umap1->dim->ctx, &umap1->table,
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;
isl_union_map_from_range(range));
}
-static int reverse_entry(void **entry, void *user)
-{
- isl_map **map = (isl_map **)entry;
-
- *map = isl_map_reverse(*map);
-
- return *map ? 0 : -1;
-}
-
static __isl_give isl_union_map *un_op(__isl_take isl_union_map *umap,
int (*fn)(void **, void *), int cow)
{
return NULL;
}
-__isl_give isl_union_map *isl_union_map_reverse(__isl_take isl_union_map *umap)
-{
- return un_op(umap, &reverse_entry, 1);
-}
-
static int affine_entry(void **entry, void *user)
{
isl_map **map = (isl_map **)entry;
return isl_union_map_lexmax(uset);
}
+static __isl_give isl_union_set *cond_un_op(__isl_take isl_union_map *umap,
+ int (*fn)(void **, void *))
+{
+ isl_union_set *res;
+
+ if (!umap)
+ return NULL;
+
+ res = isl_union_map_alloc(isl_dim_copy(umap->dim), umap->table.n);
+ if (isl_hash_table_foreach(umap->dim->ctx, &umap->table, fn, &res) < 0)
+ goto error;
+
+ isl_union_map_free(umap);
+ return res;
+error:
+ isl_union_map_free(umap);
+ isl_union_set_free(res);
+ return NULL;
+}
+
+static int reverse_entry(void **entry, void *user)
+{
+ isl_map *map = *entry;
+ isl_union_map **res = user;
+
+ *res = isl_union_map_add_map(*res, isl_map_reverse(isl_map_copy(map)));
+
+ return 0;
+}
+
+__isl_give isl_union_map *isl_union_map_reverse(__isl_take isl_union_map *umap)
+{
+ return cond_un_op(umap, &reverse_entry);
+}
+
static int domain_entry(void **entry, void *user)
{
- *entry = isl_map_domain(*entry);
+ isl_map *map = *entry;
+ isl_union_set **res = user;
- return *entry ? 0 : -1;
+ *res = isl_union_set_add_set(*res, isl_map_domain(isl_map_copy(map)));
+
+ return 0;
}
__isl_give isl_union_set *isl_union_map_domain(__isl_take isl_union_map *umap)
{
- return un_op(umap, &domain_entry, 1);
+ return cond_un_op(umap, &domain_entry);
}
static int range_entry(void **entry, void *user)
{
- *entry = isl_map_range(*entry);
+ isl_map *map = *entry;
+ isl_union_set **res = user;
- return *entry ? 0 : -1;
+ *res = isl_union_set_add_set(*res, isl_map_range(isl_map_copy(map)));
+
+ return 0;
}
__isl_give isl_union_set *isl_union_map_range(__isl_take isl_union_map *umap)
{
- return un_op(umap, &range_entry, 1);
+ return cond_un_op(umap, &range_entry);
}
-static __isl_give isl_union_set *cond_un_op(__isl_take isl_union_map *umap,
- int (*fn)(void **, void *))
+static int domain_map_entry(void **entry, void *user)
{
- isl_union_set *res;
+ isl_map *map = *entry;
+ isl_union_set **res = user;
- if (!umap)
- return NULL;
+ *res = isl_union_map_add_map(*res,
+ isl_map_domain_map(isl_map_copy(map)));
- res = isl_union_map_alloc(isl_dim_copy(umap->dim), umap->table.n);
- if (isl_hash_table_foreach(umap->dim->ctx, &umap->table, fn, &res) < 0)
- goto error;
+ return 0;
+}
- isl_union_map_free(umap);
- return res;
-error:
- isl_union_map_free(umap);
- isl_union_set_free(res);
- return NULL;
+__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)
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;
static int wrap_entry(void **entry, void *user)
{
- isl_map **map = (isl_map **)entry;
+ isl_map *map = *entry;
+ isl_union_set **res = user;
- *map = (isl_map *)isl_map_wrap(*map);
+ *res = isl_union_set_add_set(*res, isl_map_wrap(isl_map_copy(map)));
- return *map ? 0 : -1;
+ return 0;
}
__isl_give isl_union_set *isl_union_map_wrap(__isl_take isl_union_map *umap)
{
- return un_op(umap, &wrap_entry, 1);
+ return cond_un_op(umap, &wrap_entry);
}
struct isl_union_map_is_subset_data {
int isl_union_map_is_subset(__isl_keep isl_union_map *umap1,
__isl_keep isl_union_map *umap2)
{
- struct isl_union_map_is_subset_data data = { umap2, 1 };
+ struct isl_union_map_is_subset_data data = { NULL, 1 };
+
+ umap1 = isl_union_map_copy(umap1);
+ umap2 = isl_union_map_copy(umap2);
+ umap1 = isl_union_map_align_params(umap1, isl_union_map_get_dim(umap2));
+ umap2 = isl_union_map_align_params(umap2, isl_union_map_get_dim(umap1));
if (!umap1 || !umap2)
- return -1;
+ goto error;
+ data.umap2 = umap2;
if (isl_hash_table_foreach(umap1->dim->ctx, &umap1->table,
&is_subset_entry, &data) < 0 &&
data.is_subset)
- return -1;
+ goto error;
+
+ isl_union_map_free(umap1);
+ isl_union_map_free(umap2);
return data.is_subset;
+error:
+ isl_union_map_free(umap1);
+ isl_union_map_free(umap2);
+ return -1;
}
int isl_union_map_is_equal(__isl_keep isl_union_map *umap1,
return -1;
if (isl_hash_table_foreach(umap->dim->ctx, &umap->table,
- &sample_entry, &empty) < 0 && empty)
+ &empty_entry, &empty) < 0 && empty)
return -1;
return empty;