###########################################################################
#
#  Copyright (c) 2019-2019 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 HD_TestScript.py
#
#  @date 19-Nov-2019
#  @author S. Nash
#
#  @brief This is an example test script for the HD.
#
############################################################################

from HemodialysisDevice import HD
from time import sleep

if __name__ == "__main__":
    hd = HD()
    sleep(2)

    if hd.Basics.CmdLogInToHD() == 0:
        exit(1)

    hd.BloodFlow.CmdBloodFlowBroadcastIntervalOverride(hd.RESET, 0)

    sleep(2)
    print("Blood Flow Target = " + str(hd.BloodFlow.TargetBloodFlowRate))
    print("Blood Pump Current= " + str(hd.BloodFlow.MeasuredBloodPumpMCCurrent))
    sleep(5)
    print("Blood Pump Current= " + str(hd.BloodFlow.MeasuredBloodPumpMCCurrent))

    hd.BloodFlow.CmdBloodPumpMeasuredCurrentOverride(hd.NO_RESET, 140)

    sleep(1)
    print("Blood Pump Current= " + str(hd.BloodFlow.MeasuredBloodPumpMCCurrent))
    sleep(5)
    hd.BloodFlow.CmdBloodPumpMeasuredCurrentOverride(hd.RESET, 0)

    while True:
        sleep(0.5)
        print("Measured Flow = " + str(hd.BloodFlow.MeasuredBloodFlowRate) + " mL/min")
    tgtRate = 0
    hd.BloodFlow.CmdBloodFlowBroadcastIntervalOverride(hd.NO_RESET, 2000)

    while True:
        if hd.BloodFlow.TargetBloodFlowRate == 0:
            if tgtRate != 0:
                hd.BloodFlow.CmdBloodFlowBroadcastIntervalOverride(hd.NO_RESET, 2000)
                tgtRate = 0
        else:
            if tgtRate == 0:
                hd.BloodFlow.CmdBloodFlowBroadcastIntervalOverride(hd.NO_RESET, 200)
                tgtRate = hd.BloodFlow.TargetBloodFlowRate

#    hd.BloodFlow.CmdBloodFlowBroadcastIntervalOverride(hd.RESET,0)