Subversion Repositories MK3Mag

/*#######################################################################################

MK3Mag 3D-Magnet sensor

!!! THIS IS NOT FREE SOFTWARE !!!

#######################################################################################*/

// ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

// + Copyright (c) 05.2008 Holger Buss

// + Thanks to Ilja Fähnrich (P_Latzhalter)

// + Nur für den privaten Gebrauch

// + www.MikroKopter.com

// ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

// + Die Portierung der Software (oder Teile davon) auf andere Systeme (ausser der Hardware von www.mikrokopter.de) ist nur

// + mit unserer Zustimmung zulässig

// ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

// + Es gilt für das gesamte Projekt (Hardware, Software, Binärfiles, Sourcecode und Dokumentation),

// + dass eine Nutzung (auch auszugsweise) nur für den privaten (nicht-kommerziellen) Gebrauch zulässig ist.

// + AUSNAHME: Ein bei www.mikrokopter.de erworbener vorbestückter MK3Mag darf als Baugruppe auch in kommerziellen Systemen verbaut werden

// + Im Zweifelsfall bitte anfragen bei: info@mikrokopter.de

// ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

// + Werden Teile des Quellcodes (mit oder ohne Modifikation) weiterverwendet oder veröffentlicht,

// + unterliegen sie auch diesen Nutzungsbedingungen und diese Nutzungsbedingungen incl. Copyright müssen dann beiliegen

// ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

// + Sollte die Software (auch auszugesweise) oder sonstige Informationen des MikroKopter-Projekts

// + auf anderen Webseiten oder sonstigen Medien veröffentlicht werden, muss unsere Webseite "http://www.mikrokopter.de"

// + eindeutig als Ursprung verlinkt werden

// ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

// + Keine Gewähr auf Fehlerfreiheit, Vollständigkeit oder Funktion

// + Benutzung auf eigene Gefahr

// + Wir übernehmen keinerlei Haftung für direkte oder indirekte Personen- oder Sachschäden

// ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

// + Die Funktion printf_P() unterliegt ihrer eigenen Lizenz und ist hiervon nicht betroffen

// ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

// + Redistributions of source code (with or without modifications) must retain the above copyright notice,

// + this list of conditions and the following disclaimer.

// +   * PORTING this software (or parts of it) to systems (other than hardware from www.mikrokopter.de) is NOT allowed

// +   * Neither the name of the copyright holders nor the names of contributors may be used to endorse or promote products derived

// +     from this software without specific prior written permission.

// +   * The use of this project (hardware, software, binary files, sources and documentation) is only permittet

// +     for non-commercial use (directly or indirectly)

// +     Commercial use (for excample: selling of MikroKopters, selling of PCBs, assembly, ...) is only permitted

// +     with our written permission

// +     Exception: A preassembled MK3Mag, purchased from www.mikrokopter.de may be used as a part of commercial systems

// +     In case of doubt please contact: info@MikroKopter.de

// +   * If sources or documentations are redistributet on other webpages, our webpage (http://www.MikroKopter.de) must be

// +     clearly linked as origin

// +  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"

// +  AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

// +  IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE

// +  ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE

// +  LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR

// +  CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF

// +  SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS

// +  INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN

// +  CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)

// +  ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

// +  POSSIBILITY OF SUCH DAMAGE.

// ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

#include <avr/interrupt.h>

#include <math.h>

#include <stdlib.h>

#include <stdio.h>

#include "main.h"

#include "timer0.h"

#include "twislave.h"

#include "led.h"

#include "analog.h"

#include "uart.h"

AttitudeSource_t AttitudeSource = ATTITUDE_SOURCE_ACC;

uint16_t Led_Timer = 0;

struct Scaling_t

{

        int16_t Range;

        int16_t Offset;

} ;

struct Calibration_t

{

        struct Scaling_t MagX;

        struct Scaling_t MagY;

        struct Scaling_t MagZ;

        struct Scaling_t AccX;

        struct Scaling_t AccY;

        struct Scaling_t AccZ;

} ;

struct Calibration_t eeCalibration EEMEM;       // calibration data in EEProm

