/*
 * 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_wdt.h"
#include "mk_reset.h"
#include "mk_gpio.h"
#include "mk_misc.h"
#include "mk_sleep_timer.h"
#include "mk_power.h"
#include "mk_uwb.h"
#include "mk_calib.h"
#include "mk_flash.h"

#include "board.h"

#include "pal_sys.h"
#include "wsf_os.h"
#include "wsf_timer.h"
#include "wsf_buf.h"
#include "wsf_nvm.h"

#include "app.h"
#include "ranging_fira.h"
#include "uwb_api.h"
#include "lib_ranging.h"
#include "libc_rom.h"

#define INITIATOR_ADDR (0xAAA1)
#define RESPONDER0_ADDR (0xAAA2)
#define RESPONDER1_ADDR (0xAAA3)
#define RESPONDER_NUM (1)

#ifndef ANT_DELAY_CAL_EN
#define ANT_DELAY_CAL_EN (0)
#endif

static uint32_t session_id = 0x0001;
static uint8_t ranging_on_flag = 0;

//*****************************************************************************
//
// WSF buffer pools.
//
//*****************************************************************************
#define WSF_BUF_POOLS 5

// Default pool descriptor.
static wsfBufPoolDesc_t poolDescriptors[WSF_BUF_POOLS] = {
    {32, 26}, {64, 24}, {128, 4}, {256 + 32, 4}, {1024 + 32, 2},
};

static void sleep_timer_callback(void *dev, uint32_t time)
{
    // LOG_INFO(TRACE_MODULE_APP, "Wake up by sleep timer %d\r\n", time);
}

static void app_ranging_report_callback(void *report)
{
    struct RANGE_DATA_T *rpt = (struct RANGE_DATA_T *)report;
    // LOG_INFO(TRACE_MODULE_APP, "Report measurements number %d\r\n", rpt->measurements_num);
    for (uint8_t i = 0; i < RESPONDER_NUM; i++)
    {
        if ((rpt->mac_addr_mode == MAC_ADDR_SHORT) && (rpt->measurements[i].status == STATUS_OK))
        {
            LOG_INFO(TRACE_MODULE_APP, "Peer %X, Distance %ucm\r\n", READ_SHORT(rpt->measurements[i].mac_addr), rpt->measurements[i].distance);
        }
    }
}

//
// Button Handlers
//
static void GPIO_UserHandler(enum IO_PIN_T pin)
{
    // LOG_INFO(TRACE_MODULE_APP, "GPIO Interrupt happen\r\n");

    if (pin == BOARD_SW_1)
    {
        app_button_event_set(APP_BUTTON1_EVT);
    }
}

void app_process_handle(uint8_t msg_id, const void *param)
{
    switch (msg_id)
    {
        case APP_DEBOUNCE_TIMER1_MSG:
        {
            if (gpio_pin_get_val(BOARD_SW_1))
            {
                return;
            }

            LOG_INFO(TRACE_MODULE_APP, "DEBOUNCE TIMER1\r\n");

            if (ranging_on_flag)
            {
                uwbapi_session_stop(session_id);
                ranging_on_flag = 0;
            }
            else
            {
                uwbapi_session_start(session_id, app_ranging_report_callback);
                ranging_on_flag = 1;
            }
        }
        break;

        default:
            break;
    }
}

static void board_init(void)
{
    // Clock configuration
    board_clock_run();
    // Pin configuration
    board_pins_config();
    // Trace configuration
    board_debug_console_open(TRACE_PORT_UART0);
    // Reset reason
    reset_cause_get();
    reset_cause_clear();

    // Load calibration parameters from NVM
    uint32_t internal_flash = (REG_READ(0x40000018) >> 17) & 0x1;
    uint32_t external_flash = (REG_READ(0x40010030) >> 28) & 0x3;
    if (internal_flash || external_flash == 1)
    {
        WsfNvmInit();
        board_calibration_params_load();
        flash_close(FLASH_ID0);
    }
    else
    {
        board_calibration_params_default();
    }

    // Chip calibration
    calib_chip();

    // Configure IO_02 for role selection
    gpio_open();
    gpio_pin_set_dir(IO_PIN_2, GPIO_DIR_IN, 0);
    io_pull_set(IO_PIN_2, IO_PULL_UP, IO_PULL_UP_LEVEL1);

    board_led_init();
    board_led_on(BOARD_LED_1);
    board_button_init(GPIO_UserHandler);
    board_configure();
}

int main(void)
{
    // Initialize MCU system
    board_init();

    // Disable watchdog timer
    wdt_close(WDT_ID0);
    LOG_INFO(TRACE_MODULE_APP, "Ranging FiRa example\r\n");

    // Platform init for WSF
    PalSysInit();

    // Initialize os

    //
    // Set up timers for the WSF scheduler.
    //
    WsfOsInit();
    WsfTimerInit();
    sys_tick_callback_set(WsfTimerUpdateTicks);

    //
    // Initialize a buffer pool for WSF dynamic memory needs.
    //
    uint32_t wsfBufMemLen = WsfBufInit(WSF_BUF_POOLS, poolDescriptors);

    if (wsfBufMemLen > FREE_MEM_SIZE)
    {
        LOG_INFO(TRACE_MODULE_APP, "Memory pool is not enough %d\r\n", wsfBufMemLen - FREE_MEM_SIZE);
    }

    //
    // Create app task
    //
    wsfHandlerId_t handlerId = WsfOsSetNextHandler(app_handler);
    app_init(handlerId);

    //
    // Create ranging task
    //
    handlerId = WsfOsSetNextHandler(ranging_handler);
    ranging_init(handlerId);

    uwb_open();

    // set advanced parameters
    struct PHY_ADV_CONFIG_T adv_config =
    {
        // RPM0: 40, RPM3: 60
        .thres_fap_detect = 40,
        // RPM0: 4, RPM3: 8
        .nth_scale_factor = 4,
        // RFrame SP0: 0/1, Others: 0/1/2/3
        .ranging_performance_mode = 0,
#if RX_ANT_PORTS_NUM == 4
        .skip_weakest_port_en = 1,
#else
        .skip_weakest_port_en = 0,
#endif
    };
    phy_adv_params_configure(&adv_config);

    // which RX ports will be used for AoA/PDoA
    phy_rx_ant_mode_set(RX_ANT_PORTS_COMBINATION);

    uwbs_init();
    uwb_app_config.ranging_flow_mode = (uint8_t)(RANGING_FLOW_FIRA);
    uwb_app_config.filter_en = (uint8_t)(FILTER_EN);
    uwb_app_config.session_param.tx_power_level = board_param.tx_power_fcc[CALIB_CH(uwb_app_config.ppdu_params.ch_num)];
    uwb_app_config.ppdu_params.rx_ant_id = (uint8_t)(RX_MAIN_ANT_PORT);

    // Initialize ranging session
    uwbapi_session_init(session_id, SESSION_TYPE_RANGING);

    // Initialize ranging parameters
    struct APP_CFG_PARAM_T param = {0};
    param.ch_num = UWB_CH_NUM;
    param.prf_mode = UWB_MEAN_PRF;
    param.preamble_code_index = UWB_PREAMBLE_CODE_IDX;
    param.preamble_duration = UWB_PREAMBLE_DURATION;
    param.sfd_id = UWB_SFD_ID;
    param.psdu_data_rate = UWB_PSDU_DATA_RATE;
    param.sts_segment_num = UWB_STS_SEGMENT_NUM;
    param.sts_segment_len = UWB_STS_SEGMENT_LEN;
    param.aoa_result_req = 0;
    param.rframe_config = UWB_RFRAME_TYPE;
    param.sts_config = STS_STATIC;
    memcpy(param.session_key, (uint8_t *)SESSION_KEY_DYNAMIC_STS_TEST, 16);
    param.key_rotation = 0;
    param.key_rotation_rate = 8;
#if ANT_DELAY_CAL_EN
    param.ranging_round_usage = DS_TWR_DEFERRED;
#else
    param.ranging_round_usage = DS_TWR_DEFERRED;
    // param.ranging_round_usage = SS_TWR_DEFERRED;
    // param.ranging_round_usage = DS_TWR;
    // param.ranging_round_usage = SS_TWR;
#endif
    param.mac_address_mode = ARRD_SHORT_USE_SHORT;
    param.controlees_num = RESPONDER_NUM;
    param.multi_node_mode = param.controlees_num > 1 ? ONE_TO_MANY : UNICAST;
    param.result_report_config = 0x0F;
    param.ranging_round_control = 0x3; // bit0: ranging result report message enable
    param.hopping_mode = 0x00;
    param.stride_length = 0x00;

    // Load local & peer short address
    uint16_t local_addr = 0;
    uint16_t peer_addr = 0;

#if ANT_DELAY_CAL_EN
    param.device_role = DEV_ROLE_INITIATOR;
    local_addr = INITIATOR_ADDR;
    peer_addr = RESPONDER0_ADDR;

    param.src_dev_mac_addr[0] = local_addr & 0xff;
    param.src_dev_mac_addr[1] = (local_addr >> 8) & 0xff;
    param.dst_dev_mac_addr[0] = peer_addr & 0xff;
    param.dst_dev_mac_addr[1] = (peer_addr >> 8) & 0xff;
#else
    if (gpio_pin_get_val(IO_PIN_2) == 0)
    {
        param.device_role = DEV_ROLE_INITIATOR;
        local_addr = INITIATOR_ADDR;
        if (board_param.flag & (1 << BOARD_LOCAL_SHORT_ADDR))
        {
            local_addr = board_param.local_short_addr;
        }
        param.src_dev_mac_addr[0] = local_addr & 0xff;
        param.src_dev_mac_addr[1] = (local_addr >> 8) & 0xff;

        peer_addr = RESPONDER0_ADDR;
        param.dst_dev_mac_addr[0] = peer_addr & 0xff;
        param.dst_dev_mac_addr[1] = (peer_addr >> 8) & 0xff;
        peer_addr = RESPONDER1_ADDR;
        param.dst_dev_mac_addr[2] = peer_addr & 0xff;
        param.dst_dev_mac_addr[3] = (peer_addr >> 8) & 0xff;
    }
    else
    {
        param.device_role = DEV_ROLE_RESPONDER;
        local_addr = RESPONDER0_ADDR;
        if (board_param.flag & (1 << BOARD_LOCAL_SHORT_ADDR))
        {
            local_addr = board_param.local_short_addr;
        }

        param.src_dev_mac_addr[0] = local_addr & 0xff;
        param.src_dev_mac_addr[1] = (local_addr >> 8) & 0xff;

        peer_addr = INITIATOR_ADDR;
        if (board_param.flag & (1 << BOARD_PEER_SHORT_ADDR))
        {
            peer_addr = board_param.peer_short_addr;
        }

        param.dst_dev_mac_addr[0] = peer_addr & 0xff;
        param.dst_dev_mac_addr[1] = (peer_addr >> 8) & 0xff;
    }
#endif

    param.device_type = (param.device_role == DEV_ROLE_INITIATOR ? DEV_TYPE_CONTROLLER : DEV_TYPE_CONTROLEE);

    if (param.ranging_round_usage == DS_TWR_DEFERRED)
    {
        param.slots_per_round = SLOT_NUM_PER_ROUND(param.controlees_num);
    }
    else if (param.ranging_round_usage == SS_TWR_DEFERRED)
    {
        param.slots_per_round = SLOT_NUM_PER_ROUND_FOR_SS_TWR(param.controlees_num);
    }
    else if (param.ranging_round_usage == DS_TWR)
    {
        param.slots_per_round = SLOT_NUM_PER_ROUND_FOR_NON_DEFERRED_DS_TWR(param.controlees_num);
    }
    else if (param.ranging_round_usage == SS_TWR)
    {
        param.slots_per_round = SLOT_NUM_PER_ROUND_FOR_NON_DEFERRED_SS_TWR(param.controlees_num);
    }
    param.slot_duration = UWB_RANGING_SLOT_DURATION;
    param.ranging_interval = UWB_RANGING_INTERVAL;

    // Configure ranging parameters
    uwbapi_session_set_app_config(session_id, &param);

    // Start ranging
    uwbapi_session_start(session_id, app_ranging_report_callback);
    ranging_on_flag = 1;

    // Initialize low power mode
    power_init();

#if LOW_POWER_EN
    power_mode_request(POWER_UNIT_USER, POWER_MODE_POWER_DOWN);
#else
    power_mode_request(POWER_UNIT_USER, POWER_MODE_SLEEP);
#endif

    // Enable sleep timer
    sleep_timer_open(true, SLEEP_TIMER_MODE_ONESHOT, sleep_timer_callback);

    while (1)
    {
        wsfOsDispatcher();
        power_manage();
    }
}

void app_restore_from_power_down(void)
{
}