isl_pw_lift_templ.c

   1 /*
   2  * Copyright 2010      INRIA Saclay
   3  *
   4  * Use of this software is governed by the MIT license
   5  *
   6  * Written by Sven Verdoolaege, INRIA Saclay - Ile-de-France,
   7  * Parc Club Orsay Universite, ZAC des vignes, 4 rue Jacques Monod,
   8  * 91893 Orsay, France
   9  */
  10
  11 #include <isl_pw_macro.h>
  12
  13 static isl_bool any_divs(__isl_keep isl_set *set)
  14 {
  15         int i;
  16
  17         if (!set)
  18                 return isl_bool_error;
  19
  20         for (i = 0; i < set->n; ++i)
  21                 if (set->p[i]->n_div > 0)
  22                         return isl_bool_true;
  23
  24         return isl_bool_false;
  25 }
  26
  27 static isl_stat foreach_lifted_subset(__isl_take isl_set *set,
  28         __isl_take EL *el,
  29         isl_stat (*fn)(__isl_take isl_set *set, __isl_take EL *el,
  30                 void *user), void *user)
  31 {
  32         int i;
  33
  34         if (!set || !el)
  35                 goto error;
  36
  37         for (i = 0; i < set->n; ++i) {
  38                 isl_set *lift;
  39                 EL *copy;
  40
  41                 lift = isl_set_from_basic_set(isl_basic_set_copy(set->p[i]));
  42                 lift = isl_set_lift(lift);
  43
  44                 copy = FN(EL,copy)(el);
  45                 copy = FN(EL,lift)(copy, isl_set_get_space(lift));
  46
  47                 if (fn(lift, copy, user) < 0)
  48                         goto error;
  49         }
  50
  51         isl_set_free(set);
  52         FN(EL,free)(el);
  53
  54         return isl_stat_ok;
  55 error:
  56         isl_set_free(set);
  57         FN(EL,free)(el);
  58         return isl_stat_error;
  59 }
  60
  61 isl_stat FN(PW,foreach_lifted_piece)(__isl_keep PW *pw,
  62         isl_stat (*fn)(__isl_take isl_set *set, __isl_take EL *el,
  63                     void *user), void *user)
  64 {
  65         int i;
  66
  67         if (!pw)
  68                 return isl_stat_error;
  69
  70         for (i = 0; i < pw->n; ++i) {
  71                 isl_bool any;
  72                 isl_set *set;
  73                 EL *el;
  74
  75                 any = any_divs(pw->p[i].set);
  76                 if (any < 0)
  77                         return isl_stat_error;
  78                 set = isl_set_copy(pw->p[i].set);
  79                 el = FN(EL,copy)(pw->p[i].FIELD);
  80                 if (!any) {
  81                         if (fn(set, el, user) < 0)
  82                                 return isl_stat_error;
  83                         continue;
  84                 }
  85                 if (foreach_lifted_subset(set, el, fn, user) < 0)
  86                         return isl_stat_error;
  87         }
  88
  89         return isl_stat_ok;
  90 }