-
Notifications
You must be signed in to change notification settings - Fork 2
/
utils.go
86 lines (76 loc) · 1.92 KB
/
utils.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
84
85
86
package main
import (
"encoding/binary"
"os"
"fmt"
ol "github.com/ossrs/go-oryx-lib/logger"
)
// intDataSize returns the size of the data required to represent the data when encoded.
// It returns zero if the type cannot be implemented by the fast path in Read or Write.
func uint64DataSize(data interface{}) uint64 {
switch data.(type) {
case int8, uint8, *int8, *uint8:
return uint64(1)
case int16, uint16, *int16, *uint16:
return uint64(2)
case int32, uint32, *int32, *uint32:
return uint64(4)
case int64, uint64, *int64, *uint64:
return uint64(8)
case []uint8:
arru8 := data.([]uint8)
return uint64(len(arru8))
}
return 0
}
func Bytes3ToUint32(b []byte) uint32 {
nb := []byte{}
nb = append(nb, 0)
nb = append(nb, b...)
return binary.BigEndian.Uint32(nb)
}
func Uint32To3Bytes(data uint32) (res []byte) {
tmp0 := byte((data & 0x00FF0000) >> 16)
res = append(res, tmp0)
tmp1 := byte((data & 0x0000FF00) >> 8)
res = append(res, tmp1)
tmp2 := byte(data & 0x000000FF)
res = append(res, tmp2)
return res
}
func max(x, y int32) int32 {
if x > y {
return x
}
return y
}
func min(x, y int32) int32 {
if x > y {
return y
}
return x
}
func readAt(mp4 string, offset int64, length int) (data []byte, err error) {
var f *os.File
if f, err = os.Open(mp4); err != nil {
ol.E(nil, fmt.Sprintf("open mp4 file failed, err is %v", err))
return
}
defer f.Close()
data = make([]byte, length)
var n int
if n, err = f.ReadAt(data, offset); err != nil {
return
}
if n != length {
err = fmt.Errorf("read at offset:%x, exp len=%v, actual len=%v", offset, length, n)
ol.E(nil, err.Error())
return
}
return
}
//TODO: need complete
func to3Bytes(from uint32) (to []byte) {
to = make([]byte, 3)
return
}