forked from zietzm/Helmholtz_Test_Bench
Moved CSV save function.
This commit is contained in:
+4
-13
@@ -27,7 +27,7 @@ import src.config_handling as config
|
||||
import src.csv_logging as log
|
||||
from src.calibration import AmbientFieldCalibration, CoilConstantCalibration, MagnetometerCalibration
|
||||
from src.exceptions import DeviceAccessError
|
||||
from src.utility import ui_print
|
||||
from src.utility import ui_print, save_dict_list_to_csv
|
||||
import src.helmholtz_cage_device as helmholtz_cage_device
|
||||
|
||||
# define font styles:
|
||||
@@ -845,7 +845,7 @@ class CalibrateAmbientField(Frame):
|
||||
ui_print("Error: Failed to export non-existent calibration data.")
|
||||
return
|
||||
|
||||
self.save_to_csv('coil_constant_calibration.csv', self.coil_constant_raw_data)
|
||||
save_dict_list_to_csv('coil_constant_calibration.csv', self.coil_constant_raw_data, query_path=True)
|
||||
ui_print("Saved calibration results to coil_constant_calibration.csv.")
|
||||
|
||||
def save_to_csv_ambient_field(self):
|
||||
@@ -853,19 +853,9 @@ class CalibrateAmbientField(Frame):
|
||||
ui_print("Error: Failed to export non-existent calibration data.")
|
||||
return
|
||||
|
||||
self.save_to_csv('ambient_field_calibration.csv', self.ambient_field_raw_data)
|
||||
save_dict_list_to_csv('ambient_field_calibration.csv', self.ambient_field_raw_data, query_path=True)
|
||||
ui_print("Saved calibration results to ambient_field_calibration.csv.")
|
||||
|
||||
def save_to_csv(self, filename, data):
|
||||
with open(filename, mode='w', newline='') as csv_file:
|
||||
fieldnames = data[0].keys()
|
||||
csv_writer = csv.DictWriter(csv_file, fieldnames=fieldnames, delimiter=',', quotechar='"',
|
||||
quoting=csv.QUOTE_NONNUMERIC)
|
||||
|
||||
csv_writer.writeheader()
|
||||
for row in data:
|
||||
csv_writer.writerow(row)
|
||||
|
||||
def save_and_apply_ambient_calibration(self):
|
||||
if self.ambient_field_result is not None:
|
||||
ui_print("Saving ambient field calibration data")
|
||||
@@ -886,6 +876,7 @@ class CalibrateAmbientField(Frame):
|
||||
ui_print("Reinitializing devices...")
|
||||
g.CAGE_DEVICE.reconnect_hardware_async() # setup everything with the defaults
|
||||
|
||||
|
||||
class CalibrateMagnetometer(Frame):
|
||||
def __init__(self, parent, controller):
|
||||
Frame.__init__(self, parent)
|
||||
|
||||
Reference in New Issue
Block a user