/*

 * Test various AVR bits and pieces

 *

 * Copyright (c) 2008

 *      Daniel O'Connor <darius@dons.net.au>.  All rights reserved.

 *

 * Redistribution and use in source and binary forms, with or without

 * modification, are permitted provided that the following conditions

 * are met:

 * 1. Redistributions of source code must retain the above copyright

 *    notice, this list of conditions and the following disclaimer.

 * 2. Redistributions in binary form must reproduce the above copyright

 *    notice, this list of conditions and the following disclaimer in the

 *    documentation and/or other materials provided with the distribution.

 *

 * THIS SOFTWARE IS PROVIDED BY AUTHOR AND CONTRIBUTORS ``AS IS'' AND

 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE

 * ARE DISCLAIMED.  IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE

 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL

 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS

 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)

 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY

 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF

 * SUCH DAMAGE.

 */

#include <avr/io.h>

#include <avr/interrupt.h>

#include <avr/pgmspace.h>

#include <stdio.h>

#include <string.h>

#include <ctype.h>

#include <stdlib.h>

#include <util/delay.h>

#include <avr/wdt.h>

#include "cons.h"

#include "1wire.h"

#ifdef WITHUSB

#include "usb.h"

#endif

#include "tempctrl.h"

#include "ds1307.h"

/*

 * Mirror of the MCUCSR register, taken early during startup.

 */

uint8_t mcucsr __attribute__((section(".noinit")));

void		process_cmd(void);

/*

 * Read out and reset MCUCSR early during startup.

 */

void handle_mcucsr(void)

    __attribute__((section(".init3")))

    __attribute__((naked));

void handle_mcucsr(void) {

    wdt_disable();

    mcucsr = MCUCSR;

    MCUCSR = 0;

}

int

main(void) {

    /* Disable interrupts while we frob stuff */

    cli();

#if 0

    /* Disable JTAG (yes twice) */

    MCUCSR |= _BV(JTD);

    MCUCSR |= _BV(JTD);

#endif

#ifdef WITHUSB

    /* USB data bus (7:0) */

    DDRA = 0x00;

    PORTA = 0x00;

    /* USB control (4:0) */

    DDRB = 0x0d;

    PORTB = 0x00;

#else

    DDRA = 0xff;

    PORTA = 0x00;

    DDRB = 0x00;

    PORTB = 0x00;

#endif    

    /* GPIO (0:7) */

    DDRC = 0xff;

    PORTC = 0x00;

    /* USART (0:1), IDBus (2:5), 485 (6:6), GPIO (7:7) */

    DDRD = 0xf7;

    PORTD = 0xf7;

#ifndef WITHUSB

    /* Beep

     *

     * Fpwm = Fclk / (N * 510)

     *      = 16e6 / (8 * 510)

     *      = 3921Hz

     *

     * Behind ifndef because PB3 is A0 on PDI chip

     *

     * Phase correct PWM, non-inverted output, divide by 8 */

    TCCR0 = _BV(WGM00) | _BV(WGM11) | _BV(COM00) | _BV(COM01) | _BV(CS01);

    OCR0 = 128;

#endif

    /* Set up the one wire stuff */

    OWInit();

    /* Setup IIC */

    ds1307_init();

    /* Init UART */

    cons_init();

    /* Init temperature control stuff */

    tempctrl_init();

    printf_P(PSTR("\r\n\r\n===============\r\n"

		  "Inited!\r\n\r\n"));

    if ((mcucsr & _BV(PORF)) == _BV(PORF))

	printf_P(PSTR("Power on reset\r\n"));

    if ((mcucsr & _BV(EXTRF)) == _BV(EXTRF))

	printf_P(PSTR("External reset\r\n"));

    if ((mcucsr & _BV(BORF)) == _BV(BORF))

	printf_P(PSTR("Brown-out reset\r\n"));

    if ((mcucsr & _BV(WDRF)) == _BV(WDRF))

	printf_P(PSTR("Watchdog reset\r\n"));

    if ((mcucsr & _BV(JTRF)) == _BV(JTRF))

	printf_P(PSTR("JTAG reset\r\n"));

    /* Ready to go! */

    sei();

    /*

     * Enable the watchdog with the largest prescaler.  Will cause a

     * watchdog reset after approximately 2 s @ Vcc = 5 V

     *

     * Gets reset in the loop below and in the tempctrl.c timer IRQ

     */

    wdt_enable(WDTO_2S);

#ifdef WITHUSB

    printf_P(PSTR("Calling usb_init\r\n"));

    usb_init();

    printf_P(PSTR("done\r\n"));

    _delay_us(1000);

#endif

    printf_P(PSTR("> "));

    cmd.state = 0;

    /* Wait for user input or an "interrupt" */

    while (1) {

	wdt_reset();

	tempctrl_update();

	if (cmd.state == 255) {

	    process_cmd();

	    printf_P(PSTR("> "));

	    /* Allow new characters to be processed */

	    cmd.state = 0;

	}

#ifdef WITHUSB

	if (!(PDICTL & _BV(PDIINT)))

	    usb_intr();

#endif

    }

}

