isl_map_subtract.c

   1 /*
   2  * Copyright 2008-2009 Katholieke Universiteit Leuven
   3  *
   4  * Use of this software is governed by the MIT license
   5  *
   6  * Written by Sven Verdoolaege, K.U.Leuven, Departement
   7  * Computerwetenschappen, Celestijnenlaan 200A, B-3001 Leuven, Belgium
   8  */
   9
  10 #include <isl_map_private.h>
  11 #include <isl_seq.h>
  12 #include <isl/set.h>
  13 #include <isl/map.h>
  14 #include "isl_tab.h"
  15 #include <isl_point_private.h>
  16 #include <isl_vec_private.h>
  17
  18 /* Expand the constraint "c" into "v".  The initial "dim" dimensions
  19  * are the same, but "v" may have more divs than "c" and the divs of "c"
  20  * may appear in different positions in "v".
  21  * The number of divs in "c" is given by "n_div" and the mapping
  22  * of divs in "c" to divs in "v" is given by "div_map".
  23  *
  24  * Although it shouldn't happen in practice, it is theoretically
  25  * possible that two or more divs in "c" are mapped to the same div in "v".
  26  * These divs are then necessarily the same, so we simply add their
  27  * coefficients.
  28  */
  29 static void expand_constraint(isl_vec *v, unsigned dim,
  30         isl_int *c, int *div_map, unsigned n_div)
  31 {
  32         int i;
  33
  34         isl_seq_cpy(v->el, c, 1 + dim);
  35         isl_seq_clr(v->el + 1 + dim, v->size - (1 + dim));
  36
  37         for (i = 0; i < n_div; ++i) {
  38                 int pos = 1 + dim + div_map[i];
  39                 isl_int_add(v->el[pos], v->el[pos], c[1 + dim + i]);
  40         }
  41 }
  42
  43 /* Add all constraints of bmap to tab.  The equalities of bmap
  44  * are added as a pair of inequalities.
  45  */
  46 static int tab_add_constraints(struct isl_tab *tab,
  47         __isl_keep isl_basic_map *bmap, int *div_map)
  48 {
  49         int i;
  50         unsigned dim;
  51         unsigned tab_total;
  52         unsigned bmap_total;
  53         isl_vec *v;
  54
  55         if (!tab || !bmap)
  56                 return -1;
  57
  58         tab_total = isl_basic_map_total_dim(tab->bmap);
  59         bmap_total = isl_basic_map_total_dim(bmap);
  60         dim = isl_space_dim(tab->bmap->dim, isl_dim_all);
  61
  62         if (isl_tab_extend_cons(tab, 2 * bmap->n_eq + bmap->n_ineq) < 0)
  63                 return -1;
  64
  65         v = isl_vec_alloc(bmap->ctx, 1 + tab_total);
  66         if (!v)
  67                 return -1;
  68
  69         for (i = 0; i < bmap->n_eq; ++i) {
  70                 expand_constraint(v, dim, bmap->eq[i], div_map, bmap->n_div);
  71                 if (isl_tab_add_ineq(tab, v->el) < 0)
  72                         goto error;
  73                 isl_seq_neg(bmap->eq[i], bmap->eq[i], 1 + bmap_total);
  74                 expand_constraint(v, dim, bmap->eq[i], div_map, bmap->n_div);
  75                 if (isl_tab_add_ineq(tab, v->el) < 0)
  76                         goto error;
  77                 isl_seq_neg(bmap->eq[i], bmap->eq[i], 1 + bmap_total);
  78                 if (tab->empty)
  79                         break;
  80         }
  81
  82         for (i = 0; i < bmap->n_ineq; ++i) {
  83                 expand_constraint(v, dim, bmap->ineq[i], div_map, bmap->n_div);
  84                 if (isl_tab_add_ineq(tab, v->el) < 0)
  85                         goto error;
  86                 if (tab->empty)
  87                         break;
  88         }
  89
  90         isl_vec_free(v);
  91         return 0;
  92 error:
  93         isl_vec_free(v);
  94         return -1;
  95 }
  96
  97 /* Add a specific constraint of bmap (or its opposite) to tab.
  98  * The position of the constraint is specified by "c", where
  99  * the equalities of bmap are counted twice, once for the inequality
 100  * that is equal to the equality, and once for its negation.
 101  */
 102 static int tab_add_constraint(struct isl_tab *tab,
 103         __isl_keep isl_basic_map *bmap, int *div_map, int c, int oppose)
 104 {
 105         unsigned dim;
 106         unsigned tab_total;
 107         unsigned bmap_total;
 108         isl_vec *v;
 109         int r;
 110
 111         if (!tab || !bmap)
 112                 return -1;
 113
 114         tab_total = isl_basic_map_total_dim(tab->bmap);
 115         bmap_total = isl_basic_map_total_dim(bmap);
 116         dim = isl_space_dim(tab->bmap->dim, isl_dim_all);
 117
 118         v = isl_vec_alloc(bmap->ctx, 1 + tab_total);
 119         if (!v)
 120                 return -1;
 121
 122         if (c < 2 * bmap->n_eq) {
 123                 if ((c % 2) != oppose)
 124                         isl_seq_neg(bmap->eq[c/2], bmap->eq[c/2],
 125                                         1 + bmap_total);
 126                 if (oppose)
 127                         isl_int_sub_ui(bmap->eq[c/2][0], bmap->eq[c/2][0], 1);
 128                 expand_constraint(v, dim, bmap->eq[c/2], div_map, bmap->n_div);
 129                 r = isl_tab_add_ineq(tab, v->el);
 130                 if (oppose)
 131                         isl_int_add_ui(bmap->eq[c/2][0], bmap->eq[c/2][0], 1);
 132                 if ((c % 2) != oppose)
 133                         isl_seq_neg(bmap->eq[c/2], bmap->eq[c/2],
 134                                         1 + bmap_total);
 135         } else {
 136                 c -= 2 * bmap->n_eq;
 137                 if (oppose) {
 138                         isl_seq_neg(bmap->ineq[c], bmap->ineq[c],
 139                                         1 + bmap_total);
 140                         isl_int_sub_ui(bmap->ineq[c][0], bmap->ineq[c][0], 1);
 141                 }
 142                 expand_constraint(v, dim, bmap->ineq[c], div_map, bmap->n_div);
 143                 r = isl_tab_add_ineq(tab, v->el);
 144                 if (oppose) {
 145                         isl_int_add_ui(bmap->ineq[c][0], bmap->ineq[c][0], 1);
 146                         isl_seq_neg(bmap->ineq[c], bmap->ineq[c],
 147                                         1 + bmap_total);
 148                 }
 149         }
 150
 151         isl_vec_free(v);
 152         return r;
 153 }
 154
 155 static int tab_add_divs(struct isl_tab *tab, __isl_keep isl_basic_map *bmap,
 156         int **div_map)
 157 {
 158         int i, j;
 159         struct isl_vec *vec;
 160         unsigned total;
 161         unsigned dim;
 162
 163         if (!bmap)
 164                 return -1;
 165         if (!bmap->n_div)
 166                 return 0;
 167
 168         if (!*div_map)
 169                 *div_map = isl_alloc_array(bmap->ctx, int, bmap->n_div);
 170         if (!*div_map)
 171                 return -1;
 172
 173         total = isl_basic_map_total_dim(tab->bmap);
 174         dim = total - tab->bmap->n_div;
 175         vec = isl_vec_alloc(bmap->ctx, 2 + total + bmap->n_div);
 176         if (!vec)
 177                 return -1;
 178
 179         for (i = 0; i < bmap->n_div; ++i) {
 180                 isl_seq_cpy(vec->el, bmap->div[i], 2 + dim);
 181                 isl_seq_clr(vec->el + 2 + dim, tab->bmap->n_div);
 182                 for (j = 0; j < i; ++j)
 183                         isl_int_set(vec->el[2 + dim + (*div_map)[j]],
 184                                         bmap->div[i][2 + dim + j]);
 185                 for (j = 0; j < tab->bmap->n_div; ++j)
 186                         if (isl_seq_eq(tab->bmap->div[j],
 187                                         vec->el, 2 + dim + tab->bmap->n_div))
 188                                 break;
 189                 (*div_map)[i] = j;
 190                 if (j == tab->bmap->n_div) {
 191                         vec->size = 2 + dim + tab->bmap->n_div;
 192                         if (isl_tab_add_div(tab, vec, NULL, NULL) < 0)
 193                                 goto error;
 194                 }
 195         }
 196
 197         isl_vec_free(vec);
 198
 199         return 0;
 200 error:
 201         isl_vec_free(vec);
 202
 203         return -1;
 204 }
 205
 206 /* Freeze all constraints of tableau tab.
 207  */
 208 static int tab_freeze_constraints(struct isl_tab *tab)
 209 {
 210         int i;
 211
 212         for (i = 0; i < tab->n_con; ++i)
 213                 if (isl_tab_freeze_constraint(tab, i) < 0)
 214                         return -1;
 215
 216         return 0;
 217 }
 218
 219 /* Check for redundant constraints starting at offset.
 220  * Put the indices of the redundant constraints in index
 221  * and return the number of redundant constraints.
 222  */
 223 static int n_non_redundant(isl_ctx *ctx, struct isl_tab *tab,
 224         int offset, int **index)
 225 {
 226         int i, n;
 227         int n_test = tab->n_con - offset;
 228
 229         if (isl_tab_detect_redundant(tab) < 0)
 230                 return -1;
 231
 232         if (n_test == 0)
 233                 return 0;
 234         if (!*index)
 235                 *index = isl_alloc_array(ctx, int, n_test);
 236         if (!*index)
 237                 return -1;
 238
 239         for (n = 0, i = 0; i < n_test; ++i) {
 240                 int r;
 241                 r = isl_tab_is_redundant(tab, offset + i);
 242                 if (r < 0)
 243                         return -1;
 244                 if (r)
 245                         continue;
 246                 (*index)[n++] = i;
 247         }
 248
 249         return n;
 250 }
 251
 252 /* basic_map_collect_diff calls add on each of the pieces of
 253  * the set difference between bmap and map until the add method
 254  * return a negative value.
 255  */
 256 struct isl_diff_collector {
 257         int (*add)(struct isl_diff_collector *dc,
 258                     __isl_take isl_basic_map *bmap);
 259 };
 260
 261 /* Compute the set difference between bmap and map and call
 262  * dc->add on each of the piece until this function returns
 263  * a negative value.
 264  * Return 0 on success and -1 on error.  dc->add returning
 265  * a negative value is treated as an error, but the calling
 266  * function can interpret the results based on the state of dc.
 267  *
 268  * Assumes that map has known divs.
 269  *
 270  * The difference is computed by a backtracking algorithm.
 271  * Each level corresponds to a basic map in "map".
 272  * When a node in entered for the first time, we check
 273  * if the corresonding basic map intersects the current piece
 274  * of "bmap".  If not, we move to the next level.
 275  * Otherwise, we split the current piece into as many
 276  * pieces as there are non-redundant constraints of the current
 277  * basic map in the intersection.  Each of these pieces is
 278  * handled by a child of the current node.
 279  * In particular, if there are n non-redundant constraints,
 280  * then for each 0 <= i < n, a piece is cut off by adding
 281  * constraints 0 <= j < i and adding the opposite of constraint i.
 282  * If there are no non-redundant constraints, meaning that the current
 283  * piece is a subset of the current basic map, then we simply backtrack.
 284  *
 285  * In the leaves, we check if the remaining piece has any integer points
 286  * and if so, pass it along to dc->add.  As a special case, if nothing
 287  * has been removed when we end up in a leaf, we simply pass along
 288  * the original basic map.
 289  */
 290 static isl_stat basic_map_collect_diff(__isl_take isl_basic_map *bmap,
 291         __isl_take isl_map *map, struct isl_diff_collector *dc)
 292 {
 293         int i;
 294         int modified;
 295         int level;
 296         int init;
 297         isl_bool empty;
 298         isl_ctx *ctx;
 299         struct isl_tab *tab = NULL;
 300         struct isl_tab_undo **snap = NULL;
 301         int *k = NULL;
 302         int *n = NULL;
 303         int **index = NULL;
 304         int **div_map = NULL;
 305
 306         empty = isl_basic_map_is_empty(bmap);
 307         if (empty) {
 308                 isl_basic_map_free(bmap);
 309                 isl_map_free(map);
 310                 return empty < 0 ? isl_stat_error : isl_stat_ok;
 311         }
 312
 313         bmap = isl_basic_map_cow(bmap);
 314         map = isl_map_cow(map);
 315
 316         if (!bmap || !map)
 317                 goto error;
 318
 319         ctx = map->ctx;
 320         snap = isl_alloc_array(map->ctx, struct isl_tab_undo *, map->n);
 321         k = isl_alloc_array(map->ctx, int, map->n);
 322         n = isl_alloc_array(map->ctx, int, map->n);
 323         index = isl_calloc_array(map->ctx, int *, map->n);
 324         div_map = isl_calloc_array(map->ctx, int *, map->n);
 325         if (!snap || !k || !n || !index || !div_map)
 326                 goto error;
 327
 328         bmap = isl_basic_map_order_divs(bmap);
 329         map = isl_map_order_divs(map);
 330
 331         tab = isl_tab_from_basic_map(bmap, 1);
 332         if (!tab)
 333                 goto error;
 334
 335         modified = 0;
 336         level = 0;
 337         init = 1;
 338
 339         while (level >= 0) {
 340                 if (level >= map->n) {
 341                         int empty;
 342                         struct isl_basic_map *bm;
 343                         if (!modified) {
 344                                 if (dc->add(dc, isl_basic_map_copy(bmap)) < 0)
 345                                         goto error;
 346                                 break;
 347                         }
 348                         bm = isl_basic_map_copy(tab->bmap);
 349                         bm = isl_basic_map_cow(bm);
 350                         bm = isl_basic_map_update_from_tab(bm, tab);
 351                         bm = isl_basic_map_simplify(bm);
 352                         bm = isl_basic_map_finalize(bm);
 353                         empty = isl_basic_map_is_empty(bm);
 354                         if (empty)
 355                                 isl_basic_map_free(bm);
 356                         else if (dc->add(dc, bm) < 0)
 357                                 goto error;
 358                         if (empty < 0)
 359                                 goto error;
 360                         level--;
 361                         init = 0;
 362                         continue;
 363                 }
 364                 if (init) {
 365                         int offset;
 366                         struct isl_tab_undo *snap2;
 367                         snap2 = isl_tab_snap(tab);
 368                         if (tab_add_divs(tab, map->p[level],
 369                                          &div_map[level]) < 0)
 370                                 goto error;
 371                         offset = tab->n_con;
 372                         snap[level] = isl_tab_snap(tab);
 373                         if (tab_freeze_constraints(tab) < 0)
 374                                 goto error;
 375                         if (tab_add_constraints(tab, map->p[level],
 376                                                 div_map[level]) < 0)
 377                                 goto error;
 378                         k[level] = 0;
 379                         n[level] = 0;
 380                         if (tab->empty) {
 381                                 if (isl_tab_rollback(tab, snap2) < 0)
 382                                         goto error;
 383                                 level++;
 384                                 continue;
 385                         }
 386                         modified = 1;
 387                         n[level] = n_non_redundant(ctx, tab, offset,
 388                                                     &index[level]);
 389                         if (n[level] < 0)
 390                                 goto error;
 391                         if (n[level] == 0) {
 392                                 level--;
 393                                 init = 0;
 394                                 continue;
 395                         }
 396                         if (isl_tab_rollback(tab, snap[level]) < 0)
 397                                 goto error;
 398                         if (tab_add_constraint(tab, map->p[level],
 399                                         div_map[level], index[level][0], 1) < 0)
 400                                 goto error;
 401                         level++;
 402                         continue;
 403                 } else {
 404                         if (k[level] + 1 >= n[level]) {
 405                                 level--;
 406                                 continue;
 407                         }
 408                         if (isl_tab_rollback(tab, snap[level]) < 0)
 409                                 goto error;
 410                         if (tab_add_constraint(tab, map->p[level],
 411                                                 div_map[level],
 412                                                 index[level][k[level]], 0) < 0)
 413                                 goto error;
 414                         snap[level] = isl_tab_snap(tab);
 415                         k[level]++;
 416                         if (tab_add_constraint(tab, map->p[level],
 417                                                 div_map[level],
 418                                                 index[level][k[level]], 1) < 0)
 419                                 goto error;
 420                         level++;
 421                         init = 1;
 422                         continue;
 423                 }
 424         }
 425
 426         isl_tab_free(tab);
 427         free(snap);
 428         free(n);
 429         free(k);
 430         for (i = 0; index && i < map->n; ++i)
 431                 free(index[i]);
 432         free(index);
 433         for (i = 0; div_map && i < map->n; ++i)
 434                 free(div_map[i]);
 435         free(div_map);
 436
 437         isl_basic_map_free(bmap);
 438         isl_map_free(map);
 439
 440         return isl_stat_ok;
 441 error:
 442         isl_tab_free(tab);
 443         free(snap);
 444         free(n);
 445         free(k);
 446         for (i = 0; index && i < map->n; ++i)
 447                 free(index[i]);
 448         free(index);
 449         for (i = 0; div_map && i < map->n; ++i)
 450                 free(div_map[i]);
 451         free(div_map);
 452         isl_basic_map_free(bmap);
 453         isl_map_free(map);
 454         return isl_stat_error;
 455 }
 456
 457 /* A diff collector that actually collects all parts of the
 458  * set difference in the field diff.
 459  */
 460 struct isl_subtract_diff_collector {
 461         struct isl_diff_collector dc;
 462         struct isl_map *diff;
 463 };
 464
 465 /* isl_subtract_diff_collector callback.
 466  */
 467 static int basic_map_subtract_add(struct isl_diff_collector *dc,
 468                             __isl_take isl_basic_map *bmap)
 469 {
 470         struct isl_subtract_diff_collector *sdc;
 471         sdc = (struct isl_subtract_diff_collector *)dc;
 472
 473         sdc->diff = isl_map_union_disjoint(sdc->diff,
 474                         isl_map_from_basic_map(bmap));
 475
 476         return sdc->diff ? 0 : -1;
 477 }
 478
 479 /* Return the set difference between bmap and map.
 480  */
 481 static __isl_give isl_map *basic_map_subtract(__isl_take isl_basic_map *bmap,
 482         __isl_take isl_map *map)
 483 {
 484         struct isl_subtract_diff_collector sdc;
 485         sdc.dc.add = &basic_map_subtract_add;
 486         sdc.diff = isl_map_empty(isl_basic_map_get_space(bmap));
 487         if (basic_map_collect_diff(bmap, map, &sdc.dc) < 0) {
 488                 isl_map_free(sdc.diff);
 489                 sdc.diff = NULL;
 490         }
 491         return sdc.diff;
 492 }
 493
 494 /* Return an empty map living in the same space as "map1" and "map2".
 495  */
 496 static __isl_give isl_map *replace_pair_by_empty( __isl_take isl_map *map1,
 497         __isl_take isl_map *map2)
 498 {
 499         isl_space *space;
 500
 501         space = isl_map_get_space(map1);
 502         isl_map_free(map1);
 503         isl_map_free(map2);
 504         return isl_map_empty(space);
 505 }
 506
 507 /* Return the set difference between map1 and map2.
 508  * (U_i A_i) \ (U_j B_j) is computed as U_i (A_i \ (U_j B_j))
 509  *
 510  * If "map1" and "map2" are obviously equal to each other,
 511  * then return an empty map in the same space.
 512  *
 513  * If "map1" and "map2" are disjoint, then simply return "map1".
 514  */
 515 static __isl_give isl_map *map_subtract( __isl_take isl_map *map1,
 516         __isl_take isl_map *map2)
 517 {
 518         int i;
 519         int equal, disjoint;
 520         struct isl_map *diff;
 521
 522         if (!map1 || !map2)
 523                 goto error;
 524
 525         isl_assert(map1->ctx, isl_space_is_equal(map1->dim, map2->dim), goto error);
 526
 527         equal = isl_map_plain_is_equal(map1, map2);
 528         if (equal < 0)
 529                 goto error;
 530         if (equal)
 531                 return replace_pair_by_empty(map1, map2);
 532
 533         disjoint = isl_map_is_disjoint(map1, map2);
 534         if (disjoint < 0)
 535                 goto error;
 536         if (disjoint) {
 537                 isl_map_free(map2);
 538                 return map1;
 539         }
 540
 541         map1 = isl_map_compute_divs(map1);
 542         map2 = isl_map_compute_divs(map2);
 543         if (!map1 || !map2)
 544                 goto error;
 545
 546         map1 = isl_map_remove_empty_parts(map1);
 547         map2 = isl_map_remove_empty_parts(map2);
 548
 549         diff = isl_map_empty(isl_map_get_space(map1));
 550         for (i = 0; i < map1->n; ++i) {
 551                 struct isl_map *d;
 552                 d = basic_map_subtract(isl_basic_map_copy(map1->p[i]),
 553                                        isl_map_copy(map2));
 554                 if (ISL_F_ISSET(map1, ISL_MAP_DISJOINT))
 555                         diff = isl_map_union_disjoint(diff, d);
 556                 else
 557                         diff = isl_map_union(diff, d);
 558         }
 559
 560         isl_map_free(map1);
 561         isl_map_free(map2);
 562
 563         return diff;
 564 error:
 565         isl_map_free(map1);
 566         isl_map_free(map2);
 567         return NULL;
 568 }
 569
 570 __isl_give isl_map *isl_map_subtract( __isl_take isl_map *map1,
 571         __isl_take isl_map *map2)
 572 {
 573         return isl_map_align_params_map_map_and(map1, map2, &map_subtract);
 574 }
 575
 576 struct isl_set *isl_set_subtract(struct isl_set *set1, struct isl_set *set2)
 577 {
 578         return (struct isl_set *)
 579                 isl_map_subtract(
 580                         (struct isl_map *)set1, (struct isl_map *)set2);
 581 }
 582
 583 /* Remove the elements of "dom" from the domain of "map".
 584  */
 585 static __isl_give isl_map *map_subtract_domain(__isl_take isl_map *map,
 586         __isl_take isl_set *dom)
 587 {
 588         isl_map *ext_dom;
 589
 590         if (!isl_map_compatible_domain(map, dom))
 591                 isl_die(isl_set_get_ctx(dom), isl_error_invalid,
 592                         "incompatible spaces", goto error);
 593
 594         ext_dom = isl_map_universe(isl_map_get_space(map));
 595         ext_dom = isl_map_intersect_domain(ext_dom, dom);
 596         return isl_map_subtract(map, ext_dom);
 597 error:
 598         isl_map_free(map);
 599         isl_set_free(dom);
 600         return NULL;
 601 }
 602
 603 __isl_give isl_map *isl_map_subtract_domain(__isl_take isl_map *map,
 604         __isl_take isl_set *dom)
 605 {
 606         return isl_map_align_params_map_map_and(map, dom, &map_subtract_domain);
 607 }
 608
 609 /* Remove the elements of "dom" from the range of "map".
 610  */
 611 static __isl_give isl_map *map_subtract_range(__isl_take isl_map *map,
 612         __isl_take isl_set *dom)
 613 {
 614         isl_map *ext_dom;
 615
 616         if (!isl_map_compatible_range(map, dom))
 617                 isl_die(isl_set_get_ctx(dom), isl_error_invalid,
 618                         "incompatible spaces", goto error);
 619
 620         ext_dom = isl_map_universe(isl_map_get_space(map));
 621         ext_dom = isl_map_intersect_range(ext_dom, dom);
 622         return isl_map_subtract(map, ext_dom);
 623 error:
 624         isl_map_free(map);
 625         isl_set_free(dom);
 626         return NULL;
 627 }
 628
 629 __isl_give isl_map *isl_map_subtract_range(__isl_take isl_map *map,
 630         __isl_take isl_set *dom)
 631 {
 632         return isl_map_align_params_map_map_and(map, dom, &map_subtract_range);
 633 }
 634
 635 /* A diff collector that aborts as soon as its add function is called,
 636  * setting empty to 0.
 637  */
 638 struct isl_is_empty_diff_collector {
 639         struct isl_diff_collector dc;
 640         isl_bool empty;
 641 };
 642
 643 /* isl_is_empty_diff_collector callback.
 644  */
 645 static int basic_map_is_empty_add(struct isl_diff_collector *dc,
 646                             __isl_take isl_basic_map *bmap)
 647 {
 648         struct isl_is_empty_diff_collector *edc;
 649         edc = (struct isl_is_empty_diff_collector *)dc;
 650
 651         edc->empty = 0;
 652
 653         isl_basic_map_free(bmap);
 654         return -1;
 655 }
 656
 657 /* Check if bmap \ map is empty by computing this set difference
 658  * and breaking off as soon as the difference is known to be non-empty.
 659  */
 660 static isl_bool basic_map_diff_is_empty(__isl_keep isl_basic_map *bmap,
 661         __isl_keep isl_map *map)
 662 {
 663         isl_bool empty;
 664         isl_stat r;
 665         struct isl_is_empty_diff_collector edc;
 666
 667         empty = isl_basic_map_plain_is_empty(bmap);
 668         if (empty)
 669                 return empty;
 670
 671         edc.dc.add = &basic_map_is_empty_add;
 672         edc.empty = isl_bool_true;
 673         r = basic_map_collect_diff(isl_basic_map_copy(bmap),
 674                                    isl_map_copy(map), &edc.dc);
 675         if (!edc.empty)
 676                 return isl_bool_false;
 677
 678         return r < 0 ? isl_bool_error : isl_bool_true;
 679 }
 680
 681 /* Check if map1 \ map2 is empty by checking if the set difference is empty
 682  * for each of the basic maps in map1.
 683  */
 684 static isl_bool map_diff_is_empty(__isl_keep isl_map *map1,
 685         __isl_keep isl_map *map2)
 686 {
 687         int i;
 688         isl_bool is_empty = isl_bool_true;
 689
 690         if (!map1 || !map2)
 691                 return isl_bool_error;
 692
 693         for (i = 0; i < map1->n; ++i) {
 694                 is_empty = basic_map_diff_is_empty(map1->p[i], map2);
 695                 if (is_empty < 0 || !is_empty)
 696                          break;
 697         }
 698
 699         return is_empty;
 700 }
 701
 702 /* Return 1 if "bmap" contains a single element.
 703  */
 704 int isl_basic_map_plain_is_singleton(__isl_keep isl_basic_map *bmap)
 705 {
 706         if (!bmap)
 707                 return -1;
 708         if (bmap->n_div)
 709                 return 0;
 710         if (bmap->n_ineq)
 711                 return 0;
 712         return bmap->n_eq == isl_basic_map_total_dim(bmap);
 713 }
 714
 715 /* Return 1 if "map" contains a single element.
 716  */
 717 int isl_map_plain_is_singleton(__isl_keep isl_map *map)
 718 {
 719         if (!map)
 720                 return -1;
 721         if (map->n != 1)
 722                 return 0;
 723
 724         return isl_basic_map_plain_is_singleton(map->p[0]);
 725 }
 726
 727 /* Given a singleton basic map, extract the single element
 728  * as an isl_point.
 729  */
 730 static __isl_give isl_point *singleton_extract_point(
 731         __isl_keep isl_basic_map *bmap)
 732 {
 733         int j;
 734         unsigned dim;
 735         struct isl_vec *point;
 736         isl_int m;
 737
 738         if (!bmap)
 739                 return NULL;
 740
 741         dim = isl_basic_map_total_dim(bmap);
 742         isl_assert(bmap->ctx, bmap->n_eq == dim, return NULL);
 743         point = isl_vec_alloc(bmap->ctx, 1 + dim);
 744         if (!point)
 745                 return NULL;
 746
 747         isl_int_init(m);
 748
 749         isl_int_set_si(point->el[0], 1);
 750         for (j = 0; j < bmap->n_eq; ++j) {
 751                 int i = dim - 1 - j;
 752                 isl_assert(bmap->ctx,
 753                     isl_seq_first_non_zero(bmap->eq[j] + 1, i) == -1,
 754                     goto error);
 755                 isl_assert(bmap->ctx,
 756                     isl_int_is_one(bmap->eq[j][1 + i]) ||
 757                     isl_int_is_negone(bmap->eq[j][1 + i]),
 758                     goto error);
 759                 isl_assert(bmap->ctx,
 760                     isl_seq_first_non_zero(bmap->eq[j]+1+i+1, dim-i-1) == -1,
 761                     goto error);
 762
 763                 isl_int_gcd(m, point->el[0], bmap->eq[j][1 + i]);
 764                 isl_int_divexact(m, bmap->eq[j][1 + i], m);
 765                 isl_int_abs(m, m);
 766                 isl_seq_scale(point->el, point->el, m, 1 + i);
 767                 isl_int_divexact(m, point->el[0], bmap->eq[j][1 + i]);
 768                 isl_int_neg(m, m);
 769                 isl_int_mul(point->el[1 + i], m, bmap->eq[j][0]);
 770         }
 771
 772         isl_int_clear(m);
 773         return isl_point_alloc(isl_basic_map_get_space(bmap), point);
 774 error:
 775         isl_int_clear(m);
 776         isl_vec_free(point);
 777         return NULL;
 778 }
 779
 780 /* Return isl_bool_true if the singleton map "map1" is a subset of "map2",
 781  * i.e., if the single element of "map1" is also an element of "map2".
 782  * Assumes "map2" has known divs.
 783  */
 784 static isl_bool map_is_singleton_subset(__isl_keep isl_map *map1,
 785         __isl_keep isl_map *map2)
 786 {
 787         int i;
 788         isl_bool is_subset = isl_bool_false;
 789         struct isl_point *point;
 790
 791         if (!map1 || !map2)
 792                 return isl_bool_error;
 793         if (map1->n != 1)
 794                 isl_die(isl_map_get_ctx(map1), isl_error_invalid,
 795                         "expecting single-disjunct input",
 796                         return isl_bool_error);
 797
 798         point = singleton_extract_point(map1->p[0]);
 799         if (!point)
 800                 return isl_bool_error;
 801
 802         for (i = 0; i < map2->n; ++i) {
 803                 is_subset = isl_basic_map_contains_point(map2->p[i], point);
 804                 if (is_subset)
 805                         break;
 806         }
 807
 808         isl_point_free(point);
 809         return is_subset;
 810 }
 811
 812 static isl_bool map_is_subset(__isl_keep isl_map *map1,
 813         __isl_keep isl_map *map2)
 814 {
 815         isl_bool is_subset = isl_bool_false;
 816         isl_bool empty;
 817         int rat1, rat2;
 818
 819         if (!map1 || !map2)
 820                 return isl_bool_error;
 821
 822         if (!isl_map_has_equal_space(map1, map2))
 823                 return isl_bool_false;
 824
 825         empty = isl_map_is_empty(map1);
 826         if (empty < 0)
 827                 return isl_bool_error;
 828         if (empty)
 829                 return isl_bool_true;
 830
 831         empty = isl_map_is_empty(map2);
 832         if (empty < 0)
 833                 return isl_bool_error;
 834         if (empty)
 835                 return isl_bool_false;
 836
 837         rat1 = isl_map_has_rational(map1);
 838         rat2 = isl_map_has_rational(map2);
 839         if (rat1 < 0 || rat2 < 0)
 840                 return isl_bool_error;
 841         if (rat1 && !rat2)
 842                 return isl_bool_false;
 843
 844         if (isl_map_plain_is_universe(map2))
 845                 return isl_bool_true;
 846
 847         map2 = isl_map_compute_divs(isl_map_copy(map2));
 848         if (isl_map_plain_is_singleton(map1)) {
 849                 is_subset = map_is_singleton_subset(map1, map2);
 850                 isl_map_free(map2);
 851                 return is_subset;
 852         }
 853         is_subset = map_diff_is_empty(map1, map2);
 854         isl_map_free(map2);
 855
 856         return is_subset;
 857 }
 858
 859 isl_bool isl_map_is_subset(__isl_keep isl_map *map1, __isl_keep isl_map *map2)
 860 {
 861         return isl_map_align_params_map_map_and_test(map1, map2,
 862                                                         &map_is_subset);
 863 }
 864
 865 isl_bool isl_set_is_subset(__isl_keep isl_set *set1, __isl_keep isl_set *set2)
 866 {
 867         return isl_map_is_subset(
 868                         (struct isl_map *)set1, (struct isl_map *)set2);
 869 }
 870
 871 __isl_give isl_map *isl_map_make_disjoint(__isl_take isl_map *map)
 872 {
 873         int i;
 874         struct isl_subtract_diff_collector sdc;
 875         sdc.dc.add = &basic_map_subtract_add;
 876
 877         if (!map)
 878                 return NULL;
 879         if (ISL_F_ISSET(map, ISL_MAP_DISJOINT))
 880                 return map;
 881         if (map->n <= 1)
 882                 return map;
 883
 884         map = isl_map_compute_divs(map);
 885         map = isl_map_remove_empty_parts(map);
 886
 887         if (!map || map->n <= 1)
 888                 return map;
 889
 890         sdc.diff = isl_map_from_basic_map(isl_basic_map_copy(map->p[0]));
 891
 892         for (i = 1; i < map->n; ++i) {
 893                 struct isl_basic_map *bmap = isl_basic_map_copy(map->p[i]);
 894                 struct isl_map *copy = isl_map_copy(sdc.diff);
 895                 if (basic_map_collect_diff(bmap, copy, &sdc.dc) < 0) {
 896                         isl_map_free(sdc.diff);
 897                         sdc.diff = NULL;
 898                         break;
 899                 }
 900         }
 901
 902         isl_map_free(map);
 903
 904         return sdc.diff;
 905 }
 906
 907 __isl_give isl_set *isl_set_make_disjoint(__isl_take isl_set *set)
 908 {
 909         return (struct isl_set *)isl_map_make_disjoint((struct isl_map *)set);
 910 }
 911
 912 __isl_give isl_map *isl_map_complement(__isl_take isl_map *map)
 913 {
 914         isl_map *universe;
 915
 916         if (!map)
 917                 return NULL;
 918
 919         universe = isl_map_universe(isl_map_get_space(map));
 920
 921         return isl_map_subtract(universe, map);
 922 }
 923
 924 __isl_give isl_set *isl_set_complement(__isl_take isl_set *set)
 925 {
 926         return isl_map_complement(set);
 927 }