]> gitweb.factorcode.org Git - factor.git/blobdiff - vm/math.cpp
io.streams.256color: faster by caching styles
[factor.git] / vm / math.cpp
index 35610186e00be452afeaad7c163833d774f33e65..ef56d265dcf95eaae7962feafbc0ef8dd03a48e5 100644 (file)
@@ -1,4 +1,7 @@
 #include "master.hpp"
+#include <sstream>
+#include <iomanip>
+#include <stdexcept>
 
 namespace factor {
 
@@ -14,28 +17,28 @@ void factor_vm::primitive_float_to_fixnum() {
   ctx->replace(tag_fixnum(float_to_fixnum(ctx->peek())));
 }
 
-/* does not allocate, even though from_signed_cell can allocate */
-/* Division can only overflow when we are dividing the most negative fixnum
-by -1. */
+// does not allocate, even though from_signed_cell can allocate
+// Division can only overflow when we are dividing the most negative fixnum
+// by -1.
 void factor_vm::primitive_fixnum_divint() {
   fixnum y = untag_fixnum(ctx->pop());
   fixnum x = untag_fixnum(ctx->peek());
   fixnum result = x / y;
   if (result == -fixnum_min)
-    /* Does not allocate */
+    // Does not allocate
     ctx->replace(from_signed_cell(-fixnum_min));
   else
     ctx->replace(tag_fixnum(result));
 }
 
-/* does not allocate, even though from_signed_cell can allocate */
+// does not allocate, even though from_signed_cell can allocate
 void factor_vm::primitive_fixnum_divmod() {
   cell* s0 = (cell*)(ctx->datastack);
   cell* s1 = (cell*)(ctx->datastack - sizeof(cell));
   fixnum y = untag_fixnum(*s0);
   fixnum x = untag_fixnum(*s1);
   if (y == -1 && x == fixnum_min) {
-    /* Does not allocate */
+    // Does not allocate
     *s1 = from_signed_cell(-fixnum_min);
     *s0 = tag_fixnum(0);
   } else {
@@ -44,10 +47,9 @@ void factor_vm::primitive_fixnum_divmod() {
   }
 }
 
-/*
- * If we're shifting right by n bits, we won't overflow as long as none of the
- * high WORD_SIZE-TAG_BITS-n bits are set.
- */
+
+// If we're shifting right by n bits, we won't overflow as long as none of the
+// high WORD_SIZE-TAG_BITS-n bits are set.
 inline fixnum factor_vm::sign_mask(fixnum x) {
     return x >> (WORD_SIZE - 1);
 }
@@ -60,7 +62,7 @@ inline fixnum factor_vm::branchless_abs(fixnum x) {
   return (x ^ sign_mask(x)) - sign_mask(x);
 }
 
-/* Allocates memory */
+// Allocates memory
 void factor_vm::primitive_fixnum_shift() {
   fixnum y = untag_fixnum(ctx->pop());
   fixnum x = untag_fixnum(ctx->peek());
@@ -82,12 +84,12 @@ void factor_vm::primitive_fixnum_shift() {
   ctx->replace(tag<bignum>(bignum_arithmetic_shift(fixnum_to_bignum(x), y)));
 }
 
-/* Allocates memory */
+// Allocates memory
 void factor_vm::primitive_fixnum_to_bignum() {
   ctx->replace(tag<bignum>(fixnum_to_bignum(untag_fixnum(ctx->peek()))));
 }
 
-/* Allocates memory */
+// Allocates memory
 void factor_vm::primitive_float_to_bignum() {
   ctx->replace(tag<bignum>(float_to_bignum(ctx->peek())));
 }
@@ -101,31 +103,31 @@ void factor_vm::primitive_bignum_eq() {
   ctx->replace(tag_boolean(bignum_equal_p(x, y)));
 }
 
-/* Allocates memory */
+// Allocates memory
 void factor_vm::primitive_bignum_add() {
   POP_BIGNUMS(x, y);
   ctx->replace(tag<bignum>(bignum_add(x, y)));
 }
 
-/* Allocates memory */
+// Allocates memory
 void factor_vm::primitive_bignum_subtract() {
   POP_BIGNUMS(x, y);
   ctx->replace(tag<bignum>(bignum_subtract(x, y)));
 }
 
-/* Allocates memory */
+// Allocates memory
 void factor_vm::primitive_bignum_multiply() {
   POP_BIGNUMS(x, y);
   ctx->replace(tag<bignum>(bignum_multiply(x, y)));
 }
 
-/* Allocates memory */
+// Allocates memory
 void factor_vm::primitive_bignum_divint() {
   POP_BIGNUMS(x, y);
   ctx->replace(tag<bignum>(bignum_quotient(x, y)));
 }
 
-/* Allocates memory */
+// Allocates memory
 void factor_vm::primitive_bignum_divmod() {
   cell* s0 = (cell*)(ctx->datastack);
   cell* s1 = (cell*)(ctx->datastack - sizeof(cell));
@@ -134,12 +136,13 @@ void factor_vm::primitive_bignum_divmod() {
   bignum* q, *r;
   bignum_divide(x, y, &q, &r);
   *s1 = tag<bignum>(q);
-  *s0 = tag<bignum>(r);
+  *s0 = bignum_maybe_to_fixnum(r);
 }
 
 void factor_vm::primitive_bignum_mod() {
   POP_BIGNUMS(x, y);
-  ctx->replace(tag<bignum>(bignum_remainder(x, y)));
+  cell val = bignum_maybe_to_fixnum(bignum_remainder(x, y));
+  ctx->replace(val);
 }
 
 void factor_vm::primitive_bignum_gcd() {
@@ -162,7 +165,7 @@ void factor_vm::primitive_bignum_xor() {
   ctx->replace(tag<bignum>(bignum_bitwise_xor(x, y)));
 }
 
-/* Allocates memory */
+// Allocates memory
 void factor_vm::primitive_bignum_shift() {
   fixnum y = untag_fixnum(ctx->pop());
   bignum* x = untag<bignum>(ctx->peek());
@@ -171,22 +174,22 @@ void factor_vm::primitive_bignum_shift() {
 
 void factor_vm::primitive_bignum_less() {
   POP_BIGNUMS(x, y);
-  ctx->replace(tag_boolean(bignum_compare(x, y) == bignum_comparison_less));
+  ctx->replace(tag_boolean(bignum_compare(x, y) == BIGNUM_COMPARISON_LESS));
 }
 
 void factor_vm::primitive_bignum_lesseq() {
   POP_BIGNUMS(x, y);
-  ctx->replace(tag_boolean(bignum_compare(x, y) != bignum_comparison_greater));
+  ctx->replace(tag_boolean(bignum_compare(x, y) != BIGNUM_COMPARISON_GREATER));
 }
 
 void factor_vm::primitive_bignum_greater() {
   POP_BIGNUMS(x, y);
-  ctx->replace(tag_boolean(bignum_compare(x, y) == bignum_comparison_greater));
+  ctx->replace(tag_boolean(bignum_compare(x, y) == BIGNUM_COMPARISON_GREATER));
 }
 
 void factor_vm::primitive_bignum_greatereq() {
   POP_BIGNUMS(x, y);
-  ctx->replace(tag_boolean(bignum_compare(x, y) != bignum_comparison_less));
+  ctx->replace(tag_boolean(bignum_compare(x, y) != BIGNUM_COMPARISON_LESS));
 }
 
 void factor_vm::primitive_bignum_not() {
@@ -203,17 +206,49 @@ void factor_vm::primitive_bignum_log2() {
   ctx->replace(tag<bignum>(bignum_integer_length(untag<bignum>(ctx->peek()))));
 }
 
-/* Allocates memory */
+// Allocates memory
 void factor_vm::primitive_fixnum_to_float() {
   ctx->replace(allot_float(fixnum_to_float(ctx->peek())));
 }
 
-/* Allocates memory */
+// Allocates memory
 void factor_vm::primitive_format_float() {
-  byte_array* array = allot_byte_array(100);
+  char* locale = alien_offset(ctx->pop());
   char* format = alien_offset(ctx->pop());
+  fixnum precision = untag_fixnum(ctx->pop());
+  fixnum width = untag_fixnum(ctx->pop());
+  char* fill = alien_offset(ctx->pop());
   double value = untag_float_check(ctx->peek());
-  SNPRINTF(array->data<char>(), 99, format, value);
+  std::ostringstream localized_stream;
+  try {
+    localized_stream.imbue(std::locale(locale));
+  } catch (const runtime_error&) {
+    byte_array* array = allot_byte_array(0);
+    ctx->replace(tag<byte_array>(array));
+    return;
+  }
+  switch (format[0]) {
+    case 'f': localized_stream << std::fixed; break;
+    case 'e': localized_stream << std::scientific; break;
+  }
+  if (isupper(format[0])) {
+    localized_stream << std::uppercase;
+  }
+  if (fill[0] != '\0') {
+    localized_stream << std::setfill(fill[0]);
+  }
+  if (width >= 0) {
+    localized_stream << std::setw(static_cast<int>(width));
+  }
+  if (precision >= 0) {
+    localized_stream << std::setprecision(static_cast<int>(precision));
+  }
+  localized_stream << value;
+  const std::string& tmp = localized_stream.str();
+  const char* cstr = tmp.c_str();
+  size_t size = tmp.length();
+  byte_array* array = allot_byte_array(size);
+  memcpy(array->data<char>(), cstr, size);
   ctx->replace(tag<byte_array>(array));
 }
 
@@ -226,25 +261,25 @@ void factor_vm::primitive_float_eq() {
   ctx->replace(tag_boolean(x == y));
 }
 
-/* Allocates memory */
+// Allocates memory
 void factor_vm::primitive_float_add() {
   POP_FLOATS(x, y);
   ctx->replace(allot_float(x + y));
 }
 
-/* Allocates memory */
+// Allocates memory
 void factor_vm::primitive_float_subtract() {
   POP_FLOATS(x, y);
   ctx->replace(allot_float(x - y));
 }
 
-/* Allocates memory */
+// Allocates memory
 void factor_vm::primitive_float_multiply() {
   POP_FLOATS(x, y);
   ctx->replace(allot_float(x * y));
 }
 
-/* Allocates memory */
+// Allocates memory
 void factor_vm::primitive_float_divfloat() {
   POP_FLOATS(x, y);
   ctx->replace(allot_float(x / y));
@@ -270,13 +305,13 @@ void factor_vm::primitive_float_greatereq() {
   ctx->replace(tag_boolean(x >= y));
 }
 
-/* Allocates memory */
+// Allocates memory
 void factor_vm::primitive_float_bits() {
   ctx->replace(
       from_unsigned_cell(float_bits((float)untag_float_check(ctx->peek()))));
 }
 
-/* Allocates memory */
+// Allocates memory
 void factor_vm::primitive_bits_float() {
   ctx->replace(allot_float(bits_float((uint32_t)to_cell(ctx->peek()))));
 }
@@ -285,12 +320,12 @@ void factor_vm::primitive_double_bits() {
   ctx->replace(from_unsigned_8(double_bits(untag_float_check(ctx->peek()))));
 }
 
-/* Allocates memory */
+// Allocates memory
 void factor_vm::primitive_bits_double() {
   ctx->replace(allot_float(bits_double(to_unsigned_8(ctx->peek()))));
 }
 
-/* Cannot allocate. */
+// Cannot allocate.
 #define CELL_TO_FOO(name, type, converter)              \
   type factor_vm::name(cell tagged) {                   \
     switch (TAG(tagged)) {                              \
@@ -313,17 +348,17 @@ CELL_TO_FOO(to_cell, cell, bignum_to_cell)
 CELL_TO_FOO(to_signed_8, int64_t, bignum_to_long_long)
 CELL_TO_FOO(to_unsigned_8, uint64_t, bignum_to_ulong_long)
 
-/* Allocates memory */
+// Allocates memory
 VM_C_API cell from_signed_cell(fixnum integer, factor_vm* parent) {
   return parent->from_signed_cell(integer);
 }
 
-/* Allocates memory */
+// Allocates memory
 VM_C_API cell from_unsigned_cell(cell integer, factor_vm* parent) {
   return parent->from_unsigned_cell(integer);
 }
 
-/* Allocates memory */
+// Allocates memory
 cell factor_vm::from_signed_8(int64_t n) {
   if (n < fixnum_min || n > fixnum_max)
     return tag<bignum>(long_long_to_bignum(n));
@@ -335,7 +370,7 @@ VM_C_API cell from_signed_8(int64_t n, factor_vm* parent) {
   return parent->from_signed_8(n);
 }
 
-/* Allocates memory */
+// Allocates memory
 cell factor_vm::from_unsigned_8(uint64_t n) {
   if (n > (uint64_t)fixnum_max)
     return tag<bignum>(ulong_long_to_bignum(n));
@@ -347,17 +382,17 @@ VM_C_API cell from_unsigned_8(uint64_t n, factor_vm* parent) {
   return parent->from_unsigned_8(n);
 }
 
-/* Cannot allocate */
+// Cannot allocate
 float factor_vm::to_float(cell value) {
   return (float)untag_float_check(value);
 }
 
-/* Cannot allocate */
+// Cannot allocate
 double factor_vm::to_double(cell value) { return untag_float_check(value); }
 
-/* The fixnum+, fixnum- and fixnum* primitives are defined in cpu_*.S. On
-   overflow, they call these functions. */
-/* Allocates memory */
+// The fixnum+, fixnum- and fixnum* primitives are defined in cpu_*.S. On
+// overflow, they call these functions.
+// Allocates memory
 inline void factor_vm::overflow_fixnum_add(fixnum x, fixnum y) {
   ctx->replace(
       tag<bignum>(fixnum_to_bignum(untag_fixnum(x) + untag_fixnum(y))));
@@ -367,7 +402,7 @@ VM_C_API void overflow_fixnum_add(fixnum x, fixnum y, factor_vm* parent) {
   parent->overflow_fixnum_add(x, y);
 }
 
-/* Allocates memory */
+// Allocates memory
 inline void factor_vm::overflow_fixnum_subtract(fixnum x, fixnum y) {
   ctx->replace(
       tag<bignum>(fixnum_to_bignum(untag_fixnum(x) - untag_fixnum(y))));
@@ -377,7 +412,7 @@ VM_C_API void overflow_fixnum_subtract(fixnum x, fixnum y, factor_vm* parent) {
   parent->overflow_fixnum_subtract(x, y);
 }
 
-/* Allocates memory */
+// Allocates memory
 inline void factor_vm::overflow_fixnum_multiply(fixnum x, fixnum y) {
   data_root<bignum> bx(fixnum_to_bignum(x), this);
   data_root<bignum> by(fixnum_to_bignum(y), this);