Add M3FS in GUI

Ledger-Donjon · Jun 20, 2024 · 470610a · 470610a
1 parent ef69aac
commit 470610a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pystages/gui/gui.py b/pystages/gui/gui.py
@@ -17,6 +17,7 @@
 from ..corvus import Corvus
 from ..smc100 import SMC100
 from ..stage import Stage
+from ..m3fs import M3FS
 from enum import Enum
 from serial.tools.list_ports import comports
 from serial.tools.list_ports_common import ListPortInfo
@@ -26,6 +27,7 @@ class StageType(str, Enum):
     CNC = "CNC"
     Corvus = "Corvus"
     SMC = "SMC100"
+    M3FS = "M3FS"
 
 
 class StageWindow(QWidget):
@@ -46,6 +48,8 @@ def connect(self, on_off):
                 self.stage = Corvus(dev)
             elif selected == StageType.SMC:
                 self.stage = SMC100(dev, [1, 2])
+            elif selected == StageType.M3FS:
+                self.stage = M3FS(dev, baudrate=115200)
             self.position_timer.start(100)
 
         else:
@@ -83,7 +87,7 @@ def __init__(self):
         w = QLabel("Stage Selection")
         box.addWidget(w)
         self.stage_selection = w = QComboBox()
-        w.addItems([StageType.CNC, StageType.Corvus, StageType.SMC])
+        w.addItems([StageType.CNC, StageType.Corvus, StageType.SMC, StageType.M3FS])
         box.addWidget(w)
         self.port_selection = w = QComboBox()
         w.addItem("Auto detection", None)