2 * Copyright 2008-2009 Katholieke Universiteit Leuven
4 * Use of this software is governed by the GNU LGPLv2.1 license
6 * Written by Sven Verdoolaege, K.U.Leuven, Departement
7 * Computerwetenschappen, Celestijnenlaan 200A, B-3001 Leuven, Belgium
13 struct isl_dim
*isl_dim_alloc(struct isl_ctx
*ctx
,
14 unsigned nparam
, unsigned n_in
, unsigned n_out
)
18 dim
= isl_alloc_type(ctx
, struct isl_dim
);
35 struct isl_dim
*isl_dim_set_alloc(struct isl_ctx
*ctx
,
36 unsigned nparam
, unsigned dim
)
38 return isl_dim_alloc(ctx
, nparam
, 0, dim
);
41 static unsigned global_pos(struct isl_dim
*dim
,
42 enum isl_dim_type type
, unsigned pos
)
44 struct isl_ctx
*ctx
= dim
->ctx
;
48 isl_assert(ctx
, pos
< dim
->nparam
, return isl_dim_total(dim
));
51 isl_assert(ctx
, pos
< dim
->n_in
, return isl_dim_total(dim
));
52 return pos
+ dim
->nparam
;
54 isl_assert(ctx
, pos
< dim
->n_out
, return isl_dim_total(dim
));
55 return pos
+ dim
->nparam
+ dim
->n_in
;
57 isl_assert(ctx
, 0, return isl_dim_total(dim
));
59 return isl_dim_total(dim
);
62 /* Extend length of names array to the total number of dimensions.
64 static __isl_give isl_dim
*extend_names(__isl_take isl_dim
*dim
)
66 struct isl_name
**names
;
69 if (isl_dim_total(dim
) <= dim
->n_name
)
73 dim
->names
= isl_calloc_array(dim
->ctx
,
74 struct isl_name
*, isl_dim_total(dim
));
78 names
= isl_realloc_array(dim
->ctx
, dim
->names
,
79 struct isl_name
*, isl_dim_total(dim
));
83 for (i
= dim
->n_name
; i
< isl_dim_total(dim
); ++i
)
87 dim
->n_name
= isl_dim_total(dim
);
95 static struct isl_dim
*set_name(struct isl_dim
*dim
,
96 enum isl_dim_type type
, unsigned pos
,
97 struct isl_name
*name
)
99 struct isl_ctx
*ctx
= dim
->ctx
;
100 dim
= isl_dim_cow(dim
);
105 pos
= global_pos(dim
, type
, pos
);
106 isl_assert(ctx
, pos
!= isl_dim_total(dim
), goto error
);
108 if (pos
>= dim
->n_name
) {
111 dim
= extend_names(dim
);
116 dim
->names
[pos
] = name
;
120 isl_name_free(ctx
, name
);
125 static struct isl_name
*get_name(struct isl_dim
*dim
,
126 enum isl_dim_type type
, unsigned pos
)
131 pos
= global_pos(dim
, type
, pos
);
132 if (pos
== isl_dim_total(dim
))
134 if (pos
>= dim
->n_name
)
136 return dim
->names
[pos
];
139 static unsigned offset(struct isl_dim
*dim
, enum isl_dim_type type
)
142 case isl_dim_param
: return 0;
143 case isl_dim_in
: return dim
->nparam
;
144 case isl_dim_out
: return dim
->nparam
+ dim
->n_in
;
148 static unsigned n(struct isl_dim
*dim
, enum isl_dim_type type
)
151 case isl_dim_param
: return dim
->nparam
;
152 case isl_dim_in
: return dim
->n_in
;
153 case isl_dim_out
: return dim
->n_out
;
157 unsigned isl_dim_size(struct isl_dim
*dim
, enum isl_dim_type type
)
164 unsigned isl_dim_offset(__isl_keep isl_dim
*dim
, enum isl_dim_type type
)
168 return offset(dim
, type
);
171 static struct isl_dim
*copy_names(struct isl_dim
*dst
,
172 enum isl_dim_type dst_type
, unsigned offset
, struct isl_dim
*src
,
173 enum isl_dim_type src_type
)
176 struct isl_name
*name
;
178 for (i
= 0; i
< n(src
, src_type
); ++i
) {
179 name
= get_name(src
, src_type
, i
);
182 dst
= set_name(dst
, dst_type
, offset
+ i
,
183 isl_name_copy(dst
->ctx
, name
));
190 struct isl_dim
*isl_dim_dup(struct isl_dim
*dim
)
193 dup
= isl_dim_alloc(dim
->ctx
, dim
->nparam
, dim
->n_in
, dim
->n_out
);
196 dup
= copy_names(dup
, isl_dim_param
, 0, dim
, isl_dim_param
);
197 dup
= copy_names(dup
, isl_dim_in
, 0, dim
, isl_dim_in
);
198 dup
= copy_names(dup
, isl_dim_out
, 0, dim
, isl_dim_out
);
202 struct isl_dim
*isl_dim_cow(struct isl_dim
*dim
)
210 return isl_dim_dup(dim
);
213 struct isl_dim
*isl_dim_copy(struct isl_dim
*dim
)
222 void isl_dim_free(struct isl_dim
*dim
)
232 for (i
= 0; i
< dim
->n_name
; ++i
)
233 isl_name_free(dim
->ctx
, dim
->names
[i
]);
235 isl_ctx_deref(dim
->ctx
);
240 struct isl_dim
*isl_dim_set_name(struct isl_dim
*dim
,
241 enum isl_dim_type type
, unsigned pos
,
244 struct isl_name
*name
;
247 name
= isl_name_get(dim
->ctx
, s
);
250 return set_name(dim
, type
, pos
, name
);
256 const char *isl_dim_get_name(struct isl_dim
*dim
,
257 enum isl_dim_type type
, unsigned pos
)
259 struct isl_name
*name
= get_name(dim
, type
, pos
);
260 return name
? name
->name
: NULL
;
263 static int match(struct isl_dim
*dim1
, enum isl_dim_type dim1_type
,
264 struct isl_dim
*dim2
, enum isl_dim_type dim2_type
)
268 if (n(dim1
, dim1_type
) != n(dim2
, dim2_type
))
271 if (!dim1
->names
&& !dim2
->names
)
274 for (i
= 0; i
< n(dim1
, dim1_type
); ++i
) {
275 if (get_name(dim1
, dim1_type
, i
) !=
276 get_name(dim2
, dim2_type
, i
))
282 int isl_dim_match(struct isl_dim
*dim1
, enum isl_dim_type dim1_type
,
283 struct isl_dim
*dim2
, enum isl_dim_type dim2_type
)
285 return match(dim1
, dim1_type
, dim2
, dim2_type
);
288 static void get_names(struct isl_dim
*dim
, enum isl_dim_type type
,
289 unsigned first
, unsigned n
, struct isl_name
**names
)
293 for (i
= 0; i
< n
; ++i
)
294 names
[i
] = get_name(dim
, type
, first
+i
);
297 struct isl_dim
*isl_dim_extend(struct isl_dim
*dim
,
298 unsigned nparam
, unsigned n_in
, unsigned n_out
)
300 struct isl_name
**names
= NULL
;
304 if (dim
->nparam
== nparam
&& dim
->n_in
== n_in
&& dim
->n_out
== n_out
)
307 isl_assert(dim
->ctx
, dim
->nparam
<= nparam
, goto error
);
308 isl_assert(dim
->ctx
, dim
->n_in
<= n_in
, goto error
);
309 isl_assert(dim
->ctx
, dim
->n_out
<= n_out
, goto error
);
311 dim
= isl_dim_cow(dim
);
314 names
= isl_calloc_array(dim
->ctx
, struct isl_name
*,
315 nparam
+ n_in
+ n_out
);
318 get_names(dim
, isl_dim_param
, 0, dim
->nparam
, names
);
319 get_names(dim
, isl_dim_in
, 0, dim
->n_in
, names
+ nparam
);
320 get_names(dim
, isl_dim_out
, 0, dim
->n_out
,
321 names
+ nparam
+ n_in
);
324 dim
->n_name
= nparam
+ n_in
+ n_out
;
326 dim
->nparam
= nparam
;
337 struct isl_dim
*isl_dim_add(struct isl_dim
*dim
, enum isl_dim_type type
,
342 return isl_dim_extend(dim
,
343 dim
->nparam
+ n
, dim
->n_in
, dim
->n_out
);
345 return isl_dim_extend(dim
,
346 dim
->nparam
, dim
->n_in
+ n
, dim
->n_out
);
348 return isl_dim_extend(dim
,
349 dim
->nparam
, dim
->n_in
, dim
->n_out
+ n
);
354 __isl_give isl_dim
*isl_dim_insert(__isl_take isl_dim
*dim
,
355 enum isl_dim_type type
, unsigned pos
, unsigned n
)
357 struct isl_name
**names
= NULL
;
364 isl_assert(dim
->ctx
, pos
<= isl_dim_size(dim
, type
), goto error
);
366 dim
= isl_dim_cow(dim
);
374 names
= isl_calloc_array(dim
->ctx
, struct isl_name
*,
375 dim
->nparam
+ dim
->n_in
+ dim
->n_out
+ n
);
379 size
[isl_dim_param
] = dim
->nparam
;
380 size
[isl_dim_in
] = dim
->n_in
;
381 size
[isl_dim_out
] = dim
->n_out
;
382 for (t
= isl_dim_param
; t
<= isl_dim_out
; ++t
) {
384 get_names(dim
, t
, 0, size
[t
], names
+ off
);
387 get_names(dim
, t
, 0, pos
, names
+ off
);
389 get_names(dim
, t
, pos
, size
[t
]-pos
, names
+off
);
390 off
+= size
[t
] - pos
;
395 dim
->n_name
= dim
->nparam
+ dim
->n_in
+ dim
->n_out
+ n
;
398 case isl_dim_param
: dim
->nparam
+= n
; break;
399 case isl_dim_in
: dim
->n_in
+= n
; break;
400 case isl_dim_out
: dim
->n_out
+= n
; break;
409 __isl_give isl_dim
*isl_dim_move(__isl_take isl_dim
*dim
,
410 enum isl_dim_type dst_type
, unsigned dst_pos
,
411 enum isl_dim_type src_type
, unsigned src_pos
, unsigned n
)
418 isl_assert(dim
->ctx
, src_pos
+ n
<= isl_dim_size(dim
, src_type
),
421 if (dst_type
== src_type
&& dst_pos
== src_pos
)
424 isl_assert(dim
->ctx
, dst_type
!= src_type
, goto error
);
426 dim
= isl_dim_cow(dim
);
431 struct isl_name
**names
;
435 names
= isl_calloc_array(dim
->ctx
, struct isl_name
*,
436 dim
->nparam
+ dim
->n_in
+ dim
->n_out
);
440 size
[isl_dim_param
] = dim
->nparam
;
441 size
[isl_dim_in
] = dim
->n_in
;
442 size
[isl_dim_out
] = dim
->n_out
;
443 for (t
= isl_dim_param
; t
<= isl_dim_out
; ++t
) {
445 get_names(dim
, t
, 0, dst_pos
, names
+ off
);
447 get_names(dim
, src_type
, src_pos
, n
, names
+off
);
449 get_names(dim
, t
, dst_pos
, size
[t
] - dst_pos
,
451 off
+= size
[t
] - dst_pos
;
452 } else if (t
== src_type
) {
453 get_names(dim
, t
, 0, src_pos
, names
+ off
);
455 get_names(dim
, t
, src_pos
+ n
,
456 size
[t
] - src_pos
- n
, names
+ off
);
457 off
+= size
[t
] - src_pos
- n
;
459 get_names(dim
, t
, 0, size
[t
], names
+ off
);
465 dim
->n_name
= dim
->nparam
+ dim
->n_in
+ dim
->n_out
;
469 case isl_dim_param
: dim
->nparam
+= n
; break;
470 case isl_dim_in
: dim
->n_in
+= n
; break;
471 case isl_dim_out
: dim
->n_out
+= n
; break;
475 case isl_dim_param
: dim
->nparam
-= n
; break;
476 case isl_dim_in
: dim
->n_in
-= n
; break;
477 case isl_dim_out
: dim
->n_out
-= n
; break;
486 struct isl_dim
*isl_dim_join(struct isl_dim
*left
, struct isl_dim
*right
)
493 isl_assert(left
->ctx
, match(left
, isl_dim_param
, right
, isl_dim_param
),
495 isl_assert(left
->ctx
, n(left
, isl_dim_out
) == n(right
, isl_dim_in
),
498 dim
= isl_dim_alloc(left
->ctx
, left
->nparam
, left
->n_in
, right
->n_out
);
502 dim
= copy_names(dim
, isl_dim_param
, 0, left
, isl_dim_param
);
503 dim
= copy_names(dim
, isl_dim_in
, 0, left
, isl_dim_in
);
504 dim
= copy_names(dim
, isl_dim_out
, 0, right
, isl_dim_out
);
516 struct isl_dim
*isl_dim_product(struct isl_dim
*left
, struct isl_dim
*right
)
523 isl_assert(left
->ctx
, match(left
, isl_dim_param
, right
, isl_dim_param
),
526 dim
= isl_dim_alloc(left
->ctx
, left
->nparam
,
527 left
->n_in
+ right
->n_in
, left
->n_out
+ right
->n_out
);
531 dim
= copy_names(dim
, isl_dim_param
, 0, left
, isl_dim_param
);
532 dim
= copy_names(dim
, isl_dim_in
, 0, left
, isl_dim_in
);
533 dim
= copy_names(dim
, isl_dim_in
, left
->n_in
, right
, isl_dim_in
);
534 dim
= copy_names(dim
, isl_dim_out
, 0, left
, isl_dim_out
);
535 dim
= copy_names(dim
, isl_dim_out
, left
->n_out
, right
, isl_dim_out
);
547 struct isl_dim
*isl_dim_map(struct isl_dim
*dim
)
549 struct isl_name
**names
= NULL
;
553 isl_assert(dim
->ctx
, dim
->n_in
== 0, goto error
);
556 dim
= isl_dim_cow(dim
);
560 names
= isl_calloc_array(dim
->ctx
, struct isl_name
*,
561 dim
->nparam
+ dim
->n_out
+ dim
->n_out
);
564 get_names(dim
, isl_dim_param
, 0, dim
->nparam
, names
);
565 get_names(dim
, isl_dim_out
, 0, dim
->n_out
, names
+ dim
->nparam
);
567 dim
->n_in
= dim
->n_out
;
571 dim
->n_name
= dim
->nparam
+ dim
->n_out
+ dim
->n_out
;
572 dim
= copy_names(dim
, isl_dim_out
, 0, dim
, isl_dim_in
);
580 static struct isl_dim
*set_names(struct isl_dim
*dim
, enum isl_dim_type type
,
581 unsigned first
, unsigned n
, struct isl_name
**names
)
585 for (i
= 0; i
< n
; ++i
)
586 dim
= set_name(dim
, type
, first
+i
, names
[i
]);
591 struct isl_dim
*isl_dim_reverse(struct isl_dim
*dim
)
594 struct isl_name
**names
= NULL
;
598 if (match(dim
, isl_dim_in
, dim
, isl_dim_out
))
601 dim
= isl_dim_cow(dim
);
606 names
= isl_alloc_array(dim
->ctx
, struct isl_name
*,
607 dim
->n_in
+ dim
->n_out
);
610 get_names(dim
, isl_dim_in
, 0, dim
->n_in
, names
);
611 get_names(dim
, isl_dim_out
, 0, dim
->n_out
, names
+ dim
->n_in
);
615 dim
->n_in
= dim
->n_out
;
619 dim
= set_names(dim
, isl_dim_out
, 0, dim
->n_out
, names
);
620 dim
= set_names(dim
, isl_dim_in
, 0, dim
->n_in
, names
+ dim
->n_out
);
631 struct isl_dim
*isl_dim_drop(struct isl_dim
*dim
, enum isl_dim_type type
,
632 unsigned first
, unsigned num
)
642 isl_assert(dim
->ctx
, first
+ num
<= n(dim
, type
), goto error
);
643 dim
= isl_dim_cow(dim
);
647 dim
= extend_names(dim
);
650 for (i
= 0; i
< num
; ++i
)
651 isl_name_free(dim
->ctx
, get_name(dim
, type
, first
+i
));
652 for (i
= first
+num
; i
< n(dim
, type
); ++i
)
653 set_name(dim
, type
, i
- num
, get_name(dim
, type
, i
));
656 get_names(dim
, isl_dim_in
, 0, dim
->n_in
,
657 dim
->names
+ offset(dim
, isl_dim_in
) - num
);
659 get_names(dim
, isl_dim_out
, 0, dim
->n_out
,
660 dim
->names
+ offset(dim
, isl_dim_out
) - num
);
667 case isl_dim_param
: dim
->nparam
-= num
; break;
668 case isl_dim_in
: dim
->n_in
-= num
; break;
669 case isl_dim_out
: dim
->n_out
-= num
; break;
677 struct isl_dim
*isl_dim_drop_inputs(struct isl_dim
*dim
,
678 unsigned first
, unsigned n
)
680 return isl_dim_drop(dim
, isl_dim_in
, first
, n
);
683 struct isl_dim
*isl_dim_drop_outputs(struct isl_dim
*dim
,
684 unsigned first
, unsigned n
)
686 return isl_dim_drop(dim
, isl_dim_out
, first
, n
);
689 struct isl_dim
*isl_dim_domain(struct isl_dim
*dim
)
693 dim
= isl_dim_drop_outputs(dim
, 0, dim
->n_out
);
694 return isl_dim_reverse(dim
);
697 struct isl_dim
*isl_dim_range(struct isl_dim
*dim
)
701 return isl_dim_drop_inputs(dim
, 0, dim
->n_in
);
704 struct isl_dim
*isl_dim_underlying(struct isl_dim
*dim
, unsigned n_div
)
711 dim
->nparam
== 0 && dim
->n_in
== 0 && dim
->n_name
== 0)
713 dim
= isl_dim_cow(dim
);
716 dim
->n_out
+= dim
->nparam
+ dim
->n_in
+ n_div
;
720 for (i
= 0; i
< dim
->n_name
; ++i
)
721 isl_name_free(dim
->ctx
, get_name(dim
, isl_dim_out
, i
));
727 unsigned isl_dim_total(struct isl_dim
*dim
)
729 return dim
->nparam
+ dim
->n_in
+ dim
->n_out
;
732 int isl_dim_equal(struct isl_dim
*dim1
, struct isl_dim
*dim2
)
734 return match(dim1
, isl_dim_param
, dim2
, isl_dim_param
) &&
735 n(dim1
, isl_dim_in
) == n(dim2
, isl_dim_in
) &&
736 n(dim1
, isl_dim_out
) == n(dim2
, isl_dim_out
);
739 int isl_dim_compatible(struct isl_dim
*dim1
, struct isl_dim
*dim2
)
741 return dim1
->nparam
== dim2
->nparam
&&
742 dim1
->n_in
+ dim1
->n_out
== dim2
->n_in
+ dim2
->n_out
;