#!/usr/bin/python

#from katcp import BlockingClient, Message
import katcp, numpy, pylab, time, corr, struct, sys, logging

#device_host = "192.168.5.19"
#device_host = "192.168.100.72"
#device_host = "192.168.4.91"
#device_port = 7147

device_port= 7147

if __name__ == '__main__':
    from optparse import OptionParser

    p = OptionParser()
    p.set_usage('tut4_poco_config.py <ROACH_HOSTNAME_or_IP> [options]')
    p.set_description(__doc__)
    p.add_option('-b', action='store', type='string',dest='boffile', default="", help='''Specify the .bof file''')
    opts, args = p.parse_args(sys.argv[1:])

    if args==[]:
        print 'Please specify a ROACH board. \nExiting.'
        exit()
    else:
        roach = args[0]

    if opts.boffile != '':
        boffile = opts.boffile
    else:
        boffile = 'tut4_wideband_poco_2011_Sep_14_1639.bof'

my_corr=corr.katcp_wrapper.FpgaClient(roach,device_port)

if (opts.boffile==None):
    print "Please specify the .bof to Program the ROACH board"
    exit()
while not (my_corr.is_connected()):
        pass
print "Connected Successfully...!!!!"
print "Programming FPGA"
my_corr.progdev(boffile)
print "Programming Done.......>!!!"
time.sleep(2)

my_corr.write_int("scale",500)
print "scale = %i" %my_corr.read_int("scale")

my_corr.write_int("scale1",500)
print "scale1 = %i" %my_corr.read_int("scale1")


my_corr.write_int("fft_shift",0x03f00000)
print "fft_shift = %x" %my_corr.read_int("fft_shift")

my_corr.write_int("acc_len",524288)
print "acc_len = %x" %my_corr.read_int("acc_len")

my_corr.write_int("sys_rst",1)
print "sys_rst = %i" %my_corr.read_int("sys_rst")
my_corr.write_int("sys_rst",0)
print "sys_rst = %i" %my_corr.read_int("sys_rst")

my_corr.write_int("delay_a",0)
print "delay_a = %i" %my_corr.read_int("delay_a")
my_corr.write_int("delay_b",0)
print "delay_b = %i" %my_corr.read_int("delay_b")

my_corr.write_int("theta_fract",0)
print "theta_fract = %i" %my_corr.read_int("theta_fract")
my_corr.write_int("theta_fract1",0)
print "theta_fract1 = %i" %my_corr.read_int("theta_fract1")

my_corr.write_int("theta_fs",0)
print "theta_fs = %i" %my_corr.read_int("theta_fs")
my_corr.write_int("en_theta_fs",1)
print "en_theta_fs = %i" %my_corr.read_int("en_theta_fs")
my_corr.write_int("en_theta_fs",0)
print "en_theta_fs = %i" %my_corr.read_int("en_theta_fs")

my_corr.write_int("theta_fs1",0)
print "theta_fs1 = %i" %my_corr.read_int("theta_fs1")
my_corr.write_int("en_theta_fs1",1)
print "en_theta_fs1 = %i" %my_corr.read_int("en_theta_fs1")
my_corr.write_int("en_theta_fs1",0)
print "en_theta_fs1 = %i" %my_corr.read_int("en_theta_fs1")

my_corr.write_int("fft_fs",0)
print "fft_fs = %i" %my_corr.read_int("fft_fs")
my_corr.write_int("fft_fs1",0)
print "fft_fs1 = %i" %my_corr.read_int("fft_fs1")

my_corr.stop()