/**************************************************************************
*
* 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 ModeRecirculate.c
*
* @author (last) Quang Nguyen
* @date (last) 13-Aug-2020
*
* @author (original) Sean
* @date (original) 04-Apr-2020
*
***************************************************************************/

#include "etpwm.h"

#include "ConductivitySensors.h"
#include "DrainPump.h"
#include "FPGA.h"
#include "Heaters.h"
#include "ModeRecirculate.h"
#include "OperationModes.h"
#include "Pressures.h"
#include "ROPump.h"
#include "TaskGeneral.h"
#include "Timers.h"
#include "Valves.h"
#include "TemperatureSensors.h"

/**
 *  @addtogroup DGRecirculateMode
 *  @{
 */

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

#define TARGET_RO_PRESSURE_PSI          120                    ///< Target pressure for RO pump.
#define FLUSH_LINES_VOLUME_L            0.1                    ///< Water volume (in Liters) to flush when starting re-circulate mode.

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

static DG_RECIRCULATE_MODE_STATE_T recircState;                ///< Currently active re-circulation state.
static F32 flushLinesVolumeL = 0.0;                            ///< Volume of water pumped by RO pump during flush lines state.

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

static DG_RECIRCULATE_MODE_STATE_T handleFlushLinesState( void );
static DG_RECIRCULATE_MODE_STATE_T handleRecircWaterState( void );
static DG_RECIRCULATE_MODE_STATE_T handleRecircPauseState( void );

/*********************************************************************//**
 * @brief
 * The initRecirculateMode function initializes the Fill Mode module.
 * @details
 * Inputs : none
 * Outputs : Fill Mode module initialized.
 * @return none
 *************************************************************************/
void initRecirculateMode( void )
{
    recircState = DG_RECIRCULATE_MODE_STATE_START;
    flushLinesVolumeL = 0.0;
}

/*********************************************************************//**
 * @brief
 * The transitionToRecirculateMode function prepares for transition to \n
 * fill mode.
 * @details
 * Inputs : none
 * Outputs : recircState
 * @return none
 *************************************************************************/
void transitionToRecirculateMode( void )
{
    // re-initialize each time we transition to re-circulate mode
    initRecirculateMode();

    // set initial actuator states
    setValveState( VSP, VALVE_STATE_CLOSED );
    setValveState( VPI, VALVE_STATE_OPEN );
    setValveState( VRC, VALVE_STATE_DRAIN_C_TO_NO );
    setValveState( VDR, VALVE_STATE_DRAIN_C_TO_NO );
    setValveState( VPO, VALVE_STATE_NOFILL_C_TO_NO );
    setROPumpTargetPressure( TARGET_RO_PRESSURE_PSI, PUMP_CONTROL_MODE_CLOSED_LOOP );
    signalDrainPumpHardStop();
    startPrimaryHeater();
    // UV on
    // conc. pumps off
#ifndef _VECTORCAST_
    {   // TODO - test code to start the fan since we're turning the heater on
        F32 fanPWM = 0.25;
        etpwmSetCmpA( etpwmREG6, (U32)( (S32)( ( fanPWM * (F32)(etpwmREG6->TBPRD) ) + FLOAT_TO_INT_ROUNDUP_OFFSET ) ) );
        etpwmSetCmpB( etpwmREG6, (U32)( (S32)( ( fanPWM * (F32)(etpwmREG6->TBPRD) ) + FLOAT_TO_INT_ROUNDUP_OFFSET ) ) );
    }
#endif
}

/*********************************************************************//**
 * @brief
 * The execRecirculateMode function executes the Re-circulate Mode state machine.
 * @details
 * Inputs : recircState
 * Outputs : recircState
 * @return current state
 *************************************************************************/
