git.gag.com Git - fw/altos/blob - src/avr/ao_avr_stdio.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 int
  21 stdio_put(char c, FILE *stream)
  22 {
  23         if (ao_cur_task && ao_num_stdios)
  24                 putchar(c);
  25         else
  26         {
  27                 if (c == '\n')
  28                         stdio_put('\r', stream);
  29                 loop_until_bit_is_set(UCSR1A, UDRE1);
  30                 UDR1 = c;
  31         }
  32
  33         return 0;
  34 }
  35
  36 int
  37 stdio_get(FILE *stream)
  38 {
  39         return (int) getchar() & 0xff;
  40 }
  41
  42 static FILE mystdout = FDEV_SETUP_STREAM(stdio_put, NULL, _FDEV_SETUP_WRITE);
  43
  44 static FILE mystdin = FDEV_SETUP_STREAM(NULL, stdio_get, _FDEV_SETUP_READ);
  45
  46 void
  47 ao_avr_stdio_init(void)
  48 {
  49         stdout = &mystdout;
  50         stdin = &mystdin;
  51 }