Subversion Repositories Projects

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

 *   Copyright (C) 2009-2012 by Claas Anders "CaScAdE" Rathje               *

 *   admiralcascade@gmail.com                                               *

 *   Project-URL: http://www.mylifesucks.de/oss/c-strom/                    *

 *                                                                          *

 *   This program is free software; you can redistribute it and/or modify   *

 *   it under the terms of the GNU General Public License as published by   *

 *   the Free Software Foundation; either version 2 of the License.         *

 *                                                                          *

 *   This program is distributed in the hope that it will be useful,        *

 *   but WITHOUT ANY WARRANTY; without even the implied warranty of         *

 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the          *

 *   GNU General Public License for more details.                           *

 *                                                                          *

 *   You should have received a copy of the GNU General Public License      *

 *   along with this program; if not, write to the                          *

 *   Free Software Foundation, Inc.,                                        *

 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.              *

 *                                                                          *

 *   Thanks to:                                                             *

 *   Klaus "akku" Buettner for the hardware                                 *

 *   All people at http://www.rn-wissen.de especially for the i2c stuff     *

 *                                                                          *

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

#include <avr/io.h>

#include <avr/eeprom.h>

#include <avr/pgmspace.h>

#include <avr/interrupt.h>

#include <util/delay.h>

#include <stdlib.h>

#include <string.h>

#include "C-Strom.h"

#include "spi_union.h"

#include "i2c_slave.h"

uint8_t EEMEM ee_checkbyte1 = CHECKBYTE1;

uint8_t EEMEM ee_checkbyte2 = CHECKBYTE2;

uint16_t EEMEM ee_cal_ampere = 512;

uint8_t EEMEM ee_sensor = 50;

uint8_t EEMEM ee_prim_r1 = 47, ee_prim_r2 = 150;

uint8_t EEMEM ee_anin_r1 = 47, ee_anin_r2 = 150;

uint8_t EEMEM ee_config = 0;

volatile uint8_t CSTROM_FLAGS = 0;

volatile uint8_t CSTROM_CONFIG = 0;

// we could use ee_cal_ampere but eeprom is slow :)

volatile uint16_t cal_ampere = 512;

volatile uint8_t sensor = 50;

volatile uint8_t prim_r1 = 47, prim_r2 = 150;

volatile uint8_t anin_r1 = 47, anin_r2 = 150;

volatile int16_t ampere, volt, anin_volt, transfer_ampere;

volatile int32_t transfer_mah, mah;

volatile int16_t average_ampere = 0;

volatile uint8_t hwver = 10;

// global space for int conversion to string

char s[10];

// spi buffer

union SPI_buffer_t SPI_buffer;

// PD7 High

void PD7_H() {

        PORTD |=  (1 << PD7);

}

// PD7 Low

void PD7_L() {

        PORTD &= ~(1 << PD7);

}

void (*LED_ON)(void) = PD7_H;

void (*LED_OFF)(void) = PD7_L;

void ampere_calibrate();

void save_eeprom();

void help(uint8_t);

/*ISR(__vector_default) {

    asm("nop");

}*/

/**

 * decimal itoa for 10th values

 */

char *itoa_dec(int val, char* s) {

        itoa(val, s, 10);

        //char x = 0;

        for (uint8_t i = 0; i < 9; i++) {

                if (s[i] == 0 && i > 0) {

                        if (i == 1) {

                                s[i+1] = s[i-1];       

                                s[i-1] = '0';

                                s[i] = '.';

                                s[i+2] = 0;    

                        } else {

                                s[i] = s[i-1];

                                s[i-1] = '.';

                                s[i+1] = 0;

                        }

                        break;

                }

        }

        return s;

}

/**

 * init uart

 */

void uart_init() {

    UBRRL = (F_CPU / (16UL * BAUD_RATE)) - 1;

    // Enable receiver and transmitter; enable RX interrupt

    UCSRB = (1 << RXEN) | (1 << TXEN) | (1 << RXCIE);

    //asynchronous 8N1

    UCSRC = (1 << URSEL) | (3 << UCSZ0);

}

/**

 * send a single <character> through uart

 */

void uart_putc(unsigned char character) {

    // wait until UDR ready

    while (!(UCSRA & (1 << UDRE)));

    UDR = character;

}

