forked from OSchip/llvm-project
82 lines
2.1 KiB
C
82 lines
2.1 KiB
C
/*
|
||
* Copyright 2011 Sven Verdoolaege
|
||
* Copyright 2012-2013 Ecole Normale Superieure
|
||
*
|
||
* Use of this software is governed by the MIT license
|
||
*
|
||
* Written by Sven Verdoolaege,
|
||
* Ecole Normale Superieure, 45 rue d’Ulm, 75230 Paris, France
|
||
*/
|
||
|
||
#include <isl_multi_macro.h>
|
||
|
||
/* Transform the elements of "multi" by applying "fn" to them
|
||
* with extra argument "set".
|
||
*
|
||
* The parameters of "multi" and "set" are assumed to have been aligned.
|
||
*/
|
||
__isl_give MULTI(BASE) *FN(FN(MULTI(BASE),apply_aligned),APPLY_DOMBASE)(
|
||
__isl_take MULTI(BASE) *multi, __isl_take APPLY_DOM *set,
|
||
__isl_give EL *(*fn)(EL *el, __isl_take APPLY_DOM *set))
|
||
{
|
||
int i;
|
||
|
||
if (!multi || !set)
|
||
goto error;
|
||
|
||
if (multi->n == 0) {
|
||
FN(APPLY_DOM,free)(set);
|
||
return multi;
|
||
}
|
||
|
||
multi = FN(MULTI(BASE),cow)(multi);
|
||
if (!multi)
|
||
goto error;
|
||
|
||
for (i = 0; i < multi->n; ++i) {
|
||
multi->p[i] = fn(multi->p[i], FN(APPLY_DOM,copy)(set));
|
||
if (!multi->p[i])
|
||
goto error;
|
||
}
|
||
|
||
FN(APPLY_DOM,free)(set);
|
||
return multi;
|
||
error:
|
||
FN(APPLY_DOM,free)(set);
|
||
FN(MULTI(BASE),free)(multi);
|
||
return NULL;
|
||
}
|
||
|
||
/* Transform the elements of "multi" by applying "fn" to them
|
||
* with extra argument "set".
|
||
*
|
||
* Align the parameters if needed and call apply_set_aligned.
|
||
*/
|
||
static __isl_give MULTI(BASE) *FN(FN(MULTI(BASE),apply),APPLY_DOMBASE)(
|
||
__isl_take MULTI(BASE) *multi, __isl_take APPLY_DOM *set,
|
||
__isl_give EL *(*fn)(EL *el, __isl_take APPLY_DOM *set))
|
||
{
|
||
isl_ctx *ctx;
|
||
|
||
if (!multi || !set)
|
||
goto error;
|
||
|
||
if (isl_space_match(multi->space, isl_dim_param,
|
||
set->dim, isl_dim_param))
|
||
return FN(FN(MULTI(BASE),apply_aligned),APPLY_DOMBASE)(multi,
|
||
set, fn);
|
||
ctx = FN(MULTI(BASE),get_ctx)(multi);
|
||
if (!isl_space_has_named_params(multi->space) ||
|
||
!isl_space_has_named_params(set->dim))
|
||
isl_die(ctx, isl_error_invalid,
|
||
"unaligned unnamed parameters", goto error);
|
||
multi = FN(MULTI(BASE),align_params)(multi,
|
||
FN(APPLY_DOM,get_space)(set));
|
||
set = FN(APPLY_DOM,align_params)(set, FN(MULTI(BASE),get_space)(multi));
|
||
return FN(FN(MULTI(BASE),apply_aligned),APPLY_DOMBASE)(multi, set, fn);
|
||
error:
|
||
FN(MULTI(BASE),free)(multi);
|
||
FN(APPLY_DOM,free)(set);
|
||
return NULL;
|
||
}
|