/******************************************************************

 * reads some digital inputs and make them readable over i2c bus.
 *
 *                         Atmega 32
 *                         +-------+
 *  S17      (XCK/T0) PB0 -|1  O 40|- PA0 (ADC0) S9
 *  S18          (T1) PB1 -|2    39|- PA1 (ADC1) S10
 *  S19   (INT2/AIN0) PB2 -|3    38|- PA2 (ADC2) S11
 *  S20    (OC0/AIN1) PB3 -|4    37|- PA3 (ADC3) S12
 *  S21          (SS) PB4 -|5    36|- PA4 (ADC4) S13
 *  S22        (MOSI) PB5 -|6    35|- PA5 (ADC5) S14
 *  S23        (MISO) PB6 -|7    34|- PA6 (ADC6) S15
 *  S24         (SCK) PB7 -|8    33|- PA7 (ADC7) S16
 *                  RESET -|9    32|- AREF
 *                    VCC -|10   31|- GND
 *                    GND -|11   30|- AVCC
 *                  XTAL2 -|12   29|- PC7 (TOSC2)
 *                  XTAL1 -|13   28|- PC6 (TOSC1)
 *  S1          (RXD) PD0 -|14   27|- PC5 (TDI)
 *  S2          (TXD) PD1 -|15   26|- PC4 (TDO)
 *  S3         (INT0) PD2 -|16   25|- PC3 (TMS)
 *  S4         (INT1) PD3 -|17   24|- PC2 (TCK)
 *  S5         (OC1B) PD4 -|18   23|- PC1 (DSA)   I2C
 *  S6         (OC1A) PD5 -|19   22|- PC0 (SCL)   I2C
 *  S7         (ICP1) PD6 -|20   21|- PD7 (OC2)   S8
 *                         +-------+ */

#include <avr/io.h>
#include <avr/interrupt.h>
#include <stdarg.h>
#include <stdio.h>

#include <util/delay.h>
#include <util/twi.h>

#include "eprom.h"

#define TWCR_ACK TWCR = (1<<TWEN)|(1<<TWIE)|(1<<TWINT)|(1<<TWEA);
#define TWCR_NACK TWCR = (1<<TWEN)|(1<<TWIE)|(1<<TWINT)
#define TWCR_RESET TWCR = (1<<TWEN)|(1<<TWIE)|(1<<TWINT)|(1<<TWEA)|(1<<TWSTO);  

#define I2C_NUM_REGS 8
#define I2C_DEFAULT_ADDR 0x2F
#define VERSION 4

#define I2C_SREGS_OFFSET 0x20

enum {
	I2C_SREG_VERSION = 0,
	I2C_SREG_CMD,
	I2C_SREG_ADDR,
	I2C_SREG_DIOFFDELAY,

	I2C_SREG_MAX
};

enum {
	I2C_CMD_NONE = 0,
	I2C_CMD_EEPROMREAD,
	I2C_CMD_EEPROMWRITE
};

volatile uint8_t i2c_reg;
volatile uint8_t i2c_system[I2C_SREG_MAX];
volatile uint8_t i2c_regs[I2C_NUM_REGS];

//
// i2c slave addresse festlegen und interrupts einschalten
void i2c_slave_init(uint8_t adr) {
	int i;
	
	for (i = 0; i < I2C_NUM_REGS; i++)  {
		i2c_regs[i] = 0;
	}
	
	TWAR = adr << 1;
	TWCR &= ~(1<<TWSTA) | (1<<TWSTO);
	TWCR |= (1<<TWEA) | (1<<TWEN) | (1<<TWIE);

	i2c_reg = 0x0;
	
	sei();
}

