import numpy as np

from kid_readout.interactive import *
from kid_readout.measurement import acquire
from kid_readout.roach import r2heterodyne, attenuator, hardware_tools

from equipment.custom import mmwave_source
from equipment.hittite import signal_generator
from equipment.srs import lockin
from kid_readout.equipment.agilent_33220 import FunctionGenerator

logger.setLevel(logging.DEBUG)

fg = FunctionGenerator()
fg.set_load_ohms(1e6)

hittite = signal_generator.Hittite(ipaddr='192.168.0.200')
hittite.set_power(0)
hittite.on()
hittite.set_freq(148e9 / 12.)

lockin = lockin.Lockin(LOCKIN_SERIAL_PORT)
tic = time.time()
print lockin.identification
print time.time() - tic
tic = time.time()
print lockin.fast_state
print time.time() - tic
source = mmwave_source.MMWaveSource()
source.set_attenuator_turns(3.0, 3.0)