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

#include "twi.h"

static uint8_t TWI_buf[TWI_BUFFER_SIZE];     // Transceiver buffer. Set the size in the header file
static uint8_t TWI_msgSize  = 0;             // Number of bytes to be transmitted.
static uint8_t TWI_state    = TWI_NO_STATE;  // State byte. Default set to TWI_NO_STATE.

// This is true when the TWI is in the middle of a transfer
// and set to false when all bytes have been transmitted/received
// Also used to determine how deep we can sleep.
static volatile uint8_t TWI_busy = 0;

union TWI_statusReg_t TWI_statusReg = {0};           // TWI_statusReg is defined in twi.h

/****************************************************************************
Call this function to set up the TWI slave to its initial standby state.
Remember to enable interrupts from the main application after initializing the TWI.
Pass both the slave address and the requrements for triggering on a general call in the
same byte. Use e.g. this notation when calling this function:
TWI_Slave_Initialise((TWI_slaveAddress<<TWI_ADR_BITS) | (TRUE<<TWI_GEN_BIT));
The TWI module is configured to NACK on any requests. Use a TWI_Start_Transceiver function to 
start the TWI.
****************************************************************************/
void TWI_Slave_Initialize(uint8_t TWI_ownAddress)
{
    TWAR = TWI_ownAddress;                            // Set own TWI slave address. Accept TWI General Calls.
    TWCR = (1<<TWEN)|                                 // Enable TWI-interface and release TWI pins.
                  (0<<TWIE)|(0<<TWINT)|                      // Disable TWI Interupt.
                  (0<<TWEA)|(0<<TWSTA)|(0<<TWSTO)|           // Do not ACK on any requests, yet.
                  (0<<TWWC);                                 //
    TWI_busy = 0;
}    
        
/****************************************************************************
Call this function to test if the TWI_ISR is busy transmitting.
****************************************************************************/
uint8_t TWI_Transceiver_Busy(void)
{
    return TWI_busy;
}

/****************************************************************************
Call this function to fetch the state information of the previous operation. The function will hold execution (loop)
until the TWI_ISR has completed with the previous operation. If there was an error, then the function 
will return the TWI State code. 
****************************************************************************/
uint8_t TWI_Get_State_Info(void)
{
    while (TWI_Transceiver_Busy()) {}             // Wait until TWI has completed the transmission.
    return (TWI_state);                         // Return error state. 
}

/****************************************************************************
Call this function to send a prepared message, or start the Transceiver for reception. Include
a pointer to the data to be sent if a SLA+W is received. The data will be copied to the TWI buffer. 
Also include how many bytes that should be sent. Note that unlike the similar Master function, the
Address byte is not included in the message buffers.
The function will hold execution (loop) until the TWI_ISR has completed with the previous operation,
then initialize the next operation and return.
****************************************************************************/
void TWI_Start_Transceiver_With_Data(uint8_t *msg, uint8_t msgSize)
{
    uint8_t temp;

    while (TWI_Transceiver_Busy()) {}             // Wait until TWI is ready for next transmission.

    TWI_msgSize = msgSize;                        // Number of data to transmit.
    for (temp = 0; temp < msgSize; temp++)      // Copy data that may be transmitted if the TWI Master requests data.
    {
        TWI_buf[ temp ] = msg[ temp ];
    }
    TWI_statusReg.all = 0;      
    TWI_state         = TWI_NO_STATE ;
    TWCR = (1<<TWEN)|                             // TWI Interface enabled.
                  (1<<TWIE)|(1<<TWINT)|                  // Enable TWI Interupt and clear the flag.
                  (1<<TWEA)|(0<<TWSTA)|(0<<TWSTO)|       // Prepare to ACK next time the Slave is addressed.
                  (0<<TWWC);                             //
    TWI_busy = 1;
}

