Fix MBL "Click to continue" on Color UI touchscreen (MarlinFirmware#1…

…9514)
yorickchan · Sep 26, 2020 · a7f5d38 · a7f5d38
1 parent ec4a0a3
commit a7f5d38
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/Marlin/src/lcd/menu/menu_bed_leveling.cpp b/Marlin/src/lcd/menu/menu_bed_leveling.cpp
@@ -36,6 +36,11 @@
   #include "../../module/probe.h"
 #endif
 
+#if HAS_GRAPHICAL_TFT
+  #include "../tft/touch.h"
+  #include "../tft/tft.h"
+#endif
+
 #if EITHER(PROBE_MANUALLY, MESH_BED_LEVELING)
 
   #include "../../module/motion.h"
@@ -159,7 +164,11 @@
   //         Move to the first probe position
   //
   void _lcd_level_bed_homing_done() {
-    if (ui.should_draw()) MenuItem_static::draw(1, GET_TEXT(MSG_LEVEL_BED_WAITING));
+    if (ui.should_draw()) {
+      MenuItem_static::draw(1, GET_TEXT(MSG_LEVEL_BED_WAITING));
+      // Color UI needs a control to detect a touch
+      TERN_(HAS_GRAPHICAL_TFT, touch.add_control(CLICK, 0, 0, TFT_WIDTH, TFT_HEIGHT));
+    }
     if (ui.use_click()) {
       manual_probe_index = 0;
       _lcd_level_goto_next_point();