struct Calibration_t Calibration;               // calibration data in RAM

// magnet sensor variable

int16_t RawMagnet1a, RawMagnet1b;                       // raw magnet sensor data

int16_t RawMagnet2a, RawMagnet2b;

int16_t RawMagnet3a, RawMagnet3b;

int16_t UncalMagX, UncalMagY, UncalMagZ;        // sensor signal difference without Scaling

int16_t MagX, MagY, MagZ;                                       // rescaled magnetic field readings

// acc sensor variables

int16_t RawAccX, RawAccY, RawAccZ;                      // raw acceleration readings

int16_t AccX, AccY, AccZ;                                       // rescaled acceleration readings

uint8_t AccPresent = 0;

uint8_t PC_Connected = 0;

int16_t Heading = -1;

void CalcFields(void)

{

        UncalMagX = (RawMagnet1a - RawMagnet1b);

        UncalMagY = (RawMagnet3a - RawMagnet3b);

        UncalMagZ = (RawMagnet2a - RawMagnet2b);

        if(Calibration.MagX.Range != 0) MagX = (1024L * (int32_t)(UncalMagX - Calibration.MagX.Offset)) / (Calibration.MagX.Range);

        else MagX = 0;

        if(Calibration.MagY.Range != 0) MagY = (1024L * (int32_t)(UncalMagY - Calibration.MagY.Offset)) / (Calibration.MagY.Range);

        else MagY = 0;

        if(Calibration.MagY.Range != 0) MagZ = (1024L * (int32_t)(UncalMagZ - Calibration.MagZ.Offset)) / (Calibration.MagZ.Range);

        else MagZ = 0;

}

void CalcHeading(void)

{

        double nick_rad, roll_rad, Hx, Hy, Cx, Cy, Cz;

        int16_t heading = -1;

        // blink code for normal operation

        if(CheckDelay(Led_Timer))

        {

                LED_GRN_TOGGLE;

                Led_Timer = SetDelay(500);

        }

        Cx = MagX;

        Cy = MagY;

        Cz = MagZ;

        if(ExternData.Orientation == 1)

        {

                Cx = MagX;

                Cy = -MagY;

                Cz = MagZ;

        }

        // calculate nick and roll angle in rad

        switch(AttitudeSource)

        {

                case ATTITUDE_SOURCE_I2C:

                        nick_rad = ((double)I2C_WriteAttitude.Nick) * M_PI / (double)(1800.0);

                        roll_rad = ((double)I2C_WriteAttitude.Roll) * M_PI / (double)(1800.0);

                        break;

                case ATTITUDE_SOURCE_UART:

                        nick_rad = ((double)ExternData.Attitude[NICK]) * M_PI / (double)(1800.0);

                        roll_rad = ((double)ExternData.Attitude[ROLL]) * M_PI / (double)(1800.0);

                        break;

                case ATTITUDE_SOURCE_ACC:

                        if(AccX >  125) nick_rad = M_PI / 2;

                        else

                        if(AccX < -125) nick_rad = -M_PI / 2;

                        else

                        {

                                nick_rad = asin((double) AccX / 125.0);

                        }

                        if(AccY >  125) roll_rad = M_PI / 2;

                        else

                        if(AccY < -125) roll_rad = -M_PI / 2;

                        else

                        {

                        roll_rad = asin((double) AccY / 125.0);

                        }

                        break;

        }

        // calculate attitude correction

        Hx = Cx * cos(nick_rad) - Cz * sin(nick_rad);

        Hy = Cy * cos(roll_rad) + Cz * sin(roll_rad);

        // calculate Heading

        heading = (int16_t)((180.0 * atan2(Hy, Hx)) / M_PI);

        // atan2 returns angular range from -180 deg to 180 deg in counter clockwise notation

        // but the compass course is defined in a range from 0 deg to 360 deg clockwise notation.

        if (heading < 0) heading = -heading;

        else heading = 360 - heading;

        if(abs(heading) < 361) Heading = heading;

        else (Heading = -1);

}

void Calibrate(void)

