71 lines
2.3 KiB
Python
Executable File
71 lines
2.3 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import sys
|
|
import yaml
|
|
import csv
|
|
from collections import OrderedDict
|
|
|
|
def repr_odict(dumper, data):
|
|
"""
|
|
>>> data = OrderedDict([('foo', 'bar'), ('mumble', 'quux'), ('baz', 'gorp')])
|
|
>>> yaml.dump(data, default_flow_style=False)
|
|
'!!omap\\n- foo: bar\\n- mumble: quux\\n- baz: gorp\\n'
|
|
>>> yaml.dump(data, default_flow_style=True)
|
|
'!!omap [foo: bar, mumble: quux, baz: gorp]\\n'
|
|
"""
|
|
return repr_pairs(dumper, u'tag:yaml.org,2002:omap', data.iteritems())
|
|
|
|
yaml.add_representer(OrderedDict, repr_odict)
|
|
|
|
def repr_pairs(dump, tag, sequence, flow_style=None):
|
|
"""This is the same code as BaseRepresenter.represent_sequence(),
|
|
but the value passed to dump.represent_data() in the loop is a
|
|
dictionary instead of a tuple."""
|
|
|
|
value = []
|
|
node = yaml.SequenceNode(tag, value, flow_style=flow_style)
|
|
if dump.alias_key is not None:
|
|
dump.represented_objects[dump.alias_key] = node
|
|
best_style = True
|
|
for (key, val) in sequence:
|
|
item = dump.represent_data({key: val})
|
|
if not (isinstance(item, yaml.ScalarNode) and not item.style):
|
|
best_style = False
|
|
value.append(item)
|
|
if flow_style is None:
|
|
if dump.default_flow_style is not None:
|
|
node.flow_style = dump.default_flow_style
|
|
else:
|
|
node.flow_style = best_style
|
|
return node
|
|
|
|
def convert_file(fname):
|
|
reader = csv.reader(open(fname, 'r'))
|
|
|
|
registers = OrderedDict()
|
|
for register_name, lsb, width, field_name, description, reset_value, access in reader:
|
|
if register_name not in registers:
|
|
registers[register_name] = {
|
|
'fields': OrderedDict(),
|
|
}
|
|
|
|
register = registers[register_name]
|
|
fields = register['fields']
|
|
if field_name in fields:
|
|
raise RuntimeError('Duplicate field name "%s" in register "%s"' %
|
|
field_name, register_name)
|
|
else:
|
|
fields[field_name] = {
|
|
'lsb': int(lsb),
|
|
'width': int(width),
|
|
'description': description,
|
|
'reset_value': reset_value,
|
|
'access': access,
|
|
}
|
|
|
|
with open(fname.replace('.csv', '.yaml'), 'w') as out_file:
|
|
yaml.dump(registers, out_file, default_flow_style=False)
|
|
|
|
for fname in sys.argv[1:]:
|
|
convert_file(fname)
|