git.gag.com Git - fw/altos/blob - src/avr/ao_serial_avr.c

   1 /*
   2  * Copyright © 2011 Keith Packard <keithp@keithp.com>
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; version 2 of the License.
   7  *
   8  * This program is distributed in the hope that it will be useful, but
   9  * WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11  * General Public License for more details.
  12  *
  13  * You should have received a copy of the GNU General Public License along
  14  * with this program; if not, write to the Free Software Foundation, Inc.,
  15  * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
  16  */
  17
  18 #include "ao.h"
  19
  20 __xdata struct ao_fifo  ao_usart1_rx_fifo;
  21 __xdata struct ao_fifo  ao_usart1_tx_fifo;
  22
  23 void
  24 ao_debug_out(char c)
  25 {
  26         if (c == '\n')
  27                 ao_debug_out('\r');
  28         loop_until_bit_is_set(UCSR1A, UDRE1);
  29         UDR1 = c;
  30 }
  31
  32 ISR(USART1_RX_vect)
  33 {
  34         if (!ao_fifo_full(ao_usart1_rx_fifo))
  35                 ao_fifo_insert(ao_usart1_rx_fifo, UDR1);
  36         ao_wakeup(&ao_usart1_rx_fifo);
  37 #if USE_SERIAL_STDIN
  38         ao_wakeup(&ao_stdin_ready);
  39 #endif
  40 }
  41
  42 static __xdata uint8_t ao_serial_tx1_started;
  43
  44 static void
  45 ao_serial_tx1_start(void)
  46 {
  47         if (!ao_fifo_empty(ao_usart1_tx_fifo) &&
  48             !ao_serial_tx1_started)
  49         {
  50                 ao_serial_tx1_started = 1;
  51                 ao_fifo_remove(ao_usart1_tx_fifo, UDR1);
  52         }
  53 }
  54
  55 ISR(USART1_UDRE_vect)
  56 {
  57         ao_serial_tx1_started = 0;
  58         ao_serial_tx1_start();
  59         ao_wakeup(&ao_usart1_tx_fifo);
  60 }
  61
  62 char
  63 ao_serial_getchar(void) __critical
  64 {
  65         char    c;
  66         cli();
  67         while (ao_fifo_empty(ao_usart1_rx_fifo))
  68                 ao_sleep(&ao_usart1_rx_fifo);
  69         ao_fifo_remove(ao_usart1_rx_fifo, c);
  70         sei();
  71         return c;
  72 }
  73
  74 #if USE_SERIAL_STDIN
  75 char
  76 ao_serial_pollchar(void) __critical
  77 {
  78         char    c;
  79         cli();
  80         if (ao_fifo_empty(ao_usart1_rx_fifo)) {
  81                 sei();
  82                 return AO_READ_AGAIN;
  83         }
  84         ao_fifo_remove(ao_usart1_rx_fifo,c);
  85         sei();
  86         return c;
  87 }
  88 #endif
  89
  90 void
  91 ao_serial_putchar(char c) __critical
  92 {
  93         cli();
  94         while (ao_fifo_full(ao_usart1_tx_fifo))
  95                 ao_sleep(&ao_usart1_tx_fifo);
  96         ao_fifo_insert(ao_usart1_tx_fifo, c);
  97         ao_serial_tx1_start();
  98         sei();
  99 }
 100
 101 void
 102 ao_serial_drain(void) __critical
 103 {
 104         cli();
 105         while (!ao_fifo_empty(ao_usart1_tx_fifo))
 106                 ao_sleep(&ao_usart1_tx_fifo);
 107         sei();
 108 }
 109
 110 static const struct {
 111         uint16_t ubrr;
 112 } ao_serial_speeds[] = {
 113         /* [AO_SERIAL_SPEED_4800] = */ {
 114                 F_CPU / (16UL * 4800UL) - 1
 115         },
 116         /* [AO_SERIAL_SPEED_9600] = */ {
 117                 F_CPU / (16UL * 9600UL) - 1
 118         },
 119         /* [AO_SERIAL_SPEED_19200] = */ {
 120                 F_CPU / (16UL * 19200UL) - 1
 121         },
 122         /* [AO_SERIAL_SPEED_57600] = */ {
 123                 F_CPU / (16UL * 57600UL) - 1
 124         },
 125 };
 126
 127 void
 128 ao_serial_set_speed(uint8_t speed)
 129 {
 130         ao_serial_drain();
 131         if (speed > AO_SERIAL_SPEED_57600)
 132                 return;
 133         UBRR1L = ao_serial_speeds[speed].ubrr;
 134         UBRR1H = ao_serial_speeds[speed].ubrr >> 8;
 135 }
 136
 137 void
 138 ao_serial_init(void)
 139 {
 140         /* Ensure the uart is powered up */
 141
 142         PRR1 &= ~(1 << PRUSART1);
 143
 144         /* Pick a 9600 baud rate */
 145         ao_serial_set_speed(AO_SERIAL_SPEED_9600);
 146
 147         UCSR1A = 0;
 148         UCSR1C = ((0 << UMSEL10) |      /* Asynchronous mode */
 149                   (0 << UPM10) |        /* No parity */
 150                   (0 << USBS1) |        /* 1 stop bit */
 151                   (3 << UCSZ10) |       /* 8 bit characters */
 152                   (0 << UCPOL1));       /* MBZ for async mode */
 153         UCSR1B = ((1 << RXEN1) |        /* Enable receiver */
 154                   (1 << TXEN1) |        /* Enable transmitter */
 155                   (1 << RXCIE1) |       /* Enable receive interrupts */
 156                   (1 << UDRIE1));       /* Enable transmit empty interrupts */
 157 #if USE_SERIAL_STDIN
 158         ao_add_stdio(ao_serial_pollchar,
 159                      ao_serial_putchar,
 160                      NULL);
 161 #endif
 162 }