/******************************************************************************
 * Copyright (c) 2022 Telink Semiconductor (Shanghai) Co., Ltd. ("TELINK")
 * All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 *****************************************************************************/
/**	@page I2C
 *
 *	Introduction
 *	===============
 *	i2c support master mode or slave mode.
 *
 *	API Reference
 *	===============
 *	Header File: i2c.h
 */

#ifndef I2C_H
#define I2C_H
#include "gpio.h"

#include "analog.h"
#include "dma.h"
#include "reg_include/i2c_reg.h"

/**********************************************************************************************************************
 *                                         global constants                                                           *
 *********************************************************************************************************************/

/**********************************************************************************************************************
 *                                           global macro                                                             *
 *********************************************************************************************************************/

/**
 *  @brief  select pin as SDA and SCL of i2c
 */
typedef enum {
    I2C_GPIO_SDA_B3 = GPIO_PB3,
    I2C_GPIO_SDA_C2 = GPIO_PC2,
    I2C_GPIO_SDA_E2 = GPIO_PE2,
    I2C_GPIO_SDA_E3 = GPIO_PE3,
} i2c_sda_pin_e;

typedef enum {
    I2C_GPIO_SCL_B2 = GPIO_PB2,
    I2C_GPIO_SCL_C1 = GPIO_PC1,
    I2C_GPIO_SCL_E0 = GPIO_PE0,
    I2C_GPIO_SCL_E1 = GPIO_PE1,
} i2c_scl_pin_e;

typedef enum {
    I2C_RX_BUF_MASK = BIT(2),
    I2C_TX_BUF_MASK = BIT(3),
    I2C_TX_DONE_MASK = BIT(4),
    I2C_RX_DONE_MASK = BIT(5),
} i2c_irq_mask_e;

typedef enum {
    I2C_RX_BUFF_CLR = BIT(6),
    I2C_TX_BUFF_CLR = BIT(7),
} i2c_buff_clr_e;

typedef enum {
    I2C_TXDONE_STATUS = BIT(0),
    I2C_TX_BUF_STATUS = BIT(1),
    I2C_RXDONE_STATUS = BIT(2),
    I2C_RX_BUF_STATUS = BIT(3),
} i2c_irq_status_e;

typedef enum {
    I2C_TX_DONE_CLR = BIT(4),
} i2c_irq_clr_e;

/**
 * @brief      The function of this API is to determine whether the bus is busy.
 * @return     1:Indicates that the bus is busy. 0:Indicates that the bus is free
 */
static inline _Bool i2c_master_busy(void)
{
    return reg_i2c_mst & FLD_I2C_MST_BUSY;
}

/**
 * @brief      The function of this API is to Get the number of bytes in tx_buffer.
 * @return     The actual number of bytes in tx_buffer.
 */
static inline unsigned char i2c_get_tx_buf_cnt(void)
{
    return (reg_i2c_buf_cnt & FLD_I2C_TX_BUFCNT) >> 4;
}

/**
 * @brief      The function of this API is to Get the number of bytes in rx_buffer.
 * @return     The actual number of bytes in rx_buffer.
 */
static inline unsigned char i2c_get_rx_buf_cnt(void)
{
    return (reg_i2c_buf_cnt & FLD_I2C_RX_BUFCNT);
}

/**
 * @brief      The function of this API is to set the number of bytes to triggered the receive interrupt.
 *             Its default value is 4. We recommend setting it to 1 or 4.
 * @param[in]  cnt - the interrupt trigger level.
 * @return     none
 */
static inline void i2c_rx_irq_trig_cnt(unsigned char cnt)
{
    reg_i2c_trig &= (~FLD_I2C_RX_IRQ_TRIG_LEV);
    reg_i2c_trig |= cnt;
}

/**
 * @brief      The function of this interface is equivalent to
 *             that after the user finishes calling the write or read interface, the stop signal is not sent,
 * 			   and then the write or read command is executed again.
 *             The driver defaults that every write or read API will send a stop command at the end
 * @param[in]  en - Input parameters.Decide whether to disable the stop function after each write or read interface
 * @return     none
 */
void i2c_master_send_stop(unsigned char en);

/**
 * @brief      This function selects a pin port for I2C interface.
 * @param[in]  sda_pin - the pin port selected as I2C sda pin port.
 * @param[in]  scl_pin - the pin port selected as I2C scl pin port.
 * @return     none
 */
void i2c_set_pin(i2c_sda_pin_e sda_pin, i2c_scl_pin_e scl_pin);

/**
 * @brief      This function serves to enable i2c master function.
 * @return     none.
 */
void i2c_master_init(void);

/**
 * @brief      This function serves to enable i2c RX/TX/MASK_RX/MASK_TX  interrupt function.
 * @param[in]  mask - to select Interrupt type.
 * @return     none
 */
static inline void i2c_set_irq_mask(i2c_irq_mask_e mask)
{
    reg_i2c_sct0 |= mask;
}

/**
 * @brief      This function serves to disable i2c RX/TX/MASK_RX/MASK_TX  interrupt function.
 * @param[in]  mask - to select Interrupt type.
 * @return     none
 */
static inline void i2c_clr_irq_mask(i2c_irq_mask_e mask)
{
    reg_i2c_sct0 &= (~mask);
}

/**
 * @brief      This function serves to get i2c interrupt status.
 * @return     none
 *
 */
static inline unsigned char i2c_get_irq_status(i2c_irq_status_e status)
{
    return reg_i2c_irq_status & status;
}

