git.mutantstargoat.com Git - romdev/blob - fw/src/serial.c

   1 #ifndef F_CPU
   2 #ifdef XTAL
   3 #define F_CPU   XTAL
   4 #else
   5 #warning "compiled for 1mhz internal rc osc. serial comms won't work"
   6 #define F_CPU   1000000
   7 #endif
   8 #endif
   9
  10 #include <stdio.h>
  11 #include <avr/io.h>
  12 #include <avr/interrupt.h>
  13 #include <util/delay.h>
  14 #include <avr/power.h>
  15
  16 static int uart_send_char(char c, FILE *fp);
  17 static int uart_get_char(FILE *fp);
  18
  19 #define BUF_SZ  16
  20 #define BUF_IDX_MASK    (BUF_SZ - 1)
  21 #define NEXT_IDX(x)     (((x) + 1) & BUF_IDX_MASK)
  22 static char outbuf[BUF_SZ];
  23 static volatile unsigned char out_rd, out_wr;
  24 static char inbuf[BUF_SZ];
  25 static volatile unsigned char in_rd, in_wr;
  26
  27 static FILE std_stream = FDEV_SETUP_STREAM(uart_send_char, uart_get_char, _FDEV_SETUP_RW);
  28
  29
  30
  31 void init_serial(long baud)
  32 {
  33         unsigned int ubrr_val = F_CPU / 16 / baud - 1;
  34
  35         power_usart0_enable();
  36
  37         /* set baud generator timer reset value */
  38         UBRR0H = (unsigned char)(ubrr_val >> 8);
  39         UBRR0L = (unsigned char)ubrr_val;
  40
  41         /* enable rx/tx and recv interrupt */
  42         UCSR0B = (1 << RXEN0) | (1 << TXEN0) | (1 << RXCIE0);
  43         /* set frame format: 8n1 */
  44         UCSR0C = 3 << UCSZ00;
  45
  46         stdin = stdout = stderr = &std_stream;
  47 }
  48
  49 int have_input(void)
  50 {
  51         return in_wr != in_rd;
  52 }
  53
  54 static int uart_send_char(char c, FILE *fp)
  55 {
  56         /*int next;*/
  57
  58         while((UCSR0A & (1 << UDRE0)) == 0);
  59         UDR0 = (unsigned char)c;
  60 #if 0
  61         next = NEXT_IDX(out_wr);
  62         while(next == out_rd);
  63
  64         outbuf[out_wr] = c;
  65         out_wr = next;
  66
  67         /* enable the Tx data register empty interrupt */
  68         UCSR0B |= 1 << UDRIE0;
  69 #endif
  70         return 0;
  71 }
  72
  73 static int uart_get_char(FILE *fp)
  74 {
  75         char c;
  76
  77         while(in_rd == in_wr);
  78
  79         c = inbuf[in_rd];
  80         in_rd = NEXT_IDX(in_rd);
  81         return c;
  82 }
  83
  84 ISR(USART_RX_vect)
  85 {
  86         char c = UDR0;
  87
  88         inbuf[in_wr] = c;
  89         in_wr = NEXT_IDX(in_wr);
  90 }
  91
  92 /* USART Tx data register empty (can send more data) */
  93 ISR(USART_UDRE_vect)
  94 {
  95         if(out_rd != out_wr) {
  96                 UDR0 = outbuf[out_rd];
  97                 out_rd = NEXT_IDX(out_rd);
  98         } else {
  99                 /* no more data to send for now, disable the interrupt */
 100                 UCSR0B &= ~(1 << UDRIE0);
 101         }
 102 }