/**************************************************************************
*
* Copyright (c) 2019-2020 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 ModeStandby.c
*
* @author (last) Sean Nash
* @date (last) 02-Jul-2020
*
* @author (original) Dara Navaei
* @date (original) 05-Nov-2019
*
***************************************************************************/

#include "AlarmLamp.h"
#include "BloodFlow.h"
#include "Buttons.h"
#include "DGInterface.h"
#include "DialInFlow.h"
#include "DialOutFlow.h"
#include "Dialysis.h"
#include "OperationModes.h"
#include "SystemComm.h"
#include "SystemCommMessages.h"
#include "ModeStandby.h"
#ifdef EMC_TEST_BUILD     // TODO - test code
#include "FPGA.h"
#endif
#ifdef RM46_EVAL_BOARD_TARGET
    #include "Timers.h"
    static U32 start;
#endif

    /**
     *  @addtogroup HDStandbyMode
     *  @{
     */

    // ********** private definitions **********

// ********** private data **********

static HD_STANDBY_STATE_T currentStandbyState;                     ///< Current state (sub-mode) of standby mode.

// ********** private function prototypes **********

/*************************************************************************
 * @brief initStandbyMode
 * The initStandbyMode function initializes the Standby Mode module.
 * @details
 * Inputs : none
 * Outputs : Standby Mode module initialized.
 * @param none
 * @return none
 *************************************************************************/
void initStandbyMode( void )
{
    currentStandbyState = STANDBY_START_STATE;
}

/*************************************************************************
 * @brief transitionToStandbyMode
 * The transitionToStandbyMode function prepares for transition to standby mode.
 * @details
 * Inputs : none
 * Outputs :
 * @param none
 * @return none
 *************************************************************************/
void transitionToStandbyMode( void )
{
    // re-initialize when transitioning to standby mode
    initStandbyMode();
    initDGInterface();

#ifndef UF_TEST_ENABLED
    setBloodPumpTargetFlowRate( 0, MOTOR_DIR_FORWARD, PUMP_CONTROL_MODE_CLOSED_LOOP );
    setDialInPumpTargetFlowRate( 0, MOTOR_DIR_FORWARD, PUMP_CONTROL_MODE_CLOSED_LOOP );
    setDialOutPumpTargetRate( 0, MOTOR_DIR_FORWARD, PUMP_CONTROL_MODE_OPEN_LOOP );
#endif
#ifdef RM46_EVAL_BOARD_TARGET
    start = getMSTimerCount();
#endif
#ifdef EMC_TEST_BUILD     // TODO - test code
    enableValvesPIDControl(0); // enable valves
#endif
}

/*************************************************************************
 * @brief execStandbyMode
 * The execStandbyMode function executes the Standby Mode state machine.
 * @details
 * Inputs : none
 * Outputs :
 * @param none
 * @return current state (sub-mode)
 *************************************************************************/
