diff options
Diffstat (limited to 'utils')
| -rw-r--r-- | utils/bitreader.go | 41 | ||||
| -rw-r--r-- | utils/utils.go | 31 |
2 files changed, 41 insertions, 31 deletions
diff --git a/utils/bitreader.go b/utils/bitreader.go new file mode 100644 index 0000000..c4936fd --- /dev/null +++ b/utils/bitreader.go | |||
| @@ -0,0 +1,41 @@ | |||
| 1 | package utils | ||
| 2 | |||
| 3 | import ( | ||
| 4 | "fmt" | ||
| 5 | "math" | ||
| 6 | "strconv" | ||
| 7 | ) | ||
| 8 | |||
| 9 | func ReadBitStateLSB(input byte, index int) (bool, error) { | ||
| 10 | if index < 0 && index > 7 { | ||
| 11 | return false, fmt.Errorf("IndexOutOfBounds for type byte") | ||
| 12 | } | ||
| 13 | value := input & (1 << index) | ||
| 14 | return (value > 0), nil | ||
| 15 | } | ||
| 16 | |||
| 17 | func Read32BitsAfterFirstBitInt32(input []byte, index int, step int) int32 { | ||
| 18 | binary := "" | ||
| 19 | binary += fmt.Sprintf("%08b", input[step])[8-index : 8] | ||
| 20 | binary += fmt.Sprintf("%08b", input[step-1]) | ||
| 21 | binary += fmt.Sprintf("%08b", input[step-2]) | ||
| 22 | binary += fmt.Sprintf("%08b", input[step-3]) | ||
| 23 | binary += fmt.Sprintf("%08b", input[step-4])[:8-index] | ||
| 24 | output, err := strconv.ParseInt(binary, 2, 32) | ||
| 25 | CheckError(err) | ||
| 26 | return int32(output) | ||
| 27 | |||
| 28 | } | ||
| 29 | |||
| 30 | func Read32BitsAfterFirstBitFloat32(input []byte, index int, step int) float32 { | ||
| 31 | binary := "" | ||
| 32 | binary += fmt.Sprintf("%08b", input[step])[8-index : 8] | ||
| 33 | binary += fmt.Sprintf("%08b", input[step-1]) | ||
| 34 | binary += fmt.Sprintf("%08b", input[step-2]) | ||
| 35 | binary += fmt.Sprintf("%08b", input[step-3]) | ||
| 36 | binary += fmt.Sprintf("%08b", input[step-4])[:8-index] | ||
| 37 | output, err := strconv.ParseUint(binary, 2, 32) | ||
| 38 | CheckError(err) | ||
| 39 | return math.Float32frombits(uint32(output)) | ||
| 40 | |||
| 41 | } | ||
diff --git a/utils/utils.go b/utils/utils.go index 6874e6b..5226e80 100644 --- a/utils/utils.go +++ b/utils/utils.go | |||
| @@ -1,15 +1,12 @@ | |||
| 1 | package utils | 1 | package utils |
| 2 | 2 | ||
| 3 | import ( | 3 | import ( |
| 4 | "bytes" | ||
| 5 | "encoding/binary" | 4 | "encoding/binary" |
| 6 | "log" | 5 | "log" |
| 7 | "math" | 6 | "math" |
| 8 | "math/bits" | 7 | "math/bits" |
| 9 | "os" | 8 | "os" |
| 10 | "unsafe" | 9 | "unsafe" |
| 11 | |||
| 12 | "github.com/32bitkid/bitreader" | ||
| 13 | ) | 10 | ) |
| 14 | 11 | ||
| 15 | func CheckError(e error) { | 12 | func CheckError(e error) { |
| @@ -18,34 +15,6 @@ func CheckError(e error) { | |||
| 18 | } | 15 | } |
| 19 | } | 16 | } |
| 20 | 17 | ||
| 21 | func ReadBitsFromReversedByteArray1(byteArr []byte) bool { | ||
| 22 | r := bitreader.NewReader(bytes.NewBuffer(ReverseByteArrayValues(byteArr, len(byteArr)))) | ||
| 23 | value, err := r.Read1() | ||
| 24 | CheckError(err) | ||
| 25 | return value | ||
| 26 | } | ||
| 27 | |||
| 28 | func ReadBitsFromReversedByteArray8(byteArr []byte, bitLength uint) uint8 { | ||
| 29 | r := bitreader.NewReader(bytes.NewBuffer(ReverseByteArrayValues(byteArr, len(byteArr)))) | ||
| 30 | value, err := r.Read8(bitLength) | ||
| 31 | CheckError(err) | ||
| 32 | return value | ||
| 33 | } | ||
| 34 | |||
| 35 | func ReadBitsFromReversedByteArray16(byteArr []byte, bitLength uint) uint16 { | ||
| 36 | r := bitreader.NewReader(bytes.NewBuffer(ReverseByteArrayValues(byteArr, len(byteArr)))) | ||
| 37 | value, err := r.Read16(bitLength) | ||
| 38 | CheckError(err) | ||
| 39 | return value | ||
| 40 | } | ||
| 41 | |||
| 42 | func ReadBitsFromReversedByteArray32(byteArr []byte, bitLength uint) uint32 { | ||
| 43 | r := bitreader.NewReader(bytes.NewBuffer(ReverseByteArrayValues(byteArr, len(byteArr)))) | ||
| 44 | value, err := r.Read32(bitLength) | ||
| 45 | CheckError(err) | ||
| 46 | return value | ||
| 47 | } | ||
| 48 | |||
| 49 | func ReverseByteArrayValues(byteArr []byte, size int) []byte { | 18 | func ReverseByteArrayValues(byteArr []byte, size int) []byte { |
| 50 | arr := make([]byte, size) | 19 | arr := make([]byte, size) |
| 51 | for index, byteValue := range byteArr { | 20 | for index, byteValue := range byteArr { |