/**

 * send a <string> throught uart

 */

void uart_puts(char *s) {

    while (*s) {

        uart_putc(*s);

        s++;

    }

}

/**

 * send a <string> from pgm space throught uart

 */

void uart_puts_pgm(char *string) {

        while (pgm_read_byte(string) != 0x00)

                uart_putc(pgm_read_byte(string++));

}

/**

 * change the sensor type

 */

void sensor_change(uint8_t new_value) {

        if (new_value < 10) new_value = 0;

        else if (new_value > 250) new_value = 250;

        sensor = new_value;

        uart_puts_pgm(PSTR("\r\nSensor is now: "));

        uart_puts(itoa(sensor, s, 10));

        uart_puts("A\r\n");

}

/**

 * change the r2 value

 */

void r2_change(uint8_t which, uint8_t new_value) {     

        if (which == V_ANIN) {         

                uart_puts_pgm(PSTR("\r\nANIN R2 is now: "));

                anin_r2 = new_value;

                uart_puts(itoa_dec(anin_r2, s));

        } else {

                uart_puts_pgm(PSTR("\r\nPRIMARY R2 is now: "));

                prim_r2 = new_value;

                uart_puts(itoa_dec(prim_r2, s));

        }

        uart_puts_pgm(PSTR("kOhm\r\n"));

}

/**

 * enable/disable TWI

 */

void twi_change() {    

        uart_puts_pgm(PSTR("\r\nTWI turned "));

        if (CSTROM_CONFIG & CSTROM_TWI) {

                uart_puts_pgm(PSTR("ON"));

        } else {

                uart_puts_pgm(PSTR("OFF"));

        }

        uart_puts_pgm(PSTR(". Please restart...\r\n"));

}

/**

 * Interrupt handler for received data through UART1

 */

SIGNAL(SIG_UART_RECV) {

        unsigned char c = UDR;

        switch (c) {

                case 'c':

                        ampere_calibrate();

                        break;

                case 's':

                        save_eeprom();

                        break;

                case '+':

                        sensor_change(100);

                        break;

                case '-':

                        sensor_change(50);

                        break;

                case 'e':

                        if (hwver == 11) r2_change(V_ANIN, anin_r2 + 1);

                        break;

                case 'd':

                        if (hwver == 11) r2_change(V_ANIN, anin_r2 - 1);

                        break;

                case 'r':

                        r2_change(V_PRIMARY, prim_r2 + 1);

                        break;

                case 'f':

                        r2_change(V_PRIMARY, prim_r2 - 1);

                        break;

                case 'T':

                        CSTROM_CONFIG ^= CSTROM_TWI;

                        twi_change();

                        break;

                case 'h':

                        help(0);

                        break;

                default:

                        asm("nop"); // :-)

        }

}

/**

 * Interrupt handler for transmitting data through UART1

 */

SIGNAL(SIG_UART_TRANS) {

}

/**

 * Read out the ADC channel <channel>

 */

uint16_t readADC(uint8_t channel) {

        uint8_t i;

        uint16_t result = 0;

        // enable ADC and set clk div to 64

        ADCSRA = (1<<ADEN) | (1<<ADPS2) | (1<<ADPS1);

        _delay_us(5);

        // set up channel

        ADMUX = channel;

        // use internal reference

        //ADMUX |= (1<<REFS1) | (1<<REFS0);

        // init ADC for a dummy readout

        ADCSRA |= (1<<ADSC);

        // wait for conversion to be complete

        while(ADCSRA & (1<<ADSC));

        // read in three times and get the average

        for(i=0; i<3; i++) {

                // start conversion

                ADCSRA |= (1<<ADSC);

                // wait for conversion to be complete

                while(ADCSRA & (1<<ADSC));

                // add up result

                result += ADCW;

        }

        // disable ADC

        ADCSRA &= ~(1<<ADEN);

        // get average

        result /= 3;

        return result;

}

/**

 * init SPI slave interrupt conrolled

 */

