From 894549d6c1e1edf1b0a5a896bd2a00972eebacc1 Mon Sep 17 00:00:00 2001
From: Cameron Devine <cameron.devine@alleninstitute.org>
Date: Sun, 13 Oct 2024 22:56:59 -0700
Subject: [PATCH] adding some more information to the scope status message, and
 the ability to run more setup functions

---
 TEM_comms/scope.py | 1 +
 TEM_comms/ui.py    | 4 ++++
 2 files changed, 5 insertions(+)

diff --git a/TEM_comms/scope.py b/TEM_comms/scope.py
index d608641..fabc73f 100644
--- a/TEM_comms/scope.py
+++ b/TEM_comms/scope.py
@@ -25,3 +25,4 @@ class Status(BaseMessage):
     tank_voltage: int
     brightness: int
     beam_offset: Tuple[int, int]
+    screen: Literal["up", "down"] | None
diff --git a/TEM_comms/ui.py b/TEM_comms/ui.py
index 4569ca9..6077351 100644
--- a/TEM_comms/ui.py
+++ b/TEM_comms/ui.py
@@ -40,3 +40,7 @@ class Setup(BaseMessage):
     lens_correction: bool = False
     acquire_brightfield: bool = False
     acquire_darkfield: bool = False
+    center_beam: bool = False
+    spread_beam: bool = False
+    find_aperture: bool = False
+    grid: Optional[int] = None