Atmega32单片机串口驱动程序

/* ******************************************************************
* Filename: uart.c
* Author: lstzixing
* Mail: blievethink@gmail.com
* Date: 2009-5-26
* Description: 串口数据收发接口. For Atmega32
* ****************************************************************** */

#include "app.h"

// 对发送缓冲,信号计数为空闲字符数
// 对接收缓冲,计数为缓冲已有计数
typedef struct _FIFO
{
INT8U * buf; // FIFO缓冲区
INT8U * in, * out; // FIFO读写指针
OS_EVENT * sem; // FIFO读写同步信号量
}FIFO;

static INT8U UartTxBuf[ UART_TX_LEN ]; // 发送缓冲
static INT8U UartRxBuf[ UART_RX_LEN ]; // 接收缓冲
static FIFO UartTxFifo, UartRxFifo; // 收发缓冲控制FIFO结构

OS_SEM_DATA SemData;

static INT8U UartPutRxChar( INT8U c );
static INT8U UartGetTxChar( INT8U * err );

#define UartStartTx() { UCSRB |= 1UDRIE; }
#define UartStopTx() { UCSRB = ~(1UDRIE); }
#define UartStartRx() { UCSRB |= 1RXCIE; }
#define UartStopRx() { UCSRB = ~( 1RXCIE ); }
/* ****************************************************************
* UartFlush()
* 功能：缓冲清空
* 参数: isTxBuf ------ 是否为发送缓冲
* 返回值：None
* 说明：清空收发缓冲
* *************************************************************** */
void UartFlush( INT8U isTxBuf )
{
#if OS_CRITICAL_METHOD == 3
OS_CPU_SR cpu_sr;
#endif
INT8U cnt;

OS_ENTER_CRITICAL();

if( isTxBuf )
{
UartTxFifo.buf = UartTxBuf; // 复位发送缓冲读写指针
UartTxFifo.in = UartTxBuf;
UartTxFifo.out = UartTxBuf;
OSSemQuery( UartTxFifo.sem, SemData );
cnt = UART_TX_LEN - SemData.OSCnt; // 在其它地方必须保证SemData.OSCnt UART_TX_LEN
while( cnt-- )
OSSemPost( UartTxFifo.sem ); // 复位发送信号量值为UART_TX_LEN
}

else
{
UartRxFifo.buf = UartRxBuf; // 复位接收缓冲读写指针
UartRxFifo.in = UartRxBuf;
UartRxFifo.out = UartRxBuf;
while( OSSemAccept( UartRxFifo.sem ) ); // 复位接收信号量值为0

}

OS_EXIT_CRITICAL();
}

void UartPutStr( char * str )
{
char * ptr;

ptr = (char *)str;

while(*ptr != '\0')
UartPutChar( *ptr++, 0 );

}
/* *****************************************************************
* UartPutChar()
* 函数名:UartPutChar()
* 功能: 发送一字节至串口缓冲区
* 参数: c ----------- 要发送的字节
* to ---------- 指定的等待操作完成的超时时间
* 返回值: OS_NO_ERR --- 操作成功
* OS_TIMEOUT --- 操作超时
* 说明:
* ***************************************************************** */
INT8U UartPutChar( INT8U c, INT16U to )
{
#if OS_CRITICAL_METHOD == 3
OS_CPU_SR cpu_sr;
#endif
INT8U err;

OSSemPend( UartTxFifo.sem, to, err ); // 等待发送缓冲可用

if( err == OS_NO_ERR )
{
OS_ENTER_CRITICAL();
*(UartTxFifo.in)++ = c;
if( UartTxFifo.in >= UartTxBuf + UART_TX_LEN) // 调整指针
UartTxFifo.in = UartTxBuf;

OS_EXIT_CRITICAL();

// 启动发送
UartStartTx();
}

return err;
}

/* ***************************************************************
* UartGetChar()
* 函数名:UartGetChar()
* 功能:从接收缓冲区中取一字节
* 参数: to --- 指定的超时量
* err --- 指定存储错误变量
* OS_TIMEOUT / OS_NO_ERR
* 返回值: 接收的字节
* 说明:NO
* *************************************************************** */
INT8U UartGetChar( INT16U to, INT8U * err )
{
#if OS_CRITICAL_METHOD == 3
OS_CPU_SR cpu_sr;
#endif
INT8U c;

c = 0;
OSSemPend( UartRxFifo.sem, to, err );

if( *err != OS_TIMEOUT )
{
OS_ENTER_CRITICAL();
c = *(UartRxFifo.out)++; // 写入字节
if( UartRxFifo.out >= UartRxBuf + UART_RX_LEN ) // 调整指针
UartRxFifo.out = UartRxBuf;
OS_EXIT_CRITICAL();

UartStartRx(); // 接收缓冲非满,使能接收
}

return c;
}

/* ******************************************************************
* UartHdInit()
* 函数名:UartInit()
* 说明:
* 参数:baudrate ----- 波特率
* partity ------ 奇偶校验方式
* stops ------ 停止位数
* len -------- 数据帧长度
* 说明：此函数必须最先被调用
* ****************************************************************** */
void UartInit( void )
{
#if OS_CRITICAL_METHOD == 3
OS_CPU_SR cpu_sr;
#endif
UartTxFifo.sem = OSSemCreate( UART_TX_LEN ); // 发送结构初始化
UartTxFifo.buf = UartTxBuf;
UartTxFifo.in = UartTxFifo.out = UartTxBuf;