void Init_Slave_IntContr (void) {

        volatile char IOReg;

        // Set PB4(MISO) as output

        DDRB    = (1<<PB4);

                // MOSI Pullup

                PORTB |= _BV(3);

        // Enable SPI Interrupt and SPI in Slave Mode

        SPCR  = (1<<SPIE)|(1<<SPE);

        IOReg   = SPSR; // Clear SPIF bit in SPSR

        IOReg   = SPDR;

                SPCR |= _BV(SPIE); // duplicated

}

/**

 * SPI interrupt handling

 */

ISR(SPI_STC_vect) {

        LED_ON();

        unsigned char foo;

        foo = SPDR;

        //uart_putc(foo);

        switch (foo) {

                case 'A': // requested ampere high bits for next transmission

                        CSTROM_FLAGS |= CSTROM_SPILOCKED;

                        foo = SPI_buffer.buffer.c[0];

                        break;

                case 'B': // requested low bits

                        foo = SPI_buffer.buffer.c[1];

                        break;

                case 'C': // wasted ampere high bits in next

                        foo = SPI_buffer.buffer.c[2];

                        break;

                case 'D': // 2nd highest 8bits

                        foo = SPI_buffer.buffer.c[3];

                        break;

                case 'E': // 3rd highest 8bits

                        foo = SPI_buffer.buffer.c[4];

                        break;

                case 'F': // lowest 8bits

                        foo = SPI_buffer.buffer.c[5];

                        break;

                case 'G': // lowest 8bits

                        foo = SPI_buffer.buffer.c[6];

                        break;

                case 'H': // lowest 8bits

                        foo = SPI_buffer.buffer.c[7];

                        break;

                case 'I': // challange over

                        foo = 'd'; // done :)

                        CSTROM_FLAGS &= ~CSTROM_SPILOCKED;

                        break;

                default:  // what else? nothin now

                        foo = 'X';

        }

        // write back foo in next transmission

        SPDR = foo;

        //uart_putc(foo);

        LED_OFF();

}

/**

 * read data saved in eeprom

 */

void get_eeprom() {

        if (eeprom_read_byte(&ee_checkbyte1) == CHECKBYTE1 && eeprom_read_byte(&ee_checkbyte2) == CHECKBYTE2) {

                uart_puts("\tLoading data from eeprom...");

                sensor = eeprom_read_byte(&ee_sensor);

                cal_ampere = eeprom_read_word(&ee_cal_ampere);

                anin_r1 = eeprom_read_byte(&ee_anin_r1);

                anin_r2 = eeprom_read_byte(&ee_anin_r2);

                prim_r1 = eeprom_read_byte(&ee_prim_r1);

                prim_r2 = eeprom_read_byte(&ee_prim_r2);

                CSTROM_CONFIG = eeprom_read_byte(&ee_config);

                uart_puts("done\r\n");

        } else {

                uart_puts("\tNo data found in eeprom, using default data...\r\n");

        }

}

/**

 * save data to eeprom

 */

void save_eeprom() {

        uart_puts("\r\nSaving data to eeprom...");

        eeprom_write_byte(&ee_checkbyte1, CHECKBYTE1);

        eeprom_write_byte(&ee_checkbyte2, CHECKBYTE2);

        eeprom_write_byte(&ee_sensor, sensor);

        eeprom_write_word(&ee_cal_ampere, cal_ampere);

        //if (hwver == 11)  {

        // why not saving when not needed, there is space

                eeprom_write_byte(&ee_anin_r1, anin_r1);

                eeprom_write_byte(&ee_anin_r2, anin_r2);               

        //}

        eeprom_write_byte(&ee_prim_r1, prim_r1);

        eeprom_write_byte(&ee_prim_r2, prim_r2);       

        eeprom_write_byte(&ee_config, CSTROM_CONFIG);  

        uart_puts("done\r\n");

}

/**

 * calibrate the current sensor... has to be 0A during this time!

 */

void ampere_calibrate() {

        cli();

        uart_puts("\r\nCalibrating...");

        uint16_t temp_cal = 0;

        for (uint8_t i = 0; i < 10; i++) {

                temp_cal += readADC(0);

                uart_puts("#");

                _delay_ms(100);

        }

        cal_ampere = temp_cal / 10;

        uart_puts("done. Offset is now: ");

        uart_puts(itoa(cal_ampere, s, 10));

        uart_puts("\r\n");

        sei();

}

