1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
#
# Copyright 2018 Free Software Foundation, Inc.
#
# This file is part of GNU Radio
#
# SPDX-License-Identifier: GPL-3.0-or-later
#
#
""" A tool for generating YAML bindings """
from __future__ import print_function
from __future__ import absolute_import
from __future__ import unicode_literals
from collections import OrderedDict
import yaml
try:
from yaml import CLoader as Loader, CDumper as Dumper
except:
from yaml import Loader, Dumper
from .util_functions import is_number
## setup dumper for dumping OrderedDict ##
_mapping_tag = yaml.resolver.BaseResolver.DEFAULT_MAPPING_TAG
def dict_representer(dumper, data):
""" Representer to represent special OrderedDict """
return dumper.represent_dict(data.items())
def dict_constructor(loader, node):
""" Construct an OrderedDict for dumping """
return OrderedDict(loader.construct_pairs(node))
Dumper.add_representer(OrderedDict, dict_representer)
Loader.add_constructor(_mapping_tag, dict_constructor)
class GRCYAMLGenerator(object):
""" Create and write the YAML bindings for a GRC block. """
def __init__(self, modname=None, blockname=None, doc=None, params=None, iosig=None):
"""docstring for __init__"""
params_list = ['${'+s['key']+'}' for s in params if s['in_constructor']]
# Can't make a dict 'cause order matters
str_ = ', '.join(params_list)
self._header = (('id', f'{modname}_{blockname}'),
('label', blockname.replace('_', ' ')),
(f'category', '[{modname.capitalize()}]')
)
self._templates = (('imports', f'import {modname}'),
('make', f'{modname}.{blockname}({str_})')
)
self.params = params
self.iosig = iosig
self.doc = doc
self.data = None
def make_yaml(self):
""" Create the actual tag tree """
data = OrderedDict()
for tag, value in self._header:
data[tag] = value
templates = OrderedDict()
for tag, value in self._templates:
templates[tag] = value
data['templates'] = templates
parameters = []
for param in self.params:
parameter = OrderedDict()
parameter['id'] = param['key']
parameter['label'] = param['key'].capitalize()
if param['default']:
parameter['default'] = param['default']
parameter['dtype'] = param['type']
parameters.append(parameter)
if parameters:
data['parameters'] = parameters
inputs = []
outputs = []
iosig = self.iosig
for inout in sorted(iosig.keys()):
if iosig[inout]['max_ports'] == '0':
continue
for i in range(len(iosig[inout]['type'])):
s_type = {'in': 'input', 'out': 'output'}[inout]
s_obj = OrderedDict()
s_obj['label'] = inout
s_obj['domain'] = 'stream'
s_obj['dtype'] = iosig[inout]['type'][i]
if iosig[inout]['vlen'][i] != '1':
vlen = iosig[inout]['vlen'][i]
if is_number(vlen):
s_obj['vlen'] = vlen
else:
s_obj['vlen'] = '${ '+vlen+' }'
if i == len(iosig[inout]['type'])-1:
if not is_number(iosig[inout]['max_ports']):
s_obj['multiplicity'] = iosig[inout]['max_ports']
elif len(iosig[inout]['type']) < int(iosig[inout]['max_ports']):
s_obj['multiplicity'] = str(int(iosig[inout]['max_ports']) -
len(iosig[inout]['type'])+1)
if s_type == 'input':
inputs.append(s_obj)
elif s_type == 'output':
outputs.append(s_obj)
if inputs:
data['inputs'] = inputs
if outputs:
data['outputs'] = outputs
if self.doc is not None:
data['documentation'] = self.doc
self.data = data
data['file_format'] = 1
def save(self, filename):
""" Write the YAML file """
self.make_yaml()
with open(filename, 'w') as f:
yaml.dump(self.data, f, Dumper=Dumper, default_flow_style=False)
|
