/**************************************************************************
*
* 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 "AlarmMgmt.h"
#include "Buttons.h"
#include "FPGA.h"
#include "ModePreTreat.h"
#include "OperationModes.h"
#include "PreTreatmentRecirc.h"
#include "Prime.h"
#include "SelfTests.h"
#include "SystemCommMessages.h"

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

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

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

static BOOL treatStartReqReceived = FALSE;                              ///< Flag indicates user requests treatment begin.
static BOOL alarmActionStopReceived = FALSE;                            ///< Flag indicates an alarm w/ stop property was triggered.
static BOOL alarmActionResumeReceived = FALSE;                          ///< Flag indicates alarm action resume received.

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

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

static void resetSignalFlags( void );

static HD_PRE_TREATMENT_MODE_STATE_T handleSelfTestNoCartState( void );
static HD_PRE_TREATMENT_MODE_STATE_T handleInstallState( 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 handleRecirculateState( 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;

    initPrime();
    initSelfTests();

    resetSignalFlags();
}

/*********************************************************************//**
 * @brief
 * The transitionToPreTreatmentMode function prepares for transition to
 * pre-treatment mode.
 * @details Inputs: none
 * @details Outputs: none
 * @return none
*************************************************************************/
void transitionToPreTreatmentMode( void )
{
    // 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 );

    initPreTreatmentMode();
}

/*********************************************************************//**
 * @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 )
{
    BOOL stop = isStopButtonPressed();

    if ( TRUE == stop )
    {
        activateAlarmNoData( ALARM_ID_TREATMENT_STOPPED_BY_USER );
    }

    // 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:
            transitionToNoCartSelfTests();
            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 = handleInstallState();
            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 = handleRecirculateState();
            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;
    }

    // Alarm response request flags should be handled at this point, reset in case not handled in current state
    resetSignalFlags();

    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;

    // TODO: check patient connection sub-mode state
    if ( ( MODE_PRET == getCurrentOperationMode() ) && ( HD_PRE_TREATMENT_PATIENT_CONNECTION_STATE == currentPreTreatmentState ) )
    {
        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 )
{
    switch( action )
    {
        case ALARM_ACTION_STOP:
            alarmActionStopReceived = TRUE;
            break;

        case ALARM_ACTION_RESUME:
            alarmActionResumeReceived = TRUE;
            break;

        case ALARM_ACTION_END_TREATMENT:
            requestNewOperationMode( MODE_STAN );
            break;

        case ALARM_ACTION_ACK:
            // Nothing to be done here
            break;

        default:
            // Ignore
            break;
    }
}

/*********************************************************************//**
 * @brief
 * The resetSignalFlags function resets all signal flags.
 * @details Inputs: none
 * @details Outputs: signal flags set to FALSE
 * @return none
 *************************************************************************/
static void resetSignalFlags( void )
{
    treatStartReqReceived = FALSE;
    alarmActionStopReceived = FALSE;
    alarmActionResumeReceived = FALSE;
}

/*********************************************************************//**
 * @brief
 * The handleSelfTestNoCartState function handles self-test with no cartridge.
 * @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 )
{
    HD_PRE_TREATMENT_MODE_STATE_T state = HD_PRE_TREATMENT_SELF_TEST_NO_CART_STATE;

    if ( STATE_OPEN == getFPGADoorState() )
    {
        activateAlarmNoData( ALARM_ID_CARTRIDGE_DOOR_OPENED );
    }

    if ( TRUE == alarmActionResumeReceived )
    {
        signalResumeSelfTests();
    }

    if ( TRUE == alarmActionStopReceived )
    {
        signalStopSelfTests();
    }

    execNoCartSelfTests();

    if ( TRUE == isNoCartSelfTestsPassed() )
    {
        state = HD_PRE_TREATMENT_CART_INSTALL_STATE;
    }

    return state;
}

/*********************************************************************//**
 * @brief
 * The handleInstallState function handles disposable installation.
 * @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 handleInstallState( void )
{
    HD_PRE_TREATMENT_MODE_STATE_T state = HD_PRE_TREATMENT_CART_INSTALL_STATE;

    // TODO: handle install verification and transition
    transitionToDrySelfTests();
    state = HD_PRE_TREATMENT_SELF_TEST_DRY_STATE;

    return state;
}

/*********************************************************************//**
 * @brief
 * The handleSelfTestDryState function performs dry self-test.
 * @details Inputs: none
 * @details Outputs: transition to prime state on user request
 * @return current state (sub-mode)
 *************************************************************************/
static HD_PRE_TREATMENT_MODE_STATE_T handleSelfTestDryState( void )
{
    HD_PRE_TREATMENT_MODE_STATE_T state = HD_PRE_TREATMENT_SELF_TEST_DRY_STATE;

    if ( STATE_OPEN == getFPGADoorState() )
    {
        activateAlarmNoData( ALARM_ID_CARTRIDGE_DOOR_OPENED );
    }

    if ( TRUE == alarmActionResumeReceived )
    {
        signalResumeSelfTests();
    }

    if ( TRUE == alarmActionStopReceived )
    {
        signalStopSelfTests();
    }

    execDrySelfTests();

    if ( TRUE == isDrySelfTestsPassed() )
    {
        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;

    if ( STATE_OPEN == getFPGADoorState() )
    {
        activateAlarmNoData( ALARM_ID_CARTRIDGE_DOOR_OPENED );
    }

    if ( TRUE == alarmActionResumeReceived )
    {
        signalResumePrime();
    }

    if ( TRUE == alarmActionStopReceived )
    {
        signalStopPrime();
    }

    execPrime();

    if ( TRUE == isPrimeCompleted() )
    {
        state = HD_PRE_TREATMENT_RECIRCULATE_STATE;
        activateAlarmNoData( ALARM_ID_PRIME_COMPLETED_LOW_PRIORITY );
        transitionToPreTreatmentRecirc();
    }

    return state;
}

/*********************************************************************//**
 * @brief
 * The handleRecirculateState function handles blood and dialysate circuits
 * recirculation state during pre-treatment mode.
 * @details Inputs: none
 * @details Outputs: controlled valves and pumps
 * @return current state (sub-mode)
 *************************************************************************/
static HD_PRE_TREATMENT_MODE_STATE_T handleRecirculateState( void )
{
    HD_PRE_TREATMENT_MODE_STATE_T state = HD_PRE_TREATMENT_RECIRCULATE_STATE;

    if ( STATE_OPEN == getFPGADoorState() )
    {
        activateAlarmNoData( ALARM_ID_CARTRIDGE_DOOR_OPENED );
    }

    if ( TRUE == alarmActionResumeReceived )
    {
        signalResumePreTreatmentRecirc();
    }

    if ( TRUE == alarmActionStopReceived )
    {
        signalStopPreTreatmentRecirc();
    }

    execPreTreatmentRecirc();

    if ( TRUE == isPatientConnectionRequested() )
    {
        state = HD_PRE_TREATMENT_PATIENT_CONNECTION_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 )
{
    DG_RESERVOIR_ID_T const activeRes = getDGActiveReservoir();

    if ( TRUE == treatStartReqReceived )
    {
        setStartReservoirVolume( activeRes );
        requestNewOperationMode( MODE_TREA );
    }

    return HD_PRE_TREATMENT_PATIENT_CONNECTION_STATE;
}

/**@}*/