94 lines · c
1/*2 * Copyright 2010 INRIA Saclay3 * Copyright 2011 Sven Verdoolaege4 *5 * Use of this software is governed by the MIT license6 *7 * Written by Sven Verdoolaege, INRIA Saclay - Ile-de-France,8 * Parc Club Orsay Universite, ZAC des vignes, 4 rue Jacques Monod,9 * 91893 Orsay, France10 */11 12#include <isl_pw_macro.h>13 14/* Make sure "pw" has room for at least "n" more pieces.15 *16 * If there is only one reference to pw, we extend it in place.17 * Otherwise, we create a new PW and copy the pieces.18 */19static __isl_give PW *FN(PW,grow)(__isl_take PW *pw, int n)20{21 int i;22 isl_ctx *ctx;23 PW *res;24 25 if (!pw)26 return NULL;27 if (pw->n + n <= pw->size)28 return pw;29 ctx = FN(PW,get_ctx)(pw);30 n += pw->n;31 if (pw->ref == 1) {32 res = isl_realloc(ctx, pw, struct PW,33 sizeof(struct PW) + (n - 1) * sizeof(S(PW,piece)));34 if (!res)35 return FN(PW,free)(pw);36 res->size = n;37 return res;38 }39 res = FN(PW,alloc_size)(isl_space_copy(pw->dim) OPT_TYPE_ARG(pw->), n);40 if (!res)41 return FN(PW,free)(pw);42 for (i = 0; i < pw->n; ++i)43 res = FN(PW,add_piece)(res, isl_set_copy(pw->p[i].set),44 FN(EL,copy)(pw->p[i].FIELD));45 FN(PW,free)(pw);46 return res;47}48 49__isl_give PW *FN(PW,add_disjoint)(__isl_take PW *pw1, __isl_take PW *pw2)50{51 int i;52 isl_ctx *ctx;53 54 if (FN(PW,align_params_bin)(&pw1, &pw2) < 0)55 goto error;56 57 if (pw1->size < pw1->n + pw2->n && pw1->n < pw2->n)58 return FN(PW,add_disjoint)(pw2, pw1);59 60 ctx = isl_space_get_ctx(pw1->dim);61 if (!OPT_EQUAL_TYPES(pw1->, pw2->))62 isl_die(ctx, isl_error_invalid,63 "fold types don't match", goto error);64 if (FN(PW,check_equal_space)(pw1, pw2) < 0)65 goto error;66 67 if (FN(PW,IS_ZERO)(pw1)) {68 FN(PW,free)(pw1);69 return pw2;70 }71 72 if (FN(PW,IS_ZERO)(pw2)) {73 FN(PW,free)(pw2);74 return pw1;75 }76 77 pw1 = FN(PW,grow)(pw1, pw2->n);78 if (!pw1)79 goto error;80 81 for (i = 0; i < pw2->n; ++i)82 pw1 = FN(PW,add_piece)(pw1,83 isl_set_copy(pw2->p[i].set),84 FN(EL,copy)(pw2->p[i].FIELD));85 86 FN(PW,free)(pw2);87 88 return pw1;89error:90 FN(PW,free)(pw1);91 FN(PW,free)(pw2);92 return NULL;93}94