ISR (TWI_vect) {
	char data=0;

	switch (TW_STATUS) {
		case TW_SR_SLA_ACK: // 0x60 Slave Receiver, Slave wurde adressiert	
			TWCR_ACK;
			i2c_reg=0xFF;
			break;

		case TW_SR_DATA_ACK: // 0x80 Slave Receiver, ein Datenbyte wurde empfangen
			data=TWDR;
			if (i2c_reg == 0xFF) {
				if(data < I2C_NUM_REGS || (data >= I2C_SREGS_OFFSET && data < I2C_SREGS_OFFSET+I2C_SREG_MAX)) {
					i2c_reg = data;
					TWCR_ACK;
				}
				else 
					TWCR_NACK;
			}
			else {
				if(i2c_reg < I2C_NUM_REGS) {
					i2c_regs[i2c_reg] = data;
					i2c_reg++;
					TWCR_ACK;
				}
				else if (i2c_reg >= I2C_SREGS_OFFSET && i2c_reg < I2C_SREGS_OFFSET+I2C_SREG_MAX) {
					i2c_system[i2c_reg-I2C_SREGS_OFFSET] = data;
					i2c_reg++;
					TWCR_ACK;
				}
				else TWCR_NACK;
			}
			break;

		case TW_ST_SLA_ACK:  //0xA8 Slave wurde im Lesemodus adressiert und hat ein ACK zurückgegeben.
		case TW_ST_DATA_ACK: //0xB8 Slave Transmitter, Daten wurden angefordert
			if (i2c_reg == 0xFF) {
				i2c_reg = 0;
			}	
			if (i2c_reg < I2C_NUM_REGS) {
				TWDR = i2c_regs[i2c_reg];
				i2c_reg++;
				TWCR_ACK;
			}
			else if (i2c_reg >= I2C_SREGS_OFFSET && i2c_reg < I2C_SREGS_OFFSET+I2C_SREG_MAX) {
				TWDR = i2c_system[i2c_reg-I2C_SREGS_OFFSET];
				i2c_reg++;
				TWCR_ACK;
			}
			else TWCR_NACK;
			break;
	
		case TW_SR_STOP:
			TWCR_ACK;
	        break;

		case TW_ST_DATA_NACK: // 0xC0 Keine Daten mehr gefordert 
		case TW_SR_DATA_NACK: // 0x88 
		case TW_ST_LAST_DATA: // 0xC8  Last data byte in TWDR has been transmitted (TWEA = “0”); ACK has been received
		default: 	
		    TWCR_RESET;
			break;
	}
}

/////////////////////////////////////////////////////////////////////////
// 

