diff options
| -rw-r--r-- | examples/lock_eit.py | 79 |
1 files changed, 79 insertions, 0 deletions
diff --git a/examples/lock_eit.py b/examples/lock_eit.py new file mode 100644 index 0000000..90732d8 --- /dev/null +++ b/examples/lock_eit.py @@ -0,0 +1,79 @@ +import pyvisa +import numpy as np +import matplotlib as mpl +import matplotlib.pyplot as plt +import time + +from qolab.hardware.scope import SDS1104X +from qolab.hardware.rf_generator import AgilentE8257D +from qolab.feedback import PID +from qolab.data import TraceSetSameX, TraceXY, Trace + +def adjustRF(pid, scope, rfgen): + errorTrace = Trace("error") + errorTrace.config['unit']='V' + timeTrace = Trace("time") + timeTrace.config['unit']='S' + + errorLog = TraceXY("error") + errorLog.x = timeTrace + errorLog.y = errorTrace + + log = TraceSetSameX("timelog") + log.addTrace(errorLog) + + + # while True: + for i in range(0,100): + timenow = time.time() + + ch1=scope.getTrace(1, decimate=False) + err = np.mean(ch1.y.values) + fdbck = pid.feedback(err) + + freq0 = rfgen.getFreqFixed() + freq = freq0 + fdbck + rfgen.setFreqFixed(freq) + print(f'error = {err}, feedback = {fdbck}, request freq = {freq}') + + log.addPointToTrace(timenow) + log.addPointToTrace(err, "error") + plt.clf() + log.plot() + time.sleep(.01) + + return log + + + +if __name__ == '__main__': + print("testing") + rm = pyvisa.ResourceManager() + instr_scope=rm.open_resource('TCPIP::192.168.0.61::INSTR') + scope = SDS1104X(instr_scope) + instr_rfgen=rm.open_resource('TCPIP::192.168.0.114::INSTR') + rfgen = AgilentE8257D(instr_rfgen) + print('------ Header start -------------') + print(str.join('\n', scope.getHeader())) + print(str.join('\n', rfgen.getHeader())) + print('------ Header ends -------------') + ch1 = scope.getTrace(1) + traces = scope.getAllTraces() + + freqZero = 6834687190 + freqDeltaM2 = 6835387164 + SweepSpan = 10000 + + rfgen.stopFrequencySweep() + rfgen.setFreqFixed(freqDeltaM2+100) + + pid = PID(100,400,0, sign=-1) + log=adjustRF(pid, scope, rfgen) + + + + + + + + |