/******************************************************************************
 * Copyright (C) 2024 , Inc.(Gmbh) or its affiliates.
 *
 * All Rights Reserved.
 *
 * @file Flash.c
 *
 * @par dependencies
 * - Flash.h
 *
 * @author 代码移植
 *
 * @brief Functions related to reading and writing in the chip's flash area.
 *
 * Processing flow:
 *
 * call directly.
 *
 * @version V1.0 2024-09-13
 *
 * @note 1 tab == 4 spaces!
 *
 *****************************************************************************/
/* Includes ------------------------------------------------------------------*/
#include "Flash.h"
/* Private typedef -----------------------------------------------------------*/
/* Private define ------------------------------------------------------------*/
/* Private macro -------------------------------------------------------------*/
/* Private variables ---------------------------------------------------------*/
/* Private function prototypes -----------------------------------------------*/
static uint16_t STMFLASH_GetFlashSector(u32 addr);
/* Private functions ---------------------------------------------------------*/
/* extern variables ---------------------------------------------------------*/

const uint32_t Flash_Sectorsize[12] = \
{
    FLASH_Sector_0,  // Sector1
    FLASH_Sector_1,  // Sector2
    FLASH_Sector_2,  // Sector3
    FLASH_Sector_3,  // Sector4
    FLASH_Sector_4,  // Sector5
    FLASH_Sector_5,  // Sector6
    FLASH_Sector_6,  // Sector7
    FLASH_Sector_7,  // Sector8
    FLASH_Sector_8,  // Sector9
    FLASH_Sector_9,  // Sector10
    FLASH_Sector_10, // Sector11
    FLASH_Sector_11  // Sector12
};

/**
 * @brief  This function is used to erase the flash area.
 * @param  None
 * @retval 0: Success
 *         1: Fail
 */
uint8_t Flash_erase(u32 addr, u32 size)
{
    // 将App代码的所有涉及到的扇区全部擦除

    /* 判断size在几个扇区内 */
//    FLASH_Status ret = 1;
    uint32_t flash_start_sector = 0;
    uint32_t flash_end_sector = 0;
    if (size == 0U)
    {
       return 1U;
    }
    flash_start_sector = STMFLASH_GetFlashSector(addr);
    flash_end_sector = STMFLASH_GetFlashSector(addr + size - 1u);

    for (uint8_t i = 0; i < 12; i++)
    {
        if ((Flash_Sectorsize[i]) >= flash_start_sector && (Flash_Sectorsize[i]) <= flash_end_sector)
        {
            if (EreaseAppSector(Flash_Sectorsize[i]) != FLASH_COMPLETE) return 1;
        }
    }
    return 0;
}

// 通过地址获取扇区位置
static uint16_t STMFLASH_GetFlashSector(u32 addr)
{
    if (addr < ADDR_FLASH_SECTOR_1) return FLASH_Sector_0;
    else if (addr < ADDR_FLASH_SECTOR_2) return FLASH_Sector_1;
    else if (addr < ADDR_FLASH_SECTOR_3) return FLASH_Sector_2;
    else if (addr < ADDR_FLASH_SECTOR_4) return FLASH_Sector_3;
    else if (addr < ADDR_FLASH_SECTOR_5) return FLASH_Sector_4;
    else if (addr < ADDR_FLASH_SECTOR_6) return FLASH_Sector_5;
    else if (addr < ADDR_FLASH_SECTOR_7) return FLASH_Sector_6;
    else if (addr < ADDR_FLASH_SECTOR_8) return FLASH_Sector_7;
    else if (addr < ADDR_FLASH_SECTOR_9) return FLASH_Sector_8;
    else if (addr < ADDR_FLASH_SECTOR_10) return FLASH_Sector_9;
    else if (addr < ADDR_FLASH_SECTOR_11) return FLASH_Sector_10;
    return FLASH_Sector_11;
}

void Flash_Unlock(void)
{
    FLASH_Unlock();
    while (FLASH_GetStatus() == FLASH_BUSY);
}

void Flash_Lock(void)
{
    FLASH_Lock();
}

// 擦除APP区域的数据
// f4是按扇区操作，计划将app放在扇区6 FLASH_Sector_6，备份app放在扇区7
FLASH_Status EreaseAppSector(uint32_t FLASH_Sector)
{
    Flash_Unlock();
    FLASH_Status FLASHStatus = FLASH_EraseSector(FLASH_Sector, VoltageRange_3);
    Flash_Lock();
    return FLASHStatus;
}

void Flash_Write(uint32_t address, uint32_t data)
{
    // 解锁Flash
    Flash_Unlock();

    // 清除所有标志位
    // FLASH_ClearFlag(FLASH_FLAG_EOP | FLASH_FLAG_PGERR | FLASH_FLAG_WRPERR);

    // 编程一个字（32位）
    FLASH_Status status = FLASH_ProgramWord(address, data);

    // 检查编程是否成功
    if (status == FLASH_COMPLETE)
    {
        // 数据写入成功
        // log_d("Flash_Write success");
    }
    else
    {
        // 数据写入失败
        // 在这里添加错误处理代码
        // log_e("Flash_Write fail");
    }

    // 锁定Flash
    Flash_Lock();
}