import sys
sys.path.append("..")
from dialin.dg.dialysate_generator import DG
from dialin.dg.heat_disinfection import HeatDisinfectionStates
from dialin.dg.heat_disinfection import HeatDisinfectionInternalStates
from dialin.dg.drain_pump import DrainPumpStates
from dialin.dg.thermistors import Thermistors
from dialin.dg.uv_reactors import ReactorsStates
from time import sleep


if __name__ == "__main__":

    dg = DG(log_level='DEBUG')
    dg.cmd_log_in_to_dg()

    dg.uv_reactors.cmd_start_stop_inlet_uv_reactor(state=ReactorsStates.UV_REACTOR_STATE_ON.name)

    while True:
        sleep(1)
        print(dg.thermistors.onboard_temperature)

    f = open("Heat_disinfection.txt", "w")

    #dg.drain_pump.cmd_drain_pump_speed_set_point_override(2800)

    #dg.valves.cmd_valve_override(0, dg.valves.VALVE_PRESSURE_OUTLET )
    #sleep(1)
    #dg.valves.cmd_valve_override(0, dg.valves.VALVE_RESERVOIR_OUTLET)
    #dg.valves.cmd_valve_override(1, dg.valves.VALVE_RESERVOIR_INLET)
    dg.heat_disinfect.cmd_start_stop_heat_disinfection(start=False)
    sleep(1)
    time = 1

    dg.heat_disinfect.cmd_start_stop_heat_disinfection()

    try:
        while True:
            """
            var = ('Mode, {}, State, {}, Sub, {}, RO, {}, Drain, {}, VPi, {}, VBf, {}, VSP, {}, VPd, {}, VPo, {}, VDr, {}, '
                  'VRc, {}, VRo, {}, VRd, {}, VRi, {}, VRf, {}\r'
                                                        .format(dg.dg_operation_mode,
                                                        HeatDisinfectionStates(dg.dg_operation_sub_mode).name,
                                                        HeatDisinfectionInternalStates(dg.heat_disinfect.internal_state).name,
                                                        dg.ro_pump.ro_pump_state,
                                                        DrainPumpStates(dg.drain_pump.drain_pump_state).name,
                                                        dg.valves.valve_states_enum[dg.valves.VALVE_PRESSURE_INLET],
                                                        dg.valves.valve_states_enum[dg.valves.VALVE_BYPASS_FILTER],
                                                        dg.valves.valve_states_enum[dg.valves.VALVE_SAMPLING_PORT],
                                                        dg.valves.valve_states_enum[dg.valves.VALVE_PRODUCTION_DRAIN],
                                                        dg.valves.valve_states_enum[dg.valves.VALVE_PRESSURE_OUTLET],
                                                        dg.valves.valve_states_enum[dg.valves.VALVE_DRAIN],
                                                        dg.valves.valve_states_enum[dg.valves.VALVE_RECIRCULATE],
                                                        dg.valves.valve_states_enum[dg.valves.VALVE_RESERVOIR_OUTLET],
                                                        dg.valves.valve_states_enum[dg.valves.VALVE_RESERVOIR_DRAIN],
                                                        dg.valves.valve_states_enum[dg.valves.VALVE_RESERVOIR_INLET],
                                                        dg.valves.valve_states_enum[dg.valves.VALVE_RESERVOIR_FILL]))
            """
            var = ('Mode, {}, State, {}, Sub, {}, RO, {}, Flow, {:5.3f}, TgtPres, {:5.3f}, InletPres, {:5.3f}, '
                   'OutletPres, {:5.3f}, PWM, {:5.3f}, Drain, {}, DInletPres, {:5.3f}, DOutletPres, {:5.3f}, '
                   'DrainDAC, {:5.3f}, R1, {:5.3f}, R2, {:5.3f}, TPi, {:5.3f}, TPo, {:5.3f}, TDi, {:5.3f} \r'
                                            .format(dg.dg_operation_mode,
                                            HeatDisinfectionStates(dg.dg_operation_sub_mode).name,
                                            HeatDisinfectionInternalStates(dg.heat_disinfect.internal_state).name,
                                            dg.ro_pump.ro_pump_state,
                                            dg.ro_pump.measured_flow_rate_lpm,
                                            dg.ro_pump.target_pressure_psi,
                                            dg.pressures.ro_pump_inlet_pressure,
                                            dg.pressures.ro_pump_outlet_pressure,
                                            dg.ro_pump.pwm_duty_cycle_pct,
                                            DrainPumpStates(dg.drain_pump.drain_pump_state).name,
                                            dg.pressures.drain_pump_inlet_pressure,
                                            dg.pressures.drain_pump_outlet_pressure, dg.drain_pump.dac_value,
                                            dg.load_cells.load_cell_A1,
                                            dg.load_cells.load_cell_B1,
                                            dg.temperature_sensors.inlet_primary, dg.temperature_sensors.outlet_primary,
                                            dg.temperature_sensors.outlet_redundancy))

            print(var)
            f.write(var)
            sleep(0.25)

    except KeyboardInterrupt:
        dg.heat_disinfect.cmd_start_stop_heat_disinfection(start=False)
        f.close()