{

        uint8_t cal;

        static uint8_t calold = 0;

        static int16_t Xmin = 0, Xmax = 0, Ymin = 0, Ymax = 0, Zmin = 0, Zmax = 0;

        static uint8_t blinkcount = 0;

        // check both sources of communication for calibration request

        if(I2C_WriteCal.CalByte) cal = I2C_WriteCal.CalByte;

        else                     cal = ExternData.CalState;

        if(cal > 5) cal = 0;

        // blink code for current calibration state

        if(cal)

        {

                if(CheckDelay(Led_Timer) || (cal != calold))

                {

                        if(blinkcount & 0x01) LED_GRN_OFF;

                        else LED_GRN_ON;

                        // end of blinkcount sequence

                        if( (blinkcount + 1 ) >= (2 * cal) )

                        {

                                blinkcount = 0;

                                Led_Timer = SetDelay(1000);

                        }

                        else

                        {

                                blinkcount++;

                                Led_Timer = SetDelay(170);

                        }

                }

        }

        else

        {

                LED_GRN_OFF;

        }

        // calibration state machine

        switch(cal)

        {

                case 1: // 1st step of calibration

                        // initialize ranges

                        // used to change the orientation of the MK3MAG in the horizontal plane

                        Xmin =  10000;

                        Xmax = -10000;

                        Ymin =  10000;

                        Ymax = -10000;

                        Zmin =  10000;

                        Zmax = -10000;

                        Calibration.AccX.Offset = RawAccX;

                        Calibration.AccY.Offset = RawAccY;

                Calibration.AccZ.Offset = RawAccZ;

                        break;

                case 2: // 2nd step of calibration

                        // find Min and Max of the X- and Y-Sensors during rotation in the horizontal plane

                        if(UncalMagX < Xmin) Xmin = UncalMagX;

                        if(UncalMagX > Xmax) Xmax = UncalMagX;

                        if(UncalMagY < Ymin) Ymin = UncalMagY;

                        if(UncalMagY > Ymax) Ymax = UncalMagY;

                        break;

                case 3: // 3rd step of calibration

                        // used to change the orientation of the MK3MAG vertical to the horizontal plane

                        break;

                case 4:

                        // find Min and Max of the Z-Sensor

                        if(UncalMagZ < Zmin) Zmin = UncalMagZ;

                        if(UncalMagZ > Zmax) Zmax = UncalMagZ;

                        break;

                case 5:

                        // Save values

                        if(cal != calold) // avoid continously writing of eeprom!

                        {

                                Calibration.MagY.Range = Xmax - Xmin;

                                Calibration.MagX.Offset = (Xmin + Xmax) / 2;

                                Calibration.MagY.Range = Ymax - Ymin;

                                Calibration.MagY.Offset = (Ymin + Ymax) / 2;

                                Calibration.MagZ.Range = Zmax - Zmin;

                                Calibration.MagZ.Offset = (Zmin + Zmax) / 2;

                                if((Calibration.MagX.Range > 150) && (Calibration.MagY.Range > 150) && (Calibration.MagZ.Range > 150))

                                {

                                        // indicate write process by setting the led

                                        LED_GRN_ON;

                                        eeprom_write_block(&Calibration, &eeCalibration, sizeof(Calibration));

                                        Delay_ms(2000);

                                        // reset led state

                                        LED_GRN_OFF;

                                        // reset  blinkcode

                                        blinkcount = 0;

                                        Led_Timer = SetDelay(1000);

                                }

                        }

                        break;

                default:

                        break;

        }

        calold = cal;

}

void SetDebugValues(void)

