isl_multi_union_add_templ.c

   1 /*
   2  * Copyright 2017      Sven Verdoolaege
   3  *
   4  * Use of this software is governed by the MIT license
   5  *
   6  * Written by Sven Verdoolaege.
   7  */
   8
   9 #include <isl_multi_macro.h>
  10
  11 /* Compute the sum of "multi1" and "multi2" on the union of their domains,
  12  * with the actual sum on the shared domain and
  13  * the defined expression on the symmetric difference of the domains.
  14  *
  15  * We simply iterate over the elements in both arguments and
  16  * call isl_union_pw_aff_union_add on each of them, if there is
  17  * at least one element.
  18  *
  19  * Otherwise, the two expressions have an explicit domain and
  20  * the union of these explicit domains is computed.
  21  * This assumes that the explicit domains are either both in terms
  22  * of specific domains elements or both in terms of parameters.
  23  * However, if one of the expressions does not have any constraints
  24  * on its explicit domain, then this is allowed as well and the result
  25  * is the expression with no constraints on its explicit domain.
  26  */
  27 __isl_give MULTI(BASE) *FN(MULTI(BASE),union_add)(
  28         __isl_take MULTI(BASE) *multi1, __isl_take MULTI(BASE) *multi2)
  29 {
  30         isl_bool has_domain, is_params1, is_params2;
  31
  32         if (!multi1)
  33                 goto error;
  34         if (multi1->n > 0)
  35                 return FN(MULTI(BASE),bin_op)(multi1, multi2,
  36                                             &FN(EL,union_add));
  37         FN(MULTI(BASE),align_params_bin)(&multi1, &multi2);
  38         if (FN(MULTI(BASE),check_equal_space)(multi1, multi2) < 0)
  39                 goto error;
  40         if (FN(MULTI(BASE),check_has_explicit_domain)(multi1) < 0 ||
  41             FN(MULTI(BASE),check_has_explicit_domain)(multi2) < 0)
  42                 goto error;
  43
  44         has_domain = FN(MULTI(BASE),has_non_trivial_domain)(multi1);
  45         if (has_domain < 0)
  46                 goto error;
  47         if (!has_domain) {
  48                 FN(MULTI(BASE),free)(multi2);
  49                 return multi1;
  50         }
  51         has_domain = FN(MULTI(BASE),has_non_trivial_domain)(multi2);
  52         if (has_domain < 0)
  53                 goto error;
  54         if (!has_domain) {
  55                 FN(MULTI(BASE),free)(multi1);
  56                 return multi2;
  57         }
  58
  59         is_params1 = FN(DOM,is_params)(multi1->u.dom);
  60         is_params2 = FN(DOM,is_params)(multi2->u.dom);
  61         if (is_params1 < 0 || is_params2 < 0)
  62                 goto error;
  63         if (is_params1 != is_params2)
  64                 isl_die(FN(MULTI(BASE),get_ctx)(multi1),
  65                         isl_error_invalid,
  66                         "cannot compute union of concrete domain and "
  67                         "parameter constraints", goto error);
  68         multi1 = FN(MULTI(BASE),cow)(multi1);
  69         if (!multi1)
  70                 goto error;
  71         multi1->u.dom = FN(DOM,union)(multi1->u.dom,
  72                                         FN(DOM,copy)(multi2->u.dom));
  73         if (!multi1->u.dom)
  74                 goto error;
  75         FN(MULTI(BASE),free)(multi2);
  76         return multi1;
  77 error:
  78         FN(MULTI(BASE),free)(multi1);
  79         FN(MULTI(BASE),free)(multi2);
  80         return NULL;
  81 }