U32 execRecirculateMode( void )
{
    // check inlet water conductivity, temperature, and pressure
    checkInletWaterConductivity( recircState );
    checkInletWaterTemperature();
    checkInletPressure();

    // execute current re-circulate state
    switch ( recircState )
    {
        case DG_RECIRCULATE_MODE_STATE_START:
            recircState = DG_RECIRCULATE_MODE_STATE_FLUSH_LINES;
            break;

        case DG_RECIRCULATE_MODE_STATE_FLUSH_LINES:
            recircState = handleFlushLinesState();
            break;

        case DG_RECIRCULATE_MODE_STATE_RECIRC_WATER:
            recircState = handleRecircWaterState();
            break;

        case DG_RECIRCULATE_MODE_STATE_PAUSE:
            recircState = handleRecircPauseState();
            break;

        default:
            SET_ALARM_WITH_2_U32_DATA( ALARM_ID_DG_SOFTWARE_FAULT, 0, recircState )     // TODO - add s/w fault enum to 1st data param
            recircState = DG_RECIRCULATE_MODE_STATE_START;
            break;
    }

    return (U32)recircState;
}

/*********************************************************************//**
 * @brief
 * The handleCheckInletWaterState function executes the flush lines \n
 * state of the Re-circulate Mode state machine.
 * @details
 * Inputs : none
 * Outputs :
 * @return the next state
 *************************************************************************/
static DG_RECIRCULATE_MODE_STATE_T handleFlushLinesState( void )
{
    DG_RECIRCULATE_MODE_STATE_T result = DG_RECIRCULATE_MODE_STATE_FLUSH_LINES;
    F32 waterFlowRate = getMeasuredROFlowRate();
    F32 waterVolume = ( ( waterFlowRate / SEC_PER_MIN ) / ( MS_PER_SECOND / TASK_GENERAL_INTERVAL ) );

    // integrate volume of water moved through line
    flushLinesVolumeL += waterVolume;

    // when enough water volume has flowed to flush the lines, transition to re-circ state
    if ( flushLinesVolumeL >= FLUSH_LINES_VOLUME_L )
    {
        setValveState( VDR, VALVE_STATE_RECIRC_C_TO_NC );
        result = DG_RECIRCULATE_MODE_STATE_RECIRC_WATER;
    }

    return result;
}

/*********************************************************************//**
 * @brief
 * The handleRecircProductWaterState function executes the re-circulate \n
 * water state of the Re-circulate Mode state machine.
 * @details
 * Inputs : none
 * Outputs :
 * @return the next state
 *************************************************************************/
static DG_RECIRCULATE_MODE_STATE_T handleRecircWaterState( void )
{
    DG_RECIRCULATE_MODE_STATE_T result = DG_RECIRCULATE_MODE_STATE_RECIRC_WATER;

    return result;
}

/*********************************************************************//**
 * @brief
 * The handleRecircPauseState function executes the pause state of the \n
 * Re-circulate Mode state machine.
 * @details
 * Inputs : none
 * Outputs :
 * @return the next state
 *************************************************************************/
static DG_RECIRCULATE_MODE_STATE_T handleRecircPauseState( void )
{
    DG_RECIRCULATE_MODE_STATE_T result = DG_RECIRCULATE_MODE_STATE_PAUSE;

    return result;
}

/*********************************************************************//**
 * @brief
 * The requestDGStop function handles an HD request to stop (return to Standby mode).
 * @details
 * Inputs : none
 * Outputs : DG standby mode requested
 * @return TRUE if request accepted, FALSE if not.
 *************************************************************************/
BOOL requestDGStop( void )
{
    BOOL result = TRUE;

    requestNewOperationMode( DG_MODE_STAN );

    return result;
}

/*********************************************************************//**
 * @brief
 * The getCurrentRecirculateState function returns the current state of the \n
 * re-circulate mode.
 * @details
 * Inputs : recircState
 * Outputs : none
 * @return the current state of re-circulate mode.
 *************************************************************************/
DG_RECIRCULATE_MODE_STATE_T getCurrentRecirculateState( void )
{
    return recircState;
}

/**@}*/