class PointsTab(BaseWidget):
"""
A Tab for a list of points
"""
_axis = []
_update_function = None
_controller = None
_sensor = None
_lightsource = None
_saved_points = None
def __init__(self, update_function=None):
super().__init__("Points")
self._update_function = update_function
self._open_file = ControlFile(
label="Points File: "
)
self._open_file.changed_event = self._on_open_file
self._save_file = ControlButton(
label="Save"
)
self._save_file.value = self._save_points
self._points_list = ControlList(
label="Points",
add_function=self._add_point,
remove_function=self._remove_point,
auto_resize=False
)
self._points_list.data_changed_event = self._change_point
self._points_list.horizontal_headers = [
axis.get_name() for axis in self._axis]
self._pre_delay_time = ControlNumber(
label="Pre Delay Time (s)",
default=1,
minimum=0,
maximum=float('inf'),
decimals=5
)
self._measure_time = ControlNumber(
label="Measure Time (s)",
default=5,
minimum=0,
maximum=float('inf'),
decimals=5
)
self._post_delay_time = ControlNumber(
label="Post Delay Time (s)",
default=1,
minimum=0,
maximum=float('inf'),
decimals=5
)
self._scan_frequency = ControlNumber(
label="Scan Frequency (Hz)",
default=10,
minimum=0,
maximum=float('inf'),
decimals=5
)
self._out_file = ControlDir(
label="Output Folder: "
)
self._scan_button = ControlButton(
label="Scan"
)
self._scan_button.value = self._begin_scan
self.formset = [
('_open_file', '_save_file'),
'_points_list',
('_pre_delay_time', '_post_delay_time'),
('_measure_time', '_scan_frequency'),
'_out_file',
'_scan_button',
]
def update_events(self, events):
"""
Updates all events
"""
# Update the points with the most recent axis
if 'axis' in events:
self._axis = events['axis']
self._update_lists()
if 'sensor' in events:
self._sensor = events['sensor']
if 'lightsource' in events:
self._lightsource = events['lightsource']
if 'scan' in events:
state = events['scan'][0]
if state == AxisControllerState.DONE:
self._scan_button.label = "Scan"
else:
self._scan_button.label = "Stop"
if 'saved_points' in events:
self._saved_points = events['saved_points']
def _update_lists(self):
self._points_list.clear()
self._points_list.horizontal_headers = [
axis.get_name() for axis in self._axis]
self._points_list.resize_rows_contents()
length = self._max_axis_len()
for i in range(0, length):
point = []
for axis in self._axis:
assert isinstance(axis, ControlAxis)
# Add enough extra points to the axis so it matches the rest of the axis if it is missing points
if len(axis.points) <= i:
for j in range(len(axis.points) - 1, i):
axis.points.append(str(axis.get_min()))
point.append(str(axis.points[i]))
self._points_list += point
self._points_list.resize_rows_contents()
def _max_axis_len(self):
"""
Get the number of points in the axis that has the most points
"""
max_len = 0
for axis in self._axis:
assert isinstance(axis, ControlAxis)
if len(axis.points) > max_len:
max_len = len(axis.points)
return max_len
def _add_point(self):
for axis in self._axis:
assert isinstance(axis, ControlAxis)
axis.points.append("0.0")
self._update_lists()
def _remove_point(self):
index = self._points_list.selected_row_index
if index is not None:
for axis in self._axis:
assert isinstance(axis, ControlAxis)
axis.points.pop(index)
self._update_lists()
def _change_point(self, row, col, item):
if len(self._axis) > col and len(self._axis[col].points) > row:
try:
self._axis[col].points[row] = float(item)
except ValueError:
self._axis[col].points[row] = item
def _begin_scan(self):
"""
Create an AxisController and begin scanning
"""
if self._controller is None or self._controller.get_state() == AxisControllerState.DONE:
self._controller = AxisController(
self._axis, self._sensor, self._lightsource, self._pre_delay_time.value, self._measure_time.value,
self._post_delay_time.value, self._scan_frequency.value, self._saved_points, self._out_file.value, self._update_function)
self._controller.begin()
else:
self._controller.stop()
def _on_open_file(self):
"""
Open a csv file and rearange columns accoring to header
"""
print("Opening File:", self._open_file.value)
if self._open_file.value is not None and self._open_file.value != '':
with open(self._open_file.value, newline='') as csvfile:
try:
csvreader = csv.reader(csvfile)
self._parse_file(csvreader)
except:
print("Failed to read file")
def _parse_file(self, csvreader):
for axis in self._axis:
if isinstance(axis, ControlAxis):
axis.points.clear()
self._points_list.clear()
points = []
for row in csvreader:
for index, data in enumerate(row):
if len(points) <= index:
points.append([])
points[index].append(data)
#print(points)
for points_list in points:
print(points_list[0])
if isinstance(points_list[0], str):
for axis in self._axis:
if points_list[0] == axis.get_name():
axis.points.clear()
for point in points_list[1:]:
try:
axis.points.append(float(point))
except ValueError:
axis.points.append(point)
#print(axis.get_name(), axis.points)
self._update_lists()
def _save_points(self):
points_file = QFileDialog.getSaveFileName(
caption="Save Points", filter='CSV Files (*.csv)')
if points_file[0] is not None and points_file[0] != '':
out_points = []
headers = []
for axis in self._axis:
headers.append(axis.get_name())
out_points.append(headers)
length = self._max_axis_len()
for i in range(0, length):
point = []
for axis in self._axis:
assert isinstance(axis, ControlAxis)
# Add enough extra points to the axis so it matches the rest of the axis if it is missing points
if len(axis.points) <= i:
for j in range(len(axis.points) - 1, i):
axis.points.append(str(axis.get_min()))
point.append(str(axis.points[i]))
out_points.append(point)
with open(points_file[0], 'w', newline='') as csvfile:
csvwriter = csv.writer(csvfile, quoting=csv.QUOTE_MINIMAL)
csvwriter.writerows(out_points)
self._open_file.value = points_file[0]
