#!/usr/bin/env python

########################################################

##Cheetah template - gnuradio_python

##

##@param imports the import statements

##@param flow_graph the flow_graph

##@param variables the variable blocks

##@param controls the variables with gui controls

##@param parameters the paramater blocks

##@param blocks the signal blocks

##@param connections the connections

##@param generate_options the type of flow graph

##@param var_id2expr variable id map to expression

##@param var_id2deps variable id map to direct dependencies

##@param var_id2cbs variable id map to callback strings

########################################################

#import time

#set $DIVIDER = '#'*50

$DIVIDER

# Gnuradio Python Flow Graph

$('# Title: %s'%$flow_graph.get_option('title'))

$('# Author: %s'%$flow_graph.get_option('author'))

$('# Description: %s'%$flow_graph.get_option('description'))

$('# Generated: %s'%time.ctime())

$DIVIDER

########################################################

##Create Imports

########################################################

#for $imp in $imports

$imp

#end for

########################################################

##Create Class

##	Write the class declaration for a top or hier block.

##	The parameter names are the arguments to __init__.

##	Determine the absolute icon path (wx gui only).

##	Setup the IO signature (hier block only).

########################################################

#set $class_name = $flow_graph.get_option('id')

#set $param_str = ', '.join(['self'] + ['%s=%s'%(param.get_id(), param.get_make()) for param in $parameters])

#if $generate_options == 'wx_gui'

	#import gtk

	#set $icon = gtk.IconTheme().lookup_icon('gnuradio-grc', 32, 0)

class $(class_name)(grc_wxgui.top_block_gui):

	def __init__($param_str):

		grc_wxgui.top_block_gui.__init__(

			self,

			title="$flow_graph.get_parent().get_name() - Executing: $flow_graph.get_option('title')",

	#if $icon

			icon="$icon.get_filename()",

	#end if

		)

#elif $generate_options == 'no_gui'

class $(class_name)(gr.top_block):

	def __init__($param_str):

		gr.top_block.__init__(self, "$flow_graph.get_option('title')")

#elif $generate_options == 'hb'

	#set $in_sig = $flow_graph.get_input_signature()

	#set $out_sig = $flow_graph.get_output_signature()

class $(class_name)(gr.hier_block2):

	def __init__($param_str):

		gr.hier_block2.__init__(

			self,

			"$flow_graph.get_option('title')",

			gr.io_signature($in_sig.nports, $in_sig.nports, $in_sig.size*$in_sig.vlen),

			gr.io_signature($out_sig.nports, $out_sig.nports, $out_sig.size*$out_sig.vlen),

		)

#end if

########################################################

##Create Parameters

##	Set the parameter to a property of self..

########################################################

#if $parameters

		$DIVIDER

		# Parameters

		$DIVIDER

#end if

#for $param in $parameters

		self.$param.get_id() = $param.get_id()

#end for

########################################################

##Create Variables

##	Set the variable to a property of self.

##	Write the first line of the variable make.

########################################################

#if $variables

		$DIVIDER

		# Variables

		$DIVIDER

#end if

#for $var in $variables

	#set $code = $var.get_make().splitlines()[0]

		self.$var.get_id() = $var.get_id() = $code

#end for

########################################################

##Create Controls

##	Write the variable make (excluding first line).

##	Indent each line with 2 tabs.

########################################################

#if $controls

		$DIVIDER

		# Controls

		$DIVIDER

#end if

#for $ctrl in $controls

	#set $code = '\n\t\t'.join($ctrl.get_make().splitlines()[1:])

		$code

#end for

########################################################

##Create Blocks

##	Write the block make, and indent with 2 tabs.

########################################################

#if $blocks

		$DIVIDER

		# Blocks

		$DIVIDER

#end if

#for $blk in filter(lambda b: b.get_make(), $blocks)

	#set $code = '\n\t\t'.join($blk.get_make().splitlines())

		$("self.%s = %s"%($blk.get_id(), $code))

#end for

########################################################

##Create Connections

##	The port name should be the id of the parent block.

##	However, port names for IO pads should be self.

########################################################

#if $connections

		$DIVIDER

		# Connections

		$DIVIDER

#end if

#for $con in $connections

	#set $source = $con.get_source()

	#set $sink = $con.get_sink()

	#if $source.get_parent().get_key() == 'pad_source'

		#set $source_name = 'self'

	#else

		#set $source_name = 'self.' + $source.get_parent().get_id()

	#end if

	#if $sink.get_parent().get_key() == 'pad_sink'

		#set $sink_name = 'self'

	#else

		#set $sink_name = 'self.' + $sink.get_parent().get_id()

	#end if

		$("self.connect((%s, %s), (%s, %s))"%(

				$source_name,

				$source.get_key(),

				$sink_name,

				$sink.get_key(),

			)

		)

#end for

########################################################

##Create Callbacks

##	Write a set method for this variable that calls the callbacks

##	and sets the direct variable dependencies.

########################################################

#for $var in $parameters + $variables

	#set $id = $var.get_id()

	def set_$(id)(self, $id):

		self.$id = $id

	#for $dep in $var_id2deps[$id]

		self.set_$(dep)($var_id2expr[$dep])

	#end for

	#for $callback in $var_id2cbs[$id]

		self.$callback

	#end for

#end for

########################################################

##Create Main

##	For top block code, generate a main routine.

##	Instantiate the top block and run as gui or cli.

########################################################

#if $generate_options != 'hb'

if __name__ == '__main__':

	tb = $(class_name)()

	#if $generate_options == 'wx_gui'

	tb.Run()

	#elif $generate_options == 'no_gui'

	tb.start()

	raw_input('Press Enter to quit: ')

	tb.stop()

	#end if

#end if