X-Git-Url: https://git.gag.com/?p=fw%2Faltos;a=blobdiff_plain;f=src%2Fao_stdio.c;h=045dff09f340e92b7b32b62ba2ceb914cfc2724f;hp=fb8ce0937fc6cd9dd1b57b186bcf18eb4e7285ea;hb=f7ff3278bb670df59d7425a014cfe8e3718fea3f;hpb=210dbaa23cdacf3a6f2d6e23493e96ee2ac9bca7

diff --git a/src/ao_stdio.c b/src/ao_stdio.c
index fb8ce093..045dff09 100644
--- a/src/ao_stdio.c
+++ b/src/ao_stdio.c
@@ -21,22 +21,59 @@
  * Basic I/O functions to support SDCC stdio package
  */
 
+#define AO_NUM_STDIOS	2
+
+static __xdata struct ao_stdio stdios[AO_NUM_STDIOS];
+static __data int8_t ao_cur_stdio;
+static __data int8_t ao_num_stdios;
+
 void
 putchar(char c)
 {
 	if (c == '\n')
-		ao_usb_putchar('\r');
-	ao_usb_putchar(c);
+		(*stdios[ao_cur_stdio].putchar)('\r');
+	(*stdios[ao_cur_stdio].putchar)(c);
 }
 
 void
 flush(void)
 {
-	ao_usb_flush();
+	if (stdios[ao_cur_stdio].flush)
+		stdios[ao_cur_stdio].flush();
 }
 
+__xdata uint8_t ao_stdin_ready;
+
 char
-getchar(void)
+getchar(void) __reentrant
+{
+	char c;
+	int8_t stdio = ao_cur_stdio;
+
+	EA = 0;
+	for (;;) {
+		c = stdios[stdio].pollchar();
+		if (c != AO_READ_AGAIN)
+			break;
+		if (++stdio == ao_num_stdios)
+			stdio = 0;
+		if (stdio == ao_cur_stdio)
+			ao_sleep(&ao_stdin_ready);
+	}
+	EA = 1;
+	ao_cur_stdio = stdio;
+	return c;
+}
+
+void
+ao_add_stdio(char (*pollchar)(void),
+	     void (*putchar)(char),
+	     void (*flush)(void))
 {
-	return ao_usb_getchar();
+	if (ao_num_stdios == AO_NUM_STDIOS)
+		ao_panic(AO_PANIC_STDIO);
+	stdios[ao_num_stdios].pollchar = pollchar;
+	stdios[ao_num_stdios].putchar = putchar;
+	stdios[ao_num_stdios].flush = flush;
+	ao_num_stdios++;
 }