34 lines
1008 B
Python
34 lines
1008 B
Python
|
"""
|
|||
|
|
2026/4/18
|
|||
|
|
Tan Mingyan
|
|||
|
|
新建 比特 / 字节 操作工具（通用、精准）
|
|||
|
|
"""
|
|||
|
|
def set_bits(data: bytearray, byte_offset: int, bit_offset: int, bit_length: int, value: int):
|
|||
|
|
"""
|
|||
|
|
对字节数组指定位置写入指定位数的值
|
|||
|
|
:param data: 原始字节数组（会被修改）
|
|||
|
|
:param byte_offset: 起始字节（从0开始，用户数据域起始为0）
|
|||
|
|
:param bit_offset: 起始比特（0~7）
|
|||
|
|
:param bit_length: 占多少比特
|
|||
|
|
:param value: 要写入的值
|
|||
|
|
"""
|
|||
|
|
if byte_offset >= len(data):
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 生成掩码
|
|||
|
|
max_val = (1 << bit_length) - 1
|
|||
|
|
value = value & max_val
|
|||
|
|
|
|||
|
|
pos = byte_offset * 8 + bit_offset
|
|||
|
|
for i in range(bit_length):
|
|||
|
|
bit = (value >> (bit_length - 1 - i)) & 1
|
|||
|
|
byte_idx = pos // 8
|
|||
|
|
bit_idx = pos % 8
|
|||
|
|
|
|||
|
|
if byte_idx >= len(data):
|
|||
|
|
break
|
|||
|
|
|
|||
|
|
# 清位 + 置位
|
|||
|
|
data[byte_idx] &= ~(1 << (7 - bit_idx))
|
|||
|
|
data[byte_idx] |= (bit << (7 - bit_idx))
|
|||
|
|
pos += 1
|