forked from xitongsys/parquet-go
-
Notifications
You must be signed in to change notification settings - Fork 0
/
binarywrite.go
83 lines (74 loc) · 1.67 KB
/
binarywrite.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
package encoding
import (
"io"
"math"
)
//LittleEndian
func GetUint32(n interface{}) uint32 {
switch n.(type) {
case int, int8, int16, int32:
return uint32(n.(int32))
default:
return n.(uint32)
}
}
func BinaryWriteINT32(w io.Writer, nums []interface{}) {
buf := make([]byte, len(nums)*4)
for i, n := range nums {
v := GetUint32(n)
buf[i*4+0] = byte(v)
buf[i*4+1] = byte(v >> 8)
buf[i*4+2] = byte(v >> 16)
buf[i*4+3] = byte(v >> 24)
}
w.Write(buf)
}
func GetUint64(n interface{}) uint64 {
switch n.(type) {
case int, int8, int16, int32, int64:
return uint64(n.(int64))
default:
return n.(uint64)
}
}
func BinaryWriteINT64(w io.Writer, nums []interface{}) {
buf := make([]byte, len(nums)*8)
for i, n := range nums {
v := GetUint64(n)
buf[i*8+0] = byte(v)
buf[i*8+1] = byte(v >> 8)
buf[i*8+2] = byte(v >> 16)
buf[i*8+3] = byte(v >> 24)
buf[i*8+4] = byte(v >> 32)
buf[i*8+5] = byte(v >> 40)
buf[i*8+6] = byte(v >> 48)
buf[i*8+7] = byte(v >> 56)
}
w.Write(buf)
}
func BinaryWriteFLOAT32(w io.Writer, nums []interface{}) {
buf := make([]byte, len(nums)*4)
for i, n := range nums {
v := math.Float32bits(n.(float32))
buf[i*4+0] = byte(v)
buf[i*4+1] = byte(v >> 8)
buf[i*4+2] = byte(v >> 16)
buf[i*4+3] = byte(v >> 24)
}
w.Write(buf)
}
func BinaryWriteFLOAT64(w io.Writer, nums []interface{}) {
buf := make([]byte, len(nums)*8)
for i, n := range nums {
v := math.Float64bits(n.(float64))
buf[i*8+0] = byte(v)
buf[i*8+1] = byte(v >> 8)
buf[i*8+2] = byte(v >> 16)
buf[i*8+3] = byte(v >> 24)
buf[i*8+4] = byte(v >> 32)
buf[i*8+5] = byte(v >> 40)
buf[i*8+6] = byte(v >> 48)
buf[i*8+7] = byte(v >> 56)
}
w.Write(buf)
}