volatile uint16_t timer = 0, cs = 0;

/**

 * init timer0

 */

void init_timer0(void){

        // set up timer

        TCCR0 |= (1 << CS00) | (1 << CS01); // timer0 prescaler 64

        TIMSK |= (1 << TOIE0); // enable overflow timer0

}

/**

 * timer overflow handler, should be 1ms

 */

SIGNAL(SIG_OVERFLOW0) {

        TCNT0 = 131; // preload

        timer++;

        // this should be 100ms

        if (timer == 100) {

                timer = 0;

                cs++;

                average_ampere += ampere;

                CSTROM_FLAGS |= CSTROM_WRITEUART;

        }

        // this should be 1s

        if (cs == 10) {

                cs = 0;

                mah += average_ampere / 360;

                average_ampere = 0;

        }

}

/**

 * write <len> through uart spaces

 */

void write_space(uint8_t len) {

        while (len--) {

                uart_putc(' ');

        }

}

/**

 * check which hardware version we have here

 */

void check_hw() {

        // check if pin was output and has pullup

        uint8_t old_DDRD7 =  DDRD & (1 << PD7);

        uint8_t old_PORTD7 =  PORTD & (1 << PD7);

        // if it was, make it input

        if (old_DDRD7) DDRD &= ~(1 << PD7); // PD7 input (LED)

        if (!old_PORTD7) PORTD |= (1 << PD7); // PD7 enable pullup (LED)

        if (PIND & (1 << PD7)) {

                hwver = 11;

                LED_ON = PD7_L;

                LED_OFF = PD7_H;

        }

        // output again

        if (!old_PORTD7) PORTD &= ~(1 << PD7); // PD7 disable pullup (LED)

        if (old_DDRD7) DDRD |= (1 << PD7); // PD7 output (LED)

}

/**

 * call for help whenever needed

 */

void help(uint8_t load) {

    uart_puts_pgm(PSTR("\r\nC-STROM\r\n\tBUILD: "));

        uart_puts_pgm(PSTR(BUILDDATE));

        uart_puts("\r\n\tHW: ");

        uart_puts(itoa_dec(hwver, s));

        uart_puts("\r\n");

        if (load) get_eeprom();

        uart_puts_pgm(PSTR("\tSensor: "));

        uart_puts(itoa(sensor, s, 10));

        uart_puts_pgm(PSTR("A\tCalibration: "));

        uart_puts(itoa(cal_ampere, s, 10));

        uart_puts_pgm(PSTR("\r\n\tTWI is "));

        if (CSTROM_CONFIG & CSTROM_TWI) {

                uart_puts_pgm(PSTR("ON, SPI may not work!!!"));

        } else {

                uart_puts_pgm(PSTR("OFF"));

        }

        uart_puts_pgm(PSTR("\r\n\tPIMARY R2: "));

        uart_puts(itoa_dec(prim_r2, s));

        if (hwver == 11) {

                uart_puts_pgm(PSTR("kOhm"));

                uart_puts_pgm(PSTR("\tANIN R2: "));

                uart_puts(itoa_dec(anin_r2, s));

        }

        uart_puts_pgm(PSTR("kOhm\r\n"));

        uart_puts_pgm(PSTR("\tCommands available:\r\n"));

        uart_puts_pgm(PSTR("\t\th : help on commands (this)\r\n"));

        uart_puts_pgm(PSTR("\t\tc : calibrate ampere\r\n"));

        uart_puts_pgm(PSTR("\t\tT : toggle TWI (may break SPI communication!)\r\n"));

        uart_puts_pgm(PSTR("\t\t+/- : to change sensor\r\n"));

        uart_puts_pgm(PSTR("\t\tr/f : to change PRIMARY-R2 Value\r\n"));

        if (hwver == 11) {

                uart_puts_pgm(PSTR("\t\te/d : to change ANIN-R2 Value\r\n"));

        }

        uart_puts_pgm(PSTR("\t\ts : save values\r\n"));

        uart_puts_pgm(PSTR("\tnow enjoy it and have fun...\r\n\r\n"));

}

/**

 * Main

 */

