Index: firmware/App/Modes/ModeHeatDisinfect.h
===================================================================
diff -u -r66b01e138aa023829f62ddd07555d0de2013b03a -rfd84d51bd0177b49578859572d24f68571280ea4
--- firmware/App/Modes/ModeHeatDisinfect.h	(.../ModeHeatDisinfect.h)	(revision 66b01e138aa023829f62ddd07555d0de2013b03a)
+++ firmware/App/Modes/ModeHeatDisinfect.h	(.../ModeHeatDisinfect.h)	(revision fd84d51bd0177b49578859572d24f68571280ea4)
@@ -1,17 +1,17 @@
 /**************************************************************************
 *
-* Copyright (c) 2019-2020 Diality Inc. - All Rights Reserved.
+* Copyright (c) 2020-2023 Diality Inc. - All Rights Reserved.
 *
 * THIS CODE MAY NOT BE COPIED OR REPRODUCED IN ANY FORM, IN PART OR IN
 * WHOLE, WITHOUT THE EXPLICIT PERMISSION OF THE COPYRIGHT OWNER.
 *
-* @file ModeHeatDisinfect.h
+* @file    ModeHeatDisinfect.h
 *
-* @author (last) Sean Nash
-* @date (last) 29-May-2020
+* @author  (last)      Michael Garthwaite
+* @date    (last)      29-Aug-2023
 *
-* @author (original) Sean
-* @date (original) 20-Apr-2020
+* @author  (original)  Sean
+* @date    (original)  20-Apr-2020
 *
 ***************************************************************************/
 
@@ -23,22 +23,42 @@
 
 /**
  *  @defgroup DGHeatDisinfectMode DGHeatDisinfectMode
- *  @brief Heat disinfection mode module.  Manages the state machine for the heat disinfection mode.
+ *  @brief Heat disinfection mode module. Manages the state machine for the heat disinfect mode.
  *
  *  @addtogroup DGHeatDisinfectMode
  *  @{
  */
 
 // ********** public definitions **********
 
+/// Heat disinfect UI data publish struct
+typedef struct
+{
+    U32 heatDisinfectTargetTime;        ///< Target time to heat disinfect.
+    U32 ro77CountdownTimeS;             ///< Heat disinfect RO 77 count down time in seconds.
+    U32 ro82CountdownTimeS;             ///< Heat disinfect RO 82 count down time in seconds.
+    U32 r77CountdownTimeS;              ///< Heat disinfect reservoir 77 count down time in seconds.
+    U32 r82CountdownTimeS;              ///< Heat disinfect reservoir 82 count down time in seconds.
+} MODE_HEAT_DISINFECT_UI_DATA_T;
+
 // ********** public function prototypes **********
 
-void initHeatDisinfectMode( void );           // initialize this module
-void transitionToHeatDisinfectMode( void );   // prepares for transition to heat disinfect mode
-U32  execHeatDisinfectMode( void );           // execute the heat disinfect mode state machine (call from OperationModes)
+void initHeatDisinfectMode( void );                                 // Initialize this module
+U32 transitionToHeatDisinfectMode( void );                          // Prepares for transition to heat disinfect mode
+U32  execHeatDisinfectMode( void );                                 // Execute the heat disinfect mode state machine (call from OperationModes)
 
-DG_HEAT_DISINFECT_STATE_T getCurrentHeatDisinfectState( void );  // get the current state of the heat disinfect mode.
+DG_HEAT_DISINFECT_STATE_T getCurrentHeatDisinfectState( void );     // get the current state of the heat disinfect mode
 
+BOOL stopDGHeatDisinfect( void );                                   // Stop DG heat disinfect
+BOOL testSetDG77CStateTimerOverride( U32 timer );                   // set override timer for disinfection at 77C
+BOOL testResetDG77CStateTimerOverride( void );                      // reset override timer for disinfection at 77C
+BOOL testSetDG82CStateTimerOverride( U32 timer );                   // set override timer for disinfection at 82C
+BOOL testResetDG82CStateTimerOverride( void );                      // reset override timer for disinfection at 77C
+
+#ifndef _RELEASE_
+void setHeatNelsonSupportMode( NELSON_SUPPORT_T support );
+#endif
+
 /**@}*/
 
 #endif