Home | History | Annotate | Line # | Download | only in dist 
      1 /*
      2  * Copyright 2012      Ecole Normale Superieure
      3  *
      4  * Use of this software is governed by the MIT license
      5  *
      6  * Written by Sven Verdoolaege,
      7  * Ecole Normale Superieure, 45 rue d'Ulm, 75230 Paris, France
      8  */
      9 
     10 #include <isl_pw_macro.h>
     11 
     12 #undef SUFFIX
     13 #define SUFFIX	multi_aff
     14 #undef ARG1
     15 #define ARG1	PW
     16 #undef ARG2
     17 #define ARG2	isl_multi_aff
     18 
     19 static
     20 #include "isl_align_params_templ.c"
     21 
     22 #undef SUFFIX
     23 #define SUFFIX	pw_multi_aff
     24 #undef ARG1
     25 #define ARG1	PW
     26 #undef ARG2
     27 #define ARG2	isl_pw_multi_aff
     28 
     29 static
     30 #include "isl_align_params_templ.c"
     31 
     32 /* Compute the pullback of "pw" by the function represented by "ma".
     33  * In other words, plug in "ma" in "pw".
     34  */
     35 __isl_give PW *FN(PW,pullback_multi_aff)(__isl_take PW *pw,
     36 	__isl_take isl_multi_aff *ma)
     37 {
     38 	int i;
     39 	isl_size n;
     40 	isl_space *space = NULL;
     41 
     42 	FN(PW,align_params_multi_aff)(&pw, &ma);
     43 	ma = isl_multi_aff_align_divs(ma);
     44 	n = FN(PW,n_piece)(pw);
     45 	if (n < 0 || !ma)
     46 		goto error;
     47 
     48 	space = isl_space_join(isl_multi_aff_get_space(ma),
     49 				FN(PW,get_space)(pw));
     50 
     51 	for (i = 0; i < n; ++i) {
     52 		isl_set *domain;
     53 		EL *el;
     54 
     55 		domain = FN(PW,take_domain_at)(pw, i);
     56 		domain = isl_set_preimage_multi_aff(domain,
     57 						    isl_multi_aff_copy(ma));
     58 		pw = FN(PW,restore_domain_at)(pw, i, domain);
     59 		el = FN(PW,take_base_at)(pw, i);
     60 		el = FN(EL,pullback_multi_aff)(el, isl_multi_aff_copy(ma));
     61 		pw = FN(PW,restore_base_at)(pw, i, el);
     62 	}
     63 
     64 	pw = FN(PW,reset_space)(pw, space);
     65 	isl_multi_aff_free(ma);
     66 	return pw;
     67 error:
     68 	isl_space_free(space);
     69 	isl_multi_aff_free(ma);
     70 	FN(PW,free)(pw);
     71 	return NULL;
     72 }
     73 
     74 /* Compute the pullback of "pw" by the function represented by "pma".
     75  * In other words, plug in "pma" in "pw".
     76  */
     77 static __isl_give PW *FN(PW,pullback_pw_multi_aff_aligned)(__isl_take PW *pw,
     78 	__isl_take isl_pw_multi_aff *pma)
     79 {
     80 	int i;
     81 	PW *res;
     82 
     83 	if (!pma)
     84 		goto error;
     85 
     86 	if (pma->n == 0) {
     87 		isl_space *space;
     88 		space = isl_space_join(isl_pw_multi_aff_get_space(pma),
     89 					FN(PW,get_space)(pw));
     90 		isl_pw_multi_aff_free(pma);
     91 		res = FN(PW,empty)(space);
     92 		FN(PW,free)(pw);
     93 		return res;
     94 	}
     95 
     96 	res = FN(PW,pullback_multi_aff)(FN(PW,copy)(pw),
     97 					isl_multi_aff_copy(pma->p[0].maff));
     98 	res = FN(PW,intersect_domain)(res, isl_set_copy(pma->p[0].set));
     99 
    100 	for (i = 1; i < pma->n; ++i) {
    101 		PW *res_i;
    102 
    103 		res_i = FN(PW,pullback_multi_aff)(FN(PW,copy)(pw),
    104 					isl_multi_aff_copy(pma->p[i].maff));
    105 		res_i = FN(PW,intersect_domain)(res_i,
    106 					isl_set_copy(pma->p[i].set));
    107 		res = FN(PW,add_disjoint)(res, res_i);
    108 	}
    109 
    110 	isl_pw_multi_aff_free(pma);
    111 	FN(PW,free)(pw);
    112 	return res;
    113 error:
    114 	isl_pw_multi_aff_free(pma);
    115 	FN(PW,free)(pw);
    116 	return NULL;
    117 }
    118 
    119 __isl_give PW *FN(PW,pullback_pw_multi_aff)(__isl_take PW *pw,
    120 	__isl_take isl_pw_multi_aff *pma)
    121 {
    122 	FN(PW,align_params_pw_multi_aff)(&pw, &pma);
    123 	return FN(PW,pullback_pw_multi_aff_aligned)(pw, pma);
    124 }
    125