/**************************************************************************
*
* 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) 06-Oct-2020
*
* @author (original) Dara Navaei
* @date (original) 05-Nov-2019
*
***************************************************************************/

#include "ModePreTreat.h"
#include "OperationModes.h"
#include "Prime.h"
#include "SystemCommMessages.h"

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

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

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

static HD_PRE_TREATMENT_MODE_STATE_T currentPreTreatmentState;             ///< Current state of pre-treatment mode state machine.

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

static HD_PRE_TREATMENT_MODE_STATE_T handleSelfTestNoCartState( void );
static HD_PRE_TREATMENT_MODE_STATE_T handleSelfTestDryState( void );
static HD_PRE_TREATMENT_MODE_STATE_T handlePrimeState( void );
static HD_PRE_TREATMENT_MODE_STATE_T handlePatientConnectionState( void );

/*********************************************************************//**
 * @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 )
{
    currentPreTreatmentState = HD_PRE_TREATMENT_START_STATE;
}

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

    // Set user alarm recovery actions allowed in this mode
    setAlarmUserActionEnabled( ALARM_USER_ACTION_RESUME, TRUE );
    setAlarmUserActionEnabled( ALARM_USER_ACTION_RINSEBACK, FALSE );
    setAlarmUserActionEnabled( ALARM_USER_ACTION_END_TREATMENT, TRUE );

    currentPreTreatmentState = HD_PRE_TREATMENT_START_STATE;
}

/*********************************************************************//**
 * @brief
 * The execPreTreatmentMode function executes the Pre-Treatment Mode state machine.
 * @details Inputs: none
 * @details Outputs: none
 * @return current state (sub-mode)
*************************************************************************/
U32 execPreTreatmentMode( void )
{
    // execute mode state machine
    switch ( currentPreTreatmentState )
    {
        case HD_PRE_TREATMENT_START_STATE:
            cmdStartDG();
            cmdSetDGDialysateTargetTemps( 39.0, 37.0 );
            currentPreTreatmentState = HD_PRE_TREATMENT_WATER_SAMPLE_STATE;
            break;

        case HD_PRE_TREATMENT_WATER_SAMPLE_STATE:
            currentPreTreatmentState = HD_PRE_TREATMENT_SELF_TEST_NO_CART_STATE;
            break;

        case HD_PRE_TREATMENT_SELF_TEST_NO_CART_STATE:
            currentPreTreatmentState = handleSelfTestNoCartState();
            break;

        case HD_PRE_TREATMENT_CART_INSTALL_STATE:
            currentPreTreatmentState = HD_PRE_TREATMENT_SELF_TEST_DRY_STATE;
            break;

        case HD_PRE_TREATMENT_SELF_TEST_DRY_STATE:
            currentPreTreatmentState = handleSelfTestDryState();
            break;

        case HD_PRE_TREATMENT_PRIME_STATE:
            currentPreTreatmentState = handlePrimeState();
            break;

        case HD_PRE_TREATMENT_RECIRCULATE_STATE:
            currentPreTreatmentState = HD_PRE_TREATMENT_PATIENT_CONNECTION_STATE;
            break;

        case HD_PRE_TREATMENT_PATIENT_CONNECTION_STATE:
            currentPreTreatmentState = handlePatientConnectionState();
            break;

        default:
            currentPreTreatmentState = HD_PRE_TREATMENT_START_STATE;
            SET_ALARM_WITH_2_U32_DATA( ALARM_ID_HD_SOFTWARE_FAULT, SW_FAULT_ID_MODE_PRE_TREATMENT_INVALID_STATE, (U32)currentPreTreatmentState );
            break;
    }

    return (U32)currentPreTreatmentState;
}

/*********************************************************************//**
 * @brief
 * The signalUserBeginningTreatment function handles user start of a
 * treatment.
 * @details Inputs: none
 * @details Outputs: treatStartReqReceived, send response to treatment start
 * @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;
}

/*********************************************************************//**
 * @brief
 * The signalAlarmActionToPreTreatmentMode function executes the given alarm action
 * as appropriate while in PreTreatment Mode.
 * @details Inputs: none
 * @details Outputs: given alarm action executed
 * @param action ID of alarm action to execute
 * @return none
 *************************************************************************/
void signalAlarmActionToPreTreatmentMode( ALARM_ACTION_T action )
{
    // TODO - implement
}

/*********************************************************************//**
 * @brief
 * The handleSelfTestNoCartState function handles self-test with no cartrige.
 * @details Inputs: none
 * @details Outputs: home blood pump and dialysate pumps
 * @return current state (sub-mode)
 *************************************************************************/
static HD_PRE_TREATMENT_MODE_STATE_T handleSelfTestNoCartState( void )
{
    // TODO: Prompt user to close door

    homeBloodPump();
    homeDialInPump();
    homeDialOutPump();

    // TODO: Prompt user to open door

    return HD_PRE_TREATMENT_CART_INSTALL_STATE;
}

/*********************************************************************//**
 * @brief
 * The handleSelfTestDryState function handles dry self-test.
 * @details Inputs: none
 * @details Outputs: transition to prime sub-mode when blood pump finished homing
 * @return current state (sub-mode)
 *************************************************************************/
static HD_PRE_TREATMENT_MODE_STATE_T handleSelfTestDryState( void )
{
    HD_PRE_TREATMENT_MODE_STATE_T state = HD_PRE_TREATMENT_CART_INSTALL_STATE;

    if ( FALSE == isBloodPumpRunning() )
    {
        transitionToPrime();
        state = HD_PRE_TREATMENT_PRIME_STATE;
    }

    return state;
}

/*********************************************************************//**
 * @brief
 * The handlePrimeState function handles priming the blood and dialysate circuits.
 * @details Inputs: none
 * @details Outputs: transition to self test wet state after priming passed
 * @return current state (sub-mode)
 *************************************************************************/
static HD_PRE_TREATMENT_MODE_STATE_T handlePrimeState( void )
{
    HD_PRE_TREATMENT_MODE_STATE_T state = HD_PRE_TREATMENT_PRIME_STATE;

    execPrime();

    if ( TRUE == isPrimingPassed() )
    {
        state = HD_PRE_TREATMENT_RECIRCULATE_STATE;
    }

    return state;
}

/*********************************************************************//**
 * @brief
 * The handlePatientConnectionState function handles patient connection state
 * during pre-treatment mode.
 * @details Inputs: none
 * @details Outputs: requested mode transition to treatment mode
 * @return current state (sub-mode)
 *************************************************************************/
static HD_PRE_TREATMENT_MODE_STATE_T handlePatientConnectionState( void )
{
    if ( TRUE == treatStartReqReceived )
    {
        requestNewOperationMode( MODE_TREA );
    }

    return HD_PRE_TREATMENT_PATIENT_CONNECTION_STATE;
}

/**@}*/