From f2615b7972d514ca90f615e1d7b0fa2f08a780f8 Mon Sep 17 00:00:00 2001
From: Knutwurst <36196269+knutwurst@users.noreply.github.com>
Date: Wed, 17 Nov 2021 16:56:10 +0100
Subject: [PATCH] Add missing Touchscreen implementation to M104 and M109
 temperature controller.

---
 Marlin/src/gcode/temp/M104_M109.cpp | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/Marlin/src/gcode/temp/M104_M109.cpp b/Marlin/src/gcode/temp/M104_M109.cpp
index efda04de..2a91a6b5 100644
--- a/Marlin/src/gcode/temp/M104_M109.cpp
+++ b/Marlin/src/gcode/temp/M104_M109.cpp
@@ -38,6 +38,10 @@
 
 #include "../../MarlinCore.h" // for startOrResumeJob, etc.
 
+#ifdef ANYCUBIC_TOUCHSCREEN
+  #include "../../lcd/anycubic_touchscreen.h"
+#endif
+
 #if ENABLED(PRINTJOB_TIMER_AUTOSTART)
   #include "../../module/printcounter.h"
   #if ENABLED(CANCEL_OBJECTS)
@@ -129,10 +133,22 @@ void GcodeSuite::M104_M109(const bool isM109) {
       thermalManager.set_heating_message(target_extruder);
   }
 
+  #ifdef ANYCUBIC_TOUCHSCREEN
+    AnycubicTouchscreen.HeatingStart();
+  #endif
+
   TERN_(AUTOTEMP, planner.autotemp_M104_M109());
 
   if (isM109 && got_temp)
     (void)thermalManager.wait_for_hotend(target_extruder, no_wait_for_cooling);
+
+  // flush the serial buffer after heating to prevent lockup by m105
+  SERIAL_FLUSH();
+
+  #ifdef ANYCUBIC_TOUCHSCREEN
+    AnycubicTouchscreen.CommandScan();
+    AnycubicTouchscreen.BedHeatingDone();
+  #endif
 }
 
 #endif // EXTRUDERS