/*
 * 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_flash.h"
#include "mk_trace.h"
#include "mk_reset.h"
#include "mk_calib.h"
#include "mk_wdt.h"
#include "libc_rom.h"
#include "board.h"

static uint8_t test_buffer[FLASH_PAGE_SIZE] = {0};
static uint8_t test_buffer_rd[FLASH_PAGE_SIZE] = {0};

// test pages [0, 2048]
#define TEST_PAGE_START (1024)
#define TEST_PAGE_END (1024 + 1)

int main(void)
{
    board_clock_run();
    board_pins_config();
    board_debug_console_open(TRACE_PORT_UART0);
    // Reset reason
    reset_cause_get();
    reset_cause_clear();

    // Chip calibration
    calib_chip();

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

    gpio_open();
    board_configure();

    sys_timer_close();
    sys_timer_open();

    flash_open(FLASH_ID0, NULL);

    // Set the buffer
    for (uint32_t i = 0; i < FLASH_PAGE_SIZE; i++)
    {
        test_buffer[i] = i & 0xff;
    }

    // erase sector
    uint32_t sector_size = ((TEST_PAGE_END - TEST_PAGE_START) * FLASH_PAGE_SIZE) & (~(FLASH_SECTOR_SIZE - 1));
    if (((TEST_PAGE_END - TEST_PAGE_START) * FLASH_PAGE_SIZE) & (FLASH_SECTOR_SIZE - 1U))
    {
        sector_size += FLASH_SECTOR_SIZE;
    }

    flash_erase(FLASH_ID0, FLASH_BASE + TEST_PAGE_START * FLASH_PAGE_SIZE, sector_size);

    // read for check
    uint32_t err = 0;

    for (uint32_t i = TEST_PAGE_START; i < TEST_PAGE_END; i++)
    {
        for (uint32_t j = 0; j < FLASH_PAGE_SIZE; j++)
        {
            uint32_t addr = FLASH_BASE + FLASH_PAGE_SIZE * i + j;
            uint8_t val = *(uint8_t *)(addr);
            if (val != 0xFF)
            {
                LOG_INFO(TRACE_MODULE_APP, "R %02x : %02x, expected %02x\r\n", addr, val, 0xFF);
                err++;
            }
        }
    }

    if (err == 0)
    {
        LOG_INFO(TRACE_MODULE_APP, "Erase check pass!\r\n");
    }

    // write data to the page
    for (uint32_t i = TEST_PAGE_START; i < TEST_PAGE_END; i++)
    {
        flash_write(FLASH_ID0, FLASH_BASE + i * FLASH_PAGE_SIZE, test_buffer, FLASH_PAGE_SIZE);
        while (flash_check_busy(FLASH_ID0))
        {
        }
    }
    // read for check
    for (uint32_t i = TEST_PAGE_START; i < TEST_PAGE_END; i++)
    {
        flash_read(FLASH_ID0, FLASH_BASE + i * FLASH_PAGE_SIZE, test_buffer_rd, FLASH_PAGE_SIZE);
        while (flash_check_busy(FLASH_ID0))
        {
        }

        err = 0;
        for (uint32_t j = 0; j < FLASH_PAGE_SIZE; j++)
        {
            if (test_buffer_rd[j] != test_buffer[j])
            {
                uint32_t addr = FLASH_BASE + FLASH_PAGE_SIZE * i + j;
                LOG_INFO(TRACE_MODULE_APP, "R %02x : %02x, expected %02x\r\n", addr, test_buffer_rd[j], test_buffer[j]);
                err++;
            }
        }

        if (err == 0)
        {
            LOG_INFO(TRACE_MODULE_APP, "Page %d written data verify pass!\r\n", i);
        }

        // another way to verify data
        err = 0;
        for (uint32_t j = 0; j < FLASH_PAGE_SIZE; j++)
        {
            uint32_t addr = FLASH_BASE + FLASH_PAGE_SIZE * i + j;
            uint8_t val = *(uint8_t *)(addr);
            if (val != test_buffer[j])
            {
                LOG_INFO(TRACE_MODULE_APP, "R %02x : %02x, expected %02x\r\n", addr, val, test_buffer[j]);
                err++;
            }
        }

        if (err == 0)
        {
            LOG_INFO(TRACE_MODULE_APP, "Page %d XIP verify pass!\r\n", i);
        }
    }

    LOG_INFO(TRACE_MODULE_APP, "Flash example finished!\r\n");

    flash_close(FLASH_ID0);
    LOG_INFO(TRACE_MODULE_APP, "Flash close\r\n");

    while (1)
    {
    }
}