Index: dialin/common/hd_defs.py
===================================================================
diff -u -ra505f80795643e531cb0b0120a38a7765cbf7b9c -ra5d64dff71ebee498a2b993640f13b71c4286fea
--- dialin/common/hd_defs.py	(.../hd_defs.py)	(revision a505f80795643e531cb0b0120a38a7765cbf7b9c)
+++ dialin/common/hd_defs.py	(.../hd_defs.py)	(revision a5d64dff71ebee498a2b993640f13b71c4286fea)
@@ -273,3 +273,60 @@
     STANDBY_DG_HEAT_DISINFECT_IN_PROGRESS_STATE = 4     # DG heat disinfection operation is in progress
     STANDBY_DG_CHEM_DISINFECT_IN_PROGRESS_STATE = 5     # DG chemical disinfection operation is in progress
     NUM_OF_STANDBY_STATES = 6                           # Number of standby states (sub-modes)
+
+
+@unique
+class TreatmentStates(DialinEnum):
+    TREATMENT_START_STATE = 0
+    TREATMENT_BLOOD_PRIME_STATE = 1
+    TREATMENT_DIALYSIS_STATE = 2
+    TREATMENT_STOP_STATE = 3
+    TREATMENT_RINSEBACK_STATE = 4
+    TREATMENT_RECIRC_STATE = 5
+    TREATMENT_END_STATE = 6
+    NUM_OF_TREATMENT_STATES = 7
+
+
+@unique
+class TreatmentBloodPrimeStates(DialinEnum):
+    BLOOD_PRIME_RAMP_STATE = 0
+    NUM_OF_BLOOD_PRIME_STATES = 1
+
+
+@unique
+class TreatmentDialysisStates(DialinEnum):
+    DIALYSIS_START_STATE = 0
+    DIALYSIS_UF_STATE = 1
+    DIALYSIS_SALINE_BOLUS_STATE = 2
+    NUM_OF_DIALYSIS_STATES = 3
+
+
+@unique
+class TreatmentStopStates(DialinEnum):
+    TREATMENT_STOP_RECIRC_STATE = 0
+    TREATMENT_STOP_NO_RECIRC_STATE = 1
+    NUM_OF_TREATMENT_STOP_STATES = 2
+
+
+@unique
+class TreatmentRinsebackStates(DialinEnum):
+    RINSEBACK_STOP_INIT_STATE = 0
+    RINSEBACK_RUN_STATE = 1
+    RINSEBACK_PAUSED_STATE = 2
+    RINSEBACK_STOP_STATE = 3
+    RINSEBACK_RUN_ADDITIONAL_STATE = 4
+    NUM_OF_RINSEBACK_STATES = 5
+
+
+@unique
+class TreatmentRecircStates(DialinEnum):
+    TREATMENT_RECIRC_RECIRC_STATE = 0
+    TREATMENT_RECIRC_STOPPED_STATE = 1
+    NUM_OF_TREATMENT_RECIRC_STATES = 2
+
+
+@unique
+class TreatmentEndStates(DialinEnum):
+    TREATMENT_END_WAIT_FOR_RINSEBACK_STATE = 0
+    TREATMENT_END_PAUSED_STATE = 1
+    NUM_OF_TREATMENT_END_STATES = 2