#include "nRF24L01_API.h"

void delay_us ( uchar num )
{
    uchar i;

    for ( i = 0; i < num; i++ )
    {
#ifdef N76E003_IAR
        asm ( "nop" );
#else
        nop;
#endif
    }
}

/***************************************************************/

/*******************************************************************/
uchar SPI_RW ( uchar byte )
{
    uchar bit_ctr;

    for ( bit_ctr = 0; bit_ctr < 8; bit_ctr++ ) // 输出8位
    {
        NRF_MOSI = ( byte & 0x80 ); 			// MSB TO MOSI
        byte = ( byte << 1 );					// shift next bit to MSB
        NRF_SCK = 1;
        byte |= NRF_MISO;	        		// capture current MISO bit
        NRF_SCK = 0;
    }

    return byte;
}

/*********************************************/
/* 函数功能：给24L01的寄存器写值（一个字节） */
/* 入口参数：reg   要写的寄存器地址          */
/*           value 给寄存器写的值            */
/* 出口参数：status 状态值                   */
/*********************************************/
uchar NRF24L01_Write_Reg ( uchar reg, uchar value )
{
    uchar status;
    NRF_CSN = 0;                //CSN=0;
    status = SPI_RW ( reg );		//发送寄存器地址,并读取状态值
    SPI_RW ( value );
    NRF_CSN = 1;                //CSN=1;

    return status;
}
/*************************************************/
/* 函数功能：读24L01的寄存器值 （一个字节）      */
/* 入口参数：reg  要读的寄存器地址               */
/* 出口参数：value 读出寄存器的值                */
/*************************************************/
uchar NRF24L01_Read_Reg ( uchar reg )
{
    uchar value;

    NRF_CSN = 0;            //CSN=0;
    SPI_RW ( reg );			//发送寄存器值(位置),并读取状态值
    value = SPI_RW ( NOP );
    NRF_CSN = 1;             	//CSN=1;

    return value;
}
/*********************************************/
/* 函数功能：读24L01的寄存器值（多个字节）   */
/* 入口参数：reg   寄存器地址                */
/*           *pBuf 读出寄存器值的存放数组    */
/*           len   数组字节长度              */
/* 出口参数：status 状态值                   */
/*********************************************/
uchar NRF24L01_Read_Buf ( uchar reg, uchar *pBuf, uchar len )
{
    uchar status, u8_ctr;
    NRF_CSN = 0;                   	//CSN=0
    status = SPI_RW ( reg );				//发送寄存器地址,并读取状态值

    for ( u8_ctr = 0; u8_ctr < len; u8_ctr++ )
    {
        pBuf[u8_ctr] = SPI_RW ( 0XFF );		//读出数据
    }

    NRF_CSN = 1;                 		//CSN=1
    return status;        			//返回读到的状态值
}
/**********************************************/
/* 函数功能：给24L01的寄存器写值（多个字节）  */
/* 入口参数：reg  要写的寄存器地址            */
/*           *pBuf 值的存放数组               */
/*           len   数组字节长度               */
/**********************************************/
uchar NRF24L01_Write_Buf ( uchar reg, uchar *pBuf, uchar len )
{
    uchar status, u8_ctr;
    NRF_CSN = 0;
    status = SPI_RW ( reg );			//发送寄存器值(位置),并读取状态值

    for ( u8_ctr = 0; u8_ctr < len; u8_ctr++ )
    {
        SPI_RW ( *pBuf++ ); 				//写入数据
    }

    NRF_CSN = 1;
    return status;          		//返回读到的状态值
}

/*********************************************/
/* 函数功能：24L01接收数据                   */
/* 入口参数：rxbuf 接收数据数组              */
/* 返回值： 0   成功收到数据                 */
/*          1   没有收到数据                 */
/*********************************************/
uchar NRF24L01_RxPacket ( uchar *rxbuf )
{
    uchar state0, state1;
    uchar res = 0;

    state0 = state1 = NRF24L01_Read_Reg ( STATUS );  			//读取状态寄存器的值
    NRF24L01_Write_Reg ( WRITE_REG + STATUS, state1 ); //清除TX_DS或MAX_RT中断标志

    if ( state0 & RX_OK )								//接收到数据
    {
        NRF_CE = 0;
        NRF24L01_Read_Buf ( RD_RX_PLOAD, rxbuf, RX_PLOAD_WIDTH ); //读取数据
        NRF24L01_Write_Reg ( FLUSH_RX, 0xff );					//清除RX FIFO寄存器
        NRF_CE = 1;
        delay_us ( 150 );
        res = 1;
    }

    return res;//没收到任何数据
}
/**********************************************/
/* 函数功能：设置24L01为发送模式              */
/* 入口参数：txbuf  发送数据数组              */
/* 返回值； 0x10    达到最大重发次数，发送失败*/
/*          0x20    成功发送完成              */
/*          0xff    发送失败                  */
/**********************************************/
uchar NRF24L01_TxPacket ( uchar *txbuf )
{
    uchar state;

    NRF_CE = 0;												//CE拉低，使能24L01配置
    NRF24L01_Write_Buf ( WR_TX_PLOAD, txbuf, TX_PLOAD_WIDTH );	//写数据到TX BUF  32个字节
    NRF_CE = 1;												//CE置高，使能发送

    while ( NRF_IRQ == 1 );										//等待发送完成

    state = NRF24L01_Read_Reg ( STATUS );  						//读取状态寄存器的值
    NRF24L01_Write_Reg ( WRITE_REG + STATUS, state ); 			//清除TX_DS或MAX_RT中断标志

    if ( state & MAX_TX )										//达到最大重发次数
    {
        NRF24L01_Write_Reg ( FLUSH_TX, 0xff );					//清除TX FIFO寄存器
        return MAX_TX;
    }

    if ( state & TX_OK )											//发送完成
    {
        return TX_OK;
    }

    return 0xff;											//发送失败
}

