isl_multi_identity_templ.c
2.05 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
/*
* Copyright 2012 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/space.h>
#include <isl/local_space.h>
#include <isl_multi_macro.h>
/* Create a multi expression in the given space that maps each
* input dimension to the corresponding output dimension.
*/
__isl_give MULTI(BASE) *FN(MULTI(BASE),identity)(__isl_take isl_space *space)
{
int i;
isl_size n_in, n_out;
isl_local_space *ls;
MULTI(BASE) *multi;
if (!space)
return NULL;
if (isl_space_is_set(space))
isl_die(isl_space_get_ctx(space), isl_error_invalid,
"expecting map space", goto error);
n_in = isl_space_dim(space, isl_dim_in);
n_out = isl_space_dim(space, isl_dim_out);
if (n_in < 0 || n_out < 0)
goto error;
if (n_in != n_out)
isl_die(isl_space_get_ctx(space), isl_error_invalid,
"number of input and output dimensions needs to be "
"the same", goto error);
multi = FN(MULTI(BASE),alloc)(isl_space_copy(space));
if (!n_out) {
isl_space_free(space);
return multi;
}
space = isl_space_domain(space);
ls = isl_local_space_from_space(space);
for (i = 0; i < n_out; ++i) {
EL *el;
el = FN(EL,var_on_domain)(isl_local_space_copy(ls),
isl_dim_set, i);
multi = FN(FN(MULTI(BASE),set),BASE)(multi, i, el);
}
isl_local_space_free(ls);
return multi;
error:
isl_space_free(space);
return NULL;
}
/* Create a multi expression that maps elements in the given space
* to themselves.
*/
__isl_give MULTI(BASE) *FN(MULTI(BASE),identity_on_domain_space)(
__isl_take isl_space *space)
{
return FN(MULTI(BASE),identity)(isl_space_map_from_set(space));
}
/* Create a multi expression in the same space as "multi" that maps each
* input dimension to the corresponding output dimension.
*/
__isl_give MULTI(BASE) *FN(FN(MULTI(BASE),identity_multi),BASE)(
__isl_take MULTI(BASE) *multi)
{
isl_space *space;
space = FN(MULTI(BASE),get_space)(multi);
FN(MULTI(BASE),free)(multi);
return FN(MULTI(BASE),identity)(space);
}