import sys
sys.path.append("..")
from time import sleep

from leahi_dialin.td.treatment_delivery import TD

if __name__ == "__main__":
    # create an TD object called td
    td = TD(log_level="DEBUG")

    # log in to TD as a tester
    if td.cmd_log_in_to_td() == 0:
        exit(1)

    # set BP RPM
    td.blood_flow.cmd_blood_flow_set_speed_rate_request(1000)

    # create log file
    with open("TD_test.log", "w") as f:

        # collect data and display/log it
        while True:
            sleep(1)
            modes = "TD.m, " + '{:2d}'.format(td.td_operation_mode) + \
                ", TD.s, " + '{:2d}'.format(td.td_operation_sub_mode)
            pumpSetPts = ", BP.sf, " + '{:4d}'.format(td.blood_flow.set_blood_flow_rate) + \
                         ", BP.ss, " + '{:9.2f}'.format(td.blood_flow.set_rpm)
            pumpMeasSpds = ", BP.ms, " + '{:7.1f}'.format(td.blood_flow.measured_blood_pump_speed)
            press = ", PBA, " + '{:9.2f}'.format(td.pressure_sensors.arterial_pressure) + \
                ", PBV, " + '{:9.2f}'.format(td.pressure_sensors.venous_pressure)

            # log data
            f.write(modes)
            f.write(pumpSetPts)
            f.write(pumpMeasSpds)
            f.write(press)
            f.write("\n")

            # print to console
#            print("       Modes: "+modes)
            print("       Pumps: "+pumpSetPts+pumpMeasSpds)
            print("   Pressures: "+press)