/********************************************/
/* 函数功能：检测24L01是否存在              */
/* 返回值；  0  存在                        */
/*           1  不存在                      */
/********************************************/
uchar NRF24L01_Check ( void )
{
    uchar check_in_buf[5] = {0x11, 0x22, 0x33, 0x44, 0x55};
    uchar check_out_buf[5] = {0x00};
    NRF_SCK = 0;
    NRF_CSN = 1;
    NRF_CE = 0;

    NRF24L01_Write_Buf ( WRITE_REG + TX_ADDR, check_in_buf, 5 );

    NRF24L01_Read_Buf ( READ_REG + TX_ADDR, check_out_buf, 5 );

    if ( ( check_out_buf[0] == 0x11 ) && \
            ( check_out_buf[1] == 0x22 ) && \
            ( check_out_buf[2] == 0x33 ) && \
            ( check_out_buf[3] == 0x44 ) && \
            ( check_out_buf[4] == 0x55 ) ) return 0;
    else return 1;
}

/**********************************************/
/* 函数功能：24L01初始化                      */
/* 入口参数：PLOAD_WIDTH  有效数据宽度        */
/* 			 TX_ADR  发送地址		          */
/* 		     TX_ADDR_WIDTH  发送地址宽度      */
/* 			 RX_ADR  接收发送地址        	  */
/* 			 RX_ADDR_WIDTH  接收地址宽度      */
/* 			 RATE  有效数据宽度        		  */
/* 返回值  ；无	     	                      */
/**********************************************/
void NRF24L01_RT_Init ( uchar PLOAD_WIDTH, uchar *TX_ADR, uchar TX_ADDR_WIDTH, uchar *RX_ADR, uchar RX_ADDR_WIDTH, uchar RATE )
{
    NRF_CE = 0;
    NRF24L01_Write_Reg ( WRITE_REG + RX_PW_P0, PLOAD_WIDTH ); //设置接收数据长度，本次设置为32字节
    NRF24L01_Write_Reg ( FLUSH_RX, 0xff ); //清除RX FIFO寄存器
    NRF24L01_Write_Buf ( WRITE_REG + TX_ADDR, TX_ADR, TX_ADDR_WIDTH ); //写TX节点地址
    NRF24L01_Write_Buf ( WRITE_REG + RX_ADDR_P0, RX_ADR, RX_ADDR_WIDTH ); //设置RX节点地址,主要为了使能ACK

    NRF24L01_Write_Reg ( WRITE_REG + EN_AA, 0x01 ); //使能通道0的自动应答
    NRF24L01_Write_Reg ( WRITE_REG + EN_RXADDR, 0x01 ); //使能通道0的接收地址

    NRF24L01_Write_Reg ( WRITE_REG + SETUP_RETR, 0x1a ); //设置自动重发间隔时间:500us + 86us;最大自动重发次数:10(a)次
    NRF24L01_Write_Reg ( WRITE_REG + RF_CH, 0 ); //设置RF通道为40 收发必须一致，0为2.4GHz
    NRF24L01_Write_Reg ( WRITE_REG + RF_SETUP, RATE ); //设置TX发射参数,0db增益,1Mbps,低噪声增益开启
    NRF24L01_Write_Reg ( WRITE_REG + CONFIG, 0x0f ); //配置基本工作模式的参数;PWR_UP,EN_CRC,16BIT_CRC,接收模式,开启所有中断
    NRF_CE = 1;					//CE置高，使能发送
}
/**********************************************/
/* 函数功能：发射函数                     	  */
/* 入口参数：buf  发送寄存器	          	  */
/* 返回值  ；无	     	                      */
/**********************************************/
uchar SEND_BUF ( uchar *buf )
{
    uchar state;
    NRF_CE = 0;
    NRF24L01_Write_Reg ( WRITE_REG + CONFIG, 0x0e );
    NRF_CE = 1;
    delay_us ( 15 );
    state = NRF24L01_TxPacket ( buf );
    NRF_CE = 0;
    NRF24L01_Write_Reg ( WRITE_REG + CONFIG, 0x0f );
    NRF_CE = 1;
    return state;
}