1 files changed, 127 insertions, 0 deletions
diff --git a/src-avr/ao_stdio.c b/src-avr/ao_stdio.c
new file mode 100644
index 00000000..bfe418d1
--- /dev/null
+++ b/src-avr/ao_stdio.c
@@ -0,0 +1,127 @@
+/*
+ * Copyright © 2009 Keith Packard <keithp@keithp.com>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; version 2 of the License.
+ *
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
+ */
+
+#include "ao.h"
+
+#ifdef AVR
+#undef putchar
+#undef getchar
+#define putchar(c)	ao_putchar(c)
+#define getchar() 	ao_getchar()
+#endif
+
+/*
+ * Basic I/O functions to support SDCC stdio package
+ */
+
+#define AO_NUM_STDIOS	(HAS_USB + PACKET_HAS_SLAVE + USE_SERIAL_STDIN)
+
+__xdata struct ao_stdio ao_stdios[AO_NUM_STDIOS];
+__data int8_t ao_cur_stdio;
+__data int8_t ao_num_stdios;
+
+void
+putchar(char c)
+{
+	if (c == '\n')
+		(*ao_stdios[ao_cur_stdio].putchar)('\r');
+	(*ao_stdios[ao_cur_stdio].putchar)(c);
+}
+
+void
+flush(void)
+{
+	if (ao_stdios[ao_cur_stdio].flush)
+		ao_stdios[ao_cur_stdio].flush();
+}
+
+__xdata uint8_t ao_stdin_ready;
+
+char
+ao_getchar(void) __reentrant __critical
+{
+	char c;
+	int8_t stdio = ao_cur_stdio;
+
+	for (;;) {
+		c = ao_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);
+	}
+	ao_cur_stdio = stdio;
+	return c;
+}
+
+uint8_t
+ao_echo(void)
+{
+	return ao_stdios[ao_cur_stdio].echo;
+}
+
+int8_t
+ao_add_stdio(char (*pollchar)(void),
+	     void (*putchar)(char),
+	     void (*flush)(void)) __reentrant
+{
+	if (ao_num_stdios == AO_NUM_STDIOS)
+		ao_panic(AO_PANIC_STDIO);
+	ao_stdios[ao_num_stdios].pollchar = pollchar;
+	ao_stdios[ao_num_stdios].putchar = putchar;
+	ao_stdios[ao_num_stdios].flush = flush;
+	ao_stdios[ao_num_stdios].echo = 1;
+	return ao_num_stdios++;
+}
+
+#ifdef AVR
+int
+stdio_put(char c, FILE *stream)
+{
+	if (ao_cur_task && ao_num_stdios)
+		putchar(c);
+	else
+	{
+		if (c == '\n')
+			stdio_put('\r', stream);
+		loop_until_bit_is_set(UCSR1A, UDRE1);
+		UDR1 = c;
+	}
+
+	return 0;
+}
+
+int
+stdio_get(FILE *stream)
+{
+	return (int) getchar() & 0xff;
+}
+
+static FILE mystdout = FDEV_SETUP_STREAM(stdio_put, NULL, _FDEV_SETUP_WRITE);
+
+static FILE mystdin = FDEV_SETUP_STREAM(NULL, stdio_get, _FDEV_SETUP_READ);
+
+void
+ao_stdio_init(void)
+{
+	stdout = &mystdout;
+	stdin = &mystdin;
+	printf("%d stdios registered\n", ao_num_stdios);
+}
+#endif