aboutsummaryrefslogtreecommitdiff
path: root/examples/Bfield_driver.py
diff options
context:
space:
mode:
Diffstat (limited to 'examples/Bfield_driver.py')
-rw-r--r--examples/Bfield_driver.py105
1 files changed, 59 insertions, 46 deletions
diff --git a/examples/Bfield_driver.py b/examples/Bfield_driver.py
index 83f2145..362524f 100644
--- a/examples/Bfield_driver.py
+++ b/examples/Bfield_driver.py
@@ -1,6 +1,10 @@
import logging
+
# this should be done before justpy is called or log formatter does not work
-logging.basicConfig(format='%(asctime)s %(levelname)8s %(name)s: %(message)s', datefmt='%m/%d/%Y %H:%M:%S')
+logging.basicConfig(
+ format="%(asctime)s %(levelname)8s %(name)s: %(message)s",
+ datefmt="%m/%d/%Y %H:%M:%S",
+)
import pyvisa
import numpy as np
@@ -17,18 +21,24 @@ from qolab.hardware.daq import LabJackUE9
from qolab.hardware.power_supply.keysight_e3612a import KeysightE3612A
import qolab.tsdb as tsdb
-l = logging.getLogger('qolab.tsdb')
+l = logging.getLogger("qolab.tsdb")
l.setLevel(logging.INFO)
-logger = logging.getLogger('BfieldDriver')
+logger = logging.getLogger("BfieldDriver")
logger.setLevel(logging.INFO)
+
class BfieldDriver(KeysightE3612A):
- """ need to set power supply """
+ """need to set power supply"""
+
def __init__(self, *args, **kwds):
super().__init__(*args, **kwds)
- self.config['Device type'] = 'B field coil driver based on Keysight E3612A power supply'
- self.config['Device model'] = 'v0.1'
- self.deviceProperties = self.deviceProperties.union({'B', 'Bslope_TperA', 'CoilAssignment'})
+ self.config[
+ "Device type"
+ ] = "B field coil driver based on Keysight E3612A power supply"
+ self.config["Device model"] = "v0.1"
+ self.deviceProperties = self.deviceProperties.union(
+ {"B", "Bslope_TperA", "CoilAssignment"}
+ )
""""
Rough magnetic field calibration of the 3 axes coils in large magnetic shield
- Ch1: 70mA -> 650 kHz shift for delta m = 2
@@ -40,11 +50,7 @@ class BfieldDriver(KeysightE3612A):
# B response to current in a given channel
# Olivia's Calibration with flux gate magnetometer used from 2023/04/28
# +/- 0.0000007
- self._Bslope_TperA = {
- 1: 0.00065859269,
- 2: 0.00070732580,
- 3: 0.00066754994
- }
+ self._Bslope_TperA = {1: 0.00065859269, 2: 0.00070732580, 3: 0.00066754994}
"""
# Irina's Calibration used prior 2023/04/27
self._Bslope_TperA = {
@@ -60,7 +66,7 @@ class BfieldDriver(KeysightE3612A):
}
"""
# assuming that Ch1 controls Bz, Ch2 -> Bx, Ch3 -> By
- self._coil_assignment = { 'chX': 2, 'chY': 3, 'chZ': 1 }
+ self._coil_assignment = {"chX": 2, "chY": 3, "chZ": 1}
def getBslope_TperA(self):
return self._Bslope_TperA
@@ -71,9 +77,9 @@ class BfieldDriver(KeysightE3612A):
def getB(self):
Bslope_TperA = self.getBslope_TperA()
coil_assignment = self.getCoilAssignment()
- chX = coil_assignment['chX']
- chY = coil_assignment['chY']
- chZ = coil_assignment['chZ']
+ chX = coil_assignment["chX"]
+ chY = coil_assignment["chY"]
+ chZ = coil_assignment["chZ"]
Ix = self.getChanIout(chX)
Iy = self.getChanIout(chY)
@@ -81,40 +87,43 @@ class BfieldDriver(KeysightE3612A):
Bx = Ix * Bslope_TperA[chX]
By = Iy * Bslope_TperA[chY]
Bz = Iz * Bslope_TperA[chZ]
- Bmag = float(np.sqrt(Bx*Bx + By*By + Bz*Bz))
- theta = float(np.arccos(Bz/Bmag))
- phi = float(np.arctan2(By, Bx))
- theta_degree = float(theta/np.pi*180)
- phi_degree = float(phi/np.pi*180)
- return { 'Bmag': Bmag,
- 'theta': theta, 'phi': phi,
- 'theta_degree': theta_degree, 'phi_degree': phi_degree,
- 'Bx': float(Bx),
- 'By': float(By),
- 'Bz': float(Bz),
- }
+ Bmag = float(np.sqrt(Bx * Bx + By * By + Bz * Bz))
+ theta = float(np.arccos(Bz / Bmag))
+ phi = float(np.arctan2(By, Bx))
+ theta_degree = float(theta / np.pi * 180)
+ phi_degree = float(phi / np.pi * 180)
+ return {
+ "Bmag": Bmag,
+ "theta": theta,
+ "phi": phi,
+ "theta_degree": theta_degree,
+ "phi_degree": phi_degree,
+ "Bx": float(Bx),
+ "By": float(By),
+ "Bz": float(Bz),
+ }
def setB(self, Bmag=50e-6, theta=0, phi=0):
- """ Sets B field currents based on B (in T) and angles theta, and phi """
+ """Sets B field currents based on B (in T) and angles theta, and phi"""
self._Bmag = Bmag
self._theta = theta
self._phi = phi
- Bx = Bmag*np.sin(theta)*np.cos(phi)
- By = Bmag*np.sin(theta)*np.sin(phi)
- Bz = Bmag*np.cos(theta)
+ Bx = Bmag * np.sin(theta) * np.cos(phi)
+ By = Bmag * np.sin(theta) * np.sin(phi)
+ Bz = Bmag * np.cos(theta)
Bslope_TperA = self.getBslope_TperA()
coil_assignment = self.getCoilAssignment()
- chX = coil_assignment['chX']
- chY = coil_assignment['chY']
- chZ = coil_assignment['chZ']
+ chX = coil_assignment["chX"]
+ chY = coil_assignment["chY"]
+ chZ = coil_assignment["chZ"]
Ix = Bx / Bslope_TperA[chX]
Iy = By / Bslope_TperA[chY]
Iz = Bz / Bslope_TperA[chZ]
- logger.info(f'Setting {Bmag=}, {theta=} {phi=} in radians')
+ logger.info(f"Setting {Bmag=}, {theta=} {phi=} in radians")
logger.info(f"Setting {chX=} to {Ix}")
logger.info(f"Setting {chY=} to {Iy}")
logger.info(f"Setting {chZ=} to {Iz}")
@@ -124,20 +133,24 @@ class BfieldDriver(KeysightE3612A):
return Ix, Iy, Iz
def setBinDegrees(self, Bmag=50e-6, theta=0, phi=0):
- logger.info(f'Setting {Bmag=}, {theta=} {phi=} in degrees')
- return self.setB(Bmag=Bmag, theta=theta/180*np.pi, phi=phi/180*np.pi)
+ logger.info(f"Setting {Bmag=}, {theta=} {phi=} in degrees")
+ return self.setB(Bmag=Bmag, theta=theta / 180 * np.pi, phi=phi / 180 * np.pi)
+
# TSDB logger setting
-tsdb_client = tsdb.Client('influx', 'http://qo.physics.wm.edu:8428', database='qolab')
-tsdb_ingester = tsdb.Ingester(tsdb_client, batch=11, measurement_prefix='VAMPIRE.VCSEL')
+tsdb_client = tsdb.Client("influx", "http://qo.physics.wm.edu:8428", database="qolab")
+tsdb_ingester = tsdb.Ingester(tsdb_client, batch=11, measurement_prefix="VAMPIRE.VCSEL")
logger.info("Accessing hardware")
rm = pyvisa.ResourceManager()
-instr=rm.open_resource('USB0::10893::4354::MY61001869::0::INSTR')
-app_nickname = 'BfieldDriver'
-#Bfield = BfieldDriver(instr, device_nickname='.'.join([app_nickname, 'b_field_driver']), tsdb_ingester=tsdb_ingester)
-Bfield = BfieldDriver(instr, device_nickname='.'.join([app_nickname, 'b_field_driver']), tsdb_ingester=None)
-
+instr = rm.open_resource("USB0::10893::4354::MY61001869::0::INSTR")
+app_nickname = "BfieldDriver"
+# Bfield = BfieldDriver(instr, device_nickname='.'.join([app_nickname, 'b_field_driver']), tsdb_ingester=tsdb_ingester)
+Bfield = BfieldDriver(
+ instr,
+ device_nickname=".".join([app_nickname, "b_field_driver"]),
+ tsdb_ingester=None,
+)
-print('Use me as : Bfield.setBinDegrees(theta=Angle, phi=Angle)')
+print("Use me as : Bfield.setBinDegrees(theta=Angle, phi=Angle)")
generated by cgit v1.2.3 (git 2.39.1) at 2024-11-16 13:59:49 +0000