Index: firmware/App/Modes/ModeStandby.h
===================================================================
diff -u -r66b01e138aa023829f62ddd07555d0de2013b03a -re6c60e07b450ec8d58e87bf13d45f96efab54d8b
--- firmware/App/Modes/ModeStandby.h	(.../ModeStandby.h)	(revision 66b01e138aa023829f62ddd07555d0de2013b03a)
+++ firmware/App/Modes/ModeStandby.h	(.../ModeStandby.h)	(revision e6c60e07b450ec8d58e87bf13d45f96efab54d8b)
@@ -1,17 +1,17 @@
 /**************************************************************************
 *
-* Copyright (c) 2019-2020 Diality Inc. - All Rights Reserved.
+* Copyright (c) 2019-2024 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 ModeStandby.h
+* @file    ModeStandby.h
 *
-* @author (last) Sean Nash
-* @date (last) 29-May-2020
+* @author  (last)      Dara Navaei
+* @date    (last)      23-Jul-2024
 *
-* @author (original) Dara Navaei
-* @date (original) 05-Nov-2019
+* @author  (original)  Dara Navaei
+* @date    (original)  05-Nov-2019
 *
 ***************************************************************************/
 
@@ -31,17 +31,36 @@
 
 // ********** public definitions **********
 
+/// Standby mode data structure
+typedef struct
+{
+    U32 timeout;                                        ///< Timeout.
+    U32 countdown;                                      ///< Count down.
+} STANDBY_MODE_DATA_T;
+
 // ********** public function prototypes **********
 
-void initStandbyMode( void );           // initialize this module
-void transitionToStandbyMode( void );   // prepares for transition to standby mode
-U32  execStandbyMode( void );           // execute the standby mode state machine (call from OperationModes)
+void initStandbyMode( void );                           // initialize this module
+U32 transitionToStandbyMode( void );                    // prepares for transition to standby mode
+U32  execStandbyMode( void );                           // execute the standby mode state machine (call from OperationModes)
 
-DG_STANDBY_MODE_STATE_T getCurrentStandbyState( void );  // get the current state of the standby mode.
+void waterSampleCommandHandler( SAMPLE_WATER_CMD_T sampleWaterCmd );  // handle water sample command
+BOOL requestDGStart( void );                            // HD requests DG start (go to generation idle mode)
 
-BOOL requestWaterSample( void );        // HD requests water sample
-BOOL requestDGStart( void );            // HD requests DG start (go to re-circulate mode)
+BOOL signalAbortWaterSampling( void );                  // HD signal to abort water sampling.
+BOOL startDGFlush( void );                              // HD start flush mode
+BOOL startDGHeatDisinfect( BOOL isPassive );            // HD start heat disinfect mode
+BOOL startDGHeatDisinfectActiveCool( void );
+DG_STANDBY_MODE_STATE_T getCurrentStandbyState( void ); // get the current state of the standby mode.
 
+BOOL startDGChemicalDisinfect( void );                  // HD start chemical disinfect mode
+BOOL startDGChemicalDisinfectFlush( void );             // HD start chemical disinfect flush mode
+
+BOOL startDGROPermeateSample( void );                   // HD start RO permeate sample
+
+BOOL testSetFilterFlushTimePeriodOverride( U32 value ); // set filter flush time period override.
+BOOL testResetFilterFlushTimePeriodOverride( void );    // reset filter flush time period override.
+
 /**@}*/
 
 #endif