mirror of
https://github.com/AlexxIT/go2rtc.git
synced 2026-04-22 23:57:20 +08:00
Sergey Krashevich
2b7682cdb3
- replace traditional for loop with range-based for loop for clarity
refactor(ffmpeg): simplify cut function loop
- utilize range-based for loop instead of traditional for loop
refactor(ring): update API response mapping type
- change map type from `interface{}` to `any` for better type safety
refactor(stream): handle nil source in NewStream
- add nil check for source elements before processing
refactor(webrtc): unify payload handling in GetToken
- change map type from `interface{}` to `any` for consistency
refactor(ascii): optimize nested loops in Write function
- replace traditional for loops with range-based for loops for readability
refactor(bits): enhance readability in Reader methods
- replace traditional for loops with range-based for loops in Read functions
refactor(h264): modernize loop structures in DecodeConfig
- switch to range-based for loops for cleaner code
refactor(h265): streamline profile_tier_level loops
- utilize range-based for loops instead of traditional for loops
chore(core): remove commented-out test function for clarity
refactor(core): simplify RandString function loop
- change traditional for loop to range-based for loop
refactor(flvt): optimize timestamp handling in TestTimeToRTP
- switch to range-based for loop for iterating frames
refactor(nest): improve error handling in ExchangeSDP
- format error message with printf-style formatting for clarity
refactor(tapo): enhance securityEncode function
- change traditional for loop to range-based for loop for readability
fix(tcp): correct masking in websocket Write method
- replace traditional for loop with range-based for loop
refactor(tutk): modernize encoding loops in crypto functions
- utilize range-based for loops for better readability
refactor(tuya): unify data types in MQTT message struct
- change map type from `interface{}` to `any` for consistency
refactor(webrtc): standardize codec registration
- change map type from `interface{}` to `any` for type safety
refactor(yaml): simplify Unmarshal function signature
- update parameter type from `interface{}` to `any` for better clarity
96 lines
1.5 KiB
Go
96 lines
1.5 KiB
Go
package bits
|
|
|
|
type Writer struct {
|
|
buf []byte // total buf
|
|
byte *byte // pointer to current byte
|
|
bits byte // bits left in byte
|
|
}
|
|
|
|
func NewWriter(buf []byte) *Writer {
|
|
return &Writer{buf: buf}
|
|
}
|
|
|
|
//goland:noinspection GoStandardMethods
|
|
func (w *Writer) WriteUint8(b byte) {
|
|
if w.bits != 0 {
|
|
w.WriteBits8(b, 8)
|
|
}
|
|
|
|
w.buf = append(w.buf, b)
|
|
}
|
|
|
|
func (w *Writer) WriteBit(b byte) {
|
|
if w.bits == 0 {
|
|
w.buf = append(w.buf, 0)
|
|
w.byte = &w.buf[len(w.buf)-1]
|
|
w.bits = 7
|
|
} else {
|
|
w.bits--
|
|
}
|
|
|
|
*w.byte |= (b & 1) << w.bits
|
|
}
|
|
|
|
func (w *Writer) WriteBits(v uint32, n byte) {
|
|
for i := n - 1; i != 255; i-- {
|
|
w.WriteBit(byte(v>>i) & 0b1)
|
|
}
|
|
}
|
|
|
|
func (w *Writer) WriteBits16(v uint16, n byte) {
|
|
for i := n - 1; i != 255; i-- {
|
|
w.WriteBit(byte(v>>i) & 0b1)
|
|
}
|
|
}
|
|
|
|
func (w *Writer) WriteBits8(v, n byte) {
|
|
for i := n - 1; i != 255; i-- {
|
|
w.WriteBit((v >> i) & 0b1)
|
|
}
|
|
}
|
|
|
|
func (w *Writer) WriteAllBits(bit, n byte) {
|
|
for range n {
|
|
w.WriteBit(bit)
|
|
}
|
|
}
|
|
|
|
func (w *Writer) WriteBool(b bool) {
|
|
if b {
|
|
w.WriteBit(1)
|
|
} else {
|
|
w.WriteBit(0)
|
|
}
|
|
}
|
|
|
|
func (w *Writer) WriteUint16(v uint16) {
|
|
if w.bits != 0 {
|
|
w.WriteBits16(v, 16)
|
|
}
|
|
|
|
w.buf = append(w.buf, byte(v>>8), byte(v))
|
|
}
|
|
|
|
func (w *Writer) WriteBytes(bytes ...byte) {
|
|
if w.bits != 0 {
|
|
for _, b := range bytes {
|
|
w.WriteUint8(b)
|
|
}
|
|
}
|
|
|
|
w.buf = append(w.buf, bytes...)
|
|
}
|
|
|
|
func (w *Writer) Bytes() []byte {
|
|
return w.buf
|
|
}
|
|
|
|
func (w *Writer) Len() int {
|
|
return len(w.buf)
|
|
}
|
|
|
|
func (w *Writer) Reset() {
|
|
w.buf = w.buf[:0]
|
|
w.bits = 0
|
|
}
|