#!/usr/bin/env python
##################################################
# Gnuradio Python Flow Graph
# Title: Top Block
# Generated: Wed Jun 18 18:45:13 2014
##################################################

from gnuradio import blocks
from gnuradio import eng_notation
from gnuradio import gr
from gnuradio import uhd
from gnuradio import wxgui
from gnuradio.eng_option import eng_option
from gnuradio.filter import firdes
from gnuradio.wxgui import scopesink2
from grc_gnuradio import wxgui as grc_wxgui
from optparse import OptionParser
import time
import wx

class top_block(grc_wxgui.top_block_gui):

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

        ##################################################
        # Variables
        ##################################################
        self.samp_rate = samp_rate = 5000000
        self.center_freq = center_freq = 10000000

        ##################################################
        # Blocks
        ##################################################
        self.wxgui_scopesink2_0_0 = scopesink2.scope_sink_f(
        	self.GetWin(),
        	title="Scope Plot",
        	sample_rate=samp_rate,
        	v_scale=0,
        	v_offset=0,
        	t_scale=0,
        	ac_couple=False,
        	xy_mode=False,
        	num_inputs=2,
        	trig_mode=wxgui.TRIG_MODE_AUTO,
        	y_axis_label="Counts",
        )
        self.Add(self.wxgui_scopesink2_0_0.win)
    
        self.blocks_head_1 = blocks.head(gr.sizeof_gr_complex*1, 10000)
        self.blocks_head_0 = blocks.head(gr.sizeof_gr_complex*1, 10000)
        
	self.blocks_file_sink_0_0 = blocks.file_sink(gr.sizeof_gr_complex*1, "/home/benlapointe/Public/grc_stuff/a7_2", False)
        self.blocks_file_sink_0_0.set_unbuffered(False)
        
	self.blocks_file_sink_0 = blocks.file_sink(gr.sizeof_gr_complex*1, "/home/benlapointe/Public/grc_stuff/a7_1", False)
        self.blocks_file_sink_0.set_unbuffered(False)
        
	self.blocks_complex_to_mag_2 = blocks.complex_to_mag(1)
        self.blocks_complex_to_mag_0 = blocks.complex_to_mag(1)

	# moved the source creations of the two devices so that they are near each other        
	self.uhd_usrp_source_1 = uhd.usrp_source(
        	device_addr="addr=192.168.11.2",
        	stream_args=uhd.stream_args(
        		cpu_format="fc32",
        		channels=range(1),
        	),
        )
	self.uhd_usrp_source_0 = uhd.usrp_source(
        	device_addr="addr=192.168.10.2",
        	stream_args=uhd.stream_args(
        		cpu_format="fc32",
        		channels=range(1),
        	),
        )

	# set usrp_source_1 to get clock from GPSDO
	# set time to 0.0
        self.uhd_usrp_source_1.set_clock_source("gpsdo", 0)
        self.uhd_usrp_source_1.set_time_source("gpsdo", 0)
        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_gain(0, 0)
        
	# set usrp_source_0 to get clock from usrp_source_1
	# set time to 0.0
        self.uhd_usrp_source_0.set_clock_source("external", 0)
        self.uhd_usrp_source_0.set_time_source("external", 0)
        self.uhd_usrp_source_0.set_time_unknown_pps(uhd.time_spec())
        self.uhd_usrp_source_0.set_samp_rate(samp_rate)
  	self.uhd_usrp_source_0.set_gain(0, 0)
        
	# go to sleep (for at least a second) to let the clocks settle
	# (bad problems happen if you don't wait)
	time.sleep(2);

	# ask each USRP what time it thinks it was when it saw the last PPS
	cmd_time = self.uhd_usrp_source_1.get_time_last_pps()
	cmd_time0 = self.uhd_usrp_source_0.get_time_last_pps()

	# convert the "uhd.time_spec_t" to a "double" so that it works with python
	real_seconds = uhd.time_spec_t.get_real_secs(cmd_time)
	real_seconds0 = uhd.time_spec_t.get_real_secs(cmd_time0)

	# optionally print out the time that you saw your last PPS
	# (Nicholas always saw 2.999999995 and 0.999999995
	#print real_seconds
	#print real_seconds0

	# add one second to indicate when in the future you want your command to execute
	future_real_seconds = real_seconds + 2
	future_real_seconds0 = real_seconds0 + 2

	# optionally print out the future time, just to verify that it worked
	#print future_real_seconds
	#print future_real_seconds0

	# convert the "double" back to a "uhd.time_spec_t"
	future_cmd_time = uhd.time_spec_t(future_real_seconds)
	future_cmd_time0 = uhd.time_spec_t(future_real_seconds0)

	# we are going to indicate to the X310s when they are going to 
	#  start issuing the stream commands that we are going to queue
	self.uhd_usrp_source_1.set_command_time(future_cmd_time)
	self.uhd_usrp_source_0.set_command_time(future_cmd_time0)

	# we are going to set the center frequency on all daughter-cards
	#  which is the last step to start streaming
	#  This will be executed at "future_cmd_time" and "future_cmd_time0"
	self.uhd_usrp_source_1.set_center_freq(center_freq, 0)
	self.uhd_usrp_source_0.set_center_freq(center_freq, 0)

	# for completeness we indicate to the USRPs that any further commands sent
	#  should not be queued
	self.uhd_usrp_source_1.clear_command_time()
	self.uhd_usrp_source_0.clear_command_time()

        ##################################################
        # Connections
        ##################################################
        self.connect((self.uhd_usrp_source_1, 0), (self.blocks_complex_to_mag_2, 0))
        self.connect((self.blocks_complex_to_mag_2, 0), (self.wxgui_scopesink2_0_0, 1))
        self.connect((self.blocks_complex_to_mag_0, 0), (self.wxgui_scopesink2_0_0, 0))
        self.connect((self.blocks_head_0, 0), (self.blocks_file_sink_0, 0))
        self.connect((self.blocks_head_1, 0), (self.blocks_file_sink_0_0, 0))
        self.connect((self.uhd_usrp_source_0, 0), (self.blocks_head_0, 0))
        self.connect((self.uhd_usrp_source_1, 0), (self.blocks_head_1, 0))
        self.connect((self.uhd_usrp_source_0, 0), (self.blocks_complex_to_mag_0, 0))


# QT sink close method reimplementation

    def get_samp_rate(self):
        return self.samp_rate

    def set_samp_rate(self, samp_rate):
        self.samp_rate = samp_rate
        self.wxgui_scopesink2_0_0.set_sample_rate(self.samp_rate)
        self.uhd_usrp_source_0.set_samp_rate(self.samp_rate)
        self.uhd_usrp_source_1.set_samp_rate(self.samp_rate)

    def get_center_freq(self):
        return self.center_freq

    def set_center_freq(self, center_freq):
        self.center_freq = center_freq
        self.uhd_usrp_source_0.set_center_freq(self.center_freq, 0)
        self.uhd_usrp_source_1.set_center_freq(self.center_freq, 0)

if __name__ == '__main__':
    import ctypes
    import sys
    if sys.platform.startswith('linux'):
        try:
            x11 = ctypes.cdll.LoadLibrary('libX11.so')
            x11.XInitThreads()
        except:
            print "Warning: failed to XInitThreads()"
    parser = OptionParser(option_class=eng_option, usage="%prog: [options]")
    (options, args) = parser.parse_args()
    tb = top_block()
    tb.Start(True)
    tb.Wait()