/****************************************************************************
Call this function to start the Transceiver without specifing new transmission data. Useful for restarting
a transmission, or just starting the transceiver for reception. The driver will reuse the data previously put
in the transceiver buffers. The function will hold execution (loop) until the TWI_ISR has completed with the 
previous operation, then initialize the next operation and return.
****************************************************************************/
void TWI_Start_Transceiver(void)
{
    while (TWI_Transceiver_Busy()) {}             // Wait until TWI is ready for next transmission.
    TWI_statusReg.all = 0;      
    TWI_state         = TWI_NO_STATE ;
    TWCR = (1<<TWEN)|                             // TWI Interface enabled.
                  (1<<TWIE)|(1<<TWINT)|                  // Enable TWI Interupt and clear the flag.
                  (1<<TWEA)|(0<<TWSTA)|(0<<TWSTO)|       // Prepare to ACK next time the Slave is addressed.
                  (0<<TWWC);                             //
    TWI_busy = 0;
}
/****************************************************************************
Call this function to read out the received data from the TWI transceiver buffer. I.e. first call
TWI_Start_Transceiver to get the TWI Transceiver to fetch data. Then Run this function to collect the
data when they have arrived. Include a pointer to where to place the data and the number of bytes
to fetch in the function call. The function will hold execution (loop) until the TWI_ISR has completed 
with the previous operation, before reading out the data and returning.
If there was an error in the previous transmission the function will return the TWI State code.
****************************************************************************/
uint8_t TWI_Get_Data_From_Transceiver(uint8_t *msg, uint8_t msgSize)
{
    uint8_t i;

    while (TWI_Transceiver_Busy()) {}             // Wait until TWI is ready for next transmission.

    if(TWI_statusReg.lastTransOK)               // Last transmission completed successfully.              
    {                                             
        for (i=0; i<msgSize; i++)                 // Copy data from Transceiver buffer.
        {
            msg[ i ] = TWI_buf[ i ];
        }
        TWI_statusReg.RxDataInBuf = FALSE;          // Slave Receive data has been read from buffer.
    }
    return(TWI_statusReg.lastTransOK);                                   
}


