//
// Created by MightyPork on 2017/11/25.
//

#include "unit_base.h"
#include "unit_fcap.h"

#define FCAP_INTERNAL
#include "_fcap_internal.h"

// ------------------------------------------------------------------------

enum FcapCmd_ {
    CMD_STOP = 0,
    CMD_INDIRECT_CONT_START = 1,
    CMD_INDIRECT_BURST_START = 2,
    CMD_DIRECT_CONT_START = 3,
    CMD_INDIRECT_CONT_READ = 10,
    CMD_DIRECT_CONT_READ = 11,
};

/** Handle a request message */
static error_t UFCAP_handleRequest(Unit *unit, TF_ID frame_id, uint8_t command, PayloadParser *pp)
{
    struct priv *priv = unit->data;
    PayloadBuilder pb = pb_start(unit_tmp512, UNIT_TMP_LEN, NULL);

    switch (command) {
        case CMD_STOP:
            UFCAP_SwitchMode(unit, OPMODE_IDLE);
            return E_SUCCESS;

        case CMD_INDIRECT_CONT_START:
            if (priv->opmode == OPMODE_PWM_CONT) return E_SUCCESS; // no-op
            if (priv->opmode != OPMODE_IDLE) return E_BUSY;

            UFCAP_SwitchMode(unit, OPMODE_PWM_CONT);
            return E_SUCCESS;

        case CMD_INDIRECT_BURST_START:
            if (priv->opmode != OPMODE_IDLE) return E_BAD_MODE;

            uint16_t count = pp_u16(pp);
            priv->pwm_burst.n_target = count;
            priv->request_id = frame_id;
            UFCAP_SwitchMode(unit, OPMODE_PWM_BURST);
            return E_SUCCESS;

        case CMD_INDIRECT_CONT_READ:
            if (priv->opmode != OPMODE_PWM_CONT) {
                return E_BAD_MODE;
            }
            if (priv->pwm_cont.last_period == 0) {
                return E_BUSY;
            }

            pb_u16(&pb, PLAT_AHB_MHZ);
            pb_u32(&pb, priv->pwm_cont.last_period);
            pb_u32(&pb, priv->pwm_cont.last_ontime);

            com_respond_pb(frame_id, MSG_SUCCESS, &pb);
            return E_SUCCESS;

        case CMD_DIRECT_CONT_START:
            if (priv->opmode == OPMODE_COUNTER_CONT) return E_SUCCESS; // no-op
            if (priv->opmode != OPMODE_IDLE) return E_BUSY;

            UFCAP_SwitchMode(unit, OPMODE_COUNTER_CONT);
            return E_SUCCESS;

        case CMD_DIRECT_CONT_READ:
            if (priv->opmode != OPMODE_COUNTER_CONT) {
                return E_BAD_MODE;
            }
            if (priv->cnt_cont.last_count == 0) {
                return E_BUSY;
            }

            // TODO also add the window len - may need to shorten it for fast signals (or if fast capture is required)
            pb_u32(&pb, priv->cnt_cont.last_count);

            com_respond_pb(frame_id, MSG_SUCCESS, &pb);
            return E_SUCCESS;

        default:
            return E_UNKNOWN_COMMAND;
    }
}

// ------------------------------------------------------------------------

/** Frequency capture */
const UnitDriver UNIT_FCAP = {
    .name = "FCAP",
    .description = "Frequency and pulse measurement",
    // Settings
    .preInit = UFCAP_preInit,
    .cfgLoadBinary = UFCAP_loadBinary,
    .cfgWriteBinary = UFCAP_writeBinary,
    .cfgLoadIni = UFCAP_loadIni,
    .cfgWriteIni = UFCAP_writeIni,
    // Init
    .init = UFCAP_init,
    .deInit = UFCAP_deInit,
    // Function
    .handleRequest = UFCAP_handleRequest,
};