如何在atmega2560上使用串行rx/tx

#include <avr/io.h>

#define F_CPU 16000000UL
#define BAUD 9600
#define UBRR_VAL ((F_CPU / (16UL * BAUD)) - 1)

void USART_Init() {
    // 设置波特率
    UBRR0H = (UBRR_VAL >> 8);
    UBRR0L = UBRR_VAL;
    
    // 启用接收和发送
    UCSR0B = (1 << RXEN0) | (1 << TXEN0);
    
    // 设置数据位和停止位
    UCSR0C = (1 << UCSZ01) | (1 << UCSZ00);
}

void USART_Transmit(unsigned char data) {
    // 等待发送缓冲区为空
    while (!(UCSR0A & (1 << UDRE0)));
    
    // 将数据发送到串行端口
    UDR0 = data;
}

unsigned char USART_Receive() {
    // 等待接收缓冲区有数据
    while (!(UCSR0A & (1 << RXC0)));
    
    // 从串行端口接收数据
    return UDR0;
}

int main() {
    USART_Init();
    
    while (1) {
        unsigned char receivedData = USART_Receive();
        // 处理接收到的数据
        
        // 发送数据
        USART_Transmit('A');
    }
    
    return 0;
}