75 lines
1.4 KiB
C
75 lines
1.4 KiB
C
/*
|
|
* Copyright 2011 Mika Tuupola
|
|
*
|
|
* Licensed under the MIT license:
|
|
* http://www.opensource.org/licenses/mit-license.php
|
|
*
|
|
*/
|
|
|
|
#include <avr/io.h>
|
|
#include <stdio.h>
|
|
|
|
#ifndef F_CPU
|
|
#define F_CPU 16000000UL
|
|
#endif
|
|
|
|
#ifndef BAUD
|
|
#define BAUD 9600
|
|
#endif
|
|
#include <util/setbaud.h>
|
|
|
|
/* http://www.cs.mun.ca/~rod/Winter2007/4723/notes/serial/serial.html */
|
|
|
|
#if defined (__AVR_ATmega32U4__)
|
|
void uart_init(void) {
|
|
UBRR1H = UBRRH_VALUE;
|
|
UBRR1L = UBRRL_VALUE;
|
|
|
|
#if USE_2X
|
|
UCSR1A |= _BV(U2X1);
|
|
#else
|
|
UCSR1A &= ~(_BV(U2X1));
|
|
#endif
|
|
|
|
UCSR1C = _BV(UCSZ11) | _BV(UCSZ10); /* 8-bit data */
|
|
UCSR1B = _BV(RXEN1) | _BV(TXEN1); /* Enable RX and TX */
|
|
}
|
|
#else
|
|
void uart_init(void) {
|
|
UBRR0H = UBRRH_VALUE;
|
|
UBRR0L = UBRRL_VALUE;
|
|
|
|
#if USE_2X
|
|
UCSR0A |= _BV(U2X0);
|
|
#else
|
|
UCSR0A &= ~(_BV(U2X0));
|
|
#endif
|
|
|
|
UCSR0C = _BV(UCSZ01) | _BV(UCSZ00); /* 8-bit data */
|
|
UCSR0B = _BV(RXEN0) | _BV(TXEN0); /* Enable RX and TX */
|
|
}
|
|
#endif
|
|
|
|
void uart_putchar(char c, FILE *stream) {
|
|
if (c == '\n') {
|
|
uart_putchar('\r', stream);
|
|
}
|
|
#if defined (__AVR_ATmega32U4__)
|
|
loop_until_bit_is_set(UCSR1A, UDRE1);
|
|
UDR1 = c;
|
|
#else
|
|
loop_until_bit_is_set(UCSR0A, UDRE0);
|
|
UDR0 = c;
|
|
#endif
|
|
}
|
|
|
|
char uart_getchar(FILE *stream) {
|
|
#if defined (__AVR_ATmega32U4__)
|
|
loop_until_bit_is_set(UCSR1A, RXC1);
|
|
return UDR1;
|
|
#else
|
|
loop_until_bit_is_set(UCSR0A, RXC0);
|
|
return UDR0;
|
|
#endif
|
|
}
|