/*
 * Copyright (c) 2023, Artinchip Technology Co., Ltd
 *
 * SPDX-License-Identifier: Apache-2.0
 *
 * Authors: xuan.wen <xuan.wen@artinchip.com>
 */

#include <rtdevice.h>
#include <drv_qspi.h>
#include <aic_log.h>
#include "spi_flash_sfud.h"

#if defined(RT_USING_SFUD)

#define SPI_FLASH_DEVICE_NAME "qspi01"

#define USING_NOR_FLASH_DEV_NAME "norflash0"

#ifndef RT_SFUD_DEFAULT_SPI_CFG

#ifndef RT_SFUD_SPI_MAX_HZ
#define RT_SFUD_SPI_MAX_HZ 50000000
#endif

/* read the JEDEC SFDP command must run at 50 MHz or less */
#define RT_SFUD_DEFAULT_SPI_CFG                  \
{                                                \
    .mode = RT_SPI_MODE_0 | RT_SPI_MSB,          \
    .data_width = 8,                             \
    .max_hz = RT_SFUD_SPI_MAX_HZ,                \
}
#endif /* RT_SFUD_DEFAULT_SPI_CFG */

#ifdef SFUD_USING_QSPI
#define RT_SFUD_DEFAULT_QSPI_CFG                 \
{                                                \
    RT_SFUD_DEFAULT_SPI_CFG,                     \
    .medium_size = 0x800000,                     \
    .ddr_mode = 0,                               \
    .qspi_dl_width = 4,                          \
}
#endif /* SFUD_USING_QSPI */

int rt_hw_spi_flash_with_sfud_init(void)
{
    struct rt_spi_configuration cfg = RT_SFUD_DEFAULT_SPI_CFG;
    rt_err_t ret;
    rt_spi_flash_device_t flash_dev;

    ret = aic_qspi_bus_attach_device("qspi0", SPI_FLASH_DEVICE_NAME, 0, 4,
                                     RT_NULL, RT_NULL);
    if (ret < 0) {
        pr_err("attach qspi device failed.\n");
        return RT_ERROR;
    }
#ifndef SFUD_USING_QSPI
    cfg.max_hz = AIC_QSPI0_DEVICE_SPINOR_FREQ;
    flash_dev = rt_sfud_flash_probe_ex(USING_NOR_FLASH_DEV_NAME,
                                       SPI_FLASH_DEVICE_NAME, &cfg, RT_NULL);
#else
    struct rt_qspi_configuration qspi_cfg = RT_SFUD_DEFAULT_QSPI_CFG;

    qspi_cfg.parent.max_hz = AIC_QSPI0_DEVICE_SPINOR_FREQ;
    flash_dev = rt_sfud_flash_probe_ex(USING_NOR_FLASH_DEV_NAME,
                                       SPI_FLASH_DEVICE_NAME, &cfg, &qspi_cfg);
#endif
    if (flash_dev == RT_NULL) {
        pr_err("sfud flash probe  failed.\n");
        return RT_ERROR;
    };

    return RT_EOK;
}
INIT_PREV_EXPORT(rt_hw_spi_flash_with_sfud_init);
#endif