r500/regs/3d_registers.py

92 lines
2.7 KiB
Python

with open('3d_registers.txt') as f:
lines = f.read().split('\n')
def parse_reg_value(value):
if '-' in value:
start, end = value.split('-')
return int(start, 16), int(end, 16)
else:
return int(value, 16)
def reg_name_value(lines):
for line in lines:
if not line.strip():
continue
reg_name = line.split(':')[1].split(' ')[0]
reg_value = line.split(' MMReg:')[1:]
for value in reg_value:
yield reg_name, parse_reg_value(value.removesuffix(','))
#for reg_name, reg_value in reg_name_value(lines):
# print("#define", reg_name, reg_value)
aos_order_table = [
"VTX_AOS_ATTR01",
"VTX_AOS_ADDR0",
"VTX_AOS_ADDR1",
"VTX_AOS_ATTR23",
"VTX_AOS_ADDR2",
"VTX_AOS_ADDR3",
"VTX_AOS_ATTR45",
"VTX_AOS_ADDR4",
"VTX_AOS_ADDR5",
"VTX_AOS_ATTR67",
"VTX_AOS_ADDR6",
"VTX_AOS_ADDR7",
"VTX_AOS_ATTR89",
"VTX_AOS_ADDR8",
"VTX_AOS_ADDR8",
"VTX_AOS_ATTR1011",
"VTX_AOS_ADDR10",
"VTX_AOS_ADDR11",
"VTX_AOS_ATTR1213",
"VTX_AOS_ADDR12",
"VTX_AOS_ADDR13",
"VTX_AOS_ATTR1415",
"VTX_AOS_ADDR14",
"VTX_AOS_ADDR15",
]
print("registers_lookup = {")
for reg_name, reg_value in reg_name_value(lines):
if type(reg_value) is int:
print(f' 0x{reg_value:04x}: "{reg_name}",')
else:
start, end = reg_value
if '[' not in reg_name:
offset = start
while offset <= end:
ix = (offset - start) // 4
print(f' 0x{offset:04x}: "{reg_name}[{ix}]",')
offset += 4
else:
reg_basename = reg_name.split('[')[0]
index_range = reg_name.split('[')[1].split(']')[0]
start_ix, end_ix = map(int, index_range.split('-'))
offset_increment = 4
if (end - start) // 4 != end_ix - start_ix:
# guess the offset increment
if start + end_ix * 8 == end:
offset_increment = 8
elif start + end_ix * 16 == end:
offset_increment = 16
elif reg_basename == 'VAP_VTX_AOS_ADDR':
pass
elif reg_basename == 'VAP_VTX_AOS_ATTR':
for i, name in enumerate(aos_order_table):
print(f' 0x{i * 4 + start}: "{name}",')
continue
else:
print(reg_name, reg_value)
assert False
offset = start
while offset <= end:
ix = (offset - start) // offset_increment
print(f' 0x{offset:04x}: "{reg_basename}{ix}",')
offset += offset_increment
print('}')