isl_point.c

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