isl_space.c

   1 /*
   2  * Copyright 2008-2009 Katholieke Universiteit Leuven
   3  * Copyright 2010      INRIA Saclay
   4  *
   5  * Use of this software is governed by the GNU LGPLv2.1 license
   6  *
   7  * Written by Sven Verdoolaege, K.U.Leuven, Departement
   8  * Computerwetenschappen, Celestijnenlaan 200A, B-3001 Leuven, Belgium
   9  * and INRIA Saclay - Ile-de-France, Parc Club Orsay Universite,
  10  * ZAC des vignes, 4 rue Jacques Monod, 91893 Orsay, France
  11  */
  12
  13 #include <stdlib.h>
  14 #include <isl_space_private.h>
  15 #include <isl_id_private.h>
  16 #include <isl_reordering.h>
  17
  18 isl_ctx *isl_space_get_ctx(__isl_keep isl_space *dim)
  19 {
  20         return dim ? dim->ctx : NULL;
  21 }
  22
  23 __isl_give isl_space *isl_space_alloc(isl_ctx *ctx,
  24                         unsigned nparam, unsigned n_in, unsigned n_out)
  25 {
  26         isl_space *dim;
  27
  28         dim = isl_alloc_type(ctx, struct isl_space);
  29         if (!dim)
  30                 return NULL;
  31
  32         dim->ctx = ctx;
  33         isl_ctx_ref(ctx);
  34         dim->ref = 1;
  35         dim->nparam = nparam;
  36         dim->n_in = n_in;
  37         dim->n_out = n_out;
  38
  39         dim->tuple_id[0] = NULL;
  40         dim->tuple_id[1] = NULL;
  41
  42         dim->nested[0] = NULL;
  43         dim->nested[1] = NULL;
  44
  45         dim->n_id = 0;
  46         dim->ids = NULL;
  47
  48         return dim;
  49 }
  50
  51 __isl_give isl_space *isl_space_set_alloc(isl_ctx *ctx,
  52                         unsigned nparam, unsigned dim)
  53 {
  54         return isl_space_alloc(ctx, nparam, 0, dim);
  55 }
  56
  57 static unsigned global_pos(__isl_keep isl_space *dim,
  58                                  enum isl_dim_type type, unsigned pos)
  59 {
  60         struct isl_ctx *ctx = dim->ctx;
  61
  62         switch (type) {
  63         case isl_dim_param:
  64                 isl_assert(ctx, pos < dim->nparam,
  65                             return isl_space_dim(dim, isl_dim_all));
  66                 return pos;
  67         case isl_dim_in:
  68                 isl_assert(ctx, pos < dim->n_in,
  69                             return isl_space_dim(dim, isl_dim_all));
  70                 return pos + dim->nparam;
  71         case isl_dim_out:
  72                 isl_assert(ctx, pos < dim->n_out,
  73                             return isl_space_dim(dim, isl_dim_all));
  74                 return pos + dim->nparam + dim->n_in;
  75         default:
  76                 isl_assert(ctx, 0, return isl_space_dim(dim, isl_dim_all));
  77         }
  78         return isl_space_dim(dim, isl_dim_all);
  79 }
  80
  81 /* Extend length of ids array to the total number of dimensions.
  82  */
  83 static __isl_give isl_space *extend_ids(__isl_take isl_space *dim)
  84 {
  85         isl_id **ids;
  86         int i;
  87
  88         if (isl_space_dim(dim, isl_dim_all) <= dim->n_id)
  89                 return dim;
  90
  91         if (!dim->ids) {
  92                 dim->ids = isl_calloc_array(dim->ctx,
  93                                 isl_id *, isl_space_dim(dim, isl_dim_all));
  94                 if (!dim->ids)
  95                         goto error;
  96         } else {
  97                 ids = isl_realloc_array(dim->ctx, dim->ids,
  98                                 isl_id *, isl_space_dim(dim, isl_dim_all));
  99                 if (!ids)
 100                         goto error;
 101                 dim->ids = ids;
 102                 for (i = dim->n_id; i < isl_space_dim(dim, isl_dim_all); ++i)
 103                         dim->ids[i] = NULL;
 104         }
 105
 106         dim->n_id = isl_space_dim(dim, isl_dim_all);
 107
 108         return dim;
 109 error:
 110         isl_space_free(dim);
 111         return NULL;
 112 }
 113
 114 static __isl_give isl_space *set_id(__isl_take isl_space *dim,
 115         enum isl_dim_type type, unsigned pos, __isl_take isl_id *id)
 116 {
 117         dim = isl_space_cow(dim);
 118
 119         if (!dim)
 120                 goto error;
 121
 122         pos = global_pos(dim, type, pos);
 123         if (pos == isl_space_dim(dim, isl_dim_all))
 124                 goto error;
 125
 126         if (pos >= dim->n_id) {
 127                 if (!id)
 128                         return dim;
 129                 dim = extend_ids(dim);
 130                 if (!dim)
 131                         goto error;
 132         }
 133
 134         dim->ids[pos] = id;
 135
 136         return dim;
 137 error:
 138         isl_id_free(id);
 139         isl_space_free(dim);
 140         return NULL;
 141 }
 142
 143 static __isl_keep isl_id *get_id(__isl_keep isl_space *dim,
 144                                  enum isl_dim_type type, unsigned pos)
 145 {
 146         if (!dim)
 147                 return NULL;
 148
 149         pos = global_pos(dim, type, pos);
 150         if (pos == isl_space_dim(dim, isl_dim_all))
 151                 return NULL;
 152         if (pos >= dim->n_id)
 153                 return NULL;
 154         return dim->ids[pos];
 155 }
 156
 157 static unsigned offset(__isl_keep isl_space *dim, enum isl_dim_type type)
 158 {
 159         switch (type) {
 160         case isl_dim_param:     return 0;
 161         case isl_dim_in:        return dim->nparam;
 162         case isl_dim_out:       return dim->nparam + dim->n_in;
 163         default:                return 0;
 164         }
 165 }
 166
 167 static unsigned n(__isl_keep isl_space *dim, enum isl_dim_type type)
 168 {
 169         switch (type) {
 170         case isl_dim_param:     return dim->nparam;
 171         case isl_dim_in:        return dim->n_in;
 172         case isl_dim_out:       return dim->n_out;
 173         case isl_dim_all:       return dim->nparam + dim->n_in + dim->n_out;
 174         default:                return 0;
 175         }
 176 }
 177
 178 unsigned isl_space_dim(__isl_keep isl_space *dim, enum isl_dim_type type)
 179 {
 180         if (!dim)
 181                 return 0;
 182         return n(dim, type);
 183 }
 184
 185 unsigned isl_space_offset(__isl_keep isl_space *dim, enum isl_dim_type type)
 186 {
 187         if (!dim)
 188                 return 0;
 189         return offset(dim, type);
 190 }
 191
 192 static __isl_give isl_space *copy_ids(__isl_take isl_space *dst,
 193         enum isl_dim_type dst_type, unsigned offset, __isl_keep isl_space *src,
 194         enum isl_dim_type src_type)
 195 {
 196         int i;
 197         isl_id *id;
 198
 199         if (!dst)
 200                 return NULL;
 201
 202         for (i = 0; i < n(src, src_type); ++i) {
 203                 id = get_id(src, src_type, i);
 204                 if (!id)
 205                         continue;
 206                 dst = set_id(dst, dst_type, offset + i, isl_id_copy(id));
 207                 if (!dst)
 208                         return NULL;
 209         }
 210         return dst;
 211 }
 212
 213 __isl_take isl_space *isl_space_dup(__isl_keep isl_space *dim)
 214 {
 215         isl_space *dup;
 216         if (!dim)
 217                 return NULL;
 218         dup = isl_space_alloc(dim->ctx, dim->nparam, dim->n_in, dim->n_out);
 219         if (dim->tuple_id[0] &&
 220             !(dup->tuple_id[0] = isl_id_copy(dim->tuple_id[0])))
 221                 goto error;
 222         if (dim->tuple_id[1] &&
 223             !(dup->tuple_id[1] = isl_id_copy(dim->tuple_id[1])))
 224                 goto error;
 225         if (dim->nested[0] && !(dup->nested[0] = isl_space_copy(dim->nested[0])))
 226                 goto error;
 227         if (dim->nested[1] && !(dup->nested[1] = isl_space_copy(dim->nested[1])))
 228                 goto error;
 229         if (!dim->ids)
 230                 return dup;
 231         dup = copy_ids(dup, isl_dim_param, 0, dim, isl_dim_param);
 232         dup = copy_ids(dup, isl_dim_in, 0, dim, isl_dim_in);
 233         dup = copy_ids(dup, isl_dim_out, 0, dim, isl_dim_out);
 234         return dup;
 235 error:
 236         isl_space_free(dup);
 237         return NULL;
 238 }
 239
 240 __isl_give isl_space *isl_space_cow(__isl_take isl_space *dim)
 241 {
 242         if (!dim)
 243                 return NULL;
 244
 245         if (dim->ref == 1)
 246                 return dim;
 247         dim->ref--;
 248         return isl_space_dup(dim);
 249 }
 250
 251 __isl_give isl_space *isl_space_copy(__isl_keep isl_space *dim)
 252 {
 253         if (!dim)
 254                 return NULL;
 255
 256         dim->ref++;
 257         return dim;
 258 }
 259
 260 void isl_space_free(__isl_take isl_space *dim)
 261 {
 262         int i;
 263
 264         if (!dim)
 265                 return;
 266
 267         if (--dim->ref > 0)
 268                 return;
 269
 270         isl_id_free(dim->tuple_id[0]);
 271         isl_id_free(dim->tuple_id[1]);
 272
 273         isl_space_free(dim->nested[0]);
 274         isl_space_free(dim->nested[1]);
 275
 276         for (i = 0; i < dim->n_id; ++i)
 277                 isl_id_free(dim->ids[i]);
 278         free(dim->ids);
 279         isl_ctx_deref(dim->ctx);
 280
 281         free(dim);
 282 }
 283
 284 static int name_ok(isl_ctx *ctx, const char *s)
 285 {
 286         char *p;
 287         long dummy;
 288
 289         dummy = strtol(s, &p, 0);
 290         if (p != s)
 291                 isl_die(ctx, isl_error_invalid, "name looks like a number",
 292                         return 0);
 293
 294         return 1;
 295 }
 296
 297 int isl_space_has_tuple_id(__isl_keep isl_space *dim, enum isl_dim_type type)
 298 {
 299         if (!dim)
 300                 return -1;
 301         if (type != isl_dim_in && type != isl_dim_out)
 302                 isl_die(dim->ctx, isl_error_invalid,
 303                         "only input, output and set tuples can have ids",
 304                         return -1);
 305         return dim->tuple_id[type - isl_dim_in] != NULL;
 306 }
 307
 308 __isl_give isl_id *isl_space_get_tuple_id(__isl_keep isl_space *dim,
 309         enum isl_dim_type type)
 310 {
 311         if (!dim)
 312                 return NULL;
 313         if (type != isl_dim_in && type != isl_dim_out)
 314                 isl_die(dim->ctx, isl_error_invalid,
 315                         "only input, output and set tuples can have ids",
 316                         return NULL);
 317         if (!dim->tuple_id[type - isl_dim_in])
 318                 isl_die(dim->ctx, isl_error_invalid,
 319                         "tuple has no id", return NULL);
 320         return isl_id_copy(dim->tuple_id[type - isl_dim_in]);
 321 }
 322
 323 __isl_give isl_space *isl_space_set_tuple_id(__isl_take isl_space *dim,
 324         enum isl_dim_type type, __isl_take isl_id *id)
 325 {
 326         dim = isl_space_cow(dim);
 327         if (!dim || !id)
 328                 goto error;
 329         if (type != isl_dim_in && type != isl_dim_out)
 330                 isl_die(dim->ctx, isl_error_invalid,
 331                         "only input, output and set tuples can have names",
 332                         goto error);
 333
 334         isl_id_free(dim->tuple_id[type - isl_dim_in]);
 335         dim->tuple_id[type - isl_dim_in] = id;
 336
 337         return dim;
 338 error:
 339         isl_id_free(id);
 340         isl_space_free(dim);
 341         return NULL;
 342 }
 343
 344 __isl_give isl_space *isl_space_reset_tuple_id(__isl_take isl_space *dim,
 345         enum isl_dim_type type)
 346 {
 347         dim = isl_space_cow(dim);
 348         if (!dim)
 349                 return NULL;
 350         if (type != isl_dim_in && type != isl_dim_out)
 351                 isl_die(dim->ctx, isl_error_invalid,
 352                         "only input, output and set tuples can have names",
 353                         goto error);
 354
 355         isl_id_free(dim->tuple_id[type - isl_dim_in]);
 356         dim->tuple_id[type - isl_dim_in] = NULL;
 357
 358         return dim;
 359 error:
 360         isl_space_free(dim);
 361         return NULL;
 362 }
 363
 364 __isl_give isl_space *isl_space_set_dim_id(__isl_take isl_space *dim,
 365         enum isl_dim_type type, unsigned pos, __isl_take isl_id *id)
 366 {
 367         dim = isl_space_cow(dim);
 368         if (!dim || !id)
 369                 goto error;
 370         isl_id_free(get_id(dim, type, pos));
 371         return set_id(dim, type, pos, id);
 372 error:
 373         isl_id_free(id);
 374         isl_space_free(dim);
 375         return NULL;
 376 }
 377
 378 int isl_space_has_dim_id(__isl_keep isl_space *dim,
 379         enum isl_dim_type type, unsigned pos)
 380 {
 381         if (!dim)
 382                 return -1;
 383         return get_id(dim, type, pos) != NULL;
 384 }
 385
 386 __isl_give isl_id *isl_space_get_dim_id(__isl_keep isl_space *dim,
 387         enum isl_dim_type type, unsigned pos)
 388 {
 389         if (!dim)
 390                 return NULL;
 391         if (!get_id(dim, type, pos))
 392                 isl_die(dim->ctx, isl_error_invalid,
 393                         "dim has no id", return NULL);
 394         return isl_id_copy(get_id(dim, type, pos));
 395 }
 396
 397 __isl_give isl_space *isl_space_set_tuple_name(__isl_take isl_space *dim,
 398         enum isl_dim_type type, const char *s)
 399 {
 400         isl_id *id;
 401
 402         if (!dim)
 403                 return NULL;
 404
 405         if (!s)
 406                 return isl_space_reset_tuple_id(dim, type);
 407
 408         if (!name_ok(dim->ctx, s))
 409                 goto error;
 410
 411         id = isl_id_alloc(dim->ctx, s, NULL);
 412         return isl_space_set_tuple_id(dim, type, id);
 413 error:
 414         isl_space_free(dim);
 415         return NULL;
 416 }
 417
 418 const char *isl_space_get_tuple_name(__isl_keep isl_space *dim,
 419          enum isl_dim_type type)
 420 {
 421         isl_id *id;
 422         if (!dim)
 423                 return NULL;
 424         if (type != isl_dim_in && type != isl_dim_out)
 425                 return NULL;
 426         id = dim->tuple_id[type - isl_dim_in];
 427         return id ? id->name : NULL;
 428 }
 429
 430 __isl_give isl_space *isl_space_set_dim_name(__isl_take isl_space *dim,
 431                                  enum isl_dim_type type, unsigned pos,
 432                                  const char *s)
 433 {
 434         isl_id *id;
 435
 436         if (!dim)
 437                 return NULL;
 438         if (!name_ok(dim->ctx, s))
 439                 goto error;
 440         id = isl_id_alloc(dim->ctx, s, NULL);
 441         return isl_space_set_dim_id(dim, type, pos, id);
 442 error:
 443         isl_space_free(dim);
 444         return NULL;
 445 }
 446
 447 __isl_keep const char *isl_space_get_dim_name(__isl_keep isl_space *dim,
 448                                  enum isl_dim_type type, unsigned pos)
 449 {
 450         isl_id *id = get_id(dim, type, pos);
 451         return id ? id->name : NULL;
 452 }
 453
 454 int isl_space_find_dim_by_id(__isl_keep isl_space *dim, enum isl_dim_type type,
 455         __isl_keep isl_id *id)
 456 {
 457         int i;
 458         int offset;
 459         int n;
 460
 461         if (!dim || !id)
 462                 return -1;
 463
 464         offset = isl_space_offset(dim, type);
 465         n = isl_space_dim(dim, type);
 466         for (i = 0; i < n && offset + i < dim->n_id; ++i)
 467                 if (dim->ids[offset + i] == id)
 468                         return i;
 469
 470         return -1;
 471 }
 472
 473 static __isl_keep isl_id *tuple_id(__isl_keep isl_space *dim,
 474         enum isl_dim_type type)
 475 {
 476         if (!dim)
 477                 return NULL;
 478         if (type == isl_dim_in)
 479                 return dim->tuple_id[0];
 480         if (type == isl_dim_out)
 481                 return dim->tuple_id[1];
 482         return NULL;
 483 }
 484
 485 static __isl_keep isl_space *nested(__isl_keep isl_space *dim,
 486         enum isl_dim_type type)
 487 {
 488         if (!dim)
 489                 return NULL;
 490         if (type == isl_dim_in)
 491                 return dim->nested[0];
 492         if (type == isl_dim_out)
 493                 return dim->nested[1];
 494         return NULL;
 495 }
 496
 497 int isl_space_tuple_match(__isl_keep isl_space *dim1, enum isl_dim_type dim1_type,
 498                         __isl_keep isl_space *dim2, enum isl_dim_type dim2_type)
 499 {
 500         isl_id *id1, *id2;
 501         isl_space *nested1, *nested2;
 502
 503         if (!dim1 || !dim2)
 504                 return -1;
 505
 506         if (n(dim1, dim1_type) != n(dim2, dim2_type))
 507                 return 0;
 508         id1 = tuple_id(dim1, dim1_type);
 509         id2 = tuple_id(dim2, dim2_type);
 510         if (!id1 ^ !id2)
 511                 return 0;
 512         if (id1 && id1 != id2)
 513                 return 0;
 514         nested1 = nested(dim1, dim1_type);
 515         nested2 = nested(dim2, dim2_type);
 516         if (!nested1 ^ !nested2)
 517                 return 0;
 518         if (nested1 && !isl_space_is_equal(nested1, nested2))
 519                 return 0;
 520         return 1;
 521 }
 522
 523 static int match(__isl_keep isl_space *dim1, enum isl_dim_type dim1_type,
 524         __isl_keep isl_space *dim2, enum isl_dim_type dim2_type)
 525 {
 526         int i;
 527
 528         if (!isl_space_tuple_match(dim1, dim1_type, dim2, dim2_type))
 529                 return 0;
 530
 531         if (!dim1->ids && !dim2->ids)
 532                 return 1;
 533
 534         for (i = 0; i < n(dim1, dim1_type); ++i) {
 535                 if (get_id(dim1, dim1_type, i) != get_id(dim2, dim2_type, i))
 536                         return 0;
 537         }
 538         return 1;
 539 }
 540
 541 int isl_space_match(__isl_keep isl_space *dim1, enum isl_dim_type dim1_type,
 542         __isl_keep isl_space *dim2, enum isl_dim_type dim2_type)
 543 {
 544         return match(dim1, dim1_type, dim2, dim2_type);
 545 }
 546
 547 static void get_ids(__isl_keep isl_space *dim, enum isl_dim_type type,
 548         unsigned first, unsigned n, __isl_keep isl_id **ids)
 549 {
 550         int i;
 551
 552         for (i = 0; i < n ; ++i)
 553                 ids[i] = get_id(dim, type, first + i);
 554 }
 555
 556 __isl_give isl_space *isl_space_extend(__isl_take isl_space *dim,
 557                         unsigned nparam, unsigned n_in, unsigned n_out)
 558 {
 559         isl_id **ids = NULL;
 560
 561         if (!dim)
 562                 return NULL;
 563         if (dim->nparam == nparam && dim->n_in == n_in && dim->n_out == n_out)
 564                 return dim;
 565
 566         isl_assert(dim->ctx, dim->nparam <= nparam, goto error);
 567         isl_assert(dim->ctx, dim->n_in <= n_in, goto error);
 568         isl_assert(dim->ctx, dim->n_out <= n_out, goto error);
 569
 570         dim = isl_space_cow(dim);
 571
 572         if (dim->ids) {
 573                 ids = isl_calloc_array(dim->ctx, isl_id *,
 574                                          nparam + n_in + n_out);
 575                 if (!ids)
 576                         goto error;
 577                 get_ids(dim, isl_dim_param, 0, dim->nparam, ids);
 578                 get_ids(dim, isl_dim_in, 0, dim->n_in, ids + nparam);
 579                 get_ids(dim, isl_dim_out, 0, dim->n_out, ids + nparam + n_in);
 580                 free(dim->ids);
 581                 dim->ids = ids;
 582                 dim->n_id = nparam + n_in + n_out;
 583         }
 584         dim->nparam = nparam;
 585         dim->n_in = n_in;
 586         dim->n_out = n_out;
 587
 588         return dim;
 589 error:
 590         free(ids);
 591         isl_space_free(dim);
 592         return NULL;
 593 }
 594
 595 __isl_give isl_space *isl_space_add_dims(__isl_take isl_space *dim,
 596         enum isl_dim_type type, unsigned n)
 597 {
 598         if (!dim)
 599                 return NULL;
 600         dim = isl_space_reset(dim, type);
 601         switch (type) {
 602         case isl_dim_param:
 603                 dim = isl_space_extend(dim,
 604                                         dim->nparam + n, dim->n_in, dim->n_out);
 605                 if (dim && dim->nested[0] &&
 606                     !(dim->nested[0] = isl_space_add_dims(dim->nested[0],
 607                                                     isl_dim_param, n)))
 608                         goto error;
 609                 if (dim && dim->nested[1] &&
 610                     !(dim->nested[1] = isl_space_add_dims(dim->nested[1],
 611                                                     isl_dim_param, n)))
 612                         goto error;
 613                 return dim;
 614         case isl_dim_in:
 615                 return isl_space_extend(dim,
 616                                         dim->nparam, dim->n_in + n, dim->n_out);
 617         case isl_dim_out:
 618                 return isl_space_extend(dim,
 619                                         dim->nparam, dim->n_in, dim->n_out + n);
 620         default:
 621                 isl_die(dim->ctx, isl_error_invalid,
 622                         "cannot add dimensions of specified type", goto error);
 623         }
 624 error:
 625         isl_space_free(dim);
 626         return NULL;
 627 }
 628
 629 static int valid_dim_type(enum isl_dim_type type)
 630 {
 631         switch (type) {
 632         case isl_dim_param:
 633         case isl_dim_in:
 634         case isl_dim_out:
 635                 return 1;
 636         default:
 637                 return 0;
 638         }
 639 }
 640
 641 __isl_give isl_space *isl_space_insert_dims(__isl_take isl_space *dim,
 642         enum isl_dim_type type, unsigned pos, unsigned n)
 643 {
 644         isl_id **ids = NULL;
 645
 646         if (!dim)
 647                 return NULL;
 648         if (n == 0)
 649                 return isl_space_reset(dim, type);
 650
 651         if (!valid_dim_type(type))
 652                 isl_die(dim->ctx, isl_error_invalid,
 653                         "cannot insert dimensions of specified type",
 654                         goto error);
 655
 656         isl_assert(dim->ctx, pos <= isl_space_dim(dim, type), goto error);
 657
 658         dim = isl_space_cow(dim);
 659         if (!dim)
 660                 return NULL;
 661
 662         if (dim->ids) {
 663                 enum isl_dim_type t;
 664                 int off;
 665                 int s[3];
 666                 int *size = s - isl_dim_param;
 667                 ids = isl_calloc_array(dim->ctx, isl_id *,
 668                                      dim->nparam + dim->n_in + dim->n_out + n);
 669                 if (!ids)
 670                         goto error;
 671                 off = 0;
 672                 size[isl_dim_param] = dim->nparam;
 673                 size[isl_dim_in] = dim->n_in;
 674                 size[isl_dim_out] = dim->n_out;
 675                 for (t = isl_dim_param; t <= isl_dim_out; ++t) {
 676                         if (t != type) {
 677                                 get_ids(dim, t, 0, size[t], ids + off);
 678                                 off += size[t];
 679                         } else {
 680                                 get_ids(dim, t, 0, pos, ids + off);
 681                                 off += pos + n;
 682                                 get_ids(dim, t, pos, size[t] - pos, ids + off);
 683                                 off += size[t] - pos;
 684                         }
 685                 }
 686                 free(dim->ids);
 687                 dim->ids = ids;
 688                 dim->n_id = dim->nparam + dim->n_in + dim->n_out + n;
 689         }
 690         switch (type) {
 691         case isl_dim_param:     dim->nparam += n; break;
 692         case isl_dim_in:        dim->n_in += n; break;
 693         case isl_dim_out:       dim->n_out += n; break;
 694         default:                ;
 695         }
 696         dim = isl_space_reset(dim, type);
 697
 698         return dim;
 699 error:
 700         isl_space_free(dim);
 701         return NULL;
 702 }
 703
 704 __isl_give isl_space *isl_space_move_dims(__isl_take isl_space *dim,
 705         enum isl_dim_type dst_type, unsigned dst_pos,
 706         enum isl_dim_type src_type, unsigned src_pos, unsigned n)
 707 {
 708         int i;
 709
 710         if (!dim)
 711                 return NULL;
 712         if (n == 0)
 713                 return dim;
 714
 715         isl_assert(dim->ctx, src_pos + n <= isl_space_dim(dim, src_type),
 716                 goto error);
 717
 718         if (dst_type == src_type && dst_pos == src_pos)
 719                 return dim;
 720
 721         isl_assert(dim->ctx, dst_type != src_type, goto error);
 722
 723         dim = isl_space_reset(dim, src_type);
 724         dim = isl_space_reset(dim, dst_type);
 725
 726         dim = isl_space_cow(dim);
 727         if (!dim)
 728                 return NULL;
 729
 730         if (dim->ids) {
 731                 isl_id **ids;
 732                 enum isl_dim_type t;
 733                 int off;
 734                 int s[3];
 735                 int *size = s - isl_dim_param;
 736                 ids = isl_calloc_array(dim->ctx, isl_id *,
 737                                          dim->nparam + dim->n_in + dim->n_out);
 738                 if (!ids)
 739                         goto error;
 740                 off = 0;
 741                 size[isl_dim_param] = dim->nparam;
 742                 size[isl_dim_in] = dim->n_in;
 743                 size[isl_dim_out] = dim->n_out;
 744                 for (t = isl_dim_param; t <= isl_dim_out; ++t) {
 745                         if (t == dst_type) {
 746                                 get_ids(dim, t, 0, dst_pos, ids + off);
 747                                 off += dst_pos;
 748                                 get_ids(dim, src_type, src_pos, n, ids + off);
 749                                 off += n;
 750                                 get_ids(dim, t, dst_pos, size[t] - dst_pos,
 751                                                 ids + off);
 752                                 off += size[t] - dst_pos;
 753                         } else if (t == src_type) {
 754                                 get_ids(dim, t, 0, src_pos, ids + off);
 755                                 off += src_pos;
 756                                 get_ids(dim, t, src_pos + n,
 757                                             size[t] - src_pos - n, ids + off);
 758                                 off += size[t] - src_pos - n;
 759                         } else {
 760                                 get_ids(dim, t, 0, size[t], ids + off);
 761                                 off += size[t];
 762                         }
 763                 }
 764                 free(dim->ids);
 765                 dim->ids = ids;
 766                 dim->n_id = dim->nparam + dim->n_in + dim->n_out;
 767         }
 768
 769         switch (dst_type) {
 770         case isl_dim_param:     dim->nparam += n; break;
 771         case isl_dim_in:        dim->n_in += n; break;
 772         case isl_dim_out:       dim->n_out += n; break;
 773         default:                ;
 774         }
 775
 776         switch (src_type) {
 777         case isl_dim_param:     dim->nparam -= n; break;
 778         case isl_dim_in:        dim->n_in -= n; break;
 779         case isl_dim_out:       dim->n_out -= n; break;
 780         default:                ;
 781         }
 782
 783         if (dst_type != isl_dim_param && src_type != isl_dim_param)
 784                 return dim;
 785
 786         for (i = 0; i < 2; ++i) {
 787                 if (!dim->nested[i])
 788                         continue;
 789                 dim->nested[i] = isl_space_replace(dim->nested[i],
 790                                                  isl_dim_param, dim);
 791                 if (!dim->nested[i])
 792                         goto error;
 793         }
 794
 795         return dim;
 796 error:
 797         isl_space_free(dim);
 798         return NULL;
 799 }
 800
 801 __isl_give isl_space *isl_space_join(__isl_take isl_space *left,
 802         __isl_take isl_space *right)
 803 {
 804         isl_space *dim;
 805
 806         if (!left || !right)
 807                 goto error;
 808
 809         isl_assert(left->ctx, match(left, isl_dim_param, right, isl_dim_param),
 810                         goto error);
 811         isl_assert(left->ctx,
 812                 isl_space_tuple_match(left, isl_dim_out, right, isl_dim_in),
 813                 goto error);
 814
 815         dim = isl_space_alloc(left->ctx, left->nparam, left->n_in, right->n_out);
 816         if (!dim)
 817                 goto error;
 818
 819         dim = copy_ids(dim, isl_dim_param, 0, left, isl_dim_param);
 820         dim = copy_ids(dim, isl_dim_in, 0, left, isl_dim_in);
 821         dim = copy_ids(dim, isl_dim_out, 0, right, isl_dim_out);
 822
 823         if (dim && left->tuple_id[0] &&
 824             !(dim->tuple_id[0] = isl_id_copy(left->tuple_id[0])))
 825                 goto error;
 826         if (dim && right->tuple_id[1] &&
 827             !(dim->tuple_id[1] = isl_id_copy(right->tuple_id[1])))
 828                 goto error;
 829         if (dim && left->nested[0] &&
 830             !(dim->nested[0] = isl_space_copy(left->nested[0])))
 831                 goto error;
 832         if (dim && right->nested[1] &&
 833             !(dim->nested[1] = isl_space_copy(right->nested[1])))
 834                 goto error;
 835
 836         isl_space_free(left);
 837         isl_space_free(right);
 838
 839         return dim;
 840 error:
 841         isl_space_free(left);
 842         isl_space_free(right);
 843         return NULL;
 844 }
 845
 846 __isl_give isl_space *isl_space_product(__isl_take isl_space *left,
 847         __isl_take isl_space *right)
 848 {
 849         isl_space *dom1, *dom2, *nest1, *nest2;
 850
 851         if (!left || !right)
 852                 goto error;
 853
 854         isl_assert(left->ctx, match(left, isl_dim_param, right, isl_dim_param),
 855                         goto error);
 856
 857         dom1 = isl_space_domain(isl_space_copy(left));
 858         dom2 = isl_space_domain(isl_space_copy(right));
 859         nest1 = isl_space_wrap(isl_space_join(isl_space_reverse(dom1), dom2));
 860
 861         dom1 = isl_space_range(left);
 862         dom2 = isl_space_range(right);
 863         nest2 = isl_space_wrap(isl_space_join(isl_space_reverse(dom1), dom2));
 864
 865         return isl_space_join(isl_space_reverse(nest1), nest2);
 866 error:
 867         isl_space_free(left);
 868         isl_space_free(right);
 869         return NULL;
 870 }
 871
 872 __isl_give isl_space *isl_space_range_product(__isl_take isl_space *left,
 873         __isl_take isl_space *right)
 874 {
 875         isl_space *dom, *ran1, *ran2, *nest;
 876
 877         if (!left || !right)
 878                 goto error;
 879
 880         isl_assert(left->ctx, match(left, isl_dim_param, right, isl_dim_param),
 881                         goto error);
 882         if (!isl_space_tuple_match(left, isl_dim_in, right, isl_dim_in))
 883                 isl_die(left->ctx, isl_error_invalid,
 884                         "domains need to match", goto error);
 885
 886         dom = isl_space_domain(isl_space_copy(left));
 887
 888         ran1 = isl_space_range(left);
 889         ran2 = isl_space_range(right);
 890         nest = isl_space_wrap(isl_space_join(isl_space_reverse(ran1), ran2));
 891
 892         return isl_space_join(isl_space_reverse(dom), nest);
 893 error:
 894         isl_space_free(left);
 895         isl_space_free(right);
 896         return NULL;
 897 }
 898
 899 __isl_give isl_space *isl_space_map_from_set(__isl_take isl_space *dim)
 900 {
 901         isl_id **ids = NULL;
 902
 903         if (!dim)
 904                 return NULL;
 905         isl_assert(dim->ctx, dim->n_in == 0, goto error);
 906         if (dim->n_out == 0 && !isl_space_is_named_or_nested(dim, isl_dim_out))
 907                 return dim;
 908         dim = isl_space_cow(dim);
 909         if (!dim)
 910                 return NULL;
 911         if (dim->ids) {
 912                 ids = isl_calloc_array(dim->ctx, isl_id *,
 913                                         dim->nparam + dim->n_out + dim->n_out);
 914                 if (!ids)
 915                         goto error;
 916                 get_ids(dim, isl_dim_param, 0, dim->nparam, ids);
 917                 get_ids(dim, isl_dim_out, 0, dim->n_out, ids + dim->nparam);
 918         }
 919         dim->n_in = dim->n_out;
 920         if (ids) {
 921                 free(dim->ids);
 922                 dim->ids = ids;
 923                 dim->n_id = dim->nparam + dim->n_out + dim->n_out;
 924                 dim = copy_ids(dim, isl_dim_out, 0, dim, isl_dim_in);
 925         }
 926         isl_id_free(dim->tuple_id[0]);
 927         dim->tuple_id[0] = isl_id_copy(dim->tuple_id[1]);
 928         isl_space_free(dim->nested[0]);
 929         dim->nested[0] = isl_space_copy(dim->nested[1]);
 930         return dim;
 931 error:
 932         isl_space_free(dim);
 933         return NULL;
 934 }
 935
 936 static __isl_give isl_space *set_ids(__isl_take isl_space *dim,
 937         enum isl_dim_type type,
 938         unsigned first, unsigned n, __isl_take isl_id **ids)
 939 {
 940         int i;
 941
 942         for (i = 0; i < n ; ++i)
 943                 dim = set_id(dim, type, first + i, ids[i]);
 944
 945         return dim;
 946 }
 947
 948 __isl_give isl_space *isl_space_reverse(__isl_take isl_space *dim)
 949 {
 950         unsigned t;
 951         isl_space *nested;
 952         isl_id **ids = NULL;
 953         isl_id *id;
 954
 955         if (!dim)
 956                 return NULL;
 957         if (match(dim, isl_dim_in, dim, isl_dim_out))
 958                 return dim;
 959
 960         dim = isl_space_cow(dim);
 961         if (!dim)
 962                 return NULL;
 963
 964         id = dim->tuple_id[0];
 965         dim->tuple_id[0] = dim->tuple_id[1];
 966         dim->tuple_id[1] = id;
 967
 968         nested = dim->nested[0];
 969         dim->nested[0] = dim->nested[1];
 970         dim->nested[1] = nested;
 971
 972         if (dim->ids) {
 973                 ids = isl_alloc_array(dim->ctx, isl_id *,
 974                                         dim->n_in + dim->n_out);
 975                 if (!ids)
 976                         goto error;
 977                 get_ids(dim, isl_dim_in, 0, dim->n_in, ids);
 978                 get_ids(dim, isl_dim_out, 0, dim->n_out, ids + dim->n_in);
 979         }
 980
 981         t = dim->n_in;
 982         dim->n_in = dim->n_out;
 983         dim->n_out = t;
 984
 985         if (dim->ids) {
 986                 dim = set_ids(dim, isl_dim_out, 0, dim->n_out, ids);
 987                 dim = set_ids(dim, isl_dim_in, 0, dim->n_in, ids + dim->n_out);
 988                 free(ids);
 989         }
 990
 991         return dim;
 992 error:
 993         free(ids);
 994         isl_space_free(dim);
 995         return NULL;
 996 }
 997
 998 __isl_give isl_space *isl_space_drop_dims(__isl_take isl_space *dim,
 999         enum isl_dim_type type, unsigned first, unsigned num)
