79 lines
1.8 KiB
Python
79 lines
1.8 KiB
Python
#!/usr/bin/python3
|
|
import json
|
|
import sys
|
|
|
|
"""
|
|
This file takes the csr.json file output by LiteX and extracts all
|
|
CSRs that are handled by Upsilon directly. See the output file csr.json
|
|
for layout.
|
|
"""
|
|
|
|
class CSRGenerator:
|
|
def __init__(self, json_file, registers, f):
|
|
self.registers = registers
|
|
self.j = json.load(open(json_file))
|
|
self.file = f
|
|
|
|
def get_reg(self, name, num):
|
|
if num is None:
|
|
regname = f"base_{name}"
|
|
else:
|
|
regname = f"base_{name}_{num}"
|
|
return f'{self.j["csr_registers"][regname]["addr"]}'
|
|
def print(self, *args):
|
|
print(*args, end='', file=self.file)
|
|
|
|
def print_array(self, name, num):
|
|
if num == 1:
|
|
self.print(f'static const uintptr_t {name} = {self.get_reg(name, None)};\n')
|
|
else:
|
|
self.print(f'static const uintptr_t {name}[{num}] = {{', self.get_reg(name, 0))
|
|
for i in range(1,num):
|
|
self.print(',', self.get_reg(name, i))
|
|
self.print('};\n\n')
|
|
|
|
def print_registers(self):
|
|
for name,num in self.registers:
|
|
self.print_array(name, num)
|
|
def print_file(self):
|
|
self.print(f'''#pragma once
|
|
#define ADC_MAX {adc_num}
|
|
#define DAC_MAX {dac_num}
|
|
''')
|
|
self.print_registers()
|
|
|
|
if __name__ == "__main__":
|
|
dac_num = 8
|
|
adc_num = 8
|
|
|
|
registers = [
|
|
("dac_sel", dac_num),
|
|
("dac_finished", dac_num),
|
|
("dac_arm", dac_num),
|
|
("from_dac", dac_num),
|
|
("to_dac", dac_num),
|
|
("wf_arm", dac_num),
|
|
("wf_halt_on_finish", dac_num),
|
|
("wf_finished", dac_num),
|
|
("wf_running", dac_num),
|
|
("wf_time_to_wait", dac_num),
|
|
("wf_refresh_start", dac_num),
|
|
("wf_refresh_finished", dac_num),
|
|
("wf_start_addr", dac_num),
|
|
|
|
("adc_finished", adc_num),
|
|
("adc_arm", adc_num),
|
|
("from_adc", adc_num),
|
|
|
|
("adc_sel_0", 1),
|
|
("cl_in_loop", 1),
|
|
("cl_cmd", 1),
|
|
("cl_word_in", 1),
|
|
("cl_word_out", 1),
|
|
("cl_start_cmd", 1),
|
|
("cl_finish_cmd", 1),
|
|
]
|
|
|
|
CSRGenerator("csr.json", registers, sys.stdout).print_file()
|
|
|