blackmagic/scripts/get_openocd_nrf51_ids.py

57 lines
1.8 KiB
Python
Executable File

#!/usr/bin/python
"""Pulls nRF51 IDs from openocd's nrf51.c in a form suitable for
pasting into blackmagic's nrf51.c
"""
import subprocess
import re
import io
cmd = 'git archive --remote=git://git.code.sf.net/p/openocd/code HEAD src/flash/nor/nrf5.c | tar -xO'
class Spec():
def __repr__(self):
return "0x%04X: /* %s %s %s */"%(self.hwid,self.comment, self.variant,self.build_code)
proc = subprocess.Popen(cmd,shell=True,stdout=subprocess.PIPE)
specdict = {}
specs = []
spec = Spec()
for line in io.TextIOWrapper(proc.stdout, encoding="utf-8"):
m = re.search('/\*(.*)\*/',line)
if m:
lastcomment=m.group(1)
m = re.search('NRF51_DEVICE_DEF\((0x[0-9A-F]*),\s*"(.*)",\s*"(.*)",\s*"(.*)",\s*([0-9]*)\s*\),', line)
if m:
spec.hwid = int(m.group(1), base=0)
spec.variant = m.group(3)
spec.build_code = m.group(4)
spec.flash_size_kb = int(m.group(5), base=0)
ram, flash = {'AA':(16,256),
'AB':(16,128),
'AC':(32,256)}[spec.variant[-2:]]
assert flash == spec.flash_size_kb
spec.ram_size_kb = ram
nicecomment = lastcomment.strip().replace('IC ','').replace('Devices ','').replace('.','')
spec.comment = nicecomment
specdict.setdefault((ram,flash),[]).append(spec)
specs.append(spec)
spec=Spec()
for (ram,flash),specs in specdict.items():
specs.sort(key=lambda x:x.hwid)
for spec in specs:
print("\tcase",spec)
print('\t\tt->driver = "Nordic nRF51";')
print('\t\ttarget_add_ram(t, 0x20000000, 0x%X);'%(1024*ram))
print('\t\tnrf51_add_flash(t, 0x00000000, 0x%X, NRF51_PAGE_SIZE);'%(1024*flash))
print('\t\tnrf51_add_flash(t, NRF51_UICR, 0x100, 0x100);')
print('\t\ttarget_add_commands(t, nrf51_cmd_list, "nRF51");')
print('\t\treturn true;')