value_number.go 6.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367
  1. package otto
  2. import (
  3. "fmt"
  4. "math"
  5. "regexp"
  6. "strconv"
  7. "strings"
  8. )
  9. var stringToNumberParseInteger = regexp.MustCompile(`^(?:0[xX])`)
  10. func parseNumber(value string) float64 {
  11. value = strings.Trim(value, builtinString_trim_whitespace)
  12. if value == "" {
  13. return 0
  14. }
  15. parseFloat := false
  16. if strings.IndexRune(value, '.') != -1 {
  17. parseFloat = true
  18. } else if stringToNumberParseInteger.MatchString(value) {
  19. parseFloat = false
  20. } else {
  21. parseFloat = true
  22. }
  23. if parseFloat {
  24. number, err := strconv.ParseFloat(value, 64)
  25. if err != nil && err.(*strconv.NumError).Err != strconv.ErrRange {
  26. return math.NaN()
  27. }
  28. return number
  29. }
  30. number, err := strconv.ParseInt(value, 0, 64)
  31. if err != nil {
  32. return math.NaN()
  33. }
  34. return float64(number)
  35. }
  36. func (value Value) float64() float64 {
  37. switch value.kind {
  38. case valueUndefined:
  39. return math.NaN()
  40. case valueNull:
  41. return 0
  42. }
  43. switch value := value.value.(type) {
  44. case bool:
  45. if value {
  46. return 1
  47. }
  48. return 0
  49. case int:
  50. return float64(value)
  51. case int8:
  52. return float64(value)
  53. case int16:
  54. return float64(value)
  55. case int32:
  56. return float64(value)
  57. case int64:
  58. return float64(value)
  59. case uint:
  60. return float64(value)
  61. case uint8:
  62. return float64(value)
  63. case uint16:
  64. return float64(value)
  65. case uint32:
  66. return float64(value)
  67. case uint64:
  68. return float64(value)
  69. case float64:
  70. return value
  71. case string:
  72. return parseNumber(value)
  73. case *_object:
  74. return value.DefaultValue(defaultValueHintNumber).float64()
  75. }
  76. panic(fmt.Errorf("toFloat(%T)", value.value))
  77. }
  78. const (
  79. float_2_32 float64 = 4294967296.0
  80. float_2_31 float64 = 2147483648.0
  81. float_2_16 float64 = 65536.0
  82. sqrt1_2 float64 = math.Sqrt2 / 2
  83. )
  84. const (
  85. maxUint32 = math.MaxUint32
  86. maxInt = int(^uint(0) >> 1)
  87. // int64
  88. int64_maxInt8 int64 = math.MaxInt8
  89. int64_minInt8 int64 = math.MinInt8
  90. int64_maxInt16 int64 = math.MaxInt16
  91. int64_minInt16 int64 = math.MinInt16
  92. int64_maxInt32 int64 = math.MaxInt32
  93. int64_minInt32 int64 = math.MinInt32
  94. int64_maxUint8 int64 = math.MaxUint8
  95. int64_maxUint16 int64 = math.MaxUint16
  96. int64_maxUint32 int64 = math.MaxUint32
  97. // float64
  98. float_maxInt float64 = float64(int(^uint(0) >> 1))
  99. float_minInt float64 = float64(int(-maxInt - 1))
  100. float_maxUint float64 = float64(uint(^uint(0)))
  101. float_maxUint64 float64 = math.MaxUint64
  102. float_maxInt64 float64 = math.MaxInt64
  103. float_minInt64 float64 = math.MinInt64
  104. )
  105. func toIntegerFloat(value Value) float64 {
  106. float := value.float64()
  107. if math.IsInf(float, 0) {
  108. } else if math.IsNaN(float) {
  109. float = 0
  110. } else if float > 0 {
  111. float = math.Floor(float)
  112. } else {
  113. float = math.Ceil(float)
  114. }
  115. return float
  116. }
  117. type _numberKind int
  118. const (
  119. numberInteger _numberKind = iota // 3.0 => 3.0
  120. numberFloat // 3.14159 => 3.0, 1+2**63 > 2**63-1
  121. numberInfinity // Infinity => 2**63-1
  122. numberNaN // NaN => 0
  123. )
  124. type _number struct {
  125. kind _numberKind
  126. int64 int64
  127. float64 float64
  128. }
  129. // FIXME
  130. // http://www.goinggo.net/2013/08/gustavos-ieee-754-brain-teaser.html
  131. // http://bazaar.launchpad.net/~niemeyer/strepr/trunk/view/6/strepr.go#L160
  132. func (value Value) number() (number _number) {
  133. switch value := value.value.(type) {
  134. case int8:
  135. number.int64 = int64(value)
  136. return
  137. case int16:
  138. number.int64 = int64(value)
  139. return
  140. case uint8:
  141. number.int64 = int64(value)
  142. return
  143. case uint16:
  144. number.int64 = int64(value)
  145. return
  146. case uint32:
  147. number.int64 = int64(value)
  148. return
  149. case int:
  150. number.int64 = int64(value)
  151. return
  152. case int64:
  153. number.int64 = value
  154. return
  155. }
  156. float := value.float64()
  157. if float == 0 {
  158. return
  159. }
  160. number.kind = numberFloat
  161. number.float64 = float
  162. if math.IsNaN(float) {
  163. number.kind = numberNaN
  164. return
  165. }
  166. if math.IsInf(float, 0) {
  167. number.kind = numberInfinity
  168. }
  169. if float >= float_maxInt64 {
  170. number.int64 = math.MaxInt64
  171. return
  172. }
  173. if float <= float_minInt64 {
  174. number.int64 = math.MinInt64
  175. return
  176. }
  177. var integer float64
  178. if float > 0 {
  179. integer = math.Floor(float)
  180. } else {
  181. integer = math.Ceil(float)
  182. }
  183. if float == integer {
  184. number.kind = numberInteger
  185. }
  186. number.int64 = int64(float)
  187. return
  188. }
  189. // ECMA 262: 9.5
  190. func toInt32(value Value) int32 {
  191. {
  192. switch value := value.value.(type) {
  193. case int8:
  194. return int32(value)
  195. case int16:
  196. return int32(value)
  197. case int32:
  198. return value
  199. }
  200. }
  201. floatValue := value.float64()
  202. if math.IsNaN(floatValue) || math.IsInf(floatValue, 0) {
  203. return 0
  204. }
  205. if floatValue == 0 { // This will work for +0 & -0
  206. return 0
  207. }
  208. remainder := math.Mod(floatValue, float_2_32)
  209. if remainder > 0 {
  210. remainder = math.Floor(remainder)
  211. } else {
  212. remainder = math.Ceil(remainder) + float_2_32
  213. }
  214. if remainder > float_2_31 {
  215. return int32(remainder - float_2_32)
  216. }
  217. return int32(remainder)
  218. }
  219. func toUint32(value Value) uint32 {
  220. {
  221. switch value := value.value.(type) {
  222. case int8:
  223. return uint32(value)
  224. case int16:
  225. return uint32(value)
  226. case uint8:
  227. return uint32(value)
  228. case uint16:
  229. return uint32(value)
  230. case uint32:
  231. return value
  232. }
  233. }
  234. floatValue := value.float64()
  235. if math.IsNaN(floatValue) || math.IsInf(floatValue, 0) {
  236. return 0
  237. }
  238. if floatValue == 0 {
  239. return 0
  240. }
  241. remainder := math.Mod(floatValue, float_2_32)
  242. if remainder > 0 {
  243. remainder = math.Floor(remainder)
  244. } else {
  245. remainder = math.Ceil(remainder) + float_2_32
  246. }
  247. return uint32(remainder)
  248. }
  249. func toUint16(value Value) uint16 {
  250. {
  251. switch value := value.value.(type) {
  252. case int8:
  253. return uint16(value)
  254. case uint8:
  255. return uint16(value)
  256. case uint16:
  257. return value
  258. }
  259. }
  260. floatValue := value.float64()
  261. if math.IsNaN(floatValue) || math.IsInf(floatValue, 0) {
  262. return 0
  263. }
  264. if floatValue == 0 {
  265. return 0
  266. }
  267. remainder := math.Mod(floatValue, float_2_16)
  268. if remainder > 0 {
  269. remainder = math.Floor(remainder)
  270. } else {
  271. remainder = math.Ceil(remainder) + float_2_16
  272. }
  273. return uint16(remainder)
  274. }
  275. // toIntSign returns sign of a number converted to -1, 0 ,1
  276. func toIntSign(value Value) int {
  277. switch value := value.value.(type) {
  278. case int8:
  279. if int8(value) > 0 {
  280. return 1
  281. } else if int8(value) < 0 {
  282. return -1
  283. }
  284. return 0
  285. case int16:
  286. if int16(value) > 0 {
  287. return 1
  288. } else if int16(value) < 0 {
  289. return -1
  290. }
  291. return 0
  292. case int32:
  293. if int32(value) > 0 {
  294. return 1
  295. } else if int32(value) < 0 {
  296. return -1
  297. }
  298. return 0
  299. case uint8:
  300. if uint8(value) > 0 {
  301. return 1
  302. } else if uint8(value) < 0 {
  303. return -1
  304. }
  305. return 0
  306. case uint16:
  307. if uint16(value) > 0 {
  308. return 1
  309. } else if uint16(value) < 0 {
  310. return -1
  311. }
  312. return 0
  313. case uint32:
  314. if uint32(value) > 0 {
  315. return 1
  316. } else if uint32(value) < 0 {
  317. return -1
  318. }
  319. return 0
  320. }
  321. floatValue := value.float64()
  322. switch {
  323. case math.IsNaN(floatValue), math.IsInf(floatValue, 0):
  324. return 0
  325. case floatValue == 0:
  326. return 0
  327. case floatValue > 0:
  328. return 1
  329. default:
  330. return -1
  331. }
  332. }