int.go 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152
  1. package encoder
  2. import (
  3. "unsafe"
  4. )
  5. var endianness int
  6. func init() {
  7. var b [2]byte
  8. *(*uint16)(unsafe.Pointer(&b)) = uint16(0xABCD)
  9. switch b[0] {
  10. case 0xCD:
  11. endianness = 0 // LE
  12. case 0xAB:
  13. endianness = 1 // BE
  14. default:
  15. panic("could not determine endianness")
  16. }
  17. }
  18. // "00010203...96979899" cast to []uint16
  19. var intLELookup = [100]uint16{
  20. 0x3030, 0x3130, 0x3230, 0x3330, 0x3430, 0x3530, 0x3630, 0x3730, 0x3830, 0x3930,
  21. 0x3031, 0x3131, 0x3231, 0x3331, 0x3431, 0x3531, 0x3631, 0x3731, 0x3831, 0x3931,
  22. 0x3032, 0x3132, 0x3232, 0x3332, 0x3432, 0x3532, 0x3632, 0x3732, 0x3832, 0x3932,
  23. 0x3033, 0x3133, 0x3233, 0x3333, 0x3433, 0x3533, 0x3633, 0x3733, 0x3833, 0x3933,
  24. 0x3034, 0x3134, 0x3234, 0x3334, 0x3434, 0x3534, 0x3634, 0x3734, 0x3834, 0x3934,
  25. 0x3035, 0x3135, 0x3235, 0x3335, 0x3435, 0x3535, 0x3635, 0x3735, 0x3835, 0x3935,
  26. 0x3036, 0x3136, 0x3236, 0x3336, 0x3436, 0x3536, 0x3636, 0x3736, 0x3836, 0x3936,
  27. 0x3037, 0x3137, 0x3237, 0x3337, 0x3437, 0x3537, 0x3637, 0x3737, 0x3837, 0x3937,
  28. 0x3038, 0x3138, 0x3238, 0x3338, 0x3438, 0x3538, 0x3638, 0x3738, 0x3838, 0x3938,
  29. 0x3039, 0x3139, 0x3239, 0x3339, 0x3439, 0x3539, 0x3639, 0x3739, 0x3839, 0x3939,
  30. }
  31. var intBELookup = [100]uint16{
  32. 0x3030, 0x3031, 0x3032, 0x3033, 0x3034, 0x3035, 0x3036, 0x3037, 0x3038, 0x3039,
  33. 0x3130, 0x3131, 0x3132, 0x3133, 0x3134, 0x3135, 0x3136, 0x3137, 0x3138, 0x3139,
  34. 0x3230, 0x3231, 0x3232, 0x3233, 0x3234, 0x3235, 0x3236, 0x3237, 0x3238, 0x3239,
  35. 0x3330, 0x3331, 0x3332, 0x3333, 0x3334, 0x3335, 0x3336, 0x3337, 0x3338, 0x3339,
  36. 0x3430, 0x3431, 0x3432, 0x3433, 0x3434, 0x3435, 0x3436, 0x3437, 0x3438, 0x3439,
  37. 0x3530, 0x3531, 0x3532, 0x3533, 0x3534, 0x3535, 0x3536, 0x3537, 0x3538, 0x3539,
  38. 0x3630, 0x3631, 0x3632, 0x3633, 0x3634, 0x3635, 0x3636, 0x3637, 0x3638, 0x3639,
  39. 0x3730, 0x3731, 0x3732, 0x3733, 0x3734, 0x3735, 0x3736, 0x3737, 0x3738, 0x3739,
  40. 0x3830, 0x3831, 0x3832, 0x3833, 0x3834, 0x3835, 0x3836, 0x3837, 0x3838, 0x3839,
  41. 0x3930, 0x3931, 0x3932, 0x3933, 0x3934, 0x3935, 0x3936, 0x3937, 0x3938, 0x3939,
  42. }
  43. var intLookup = [2]*[100]uint16{&intLELookup, &intBELookup}
  44. func numMask(numBitSize uint8) uint64 {
  45. return 1<<numBitSize - 1
  46. }
  47. func AppendInt(_ *RuntimeContext, out []byte, p uintptr, code *Opcode) []byte {
  48. var u64 uint64
  49. switch code.NumBitSize {
  50. case 8:
  51. u64 = (uint64)(**(**uint8)(unsafe.Pointer(&p)))
  52. case 16:
  53. u64 = (uint64)(**(**uint16)(unsafe.Pointer(&p)))
  54. case 32:
  55. u64 = (uint64)(**(**uint32)(unsafe.Pointer(&p)))
  56. case 64:
  57. u64 = **(**uint64)(unsafe.Pointer(&p))
  58. }
  59. mask := numMask(code.NumBitSize)
  60. n := u64 & mask
  61. negative := (u64>>(code.NumBitSize-1))&1 == 1
  62. if !negative {
  63. if n < 10 {
  64. return append(out, byte(n+'0'))
  65. } else if n < 100 {
  66. u := intLELookup[n]
  67. return append(out, byte(u), byte(u>>8))
  68. }
  69. } else {
  70. n = -n & mask
  71. }
  72. lookup := intLookup[endianness]
  73. var b [22]byte
  74. u := (*[11]uint16)(unsafe.Pointer(&b))
  75. i := 11
  76. for n >= 100 {
  77. j := n % 100
  78. n /= 100
  79. i--
  80. u[i] = lookup[j]
  81. }
  82. i--
  83. u[i] = lookup[n]
  84. i *= 2 // convert to byte index
  85. if n < 10 {
  86. i++ // remove leading zero
  87. }
  88. if negative {
  89. i--
  90. b[i] = '-'
  91. }
  92. return append(out, b[i:]...)
  93. }
  94. func AppendUint(_ *RuntimeContext, out []byte, p uintptr, code *Opcode) []byte {
  95. var u64 uint64
  96. switch code.NumBitSize {
  97. case 8:
  98. u64 = (uint64)(**(**uint8)(unsafe.Pointer(&p)))
  99. case 16:
  100. u64 = (uint64)(**(**uint16)(unsafe.Pointer(&p)))
  101. case 32:
  102. u64 = (uint64)(**(**uint32)(unsafe.Pointer(&p)))
  103. case 64:
  104. u64 = **(**uint64)(unsafe.Pointer(&p))
  105. }
  106. mask := numMask(code.NumBitSize)
  107. n := u64 & mask
  108. if n < 10 {
  109. return append(out, byte(n+'0'))
  110. } else if n < 100 {
  111. u := intLELookup[n]
  112. return append(out, byte(u), byte(u>>8))
  113. }
  114. lookup := intLookup[endianness]
  115. var b [22]byte
  116. u := (*[11]uint16)(unsafe.Pointer(&b))
  117. i := 11
  118. for n >= 100 {
  119. j := n % 100
  120. n /= 100
  121. i--
  122. u[i] = lookup[j]
  123. }
  124. i--
  125. u[i] = lookup[n]
  126. i *= 2 // convert to byte index
  127. if n < 10 {
  128. i++ // remove leading zero
  129. }
  130. return append(out, b[i:]...)
  131. }