1 /*
2  * Copyright 2011      Sven Verdoolaege
3  * Copyright 2013      Ecole Normale Superieure
4  *
5  * Use of this software is governed by the MIT license
6  *
7  * Written by Sven Verdoolaege,
8  * Ecole Normale Superieure, 45 rue d'Ulm, 75230 Paris, France
9  */
10 
11 #include <isl/space.h>
12 
13 #include <isl_multi_macro.h>
14 
15 /* Return the position of the dimension of the given type and name
16  * in "multi".
17  * Return -1 if no such dimension can be found.
18  */
FN(MULTI (BASE),find_dim_by_name)19 int FN(MULTI(BASE),find_dim_by_name)(__isl_keep MULTI(BASE) *multi,
20 	enum isl_dim_type type, const char *name)
21 {
22 	if (!multi)
23 		return -1;
24 	return isl_space_find_dim_by_name(multi->space, type, name);
25 }
26 
27 /* Return the position of the first dimension of "type" with id "id".
28  * Return -1 if there is no such dimension.
29  */
FN(MULTI (BASE),find_dim_by_id)30 int FN(MULTI(BASE),find_dim_by_id)(__isl_keep MULTI(BASE) *multi,
31 	enum isl_dim_type type, __isl_keep isl_id *id)
32 {
33 	if (!multi)
34 		return -1;
35 	return isl_space_find_dim_by_id(multi->space, type, id);
36 }
37 
38 /* Return the id of the given dimension.
39  */
FN(MULTI (BASE),get_dim_id)40 __isl_give isl_id *FN(MULTI(BASE),get_dim_id)(__isl_keep MULTI(BASE) *multi,
41 	enum isl_dim_type type, unsigned pos)
42 {
43 	return multi ? isl_space_get_dim_id(multi->space, type, pos) : NULL;
44 }
45 
MULTI(BASE)46 __isl_give MULTI(BASE) *FN(MULTI(BASE),set_dim_name)(
47 	__isl_take MULTI(BASE) *multi,
48 	enum isl_dim_type type, unsigned pos, const char *s)
49 {
50 	int i;
51 
52 	multi = FN(MULTI(BASE),cow)(multi);
53 	if (!multi)
54 		return NULL;
55 
56 	multi->space = isl_space_set_dim_name(multi->space, type, pos, s);
57 	if (!multi->space)
58 		return FN(MULTI(BASE),free)(multi);
59 
60 	if (type == isl_dim_out)
61 		return multi;
62 	for (i = 0; i < multi->n; ++i) {
63 		multi->u.p[i] = FN(EL,set_dim_name)(multi->u.p[i],
64 							type, pos, s);
65 		if (!multi->u.p[i])
66 			return FN(MULTI(BASE),free)(multi);
67 	}
68 
69 	return multi;
70 }
71 
72 /* Set the id of the given dimension of "multi" to "id".
73  */
MULTI(BASE)74 __isl_give MULTI(BASE) *FN(MULTI(BASE),set_dim_id)(
75 	__isl_take MULTI(BASE) *multi,
76 	enum isl_dim_type type, unsigned pos, __isl_take isl_id *id)
77 {
78 	isl_space *space;
79 
80 	multi = FN(MULTI(BASE),cow)(multi);
81 	if (!multi || !id)
82 		goto error;
83 
84 	space = FN(MULTI(BASE),get_space)(multi);
85 	space = isl_space_set_dim_id(space, type, pos, id);
86 
87 	return FN(MULTI(BASE),reset_space)(multi, space);
88 error:
89 	isl_id_free(id);
90 	FN(MULTI(BASE),free)(multi);
91 	return NULL;
92 }
93