Home | History | Annotate | Line # | Download | only in dist
isl_pw_add_disjoint_templ.c revision 1.1 
      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)(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;
     89 error:
     90 	FN(PW,free)(pw1);
     91 	FN(PW,free)(pw2);
     92 	return NULL;
     93 }
     94