import time
import numpy as np
from kid_readout.interactive import *
from kid_readout.equipment import hardware
from kid_readout.measurement import acquire
from kid_readout.roach import analog
ifboard = analog.HeterodyneMarkI()
setup = hardware.Hardware(ifboard)
ri = hardware_tools.r2_with_mk1()
ri.iq_delay = -1
ri.set_fft_gain(6)
#initial_f0s = np.load('/data/readout/resonances/2016-06-18-jpl-hex-271-32-high-qi-lo-1210-resonances.npy')
#initial_f0s = initial_f0s/1e6
initial_f0s = np.load(
'/data/readout/resonances/2016-09-13-JPL-8x8-HF-1_thirdcooldown_resonances.npy'
) / 1e6
initial_lo = 1010.
nf = len(initial_f0s)
atonce = 128
if nf % atonce > 0:
print "extending list of resonators to make a multiple of ", atonce
initial_f0s = np.concatenate(
(initial_f0s,
acquire.show_settings()
acquire.show_git_status()
logger = acquire.get_script_logger(__file__)
# Parameters
suffix = 'r2h11nc'
attenuation = 10
f_lo_MHz = 3000
df_lo_MHz = 10
num_tones = 4
f_baseband_MHz = np.linspace(10, 200, num_tones)
tone_sample_exponent = 15
length_seconds = 10
# Hardware
conditioner = analog.HeterodyneMarkI()
hw = hardware.Hardware(conditioner)
ri = hardware_tools.r2h11nc_with_mk1(initialize=True, use_config=False)
#ri.adc_valon.set_ref_select(0)
#ri.lo_valon.set_ref_select(1)
ri.set_dac_attenuator(attenuation)
ri.set_lo(f_lo_MHz, chan_spacing=df_lo_MHz)
ri.set_tone_baseband_freqs(f_baseband_MHz, nsamp=2**tone_sample_exponent)
# Run
npd = acquire.new_npy_directory(suffix=suffix)
tic = time.time()
try:
tools.optimize_fft_gain(ri)
npd.write(ri.get_measurement(num_seconds=length_seconds, state=hw.state()))
npd.write(ri.get_adc_measurement())