1000 {
1001         int i;
1002
1003         if (!dim)
1004                 return NULL;
1005
1006         if (num == 0)
1007                 return isl_space_reset(dim, type);
1008
1009         if (!valid_dim_type(type))
1010                 isl_die(dim->ctx, isl_error_invalid,
1011                         "cannot drop dimensions of specified type", goto error);
1012
1013         isl_assert(dim->ctx, first + num <= n(dim, type), goto error);
1014         dim = isl_space_cow(dim);
1015         if (!dim)
1016                 goto error;
1017         if (dim->ids) {
1018                 dim = extend_ids(dim);
1019                 if (!dim)
1020                         goto error;
1021                 for (i = 0; i < num; ++i)
1022                         isl_id_free(get_id(dim, type, first + i));
1023                 for (i = first+num; i < n(dim, type); ++i)
1024                         set_id(dim, type, i - num, get_id(dim, type, i));
1025                 switch (type) {
1026                 case isl_dim_param:
1027                         get_ids(dim, isl_dim_in, 0, dim->n_in,
1028                                 dim->ids + offset(dim, isl_dim_in) - num);
1029                 case isl_dim_in:
1030                         get_ids(dim, isl_dim_out, 0, dim->n_out,
1031                                 dim->ids + offset(dim, isl_dim_out) - num);
1032                 default:
1033                         ;
1034                 }
1035                 dim->n_id -= num;
1036         }
1037         switch (type) {
1038         case isl_dim_param:     dim->nparam -= num; break;
1039         case isl_dim_in:        dim->n_in -= num; break;
1040         case isl_dim_out:       dim->n_out -= num; break;
1041         default:                ;
1042         }
1043         dim = isl_space_reset(dim, type);
1044         if (type == isl_dim_param) {
1045                 if (dim && dim->nested[0] &&
1046                     !(dim->nested[0] = isl_space_drop_dims(dim->nested[0],
1047                                                     isl_dim_param, first, num)))
1048                         goto error;
1049                 if (dim && dim->nested[1] &&
1050                     !(dim->nested[1] = isl_space_drop_dims(dim->nested[1],
1051                                                     isl_dim_param, first, num)))
1052                         goto error;
1053         }
1054         return dim;
1055 error:
1056         isl_space_free(dim);
1057         return NULL;
1058 }
1059
1060 __isl_give isl_space *isl_space_drop_inputs(__isl_take isl_space *dim,
1061                 unsigned first, unsigned n)
1062 {
1063         if (!dim)
1064                 return NULL;
1065         return isl_space_drop_dims(dim, isl_dim_in, first, n);
1066 }
1067
1068 __isl_give isl_space *isl_space_drop_outputs(__isl_take isl_space *dim,
1069                 unsigned first, unsigned n)
1070 {
1071         if (!dim)
1072                 return NULL;
1073         return isl_space_drop_dims(dim, isl_dim_out, first, n);
1074 }
1075
1076 __isl_give isl_space *isl_space_domain(__isl_take isl_space *dim)
1077 {
1078         if (!dim)
1079                 return NULL;
1080         dim = isl_space_drop_outputs(dim, 0, dim->n_out);
1081         return isl_space_reverse(dim);
1082 }
1083
1084 __isl_give isl_space *isl_space_from_domain(__isl_take isl_space *dim)
1085 {
1086         return isl_space_reverse(dim);
1087 }
1088
1089 __isl_give isl_space *isl_space_range(__isl_take isl_space *dim)
1090 {
1091         if (!dim)
1092                 return NULL;
1093         return isl_space_drop_inputs(dim, 0, dim->n_in);
1094 }
1095
1096 __isl_give isl_space *isl_space_from_range(__isl_take isl_space *dim)
1097 {
1098         return dim;
1099 }
1100
1101 __isl_give isl_space *isl_space_params(__isl_take isl_space *space)
1102 {
1103         space = isl_space_drop_dims(space,
1104                             isl_dim_in, 0, isl_space_dim(space, isl_dim_in));
1105         space = isl_space_drop_dims(space,
1106                             isl_dim_out, 0, isl_space_dim(space, isl_dim_out));
1107         return space;
1108 }
1109
1110 __isl_give isl_space *isl_space_as_set_space(__isl_take isl_space *dim)
1111 {
1112         dim = isl_space_cow(dim);
1113         if (!dim)
1114                 return NULL;
1115
1116         dim->n_out += dim->n_in;
1117         dim->n_in = 0;
1118         dim = isl_space_reset(dim, isl_dim_in);
1119         dim = isl_space_reset(dim, isl_dim_out);
1120
1121         return dim;
1122 }
1123
1124 __isl_give isl_space *isl_space_underlying(__isl_take isl_space *dim,
1125         unsigned n_div)
1126 {
1127         int i;
1128
1129         if (!dim)
1130                 return NULL;
1131         if (n_div == 0 &&
1132             dim->nparam == 0 && dim->n_in == 0 && dim->n_id == 0)
1133                 return isl_space_reset(isl_space_reset(dim, isl_dim_in), isl_dim_out);
1134         dim = isl_space_cow(dim);
1135         if (!dim)
1136                 return NULL;
1137         dim->n_out += dim->nparam + dim->n_in + n_div;
1138         dim->nparam = 0;
1139         dim->n_in = 0;
1140
1141         for (i = 0; i < dim->n_id; ++i)
1142                 isl_id_free(get_id(dim, isl_dim_out, i));
1143         dim->n_id = 0;
1144         dim = isl_space_reset(dim, isl_dim_in);
1145         dim = isl_space_reset(dim, isl_dim_out);
1146
1147         return dim;
1148 }
1149
1150 int isl_space_is_equal(__isl_keep isl_space *dim1, __isl_keep isl_space *dim2)
1151 {
1152         if (!dim1 || !dim2)
1153                 return -1;
1154         return match(dim1, isl_dim_param, dim2, isl_dim_param) &&
1155                isl_space_tuple_match(dim1, isl_dim_in, dim2, isl_dim_in) &&
1156                isl_space_tuple_match(dim1, isl_dim_out, dim2, isl_dim_out);
1157 }
1158
1159 int isl_space_compatible(__isl_keep isl_space *dim1,
1160         __isl_keep isl_space *dim2)
1161 {
1162         return dim1->nparam == dim2->nparam &&
1163                dim1->n_in + dim1->n_out == dim2->n_in + dim2->n_out;
1164 }
1165
1166 static uint32_t isl_hash_dim(uint32_t hash, __isl_keep isl_space *dim)
1167 {
1168         int i;
1169         isl_id *id;
1170
1171         if (!dim)
1172                 return hash;
1173
1174         hash = isl_hash_builtin(hash, dim->nparam);
1175         hash = isl_hash_builtin(hash, dim->n_in);
1176         hash = isl_hash_builtin(hash, dim->n_out);
1177
1178         for (i = 0; i < dim->nparam; ++i) {
1179                 id = get_id(dim, isl_dim_param, i);
1180                 hash = isl_hash_id(hash, id);
1181         }
1182
1183         id = tuple_id(dim, isl_dim_in);
1184         hash = isl_hash_id(hash, id);
1185         id = tuple_id(dim, isl_dim_out);
1186         hash = isl_hash_id(hash, id);
1187
1188         hash = isl_hash_dim(hash, dim->nested[0]);
1189         hash = isl_hash_dim(hash, dim->nested[1]);
1190
1191         return hash;
1192 }
1193
1194 uint32_t isl_space_get_hash(__isl_keep isl_space *dim)
1195 {
1196         uint32_t hash;
1197
1198         if (!dim)
1199                 return 0;
1200
1201         hash = isl_hash_init();
1202         hash = isl_hash_dim(hash, dim);
1203
1204         return hash;
1205 }
1206
1207 int isl_space_is_wrapping(__isl_keep isl_space *dim)
1208 {
1209         if (!dim)
1210                 return -1;
1211
1212         if (dim->n_in != 0 || dim->tuple_id[0] || dim->nested[0])
1213                 return 0;
1214
1215         return dim->nested[1] != NULL;
1216 }
1217
1218 __isl_give isl_space *isl_space_wrap(__isl_take isl_space *dim)
1219 {
1220         isl_space *wrap;
1221
1222         if (!dim)
1223                 return NULL;
1224
1225         wrap = isl_space_alloc(dim->ctx, dim->nparam, 0, dim->n_in + dim->n_out);
1226
1227         wrap = copy_ids(wrap, isl_dim_param, 0, dim, isl_dim_param);
1228         wrap = copy_ids(wrap, isl_dim_set, 0, dim, isl_dim_in);
1229         wrap = copy_ids(wrap, isl_dim_set, dim->n_in, dim, isl_dim_out);
1230
1231         if (!wrap)
1232                 goto error;
1233
1234         wrap->nested[1] = dim;
1235
1236         return wrap;
1237 error:
1238         isl_space_free(dim);
1239         return NULL;
1240 }
1241
1242 __isl_give isl_space *isl_space_unwrap(__isl_take isl_space *dim)
1243 {
1244         isl_space *unwrap;
1245
1246         if (!dim)
1247                 return NULL;
1248
1249         if (!isl_space_is_wrapping(dim))
1250                 isl_die(dim->ctx, isl_error_invalid, "not a wrapping dim",
1251                         goto error);
1252
1253         unwrap = isl_space_copy(dim->nested[1]);
1254         isl_space_free(dim);
1255
1256         return unwrap;
1257 error:
1258         isl_space_free(dim);
1259         return NULL;
1260 }
1261
1262 int isl_space_is_named_or_nested(__isl_keep isl_space *dim, enum isl_dim_type type)
1263 {
1264         if (type != isl_dim_in && type != isl_dim_out)
1265                 return 0;
1266         if (!dim)
1267                 return -1;
1268         if (dim->tuple_id[type - isl_dim_in])
1269                 return 1;
1270         if (dim->nested[type - isl_dim_in])
1271                 return 1;
1272         return 0;
1273 }
1274
1275 int isl_space_may_be_set(__isl_keep isl_space *dim)
1276 {
1277         if (!dim)
1278                 return -1;
1279         if (isl_space_dim(dim, isl_dim_in) != 0)
1280                 return 0;
1281         if (isl_space_is_named_or_nested(dim, isl_dim_in))
1282                 return 0;
1283         return 1;
1284 }
1285
1286 __isl_give isl_space *isl_space_reset(__isl_take isl_space *dim,
1287         enum isl_dim_type type)
1288 {
1289         if (!isl_space_is_named_or_nested(dim, type))
1290                 return dim;
1291
1292         dim = isl_space_cow(dim);
1293         if (!dim)
1294                 return NULL;
1295
1296         isl_id_free(dim->tuple_id[type - isl_dim_in]);
1297         dim->tuple_id[type - isl_dim_in] = NULL;
1298         isl_space_free(dim->nested[type - isl_dim_in]);
1299         dim->nested[type - isl_dim_in] = NULL;
1300
1301         return dim;
1302 }
1303
1304 __isl_give isl_space *isl_space_flatten(__isl_take isl_space *dim)
1305 {
1306         if (!dim)
1307                 return NULL;
1308         if (!dim->nested[0] && !dim->nested[1])
1309                 return dim;
1310
1311         if (dim->nested[0])
1312                 dim = isl_space_reset(dim, isl_dim_in);
1313         if (dim && dim->nested[1])
1314                 dim = isl_space_reset(dim, isl_dim_out);
1315
1316         return dim;
1317 }
1318
1319 __isl_give isl_space *isl_space_flatten_range(__isl_take isl_space *dim)
1320 {
1321         if (!dim)
1322                 return NULL;
1323         if (!dim->nested[1])
1324                 return dim;
1325
1326         return isl_space_reset(dim, isl_dim_out);
1327 }
1328
1329 /* Replace the dimensions of the given type of dst by those of src.
1330  */
1331 __isl_give isl_space *isl_space_replace(__isl_take isl_space *dst,
1332         enum isl_dim_type type, __isl_keep isl_space *src)
1333 {
1334         dst = isl_space_cow(dst);
1335
1336         if (!dst || !src)
1337                 goto error;
1338
1339         dst = isl_space_drop_dims(dst, type, 0, isl_space_dim(dst, type));
1340         dst = isl_space_add_dims(dst, type, isl_space_dim(src, type));
1341         dst = copy_ids(dst, type, 0, src, type);
1342
1343         if (dst && type == isl_dim_param) {
1344                 int i;
1345                 for (i = 0; i <= 1; ++i) {
1346                         if (!dst->nested[i])
1347                                 continue;
1348                         dst->nested[i] = isl_space_replace(dst->nested[i],
1349                                                          type, src);
1350                         if (!dst->nested[i])
1351                                 goto error;
1352                 }
1353         }
1354
1355         return dst;
1356 error:
1357         isl_space_free(dst);
1358         return NULL;
1359 }
1360
1361 /* Given a dimension specification "dim" of a set, create a dimension
1362  * specification for the lift of the set.  In particular, the result
1363  * is of the form [dim -> local[..]], with n_local variables in the
1364  * range of the wrapped map.
1365  */
1366 __isl_give isl_space *isl_space_lift(__isl_take isl_space *dim, unsigned n_local)
1367 {
1368         isl_space *local_dim;
1369
1370         if (!dim)
1371                 return NULL;
1372
1373         local_dim = isl_space_dup(dim);
1374         local_dim = isl_space_drop_dims(local_dim, isl_dim_set, 0, dim->n_out);
1375         local_dim = isl_space_add_dims(local_dim, isl_dim_set, n_local);
1376         local_dim = isl_space_set_tuple_name(local_dim, isl_dim_set, "local");
1377         dim = isl_space_join(isl_space_from_domain(dim),
1378                             isl_space_from_range(local_dim));
1379         dim = isl_space_wrap(dim);
1380         dim = isl_space_set_tuple_name(dim, isl_dim_set, "lifted");
1381
1382         return dim;
1383 }
1384
1385 int isl_space_can_zip(__isl_keep isl_space *dim)
1386 {
1387         if (!dim)
1388                 return -1;
1389
1390         return dim->nested[0] && dim->nested[1];
1391 }
1392
1393 __isl_give isl_space *isl_space_zip(__isl_take isl_space *dim)
1394 {
1395         isl_space *dom, *ran;
1396         isl_space *dom_dom, *dom_ran, *ran_dom, *ran_ran;
1397
1398         if (!isl_space_can_zip(dim))
1399                 isl_die(dim->ctx, isl_error_invalid, "dim cannot be zipped",
1400                         goto error);
1401
1402         if (!dim)
1403                 return 0;
1404         dom = isl_space_unwrap(isl_space_domain(isl_space_copy(dim)));
1405         ran = isl_space_unwrap(isl_space_range(dim));
1406         dom_dom = isl_space_domain(isl_space_copy(dom));
1407         dom_ran = isl_space_range(dom);
1408         ran_dom = isl_space_domain(isl_space_copy(ran));
1409         ran_ran = isl_space_range(ran);
1410         dom = isl_space_join(isl_space_from_domain(dom_dom),
1411                            isl_space_from_range(ran_dom));
1412         ran = isl_space_join(isl_space_from_domain(dom_ran),
1413                            isl_space_from_range(ran_ran));
1414         return isl_space_join(isl_space_from_domain(isl_space_wrap(dom)),
1415                             isl_space_from_range(isl_space_wrap(ran)));
1416 error:
1417         isl_space_free(dim);
1418         return NULL;
1419 }
1420
1421 int isl_space_has_named_params(__isl_keep isl_space *dim)
1422 {
1423         int i;
1424         unsigned off;
1425
1426         if (!dim)
1427                 return -1;
1428         if (dim->nparam == 0)
1429                 return 1;
1430         off = isl_space_offset(dim, isl_dim_param);
1431         if (off + dim->nparam > dim->n_id)
1432                 return 0;
1433         for (i = 0; i < dim->nparam; ++i)
1434                 if (!dim->ids[off + i])
1435                         return 0;
1436         return 1;
1437 }
1438
1439 /* Align the initial parameters of dim1 to match the order in dim2.
1440  */
1441 __isl_give isl_space *isl_space_align_params(__isl_take isl_space *dim1,
1442         __isl_take isl_space *dim2)
1443 {
1444         isl_reordering *exp;
1445
1446         if (!isl_space_has_named_params(dim1) || !isl_space_has_named_params(dim2))
1447                 isl_die(isl_space_get_ctx(dim1), isl_error_invalid,
1448                         "parameter alignment requires named parameters",
1449                         goto error);
1450
1451         exp = isl_parameter_alignment_reordering(dim1, dim2);
1452         isl_space_free(dim1);
1453         isl_space_free(dim2);
1454         if (!exp)
1455                 return NULL;
1456         dim1 = isl_space_copy(exp->dim);
1457         isl_reordering_free(exp);
1458         return dim1;
1459 error:
1460         isl_space_free(dim1);
1461         isl_space_free(dim2);
1462         return NULL;
1463 }