Index: firmware/App/Modes/ModePreTreat.h
===================================================================
diff -u -rc137d3c7cb17b0364d745e10ff6dbd1901eb1baa -rcd5be724d5a3ba7457e761191d82f278654d7f5c
--- firmware/App/Modes/ModePreTreat.h	(.../ModePreTreat.h)	(revision c137d3c7cb17b0364d745e10ff6dbd1901eb1baa)
+++ firmware/App/Modes/ModePreTreat.h	(.../ModePreTreat.h)	(revision cd5be724d5a3ba7457e761191d82f278654d7f5c)
@@ -1,14 +1,14 @@
 /**************************************************************************
 *
-* Copyright (c) 2019-2021 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    ModePreTreat.h
 *
 * @author  (last)      Sean Nash
-* @date    (last)      11-Aug-2021
+* @date    (last)      01-Sep-2023
 *
 * @author  (original)  Dara Navaei
 * @date    (original)  05-Nov-2019
@@ -37,13 +37,15 @@
 {
     U32 preTreatmentSubMode;        ///< Current pre-treatment sub-mode
     U32 sampleWaterState;           ///< Current sample water state
-    U32 consumableSelfTestsState;   ///< Current consumable sefl-tests state
+    U32 consumableSelfTestsState;   ///< Current consumable self-tests state
     U32 noCartSelfTestsState;       ///< Current no cartridge self-tests state
     U32 installState;               ///< Current disposable installation state
     U32 drySelfTestsState;          ///< Current dry self-tests state
     U32 primeState;                 ///< Current prime state
     U32 recircState;                ///< Current re-circulate state
     U32 patientConnectionState;     ///< Current patient connection state
+    U32 wetSelfTestsState;          ///< CUrrent wet self-tests state
+    U32 preTreatmentRsrvrState;     ///< Current pre-treatment reservoir state
 } PRE_TREATMENT_STATE_DATA_T;
 
 // ********** public function prototypes **********
@@ -57,10 +59,14 @@
 void setUserSetUFVolumeStatus( BOOL status );                       // Set status that indicates UF volume setting has been set
 void signalUserConfirmPatientConnection( void );                    // Signal that user confirms patient connection has been completed
 void signalUserStartTreatment( void );                              // Signal that user requests to start treatment
+void signalAllowDGFlushFills( void );                               // Signal that indicates the HD may begin flush fills at DG
+void signalAllowDGFillRes1( void );                                 // Signal that indicates the HD may begin filling reservoir 1 (after flushes completed)
+void signalAllowDGFillRes2( void );                                 // Signal that indicates the HD may begin filling reservoir 2 (after flushes completed)
 
 void signalAlarmActionToPreTreatmentMode( ALARM_ACTION_T action );  // Execute alarm action as appropriate for pre-treatment mode
 
-BOOL getReservoirFillStatus( DG_RESERVOIR_ID_T reservoirID );
+BOOL getReservoirFillStatus( DG_RESERVOIR_ID_T reservoirID );       // Get the fill complete status for given reservoir
+HD_PRE_TREATMENT_MODE_STATE_T getPreTreatmentSubState( void );      // Get the current pre-treatment sub mode.
 
 BOOL testSetPreTreatmentModePublishIntervalOverride( U32 ms );
 BOOL testResetPreTreatmentModePublishIntervalOverride( void );