#!/usr/bin/env python
##################################################
# Gnuradio Python Flow Graph
# Title: Top Block
# Generated: Tue May  3 08:59:48 2011
##################################################

from gnuradio import eng_notation
from gnuradio import gr
from gnuradio import uhd
from gnuradio.eng_option import eng_option
from gnuradio.gr import firdes
from grc_gnuradio import wxgui as grc_wxgui
from optparse import OptionParser
import wx
import time
class top_block(grc_wxgui.top_block_gui):

	def __init__(self,f0,f1):
		grc_wxgui.top_block_gui.__init__(self, title="Top Block")

		##################################################
		# Variables
		##################################################
		self.samp_rate = samp_rate = 1e6

		##################################################
		# Blocks
		##################################################
		self.uhd_usrp_source_1 = uhd.usrp_source(
			device_addr="addr0=192.168.10.2,addr1=192.168.40.2",
			io_type=uhd.io_type.COMPLEX_FLOAT32,
			num_channels=2,
		)
		self.uhd_usrp_source_1.set_clock_config(uhd.clock_config.external(), uhd.ALL_MBOARDS)
		self.uhd_usrp_source_1.set_time_unknown_pps(uhd.time_spec())
		self.uhd_usrp_source_1.set_samp_rate(samp_rate)
		self.uhd_usrp_source_1.set_center_freq(300e6, 0)
		self.uhd_usrp_source_1.set_gain(0, 0)
		self.uhd_usrp_source_1.set_center_freq(300e6, 1)
		self.uhd_usrp_source_1.set_gain(0, 1)
		#self.gr_head_0_0 = gr.head(gr.sizeof_gr_complex*1, 1024)
		#self.gr_head_0 = gr.head(gr.sizeof_gr_complex*1, 1024)
		self.gr_file_sink_0_0 = gr.file_sink(gr.sizeof_gr_complex*1, f0)
		self.gr_file_sink_0_0.set_unbuffered(False)
		self.gr_file_sink_0 = gr.file_sink(gr.sizeof_gr_complex*1, f1)
		self.gr_file_sink_0.set_unbuffered(False)

		##################################################
		# Connections
		##################################################
		#self.connect((self.gr_head_0, 0), (self.gr_file_sink_0, 0))
		#self.connect((self.uhd_usrp_source_1, 0), (self.gr_head_0, 0))
		#self.connect((self.gr_head_0_0, 0), (self.gr_file_sink_0_0, 0))
		#self.connect((self.uhd_usrp_source_1, 1), (self.gr_head_0_0, 0))
		self.connect((self.uhd_usrp_source_1, 0), (self.gr_file_sink_0, 0))
		self.connect((self.uhd_usrp_source_1, 1), (self.gr_file_sink_0_0, 0))
	def get_samp_rate(self):
		return self.samp_rate

	def set_samp_rate(self, samp_rate):
		self.samp_rate = samp_rate
		self.uhd_usrp_source_1.set_samp_rate(self.samp_rate)

if __name__ == '__main__':
	parser = OptionParser(option_class=eng_option, usage="%prog: [options]")
	(options, args) = parser.parse_args()

	for i in range(6):
		filename0 = "run0_"+str(i)+".bin"
		filename1 = "run1_"+str(i)+".bin"
		print filename0
		tb = top_block(filename0,filename1)
		tb.start()
		time.sleep(4)
		tb.stop()
#		tb.wait()
		time.sleep(1)