isl_point.c

   1 #include <isl_map_private.h>
   2 #include <isl_point_private.h>
   3 #include <isl/set.h>
   4 #include <isl/union_set.h>
   5 #include <isl_sample.h>
   6 #include <isl_scan.h>
   7 #include <isl_seq.h>
   8 #include <isl_space_private.h>
   9 #include <isl_val_private.h>
  10 #include <isl_vec_private.h>
  11 #include <isl_output_private.h>
  12 #include <isl/deprecated/point_int.h>
  13
  14 #include <set_to_map.c>
  15
  16 isl_ctx *isl_point_get_ctx(__isl_keep isl_point *pnt)
  17 {
  18         return pnt ? isl_space_get_ctx(pnt->dim) : NULL;
  19 }
  20
  21 __isl_give isl_space *isl_point_get_space(__isl_keep isl_point *pnt)
  22 {
  23         return pnt ? isl_space_copy(pnt->dim) : NULL;
  24 }
  25
  26 __isl_give isl_point *isl_point_alloc(__isl_take isl_space *dim,
  27         __isl_take isl_vec *vec)
  28 {
  29         struct isl_point *pnt;
  30
  31         if (!dim || !vec)
  32                 goto error;
  33
  34         if (vec->size > 1 + isl_space_dim(dim, isl_dim_all)) {
  35                 vec = isl_vec_cow(vec);
  36                 if (!vec)
  37                         goto error;
  38                 vec->size = 1 + isl_space_dim(dim, isl_dim_all);
  39         }
  40
  41         pnt = isl_alloc_type(dim->ctx, struct isl_point);
  42         if (!pnt)
  43                 goto error;
  44
  45         pnt->ref = 1;
  46         pnt->dim = dim;
  47         pnt->vec = vec;
  48
  49         return pnt;
  50 error:
  51         isl_space_free(dim);
  52         isl_vec_free(vec);
  53         return NULL;
  54 }
  55
  56 __isl_give isl_point *isl_point_zero(__isl_take isl_space *dim)
  57 {
  58         isl_vec *vec;
  59
  60         if (!dim)
  61                 return NULL;
  62         vec = isl_vec_alloc(dim->ctx, 1 + isl_space_dim(dim, isl_dim_all));
  63         if (!vec)
  64                 goto error;
  65         isl_int_set_si(vec->el[0], 1);
  66         isl_seq_clr(vec->el + 1, vec->size - 1);
  67         return isl_point_alloc(dim, vec);
  68 error:
  69         isl_space_free(dim);
  70         return NULL;
  71 }
  72
  73 __isl_give isl_point *isl_point_dup(__isl_keep isl_point *pnt)
  74 {
  75         struct isl_point *pnt2;
  76
  77         if (!pnt)
  78                 return NULL;
  79         pnt2 = isl_point_alloc(isl_space_copy(pnt->dim), isl_vec_copy(pnt->vec));
  80         return pnt2;
  81 }
  82
  83 __isl_give isl_point *isl_point_cow(__isl_take isl_point *pnt)
  84 {
  85         struct isl_point *pnt2;
  86         if (!pnt)
  87                 return NULL;
  88
  89         if (pnt->ref == 1)
  90                 return pnt;
  91
  92         pnt2 = isl_point_dup(pnt);
  93         isl_point_free(pnt);
  94         return pnt2;
  95 }
  96
  97 __isl_give isl_point *isl_point_copy(__isl_keep isl_point *pnt)
  98 {
  99         if (!pnt)
 100                 return NULL;
 101
 102         pnt->ref++;
 103         return pnt;
 104 }
 105
 106 void isl_point_free(__isl_take isl_point *pnt)
 107 {
 108         if (!pnt)
 109                 return;
 110
 111         if (--pnt->ref > 0)
 112                 return;
 113
 114         isl_space_free(pnt->dim);
 115         isl_vec_free(pnt->vec);
 116         free(pnt);
 117 }
 118
 119 __isl_give isl_point *isl_point_void(__isl_take isl_space *dim)
 120 {
 121         if (!dim)
 122                 return NULL;
 123
 124         return isl_point_alloc(dim, isl_vec_alloc(dim->ctx, 0));
 125 }
 126
 127 isl_bool isl_point_is_void(__isl_keep isl_point *pnt)
 128 {
 129         if (!pnt)
 130                 return isl_bool_error;
 131
 132         return pnt->vec->size == 0;
 133 }
 134
 135 int isl_point_get_coordinate(__isl_keep isl_point *pnt,
 136         enum isl_dim_type type, int pos, isl_int *v)
 137 {
 138         if (!pnt || isl_point_is_void(pnt))
 139                 return -1;
 140
 141         if (pos < 0 || pos >= isl_space_dim(pnt->dim, type))
 142                 isl_die(isl_point_get_ctx(pnt), isl_error_invalid,
 143                         "position out of bounds", return -1);
 144
 145         if (type == isl_dim_set)
 146                 pos += isl_space_dim(pnt->dim, isl_dim_param);
 147         isl_int_set(*v, pnt->vec->el[1 + pos]);
 148
 149         return 0;
 150 }
 151
 152 /* Return the value of coordinate "pos" of type "type" of "pnt".
 153  */
 154 __isl_give isl_val *isl_point_get_coordinate_val(__isl_keep isl_point *pnt,
 155         enum isl_dim_type type, int pos)
 156 {
 157         isl_ctx *ctx;
 158         isl_val *v;
 159
 160         if (!pnt)
 161                 return NULL;
 162
 163         ctx = isl_point_get_ctx(pnt);
 164         if (isl_point_is_void(pnt))
 165                 isl_die(ctx, isl_error_invalid,
 166                         "void point does not have coordinates", return NULL);
 167         if (pos < 0 || pos >= isl_space_dim(pnt->dim, type))
 168                 isl_die(ctx, isl_error_invalid,
 169                         "position out of bounds", return NULL);
 170
 171         if (type == isl_dim_set)
 172                 pos += isl_space_dim(pnt->dim, isl_dim_param);
 173
 174         v = isl_val_rat_from_isl_int(ctx, pnt->vec->el[1 + pos],
 175                                                 pnt->vec->el[0]);
 176         return isl_val_normalize(v);
 177 }
 178
 179 __isl_give isl_point *isl_point_set_coordinate(__isl_take isl_point *pnt,
 180         enum isl_dim_type type, int pos, isl_int v)
 181 {
 182         if (!pnt || isl_point_is_void(pnt))
 183                 return pnt;
 184
 185         pnt = isl_point_cow(pnt);
 186         if (!pnt)
 187                 return NULL;
 188         pnt->vec = isl_vec_cow(pnt->vec);
 189         if (!pnt->vec)
 190                 goto error;
 191
 192         if (type == isl_dim_set)
 193                 pos += isl_space_dim(pnt->dim, isl_dim_param);
 194
 195         isl_int_set(pnt->vec->el[1 + pos], v);
 196
 197         return pnt;
 198 error:
 199         isl_point_free(pnt);
 200         return NULL;
 201 }
 202
 203 /* Replace coordinate "pos" of type "type" of "pnt" by "v".
 204  */
 205 __isl_give isl_point *isl_point_set_coordinate_val(__isl_take isl_point *pnt,
 206         enum isl_dim_type type, int pos, __isl_take isl_val *v)
 207 {
 208         if (!pnt || !v)
 209                 goto error;
 210         if (isl_point_is_void(pnt))
 211                 isl_die(isl_point_get_ctx(pnt), isl_error_invalid,
 212                         "void point does not have coordinates", goto error);
 213         if (pos < 0 || pos >= isl_space_dim(pnt->dim, type))
 214                 isl_die(isl_point_get_ctx(pnt), isl_error_invalid,
 215                         "position out of bounds", goto error);
 216         if (!isl_val_is_rat(v))
 217                 isl_die(isl_point_get_ctx(pnt), isl_error_invalid,
 218                         "expecting rational value", goto error);
 219
 220         if (isl_int_eq(pnt->vec->el[1 + pos], v->n) &&
 221             isl_int_eq(pnt->vec->el[0], v->d)) {
 222                 isl_val_free(v);
 223                 return pnt;
 224         }
 225
 226         pnt = isl_point_cow(pnt);
 227         if (!pnt)
 228                 goto error;
 229         pnt->vec = isl_vec_cow(pnt->vec);
 230         if (!pnt->vec)
 231                 goto error;
 232
 233         if (isl_int_eq(pnt->vec->el[0], v->d)) {
 234                 isl_int_set(pnt->vec->el[1 + pos], v->n);
 235         } else if (isl_int_is_one(v->d)) {
 236                 isl_int_mul(pnt->vec->el[1 + pos], pnt->vec->el[0], v->n);
 237         } else {
 238                 isl_seq_scale(pnt->vec->el + 1,
 239                                 pnt->vec->el + 1, v->d, pnt->vec->size - 1);
 240                 isl_int_mul(pnt->vec->el[1 + pos], pnt->vec->el[0], v->n);
 241                 isl_int_mul(pnt->vec->el[0], pnt->vec->el[0], v->d);
 242                 pnt->vec = isl_vec_normalize(pnt->vec);
 243                 if (!pnt->vec)
 244                         goto error;
 245         }
 246
 247         isl_val_free(v);
 248         return pnt;
 249 error:
 250         isl_val_free(v);
 251         isl_point_free(pnt);
 252         return NULL;
 253 }
 254
 255 __isl_give isl_point *isl_point_add_ui(__isl_take isl_point *pnt,
 256         enum isl_dim_type type, int pos, unsigned val)
 257 {
 258         if (!pnt || isl_point_is_void(pnt))
 259                 return pnt;
 260
 261         pnt = isl_point_cow(pnt);
 262         if (!pnt)
 263                 return NULL;
 264         pnt->vec = isl_vec_cow(pnt->vec);
 265         if (!pnt->vec)
 266                 goto error;
 267
 268         if (type == isl_dim_set)
 269                 pos += isl_space_dim(pnt->dim, isl_dim_param);
 270
 271         isl_int_add_ui(pnt->vec->el[1 + pos], pnt->vec->el[1 + pos], val);
 272
 273         return pnt;
 274 error:
 275         isl_point_free(pnt);
 276         return NULL;
 277 }
 278
 279 __isl_give isl_point *isl_point_sub_ui(__isl_take isl_point *pnt,
 280         enum isl_dim_type type, int pos, unsigned val)
 281 {
 282         if (!pnt || isl_point_is_void(pnt))
 283                 return pnt;
 284
 285         pnt = isl_point_cow(pnt);
 286         if (!pnt)
 287                 return NULL;
 288         pnt->vec = isl_vec_cow(pnt->vec);
 289         if (!pnt->vec)
 290                 goto error;
 291
 292         if (type == isl_dim_set)
 293                 pos += isl_space_dim(pnt->dim, isl_dim_param);
 294
 295         isl_int_sub_ui(pnt->vec->el[1 + pos], pnt->vec->el[1 + pos], val);
 296
 297         return pnt;
 298 error:
 299         isl_point_free(pnt);
 300         return NULL;
 301 }
 302
 303 struct isl_foreach_point {
 304         struct isl_scan_callback callback;
 305         isl_stat (*fn)(__isl_take isl_point *pnt, void *user);
 306         void *user;
 307         isl_space *dim;
 308 };
 309
 310 static isl_stat foreach_point(struct isl_scan_callback *cb,
 311         __isl_take isl_vec *sample)
 312 {
 313         struct isl_foreach_point *fp = (struct isl_foreach_point *)cb;
 314         isl_point *pnt;
 315
 316         pnt = isl_point_alloc(isl_space_copy(fp->dim), sample);
 317
 318         return fp->fn(pnt, fp->user);
 319 }
 320
 321 isl_stat isl_set_foreach_point(__isl_keep isl_set *set,
 322         isl_stat (*fn)(__isl_take isl_point *pnt, void *user), void *user)
 323 {
 324         struct isl_foreach_point fp = { { &foreach_point }, fn, user };
 325         int i;
 326
 327         if (!set)
 328                 return isl_stat_error;
 329
 330         fp.dim = isl_set_get_space(set);
 331         if (!fp.dim)
 332                 return isl_stat_error;
 333
 334         set = isl_set_copy(set);
 335         set = isl_set_cow(set);
 336         set = isl_set_make_disjoint(set);
 337         set = isl_set_compute_divs(set);
 338         if (!set)
 339                 goto error;
 340
 341         for (i = 0; i < set->n; ++i)
 342                 if (isl_basic_set_scan(isl_basic_set_copy(set->p[i]),
 343                                         &fp.callback) < 0)
 344                         goto error;
 345
 346         isl_set_free(set);
 347         isl_space_free(fp.dim);
 348
 349         return isl_stat_ok;
 350 error:
 351         isl_set_free(set);
 352         isl_space_free(fp.dim);
 353         return isl_stat_error;
 354 }
 355
 356 /* Return 1 if "bmap" contains the point "point".
 357  * "bmap" is assumed to have known divs.
 358  * The point is first extended with the divs and then passed
 359  * to basic_map_contains.
 360  */
 361 isl_bool isl_basic_map_contains_point(__isl_keep isl_basic_map *bmap,
 362         __isl_keep isl_point *point)
 363 {
 364         int i;
 365         struct isl_vec *vec;
 366         unsigned dim;
 367         isl_bool contains;
 368
 369         if (!bmap || !point)
 370                 return isl_bool_error;
 371         isl_assert(bmap->ctx, isl_space_is_equal(bmap->dim, point->dim),
 372                 return isl_bool_error);
 373         if (bmap->n_div == 0)
 374                 return isl_basic_map_contains(bmap, point->vec);
 375
 376         dim = isl_basic_map_total_dim(bmap) - bmap->n_div;
 377         vec = isl_vec_alloc(bmap->ctx, 1 + dim + bmap->n_div);
 378         if (!vec)
 379                 return isl_bool_error;
 380
 381         isl_seq_cpy(vec->el, point->vec->el, point->vec->size);
 382         for (i = 0; i < bmap->n_div; ++i) {
 383                 isl_seq_inner_product(bmap->div[i] + 1, vec->el,
 384                                         1 + dim + i, &vec->el[1+dim+i]);
 385                 isl_int_fdiv_q(vec->el[1+dim+i], vec->el[1+dim+i],
 386                                 bmap->div[i][0]);
 387         }
 388
 389         contains = isl_basic_map_contains(bmap, vec);
 390
 391         isl_vec_free(vec);
 392         return contains;
 393 }
 394
 395 int isl_map_contains_point(__isl_keep isl_map *map, __isl_keep isl_point *point)
 396 {
 397         int i;
 398         int found = 0;
 399
 400         if (!map || !point)
 401                 return -1;
 402
 403         map = isl_map_copy(map);
 404         map = isl_map_compute_divs(map);
 405         if (!map)
 406                 return -1;
 407
 408         for (i = 0; i < map->n; ++i) {
 409                 found = isl_basic_map_contains_point(map->p[i], point);
 410                 if (found < 0)
 411                         goto error;
 412                 if (found)
 413                         break;
 414         }
 415         isl_map_free(map);
 416
 417         return found;
 418 error:
 419         isl_map_free(map);
 420         return -1;
 421 }
 422
 423 isl_bool isl_set_contains_point(__isl_keep isl_set *set,
 424         __isl_keep isl_point *point)
 425 {
 426         return isl_map_contains_point(set_to_map(set), point);
 427 }
 428
 429 __isl_give isl_basic_set *isl_basic_set_from_point(__isl_take isl_point *pnt)
 430 {
 431         isl_basic_set *bset;
 432         isl_basic_set *model;
 433
 434         if (!pnt)
 435                 return NULL;
 436
 437         model = isl_basic_set_empty(isl_space_copy(pnt->dim));
 438         bset = isl_basic_set_from_vec(isl_vec_copy(pnt->vec));
 439         bset = isl_basic_set_from_underlying_set(bset, model);
 440         isl_point_free(pnt);
 441
 442         return bset;
 443 }
 444
 445 __isl_give isl_set *isl_set_from_point(__isl_take isl_point *pnt)
 446 {
 447         isl_basic_set *bset;
 448         bset = isl_basic_set_from_point(pnt);
 449         return isl_set_from_basic_set(bset);
 450 }
 451
 452 /* Construct a union set, containing the single element "pnt".
 453  * If "pnt" is void, then return an empty union set.
 454  */
 455 __isl_give isl_union_set *isl_union_set_from_point(__isl_take isl_point *pnt)
 456 {
 457         if (!pnt)
 458                 return NULL;
 459         if (isl_point_is_void(pnt)) {
 460                 isl_space *space;
 461
 462                 space = isl_point_get_space(pnt);
 463                 isl_point_free(pnt);
 464                 return isl_union_set_empty(space);
 465         }
 466
 467         return isl_union_set_from_set(isl_set_from_point(pnt));
 468 }
 469
 470 __isl_give isl_basic_set *isl_basic_set_box_from_points(
 471         __isl_take isl_point *pnt1, __isl_take isl_point *pnt2)
 472 {
 473         isl_basic_set *bset;
 474         unsigned total;
 475         int i;
 476         int k;
 477         isl_int t;
 478
 479         isl_int_init(t);
 480
 481         if (!pnt1 || !pnt2)
 482                 goto error;
 483
 484         isl_assert(pnt1->dim->ctx,
 485                         isl_space_is_equal(pnt1->dim, pnt2->dim), goto error);
 486
 487         if (isl_point_is_void(pnt1) && isl_point_is_void(pnt2)) {
 488                 isl_space *dim = isl_space_copy(pnt1->dim);
 489                 isl_point_free(pnt1);
 490                 isl_point_free(pnt2);
 491                 isl_int_clear(t);
 492                 return isl_basic_set_empty(dim);
 493         }
 494         if (isl_point_is_void(pnt1)) {
 495                 isl_point_free(pnt1);
 496                 isl_int_clear(t);
 497                 return isl_basic_set_from_point(pnt2);
 498         }
 499         if (isl_point_is_void(pnt2)) {
 500                 isl_point_free(pnt2);
 501                 isl_int_clear(t);
 502                 return isl_basic_set_from_point(pnt1);
 503         }
 504
 505         total = isl_space_dim(pnt1->dim, isl_dim_all);
 506         bset = isl_basic_set_alloc_space(isl_space_copy(pnt1->dim), 0, 0, 2 * total);
 507
 508         for (i = 0; i < total; ++i) {
 509                 isl_int_mul(t, pnt1->vec->el[1 + i], pnt2->vec->el[0]);
 510                 isl_int_submul(t, pnt2->vec->el[1 + i], pnt1->vec->el[0]);
 511
 512                 k = isl_basic_set_alloc_inequality(bset);
 513                 if (k < 0)
 514                         goto error;
 515                 isl_seq_clr(bset->ineq[k] + 1, total);
 516                 if (isl_int_is_pos(t)) {
 517                         isl_int_set_si(bset->ineq[k][1 + i], -1);
 518                         isl_int_set(bset->ineq[k][0], pnt1->vec->el[1 + i]);
 519                 } else {
 520                         isl_int_set_si(bset->ineq[k][1 + i], 1);
 521                         isl_int_neg(bset->ineq[k][0], pnt1->vec->el[1 + i]);
 522                 }
 523                 isl_int_fdiv_q(bset->ineq[k][0], bset->ineq[k][0], pnt1->vec->el[0]);
 524
 525                 k = isl_basic_set_alloc_inequality(bset);
 526                 if (k < 0)
 527                         goto error;
 528                 isl_seq_clr(bset->ineq[k] + 1, total);
 529                 if (isl_int_is_pos(t)) {
 530                         isl_int_set_si(bset->ineq[k][1 + i], 1);
 531                         isl_int_neg(bset->ineq[k][0], pnt2->vec->el[1 + i]);
 532                 } else {
 533                         isl_int_set_si(bset->ineq[k][1 + i], -1);
 534                         isl_int_set(bset->ineq[k][0], pnt2->vec->el[1 + i]);
 535                 }
 536                 isl_int_fdiv_q(bset->ineq[k][0], bset->ineq[k][0], pnt2->vec->el[0]);
 537         }
 538
 539         bset = isl_basic_set_finalize(bset);
 540
 541         isl_point_free(pnt1);
 542         isl_point_free(pnt2);
 543
 544         isl_int_clear(t);
 545
 546         return bset;
 547 error:
 548         isl_point_free(pnt1);
 549         isl_point_free(pnt2);
 550         isl_int_clear(t);
 551         return NULL;
 552 }
 553
 554 __isl_give isl_set *isl_set_box_from_points(__isl_take isl_point *pnt1,
 555         __isl_take isl_point *pnt2)
 556 {
 557         isl_basic_set *bset;
 558         bset = isl_basic_set_box_from_points(pnt1, pnt2);
 559         return isl_set_from_basic_set(bset);
 560 }
 561
 562 /* Print the coordinate at position "pos" of the point "pnt".
 563  */
 564 static __isl_give isl_printer *print_coordinate(__isl_take isl_printer *p,
 565         struct isl_print_space_data *data, unsigned pos)
 566 {
 567         isl_point *pnt = data->user;
 568
 569         p = isl_printer_print_isl_int(p, pnt->vec->el[1 + pos]);
 570         if (!isl_int_is_one(pnt->vec->el[0])) {
 571                 p = isl_printer_print_str(p, "/");
 572                 p = isl_printer_print_isl_int(p, pnt->vec->el[0]);
 573         }
 574
 575         return p;
 576 }
 577
 578 __isl_give isl_printer *isl_printer_print_point(
 579         __isl_take isl_printer *p, __isl_keep isl_point *pnt)
 580 {
 581         struct isl_print_space_data data = { 0 };
 582         int i;
 583         unsigned nparam;
 584         unsigned dim;
 585
 586         if (!pnt)
 587                 return p;
 588         if (isl_point_is_void(pnt)) {
 589                 p = isl_printer_print_str(p, "void");
 590                 return p;
 591         }
 592
 593         nparam = isl_space_dim(pnt->dim, isl_dim_param);
 594         dim = isl_space_dim(pnt->dim, isl_dim_set);
 595         if (nparam > 0) {
 596                 p = isl_printer_print_str(p, "[");
 597                 for (i = 0; i < nparam; ++i) {
 598                         const char *name;
 599                         if (i)
 600                                 p = isl_printer_print_str(p, ", ");
 601                         name = isl_space_get_dim_name(pnt->dim, isl_dim_param, i);
 602                         if (name) {
 603                                 p = isl_printer_print_str(p, name);
 604                                 p = isl_printer_print_str(p, " = ");
 605                         }
 606                         p = isl_printer_print_isl_int(p, pnt->vec->el[1 + i]);
 607                         if (!isl_int_is_one(pnt->vec->el[0])) {
 608                                 p = isl_printer_print_str(p, "/");
 609                                 p = isl_printer_print_isl_int(p, pnt->vec->el[0]);
 610                         }
 611                 }
 612                 p = isl_printer_print_str(p, "]");
 613                 p = isl_printer_print_str(p, " -> ");
 614         }
 615         data.print_dim = &print_coordinate;
 616         data.user = pnt;
 617         p = isl_printer_print_str(p, "{ ");
 618         p = isl_print_space(pnt->dim, p, 0, &data);
 619         p = isl_printer_print_str(p, " }");
 620         return p;
 621 }