/**
 * @brief      This function serves to clear i2c rx/tx fifo.
 * @param[in]  clr - to select Interrupt type.
 * @return     none
 */
static inline void i2c_clr_fifo(i2c_buff_clr_e clr)
{
    reg_i2c_status = clr;
}

/**
 * @brief      This function serves to clear i2c irq status.
 * @return     none
 */
static inline void i2c_clr_irq_status(i2c_irq_clr_e status)
{
    reg_i2c_irq_status = status;
}

/**
 * @brief      This function serves to enable slave mode.
 * @param[in]  id - the id of slave device.it contains write or read bit,the laster bit is write or read bit.
 *                       ID|0x01 indicate read. ID&0xfe indicate write.
 * @return     none
 */
void i2c_slave_init(unsigned char id);

/**
 *  @brief      The function of this API is to ensure that the data can be successfully sent out.
 *  @param[in]  id - to set the slave ID.for kite slave ID=0x5c,for eagle slave ID=0x5a.
 *  @param[in]  data - The data to be sent, The first three bytes can be set as the RAM address of the slave.
 *  @param[in]  len - This length is the total length, including both the length of the slave RAM address
 *                    and the length of the data to be sent.
 *  @return     0 : the master receive NACK after sending out the id and then send stop.
 *              1: the master sent the data successfully,(master does not detect NACK in data phase)
 */
unsigned char i2c_master_write(unsigned char id, unsigned char *data, unsigned char len);

/**
 * @brief      This function serves to read a packet of data from the specified address of slave device
 * @param[in]  id - to set the slave ID.for kite slave ID=0x5c,for eagle slave ID=0x5a.
 * @param[in]  data - Store the read data
 * @param[in]  len - The total length of the data read back.
 * @return     0 : the master receive NACK after sending out the id and then send stop.
 *             1: the master receive the data successfully.
 */
unsigned char i2c_master_read(unsigned char id, unsigned char *data, unsigned char len);

/**
 * @brief      This function serves to write data and restart read data.
 * @param[in]  id - to set the slave ID.for kite slave ID=0x5c,for eagle slave ID=0x5a.
 * @param[in]  wr_data - The data to be sent, The first three bytes can be set as the RAM address of the slave.
 * @param[in]  wr_len -  This length is the total length, including both the length of the slave RAM address
 *             and the length of the data to be sent.
 * @param[in]  rd_data - Store the read data
 * @param[in]  rd_len -  The total length of the data read back.
 * @return     0 : the master receive NACK after sending out the id and then send stop.
 *             1: the master receive the data successfully.
 */
unsigned char i2c_master_write_read(unsigned char id, unsigned char *wr_data, unsigned char wr_len,
                                    unsigned char *rd_data, unsigned char rd_len);

/**
 * @brief      The function of this API is just to write data to the i2c tx_fifo by DMA.
 * @param[in]  id - to set the slave ID.for kite slave ID=0x5c,for eagle slave ID=0x5a.
 * @param[in]  data - The data to be sent, The first three bytes represent the RAM address of the slave.
 * @param[in]  len - This length is the total length, including both the length of the slave RAM address
 *             and the length of the data to be sent.
 * @return     none.
 */
void i2c_master_write_dma(unsigned char id, unsigned char *data, unsigned char len);

/**
 * @brief      This function serves to read a packet of data from the specified address of slave device.
 * @param[in]  id - to set the slave ID.for kite slave ID=0x5c,for eagle slave ID=0x5a.
 * @param[in]  data - Store the read data
 * @param[in]  len - The total length of the data read back.
 * @return     none.
 */
void i2c_master_read_dma(unsigned char id, unsigned char *data, unsigned char len);

/**
 * @brief      This function serves to send a packet of data to master device.
 *             It will trigger after the master sends the read sequence.
 * @param[in]  data - the pointer of tx_buff.
 * @param[in]  len - The total length of the data .
 * @return     none.
 */
void i2c_slave_set_tx_dma(unsigned char *data, unsigned char len);

/**
 * @brief      This function serves to receive a packet of data from master device,
 *             It will trigger after the master sends the write sequence.
 * @param[in]  data - the pointer of rx_buff.
 * @param[in]  len  - The total length of the data.
 * @return     none.
 */
void i2c_slave_set_rx_dma(unsigned char *data, unsigned char len);

/**
 * @brief     This function serves to receive data.
 * @param[in]  data - the data need read.
 * @param[in]  len - The total length of the data
 * @return    none
 */
void i2c_slave_read(unsigned char *data, unsigned char len);

/**
 * @brief     This function serves to receive uart data by byte with not DMA method.
 * @param[in]  data - the data need send.
 * @param[in]  len - The total length of the data.
 * @return    none
 */
void i2c_slave_write(unsigned char *data, unsigned char len);

/**
 * @brief      This function serves to set the i2c clock frequency. The i2c clock is consistent with the system clock.
 *             Currently, the default system clock is 48M, and the i2c clock is also 48M.
 * @param[in]  clock - the division factor of I2C clock,
 *             I2C frequency = System_clock / (4*DivClock).
 * @return     none
 */
void i2c_set_master_clk(unsigned char clock);

/**
 * @brief     This function serves to set i2c tx_dam channel and config dma tx default.
 * @param[in] chn: dma channel.
 * @return    none
 */
void i2c_set_tx_dma_config(dma_chn_e chn);

/**
 * @brief     This function serves to set i2c rx_dam channel and config dma rx default.
 * @param[in] chn: dma channel.
 * @return    none
 */
void i2c_set_rx_dma_config(dma_chn_e chn);

#endif