void

process_cmd(void) {

    uint8_t	ROM[8], crc, buf[9], temp;

    int8_t	i, arg;

    int16_t	t;

    /* User just pressed enter */

    if (cmd.len == 0)

	return;

    if (!strcasecmp_P((char *)cmd.buf, PSTR("?")) ||

	!strcasecmp_P((char *)cmd.buf, PSTR("help"))) {

        printf_P(PSTR("rs               Reset and check for presence\r\n"

		      "sr               Search the bus for ROMs\r\n"

		      "re               Read a bit\r\n"

		      "rb               Read a byte\r\n"

		      "wr bit           Write a bit\r\n"

		      "wb byte          Write a byte (hex)\r\n"

		      "wc cmd [ROMID]   Write command\r\n"

		      "te ROMID         Read the temperature from a DS1820\r\n"

		      "in port          Read from a port\r\n"

		      "ou port val      Write to a port (val in hex)\r\n"

		      "dd port [val]    Read/write DDR for a port (val in hex)\r\n"

		      "rt ROMID	        Read DS2502 status page\r\n"

		      "we ROMID adr val Write data into a DS2502 PROM (adr & val in hex)\r\n"

		      "rr ROMID         Read DS2502 PROM\r\n"

		      "zz               Reset MCU\r\n"

		      "gc               Get time of day\r\n"

		      "sc time          Set time of day (time is YYYY/MM/DD HH:MM:SS)\r\n"

#ifdef WITHUSB

		      "us               Generate USB data\r\n"

#endif

		      "tc ...           Temperature control related (tc help for more)\r\n"));

	return;

    } else if (!strncasecmp_P((char *)cmd.buf, PSTR("zz"), 2)) {

	cli();

	wdt_enable(WDTO_15MS);

	for (;;)

	    ;

    } else if (!strncasecmp_P((char *)cmd.buf, PSTR("rs"), 2)) {

	printf_P(PSTR("Resetting... "));

	if (OWTouchReset() == 1)

	    printf_P(PSTR("No presence pulse found\r\n"));

	else

	    printf_P(PSTR("Presence pulse found\r\n"));

    } else if (!strncasecmp_P((char *)cmd.buf, PSTR("re"), 2)) {

	if (OWReadBit())

	    printf_P(PSTR("Read a 1\r\n"));

	else

	    printf_P(PSTR("Read a 0\r\n"));

    } else if (!strncasecmp_P((char *)cmd.buf, PSTR("rb"), 2)) {

	printf_P(PSTR("Read a 0x%02x\r\n"), OWReadByte());

    } else if (!strncasecmp_P((char *)cmd.buf, PSTR("wr"), 2)) {

	arg = strtol((char *)cmd.buf + 3, (char **)NULL, 10);

	OWWriteBit(arg);

	printf_P(PSTR("Wrote a %c\r\n"), arg ? '1' : '0');

    } else if (!strncasecmp_P((char *)cmd.buf, PSTR("wb"), 2)) {

	arg = (int)strtol((char *)cmd.buf + 3, (char **)NULL, 16); 

	OWWriteByte(arg);

    } else if (!strncasecmp_P((char *)cmd.buf, PSTR("rt"), 2)) {

	if (sscanf_P((char *)cmd.buf, PSTR("rt %hhx:%hhx:%hhx:%hhx:%hhx:%hhx:%hhx:%hhx"), 

		     &ROM[0], &ROM[1], &ROM[2], &ROM[3],

		     &ROM[4], &ROM[5], &ROM[6], &ROM[7]) != 8) {

	    printf_P(PSTR("Unable to parse ROM ID\r\n"));

	    return;

	}

	if (ROM[0] != OW_FAMILY_ROM) {

	    printf_P(PSTR("ROM specified isn't a DS2502\r\n"));

	    return;

	}

	if (OWTouchReset() != 0) {

	    printf_P(PSTR("No presence\r\n"));

	    return;

	}

	crc = 0;

	OWCRC(OW_READ_STATUS, &crc);

	OWSendCmd(ROM, OW_READ_STATUS);

	OWWriteByte(0x00);

	OWCRC(0x00, &crc);

	OWWriteByte(0x00);

	OWCRC(0x00, &crc);

	if (crc != OWReadByte()) {

	    printf_P(PSTR("CRC mismatch on command & address\r\n"));

	    return;

	}

	crc = 0;

	for (i = 0; i < 8; i++) {

	    temp = OWReadByte();

	    printf_P(PSTR("%02x "), temp);

	    OWCRC(temp, &crc);

	}

	printf_P(PSTR("\r\n"));

	if (crc != OWReadByte()) {

	    printf_P(PSTR("CRC mismatch on data\r\n"));

	    return;

	}

    } else if (!strncasecmp_P((char *)cmd.buf, PSTR("we"), 2)) {

	uint8_t	adr, data;

	if (sscanf_P((char *)cmd.buf, PSTR("we %hhx:%hhx:%hhx:%hhx:%hhx:%hhx:%hhx:%hhx %hhx %hhx"),

		     &ROM[0], &ROM[1], &ROM[2], &ROM[3],

		     &ROM[4], &ROM[5], &ROM[6], &ROM[7],

		     &adr, &data) != 10) {

	    printf_P(PSTR("Unable to parse ROM ID\r\n"));

	    return;

	}

	if (ROM[0] != OW_FAMILY_ROM) {

	    printf_P(PSTR("ID specified isn't a ROM\r\n"));

	    return;

	}

	if (OWTouchReset() != 0) {

	    printf_P(PSTR("No presence\r\n"));

	    return;

	}

	printf_P(PSTR("OWProgROM returned %S\r\n"), OWProgROM_Status[OWProgROM(ROM, buf[0], 2, &buf[1], 0, 0)]);

    } else if (!strncasecmp_P((char *)cmd.buf, PSTR("rr"), 2)) {

	if (sscanf_P((char *)cmd.buf, PSTR("rr %hhx:%hhx:%hhx:%hhx:%hhx:%hhx:%hhx:%hhx"),

		     &ROM[0], &ROM[1], &ROM[2], &ROM[3],

		     &ROM[4], &ROM[5], &ROM[6], &ROM[7]) != 8) {

	    printf_P(PSTR("Unable to parse ROM ID\r\n"));

	    return;

	}

	if (ROM[0] != OW_FAMILY_ROM) {

	    printf_P(PSTR("ROM specified isn't a ROM\r\n"));

	    return;

	}

	crc = 0;

	OWSendCmd(ROM, OW_READ_MEMORY);

	OWCRC(OW_READ_MEMORY, &crc);

	OWWriteByte(0x00);

	OWCRC(0x00, &crc);

	OWWriteByte(0x00);

	OWCRC(0x00, &crc);

	if (crc != OWReadByte()) {

	    printf_P(PSTR("CRC mismatch on command & address\r\n"));

	    return;

	}

	crc = 0;

	for (buf[0] = 0; buf[0] < 128; buf[0]++) {

	    buf[1] = OWReadByte();

	    if (buf[0] > 0) {

		if (buf[0] % 16 != 0)

		    printf_P(PSTR(" "));

		else

		    printf_P(PSTR("\r\n"));

	    }

	    printf_P(PSTR("%02x"), buf[1]);

	    OWCRC(buf[1], &crc);

	}

	printf_P(PSTR("\r\n"));

	if (crc != OWReadByte()) {

	    printf_P(PSTR("CRC mismatch on data\r\n"));

	    return;

	}

    } else if (!strncasecmp_P((char *)cmd.buf, PSTR("wc"), 2)) {

	uint8_t c;

	i = sscanf_P((char *)cmd.buf, PSTR("wc %hhx %hhx:%hhx:%hhx:%hhx:%hhx:%hhx:%hhx:%hhx"),

		     &ROM[0], &ROM[1], &ROM[2], &ROM[3],

		     &ROM[4], &ROM[5], &ROM[6], &ROM[7],

		     &c);

	if (i != 1 && i != 9) {

	    printf_P(PSTR("Incorrect usage\r\n"));

	    return;

	}

	if (i == 1) {

	    OWSendCmd(i == 1 ? NULL : ROM, c);

	    return;

	}

    } else if (!strncasecmp_P((char *)cmd.buf, PSTR("te"), 2)) {

	if (sscanf_P((char *)cmd.buf, PSTR("te %hhx:%hhx:%hhx:%hhx:%hhx:%hhx:%hhx:%hhx"), 

		     &ROM[0], &ROM[1], &ROM[2], &ROM[3],

		     &ROM[4], &ROM[5], &ROM[6], &ROM[7]) != 8) {

	    printf_P(PSTR("Unable to parse ROM ID\r\n"));

	    return;

	}

	t = OWGetTemp(ROM);

	switch (t) {

	    case OW_TEMP_WRONG_FAM:

		printf_P(PSTR("ROM specified isn't a temperature sensor\r\n"));

		break;

	    case OW_TEMP_CRC_ERR:

		printf_P(PSTR("CRC mismatch\r\n"));

		break;

	    case OW_TEMP_NO_ROM:

		printf_P(PSTR("No ROM found\r\n"));

		break;

	    default:

		printf_P(PSTR("%d.%02d\r\n"), GETWHOLE(t), GETFRAC(t));

		break;

	}

    } else if (!strncasecmp_P((char *)cmd.buf, PSTR("sr"), 2)) {

	memset(ROM, 0, 8);

	i = OWFirst(ROM, 1, 0);

	do {

	    switch (i) {

		case OW_BADWIRE:

		    printf_P(PSTR("Presence pulse, but no module found, bad module/cabling?\r\n"));

		    break;

		case OW_NOPRESENCE:

		    printf_P(PSTR("No presence pulse found\r\n"));

		    break;

		case OW_BADCRC:

		    printf_P(PSTR("Bad CRC\r\n"));

		    break;

		case OW_NOMODULES:

		case OW_FOUND:

		    break;

		default:

		    printf_P(PSTR("Unknown error from 1 wire library\r\n"));

		    break;

	    }

	    if (i != OW_FOUND)

		break;

	    for (i = 0; i < 8; i++)

		printf_P(PSTR("%02x%S"), ROM[i], i == 7 ? PSTR("\r\n") : PSTR(":"));

	    i = OWNext(ROM, 1, 0);

	} while (1);

    } else if (!strncasecmp_P((char *)cmd.buf, PSTR("in"), 2)) {

	uint8_t inp;

	switch (tolower(cmd.buf[3])) {

	    case 'a':

		inp = PINA;

		break;

	    case 'b':

		inp = PINB;

		break;

	    case 'c':

		inp = PINC;

		break;

	    case 'd':

		inp = PIND;

		break;

	    default:

		printf_P(PSTR("Unknown port\r\n"));

		return;

	}

	printf_P(PSTR("0x%02x\r\n"), inp);

    } else if (!strncasecmp_P((char *)cmd.buf, PSTR("ou"), 2)) {

	char port;

	int val;

	if (sscanf_P((char *)cmd.buf, PSTR("ou %c %x"), &port, &val) != 2) {

	    printf_P(PSTR("Unable to parse ou arguments\r\n"));

	    return;

	}

	switch (port) {

	    case 'a':

		PORTA = val & 0xff;

		break;

	    case 'b':

		PORTB = val & 0xff;

		break;

	    case 'c':

		PORTC = val & 0xff;

		break;

	    case 'd':

		PORTD = val & 0xff;

		break;

	    default:

		printf_P(PSTR("Unknown port\r\n"));

		return;

	}

	printf_P(PSTR("PORT%c <= 0x%02x\r\n"), toupper(port), val);

    } else if (!strncasecmp_P((char *)cmd.buf, PSTR("dd"), 2)) {

	char port;

	uint8_t val;

	int num;

	num = sscanf_P((char *)cmd.buf, PSTR("dd %c %x"), &port, &val);

	if (num != 2 && num != 3) {

	    printf_P(PSTR("Unable to parse dd arguments\r\n"));

	    return;

	}

	if (num == 2) {

	    switch (port) {

		case 'a':

		    val = DDRA;

		    break;

		case 'b':

		    val = DDRB;

		    break;

		case 'c':

		    val = DDRC;

		    break;

		case 'd':

		    val = DDRD;

		    break;

		default:

		    printf_P(PSTR("Unknown port\r\n"));

		    return;

	    }

	    printf_P(PSTR("DDR%c => 0x%02x\r\n"), toupper(port), val);

	} else {

	    switch (port) {

		case 'a':

		    DDRA = val & 0xff;

		    break;

		case 'b':

		    DDRB = val & 0xff;

		    break;

		case 'c':

		    DDRC = val & 0xff;

		    break;

		case 'd':

		    DDRD = val & 0xff;

		    break;

		default:

		    printf_P(PSTR("Unknown port\r\n"));

		    return;

	    }

	    printf_P(PSTR("DDR%c <= 0x%02x\r\n"), toupper(port), val);

	}

    } else if (!strncasecmp_P((char *)cmd.buf, PSTR("tc"), 2)) {

	tempctrl_cmd((char *)cmd.buf);

#ifdef WITHUSB

    } else if (!strncasecmp_P((char *)cmd.buf, PSTR("us"), 2)) {

	usb_gendata();

#endif

    } else if (!strncasecmp_P((char *)cmd.buf, PSTR("gc"), 2)) {

	ds1307_printtime(PSTR(""), PSTR("\r\n"));

    } else if (!strncasecmp_P((char *)cmd.buf, PSTR("sc"), 2)) {

	if (cmd.len < 17) {

	    printf_P(PSTR("Command not long enough\r\n"));

	} else {

	    if (ds1307_settod((char *)cmd.buf + 3) != 1)

		printf_P(PSTR("Unable to set time of day\r\n"));

	}

    } else {

	printf_P(PSTR("Unknown command, help for a list\r\n"));

    }

}