B field sweep improved

1 files changed, 9 insertions, 4 deletions

diff --git a/examples/eit_with_vcsel.py b/examples/eit_with_vcsel.py
index 25e87fe..77104f2 100644
--- a/examples/eit_with_vcsel.py
+++ b/examples/eit_with_vcsel.py
@@ -7,6 +7,7 @@ import numpy as np
 import platform
 import time
 from tqdm import tqdm as pbar
+import matplotlib.pyplot as plt
 
 
 from qolab.hardware import BasicInstrument
@@ -211,17 +212,21 @@ def calibrateCoilsCurrent():
     getCurrentCalibrationData(ch=3)
 
 def rotateBandGetEITtrace():
-    Np=100
-    Nsweeps=1
+    Np=1000
+    Nsweeps=5
     Bmag=50e-6 # earth magnetic field in Tesla (0.5 G)
-    phiStep = 5
-    thetaStep = 5
+    phiStep = 10
+    thetaStep = 10
     phiSet = range(0,90+phiStep, phiStep)
     thetaSet = range(0,90+thetaStep, thetaStep)
     for phi in phiSet:
         for theta in thetaSet:
             Bfield.setBinDegrees(Bmag=Bmag, theta=theta, phi=phi)
             trEIT = eitSweep(central_frequency, frequency_span, Np, Nsweeps=Nsweeps)
+            plt.clf()
+            trEIT.plot()
+            plt.draw()
+            plt.pause(0.1)
             fn = apparatus.getNextDataFile()
             logger.info(f'Data ready for {Bmag=} in {theta=} {phi=}')
             logger.info(f'Data saved to {fn=}')