2 * Copyright 2011 Sven Verdoolaege
3 * Copyright 2012 Ecole Normale Superieure
5 * Use of this software is governed by the MIT license
7 * Written by Sven Verdoolaege,
8 * Ecole Normale Superieure, 45 rue d’Ulm, 75230 Paris, France
11 #define xCAT(A,B) A ## B
12 #define CAT(A,B) xCAT(A,B)
14 #define EL CAT(isl_,BASE)
15 #define xFN(TYPE,NAME) TYPE ## _ ## NAME
16 #define FN(TYPE,NAME) xFN(TYPE,NAME)
17 #define xMULTI(BASE) isl_multi_ ## BASE
18 #define MULTI(BASE) xMULTI(BASE)
19 #define MULTI_NAME(BASE) "isl_multi_" #BASE
20 #define xLIST(EL) EL ## _list
21 #define LIST(EL) xLIST(EL)
23 isl_ctx *FN(MULTI(BASE),get_ctx)(__isl_keep MULTI(BASE) *multi)
25 return multi ? isl_space_get_ctx(multi->space) : NULL;
28 __isl_give isl_space *FN(MULTI(BASE),get_space)(__isl_keep MULTI(BASE) *multi)
30 return multi ? isl_space_copy(multi->space) : NULL;
33 __isl_give isl_space *FN(MULTI(BASE),get_domain_space)(
34 __isl_keep MULTI(BASE) *multi)
36 return multi ? isl_space_domain(isl_space_copy(multi->space)) : NULL;
39 __isl_give MULTI(BASE) *FN(MULTI(BASE),alloc)(__isl_take isl_space *space)
48 ctx = isl_space_get_ctx(space);
49 n = isl_space_dim(space, isl_dim_out);
50 multi = isl_calloc(ctx, MULTI(BASE),
51 sizeof(MULTI(BASE)) + (n - 1) * sizeof(struct EL *));
60 isl_space_free(space);
64 __isl_give MULTI(BASE) *FN(MULTI(BASE),dup)(__isl_keep MULTI(BASE) *multi)
72 dup = FN(MULTI(BASE),alloc)(isl_space_copy(multi->space));
76 for (i = 0; i < multi->n; ++i)
77 dup = FN(FN(MULTI(BASE),set),BASE)(dup, i,
78 FN(EL,copy)(multi->p[i]));
83 __isl_give MULTI(BASE) *FN(MULTI(BASE),cow)(__isl_take MULTI(BASE) *multi)
92 return FN(MULTI(BASE),dup)(multi);
95 __isl_give MULTI(BASE) *FN(MULTI(BASE),copy)(__isl_keep MULTI(BASE) *multi)
104 void *FN(MULTI(BASE),free)(__isl_take MULTI(BASE) *multi)
111 if (--multi->ref > 0)
114 isl_space_free(multi->space);
115 for (i = 0; i < multi->n; ++i)
116 FN(EL,free)(multi->p[i]);
122 __isl_give MULTI(BASE) *FN(MULTI(BASE),insert_dims)(
123 __isl_take MULTI(BASE) *multi,
124 enum isl_dim_type type, unsigned first, unsigned n)
130 if (type == isl_dim_out)
131 isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
132 "cannot insert output/set dimensions",
133 return FN(MULTI(BASE),free)(multi));
134 if (n == 0 && !isl_space_is_named_or_nested(multi->space, type))
137 multi = FN(MULTI(BASE),cow)(multi);
141 multi->space = isl_space_insert_dims(multi->space, type, first, n);
143 return FN(MULTI(BASE),free)(multi);
145 for (i = 0; i < multi->n; ++i) {
146 multi->p[i] = FN(EL,insert_dims)(multi->p[i], type, first, n);
148 return FN(MULTI(BASE),free)(multi);
154 __isl_give MULTI(BASE) *FN(MULTI(BASE),add_dims)(__isl_take MULTI(BASE) *multi,
155 enum isl_dim_type type, unsigned n)
159 pos = FN(MULTI(BASE),dim)(multi, type);
161 return FN(MULTI(BASE),insert_dims)(multi, type, pos, n);
164 unsigned FN(MULTI(BASE),dim)(__isl_keep MULTI(BASE) *multi,
165 enum isl_dim_type type)
167 return multi ? isl_space_dim(multi->space, type) : 0;
170 __isl_give MULTI(BASE) *FN(MULTI(BASE),set_dim_name)(
171 __isl_take MULTI(BASE) *multi,
172 enum isl_dim_type type, unsigned pos, const char *s)
176 multi = FN(MULTI(BASE),cow)(multi);
180 multi->space = isl_space_set_dim_name(multi->space, type, pos, s);
182 return FN(MULTI(BASE),free)(multi);
184 if (type == isl_dim_out)
186 for (i = 0; i < multi->n; ++i) {
187 multi->p[i] = FN(EL,set_dim_name)(multi->p[i], type, pos, s);
189 return FN(MULTI(BASE),free)(multi);
195 const char *FN(MULTI(BASE),get_tuple_name)(__isl_keep MULTI(BASE) *multi,
196 enum isl_dim_type type)
198 return multi ? isl_space_get_tuple_name(multi->space, type) : NULL;
201 __isl_give EL *FN(FN(MULTI(BASE),get),BASE)(__isl_keep MULTI(BASE) *multi,
208 ctx = FN(MULTI(BASE),get_ctx)(multi);
209 if (pos < 0 || pos >= multi->n)
210 isl_die(ctx, isl_error_invalid,
211 "index out of bounds", return NULL);
212 return FN(EL,copy)(multi->p[pos]);
215 __isl_give MULTI(BASE) *FN(FN(MULTI(BASE),set),BASE)(
216 __isl_take MULTI(BASE) *multi, int pos, __isl_take EL *el)
218 isl_space *multi_space = NULL;
219 isl_space *el_space = NULL;
221 multi = FN(MULTI(BASE),cow)(multi);
225 multi_space = FN(MULTI(BASE),get_space)(multi);
226 el_space = FN(EL,get_space)(el);
228 if (!isl_space_match(multi_space, isl_dim_param,
229 el_space, isl_dim_param))
230 isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
231 "parameters don't match", goto error);
232 if (!isl_space_tuple_match(multi_space, isl_dim_in,
233 el_space, isl_dim_in))
234 isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
235 "domains don't match", goto error);
237 if (pos < 0 || pos >= multi->n)
238 isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
239 "index out of bounds", goto error);
241 FN(EL,free)(multi->p[pos]);
244 isl_space_free(multi_space);
245 isl_space_free(el_space);
249 FN(MULTI(BASE),free)(multi);
251 isl_space_free(multi_space);
252 isl_space_free(el_space);
256 /* Reset the space of "multi". This function is called from isl_pw_templ.c
257 * and doesn't know if the space of an element object is represented
258 * directly or through its domain. It therefore passes along both,
259 * which we pass along to the element function since we don't how
260 * that is represented either.
262 __isl_give MULTI(BASE) *FN(MULTI(BASE),reset_space_and_domain)(
263 __isl_take MULTI(BASE) *multi, __isl_take isl_space *space,
264 __isl_take isl_space *domain)
268 multi = FN(MULTI(BASE),cow)(multi);
269 if (!multi || !space || !domain)
272 for (i = 0; i < multi->n; ++i) {
273 multi->p[i] = FN(EL,reset_domain_space)(multi->p[i],
274 isl_space_copy(domain));
278 isl_space_free(domain);
279 isl_space_free(multi->space);
280 multi->space = space;
284 isl_space_free(domain);
285 isl_space_free(space);
286 FN(MULTI(BASE),free)(multi);
290 __isl_give MULTI(BASE) *FN(MULTI(BASE),reset_domain_space)(
291 __isl_take MULTI(BASE) *multi, __isl_take isl_space *domain)
295 space = isl_space_extend_domain_with_range(isl_space_copy(domain),
296 isl_space_copy(multi->space));
297 return FN(MULTI(BASE),reset_space_and_domain)(multi, space, domain);
300 __isl_give MULTI(BASE) *FN(MULTI(BASE),reset_space)(
301 __isl_take MULTI(BASE) *multi, __isl_take isl_space *space)
305 domain = isl_space_domain(isl_space_copy(space));
306 return FN(MULTI(BASE),reset_space_and_domain)(multi, space, domain);
309 __isl_give MULTI(BASE) *FN(MULTI(BASE),set_tuple_id)(
310 __isl_keep MULTI(BASE) *multi, enum isl_dim_type type,
311 __isl_take isl_id *id)
315 multi = FN(MULTI(BASE),cow)(multi);
317 return isl_id_free(id);
319 space = FN(MULTI(BASE),get_space)(multi);
320 space = isl_space_set_tuple_id(space, type, id);
322 return FN(MULTI(BASE),reset_space)(multi, space);
325 __isl_give MULTI(BASE) *FN(MULTI(BASE),realign_domain)(
326 __isl_take MULTI(BASE) *multi, __isl_take isl_reordering *exp)
330 multi = FN(MULTI(BASE),cow)(multi);
334 for (i = 0; i < multi->n; ++i) {
335 multi->p[i] = FN(EL,realign_domain)(multi->p[i],
336 isl_reordering_copy(exp));
341 multi = FN(MULTI(BASE),reset_domain_space)(multi,
342 isl_space_copy(exp->dim));
344 isl_reordering_free(exp);
347 isl_reordering_free(exp);
348 FN(MULTI(BASE),free)(multi);
352 /* Align the parameters of "multi" to those of "model".
354 __isl_give MULTI(BASE) *FN(MULTI(BASE),align_params)(
355 __isl_take MULTI(BASE) *multi, __isl_take isl_space *model)
359 if (!multi || !model)
362 ctx = isl_space_get_ctx(model);
363 if (!isl_space_has_named_params(model))
364 isl_die(ctx, isl_error_invalid,
365 "model has unnamed parameters", goto error);
366 if (!isl_space_has_named_params(multi->space))
367 isl_die(ctx, isl_error_invalid,
368 "input has unnamed parameters", goto error);
369 if (!isl_space_match(multi->space, isl_dim_param,
370 model, isl_dim_param)) {
373 model = isl_space_params(model);
374 exp = isl_parameter_alignment_reordering(multi->space, model);
375 exp = isl_reordering_extend_space(exp,
376 FN(MULTI(BASE),get_domain_space)(multi));
377 multi = FN(MULTI(BASE),realign_domain)(multi, exp);
380 isl_space_free(model);
383 isl_space_free(model);
384 FN(MULTI(BASE),free)(multi);
388 static __isl_give MULTI(BASE) *FN(MULTI(BASE),align_params_multi_set_and)(
389 __isl_take MULTI(BASE) *multi, __isl_take isl_set *set,
390 __isl_give MULTI(BASE) *(*fn)(__isl_take MULTI(BASE) *multi,
391 __isl_take isl_set *set))
397 if (isl_space_match(multi->space, isl_dim_param,
398 set->dim, isl_dim_param))
399 return fn(multi, set);
400 ctx = FN(MULTI(BASE),get_ctx)(multi);
401 if (!isl_space_has_named_params(multi->space) ||
402 !isl_space_has_named_params(set->dim))
403 isl_die(ctx, isl_error_invalid,
404 "unaligned unnamed parameters", goto error);
405 multi = FN(MULTI(BASE),align_params)(multi, isl_set_get_space(set));
406 set = isl_set_align_params(set, FN(MULTI(BASE),get_space)(multi));
407 return fn(multi, set);
409 FN(MULTI(BASE),free)(multi);
414 __isl_give MULTI(BASE) *FN(MULTI(BASE),gist_aligned)(
415 __isl_take MULTI(BASE) *multi, __isl_take isl_set *context)
419 if (!multi || !context)
422 for (i = 0; i < multi->n; ++i) {
423 multi->p[i] = FN(EL,gist)(multi->p[i], isl_set_copy(context));
428 isl_set_free(context);
431 isl_set_free(context);
432 FN(MULTI(BASE),free)(multi);
436 __isl_give MULTI(BASE) *FN(MULTI(BASE),gist)(__isl_take MULTI(BASE) *multi,
437 __isl_take isl_set *context)
439 return FN(MULTI(BASE),align_params_multi_set_and)(multi, context,
440 &FN(MULTI(BASE),gist_aligned));
443 __isl_give MULTI(BASE) *FN(MULTI(BASE),gist_params)(
444 __isl_take MULTI(BASE) *multi, __isl_take isl_set *context)
446 isl_space *space = FN(MULTI(BASE),get_domain_space)(multi);
447 isl_set *dom_context = isl_set_universe(space);
448 dom_context = isl_set_intersect_params(dom_context, context);
449 return FN(MULTI(BASE),gist)(multi, dom_context);
452 __isl_give MULTI(BASE) *FN(FN(MULTI(BASE),from),LIST(BASE))(
453 __isl_take isl_space *space, __isl_take LIST(EL) *list)
463 ctx = isl_space_get_ctx(space);
464 n = FN(FN(LIST(EL),n),BASE)(list);
465 if (n != isl_space_dim(space, isl_dim_out))
466 isl_die(ctx, isl_error_invalid,
467 "invalid number of elements in list", goto error);
469 multi = FN(MULTI(BASE),alloc)(isl_space_copy(space));
470 for (i = 0; i < n; ++i) {
471 multi = FN(FN(MULTI(BASE),set),BASE)(multi, i,
472 FN(FN(LIST(EL),get),BASE)(list, i));
475 isl_space_free(space);
476 FN(LIST(EL),free)(list);
479 isl_space_free(space);
480 FN(LIST(EL),free)(list);
484 /* Create a multi expression in the given space that maps each
485 * input dimension to the corresponding output dimension.
487 __isl_give MULTI(BASE) *FN(MULTI(BASE),identity)(__isl_take isl_space *space)
496 if (isl_space_is_set(space))
497 isl_die(isl_space_get_ctx(space), isl_error_invalid,
498 "expecting map space", goto error);
500 n = isl_space_dim(space, isl_dim_out);
501 if (n != isl_space_dim(space, isl_dim_in))
502 isl_die(isl_space_get_ctx(space), isl_error_invalid,
503 "number of input and output dimensions needs to be "
504 "the same", goto error);
506 multi = FN(MULTI(BASE),alloc)(isl_space_copy(space));
509 isl_space_free(space);
513 space = isl_space_domain(space);
514 ls = isl_local_space_from_space(space);
516 for (i = 0; i < n; ++i) {
518 el = FN(EL,var_on_domain)(isl_local_space_copy(ls),
520 multi = FN(FN(MULTI(BASE),set),BASE)(multi, i, el);
523 isl_local_space_free(ls);
527 isl_space_free(space);
531 /* Construct a multi expression in the given space with value zero in
532 * each of the output dimensions.
534 __isl_give MULTI(BASE) *FN(MULTI(BASE),zero)(__isl_take isl_space *space)
542 n = isl_space_dim(space , isl_dim_out);
543 multi = FN(MULTI(BASE),alloc)(isl_space_copy(space));
546 isl_space_free(space);
552 space = isl_space_domain(space);
553 ls = isl_local_space_from_space(space);
554 el = FN(EL,zero_on_domain)(ls);
556 for (i = 0; i < n; ++i)
557 multi = FN(FN(MULTI(BASE),set),BASE)(multi, i,
566 __isl_give MULTI(BASE) *FN(FN(MULTI(BASE),from),BASE)(__isl_take EL *el)
570 multi = FN(MULTI(BASE),alloc)(FN(EL,get_space)(el));
571 multi = FN(FN(MULTI(BASE),set),BASE)(multi, 0, el);
576 __isl_give MULTI(BASE) *FN(MULTI(BASE),drop_dims)(
577 __isl_take MULTI(BASE) *multi,
578 enum isl_dim_type type, unsigned first, unsigned n)
583 multi = FN(MULTI(BASE),cow)(multi);
587 dim = FN(MULTI(BASE),dim)(multi, type);
588 if (first + n > dim || first + n < first)
589 isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
590 "index out of bounds",
591 return FN(MULTI(BASE),cow)(multi));
593 multi->space = isl_space_drop_dims(multi->space, type, first, n);
595 return FN(MULTI(BASE),cow)(multi);
597 if (type == isl_dim_out) {
598 for (i = 0; i < n; ++i)
599 FN(EL,free)(multi->p[first + i]);
600 for (i = first; i + n < multi->n; ++i)
601 multi->p[i] = multi->p[i + n];
607 for (i = 0; i < multi->n; ++i) {
608 multi->p[i] = FN(EL,drop_dims)(multi->p[i], type, first, n);
610 return FN(MULTI(BASE),cow)(multi);
616 /* Given two MULTI(BASE)s A -> B and C -> D,
617 * construct a MULTI(BASE) (A * C) -> (B, D).
619 __isl_give MULTI(BASE) *FN(MULTI(BASE),range_product)(
620 __isl_take MULTI(BASE) *multi1, __isl_take MULTI(BASE) *multi2)
627 if (!multi1 || !multi2)
630 space = isl_space_range_product(FN(MULTI(BASE),get_space)(multi1),
631 FN(MULTI(BASE),get_space)(multi2));
632 res = FN(MULTI(BASE),alloc)(space);
634 n1 = FN(MULTI(BASE),dim)(multi1, isl_dim_out);
635 n2 = FN(MULTI(BASE),dim)(multi2, isl_dim_out);
637 for (i = 0; i < n1; ++i) {
638 el = FN(FN(MULTI(BASE),get),BASE)(multi1, i);
639 res = FN(FN(MULTI(BASE),set),BASE)(res, i, el);
642 for (i = 0; i < n2; ++i) {
643 el = FN(FN(MULTI(BASE),get),BASE)(multi2, i);
644 res = FN(FN(MULTI(BASE),set),BASE)(res, n1 + i, el);
647 FN(MULTI(BASE),free)(multi1);
648 FN(MULTI(BASE),free)(multi2);
651 FN(MULTI(BASE),free)(multi1);
652 FN(MULTI(BASE),free)(multi2);
656 __isl_give MULTI(BASE) *FN(MULTI(BASE),flatten_range)(
657 __isl_take MULTI(BASE) *multi)
662 if (!multi->space->nested[1])
665 multi = FN(MULTI(BASE),cow)(multi);
669 multi->space = isl_space_flatten_range(multi->space);
671 return FN(MULTI(BASE),free)(multi);
676 /* Given two MULTI(BASE)s A -> B and C -> D,
677 * construct a MULTI(BASE) (A * C) -> [B -> D].
679 __isl_give MULTI(BASE) *FN(MULTI(BASE),flat_range_product)(
680 __isl_take MULTI(BASE) *multi1, __isl_take MULTI(BASE) *multi2)
684 multi = FN(MULTI(BASE),range_product)(multi1, multi2);
685 multi = FN(MULTI(BASE),flatten_range)(multi);