// ********** Interrupt Handlers ********** //
/****************************************************************************
This function is the Interrupt Service Routine (ISR), and called when the TWI interrupt is triggered;
that is whenever a TWI event has occurred. This function should not be called directly from the main
application.
****************************************************************************/
ISR(TWI_vect)
{
    static uint8_t TWI_bufPtr;
    
    switch (TWSR)
    {
        case TWI_STX_ADR_ACK:            // Own SLA+R has been received; ACK has been returned
            TWI_bufPtr   = 0;                                 // Set buffer pointer to first data location
        case TWI_STX_DATA_ACK:           // Data byte in TWDR has been transmitted; ACK has been received
            TWDR = TWI_buf[TWI_bufPtr++];
            TWCR = (1<<TWEN)|                                 // TWI Interface enabled
                   (1<<TWIE)|(1<<TWINT)|                      // Enable TWI Interupt and clear the flag to send byte
                   (1<<TWEA)|(0<<TWSTA)|(0<<TWSTO)|           // 
                   (0<<TWWC);                                 //
            TWI_busy = 1;
            break;
        case TWI_STX_DATA_NACK:          // Data byte in TWDR has been transmitted; NACK has been received. 
                                                                          // I.e. this could be the end of the transmission.
            if (TWI_bufPtr == TWI_msgSize) // Have we transceived all expected data?
                TWI_statusReg.lastTransOK = TRUE;               // Set status bits to completed successfully. 
            else                          // Master has sent a NACK before all data where sent.
                TWI_state = TWSR;                               // Store TWI State as errormessage.      
                                                                                                                
            TWCR = (1<<TWEN)|                                 // Enable TWI-interface and release TWI pins
                   (1<<TWIE)|(1<<TWINT)|                      // Keep interrupt enabled and clear the flag
                   (1<<TWEA)|(0<<TWSTA)|(0<<TWSTO)|           // Answer on next address match
                   (0<<TWWC);                                 //
            
            TWI_busy = 0;   // Transmit is finished, we are not busy anymore
            break;     
        case TWI_SRX_GEN_ACK:            // General call address has been received; ACK has been returned
//    case TWI_SRX_GEN_ACK_M_ARB_LOST: // Arbitration lost in SLA+R/W as Master; General call address has been received; ACK has been returned
            TWI_statusReg.genAddressCall = TRUE;
        case TWI_SRX_ADR_ACK:            // Own SLA+W has been received ACK has been returned
//    case TWI_SRX_ADR_ACK_M_ARB_LOST: // Arbitration lost in SLA+R/W as Master; own SLA+W has been received; ACK has been returned    
                                                                                                                // Dont need to clear TWI_S_statusRegister.generalAddressCall due to that it is the default state.
            TWI_statusReg.RxDataInBuf = TRUE;      
            TWI_bufPtr   = 0;                                 // Set buffer pointer to first data location
            
                                                                                                                // Reset the TWI Interupt to wait for a new event.
            TWCR = (1<<TWEN)|                                 // TWI Interface enabled
                   (1<<TWIE)|(1<<TWINT)|                      // Enable TWI Interupt and clear the flag to send byte
                   (1<<TWEA)|(0<<TWSTA)|(0<<TWSTO)|           // Expect ACK on this transmission
                   (0<<TWWC);  
            TWI_busy = 1;
            
            break;
        case TWI_SRX_ADR_DATA_ACK:       // Previously addressed with own SLA+W; data has been received; ACK has been returned
        case TWI_SRX_GEN_DATA_ACK:       // Previously addressed with general call; data has been received; ACK has been returned
            TWI_buf[TWI_bufPtr++]     = TWDR;
            TWI_statusReg.lastTransOK = TRUE;                 // Set flag transmission successfull.       
                                                                                                                // Reset the TWI Interupt to wait for a new event.
            TWCR = (1<<TWEN)|                                 // TWI Interface enabled
                   (1<<TWIE)|(1<<TWINT)|                      // Enable TWI Interupt and clear the flag to send byte
                   (1<<TWEA)|(0<<TWSTA)|(0<<TWSTO)|           // Send ACK after next reception
                   (0<<TWWC);                                 // 
            TWI_busy = 1;
            break;
        case TWI_SRX_STOP_RESTART:       // A STOP condition or repeated START condition has been received while still addressed as Slave    
                                                                                                                // Enter not addressed mode and listen to address match
            TWCR = (1<<TWEN)|                                 // Enable TWI-interface and release TWI pins
                   (1<<TWIE)|(1<<TWINT)|                      // Enable interrupt and clear the flag
                   (1<<TWEA)|(0<<TWSTA)|(0<<TWSTO)|           // Wait for new address match
                   (0<<TWWC);                                 //
            
            TWI_busy = 0;  // We are waiting for a new address match, so we are not busy
            
            break;           
        case TWI_SRX_ADR_DATA_NACK:      // Previously addressed with own SLA+W; data has been received; NOT ACK has been returned
        case TWI_SRX_GEN_DATA_NACK:      // Previously addressed with general call; data has been received; NOT ACK has been returned
        case TWI_STX_DATA_ACK_LAST_BYTE: // Last data byte in TWDR has been transmitted (TWEA = “0”); ACK has been received
//    case TWI_NO_STATE              // No relevant state information available; TWINT = “0”
        case TWI_BUS_ERROR:         // Bus error due to an illegal START or STOP condition
            TWI_state = TWSR;                 //Store TWI State as errormessage, operation also clears noErrors bit
            TWCR =   (1<<TWSTO)|(1<<TWINT);   //Recover from TWI_BUS_ERROR, this will release the SDA and SCL pins thus enabling other devices to use the bus
            break;
        default:     
            TWI_state = TWSR;                                 // Store TWI State as errormessage, operation also clears the Success bit.      
            TWCR = (1<<TWEN)|                                 // Enable TWI-interface and release TWI pins
                   (1<<TWIE)|(1<<TWINT)|                      // Keep interrupt enabled and clear the flag
                   (1<<TWEA)|(0<<TWSTA)|(0<<TWSTO)|           // Acknowledge on any new requests.
                   (0<<TWWC);                                 //
            
            TWI_busy = 0; // Unknown status, so we wait for a new address match that might be something we can handle
    }
}