{

        DebugOut.Analog[0] =  MagX;

        DebugOut.Analog[1] =  MagY;

        DebugOut.Analog[2] =  MagZ;

        DebugOut.Analog[3] =  UncalMagX;

        DebugOut.Analog[4] =  UncalMagY;

        DebugOut.Analog[5] =  UncalMagZ;

        switch(AttitudeSource)

        {

                case ATTITUDE_SOURCE_ACC:

                        break;

                case ATTITUDE_SOURCE_UART:

                        DebugOut.Analog[6] =  ExternData.Attitude[NICK];

                        DebugOut.Analog[7] =  ExternData.Attitude[ROLL];

                        break;

                case ATTITUDE_SOURCE_I2C:

                        DebugOut.Analog[6] =  I2C_WriteAttitude.Nick;

                        DebugOut.Analog[7] =  I2C_WriteAttitude.Roll;

                        break;

        }

        DebugOut.Analog[8] =  Calibration.MagX.Offset;

        DebugOut.Analog[9] =  Calibration.MagX.Range;

        DebugOut.Analog[10] = Calibration.MagY.Offset;

        DebugOut.Analog[11] = Calibration.MagY.Range;

        DebugOut.Analog[12] = Calibration.MagZ.Offset;

        DebugOut.Analog[13] = Calibration.MagZ.Range;

        DebugOut.Analog[14] = ExternData.CalState;

        DebugOut.Analog[15] = Heading;

        DebugOut.Analog[16] = ExternData.UserParam[0];

        DebugOut.Analog[17] = ExternData.UserParam[1];

        DebugOut.Analog[18] = AccX;

        DebugOut.Analog[19] = AccY;

        DebugOut.Analog[20] = AccZ;

        DebugOut.Analog[21] = RawAccX;

        DebugOut.Analog[22] = RawAccY;

        DebugOut.Analog[23] = RawAccZ;

        DebugOut.Analog[24] = Calibration.AccX.Offset;

        DebugOut.Analog[25] = Calibration.AccY.Offset;

    DebugOut.Analog[26] = Calibration.AccZ.Offset;

}

void AccMeasurement(void)

{

        if(AccPresent)

        {

                RawAccX = (RawAccX + (int16_t)ADC_GetValue(ADC2))/2;

                RawAccY = (RawAccY + (int16_t)ADC_GetValue(ADC3))/2;

                RawAccZ = (RawAccZ + (int16_t)ADC_GetValue(ADC6))/2;

        }

        else

        {

                RawAccX = 0;

                RawAccY = 0;

                RawAccZ = 0;

        }

                AccX = ((RawAccX - Calibration.AccX.Offset) + AccX * 7) / 8;

                AccY = ((RawAccY - Calibration.AccY.Offset) + AccY * 7) / 8;

                AccZ = ((Calibration.AccZ.Offset - RawAccZ) + AccZ * 7) / 8;

}

int main (void)

{

        // reset input pullup

        DDRC &=~((1<<DDC6));

        PORTC |= (1<<PORTC6);

    LED_Init();

    TIMER0_Init();

    USART0_Init();

    ADC_Init();

        I2C_Init();

    sei(); // enable globale interrupts

    LED_GRN_ON;

    Debug_Timer = SetDelay(200);

    Led_Timer = SetDelay(200);

        // read calibration info from eeprom

        eeprom_read_block(&Calibration, &eeCalibration, sizeof(Calibration));

    ExternData.Orientation = 0;

    ExternData.CalState = 0;

    I2C_WriteCal.CalByte = 0;

        // main loop

    while (1)

    {

                FLIP_LOW;

                Delay_ms(2);

                RawMagnet1a = ADC_GetValue(ADC0);

                RawMagnet2a = -ADC_GetValue(ADC1);

                RawMagnet3a = ADC_GetValue(ADC7);

                AccMeasurement();

                Delay_ms(1);

                FLIP_HIGH;

                Delay_ms(2);

                RawMagnet1b = ADC_GetValue(ADC0);

                RawMagnet2b = -ADC_GetValue(ADC1);

                RawMagnet3b = ADC_GetValue(ADC7);

                AccMeasurement();

                Delay_ms(1);

                CalcFields();

                if(ExternData.CalState || I2C_WriteCal.CalByte) Calibrate();

                else CalcHeading();

                // check data from USART

        USART0_ProcessRxData();

                USART0_TransmitTxData();

        if(PC_Connected)

        {

            USART0_EnableTXD();

            USART0_TransmitTxData();

            PC_Connected--;

                }

                else

                {

                        USART0_DisableTXD();

                }

        } // while(1)

}