###########################################################################
#
#  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 test_demo.py
#
#  @date 15-May-2020
#  @author P. Lucia
#
#  @brief This is a demo script to exercise the HD and DG.
#
############################################################################
import sys
sys.path.append("..")
from dialin.dg.dialysate_generator import DG
from dialin.hd.hemodialysis_device import HD
from dialin.hd.constants import RESET, NO_RESET, BUTTON_PRESSED, BUTTON_RELEASED
import time


def test_dg_version():
    """
    Prints the DG version.

    @return: None
    """
    dg = DG()
    if dg.cmd_log_in_to_dg():
        dg.cmd_ui_request_dg_version()
        time.sleep(0.5)
        print(dg.dg_version)


def test_hd_version():
    """
    Prints the HD version.

    @return: None
    """
    hd = HD()

    if hd.cmd_log_in_to_hd():
        hd.ui.cmd_ui_request_hd_version()
        time.sleep(0.5)
        print(f"Current HD version: {hd.ui.hd_version}")


def test_hd_off_button():
    """
    Simulates pressing the off button

    @return: None
    """
    hd = HD()

    if hd.cmd_log_in_to_hd():
        # reset previous override
        hd.buttons.cmd_off_button_override(BUTTON_PRESSED, RESET)

        # simulate off button pressed
        hd.buttons.cmd_off_button_override(BUTTON_PRESSED, NO_RESET)


def test_hd_blood_flow(ml_per_min=110):
    """
    Sets the HD blood flow rate (mL / min)

    @return: None
    """
    hd = HD()
    if hd.cmd_log_in_to_hd():
        hd.bloodflow.cmd_blood_flow_set_point_override(0, RESET)
        hd.bloodflow.cmd_blood_flow_set_point_override(ml_per_min, NO_RESET)
        while True:
           print("measured blood flow rate: {0}".format(hd.bloodflow.measured_blood_flow_rate))
           time.sleep(0.5)


def test_stop_hd_blood_flow():
    """
    Stops the HD blood flow

    @return: None
    """
    hd = HD()
    if hd.cmd_log_in_to_hd():
        hd.bloodflow.cmd_blood_flow_set_point_override(0, NO_RESET)


def test_dialin_version():
    import dialin
    print(dialin.__version__)


def test_valves():
    dg = DG()
    if dg.cmd_log_in_to_dg():
        print(dg.valves.get_valve_states())

        for valve_id in range(0, 13):
            dg.valves.cmd_valve_override(False, valve_id, RESET)

        for valve_id in range(0, 13):
            if valve_id % 2 == 0:
                dg.valves.cmd_valve_override(True, valve_id, NO_RESET)
            else:
                dg.valves.cmd_valve_override(False, valve_id, RESET)

        print("Sleeping...")
        time.sleep(5)
        print(dg.valves.get_valve_states())


if __name__ == '__main__':
    test_valves()