Subversion Repositories FlightCtrl

#include <avr/io.h>
#include <avr/interrupt.h>
#include <util/twi.h>
#include <util/delay.h>
#include "twimaster.h"
#include "configuration.h"
#include "analog.h"
#include "printf_P.h"

volatile uint8_t twi_state = TWI_STATE_MOTOR_TX;
volatile uint8_t dac_channel = 0;
volatile uint8_t motor_write = 0;
volatile uint8_t motor_read = 0;
volatile uint16_t I2CTimeout = 100;
uint8_t missingMotor = 0;

motorData_t motor[MAX_MOTORS];

uint8_t DACChannel = 0;

#define SCL_CLOCK  200000L
#define I2C_TIMEOUT 30000

/**************************************************
 * Initialize I2C (TWI)                        
 **************************************************/
void I2C_init(void) {
        uint8_t i;
        uint8_t sreg = SREG;
        cli();

        // SDA is INPUT
        DDRC &= ~(1 << DDC1);
        // SCL is output
        DDRC |= (1 << DDC0);
        // pull up SDA
        PORTC |= (1 << PORTC0) | (1 << PORTC1);

        // TWI Status Register
        // prescaler 1 (TWPS1 = 0, TWPS0 = 0)
        TWSR &= ~((1 << TWPS1) | (1 << TWPS0));

        // set TWI Bit Rate Register
        TWBR = ((SYSCLK / SCL_CLOCK) - 16) / 2;

        twi_state = TWI_STATE_MOTOR_TX;
        motor_write = 0;
        motor_read = 0;

        for (i = 0; i < MAX_MOTORS; i++) {
                motor[i].throttle = 0;
                motor[i].present = 0;
                motor[i].maxPWM = 0;
        }

        SREG = sreg;
}

/****************************************
 * Start I2C                          
 ****************************************/
void I2C_Start(uint8_t start_state) {
        twi_state = start_state;
        // TWI Control Register
        // clear TWI interrupt flag (TWINT=1)
        // disable TWI Acknowledge Bit (TWEA = 0)
        // enable TWI START Condition Bit (TWSTA = 1), MASTER
        // disable TWI STOP Condition Bit (TWSTO = 0)
        // disable TWI Write Collision Flag (TWWC = 0)
        // enable i2c (TWEN = 1)
        // enable TWI Interrupt (TWIE = 1)
        TWCR = (1 << TWINT) | (1 << TWSTA) | (1 << TWEN) | (1 << TWIE);
}

/****************************************
 * Stop I2C                          
 ****************************************/
void I2C_Stop(uint8_t start_state) {
        twi_state = start_state;
        // TWI Control Register
        // clear TWI interrupt flag (TWINT=1)
        // disable TWI Acknowledge Bit (TWEA = 0)
        // diable TWI START Condition Bit (TWSTA = 1), no MASTER
        // enable TWI STOP Condition Bit (TWSTO = 1)
        // disable TWI Write Collision Flag (TWWC = 0)
        // enable i2c (TWEN = 1)
        // disable TWI Interrupt (TWIE = 0)
        TWCR = (1 << TWINT) | (1 << TWSTO) | (1 << TWEN);
}

/****************************************
 *    Write to I2C                      
 ****************************************/
void I2C_WriteByte(int8_t byte) {
        // move byte to send into TWI Data Register
        TWDR = byte;
        // clear interrupt flag (TWINT = 1)
        // enable i2c bus (TWEN = 1)
        // enable interrupt (TWIE = 1)
        TWCR = (1 << TWINT) | (1 << TWEN) | (1 << TWIE);
}

/****************************************
 * Receive byte and send ACK        
 ****************************************/
void I2C_ReceiveByte(void) {
        TWCR = (1 << TWINT) | (1 << TWEN) | (1 << TWIE) | (1 << TWEA);
}

/****************************************
 * I2C receive last byte and send no ACK
 ****************************************/
void I2C_ReceiveLastByte(void) {
        TWCR = (1 << TWINT) | (1 << TWEN) | (1 << TWIE);
}

/****************************************
 * Reset I2C                        
 ****************************************/
void I2C_Reset(void) {
        // stop i2c bus
        I2C_Stop(TWI_STATE_MOTOR_TX);
        twi_state = 0;
        motor_write = TWDR;
        motor_write = 0;
        motor_read = 0;
        TWCR = (1 << TWINT); // reset to original state incl. interrupt flag reset
        TWAMR = 0;
        TWAR = 0;
        TWDR = 0;
        TWSR = 0;
        TWBR = 0;
        I2C_init();
        I2C_Start(TWI_STATE_MOTOR_TX);
}

/****************************************
 * I2C ISR
 ****************************************/
