/*
 * Copyright (c) 2006-2022, RT-Thread Development Team
 *
 * SPDX-License-Identifier: Apache-2.0
 *
 * Change Logs:
 * Date           Author            Notes
 * 2020-07-27     thread-liu        the first version
 * 2022-04-13     Miaowulue         fit openmv-h7plus
 */
#include "board.h"

#ifdef BSP_USING_DCMI

#define DRV_DEBUG
#define LOG_TAG             "drv.dcmi"
#include <drv_log.h>

struct stm32_dcmi
{
    struct rt_device dev;
};
static struct stm32_dcmi rt_dcmi = {0};
DCMI_HandleTypeDef dcmi    = {0};
DMA_HandleTypeDef hdma_dcmi = {0};

extern void jpeg_data_process(void);

static void rt_hw_dmci_dma_init(void)
{
    __HAL_RCC_DMA1_CLK_ENABLE();

    hdma_dcmi.Instance = DMA1_Stream3;
    hdma_dcmi.Init.Request = DMA_REQUEST_DCMI;
    hdma_dcmi.Init.Direction = DMA_PERIPH_TO_MEMORY;
    hdma_dcmi.Init.PeriphInc = DMA_PINC_DISABLE;
    hdma_dcmi.Init.MemInc = DMA_MINC_ENABLE;
    hdma_dcmi.Init.PeriphDataAlignment = DMA_PDATAALIGN_WORD;
    hdma_dcmi.Init.MemDataAlignment = DMA_MDATAALIGN_WORD;
    hdma_dcmi.Init.Mode = DMA_CIRCULAR;
    hdma_dcmi.Init.Priority = DMA_PRIORITY_HIGH;
    hdma_dcmi.Init.FIFOMode = DMA_FIFOMODE_ENABLE;
    hdma_dcmi.Init.FIFOThreshold = DMA_FIFO_THRESHOLD_FULL;
    hdma_dcmi.Init.MemBurst = DMA_MBURST_SINGLE;
    hdma_dcmi.Init.PeriphBurst = DMA_PBURST_SINGLE;
    if (HAL_DMA_Init(&hdma_dcmi) != HAL_OK)
    {
      Error_Handler();
    }

    __HAL_LINKDMA(&dcmi, DMA_Handle, hdma_dcmi);

    HAL_NVIC_SetPriority(DMA1_Stream3_IRQn, 0x00, 0);
    HAL_NVIC_EnableIRQ(DMA1_Stream3_IRQn);
}

void rt_hw_dcmi_dma_config(rt_uint32_t dst_addr1, rt_uint32_t dst_addr2, rt_uint16_t len)
{
    HAL_DMAEx_MultiBufferStart(&hdma_dcmi, (rt_uint32_t)&DCMI->DR, dst_addr1, dst_addr2, len);
    __HAL_DMA_ENABLE_IT(&hdma_dcmi, DMA_IT_TC);
}

static rt_err_t rt_hw_dcmi_init(DCMI_HandleTypeDef *device)
{
    RT_ASSERT(device != RT_NULL);

    dcmi.Instance = DCMI;
    dcmi.Init.SynchroMode = DCMI_SYNCHRO_HARDWARE;
    dcmi.Init.PCKPolarity = DCMI_PCKPOLARITY_RISING;
    dcmi.Init.VSPolarity = DCMI_VSPOLARITY_LOW;
    dcmi.Init.HSPolarity = DCMI_HSPOLARITY_LOW;
    dcmi.Init.CaptureRate = DCMI_CR_ALL_FRAME;
    dcmi.Init.ExtendedDataMode = DCMI_EXTEND_DATA_8B;
    dcmi.Init.JPEGMode = DCMI_JPEG_ENABLE;
    dcmi.Init.ByteSelectMode = DCMI_BSM_ALL;
    dcmi.Init.ByteSelectStart = DCMI_OEBS_ODD;
    dcmi.Init.LineSelectMode = DCMI_LSM_ALL;
    dcmi.Init.LineSelectStart = DCMI_OELS_ODD;
    if (HAL_DCMI_Init(device) != HAL_OK)
    {
        LOG_E("dcmi init error!");
        return RT_ERROR;
    }

    DCMI->IER = 0x0;

    __HAL_DCMI_ENABLE_IT(device, DCMI_IT_FRAME);
    __HAL_DCMI_ENABLE(device);

    rt_hw_dmci_dma_init();

    return RT_EOK;
}

