/*
 * 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 (ao_cur_stdio >= ao_num_stdios || !ao_stdios[ao_cur_stdio].putchar)
		return;
	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 (;;) {
		if (stdio < ao_num_stdios) {
			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;
	ao_wakeup(&ao_stdin_ready);
	return ao_num_stdios++;
}

#ifdef AVR
int
stdio_put(char c, FILE *stream)
{
#if 1
	if (ao_cur_task && ao_num_stdios)
		putchar(c);
	else
#endif
	{
		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;
}
#endif