vm/math.cpp

   1 #include "master.hpp"
   2
   3 namespace factor
   4 {
   5
   6 void factor_vm::primitive_bignum_to_fixnum()
   7 {
   8         ctx->replace(tag_fixnum(bignum_to_fixnum(untag<bignum>(ctx->peek()))));
   9 }
  10
  11 void factor_vm::primitive_float_to_fixnum()
  12 {
  13         ctx->replace(tag_fixnum(float_to_fixnum(ctx->peek())));
  14 }
  15
  16 /* Division can only overflow when we are dividing the most negative fixnum
  17 by -1. */
  18 void factor_vm::primitive_fixnum_divint()
  19 {
  20         fixnum y = untag_fixnum(ctx->pop()); \
  21         fixnum x = untag_fixnum(ctx->peek());
  22         fixnum result = x / y;
  23         if(result == -fixnum_min)
  24                 ctx->replace(allot_integer(-fixnum_min));
  25         else
  26                 ctx->replace(tag_fixnum(result));
  27 }
  28
  29 void factor_vm::primitive_fixnum_divmod()
  30 {
  31         cell y = ((cell *)ctx->datastack)[0];
  32         cell x = ((cell *)ctx->datastack)[-1];
  33         if(y == tag_fixnum(-1) && x == tag_fixnum(fixnum_min))
  34         {
  35                 ((cell *)ctx->datastack)[-1] = allot_integer(-fixnum_min);
  36                 ((cell *)ctx->datastack)[0] = tag_fixnum(0);
  37         }
  38         else
  39         {
  40                 ((cell *)ctx->datastack)[-1] = tag_fixnum(untag_fixnum(x) / untag_fixnum(y));
  41                 ((cell *)ctx->datastack)[0] = (fixnum)x % (fixnum)y;
  42         }
  43 }
  44
  45 /*
  46  * If we're shifting right by n bits, we won't overflow as long as none of the
  47  * high WORD_SIZE-TAG_BITS-n bits are set.
  48  */
  49 inline fixnum factor_vm::sign_mask(fixnum x)
  50 {
  51         return x >> (WORD_SIZE - 1);
  52 }
  53
  54 inline fixnum factor_vm::branchless_max(fixnum x, fixnum y)
  55 {
  56         return (x - ((x - y) & sign_mask(x - y)));
  57 }
  58
  59 inline fixnum factor_vm::branchless_abs(fixnum x)
  60 {
  61         return (x ^ sign_mask(x)) - sign_mask(x);
  62 }
  63
  64 void factor_vm::primitive_fixnum_shift()
  65 {
  66         fixnum y = untag_fixnum(ctx->pop());
  67         fixnum x = untag_fixnum(ctx->peek());
  68
  69         if(x == 0)
  70                 return;
  71         else if(y < 0)
  72         {
  73                 y = branchless_max(y,-WORD_SIZE + 1);
  74                 ctx->replace(tag_fixnum(x >> -y));
  75                 return;
  76         }
  77         else if(y < WORD_SIZE - TAG_BITS)
  78         {
  79                 fixnum mask = -((fixnum)1 << (WORD_SIZE - 1 - TAG_BITS - y));
  80                 if(!(branchless_abs(x) & mask))
  81                 {
  82                         ctx->replace(tag_fixnum(x << y));
  83                         return;
  84                 }
  85         }
  86
  87         ctx->replace(tag<bignum>(bignum_arithmetic_shift(
  88                 fixnum_to_bignum(x),y)));
  89 }
  90
  91 void factor_vm::primitive_fixnum_to_bignum()
  92 {
  93         ctx->replace(tag<bignum>(fixnum_to_bignum(untag_fixnum(ctx->peek()))));
  94 }
  95
  96 void factor_vm::primitive_float_to_bignum()
  97 {
  98         ctx->replace(tag<bignum>(float_to_bignum(ctx->peek())));
  99 }
 100
 101 #define POP_BIGNUMS(x,y) \
 102         bignum * y = untag<bignum>(ctx->pop()); \
 103         bignum * x = untag<bignum>(ctx->pop());
 104
 105 void factor_vm::primitive_bignum_eq()
 106 {
 107         POP_BIGNUMS(x,y);
 108         ctx->push(tag_boolean(bignum_equal_p(x,y)));
 109 }
 110
 111 void factor_vm::primitive_bignum_add()
 112 {
 113         POP_BIGNUMS(x,y);
 114         ctx->push(tag<bignum>(bignum_add(x,y)));
 115 }
 116
 117 void factor_vm::primitive_bignum_subtract()
 118 {
 119         POP_BIGNUMS(x,y);
 120         ctx->push(tag<bignum>(bignum_subtract(x,y)));
 121 }
 122
 123 void factor_vm::primitive_bignum_multiply()
 124 {
 125         POP_BIGNUMS(x,y);
 126         ctx->push(tag<bignum>(bignum_multiply(x,y)));
 127 }
 128
 129 void factor_vm::primitive_bignum_divint()
 130 {
 131         POP_BIGNUMS(x,y);
 132         ctx->push(tag<bignum>(bignum_quotient(x,y)));
 133 }
 134
 135 void factor_vm::primitive_bignum_divmod()
 136 {
 137         bignum *q, *r;
 138         POP_BIGNUMS(x,y);
 139         bignum_divide(x,y,&q,&r);
 140         ctx->push(tag<bignum>(q));
 141         ctx->push(tag<bignum>(r));
 142 }
 143
 144 void factor_vm::primitive_bignum_mod()
 145 {
 146         POP_BIGNUMS(x,y);
 147         ctx->push(tag<bignum>(bignum_remainder(x,y)));
 148 }
 149
 150 void factor_vm::primitive_bignum_and()
 151 {
 152         POP_BIGNUMS(x,y);
 153         ctx->push(tag<bignum>(bignum_bitwise_and(x,y)));
 154 }
 155
 156 void factor_vm::primitive_bignum_or()
 157 {
 158         POP_BIGNUMS(x,y);
 159         ctx->push(tag<bignum>(bignum_bitwise_ior(x,y)));
 160 }
 161
 162 void factor_vm::primitive_bignum_xor()
 163 {
 164         POP_BIGNUMS(x,y);
 165         ctx->push(tag<bignum>(bignum_bitwise_xor(x,y)));
 166 }
 167
 168 void factor_vm::primitive_bignum_shift()
 169 {
 170         fixnum y = untag_fixnum(ctx->pop());
 171         bignum* x = untag<bignum>(ctx->pop());
 172         ctx->push(tag<bignum>(bignum_arithmetic_shift(x,y)));
 173 }
 174
 175 void factor_vm::primitive_bignum_less()
 176 {
 177         POP_BIGNUMS(x,y);
 178         ctx->push(tag_boolean(bignum_compare(x,y) == bignum_comparison_less));
 179 }
 180
 181 void factor_vm::primitive_bignum_lesseq()
 182 {
 183         POP_BIGNUMS(x,y);
 184         ctx->push(tag_boolean(bignum_compare(x,y) != bignum_comparison_greater));
 185 }
 186
 187 void factor_vm::primitive_bignum_greater()
 188 {
 189         POP_BIGNUMS(x,y);
 190         ctx->push(tag_boolean(bignum_compare(x,y) == bignum_comparison_greater));
 191 }
 192
 193 void factor_vm::primitive_bignum_greatereq()
 194 {
 195         POP_BIGNUMS(x,y);
 196         ctx->push(tag_boolean(bignum_compare(x,y) != bignum_comparison_less));
 197 }
 198
 199 void factor_vm::primitive_bignum_not()
 200 {
 201         ctx->replace(tag<bignum>(bignum_bitwise_not(untag<bignum>(ctx->peek()))));
 202 }
 203
 204 void factor_vm::primitive_bignum_bitp()
 205 {
 206         fixnum bit = to_fixnum(ctx->pop());
 207         bignum *x = untag<bignum>(ctx->pop());
 208         ctx->push(tag_boolean(bignum_logbitp(bit,x)));
 209 }
 210
 211 void factor_vm::primitive_bignum_log2()
 212 {
 213         ctx->replace(tag<bignum>(bignum_integer_length(untag<bignum>(ctx->peek()))));
 214 }
 215
 216 unsigned int factor_vm::bignum_producer(unsigned int digit)
 217 {
 218         unsigned char *ptr = (unsigned char *)alien_offset(ctx->peek());
 219         return *(ptr + digit);
 220 }
 221
 222 unsigned int bignum_producer(unsigned int digit, factor_vm *parent)
 223 {
 224         return parent->bignum_producer(digit);
 225 }
 226
 227 void factor_vm::primitive_byte_array_to_bignum()
 228 {
 229         cell n_digits = array_capacity(untag_check<byte_array>(ctx->peek()));
 230         bignum * result = digit_stream_to_bignum(n_digits,factor::bignum_producer,0x100,0);
 231         ctx->replace(tag<bignum>(result));
 232 }
 233
 234 cell factor_vm::unbox_array_size_slow()
 235 {
 236         if(tagged<object>(ctx->peek()).type() == BIGNUM_TYPE)
 237         {
 238                 bignum *zero = untag<bignum>(bignum_zero);
 239                 bignum *max = cell_to_bignum(array_size_max);
 240                 bignum *n = untag<bignum>(ctx->peek());
 241                 if(bignum_compare(n,zero) != bignum_comparison_less
 242                         && bignum_compare(n,max) == bignum_comparison_less)
 243                 {
 244                         ctx->pop();
 245                         return bignum_to_cell(n);
 246                 }
 247         }
 248
 249         general_error(ERROR_ARRAY_SIZE,ctx->pop(),tag_fixnum(array_size_max),NULL);
 250         return 0; /* can't happen */
 251 }
 252
 253 void factor_vm::primitive_fixnum_to_float()
 254 {
 255         ctx->replace(allot_float(fixnum_to_float(ctx->peek())));
 256 }
 257
 258 void factor_vm::primitive_bignum_to_float()
 259 {
 260         ctx->replace(allot_float(bignum_to_float(ctx->peek())));
 261 }
 262
 263 void factor_vm::primitive_str_to_float()
 264 {
 265         byte_array *bytes = untag_check<byte_array>(ctx->peek());
 266         cell capacity = array_capacity(bytes);
 267
 268         char *c_str = (char *)(bytes + 1);
 269         char *end = c_str;
 270         double f = strtod(c_str,&end);
 271         if(end == c_str + capacity - 1)
 272                 ctx->replace(allot_float(f));
 273         else
 274                 ctx->replace(false_object);
 275 }
 276
 277 void factor_vm::primitive_float_to_str()
 278 {
 279         byte_array *array = allot_byte_array(33);
 280         SNPRINTF((char *)(array + 1),32,"%.16g",untag_float_check(ctx->pop()));
 281         ctx->push(tag<byte_array>(array));
 282 }
 283
 284 #define POP_FLOATS(x,y) \
 285         double y = untag_float(ctx->pop()); \
 286         double x = untag_float(ctx->pop());
 287
 288 void factor_vm::primitive_float_eq()
 289 {
 290         POP_FLOATS(x,y);
 291         ctx->push(tag_boolean(x == y));
 292 }
 293
 294 void factor_vm::primitive_float_add()
 295 {
 296         POP_FLOATS(x,y);
 297         ctx->push(allot_float(x + y));
 298 }
 299
 300 void factor_vm::primitive_float_subtract()
 301 {
 302         POP_FLOATS(x,y);
 303         ctx->push(allot_float(x - y));
 304 }
 305
 306 void factor_vm::primitive_float_multiply()
 307 {
 308         POP_FLOATS(x,y);
 309         ctx->push(allot_float(x * y));
 310 }
 311
 312 void factor_vm::primitive_float_divfloat()
 313 {
 314         POP_FLOATS(x,y);
 315         ctx->push(allot_float(x / y));
 316 }
 317
 318 void factor_vm::primitive_float_mod()
 319 {
 320         POP_FLOATS(x,y);
 321         ctx->push(allot_float(fmod(x,y)));
 322 }
 323
 324 void factor_vm::primitive_float_less()
 325 {
 326         POP_FLOATS(x,y);
 327         ctx->push(tag_boolean(x < y));
 328 }
 329
 330 void factor_vm::primitive_float_lesseq()
 331 {
 332         POP_FLOATS(x,y);
 333         ctx->push(tag_boolean(x <= y));
 334 }
 335
 336 void factor_vm::primitive_float_greater()
 337 {
 338         POP_FLOATS(x,y);
 339         ctx->push(tag_boolean(x > y));
 340 }
 341
 342 void factor_vm::primitive_float_greatereq()
 343 {
 344         POP_FLOATS(x,y);
 345         ctx->push(tag_boolean(x >= y));
 346 }
 347
 348 void factor_vm::primitive_float_bits()
 349 {
 350         ctx->push(from_unsigned_4(float_bits((float)untag_float_check(ctx->pop()))));
 351 }
 352
 353 void factor_vm::primitive_bits_float()
 354 {
 355         ctx->push(allot_float(bits_float(to_cell(ctx->pop()))));
 356 }
 357
 358 void factor_vm::primitive_double_bits()
 359 {
 360         ctx->push(from_unsigned_8(double_bits(untag_float_check(ctx->pop()))));
 361 }
 362
 363 void factor_vm::primitive_bits_double()
 364 {
 365         ctx->push(allot_float(bits_double(to_unsigned_8(ctx->pop()))));
 366 }
 367
 368 /* Cannot allocate */
 369 fixnum factor_vm::to_fixnum(cell tagged)
 370 {
 371         switch(TAG(tagged))
 372         {
 373         case FIXNUM_TYPE:
 374                 return untag_fixnum(tagged);
 375         case BIGNUM_TYPE:
 376                 return bignum_to_fixnum(untag<bignum>(tagged));
 377         default:
 378                 type_error(FIXNUM_TYPE,tagged);
 379                 return 0; /* can't happen */
 380         }
 381 }
 382
 383 VM_C_API fixnum to_fixnum(cell tagged, factor_vm *parent)
 384 {
 385         return parent->to_fixnum(tagged);
 386 }
 387
 388 cell factor_vm::to_cell(cell tagged)
 389 {
 390         return (cell)to_fixnum(tagged);
 391 }
 392
 393 VM_C_API cell to_cell(cell tagged, factor_vm *parent)
 394 {
 395         return parent->to_cell(tagged);
 396 }
 397
 398 cell factor_vm::from_signed_1(s8 n)
 399 {
 400         return tag_fixnum(n);
 401 }
 402
 403 VM_C_API cell from_signed_1(s8 n, factor_vm *parent)
 404 {
 405         return parent->from_signed_1(n);
 406 }
 407
 408 cell factor_vm::from_unsigned_1(u8 n)
 409 {
 410         return tag_fixnum(n);
 411 }
 412
 413 VM_C_API cell from_unsigned_1(u8 n, factor_vm *parent)
 414 {
 415         return parent->from_unsigned_1(n);
 416 }
 417
 418 cell factor_vm::from_signed_2(s16 n)
 419 {
 420         return tag_fixnum(n);
 421 }
 422
 423 VM_C_API cell from_signed_2(s16 n, factor_vm *parent)
 424 {
 425         return parent->from_signed_2(n);
 426 }
 427
 428 cell factor_vm::from_unsigned_2(u16 n)
 429 {
 430         return tag_fixnum(n);
 431 }
 432
 433 VM_C_API cell from_unsigned_2(u16 n, factor_vm *parent)
 434 {
 435         return parent->from_unsigned_2(n);
 436 }
 437
 438 cell factor_vm::from_signed_4(s32 n)
 439 {
 440         return allot_integer(n);
 441 }
 442
 443 VM_C_API cell from_signed_4(s32 n, factor_vm *parent)
 444 {
 445         return parent->from_signed_4(n);
 446 }
 447
 448 cell factor_vm::from_unsigned_4(u32 n)
 449 {
 450         return allot_cell(n);
 451 }
 452
 453 VM_C_API cell from_unsigned_4(u32 n, factor_vm *parent)
 454 {
 455         return parent->from_unsigned_4(n);
 456 }
 457
 458 cell factor_vm::from_signed_cell(fixnum integer)
 459 {
 460         return allot_integer(integer);
 461 }
 462
 463 cell factor_vm::from_unsigned_cell(cell integer)
 464 {
 465         return allot_cell(integer);
 466 }
 467
 468 VM_C_API cell from_signed_cell(fixnum integer, factor_vm *parent)
 469 {
 470         return parent->from_signed_cell(integer);
 471 }
 472
 473 VM_C_API cell from_unsigned_cell(cell integer, factor_vm *parent)
 474 {
 475         return parent->from_unsigned_cell(integer);
 476 }
 477
 478 cell factor_vm::from_signed_8(s64 n)
 479 {
 480         if(n < fixnum_min || n > fixnum_max)
 481                 return tag<bignum>(long_long_to_bignum(n));
 482         else
 483                 return tag_fixnum((fixnum)n);
 484 }
 485
 486 VM_C_API cell from_signed_8(s64 n, factor_vm *parent)
 487 {
 488         return parent->from_signed_8(n);
 489 }
 490
 491 /* Cannot allocate */
 492 s64 factor_vm::to_signed_8(cell obj)
 493 {
 494         switch(tagged<object>(obj).type())
 495         {
 496         case FIXNUM_TYPE:
 497                 return untag_fixnum(obj);
 498         case BIGNUM_TYPE:
 499                 return bignum_to_long_long(untag<bignum>(obj));
 500         default:
 501                 type_error(BIGNUM_TYPE,obj);
 502                 return 0;
 503         }
 504 }
 505
 506 VM_C_API s64 to_signed_8(cell obj, factor_vm *parent)
 507 {
 508         return parent->to_signed_8(obj);
 509 }
 510
 511 cell factor_vm::from_unsigned_8(u64 n)
 512 {
 513         if(n > (u64)fixnum_max)
 514                 return tag<bignum>(ulong_long_to_bignum(n));
 515         else
 516                 return tag_fixnum((fixnum)n);
 517 }
 518
 519 VM_C_API cell from_unsigned_8(u64 n, factor_vm *parent)
 520 {
 521         return parent->from_unsigned_8(n);
 522 }
 523
 524 /* Cannot allocate */
 525 u64 factor_vm::to_unsigned_8(cell obj)
 526 {
 527         switch(tagged<object>(obj).type())
 528         {
 529         case FIXNUM_TYPE:
 530                 return untag_fixnum(obj);
 531         case BIGNUM_TYPE:
 532                 return bignum_to_ulong_long(untag<bignum>(obj));
 533         default:
 534                 type_error(BIGNUM_TYPE,obj);
 535                 return 0;
 536         }
 537 }
 538
 539 VM_C_API u64 to_unsigned_8(cell obj, factor_vm *parent)
 540 {
 541         return parent->to_unsigned_8(obj);
 542 }
 543
 544 VM_C_API cell from_float(float flo, factor_vm *parent)
 545 {
 546         return parent->allot_float(flo);
 547 }
 548
 549 /* Cannot allocate */
 550 float factor_vm::to_float(cell value)
 551 {
 552         return (float)untag_float_check(value);
 553 }
 554
 555 VM_C_API float to_float(cell value, factor_vm *parent)
 556 {
 557         return parent->to_float(value);
 558 }
 559
 560 VM_C_API cell from_double(double flo, factor_vm *parent)
 561 {
 562         return parent->allot_float(flo);
 563 }
 564
 565 /* Cannot allocate */
 566 double factor_vm::to_double(cell value)
 567 {
 568         return untag_float_check(value);
 569 }
 570
 571 VM_C_API double to_double(cell value, factor_vm *parent)
 572 {
 573         return parent->to_double(value);
 574 }
 575
 576 /* The fixnum+, fixnum- and fixnum* primitives are defined in cpu_*.S. On
 577 overflow, they call these functions. */
 578 inline void factor_vm::overflow_fixnum_add(fixnum x, fixnum y)
 579 {
 580         ctx->replace(tag<bignum>(fixnum_to_bignum(
 581                 untag_fixnum(x) + untag_fixnum(y))));
 582 }
 583
 584 VM_C_API void overflow_fixnum_add(fixnum x, fixnum y, factor_vm *parent)
 585 {
 586         parent->overflow_fixnum_add(x,y);
 587 }
 588
 589 inline void factor_vm::overflow_fixnum_subtract(fixnum x, fixnum y)
 590 {
 591         ctx->replace(tag<bignum>(fixnum_to_bignum(
 592                 untag_fixnum(x) - untag_fixnum(y))));
 593 }
 594
 595 VM_C_API void overflow_fixnum_subtract(fixnum x, fixnum y, factor_vm *parent)
 596 {
 597         parent->overflow_fixnum_subtract(x,y);
 598 }
 599
 600 inline void factor_vm::overflow_fixnum_multiply(fixnum x, fixnum y)
 601 {
 602         bignum *bx = fixnum_to_bignum(x);
 603         GC_BIGNUM(bx);
 604         bignum *by = fixnum_to_bignum(y);
 605         GC_BIGNUM(by);
 606         ctx->replace(tag<bignum>(bignum_multiply(bx,by)));
 607 }
 608
 609 VM_C_API void overflow_fixnum_multiply(fixnum x, fixnum y, factor_vm *parent)
 610 {
 611         parent->overflow_fixnum_multiply(x,y);
 612 }
 613
 614 }