135 lines · c
1/*2 * Copyright 2017 Sven Verdoolaege3 *4 * Use of this software is governed by the MIT license5 *6 * Written by Sven Verdoolaege.7 */8 9/* Initialize the explicit domain of "mpa".10 *11 * The explicit domain is initialized to a universe set12 * in the domain space.13 */14static __isl_give isl_multi_pw_aff *isl_multi_pw_aff_init_explicit_domain(15 __isl_take isl_multi_pw_aff *mpa)16{17 if (isl_multi_pw_aff_check_has_explicit_domain(mpa) < 0)18 return isl_multi_pw_aff_free(mpa);19 mpa->u.dom = isl_set_universe(isl_multi_pw_aff_get_domain_space(mpa));20 if (!mpa->u.dom)21 return isl_multi_pw_aff_free(mpa);22 return mpa;23}24 25/* Intersect the domain of "dst" with the domain product26 * of the explicit domains of "src1" and "src2".27 * This function is only called if at least one of "src1" or "src2"28 * has an explicit domain.29 */30static __isl_give isl_multi_pw_aff *31isl_multi_pw_aff_intersect_explicit_domain_product(32 __isl_take isl_multi_pw_aff *dst, __isl_keep isl_multi_pw_aff *src1,33 __isl_keep isl_multi_pw_aff *src2)34{35 isl_space *space;36 isl_set *dom;37 isl_map *map;38 39 if (!src1 || !src2)40 return FN(isl_multi_pw_aff,free)(dst);41 space = isl_multi_pw_aff_get_domain_space(dst);42 dom = isl_set_universe(space);43 map = isl_set_unwrap(dom);44 if (isl_multi_pw_aff_has_explicit_domain(src1)) {45 dom = isl_set_copy(src1->u.dom);46 map = isl_map_intersect_domain(map, dom);47 }48 if (isl_multi_pw_aff_has_explicit_domain(src2)) {49 dom = isl_set_copy(src2->u.dom);50 map = isl_map_intersect_range(map, dom);51 }52 dom = isl_map_wrap(map);53 dst = isl_multi_pw_aff_intersect_domain(dst, dom);54 return dst;55}56 57/* Check whether the explicit domain of "mpa" has non-zero coefficients58 * for any dimension in the given range or if any of these dimensions appear59 * with non-zero coefficients in any of the integer divisions involved.60 */61isl_bool isl_multi_pw_aff_involves_explicit_domain_dims(62 __isl_keep isl_multi_pw_aff *mpa,63 enum isl_dim_type type, unsigned pos, unsigned n)64{65 if (isl_multi_pw_aff_check_has_explicit_domain(mpa) < 0)66 return isl_bool_error;67 if (type == isl_dim_in)68 type = isl_dim_set;69 return isl_set_involves_dims(mpa->u.dom, type, pos, n);70}71 72/* Insert "n" dimensions of type "type" at position "pos"73 * of the explicit domain of "mpa".74 */75static __isl_give isl_multi_pw_aff *76isl_multi_pw_aff_insert_explicit_domain_dims(__isl_take isl_multi_pw_aff *mpa,77 enum isl_dim_type type, unsigned pos, unsigned n)78{79 if (isl_multi_pw_aff_check_has_explicit_domain(mpa) < 0)80 return isl_multi_pw_aff_free(mpa);81 mpa = isl_multi_pw_aff_cow(mpa);82 if (!mpa)83 return NULL;84 if (type == isl_dim_in)85 type = isl_dim_set;86 mpa->u.dom = isl_set_insert_dims(mpa->u.dom, type, pos, n);87 if (!mpa->u.dom)88 return isl_multi_pw_aff_free(mpa);89 return mpa;90}91 92/* Drop the "n" dimensions of type "type" starting at position "pos"93 * of the explicit domain of "mpa".94 */95static __isl_give isl_multi_pw_aff *96isl_multi_pw_aff_drop_explicit_domain_dims(__isl_take isl_multi_pw_aff *mpa,97 enum isl_dim_type type, unsigned pos, unsigned n)98{99 if (isl_multi_pw_aff_check_has_explicit_domain(mpa) < 0)100 return isl_multi_pw_aff_free(mpa);101 mpa = isl_multi_pw_aff_cow(mpa);102 if (!mpa)103 return NULL;104 if (type == isl_dim_in)105 type = isl_dim_set;106 mpa->u.dom = isl_set_drop(mpa->u.dom, type, pos, n);107 if (!mpa->u.dom)108 return isl_multi_pw_aff_free(mpa);109 return mpa;110}111 112/* Move the "n" dimensions of "src_type" starting at "src_pos" of113 * of the explicit domain of "mpa" to dimensions of "dst_type" at "dst_pos".114 */115static __isl_give isl_multi_pw_aff *isl_multi_pw_aff_move_explicit_domain_dims(116 __isl_take isl_multi_pw_aff *mpa,117 enum isl_dim_type dst_type, unsigned dst_pos,118 enum isl_dim_type src_type, unsigned src_pos, unsigned n)119{120 if (isl_multi_pw_aff_check_has_explicit_domain(mpa) < 0)121 return isl_multi_pw_aff_free(mpa);122 mpa = isl_multi_pw_aff_cow(mpa);123 if (!mpa)124 return NULL;125 if (dst_type == isl_dim_in)126 dst_type = isl_dim_set;127 if (src_type == isl_dim_in)128 src_type = isl_dim_set;129 mpa->u.dom = isl_set_move_dims(mpa->u.dom, dst_type, dst_pos,130 src_type, src_pos, n);131 if (!mpa->u.dom)132 return isl_multi_pw_aff_free(mpa);133 return mpa;134}135