#ifndef I2C_H
#define I2C_H

#include "std.h"

#ifndef SITL
#include "i2c_hw.h"
#endif

extern void i2c_init(void);
extern void i2c_receive(uint8_t slave_addr, uint8_t len, volatile bool_t* finished);
extern void i2c_transmit(uint8_t slave_addr, uint8_t len, volatile bool_t* finished);

#define I2C_START        0x08
#define I2C_RESTART      0x10
#define I2C_MT_SLA_ACK   0x18
#define I2C_MT_SLA_NACK  0x20
#define I2C_MT_DATA_ACK  0x28
#define I2C_MR_SLA_ACK   0x40
#define I2C_MR_SLA_NACK  0x48
#define I2C_MR_DATA_ACK  0x50
#define I2C_MR_DATA_NACK 0x58


#define I2C_IDLE 0
#define I2C_BUSY 1

extern volatile uint8_t i2c_status;

#ifndef I2C_BUF_LEN
#define I2C_BUF_LEN 8
#endif

extern volatile uint8_t i2c_buf[I2C_BUF_LEN];
extern volatile uint8_t i2c_len;
extern volatile uint8_t i2c_index;
extern volatile uint8_t i2c_slave_addr;

extern volatile bool_t* i2c_finished;

#define I2cAutomaton(state) {						\
    switch (state) {							\
    case I2C_START:							\
    case I2C_RESTART:							\
      I2cSendByte(i2c_slave_addr);					\
      I2cClearStart();							\
      i2c_index = 0;							\
      break;								\
    case I2C_MR_DATA_ACK:						\
      if (i2c_index < i2c_len) {					\
	i2c_buf[i2c_index] = I2C_DATA_REG;				\
	i2c_index++;							\
	I2cReceive(i2c_index < i2c_len - 1);				\
      }									\
      else {								\
	/* error , we should have got NACK */				\
	I2cSendStop();							\
      }									\
      break;								\
    case I2C_MR_SLA_ACK: /* At least one char */			\
      /* Wait and reply with ACK or NACK */				\
      I2cReceive(i2c_index < i2c_len - 1);				\
      break;								\
    case I2C_MR_SLA_NACK:						\
    case I2C_MT_SLA_NACK:						\
      I2cSendStart();							\
      break;								\
    case I2C_MT_SLA_ACK:						\
    case I2C_MT_DATA_ACK:						\
      if (i2c_index < i2c_len) {					\
	I2cSendByte(i2c_buf[i2c_index]);				\
	i2c_index++;							\
      } else {								\
	I2cSendStop();							\
      }									\
      break;								\
    case I2C_MR_DATA_NACK:						\
      if (i2c_index < i2c_len) {					\
	i2c_buf[i2c_index] = I2C_DATA_REG;				\
      }									\
      I2cSendStop();							\
      break;								\
    default:								\
      I2cSendStop();							\
      /* LED_ON(2); FIXME log error */					\
    }									\
  }									\
   
#endif /* I2C_H */