2967e705cf013e9c3d08fb12a6e55da1c405f62e
[fw/altos] / src / core / ao_stdio.c
1 /*
2  * Copyright © 2009 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 /*
21  * Basic I/O functions to support SDCC stdio package
22  */
23
24 #define AO_NUM_STDIOS   (HAS_USB + PACKET_HAS_SLAVE + USE_SERIAL_STDIN)
25
26 __xdata struct ao_stdio ao_stdios[AO_NUM_STDIOS];
27 __pdata int8_t ao_cur_stdio;
28 __pdata int8_t ao_num_stdios;
29
30 void
31 putchar(char c)
32 {
33 #if LOW_LEVEL_DEBUG
34         if (!ao_cur_task) {
35                 extern void ao_debug_out(char c);
36                 if (c == '\n')
37                         ao_debug_out('\r');
38                 ao_debug_out(c);
39                 return;
40         }
41 #endif
42         if (c == '\n')
43                 (*ao_stdios[ao_cur_stdio].putchar)('\r');
44         (*ao_stdios[ao_cur_stdio].putchar)(c);
45 }
46
47 void
48 flush(void)
49 {
50         if (ao_stdios[ao_cur_stdio].flush)
51                 ao_stdios[ao_cur_stdio].flush();
52 }
53
54 __xdata uint8_t ao_stdin_ready;
55
56 char
57 getchar(void) __reentrant __critical
58 {
59         char c;
60         int8_t stdio = ao_cur_stdio;
61
62         for (;;) {
63                 c = ao_stdios[stdio].pollchar();
64                 if (c != AO_READ_AGAIN)
65                         break;
66                 if (++stdio == ao_num_stdios)
67                         stdio = 0;
68                 if (stdio == ao_cur_stdio)
69                         ao_sleep(&ao_stdin_ready);
70         }
71         ao_cur_stdio = stdio;
72         return c;
73 }
74
75 uint8_t
76 ao_echo(void)
77 {
78         return ao_stdios[ao_cur_stdio].echo;
79 }
80
81 int8_t
82 ao_add_stdio(char (*pollchar)(void),
83              void (*putchar)(char),
84              void (*flush)(void)) __reentrant
85 {
86         if (ao_num_stdios == AO_NUM_STDIOS)
87                 ao_panic(AO_PANIC_STDIO);
88         ao_stdios[ao_num_stdios].pollchar = pollchar;
89         ao_stdios[ao_num_stdios].putchar = putchar;
90         ao_stdios[ao_num_stdios].flush = flush;
91         ao_stdios[ao_num_stdios].echo = 1;
92         return ao_num_stdios++;
93 }