| 1 | /* |
|---|
| 2 | * Copyright 2010 INRIA Saclay |
|---|
| 3 | * Copyright 2011 Sven Verdoolaege |
|---|
| 4 | * |
|---|
| 5 | * Use of this software is governed by the MIT license |
|---|
| 6 | * |
|---|
| 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, France |
|---|
| 10 | */ |
|---|
| 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 | */ |
|---|
| 19 | static __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)(space: isl_space_copy(space: 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)(pw: res, set: isl_set_copy(set: pw->p[i].set), |
|---|
| 44 | FN(EL,copy)(aff: 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)(obj1: &pw1, obj2: &pw2) < 0) |
|---|
| 55 | goto error; |
|---|
| 56 | |
|---|
| 57 | if (pw1->size < pw1->n + pw2->n && pw1->n < pw2->n) |
|---|
| 58 | return FN(PW,add_disjoint)(pw1: pw2, pw2: pw1); |
|---|
| 59 | |
|---|
| 60 | ctx = isl_space_get_ctx(space: 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)(obj1: pw1, obj2: pw2) < 0) |
|---|
| 65 | goto error; |
|---|
| 66 | |
|---|
| 67 | if (FN(PW,IS_ZERO)(pw: pw1)) { |
|---|
| 68 | FN(PW,free)(pw: pw1); |
|---|
| 69 | return pw2; |
|---|
| 70 | } |
|---|
| 71 | |
|---|
| 72 | if (FN(PW,IS_ZERO)(pw: pw2)) { |
|---|
| 73 | FN(PW,free)(pw: pw2); |
|---|
| 74 | return pw1; |
|---|
| 75 | } |
|---|
| 76 | |
|---|
| 77 | pw1 = FN(PW,grow)(pw: pw1, n: pw2->n); |
|---|
| 78 | if (!pw1) |
|---|
| 79 | goto error; |
|---|
| 80 | |
|---|
| 81 | for (i = 0; i < pw2->n; ++i) |
|---|
| 82 | pw1 = FN(PW,add_piece)(pw: pw1, |
|---|
| 83 | set: isl_set_copy(set: pw2->p[i].set), |
|---|
| 84 | FN(EL,copy)(aff: pw2->p[i].FIELD)); |
|---|
| 85 | |
|---|
| 86 | FN(PW,free)(pw: pw2); |
|---|
| 87 | |
|---|
| 88 | return pw1; |
|---|
| 89 | error: |
|---|
| 90 | FN(PW,free)(pw: pw1); |
|---|
| 91 | FN(PW,free)(pw: pw2); |
|---|
| 92 | return NULL; |
|---|
| 93 | } |
|---|
| 94 | |
|---|
