/*
 * Copyright (c) 2019-2023 Beijing Hanwei Innovation Technology Ltd. Co. and
 * its subsidiaries and affiliates (collectly called MKSEMI).
 *
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice,
 *    this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form, except as embedded into an MKSEMI
 *    integrated circuit in a product or a software update for such product,
 *    must reproduce the above copyright notice, this list of conditions and
 *    the following disclaimer in the documentation and/or other materials
 *    provided with the distribution.
 *
 * 3. Neither the name of MKSEMI nor the names of its contributors may be used
 *    to endorse or promote products derived from this software without
 *    specific prior written permission.
 *
 * 4. This software, with or without modification, must only be used with a
 *    MKSEMI integrated circuit.
 *
 * 5. Any software provided in binary form under this license must not be
 *    reverse engineered, decompiled, modified and/or disassembled.
 *
 * THIS SOFTWARE IS PROVIDED BY MKSEMI "AS IS" AND ANY EXPRESS OR IMPLIED
 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
 * MERCHANTABILITY, NONINFRINGEMENT, AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL MKSEMI OR CONTRIBUTORS BE LIABLE FOR ANY
 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include "mk_trace.h"
#include "mk_clock.h"
#include "mk_uwb.h"
#include "mk_calib.h"
#include "mk_misc.h"

#include "uwb_api.h"
#include "uwb_data_transfer.h"
#include "uwb_data.h"

/*************************************************************************************************/
/*!
 *  \brief  WSF event handler for ranging task.
 *
 *  \param  event   WSF event mask.
 *  \param  msg    WSF message.
 *
 *  \return None.
 */
/*************************************************************************************************/
void ranging_handler(wsfEventMask_t event, const void *param)
{
    const wsfMsgHdr_t *msg = (const wsfMsgHdr_t *)param;

    if (msg != NULL)
    {
        switch (msg->event)
        {
            case UWB_PKT_TX_DONE_MSG:
            {
                const struct UWB_PKT_TX_DONE_IND_T *ind = (const struct UWB_PKT_TX_DONE_IND_T *)param;

                if (ind->ranging_stage == RANGING_DTPCM)
                {
                    ranging_env.range_data.sequence_num = (ranging_env.phy_sts_index - fira_key.phyStsIdxInit) / ranging_env.slots_per_block;
                    LOG_INFO(TRACE_MODULE_APP | TRACE_NO_OPTION, "\r\n");
                    LOG_INFO(TRACE_MODULE_APP, "FiRa Data Transfer - Controller-Initiator SEQ NUM %u\r\n", ranging_env.range_data.sequence_num);
                    LOG_INFO(TRACE_MODULE_APP, "[TX][%d][%u] DTPCM\r\n", ind->slot_idx, ind->tx_len);
                }
                else if ((ind->ranging_stage == RANGING_DM_TX) && (ind->tx_len))
                {
                    LOG_INFO(TRACE_MODULE_APP, "[TX][%d][%u] DM\r\n", ind->slot_idx, ind->tx_len);
                }

#if DM_PRINT_PAYLOAD_EN
                if (ind->tx_len)
                {
                    LOG_INFO(TRACE_MODULE_APP, " ");
                    for (uint8_t i = 0; i < ind->tx_len; i++)
                    {
                        LOG_INFO(TRACE_NO_OPTION | TRACE_MODULE_APP, "%02x ", ind->tx_data[i]);
                    }
                    LOG_INFO(TRACE_NO_OPTION | TRACE_MODULE_APP, "\r\n");
                }
#endif
            }
            break;

            case UWB_PKT_RX_DONE_MSG:
            {
                const struct UWB_PKT_RX_DONE_IND_T *ind = (const struct UWB_PKT_RX_DONE_IND_T *)param;

                if (ind->status == UWB_RX_OK)
                {
                    if (ind->ranging_stage == RANGING_DTPCM)
                    {
                        ranging_env.range_data.sequence_num = (ranging_env.phy_sts_index - fira_key.phyStsIdxInit) / ranging_env.slots_per_block;
                        LOG_INFO(TRACE_MODULE_APP | TRACE_NO_OPTION, "\r\n");
                        LOG_INFO(TRACE_MODULE_APP, "FiRa Data Transfer - Controlee-Responder SEQ NUM %u\r\n", ranging_env.range_data.sequence_num);

                        LOG_INFO(TRACE_MODULE_APP, "[RX][%d][%u] DTPCM\r\n", ind->slot_idx, ind->rx_len);

                        int32_t freq_offset = phy_freq_offset_get();
                        int32_t freq_offset_filter = average_filter(freq_offset);
                        LOG_INFO(TRACE_MODULE_APP, "CH Freq Offset %d\r\n", freq_offset_filter);
#if XTAL_AUTO_TUNE_EN
                        int32_t ppm = freq_offset_filter / (int32_t)(ch_center_freq_map[uwb_app_config.ppdu_params.ch_num] * 1e-6);
                        calib_xtal38m4_load_cap_auto_tune(ppm);
#endif
                    }
                    else if ((ind->ranging_stage == RANGING_DM_RX) && (ind->rx_len))
                    {
                        LOG_INFO(TRACE_MODULE_APP, "[RX][%d][%u] DM\r\n", ind->slot_idx, ind->rx_len);

                        if (uwb_dm_rx_is_done())
                        {
#if UWB_SERIAL_DATA_TRANSFER_EN
                            uwb_serial_rx_msg_update(uwb_dm_rx_pkt());
#endif
                        }
                    }

#if DM_PRINT_PAYLOAD_EN
                    if (ind->rx_len)
                    {
                        LOG_INFO(TRACE_MODULE_APP, " ");
                        for (uint8_t i = 0; i < ind->rx_len; i++)
                        {
                            LOG_INFO(TRACE_NO_OPTION | TRACE_MODULE_APP, "%02x ", ind->rx_data[i]);
                        }
                        LOG_INFO(TRACE_NO_OPTION | TRACE_MODULE_APP, "\r\n");
                    }
#endif
                }
                else
                {
                    LOG_INFO(TRACE_MODULE_APP, "[RX][%d] Fail  0x%04x\r\n", ind->slot_idx, ind->status);
                }
            }
            break;

            default:
                break;
        }
#if UWB_SERIAL_DATA_TRANSFER_EN
        uwb_serial_tx_msg_check();
        uwb_serial_rx_msg_check();
#endif
    }
    // Handle events
    else
    {
    }
}