ISR (TWI_vect)
{
        static uint8_t missing_motor = 0;
        switch (twi_state++) { // First i2c_start from SendMotorData()
        // Master Transmit
        case 0: // TWI_STATE_MOTOR_TX
                // skip motor if not used in mixer
                while ((mixerMatrix.motor[motor_write][MIX_THROTTLE] <= 0) && (motor_write < MAX_MOTORS))
                        motor_write++;
                if (motor_write >= MAX_MOTORS) { // writing finished, read now
                        motor_write = 0;
                        twi_state = TWI_STATE_MOTOR_RX;
                        I2C_WriteByte(0x53 + (motor_read * 2)); // select slave adress in rx mode
                } else
                        I2C_WriteByte(0x52 + (motor_write * 2)); // select slave adress in tx mode
                break;
        case 1: // Send Data to Slave
                I2C_WriteByte(motor[motor_write].throttle); // transmit throttle value.
                break;
        case 2: // repeat case 0+1 for all motors
                if (TWSR == TW_MT_DATA_NACK) { // Data transmitted, NACK received
                        if (!missing_motor)
                                missing_motor = motor_write + 1;
                        if (++motor[motor_write].error == 0)
                                motor[motor_write].error = 255; // increment error counter and handle overflow
                }
                I2C_Stop(TWI_STATE_MOTOR_TX);
                I2CTimeout = 10;
                motor_write++; // next motor
                I2C_Start(TWI_STATE_MOTOR_TX); // Repeated start -> switch slave or switch Master Transmit -> Master Receive
                break;
                // Master Receive Data
        case 3:
                if (TWSR != TW_MR_SLA_ACK) { //  SLA+R transmitted, if not ACK received
                        // no response from the addressed slave received
                        motor[motor_read].present = 0;
                        motor_read++; // next motor
                        if (motor_read >= MAX_MOTORS)
                                motor_read = 0; // restart reading of first motor if we have reached the last one
                        I2C_Stop(TWI_STATE_MOTOR_TX);
                } else {
                        motor[motor_read].present = ('1' - '-') + motor_read;
                        I2C_ReceiveByte(); //Transmit 1st byte
                }
                missingMotor = missing_motor;
                missing_motor = 0;
                break;
        case 4: //Read 1st byte and transmit 2nd Byte
                motor[motor_read].current = TWDR;
                I2C_ReceiveLastByte(); // nack
                break;
        case 5:
                //Read 2nd byte
                motor[motor_read].maxPWM = TWDR;
                motor_read++; // next motor
                if (motor_read >= MAX_MOTORS)
                        motor_read = 0; // restart reading of first motor if we have reached the last one
                I2C_Stop(TWI_STATE_MOTOR_TX);
                break;

                // Writing ADC values.
        case 7:
                I2C_WriteByte(0x98); // Address the DAC
                break;

        case 8:
                I2C_WriteByte(0x10 + (DACChannel << 1)); // Select DAC Channel (0x10 = A, 0x12 = B, 0x14 = C)
                break;

        case 9:
                I2C_WriteByte(gyroAmplifierOffset.offsets[DACChannel]);
                break;

        case 10:
                I2C_WriteByte(0x80); // 2nd byte for all channels is 0x80
                break;

        case 11:
                I2C_Stop(TWI_STATE_MOTOR_TX);
                I2CTimeout = 10;
                // repeat case 7...10 until all DAC Channels are updated
                if (DACChannel < 2) {
                        DACChannel++; // jump to next channel
                        I2C_Start(TWI_STATE_GYRO_OFFSET_TX); // start transmission for next channel
                } else {
                        DACChannel = 0; // reset dac channel counter
                }
                break;

        default:
                I2C_Stop(TWI_STATE_MOTOR_TX);
                I2CTimeout = 10;
                motor_write = 0;
                motor_read = 0;
        }
}

extern void twi_diagnostics(void) {
        // Check connected BL-Ctrls
        uint8_t i;

        printf("\n\rFound BL-Ctrl: ");

        for (i = 0; i < MAX_MOTORS; i++) {
                motor[i].throttle = 0;
        }

        I2C_Start(TWI_STATE_MOTOR_TX);
        _delay_ms(2);

        motor_read = 0; // read the first I2C-Data

        for (i = 0; i < MAX_MOTORS; i++) {
                I2C_Start(TWI_STATE_MOTOR_TX);
                _delay_ms(2);
                if (motor[i].present)
                        printf("%d ",i+1);
        }

        for (i = 0; i < MAX_MOTORS; i++) {
                if (!motor[i].present && mixerMatrix.motor[i][MIX_THROTTLE] > 0)
                        printf("\n\r\n\r!! MISSING BL-CTRL: %d !!",i + 1);
                motor[i].error = 0;
        }
}