Experimental new menu system

Author: WesleyJ-128

FLLMaster/Main.py

@@ -9,62 +9,61 @@
 calibrate()
 robot.reflectCal()
 
-def left(state):
-    if state:
-        if not mission.is_alive():
-            minusCount()
-            display()
-        else:
-                abort()
-                minusCount()
-                display()
-def right(state):
-    if state:
-        if not mission.is_alive():
-            addCount()
-            display()
-        else:
-            abort()
-            minusCount()
-            display()
-def down(state):
-    if state:
-        if not mission.is_alive():
-            calibrate()
-        else:
-            abort()
-            minusCount()
-            display()
-def up(state):
-    if state:
-        if not mission.is_alive():
-            robot.reflectCal()
-        else:
-            abort()
-            minusCount()
-            display()
-def enter(state):
-    if state:
-        if not mission.is_alive():
-            run()
-            addCount()
-            display()
-        else:
-            abort()
-            minusCount()
-            display()
-def backspace(state):
-    pass
+def left():
+    robot.btn.wait_for_released('left, right, up, down, enter')
+    if not mission.is_alive():
+        minusCount()
+        display()
+    else:
+        abort()
+        minusCount()
+        display()
+def right():
+    robot.btn.wait_for_released('left, right, up, down, enter')
+    if not mission.is_alive():
+        addCount()
+        display()
+    else:
+        abort()
+        minusCount()
+        display()
+def down():
+    robot.btn.wait_for_released('left, right, up, down, enter')
+    if not mission.is_alive():
+        calibrate()
+    else:
+        abort()
+        minusCount()
+        display()
+def up():
+    robot.btn.wait_for_released('left, right, up, down, enter')
+    if not mission.is_alive():
+        robot.reflectCal()
+    else:
+        abort()
+        minusCount()
+        display()
+def enter():
+    robot.btn.wait_for_released('left, right, up, down, enter')
+    if not mission.is_alive():
+        run()
+        addCount()
+        display()
+    else:
+        abort()
+        minusCount()
+        display()
 
-robot.btn.on_left = left
-robot.btn.on_right = right
-robot.btn.on_up = up
-robot.btn.on_down = down
-robot.btn.on_enter = enter
-robot.btn.on_backspace = backspace      
+buttonMap = {
+    'left': left(),
+    'right': right(),
+    'enter': enter(),
+    'up': up(),
+    'down': down()
+}
 
-while True:    
-    robot.btn.process()
-    loopIndex = (loopIndex + 1) % 100
+while True:
+    buttonMap[robot.btn.buttons_pressed]()
+    checkDrift()
     displaySensor()
-    checkDrift()
+    loopIndex = (loopIndex + 1) % 100