31 lines
856 B
Python
31 lines
856 B
Python
"""
|
||
2026/4/5
|
||
TMY
|
||
比特位打包器
|
||
"""
|
||
class BitPacker:
|
||
def __init__(self, total_bytes):
|
||
self.buf = bytearray(total_bytes)
|
||
|
||
def set_bits(self, start_byte, start_bit, bit_len, value):
|
||
"""
|
||
start_byte: 起始字节
|
||
start_bit: 起始比特位（0~7）
|
||
bit_len: 比特长度
|
||
value: 无符号整数值
|
||
"""
|
||
value = int(value) & ((1 << bit_len) - 1)
|
||
bit_pos = start_byte * 8 + start_bit
|
||
|
||
for i in range(bit_len):
|
||
bit = (value >> (bit_len - 1 - i)) & 1
|
||
byte_idx = bit_pos // 8
|
||
bit_idx = 7 - (bit_pos % 8)
|
||
if bit:
|
||
self.buf[byte_idx] |= (1 << bit_idx)
|
||
else:
|
||
self.buf[byte_idx] &= ~(1 << bit_idx)
|
||
bit_pos += 1
|
||
|
||
def get_bytes(self):
|
||
return bytes(self.buf) |