2 ########################################################
3 ##Cheetah template - gnuradio_python
5 ##@param imports the import statements
6 ##@param flow_graph the flow_graph
7 ##@param variables the variable blocks
8 ##@param notebooks a list of notebook blocks
9 ##@param controls the variables with gui controls
10 ##@param parameters the paramater blocks
11 ##@param blocks the signal blocks
12 ##@param connections the connections
13 ##@param generate_options the type of flow graph
14 ##@param var_id2cbs variable id map to callback strings
15 ########################################################
17 #set $code = '\n\t\t'.join(str($code).splitlines())
21 #set $DIVIDER = '#'*50
23 # Gnuradio Python Flow Graph
25 #if $flow_graph.get_option('author')
26 # Author: $flow_graph.get_option('author')
28 #if $flow_graph.get_option('description')
29 # Description: $flow_graph.get_option('description')
31 # Generated: $time.ctime()
34 ########################################################
36 ########################################################
41 ########################################################
43 ## Write the class declaration for a top or hier block.
44 ## The parameter names are the arguments to __init__.
45 ## Determine the absolute icon path (wx gui only).
46 ## Setup the IO signature (hier block only).
47 ########################################################
48 #set $class_name = $flow_graph.get_option('id')
49 #set $param_str = ', '.join(['self'] + ['%s=%s'%(param.get_id(), param.get_make()) for param in $parameters])
50 #if $generate_options == 'wx_gui'
52 #set $icon = gtk.IconTheme().lookup_icon('gnuradio-grc', 32, 0)
53 class $(class_name)(grc_wxgui.top_block_gui):
55 def __init__($param_str):
56 grc_wxgui.top_block_gui.__init__(self, title="$title")
58 _icon_path = "$icon.get_filename()"
59 self.SetIcon(wx.Icon(_icon_path, wx.BITMAP_TYPE_ANY))
61 #elif $generate_options == 'no_gui'
62 class $(class_name)(gr.top_block):
64 def __init__($param_str):
65 gr.top_block.__init__(self, "$title")
66 #elif $generate_options == 'hb'
67 #set $in_sig = $flow_graph.get_input_signature()
68 #set $out_sig = $flow_graph.get_output_signature()
69 class $(class_name)(gr.hier_block2):
71 def __init__($param_str):
72 gr.hier_block2.__init__(
74 gr.io_signature($in_sig.nports, $in_sig.nports, $in_sig.size*$in_sig.vlen),
75 gr.io_signature($out_sig.nports, $out_sig.nports, $out_sig.size*$out_sig.vlen),
78 ########################################################
80 ## Set the parameter to a property of self.
81 ########################################################
88 #for $param in $parameters
89 $indent($param.get_var_make())
91 ########################################################
93 ########################################################
100 #for $var in $variables
101 $indent($var.get_var_make())
103 ########################################################
105 ########################################################
112 #for $notebook in $notebooks
113 $indent($notebook.get_make())
115 ########################################################
117 ########################################################
124 #for $ctrl in $controls
125 $indent($ctrl.get_make())
127 ########################################################
129 ########################################################
136 #for $blk in filter(lambda b: b.get_make(), $blocks)
137 self.$blk.get_id() = $indent($blk.get_make())
139 ########################################################
141 ## The port name should be the id of the parent block.
142 ## However, port names for IO pads should be self.
143 ########################################################
150 #for $con in $connections
151 #set $source = $con.get_source()
152 #set $sink = $con.get_sink()
153 #if $source.get_parent().get_key() == 'pad_source'
154 #set $source_name = 'self'
156 #set $source_name = 'self.' + $source.get_parent().get_id()
158 #if $sink.get_parent().get_key() == 'pad_sink'
159 #set $sink_name = 'self'
161 #set $sink_name = 'self.' + $sink.get_parent().get_id()
163 self.connect(($source_name, $source.get_key()), ($sink_name, $sink.get_key()))
166 ########################################################
168 ## Write a set method for this variable that calls the callbacks
169 ########################################################
170 #for $var in $parameters + $variables
171 #set $id = $var.get_id()
172 def set_$(id)(self, $id):
174 #for $callback in $var_id2cbs[$id]
179 ########################################################
181 ## For top block code, generate a main routine.
182 ## Instantiate the top block and run as gui or cli.
183 ########################################################
184 #if $generate_options != 'hb'
185 if __name__ == '__main__':
186 parser = OptionParser(option_class=eng_option, usage="%prog: [options]")
187 #set $params_eq_list = list()
188 #for $param in $parameters
189 #set $type = $param.get_param('type').get_value()
191 #silent $params_eq_list.append('%s=options.%s'%($param.get_id(), $param.get_id()))
192 #set $short_id = $param.get_param('short_id').get_evaluated()
194 #set $short_id = '-' + $short_id
196 parser.add_option("$short_id", "--$param.get_id()", dest="$param.get_id()", type="$type", default=$param.get_make(),
197 help="Set $($param.get_param('label').evaluate() or $param.get_id()) [default=%default]")
200 (options, args) = parser.parse_args()
201 #if $flow_graph.get_option('realtime_scheduling')
202 if gr.enable_realtime_scheduling() != gr.RT_OK:
203 print "Error: failed to enable realtime scheduling."
205 tb = $(class_name)($(', '.join($params_eq_list)))
206 #if $generate_options == 'wx_gui'
207 tb.Run($flow_graph.get_option('run'))
208 #elif $generate_options == 'no_gui'
210 raw_input('Press Enter to quit: ')