/**************************************************************************
 *
 *  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
 *
 *  @date 19-Sep-2019
 *  @author S. Nash
 *
 *  @brief Top-level state machine for the standby mode.
 *
 **************************************************************************/

#include "AlarmLamp.h"
#include "BloodFlow.h"
#include "Buttons.h"
#include "DGInterface.h"
#include "DialInFlow.h"
#include "DialOutFlow.h"
#include "OperationModes.h"
#include "SystemCommMessages.h"
#include "ModeStandby.h"
#ifdef RM46_EVAL_BOARD_TARGET
    #include "Timers.h"
    static U32 start;
#endif

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

/// Sub-mode states while in standby mode
typedef enum Standby_States
{
    STANDBY_START_STATE = 0,                ///< Start standby
    STANDBY_FLUSH_DG_LINES_STATE,           ///<
    STANDBY_DRAIN_RESERVOIR_STATE,          ///<
    STANDBY_WAIT_FOR_PRIME_STATE,           ///<
    STANDBY_FILL_RESERVOIR_STATE,           ///<
    STANDBY_WAIT_FOR_TREATMENT_STATE,       ///<
    NUM_OF_STANDBY_STATES                   ///< Number of standby states (sub-modes).
} STANDBY_STATE_T;

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

static 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();

#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
}

/*************************************************************************
 * @brief execStandbyMode
 * The execStandbyMode function executes the Standby Mode state machine.
 * @details
 * Inputs : none
 * Outputs :
 * @param none
 * @return none
 *************************************************************************/
void execStandbyMode( void )
{
    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();

    // treatment mode state machine
    switch ( currentStandbyState )
    {
        case STANDBY_START_STATE:
            // temporary test code - TODO - remove later
            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
            if ( DG_MODE_CIRC == dgOpMode )
            {
                if ( DG_RECIRCULATE_MODE_STATE_RECIRC_WATER == dgSubMode )
                {
                    cmdStartDGDrain( 100 );
                }
            }
            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( 1500 );
                }
            }
            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 )
            {
                cmdSetDGActiveReservoir( DG_RESERVOIR_1 );
                currentStandbyState = STANDBY_WAIT_FOR_TREATMENT_STATE;
            }
            break;

        case STANDBY_WAIT_FOR_TREATMENT_STATE:
            // TODO - test code
            if ( TRUE == stop )
            {
                if ( DG_MODE_CIRC == getDGOpMode() )
                {
                    if ( DG_RECIRCULATE_MODE_STATE_RECIRC_WATER == dgSubMode )
                    {
                        requestNewOperationMode( MODE_PRES );
                    }
                }
            }
            break;

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

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