Home | History | Annotate | Line # | Download | only in dist 
      1 /*
      2  * Copyright 2012      Ecole Normale Superieure
      3  * Copyright 2017      Sven Verdoolaege
      4  *
      5  * Use of this software is governed by the MIT license
      6  *
      7  * Written by Sven Verdoolaege,
      8  * Ecole Normale Superieure, 45 rue d'Ulm, 75230 Paris, France
      9  */
     10 
     11 #define xFN(TYPE,NAME) TYPE ## _ ## NAME
     12 #define FN(TYPE,NAME) xFN(TYPE,NAME)
     13 
     14 /* Drop the "n" domain dimensions starting at "first" from "obj",
     15  * after checking that they do not appear in the affine expression.
     16  */
     17 static __isl_give TYPE *FN(TYPE,drop_domain)(__isl_take TYPE *obj,
     18 	unsigned first, unsigned n)
     19 {
     20 	isl_bool involves;
     21 
     22 	involves = FN(TYPE,involves_dims)(obj, isl_dim_in, first, n);
     23 	if (involves < 0)
     24 		return FN(TYPE,free)(obj);
     25 	if (involves)
     26 		isl_die(FN(TYPE,get_ctx)(obj), isl_error_invalid,
     27 		    "affine expression involves some of the domain dimensions",
     28 		    return FN(TYPE,free)(obj));
     29 	return FN(TYPE,drop_dims)(obj, isl_dim_in, first, n);
     30 }
     31 
     32 /* Check that the domain of "obj" is a product.
     33  */
     34 static isl_stat FN(TYPE,check_domain_product)(__isl_keep TYPE *obj)
     35 {
     36 	isl_bool is_product;
     37 
     38 	is_product = FN(TYPE,domain_is_product)(obj);
     39 	if (is_product < 0)
     40 		return isl_stat_error;
     41 	if (!is_product)
     42 		isl_die(FN(TYPE,get_ctx)(obj), isl_error_invalid,
     43 			"domain is not a product", return isl_stat_error);
     44 	return isl_stat_ok;
     45 }
     46 
     47 /* Given an affine function with a domain of the form [A -> B] that
     48  * does not depend on B, return the same function on domain A.
     49  */
     50 __isl_give TYPE *FN(TYPE,domain_factor_domain)(__isl_take TYPE *obj)
     51 {
     52 	isl_space *space;
     53 	isl_size n, n_in;
     54 
     55 	if (FN(TYPE,check_domain_product)(obj) < 0)
     56 		return FN(TYPE,free)(obj);
     57 	space = FN(TYPE,get_domain_space)(obj);
     58 	n = isl_space_dim(space, isl_dim_set);
     59 	space = isl_space_factor_domain(space);
     60 	n_in = isl_space_dim(space, isl_dim_set);
     61 	if (n < 0 || n_in < 0)
     62 		obj = FN(TYPE,free)(obj);
     63 	else
     64 		obj = FN(TYPE,drop_domain)(obj, n_in, n - n_in);
     65 	obj = FN(TYPE,reset_domain_space)(obj, space);
     66 	return obj;
     67 }
     68