Lab_NHMFL
/
upsilon
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix generate_csr_locations.py

This commit is contained in:
Peter McGoron 2023-03-20 15:07:52 -04:00
parent 908be977f5
commit a7da03f4b9
1 changed files with 12 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
firmware/generate_csr_locations.py
View File

@ -3,9 +3,9 @@ import json
import sys import sys
class CSRGenerator: class CSRGenerator:
def __init__(self, json, registers, file): def __init__(self, json_file, registers, f):
self.registers = registers self.registers = registers
self.j = json.load(open("csr.json")) self.j = json.load(open(json_file))
self.file = f self.file = f
def get_reg(self, name, num): def get_reg(self, name, num):
@ -15,23 +15,22 @@ class CSRGenerator:
regname = f"base_{name}_{num}" regname = f"base_{name}_{num}"
return self.j["csr_registers"][regname]["addr"] return self.j["csr_registers"][regname]["addr"]
def print(self, *args): def print(self, *args):
print(*args, end='', file=f) print(*args, end='', file=self.file)
def print_array(self, name, num): def print_array(self, name, num):
if num == 1: if num == 1:
self.print(f'csr_t {name} = {self.get_reg(name, None)};\n') self.print(f'csr_t {name} = {self.get_reg(name, None)};\n')
else: else:
self.print(f'csr_t {name} = {{', self.get_reg(name, 0)) self.print(f'csr_t {name}[{num}] = {{', self.get_reg(name, 0))
for i in range(i,num): for i in range(0,num):
self.print(',', self.get_reg(name, i)) self.print(',', self.get_reg(name, i))
self.print('}\n\n') self.print('};\n\n')
def print_registers(self): def print_registers(self):
for name,num in self.registers: for name,num in self.registers:
self.print_array(name, num) self.print_array(name, num)
def print_file(self): def print_file(self):
self.print(f''' self.print(f'''#pragma once
#pragma once
typedef volatile uint32_t *csr_t; typedef volatile uint32_t *csr_t;
#define ADC_MAX {adc_num} #define ADC_MAX {adc_num}
#define DAC_MAX {dac_num} #define DAC_MAX {dac_num}
@ -69,5 +68,5 @@ if __name__ == "__main__":
("cl_finish_cmd", 1), ("cl_finish_cmd", 1),
] ]
CSRGenerator(self, "csr.json", registers, sys.stdout) CSRGenerator("csr.json", registers, sys.stdout).print_file()