/*
 * Copyright (c) 2023 HPMicro
 *
 * SPDX-License-Identifier: BSD-3-Clause
 *
 */

#include "board.h"
#include "i2s_master.h"
#include "hpm_i2s_drv.h"
#ifdef HPMSOC_HAS_HPMSDK_DMAV2
#include "hpm_dmav2_drv.h"
#else
#include "hpm_dma_drv.h"
#endif
#include "hpm_dmamux_drv.h"
#include "hpm_dao_drv.h"

/* DMA transfer flags */
volatile bool dma_transfer_done = false;
volatile bool dma_transfer_error;

void isr_dma(void)
{
    uint32_t stat;

    stat = dma_check_transfer_status(APP_DMA, DMA_CHANNEL);
    dma_clear_transfer_status(APP_DMA, DMA_CHANNEL);

    if (stat & DMA_CHANNEL_STATUS_TC) {
        dma_transfer_done = true;
    } else {
        dma_transfer_error = true;
    }
}
SDK_DECLARE_EXT_ISR_M(BOARD_APP_HDMA_IRQ, isr_dma)

void dma_transfer_config(uint32_t size)
{
    dma_channel_config_t ch_config = {0};

    intc_m_enable_irq_with_priority(BOARD_APP_HDMA_IRQ, ISR_PRIORITY_LEVEL);

    dma_default_channel_config(APP_DMA, &ch_config);

    ch_config.src_addr = (uint32_t)((uint8_t *)&I2S_MASTER->RXD[I2S_MASTER_RX_LINE] + sizeof(uint16_t));
    ch_config.dst_addr = (uint32_t)((uint8_t *)&I2S_DAO->TXD[I2S_DAO_DATA_LINE] + sizeof(uint16_t));
    ch_config.src_width = DMA_TRANSFER_WIDTH_HALF_WORD;
    ch_config.dst_width = DMA_TRANSFER_WIDTH_HALF_WORD;
    ch_config.src_addr_ctrl = DMA_ADDRESS_CONTROL_FIXED;
    ch_config.dst_addr_ctrl = DMA_ADDRESS_CONTROL_FIXED;
    ch_config.size_in_byte = size;
    ch_config.src_mode = DMA_HANDSHAKE_MODE_HANDSHAKE;
    ch_config.src_burst_size = DMA_NUM_TRANSFER_PER_BURST_2T;

    if (status_success != dma_setup_channel(APP_DMA, DMA_CHANNEL, &ch_config, true)) {
        printf(" dma setup channel failed\n");
        return;
    }
    dmamux_config(BOARD_APP_DMAMUX, DMA_MUX_CHANNEL, DMA_MUX_SRC, true);
}

void i2s_master_config(void)
{
    i2s_config_t i2s_config;
    i2s_transfer_config_t transfer;
    uint32_t i2s_mclk_hz;

    /* Config I2S interface */
    i2s_get_default_config(I2S_MASTER, &i2s_config);
    i2s_config.tx_fifo_threshold = I2S_MASTER_FIFO_THR;
    i2s_config.rx_fifo_threshold = I2S_MASTER_FIFO_THR;
    i2s_config.enable_mclk_out = true;
    i2s_init(I2S_MASTER, &i2s_config);

    i2s_get_default_transfer_config(&transfer);
    transfer.data_line = I2S_MASTER_RX_LINE;
    transfer.sample_rate = I2S_SAMPLE_RATE_HZ;
    transfer.audio_depth = I2S_SAMPLE_BITDEPTH;
    transfer.channel_length = I2S_CHANNEL_LENGTH;

    transfer.master_mode = true;
    i2s_mclk_hz = clock_get_frequency(BOARD_APP_I2S_CLK_NAME);
    if (status_success != i2s_config_transfer(I2S_MASTER, i2s_mclk_hz, &transfer)) {
        printf("I2S config failed!\n");
        while (1) {

        }
    }

    i2s_enable_rx_dma_request(I2S_MASTER);
    i2s_start(I2S_MASTER);
}

void i2s_dao_config(uint32_t sample_rate, uint8_t audio_depth, uint8_t channel_num)
{
    uint32_t i2s_mclk_hz;
    i2s_config_t i2s_config;
    i2s_transfer_config_t transfer;
    dao_config_t dao_config;

    i2s_get_default_config(I2S_DAO, &i2s_config);
    i2s_config.tx_fifo_threshold = 2;
    i2s_init(I2S_DAO, &i2s_config);

    /*
     * config transfer for DAO
     */
    i2s_get_default_transfer_config_for_dao(&transfer);
    transfer.sample_rate = sample_rate;
    transfer.audio_depth = audio_depth;
    transfer.channel_num_per_frame = channel_num;

    i2s_mclk_hz = clock_get_frequency(I2S_DAO_CLK_NAME);

    if (I2S_DAO_MCLK_FREQ_IN_HZ != i2s_mclk_hz) {
        printf("MCLK error for I2S_DAO!\n");
    }

    if (status_success != i2s_config_tx(I2S_DAO, i2s_mclk_hz, &transfer)) {
        printf("I2S config failed for I2S_DAO\n");
        while (1) {

        }
    }

    dao_get_default_config(HPM_DAO, &dao_config);
    dao_init(HPM_DAO, &dao_config);

    i2s_start(I2S_DAO);
    dao_start(HPM_DAO);
}

int main(void)
{
    board_init();

    board_init_i2s_clock(I2S_MASTER);
    board_init_dao_clock();

    board_init_i2s_pins(I2S_MASTER);
    board_init_dao_pins();

    printf("I2S Master example\n");

    i2s_master_config();
    i2s_dao_config(I2S_SAMPLE_RATE_HZ, I2S_SAMPLE_BITDEPTH, I2S_SAMPLE_CHANNEL);

    while (1) {
        dma_transfer_done = false;
        dma_transfer_config(DMA_TRANSFER_SIZE);
        while (dma_transfer_done == false) {

        }
    }

    return 0;
}