【转】Atmega32串口驱动(基于ucos)——循环缓冲区

原文地址：http://bbs.ednchina.com/BLOG_ARTICLE_235210.HTM

/* ******************************************************************
 * 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 |=  1<<UDRIE; }

#define UartStopTx()    { UCSRB &= ~(1<<UDRIE); } 
#define UartStartRx()   { UCSRB |=  1<<RXCIE; }
#define UartStopRx()    { UCSRB &= ~( 1<<RXCIE ); }
/* ****************************************************************
 *              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 != '')

        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;

    UartRxFifo.sem =  OSSemCreate( 0 );                         // 接收结构初始化
    UartRxFifo.buf = UartRxBuf;
    UartRxFifo.in = UartRxFifo.out = UartRxBuf;

    // 初始化串口，设置baudrate, len, parity, stops -- 

    OS_ENTER_CRITICAL();
//  UCSRA = 0xff;                                           
    UCSRB = 1<<RXCIE | 1<<RXEN | 1<<TXEN;                               // 收发使能,接收中断
    UCSRC =  1<<URSEL| 1<<UPM1| 1<<UPM0| 1<<UCSZ1| 1<<UCSZ0;            // 异步,8位字节,奇校验
    UBRRL = ((INT32U)F_OSC / (16*(INT32U)UART_BAUD ) - 1 ) & 0xff;      // 设置波特率
    UBRRH = ((INT32U)F_OSC / (16*(INT32U)UART_BAUD ) - 1 ) >> 8;
    OS_EXIT_CRITICAL();
}

void  UartTxIsr( void )                                                 // 发送寄存器空中断
{   
    INT8U   err, c;

    c = UartGetTxChar( &err );

    if( err == UART_OK )
        UDR = c;                                                        // 发送数据
}

void  UartRxIsr( void )                                                 // 发送寄存器空中断
{
    INT8U   err, c;

    err = UCSRA;

    c = UDR;
    if( !(err & ( DOR|PE|FE )) )
        UartPutRxChar( c );
}
/* *****************************************************************
 *                      UartPutRxChar()
 * 函数名: UartPutRxChar()
 * 参数：c ----- 要写入的字符
 * 返回值：UART_OK --- 操作成功
 *          UART_BUF_FULL -- 失败，缓冲满
 * 说明：此函数仅供中断系统调用
 * ***************************************************************** */
static INT8U UartPutRxChar( INT8U c )
{
    OSSemQuery( UartRxFifo.sem, &SemData );
    if( SemData.OSCnt < UART_RX_LEN )                                   // 缓冲不满
    {
        *(UartRxFifo.in)++ = c;
        if( UartRxFifo.in >= UartRxBuf + UART_RX_LEN )
            UartRxFifo.in = UartRxBuf;
        OSSemPost( UartRxFifo.sem );                                    // 向任务发送信号量

        return  UART_OK;

    }       
    else
    {
        UartStopRx();                                                   // 缓冲满,不再接收数据
        return  UART_BUF_FULL;
    }
}

/* *****************************************************************
 *                      UartGetTxChar()
 * 函数名: UartGetTxChar()
 * 参数：err ---- 错误参量
 * 返回值：UART_OK
 *          UART_BUF_EMPTY --- 失败，缓冲空
 * 说明：
 * ***************************************************************** */
static INT8U    UartGetTxChar( INT8U * err )
{
    INT8U c;

    c = 0;

    OSSemQuery( UartTxFifo.sem, &SemData );
    if( SemData.OSCnt < UART_TX_LEN )                                   // <TX_LEN, 缓冲有数据可发送
    {
        c = *(UartTxFifo.out)++;
        if( UartTxFifo.out >= UartTxBuf + UART_TX_LEN )
                UartTxFifo.out = UartTxBuf;
        OSSemPost( UartTxFifo.sem );                                    // 向任务发送信号量
        *err = UART_OK;
    }       
    else
    {
        UartStopTx();                                                   // 缓冲空,不再使能发送中断
        *err = UART_BUF_EMPTY;
    }

    return  c;

}