def plot_main():
try:
fpga = casperfpga.katcp_fpga.KatcpFpga(roach_ppc_ip, timeout=20.)
except RuntimeError:
fpga = None
# Roach interface
ri = roachInterface(sys.argv[1], fpga, gc, regs)
while 1:
plot_opt(ri)
return
def plot_main():
try:
fpga = casperfpga.katcp_fpga.KatcpFpga(roach_ppc_ip, timeout = 3.)
except (RuntimeError, AttributeError):
fpga = None
# Roach interface
ri = roachInterface(fpga, gc, regs, None)
while 1:
plot_opt(ri)
return
def systemInit():
fpga = getFPGA()
if not fpga:
print "\nROACH link is down"
return
# Valon object
#valon = getValon()
valon = None
# Roach PPC object
fpga = getFPGA()
# Roach interface
ri = roachInterface(fpga, gc, regs, valon)
if (ri.uploadfpg() < 0):
print "\nFirmware upload failed"
time.sleep(0.3)
# UDP socket
s = socket(AF_PACKET, SOCK_RAW, htons(3))
# UDP object
udp = roachDownlink(ri, fpga, gc, regs, s, ri.accum_freq)
try:
# Windfreak calibration
#cal = rfcalibration.CalibrationRF(synthID, attenID)
#cal.setSynthToDefault(clkFreq, clkPow, LOFreq, LOPow)
#cal.calibrateADC(fpga, target_rms_mv, attOut, attIn) # ADC level calibration
#initValon(valon)
print "Valon initiliazed"
except OSError:
print '\033[93mValon Synthesizer could not be initialized: Check comm port and power supply\033[93m'
return
except IndexError:
print '\033[93mValon Synthesizer could not be initialized: Check comm port and power supply\033[93m'
return
fpga.write_int(regs[np.where(regs == 'accum_len_reg')[0][0]][1],
ri.accum_len - 1)
time.sleep(0.1)
fpga.write_int(regs[np.where(regs == 'dds_shift_reg')[0][0]][1],
int(gc[np.where(gc == 'dds_shift')[0][0]][1]))
time.sleep(0.1)
#ri.lpf(ri.boxcar)
if (ri.qdrCal() < 0):
print '\033[93mQDR calibration failed... Check FPGA clock source\033[93m'
return
else:
fpga.write_int(regs[np.where(regs == 'write_qdr_status_reg')[0][0]][1],
1)
time.sleep(0.1)
try:
udp.configDownlink()
print "UDP Downlink configured!"
except AttributeError:
print "UDP Downlink could not be configured. Check ROACH connection."
return
return
def plot_main():
try:
fpga = casperfpga.CasperFpga(
gc[np.where(gc == 'roach_ppc_ip')[0][0]][1], timeout=3.)
except RuntimeError:
fpga = None
# Roach interface
ri = roachInterface(fpga, gc, regs, None)
while 1:
plot_opt(ri)
return
def vnaSweepConsole():
"""Does a wideband sweep of the RF band, saves data in vna_savepath
as .npy files"""
# UDP socket
s = socket(AF_PACKET, SOCK_RAW, htons(3))
# Valon object
#valon = getValon()
valon = None
# Roach PPC object
fpga = getFPGA()
# Roach interface
ri = roachInterface(fpga, gc, regs, valon)
# UDP object
udp = roachDownlink(ri, fpga, gc, regs, s, ri.accum_freq)
udp.configSocket()
Navg = np.int(gc[np.where(gc == 'Navg')[0][0]][1])
if not os.path.exists(vna_savepath):
os.makedirs(vna_savepath)
sweep_dir = vna_savepath + '/' + \
str(int(time.time())) + '-' + time.strftime('%b-%d-%Y-%H-%M-%S') + '.dir'
os.mkdir(sweep_dir)
np.save("./last_vna_dir.npy", sweep_dir)
print sweep_dir
#valon.set_frequency(LO, center_freq/1.0e6)
span = ri.pos_delta
print "Sweep Span =", 2 * np.round(ri.pos_delta, 2), "Hz"
start = center_freq * 1.0e6 - (span)
stop = center_freq * 1.0e6 + (span)
sweep_freqs = np.arange(start, stop, lo_step)
sweep_freqs = np.round(sweep_freqs / lo_step) * lo_step
if not np.size(ri.freq_comb):
ri.makeFreqComb()
np.save(sweep_dir + '/bb_freqs.npy', ri.freq_comb)
np.save(sweep_dir + '/sweep_freqs.npy', sweep_freqs)
Nchan = len(ri.freq_comb)
if not Nchan:
Nchan = fpga.read_int(
regs[np.where(regs == 'read_comb_len_reg')[0][0]][1])
idx = 0
while (idx < len(sweep_freqs)):
print 'LO freq =', sweep_freqs[idx] / 1.0e6
#valon.set_frequency(LO, sweep_freqs[idx]/1.0e6)
time.sleep(0.2)
#time.sleep(0.1)
if (udp.saveSweepData(Navg, sweep_dir, sweep_freqs[idx], Nchan) < 0):
continue
else:
idx += 1
#time.sleep(0.1)
#valon.set_frequency(LO, center_freq) # LO
return
def calibrateADC(target_rms_mv, outAtten, inAtten):
"""Automatically set RUDAT attenuation values to achieve desired ADC rms level
inputs:
float target_rms_mv: The target ADC rms voltage level, in mV,
for either I or Q channel
float outAtten: Starting output attenuation, dB
float inAtten: Starting input attenuation, dB"""
#setAtten(outAtten, inAtten)
fpga = getFPGA()
valon = None
ri = roachInterface(fpga, gc, regs, valon)
print "Start atten:", outAtten, inAtten
rmsI, rmsQ, __, __ = ri.rmsVoltageADC()
avg_rms_0 = (rmsI + rmsQ) / 2.
print "Target RMS:", target_rms_mv, "mV"
print "Current RMS:", avg_rms_0, "mV"
if avg_rms_0 < target_rms_mv:
avg_rms = avg_rms_0
while avg_rms < target_rms_mv:
time.sleep(0.1)
if inAtten > 1:
inAtten -= 1
else:
outAtten -= 1
if (inAtten == 1) and (outAtten == 1):
break
#setAtten(outAtten, inAtten)
rmsI, rmsQ, __, __ = ri.rmsVoltageADC()
avg_rms = (rmsI + rmsQ) / 2.
outA, inA = 0, 0 #readAtten()
print outA, inA
if avg_rms_0 > target_rms_mv:
avg_rms = avg_rms_0
while avg_rms > target_rms_mv:
time.sleep(0.1)
if outAtten < 30:
outAtten += 1
else:
inAtten += 1
if (inAtten > 30) and (outAtten > 30):
break
#setAtten(outAtten, inAtten)
rmsI, rmsQ, __, __ = ri.rmsVoltageADC()
avg_rms = (rmsI + rmsQ) / 2.
outA, inA = 0, 0 #readAtten()
print outA, inA
new_out, new_in = 0, 0 #readAtten()
print
print "Final atten:", new_out, new_in
print "Current RMS:", avg_rms, "mV"
return
def testConn(self, fpga):
"""Tests the link to Roach2 PPC, using return from getFPGA()
inputs:
casperfpga object fpga: The fpga object
outputs: the fpga object"""
if not fpga:
try:
fpga = casperfpga.CasperFpga(self.roach_ip, timeout=3.)
# Roach interface
self.ri = roachInterface(self.fpga, self.gc, self.regs, None)
except RuntimeError:
logging.warning(
"No connection to ROACH. If booting, wait 30 seconds and retry. Otherwise, check gc config."
)
return fpga
def writeVnaComb(cw=False):
# Roach PPC object
fpga = getFPGA()
if not fpga:
print "\nROACH link is down"
return
# Roach interface
ri = roachInterface(fpga, gc, regs, None)
try:
if cw:
ri.freq_comb = test_freq
else:
ri.makeFreqComb()
if (len(ri.freq_comb) > 400):
fpga.write_int(regs[np.where(regs == 'fft_shift_reg')[0][0]][1],
2**5 - 1)
time.sleep(0.1)
else:
fpga.write_int(regs[np.where(regs == 'fft_shift_reg')[0][0]][1],
2**9 - 1)
time.sleep(0.1)
ri.upconvert = np.sort(
((ri.freq_comb + (center_freq) * 1.0e6)) / 1.0e6)
print "RF tones =", ri.upconvert
ri.writeQDR(ri.freq_comb, transfunc=False)
np.save("last_freq_comb.npy", ri.freq_comb)
if not (fpga.read_int(
regs[np.where(regs == 'dds_shift_reg')[0][0]][1])):
if regs[np.where(regs == 'DDC_mixerout_bram_reg')[0]
[0]][1] in fpga.listdev():
shift = ri.return_shift(0)
if (shift < 0):
print "\nError finding dds shift: Try writing full frequency comb (N = 1000), or single test frequency. Then try again"
return
else:
fpga.write_int(
regs[np.where(regs == 'dds_shift_reg')[0][0]][1],
shift)
print "Wrote DDS shift (" + str(shift) + ")"
else:
fpga.write_int(
regs[np.where(regs == 'dds_shift_reg')[0][0]][1],
ri.dds_shift)
except KeyboardInterrupt:
return
return
def main():
pi = None
s = None
try:
fpga = casperfpga.katcp_fpga.KatcpFpga(roach_ppc_ip, timeout = 3.)
except (RuntimeError, AttributeError):
fpga = None
# UDP socket
s = socket.socket(socket.AF_PACKET, socket.SOCK_RAW, socket.htons(3))
# Roach interface
ri = roachInterface(fpga, gc, regs, None)
# GbE interface
udp = roachDownlink(ri, fpga, gc, regs, network, s, ri.accum_freq)
udp.configSocket()
#os.system('clear')
while 1:
try:
upload_status = 0
name = ''
build_time = ''
if fpga:
if fpga.is_running():
firmware_info = fpga.get_config_file_info()
name = firmware_info['name']
build_time = firmware_info['build_time']
upload_status = 1
time.sleep(0.1)
pi = piSocket(FC2)
upload_status = main_opt(fpga, ri, udp, pi, upload_status, name, build_time)
if pi:
pi.shutdown(1)
pi.close()
except KeyboardInterrupt:
if pi:
pi.shutdown(1)
pi.close()
break
return
def main():
"""Main function, try to initialize the system the first time, then open the menu"""
s = None
try:
fpga = casperfpga.CasperFpga(roach_ip, timeout = 120.)
print "\nConnected to: " + roach_ip
except RuntimeError:
fpga = None
print "\nRoach link is down"
# UDP socket
s = socket(AF_PACKET, SOCK_RAW, htons(3))
# Roach interface
ri = roachInterface(fpga, gc, regs, None)
# Windfreak synthesizer
synthRF = synthclass.Synthesizer(synthID)
# GbE interface
udp = roachDownlink(ri, fpga, gc, regs, s, ri.accum_freq)
udp.configSocket()
os.system('clear')
valon = None
while 1:
try:
upload_status = 0
if fpga:
if fpga.is_running():
upload_status = 1
time.sleep(0.1)
#fpga = None
#ri = None
#udp = None
upload_status = main_opt(fpga, ri, udp, valon, upload_status)
except TypeError:
pass
return
def main():
pi = None
s = None
try:
fpga = casperfpga.katcp_fpga.KatcpFpga(roach_ppc_ip, timeout=20.)
except (RuntimeError, AttributeError):
fpga = None
# UDP socket
s = socket(AF_PACKET, SOCK_RAW, htons(3))
# Roach interface
ri = roachInterface(sys.argv[1], fpga, gc, regs)
# GbE interface
udp = roachDownlink(ri, fpga, gc, regs, s, ri.accum_freq)
udp.configSocket()
#os.system('clear')
while 1:
upload_status = 0
name = ''
build_time = ''
if fpga and fpga.is_running():
upload_status = 1
if not pi:
try:
print "Opening pi socket"
pi = piSocket(FC1)
except (TypeError, KeyboardInterrupt):
#if pi:
#pi.shutdown(1)
#pi.close()
pass
#break
try:
upload_status = main_opt(fpga, ri, udp, pi, upload_status, name,
build_time)
except KeyboardInterrupt:
break
return
def main():
s = None
try:
fpga = casperfpga.CasperFpga(
gc[np.where(gc == 'roach_ppc_ip')[0][0]][1], timeout=120.)
except RuntimeError:
fpga = None
# UDP socket
s = socket(AF_PACKET, SOCK_RAW, htons(3))
# Valon synthesizer instance
try:
#valon = valon_synth9.Synthesizer(gc[np.where(gc == 'valon_comm_port')[0][0]][1])
valon = None
print "$NO VALON$"
except OSError:
"Valon could not be initialized. Check comm port and power supply."
# Roach interface
ri = roachInterface(fpga, gc, regs, valon)
# GbE interface
udp = roachDownlink(ri, fpga, gc, regs, s, ri.accum_freq)
udp.configSocket()
os.system('clear')
while 1:
try:
upload_status = 0
if fpga:
if fpga.is_running():
#firmware_info = fpga.get_config_file_info()
upload_status = 1
time.sleep(0.1)
upload_status = main_opt(fpga, ri, udp, valon, upload_status)
except TypeError:
pass
return
def __init__(self):
QtGui.QMainWindow.__init__(self)
# Load of main window GUI
# The GUI was developed in QT Designer
self.ui = uic.loadUi("src/gui/main.ui")
# Full Screen
self.ui.showMaximized()
screen = QtGui.QDesktopWidget().screenGeometry()
# Screen dimensions
self.size_x = screen.width()
self.size_y = screen.height()
self.ui.setWindowFlags(self.ui.windowFlags()
| QtCore.Qt.CustomizeWindowHint)
self.ui.setWindowFlags(self.ui.windowFlags()
& ~QtCore.Qt.WindowMaximizeButtonHint)
self.ui.ctrlFrame.resize(280, self.size_y - 120)
self.ui.tabPlots.resize(self.size_x - 600, self.size_y - 150)
self.ui.plotFrame.resize(self.size_x - 620, self.size_y - 210)
self.ui.plotFrame.setLayout(self.ui.MainPlot)
self.ui.plotFrame_2.resize(self.size_x - 620, self.size_y - 210)
self.ui.plotFrame_2.setLayout(self.ui.MainPlot_2)
self.ui.Terminal.move(self.size_x - 290, self.size_y / 2 - 100)
self.ui.Terminal.resize(self.size_x - 1085, self.size_y / 2 - 30)
self.ui.Terminal.setLayout(self.ui.TermVBox)
self.ui.loggsFrame.move(self.size_x - 290, 10)
self.ui.loggsFrame.resize(self.size_x - 1085, self.size_y / 2 - 120)
self.ui.loggsFrame.setLayout(self.ui.loggsText)
# Logging
logTextBox = QTextEditLogger(self)
# You can format what is printed to text box
logTextBox.setFormatter(
logging.Formatter('%(asctime)s - %(levelname)s - %(message)s'))
logging.getLogger().addHandler(logTextBox)
# You can control the logging level
logging.getLogger().setLevel(logging.INFO)
self.ui.loggsText.addWidget(logTextBox.widget)
# Initial settings
# Loading General Configuration file
# Load general settings
self.gc = {}
with open("./config/general_config") as f:
for line in f:
if line[0] != "#" and line[0] != "\n":
(key, val) = line.split()
self.gc[key] = val
logging.info('Loading configuration parameters ...')
# Load list of firmware registers (note: must manually update for different versions)
self.regs = {}
with open("./config/firmware_registers") as f:
for line in f:
if line[0] != "#" and line[0] != "\n":
(key, val) = line.split()
self.regs[key] = val
logging.info('Loading firmware registers ...')
# Paths of firmware and directories to save data
self.firmware = self.gc['FIRMWARE_FILE']
self.ui.firmEdit.setText(self.firmware)
self.vna_savepath = self.gc['VNA_SAVEPATH']
self.targ_savepath = self.gc['TARG_SAVEPATH']
self.dirfile_savepath = self.gc['DIRFILE_SAVEPATH']
self.ui.vnaEdit.setText(self.vna_savepath)
self.ui.tarEdit.setText(self.targ_savepath)
self.ui.streamEdit.setText(self.dirfile_savepath)
# UDP packet
self.buf_size = int(self.gc['buf_size'])
self.header_len = int(self.gc['header_len'])
# Ethernet port
self.eth_port = self.gc['udp_dest_device']
self.ui.ethEdit.setText(self.eth_port)
os.system("sudo ip link set " + self.eth_port + " mtu 9000")
# Source (V6) Data for V6
self.udp_src_ip = self.gc['udp_src_ip']
self.udp_src_mac = self.gc['udp_src_mac']
self.udp_src_port = self.gc['udp_src_port']
self.ui.ipSrcEdit.setText(self.udp_src_ip)
self.ui.macSrcEdit.setText(self.udp_src_mac)
self.ui.portSrcEdit.setText(self.udp_src_port)
self.dds_shift = self.gc['dds_shift']
self.udp_dst_ip = self.gc['udp_dest_ip']
self.udp_dst_mac = self.gc['udp_dest_mac']
self.udp_dst_port = self.gc['udp_dst_port']
self.ui.ipDstEdit.setText(self.udp_dst_ip)
self.ui.macDstEdit.setText(self.udp_dst_mac)
self.ui.portDstEdit.setText(self.udp_dst_port)
# About the ROACH
self.roach_ip = self.gc['roach_ppc_ip']
self.ui.roachIPEdit.setText(self.roach_ip)
# Windfreak Synthesizer params
self.synthID = self.gc['synthID']
self.clkFreq = np.float(self.gc['clkFreq'])
self.clkPow = np.float(self.gc['clkPow'])
self.LOFreq = np.float(self.gc['LOFreq'])
self.LOPow = np.float(self.gc['LOPow'])
self.center_freq = np.float(self.gc['center_freq'])
self.lo_step = np.float(self.gc['lo_step'])
self.ui.freqClk.setText(str(self.clkFreq))
self.ui.powClk.setText(str(self.clkPow))
self.ui.loFreq.setText(str(self.LOFreq))
self.ui.loPow.setText(str(self.LOPow))
# Limits of test comb
self.min_pos_freq = np.float(self.gc['min_pos_freq'])
self.max_pos_freq = np.float(self.gc['max_pos_freq'])
self.min_neg_freq = np.float(self.gc['min_neg_freq'])
self.max_neg_freq = np.float(self.gc['max_neg_freq'])
self.symm_offset = np.float(self.gc['symm_offset'])
self.Nfreq = int(self.gc['Nfreq'])
self.ui.minPosEdit.setText(str(self.min_pos_freq / 1.0e6))
self.ui.maxPosEdit.setText(str(self.max_pos_freq / 1.0e6))
self.ui.minNegEdit.setText(str(self.min_neg_freq / 1.0e6))
self.ui.maxNegEdit.setText(str(self.max_neg_freq / 1.0e6))
self.ui.offsetEdit.setText(str(self.symm_offset / 1.0e6))
self.ui.nFreqsEdit.setText(str(self.Nfreq))
# Attenuation
att_ID_1 = int(self.gc['att_ID_1'])
att_ID_2 = int(self.gc['att_ID_2'])
self.attenID = [att_ID_1, att_ID_2]
self.ui.attInIDEdit.setText(str(att_ID_1))
self.ui.attOutIDEdit.setText(str(att_ID_2))
self.att_In = int(self.gc['attIn'])
self.att_Out = int(self.gc['attOut'])
self.target_rms = np.float(self.gc['target_rms_mv'])
self.ui.attInEdit.setText(str(self.att_In))
self.ui.attOutEdit.setText(str(self.att_Out))
self.ui.tarLevelEdit.setText(str(self.target_rms))
# Optional test frequencies
self.test_freq = np.float(self.gc['test_freq'])
self.test_freq = np.array([self.test_freq])
self.freq_list = self.gc['freq_list']
# Parameters for resonator search
self.smoothing_scale = np.float(self.gc['smoothing_scale'])
self.peak_threshold = np.float(self.gc['peak_threshold'])
self.spacing_threshold = np.float(self.gc['spacing_threshold'])
# VNA Sweep
self.startVNA = -255.5e6
self.stopVNA = 255.5e6
self.ui.centralEdit.setText(str(self.center_freq))
self.ui.startEdit.setText(str(self.startVNA / 1.0e6))
self.ui.stopEdit.setText(str(self.stopVNA / 1.0e6))
self.ui.stepEdit.setText(str(self.lo_step / 1.0e6))
self.ui.nTonesEdit.setText(str(self.Nfreq))
# Tool bar
# ROACH status
self.ui.actionRoach.triggered.connect(self.roach_connection)
# ROACH network
self.ui.actionNetwork.triggered.connect(self.roach_network)
# Synthesizer
self.ui.actionSynthesizer.triggered.connect(self.roach_synth)
# Attenuattors
self.ui.actionRF_Calibration.triggered.connect(self.roach_atten)
# QDR Calibration
self.ui.actionQDR_Calibration.triggered.connect(self.qdr_cal)
# Buttons
# Roach
# Roach Settings
self.ui.firmDir.mousePressEvent = self.chooseFirmPath
self.ui.vnaDir.mousePressEvent = self.chooseVNAPath
self.ui.targDir.mousePressEvent = self.chooseTargPath
self.ui.streamDir.mousePressEvent = self.chooseStreamPath
self.ui.upFirmBtn.mousePressEvent = self.upload_firmware
self.ui.synthBtn.mousePressEvent = self.roach_synth
self.ui.udpConfBtn.mousePressEvent = self.roach_network
self.ui.udpTestBtn.mousePressEvent = self.test_udp
self.ui.attBtn.mousePressEvent = self.roach_atten
self.ui.writeTestBtn.mousePressEvent = self.write_test_comb
self.ui.plotSweepBtn.mousePressEvent = self.start_plot_VNA
self.ui.startSweepBtn.mousePressEvent = self.start_VNA_sweep
# Iniatialising
self.statusConn = 0
self.statusFirm = 0
self.statusSynth = 0
self.statusAtt = 0
self.statusNet = 0
self.s = None
self.fpga = None
try:
self.fpga = casperfpga.CasperFpga(self.roach_ip, timeout=100.)
icon.addPixmap(QPixmap('./src/icon/ok_icon.png'))
self.ui.actionRoach_Status.setIcon(icon)
logging.info("Connected to: " + self.roach_ip)
except:
self.fpga = None
self.statusConn = 1
logging.info("Roach link is down")
# Check firmware
if self.fpga:
logging.info('Firmware is uploaded')
if self.fpga.is_running():
self.ui.upFirmBtn.setStyleSheet("""QWidget {
color: white;
background-color: green
}""")
else:
self.statusFirm = 1
else:
self.statusFirm = 1
# UDP socket
# Run with root permissions
try:
self.s = socket(AF_PACKET, SOCK_RAW, htons(3))
logging.info('Socket is initialised.')
except:
logging.error(
'Socket is not initialised. Permissions are required')
# Roach interface
self.ri = roachInterface(self.fpga, self.gc, self.regs, None)
# GbE interface
try:
self.udp = roachDownlink(self.ri, self.fpga, self.gc, self.regs,
self.s, self.ri.accum_freq)
self.udp.configSocket()
logging.info('UDP configuration done.')
except:
logging.error("UDP connection couldn't be initialised.")
# Creation of Plot
self.fig1 = Figure()
self.addmpl_homodyne(self.fig1)
# Creation of Plot
self.fig1 = Figure()
self.addmpl_vna(self.fig1)
# To use LATEX in plots
matplotlib.rc('text', usetex=True)
matplotlib.rcParams['text.latex.preamble'] = [r"\usepackage{amsmath}"]
# IPython console
self.console = EmbedIPython()
self.console.kernel.shell.run_cell('%pylab qt')
self.console.execute("cd ./")
self.ui.TermVBox.addWidget(self.console)
self.ui.show()
s = None
fpga = casperfpga.katcp_fpga.KatcpFpga(gc[np.where(gc == 'roach_ppc_ip')[0][0]][1], timeout = 120.)
#fpga = kp.getFPGA()
# UDP socket
s = socket(AF_PACKET, SOCK_RAW, htons(3))
#s.setsockopt(SOL_SOCKET,SO_RCVBUF,buf_size)
#s.bind((gc[np.where(gc=='udp_dest_device')[0][0]][1],3))
# Valon synthesizer instance
valon = valon_synth9.Synthesizer(gc[np.where(gc == 'valon_comm_port')[0][0]][1])
#valon = kp.getValon()
# Roach interface
ri = roachInterface(fpga, gc, regs, valon)
# GbE interface
udp = roachDownlink(ri, fpga, gc, regs, s, ri.accum_freq)
udp.configSocket()
# At this point in the main script we go to main_opt
# 0: Test connection to ROACH
result = kp.testConn(fpga)
# 1: Upload firmware
ri.uploadfpg()
# 2: Initialize system & UDP conn
kp.initValon(valon)
def saveTimestreamDirfile(subfolder, start_chan, end_chan, time_interval):
"""Saves a dirfile containing the I and Q values for a range of channels, streamed
over a time interval specified by time_interval
inputs:
float time_interval: Time interval to integrate over, seconds"""
# Roach PPC object
fpga = getFPGA()
# UDP socket
s = socket(AF_PACKET, SOCK_RAW, htons(3))
# Roach interface
ri = roachInterface(fpga, gc, regs, None)
# UDP object
udp = roachDownlink(ri, fpga, gc, regs, s, ri.accum_freq)
udp.configSocket()
chan_range = range(start_chan, end_chan + 1)
data_path = gc[np.where(gc == 'DIRFILE_SAVEPATH')[0][0]][1]
Npackets = int(np.ceil(time_interval * ri.accum_freq))
udp.zeroPPS()
save_path = os.path.join(data_path, subfolder)
if not os.path.exists(save_path):
os.makedirs(save_path)
filename = save_path + '/' + \
str(int(time.time())) + '-' + time.strftime('%b-%d-%Y-%H-%M-%S') + '.dir'
print filename
np.save('last_data_path.npy', filename)
# make the dirfile
d = gd.dirfile(filename, gd.CREAT | gd.RDWR | gd.UNENCODED)
# add fields
I_fields = []
Q_fields = []
for chan in chan_range:
I_fields.append('I_' + str(chan))
Q_fields.append('Q_' + str(chan))
d.add_spec('I_' + str(chan) + ' RAW FLOAT64 1')
d.add_spec('Q_' + str(chan) + ' RAW FLOAT64 1')
d.close()
d = gd.dirfile(filename, gd.RDWR | gd.UNENCODED)
nfo_I = map(lambda z: filename + "/I_" + str(z), chan_range)
nfo_Q = map(lambda z: filename + "/Q_" + str(z), chan_range)
fo_I = map(lambda z: open(z, "ab"), nfo_I)
fo_Q = map(lambda z: open(z, "ab"), nfo_Q)
fo_time = open(filename + "/time", "ab")
fo_count = open(filename + "/packet_count", "ab")
count = 0
while count < Npackets:
ts = time.time()
try:
packet, data, header, saddr = udp.parsePacketData()
if not packet:
continue
except TypeError:
continue
packet_count = (np.fromstring(packet[-4:], dtype='>I'))
idx = 0
for chan in range(start_chan, end_chan + 1):
I, Q, __ = udp.parseChanData(chan, data)
fo_I[idx].write(struct.pack('d', I))
fo_Q[idx].write(struct.pack('d', Q))
fo_I[idx].flush()
fo_Q[idx].flush()
idx += 1
fo_count.write(struct.pack('L', packet_count))
fo_count.flush()
fo_time.write(struct.pack('d', ts))
fo_time.flush()
count += 1
for idx in range(len(fo_I)):
fo_I[idx].close()
fo_Q[idx].close()
fo_time.close()
fo_count.close()
d.close()
return