int main (void) {

        DDRD |= (1 << PD7); // PD7 output (LED)

        check_hw();

        uart_init();

        Init_Slave_IntContr();

        init_timer0();

        sei();   // Enable Global Interrupts

    uart_puts("\x1B[2J\x1B[H"); // clear serial

        help(1);

        if (CSTROM_CONFIG & CSTROM_TWI) init_twi_slave(CSTROM_I2C);

        int16_t raw_volt = 0, raw_ampere = 0, raw_aninvolt = 0;

        char c[10] = "         ";

        c[9] = 0;

        //strom_data  = *((SPI_strom_data_t*) &spi_buffer);

        //*spi_buffer = *((uint8_t*) (void*) &strom_data);

        LED_ON();

        while (1) { // Loop Forever

                // we have got a normal voltage measuring circuit that takes the lipo-voltage

                raw_volt = readADC(1);

                /* according to what i read about voltage divider it is

                   Uo = Ue * (R1 / (R2 + R1))

                   Ue = Uo * (R2 + R1) / R1

                   the board has got r1 = 4.7k and r2 = 15k

                   but since 1step is 0,0048828125V = 4,8828125mV and not 5mV there

                   is some conversion to do for raw_volt --**-> Uo

                   this should end up in 10th of volts */

                raw_volt = ((uint32_t)raw_volt * (uint32_t)48828) / (uint32_t)10000;

                volt = (int16_t) (((uint32_t)raw_volt * (uint32_t)(prim_r1 + prim_r2)) / (uint32_t)prim_r1) / 100;

                if (volt < 0) volt = 0;

                // and we have got a seccond voltage measuring circuit for user voltages

                raw_aninvolt = readADC(2);

                /* some conversion to do for raw_volt --**-> Uo

                   this should end up in 10th of volts */

                raw_aninvolt = ((uint32_t)raw_aninvolt * (uint32_t)48828) / (uint32_t)10000;

                anin_volt = (int16_t) (((uint32_t)raw_aninvolt * (uint32_t)(anin_r1 + anin_r2)) / (uint32_t)anin_r1) / 100;

                if (anin_volt < 0) anin_volt = 0;

                raw_ampere = readADC(0);

                /* according to datasheet sensitivity is nominal 40mV per A for the 50A chip

                   this would mean 50A ^= 2V since 0A is set to 2.5V output Voltage we get

                   a range of 0.5V till 4.5V for the full range.

                   the atmega ADC features 0...5V range divided into 10bit ^= 1024 steps

                   so 0,0048828125V, or 4,8828125mV, is one step

                   this leads us to 0,8192 steps per 0,1A and somehow the below formula

                   and i know that 32bit is evil, but what else does this device has to do? :)

                   this should end up in 100th of ampere */

                ampere = (int16_t) (((int32_t)(((int16_t)raw_ampere - (int16_t)cal_ampere)) * (int32_t)10000) / (int32_t) 819);

                if (sensor == 100) ampere *= 2;

                if ((CSTROM_FLAGS & CSTROM_WRITEUART)) {

                        uart_puts("V: ");

                        uart_puts(itoa_dec(volt, s));

                        write_space(10-strlen(s));

                        uart_puts("AN-IN V: ");

                        uart_puts(itoa_dec(anin_volt, s));

                        write_space(10-strlen(s));

                        uart_puts("A: ");

                        uart_puts(itoa(ampere, s, 10));

                        write_space(10-strlen(s));

                        uart_puts("C: ");

                        uart_puts(itoa(mah, s, 10));

                        write_space(10-strlen(s));

                        uart_puts("\r");

                        CSTROM_FLAGS &= ~CSTROM_WRITEUART;

                }

                //spi_buff

                if (!(CSTROM_FLAGS & CSTROM_SPILOCKED)) {

                        // TESTTING 

                        if (!(CSTROM_CONFIG & CSTROM_TWI)) CSTROM_FLAGS |= CSTROM_SPILOCKED;

                        SPI_buffer.data.ampere = ampere;

                        SPI_buffer.data.mah = mah;

                        if (hwver == 11) {

                                SPI_buffer.data.volt = anin_volt;

                        } else {

                                SPI_buffer.data.volt = volt;

                        }

                }

        }

   return 0;

}