void DCMI_IRQHandler(void)
{
    /* enter interrupt */
    rt_interrupt_enter();

    HAL_DCMI_IRQHandler(&dcmi);

    /* leave interrupt */
    rt_interrupt_leave();
}

/* Capture a frame of the image */
void HAL_DCMI_FrameEventCallback(DCMI_HandleTypeDef *hdcmi)
{
    /* enter interrupt */
    rt_interrupt_enter();

    jpeg_data_process();
    __HAL_DCMI_ENABLE_IT(&dcmi,DCMI_IT_FRAME);
    /* leave interrupt */
    rt_interrupt_leave();
}

void DMA1_Stream3_IRQHandler(void)
{
    extern void rt_hw_camera_rx_callback(void);
    /* enter interrupt */
    rt_interrupt_enter();

    if(__HAL_DMA_GET_FLAG(&hdma_dcmi, DMA_FLAG_TCIF3_7)!=RESET)
    {
        __HAL_DMA_CLEAR_FLAG(&hdma_dcmi, DMA_FLAG_TCIF3_7);
        rt_hw_camera_rx_callback();
    }

    /* leave interrupt */
    rt_interrupt_leave();
}

static rt_err_t rt_dcmi_init(rt_device_t dev)
{
    RT_ASSERT(dev != RT_NULL);
    rt_err_t result = RT_EOK;

    result = rt_hw_dcmi_init(&dcmi);
    if (result != RT_EOK)
    {
        return result;
    }

    return result;
}

static rt_err_t rt_dcmi_open(rt_device_t dev, rt_uint16_t oflag)
{
    RT_ASSERT(dev != RT_NULL);

    return RT_EOK;
}

static rt_err_t rt_dcmi_close(rt_device_t dev)
{
    RT_ASSERT(dev != RT_NULL);

    return RT_EOK;
}

static rt_err_t rt_dcmi_control(rt_device_t dev, int cmd, void *args)
{
    RT_ASSERT(dev != RT_NULL);

    return RT_EOK;
}

static rt_size_t rt_dcmi_read(rt_device_t dev, rt_off_t pos, void *buffer, rt_size_t size)
{
    RT_ASSERT(dev != RT_NULL);

    return RT_EOK;
}

static rt_size_t rt_dcmi_write(rt_device_t dev, rt_off_t pos, const void *buffer, rt_size_t size)
{
    RT_ASSERT(dev != RT_NULL);

    return RT_EOK;
}

int dcmi_init(void)
{
    int ret = 0;
    rt_device_t dcmi_dev = RT_NULL;

    rt_dcmi.dev.type      = RT_Device_Class_Miscellaneous;
    rt_dcmi.dev.init      = rt_dcmi_init;
    rt_dcmi.dev.open      = rt_dcmi_open;
    rt_dcmi.dev.close     = rt_dcmi_close;
    rt_dcmi.dev.read      = rt_dcmi_read;
    rt_dcmi.dev.write     = rt_dcmi_write;
    rt_dcmi.dev.control   = rt_dcmi_control;
    rt_dcmi.dev.user_data = RT_NULL;

    ret = rt_device_register(&rt_dcmi.dev, "dcmi", RT_DEVICE_FLAG_RDWR | RT_DEVICE_FLAG_REMOVABLE | RT_DEVICE_FLAG_STANDALONE);
    if(ret != RT_EOK)
    {
        LOG_E("dcmi registered fail!\n\r");
        return -RT_ERROR;
    }

    LOG_I("dcmi init success!");

    return RT_EOK;
}
INIT_BOARD_EXPORT(dcmi_init);

#endif