/*
 * 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"

/*
 * 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 = 253;
#endif

    /* Set up the one wire stuff */
    OWInit();

    /* Init UART */
    cons_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"));

    tempctrl_init();
    
    /* 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"
#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 {
	printf_P(PSTR("Unknown command, help for a list\r\n"));
    }
}