#!/usr/bin/env python2
# -*- coding: utf-8 -*-
##################################################
# GNU Radio Python Flow Graph
# Title: Top Block
# Generated: Tue Jul  4 14:16:31 2017
##################################################

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()"

from gnuradio import analog
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 forms
from gnuradio.wxgui import scopesink2
from grc_gnuradio import wxgui as grc_wxgui
from optparse import OptionParser
import math
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")
        _icon_path = "/usr/local/share/icons/hicolor/32x32/apps/gnuradio-grc.png"
        self.SetIcon(wx.Icon(_icon_path, wx.BITMAP_TYPE_ANY))

        ##################################################
        # Variables
        ##################################################
        self.samp_rate = samp_rate = 500e3
        self.phase4 = phase4 = 1.75929
        self.phase3 = phase3 = -.816814
        self.phase2 = phase2 = -0.439823
        self.phase1 = phase1 = -0.314159
        self.gain_tx = gain_tx = 0
        self.gain = gain = 20
        self.f = f = 2450e6

        ##################################################
        # Blocks
        ##################################################
        _phase4_sizer = wx.BoxSizer(wx.VERTICAL)
        self._phase4_text_box = forms.text_box(
        	parent=self.GetWin(),
        	sizer=_phase4_sizer,
        	value=self.phase4,
        	callback=self.set_phase4,
        	label='phase4',
        	converter=forms.float_converter(),
        	proportion=0,
        )
        self._phase4_slider = forms.slider(
        	parent=self.GetWin(),
        	sizer=_phase4_sizer,
        	value=self.phase4,
        	callback=self.set_phase4,
        	minimum=-math.pi,
        	maximum=math.pi,
        	num_steps=100,
        	style=wx.SL_HORIZONTAL,
        	cast=float,
        	proportion=1,
        )
        self.GridAdd(_phase4_sizer, 12, 9, 1, 8)
        _phase3_sizer = wx.BoxSizer(wx.VERTICAL)
        self._phase3_text_box = forms.text_box(
        	parent=self.GetWin(),
        	sizer=_phase3_sizer,
        	value=self.phase3,
        	callback=self.set_phase3,
        	label='phase3',
        	converter=forms.float_converter(),
        	proportion=0,
        )
        self._phase3_slider = forms.slider(
        	parent=self.GetWin(),
        	sizer=_phase3_sizer,
        	value=self.phase3,
        	callback=self.set_phase3,
        	minimum=-math.pi,
        	maximum=math.pi,
        	num_steps=100,
        	style=wx.SL_HORIZONTAL,
        	cast=float,
        	proportion=1,
        )
        self.GridAdd(_phase3_sizer, 11, 9, 1, 8)
        _phase2_sizer = wx.BoxSizer(wx.VERTICAL)
        self._phase2_text_box = forms.text_box(
        	parent=self.GetWin(),
        	sizer=_phase2_sizer,
        	value=self.phase2,
        	callback=self.set_phase2,
        	label='phase2',
        	converter=forms.float_converter(),
        	proportion=0,
        )
        self._phase2_slider = forms.slider(
        	parent=self.GetWin(),
        	sizer=_phase2_sizer,
        	value=self.phase2,
        	callback=self.set_phase2,
        	minimum=-math.pi,
        	maximum=math.pi,
        	num_steps=100,
        	style=wx.SL_HORIZONTAL,
        	cast=float,
        	proportion=1,
        )
        self.GridAdd(_phase2_sizer, 10, 9, 1, 8)
        _phase1_sizer = wx.BoxSizer(wx.VERTICAL)
        self._phase1_text_box = forms.text_box(
        	parent=self.GetWin(),
        	sizer=_phase1_sizer,
        	value=self.phase1,
        	callback=self.set_phase1,
        	label='phase1',
        	converter=forms.float_converter(),
        	proportion=0,
        )
        self._phase1_slider = forms.slider(
        	parent=self.GetWin(),
        	sizer=_phase1_sizer,
        	value=self.phase1,
        	callback=self.set_phase1,
        	minimum=-math.pi,
        	maximum=math.pi,
        	num_steps=100,
        	style=wx.SL_HORIZONTAL,
        	cast=float,
        	proportion=1,
        )
        self.GridAdd(_phase1_sizer, 9, 9, 1, 8)
        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=4,
        	trig_mode=wxgui.TRIG_MODE_AUTO,
        	y_axis_label="Counts",
        )
        self.Add(self.wxgui_scopesink2_0_0.win)


        self.transmitter = uhd.usrp_sink(
        	",".join(("addr0=192.168.10.3,addr1=192.168.10.5,addr2=192.168.10.4", "")),
        	uhd.stream_args(
        		cpu_format="fc32",
        		channels=range(1),
        	),
        )
        self.receiver = uhd.usrp_source(
        	",".join(("addr0=192.168.10.5,addr1=192.168.10.4,addr2=192.168.10.3", "")),
        	uhd.stream_args(
        		cpu_format="fc32",
        		channels=range(4),
        	),
        )



        self.transmitter.set_clock_source("external", 0)
        self.transmitter.set_time_source("external", 0)
        self.transmitter.set_subdev_spec("A:0", 0)
        self.receiver.set_clock_source("external", 0)
        self.receiver.set_time_source("external", 0)
        self.receiver.set_subdev_spec("A:0 B:0", 0)
        self.receiver.set_clock_source("external", 1)
        self.receiver.set_time_source("external", 1)
        self.receiver.set_subdev_spec("A:0 B:0", 1)


        self.transmitter.set_time_unknown_pps(uhd.time_spec())

	time.sleep(1)

	print ('Timing reference')
	print('@TX')
	print self.transmitter.get_time_last_pps(0).get_real_secs()
	print('@RX')
	print self.receiver.get_time_last_pps(0).get_real_secs()


        self.transmitter.set_samp_rate(samp_rate)

        self.transmitter.set_gain(gain_tx, 0)
        self.transmitter.set_antenna("TX/RX", 0)

        self.receiver.set_samp_rate(samp_rate)

        self.receiver.set_gain(gain, 0)
        self.receiver.set_antenna("TX/RX", 0)

        self.receiver.set_gain(gain, 1)
        self.receiver.set_antenna("TX/RX", 1)

        self.receiver.set_gain(gain, 2)
        self.receiver.set_antenna("TX/RX", 2)

        self.receiver.set_gain(gain, 3)
        self.receiver.set_antenna("TX/RX", 3)



        self.blocks_multiply_const_vxx_0_0_0_0_0 = blocks.multiply_const_vcc((complex(math.cos(phase4),math.sin(phase4)), ))
        self.blocks_multiply_const_vxx_0_0_0_0 = blocks.multiply_const_vcc((complex(math.cos(phase3),math.sin(phase3)), ))
        self.blocks_multiply_const_vxx_0_0_0 = blocks.multiply_const_vcc((complex(math.cos(phase2),math.sin(phase2)), ))
        self.blocks_multiply_const_vxx_0_0 = blocks.multiply_const_vcc((complex(math.cos(phase1),math.sin(phase1)), ))
        self.blocks_complex_to_real_0_0_1 = blocks.complex_to_real(1)
        self.blocks_complex_to_real_0_0_0_0 = blocks.complex_to_real(1)
        self.blocks_complex_to_real_0_0_0 = blocks.complex_to_real(1)
        self.blocks_complex_to_real_0_0 = blocks.complex_to_real(1)
        self.analog_sig_source_x_0 = analog.sig_source_c(samp_rate, analog.GR_SIN_WAVE, 1000, .3, 0)


	now = self.transmitter.get_time_now()
	cmd_time=now+uhd.time_spec(0.5)

	self.receiver.set_command_time(cmd_time)
	self.transmitter.set_command_time(cmd_time)


        self.transmitter.set_center_freq(f, 0)
        self.receiver.set_center_freq(f, 0)
        self.receiver.set_center_freq(f, 1)
        self.receiver.set_center_freq(f, 2)
        self.receiver.set_center_freq(f, 3)

        self.receiver.clear_command_time()
        self.transmitter.clear_command_time()

	time.sleep(1)	
	print('Transmitter LO')
	print(self.transmitter.get_sensor('lo_locked'))
	print('Receiver LO')
	print(self.receiver.get_sensor('lo_locked'))


	now = self.transmitter.get_time_now()
	starttime = now + uhd.time_spec(0.5)

	#starttime_T=now + uhd.time_spec(0.5000003)

	self.receiver.set_start_time(starttime)
	self.transmitter.set_start_time(starttime)




        ##################################################
        # Connections
        ##################################################
        self.connect((self.analog_sig_source_x_0, 0), (self.transmitter, 0))    
        self.connect((self.blocks_complex_to_real_0_0, 0), (self.wxgui_scopesink2_0_0, 0))    
        self.connect((self.blocks_complex_to_real_0_0_0, 0), (self.wxgui_scopesink2_0_0, 2))    
        self.connect((self.blocks_complex_to_real_0_0_0_0, 0), (self.wxgui_scopesink2_0_0, 3))    
        self.connect((self.blocks_complex_to_real_0_0_1, 0), (self.wxgui_scopesink2_0_0, 1))    
        self.connect((self.blocks_multiply_const_vxx_0_0, 0), (self.blocks_complex_to_real_0_0, 0))    
        self.connect((self.blocks_multiply_const_vxx_0_0_0, 0), (self.blocks_complex_to_real_0_0_1, 0))    
        self.connect((self.blocks_multiply_const_vxx_0_0_0_0, 0), (self.blocks_complex_to_real_0_0_0, 0))    
        self.connect((self.blocks_multiply_const_vxx_0_0_0_0_0, 0), (self.blocks_complex_to_real_0_0_0_0, 0))    
        self.connect((self.receiver, 0), (self.blocks_multiply_const_vxx_0_0, 0))    
        self.connect((self.receiver, 1), (self.blocks_multiply_const_vxx_0_0_0, 0))    
        self.connect((self.receiver, 2), (self.blocks_multiply_const_vxx_0_0_0_0, 0))    
        self.connect((self.receiver, 3), (self.blocks_multiply_const_vxx_0_0_0_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.analog_sig_source_x_0.set_sampling_freq(self.samp_rate)
        self.transmitter.set_samp_rate(self.samp_rate)
        self.wxgui_scopesink2_0_0.set_sample_rate(self.samp_rate)
        self.receiver.set_samp_rate(self.samp_rate)

    def get_phase4(self):
        return self.phase4

    def set_phase4(self, phase4):
        self.phase4 = phase4
        self._phase4_slider.set_value(self.phase4)
        self._phase4_text_box.set_value(self.phase4)
        self.blocks_multiply_const_vxx_0_0_0_0_0.set_k((complex(math.cos(self.phase4),math.sin(self.phase4)), ))

    def get_phase3(self):
        return self.phase3

    def set_phase3(self, phase3):
        self.phase3 = phase3
        self._phase3_slider.set_value(self.phase3)
        self._phase3_text_box.set_value(self.phase3)
        self.blocks_multiply_const_vxx_0_0_0_0.set_k((complex(math.cos(self.phase3),math.sin(self.phase3)), ))

    def get_phase2(self):
        return self.phase2

    def set_phase2(self, phase2):
        self.phase2 = phase2
        self._phase2_slider.set_value(self.phase2)
        self._phase2_text_box.set_value(self.phase2)
        self.blocks_multiply_const_vxx_0_0_0.set_k((complex(math.cos(self.phase2),math.sin(self.phase2)), ))

    def get_phase1(self):
        return self.phase1

    def set_phase1(self, phase1):
        self.phase1 = phase1
        self._phase1_slider.set_value(self.phase1)
        self._phase1_text_box.set_value(self.phase1)
        self.blocks_multiply_const_vxx_0_0.set_k((complex(math.cos(self.phase1),math.sin(self.phase1)), ))

    def get_gain_tx(self):
        return self.gain_tx

    def set_gain_tx(self, gain_tx):
        self.gain_tx = gain_tx
        self.transmitter.set_gain(self.gain_tx, 0)
        	

    def get_gain(self):
        return self.gain

    def set_gain(self, gain):
        self.gain = gain
        self.transmitter.set_gain(self.gain, 1)
        	
        self.receiver.set_gain(self.gain, 0)
        	
        self.receiver.set_gain(self.gain, 1)
        	
        self.receiver.set_gain(self.gain, 2)
        	
        self.receiver.set_gain(self.gain, 3)
        	

    def get_f(self):
        return self.f

    def set_f(self, f):
        self.f = f
        self.transmitter.set_center_freq(self.f, 0)
        self.transmitter.set_center_freq(self.f, 1)
        self.receiver.set_center_freq(self.f, 0)
        self.receiver.set_center_freq(self.f, 1)
        self.receiver.set_center_freq(self.f, 2)
        self.receiver.set_center_freq(self.f, 3)


def main(top_block_cls=top_block, options=None):

    tb = top_block_cls()
    tb.Start(True)
    tb.Wait()


if __name__ == '__main__':
    main()
