mirror of
https://gitee.com/johng/gf.git
synced 2024-11-30 03:07:45 +08:00
84 lines
2.0 KiB
Go
84 lines
2.0 KiB
Go
// Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
|
||
//
|
||
// This Source Code Form is subject to the terms of the MIT License.
|
||
// If a copy of the MIT was not distributed with this file,
|
||
// You can obtain one at https://github.com/gogf/gf.
|
||
|
||
package gbinary
|
||
|
||
// NOTE: THIS IS AN EXPERIMENTAL FEATURE!
|
||
|
||
// Bit Binary bit (0 | 1) 二进制位(0|1)
|
||
type Bit int8
|
||
|
||
// EncodeBits .
|
||
// Default coding
|
||
// 默认编码
|
||
func EncodeBits(bits []Bit, i int, l int) []Bit {
|
||
return EncodeBitsWithUint(bits, uint(i), l)
|
||
}
|
||
|
||
// EncodeBitsWithUint .
|
||
// Merge ui bitwise into the bits array and occupy the length length bits (Note: binary 0 | 1 digits are stored in the uis array)
|
||
// 将ui按位合并到bits数组中，并占length长度位(注意：uis数组中存放的是二进制的0|1数字)
|
||
func EncodeBitsWithUint(bits []Bit, ui uint, l int) []Bit {
|
||
a := make([]Bit, l)
|
||
for i := l - 1; i >= 0; i-- {
|
||
a[i] = Bit(ui & 1)
|
||
ui >>= 1
|
||
}
|
||
if bits != nil {
|
||
return append(bits, a...)
|
||
}
|
||
return a
|
||
}
|
||
|
||
// EncodeBitsToBytes .
|
||
// Convert bits to [] byte, encode from left to right, and add less than 1 byte from 0 to the end.
|
||
// 将bits转换为[]byte，从左至右进行编码，不足1 byte按0往末尾补充
|
||
func EncodeBitsToBytes(bits []Bit) []byte {
|
||
if len(bits)%8 != 0 {
|
||
for i := 0; i < len(bits)%8; i++ {
|
||
bits = append(bits, 0)
|
||
}
|
||
}
|
||
b := make([]byte, 0)
|
||
for i := 0; i < len(bits); i += 8 {
|
||
b = append(b, byte(DecodeBitsToUint(bits[i:i+8])))
|
||
}
|
||
return b
|
||
}
|
||
|
||
// DecodeBits .
|
||
// Resolve to int
|
||
// 解析为int
|
||
func DecodeBits(bits []Bit) int {
|
||
v := 0
|
||
for _, i := range bits {
|
||
v = v<<1 | int(i)
|
||
}
|
||
return v
|
||
}
|
||
|
||
// DecodeBitsToUint .
|
||
// Resolve to uint
|
||
// 解析为uint
|
||
func DecodeBitsToUint(bits []Bit) uint {
|
||
v := uint(0)
|
||
for _, i := range bits {
|
||
v = v<<1 | uint(i)
|
||
}
|
||
return v
|
||
}
|
||
|
||
// DecodeBytesToBits .
|
||
// Parsing [] byte into character array [] uint8
|
||
// 解析[]byte为字位数组[]uint8
|
||
func DecodeBytesToBits(bs []byte) []Bit {
|
||
bits := make([]Bit, 0)
|
||
for _, b := range bs {
|
||
bits = EncodeBitsWithUint(bits, uint(b), 8)
|
||
}
|
||
return bits
|
||
}
|