Home | History | Annotate | Line # | Download | only in dist 
      1 /*
      2  * Copyright 2014      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_multi_macro.h>
     11 
     12 /* Data structure that specifies how isl_multi_*_un_op should
     13  * modify its input.
     14  *
     15  * If "fn_space" is set, then it is applied to the space.
     16  *
     17  * "fn_el" is applied to each base expression.
     18  */
     19 S(MULTI(BASE),un_op_control) {
     20 	__isl_give isl_space *(*fn_space)(__isl_take isl_space *space);
     21 	__isl_give EL *(*fn_el)(__isl_take EL *el);
     22 };
     23 
     24 /* Modify "multi" based on "control".
     25  */
     26 static __isl_give MULTI(BASE) *FN(MULTI(BASE),un_op)(
     27 	__isl_take MULTI(BASE) *multi, S(MULTI(BASE),un_op_control) *control)
     28 {
     29 	int i;
     30 	isl_size n;
     31 	isl_space *space;
     32 
     33 	n = FN(MULTI(BASE),size)(multi);
     34 	if (n < 0)
     35 		return FN(MULTI(BASE),free)(multi);
     36 
     37 	for (i = 0; i < n; ++i) {
     38 		EL *el;
     39 
     40 		el = FN(MULTI(BASE),take_at)(multi, i);
     41 		el = control->fn_el(el);
     42 		multi = FN(MULTI(BASE),restore_at)(multi, i, el);
     43 	}
     44 
     45 	if (!control->fn_space)
     46 		return multi;
     47 
     48 	space = FN(MULTI(BASE),take_space)(multi);
     49 	space = control->fn_space(space);
     50 	multi = FN(MULTI(BASE),restore_space)(multi, space);
     51 
     52 	return multi;
     53 }
     54