isl_pw_add_disjoint_templ.c

Go to the documentation of this file.

/*
 * Copyright 2010      INRIA Saclay
 * Copyright 2011      Sven Verdoolaege
 *
 * Use of this software is governed by the MIT license
 *
 * Written by Sven Verdoolaege, INRIA Saclay - Ile-de-France,
 * Parc Club Orsay Universite, ZAC des vignes, 4 rue Jacques Monod,
 * 91893 Orsay, France
 */
 
#include <isl_pw_macro.h>
 
/* Make sure "pw" has room for at least "n" more pieces.
 *
 * If there is only one reference to pw, we extend it in place.
 * Otherwise, we create a new PW and copy the pieces.
 */
static __isl_give PW *FN(PW,grow)(__isl_take PW *pw, int n)
{
  int i;
  isl_ctx *ctx;
  PW *res;
 
  if (!pw)
    return NULL;
  if (pw->n + n <= pw->size)
    return pw;
  ctx = FN(PW,get_ctx)(pw);
  n += pw->n;
  if (pw->ref == 1) {
    res = isl_realloc(ctx, pw, struct PW,
          sizeof(struct PW) + (n - 1) * sizeof(S(PW,piece)));
    if (!res)
      return FN(PW,free)(pw);
    res->size = n;
    return res;
  }
  res = FN(PW,alloc_size)(isl_space_copy(pw->dim) OPT_TYPE_ARG(pw->), n);
  if (!res)
    return FN(PW,free)(pw);
  for (i = 0; i < pw->n; ++i)
    res = FN(PW,add_piece)(res, isl_set_copy(pw->p[i].set),
              FN(EL,copy)(pw->p[i].FIELD));
  FN(PW,free)(pw);
  return res;
}
 
__isl_give PW *FN(PW,add_disjoint)(__isl_take PW *pw1, __isl_take PW *pw2)
{
  int i;
  isl_ctx *ctx;
 
  if (FN(PW,align_params_bin)(&pw1, &pw2) < 0)
    goto error;
 
  if (pw1->size < pw1->n + pw2->n && pw1->n < pw2->n)
    return FN(PW,add_disjoint)(pw2, pw1);
 
  ctx = isl_space_get_ctx(pw1->dim);
  if (!OPT_EQUAL_TYPES(pw1->, pw2->))
    isl_die(ctx, isl_error_invalid,
      "fold types don't match", goto error);
  if (FN(PW,check_equal_space)(pw1, pw2) < 0)
    goto error;
 
  if (FN(PW,IS_ZERO)(pw1)) {
    FN(PW,free)(pw1);
    return pw2;
  }
 
  if (FN(PW,IS_ZERO)(pw2)) {
    FN(PW,free)(pw2);
    return pw1;
  }
 
  pw1 = FN(PW,grow)(pw1, pw2->n);
  if (!pw1)
    goto error;
 
  for (i = 0; i < pw2->n; ++i)
    pw1 = FN(PW,add_piece)(pw1,
        isl_set_copy(pw2->p[i].set),
        FN(EL,copy)(pw2->p[i].FIELD));
 
  FN(PW,free)(pw2);
 
  return pw1;
error:
  FN(PW,free)(pw1);
  FN(PW,free)(pw2);
  return NULL;
}