int main( void ) {
	uint32_t cnt[32] = { 0 };
	unsigned int i;
	uint32_t inval = 0;
	uint32_t temp = 0;
	int offdelay;

	//
	// disable JTAG
	MCUCSR |= (1<<JTD);
	MCUCSR |= (1<<JTD);

	// enable inputs
	DDRA = 0;
	PORTA = 0xFF;
	DDRB = 0;
	PORTB = 0xFF;
	DDRC = 0;
	PORTC = 0xFF;
	DDRD = 0;
	PORTD = 0xFF;

	//
	// read all variables from eeprom
	i2c_system[I2C_SREG_VERSION] = VERSION;
	i2c_system[I2C_SREG_CMD] = 0;

	i2c_system[I2C_SREG_ADDR] = EEPROM_read(I2C_SREG_ADDR);
	if (i2c_system[I2C_SREG_ADDR] > 0x4f || i2c_system[I2C_SREG_ADDR] < 0x20) {
		i2c_system[I2C_SREG_ADDR] = I2C_DEFAULT_ADDR;
	}

	i2c_system[I2C_SREG_DIOFFDELAY] = EEPROM_read(I2C_SREG_DIOFFDELAY);
	if (i2c_system[I2C_SREG_DIOFFDELAY] == 0 || i2c_system[I2C_SREG_DIOFFDELAY] == 0xFF) {
		i2c_system[I2C_SREG_DIOFFDELAY] = 0x20;
	}

	i2c_slave_init (i2c_system[I2C_SREG_ADDR]);

	for (i = 0; i < 32; i++) cnt[i] = 0;

	//
	// system loop
    while( 1 ) {
    	//
    	// digital in registers..
    	offdelay = i2c_system[I2C_SREG_DIOFFDELAY] << 8;

		//  S1 - S8
    	i = 0; if (!(PINB & (1 << 0))) cnt[i] = offdelay;
    	i++; if (!(PINB & (1 << 1))) cnt[i] = offdelay;
		i++; if (!(PINB & (1 << 2))) cnt[i] = offdelay;
		i++; if (!(PINB & (1 << 3))) cnt[i] = offdelay;
		i++; if (!(PINB & (1 << 4))) cnt[i] = offdelay;
		i++; if (!(PINB & (1 << 5))) cnt[i] = offdelay;
		i++; if (!(PINB & (1 << 6))) cnt[i] = offdelay;
		i++; if (!(PINB & (1 << 7))) cnt[i] = offdelay;

		//  S9 - S16
		i++; if (!(PINA & (1 << 0))) cnt[i] = offdelay;
		i++; if (!(PINA & (1 << 1))) cnt[i] = offdelay;
		i++; if (!(PINA & (1 << 2))) cnt[i] = offdelay;
		i++; if (!(PINA & (1 << 3))) cnt[i] = offdelay;
		i++; if (!(PINA & (1 << 4))) cnt[i] = offdelay;
		i++; if (!(PINA & (1 << 5))) cnt[i] = offdelay;
		i++; if (!(PINA & (1 << 6))) cnt[i] = offdelay;
		i++; if (!(PINA & (1 << 7))) cnt[i] = offdelay;

		//  S17 - S24
		i++; if (!(PIND & (1 << 0))) cnt[i] = offdelay;
		i++; if (!(PIND & (1 << 1))) cnt[i] = offdelay;
		i++; if (!(PIND & (1 << 2))) cnt[i] = offdelay;
		i++; if (!(PIND & (1 << 3))) cnt[i] = offdelay;
		i++; if (!(PIND & (1 << 4))) cnt[i] = offdelay;
		i++; if (!(PIND & (1 << 5))) cnt[i] = offdelay;
		i++; if (!(PIND & (1 << 6))) cnt[i] = offdelay;
		i++; if (!(PIND & (1 << 7))) cnt[i] = offdelay;

    	// all values
		for (i = 0; i < 16; i++) {
			if (cnt[i] == 0) inval &= ~(1 << (i));
			else {
				cnt[i]--;
				inval |= (1 << (i));
			}
		}

		for (i = 0; i < 16; i++) {
			if (cnt[16+i] == 0) temp &= ~(1 << (i));
			else {
				cnt[16+i]--;
				temp |= (1 << (i));
			}
		}

		i2c_regs[0] = (unsigned char) (inval & 0x000000FF);
		i2c_regs[1] = (unsigned char) ((inval & 0x0000FF00) >> 8);
		i2c_regs[2] = (unsigned char) (temp & 0x000000FF);
		i2c_regs[3] = (unsigned char) ((temp & 0x0000FF00) >> 8);


		//
		// system registers
		if (i2c_system[I2C_SREG_CMD] != 0) {
			if (i2c_system[I2C_SREG_CMD] == I2C_CMD_EEPROMREAD) {
				cli();
				i2c_system[I2C_SREG_CMD] = 0;
				for (i = 0; i < I2C_SREG_MAX; i++) {
					i2c_system[i] = EEPROM_read(i);
				}
				sei();
			}

			if (i2c_system[I2C_SREG_CMD] == I2C_CMD_EEPROMWRITE) {
				cli();
				i2c_system[I2C_SREG_CMD] = 0;
				for (i = 0; i < I2C_SREG_MAX; i++) {
					EEPROM_write(i, i2c_system[i]);
				}
				sei();
			}
		}
		i2c_system[I2C_SREG_CMD] = 0;
    }
    return 0;
};