U32 execStandbyMode( void )
{
#ifdef EMC_TEST_BUILD
    static U32 toggle = 0;
    static BOOL button_state = FALSE;
#endif
BOOL stop = isStopButtonPressed();
    DG_OP_MODE_T dgOpMode = getDGOpMode();  // TODO - the DG mode & sub-mode come as a pair at interval - they MUST be kept together.
    U32 dgSubMode = getDGSubMode();

#ifdef UF_TEST_WITH_DG
    // state machine to get DG to prep a reservoir so we can start a treatment
    switch ( currentStandbyState )
    {
        case STANDBY_START_STATE:
            // temporary test code - TODO - remove later
            if ( TRUE == isDGCommunicating() )
            {
                //homeBloodPump();
                //homeDialInPump();
                //homeDialOutPump();
                cmdSetDGDialysateTargetTemps( 39.0, 37.0 );
                cmdSetDGActiveReservoir( DG_RESERVOIR_2 );
                cmdStartDG();
                currentStandbyState = STANDBY_FLUSH_DG_LINES_STATE;
            }
            break;

        case STANDBY_FLUSH_DG_LINES_STATE:
            // temporary test code - TODO - remove later
            cmdSetDGActiveReservoir( DG_RESERVOIR_2 );
            if ( DG_MODE_CIRC == dgOpMode )
            {
                if ( DG_RECIRCULATE_MODE_STATE_RECIRC_WATER == dgSubMode )
                {
                    cmdStartDGDrain( DRAIN_RESERVOIR_TO_VOLUME_ML );
                }
            }
            else if ( DG_MODE_DRAI == dgOpMode )
            {
                currentStandbyState = STANDBY_DRAIN_RESERVOIR_STATE;
            }
            else
            {
                cmdStartDG();
            }
            break;

        case STANDBY_DRAIN_RESERVOIR_STATE:
            // temporary test code - TODO - remove later
            if ( DG_MODE_CIRC == dgOpMode )
            {
                currentStandbyState = STANDBY_WAIT_FOR_PRIME_STATE;
            }
            break;

        case STANDBY_WAIT_FOR_PRIME_STATE:
            // temporary test code - TODO - remove later
            if ( DG_MODE_CIRC == dgOpMode )
            {
                if ( DG_RECIRCULATE_MODE_STATE_RECIRC_WATER == dgSubMode )
                {
                    cmdStartDGFill( FILL_RESERVOIR_TO_VOLUME_ML );
                }
            }
            else if ( DG_MODE_FILL == dgOpMode )
            {
                currentStandbyState = STANDBY_FILL_RESERVOIR_STATE;
            }
            break;

        case STANDBY_FILL_RESERVOIR_STATE:
            // temporary test code - TODO - remove later
            if ( DG_MODE_CIRC == dgOpMode )
            {
                currentStandbyState = STANDBY_WAIT_FOR_TREATMENT_STATE;
            }
            break;

        case STANDBY_WAIT_FOR_TREATMENT_STATE:
            // TODO - test code
            if ( TRUE == stop )
            {
                if ( DG_MODE_CIRC == dgOpMode )
                {
                    if ( DG_RECIRCULATE_MODE_STATE_RECIRC_WATER == dgSubMode )
                    {
                        setStartReservoirVolume();
                        cmdSetDGActiveReservoir( DG_RESERVOIR_1 );
                        requestNewOperationMode( MODE_TPAR );
                    }
                }
            }
            break;

        default:
            // TODO - s/w fault
            currentStandbyState = STANDBY_START_STATE;
            break;
    }
#else
    // state machine to get DG to prep a reservoir so we can start a treatment
    switch ( currentStandbyState )
    {
        case STANDBY_START_STATE:
            // temporary test code - TODO - remove later
            homeBloodPump();
            homeDialInPump();
            homeDialOutPump();
            currentStandbyState = STANDBY_WAIT_FOR_TREATMENT_STATE;
            break;

        case STANDBY_WAIT_FOR_TREATMENT_STATE:
            // TODO - test code
            if ( TRUE == stop )
            {
#ifdef EMC_TEST_BUILD
                if ( stop != button_state )
                {
                    toggle = INC_WRAP( toggle, 0, 3 );
                    switch ( toggle )
                    {
                        case 0: // pumps and valves off
                            setDialyzerInletValvePosition( 0 );
                            setDialyzerOutletValvePosition( 0 );
                            setVenousBloodValvePosition( 0 );
                            setArterialBloodValvePosition( 0 );
                            break;

                        case 1: // pumps on, valves off
                            setBloodPumpTargetFlowRate( 500, MOTOR_DIR_FORWARD, PUMP_CONTROL_MODE_OPEN_LOOP );
                            setDialInPumpTargetFlowRate( 500, MOTOR_DIR_FORWARD, PUMP_CONTROL_MODE_OPEN_LOOP );
                            setDialOutPumpTargetRate( 500, MOTOR_DIR_FORWARD, PUMP_CONTROL_MODE_OPEN_LOOP );
                            break;

                        case 2: // pumps on, valves on
                            setDialyzerInletValvePosition( 12000 );
                            setDialyzerOutletValvePosition( 12000 );
                            setVenousBloodValvePosition( 12000 );
                            setArterialBloodValvePosition( 12000 );
                            break;

                        case 3: // pumps off, valves on
                            setBloodPumpTargetFlowRate( 0, MOTOR_DIR_FORWARD, PUMP_CONTROL_MODE_OPEN_LOOP );
                            setDialInPumpTargetFlowRate( 0, MOTOR_DIR_FORWARD, PUMP_CONTROL_MODE_OPEN_LOOP );
                            setDialOutPumpTargetRate( 0, MOTOR_DIR_FORWARD, PUMP_CONTROL_MODE_OPEN_LOOP );
                            break;

                        default: // shouldn't get here, reset if we do
                            toggle = 0;
                            setBloodPumpTargetFlowRate( 0, MOTOR_DIR_FORWARD, PUMP_CONTROL_MODE_OPEN_LOOP );
                            setDialInPumpTargetFlowRate( 0, MOTOR_DIR_FORWARD, PUMP_CONTROL_MODE_OPEN_LOOP );
                            setDialOutPumpTargetRate( 0, MOTOR_DIR_FORWARD, PUMP_CONTROL_MODE_OPEN_LOOP );
                            setDialyzerInletValvePosition( 0 );
                            setDialyzerOutletValvePosition( 0 );
                            setVenousBloodValvePosition( 0 );
                            setArterialBloodValvePosition( 0 );
                            break;
                    }
                }
#else
                requestNewOperationMode( MODE_TPAR );
#endif
            }
#ifdef EMC_TEST_BUILD
            button_state = stop;
#endif
            break;

        default:
            // TODO - s/w fault
            currentStandbyState = STANDBY_START_STATE;
            break;
    }
#endif

#ifdef RM46_EVAL_BOARD_TARGET
    if ( TRUE == didTimeout( start, 5000U ) )
    {
        requestNewOperationMode( MODE_TPAR );
    }
#endif

    return currentStandbyState;
}

/**@}*/