/**************************************************************************
*
* 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 ModePreTreat.c
*
* @author (last) Sean Nash
* @date (last) 28-Aug-2020
*
* @author (original) Dara Navaei
* @date (original) 05-Nov-2019
*
***************************************************************************/

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

/**
 *  @addtogroup HDPreTreatmentMode
 *  @{
 */

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

static BOOL treatStartReqReceived = FALSE;                      ///< Flag indicates user requests treatment begin.

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

/*********************************************************************//**
 * @brief
 * The initPreTreatmentMode function initializes the Pre-Treatment Mode module.
 * @details Inputs: none
 * @details Outputs: Pre-Treatment Mode module initialized.
 * @return none
*************************************************************************/
void initPreTreatmentMode( void )
{
}

/*********************************************************************//**
 * @brief
 * The transitionToPreTreatmentMode function prepares for transition to
 * pre-treatment mode.
 * @details Inputs: none
 * @details Outputs: none
 * @return none
*************************************************************************/
void transitionToPreTreatmentMode( void )
{
    treatStartReqReceived = FALSE;

    // temporary test code.  TODO - remove later
#ifndef UF_TEST_ENABLED
#ifndef RUN_PUMPS_OPEN_LOOP
    setBloodPumpTargetFlowRate( 0, MOTOR_DIR_FORWARD, PUMP_CONTROL_MODE_CLOSED_LOOP );
    setDialInPumpTargetFlowRate( 0, MOTOR_DIR_FORWARD, PUMP_CONTROL_MODE_CLOSED_LOOP );
#else
    setBloodPumpTargetFlowRate( 0, MOTOR_DIR_FORWARD, PUMP_CONTROL_MODE_OPEN_LOOP );
    setDialInPumpTargetFlowRate( 0, MOTOR_DIR_FORWARD, PUMP_CONTROL_MODE_OPEN_LOOP );
#endif
    setDialOutPumpTargetRate( 0, MOTOR_DIR_FORWARD, PUMP_CONTROL_MODE_CLOSED_LOOP );
#else
#endif
#ifdef RM46_EVAL_BOARD_TARGET
    start = getMSTimerCount();
#endif
}

/*********************************************************************//**
 * @brief
 * The execFaultMode function executes the Pre-Treatment Mode state machine.
 * @details Inputs: none
 * @details Outputs: none
 * @return current state (sub-mode)
*************************************************************************/
U32 execPreTreatmentMode( void )
{
   BOOL stop = isStopButtonPressed();

#ifndef DISABLE_UI_TREATMENT_WORKFLOW
    if ( TRUE == treatStartReqReceived )
#else
#ifndef UF_TEST_ENABLED
   if ( TRUE == stop )
#endif
#endif
    {
        startAirTrapControl();  // TODO - move to right after prime completes when prime is implemented
        requestNewOperationMode( MODE_TREA );
    }
#ifdef RM46_EVAL_BOARD_TARGET
    if ( TRUE == didTimeout( start, 5000U ) )
    {
        requestNewOperationMode( MODE_TREA );
    }
#endif

    return 0; // TODO - return current state
}

/*********************************************************************//**
 * @brief
 * The signalUserBeginningTreatment function handles user start of a
 * treatment.
 * @details Inputs: none
 * @details Outputs: requested mode transition to treatment mode
 * @return TRUE if signal accepted, FALSE if not
 *************************************************************************/
BOOL signalUserBeginningTreatment( void )
{
    BOOL result = FALSE;

    if ( MODE_PRET == getCurrentOperationMode() )
    {
        treatStartReqReceived = TRUE;
        result = TRUE;
    }
    sendTreatmentStartResponseMsg( result, 0 ); // TODO - provide reason code if rejected

    return result;
}

/**@}*/