lj_vmmath.c 3.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
  1. /*
  2. ** Math helper functions for assembler VM.
  3. ** Copyright (C) 2005-2014 Mike Pall. See Copyright Notice in luajit.h
  4. */
  5. #define lj_vmmath_c
  6. #define LUA_CORE
  7. #include <errno.h>
  8. #include <math.h>
  9. #include "lj_obj.h"
  10. #include "lj_ir.h"
  11. #include "lj_vm.h"
  12. /* -- Helper functions for generated machine code ------------------------- */
  13. #if LJ_TARGET_X86ORX64
  14. /* Wrapper functions to avoid linker issues on OSX. */
  15. LJ_FUNCA double lj_vm_sinh(double x) { return sinh(x); }
  16. LJ_FUNCA double lj_vm_cosh(double x) { return cosh(x); }
  17. LJ_FUNCA double lj_vm_tanh(double x) { return tanh(x); }
  18. #endif
  19. #if !LJ_TARGET_X86ORX64
  20. double lj_vm_foldarith(double x, double y, int op)
  21. {
  22. switch (op) {
  23. case IR_ADD - IR_ADD: return x+y; break;
  24. case IR_SUB - IR_ADD: return x-y; break;
  25. case IR_MUL - IR_ADD: return x*y; break;
  26. case IR_DIV - IR_ADD: return x/y; break;
  27. case IR_MOD - IR_ADD: return x-lj_vm_floor(x/y)*y; break;
  28. case IR_POW - IR_ADD: return pow(x, y); break;
  29. case IR_NEG - IR_ADD: return -x; break;
  30. case IR_ABS - IR_ADD: return fabs(x); break;
  31. #if LJ_HASJIT
  32. case IR_ATAN2 - IR_ADD: return atan2(x, y); break;
  33. case IR_LDEXP - IR_ADD: return ldexp(x, (int)y); break;
  34. case IR_MIN - IR_ADD: return x > y ? y : x; break;
  35. case IR_MAX - IR_ADD: return x < y ? y : x; break;
  36. #endif
  37. default: return x;
  38. }
  39. }
  40. #endif
  41. #if LJ_HASJIT
  42. #ifdef LUAJIT_NO_LOG2
  43. double lj_vm_log2(double a)
  44. {
  45. return log(a) * 1.4426950408889634074;
  46. }
  47. #endif
  48. #ifdef LUAJIT_NO_EXP2
  49. double lj_vm_exp2(double a)
  50. {
  51. return exp(a * 0.6931471805599453);
  52. }
  53. #endif
  54. #if !(LJ_TARGET_ARM || LJ_TARGET_PPC)
  55. int32_t LJ_FASTCALL lj_vm_modi(int32_t a, int32_t b)
  56. {
  57. uint32_t y, ua, ub;
  58. lua_assert(b != 0); /* This must be checked before using this function. */
  59. ua = a < 0 ? (uint32_t)-a : (uint32_t)a;
  60. ub = b < 0 ? (uint32_t)-b : (uint32_t)b;
  61. y = ua % ub;
  62. if (y != 0 && (a^b) < 0) y = y - ub;
  63. if (((int32_t)y^b) < 0) y = (uint32_t)-(int32_t)y;
  64. return (int32_t)y;
  65. }
  66. #endif
  67. #if !LJ_TARGET_X86ORX64
  68. /* Unsigned x^k. */
  69. static double lj_vm_powui(double x, uint32_t k)
  70. {
  71. double y;
  72. lua_assert(k != 0);
  73. for (; (k & 1) == 0; k >>= 1) x *= x;
  74. y = x;
  75. if ((k >>= 1) != 0) {
  76. for (;;) {
  77. x *= x;
  78. if (k == 1) break;
  79. if (k & 1) y *= x;
  80. k >>= 1;
  81. }
  82. y *= x;
  83. }
  84. return y;
  85. }
  86. /* Signed x^k. */
  87. double lj_vm_powi(double x, int32_t k)
  88. {
  89. if (k > 1)
  90. return lj_vm_powui(x, (uint32_t)k);
  91. else if (k == 1)
  92. return x;
  93. else if (k == 0)
  94. return 1.0;
  95. else
  96. return 1.0 / lj_vm_powui(x, (uint32_t)-k);
  97. }
  98. /* Computes fpm(x) for extended math functions. */
  99. double lj_vm_foldfpm(double x, int fpm)
  100. {
  101. switch (fpm) {
  102. case IRFPM_FLOOR: return lj_vm_floor(x);
  103. case IRFPM_CEIL: return lj_vm_ceil(x);
  104. case IRFPM_TRUNC: return lj_vm_trunc(x);
  105. case IRFPM_SQRT: return sqrt(x);
  106. case IRFPM_EXP: return exp(x);
  107. case IRFPM_EXP2: return lj_vm_exp2(x);
  108. case IRFPM_LOG: return log(x);
  109. case IRFPM_LOG2: return lj_vm_log2(x);
  110. case IRFPM_LOG10: return log10(x);
  111. case IRFPM_SIN: return sin(x);
  112. case IRFPM_COS: return cos(x);
  113. case IRFPM_TAN: return tan(x);
  114. default: lua_assert(0);
  115. }
  116. return 0;
  117. }
  118. #endif
  119. #if LJ_HASFFI
  120. int lj_vm_errno(void)
  121. {
  122. return errno;
  123. }
  124. #endif
  125. #endif