Rev 2225 |
Go to most recent revision |
Blame |
Compare with Previous |
Last modification |
View Log
| RSS feed
/****************************************************************************
* Copyright (C) 2009-2017 by Claas Anders "CaScAdE" Rathje *
* admiralcascade@gmail.com *
* Project-URL: http://www.mylifesucks.de/oss/c-osd/ *
* *
* 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. *
* *
* *
* Credits to: *
* Holger Buss & Ingo Busker from mikrokopter.de for the MK project + SVN *
* Gregor "killagreg" Stobrawa for making the MK code readable *
* Klaus "akku" Buettner for the hardware *
* Manuel "KeyOz" Schrape for explaining the MK protocol to me *
****************************************************************************/
#include "main.h"
#include <avr/io.h>
#include <avr/interrupt.h>
#include <util/delay.h>
#include <avr/pgmspace.h>
#include <string.h>
#include "max7456_software_spi.h"
#ifdef ANTENNATRACKTEST
#include "usart0.h"
#endif
#include "usart1.h"
#include "osd_helpers.h"
#include "config.h"
#include "spi.h"
#include "buttons.h"
#include "ppm.h"
#include "osd_ncmode_default.h"
#include "osd_ncmode_minimal.h"
#include "osd_fcmode_default.h"
#include "osd_fcmode_jopl.h"
#if WRITECHARS != -1
#include "characters.h"
#endif
/* TODO:
* - verifiy correctness of values
* - clean up code :)
*/
/* ##########################################################################
* global definitions and global vars
* ##########################################################################*/
#if !(ALLCHARSDEBUG|(WRITECHARS != -1))
#if FCONLY
volatile str_DebugOut debugData
;
#else
volatile NaviData_t naviData
;
volatile Data3D_t data3d
;
#endif
// cache old vars for blinking attribute, checkup is faster than full
// attribute write each time
volatile uint8_t last_UBat
= 255;
volatile uint8_t last_RC_Quality
= 255;
// 16bit should be enough, normal LiPos don't last that long
volatile uint16_t uptime
= 0;
volatile uint16_t timer
= 0;
volatile uint16_t flytime_fc
= 0;
// remember last time data was received
volatile uint8_t seconds_since_last_data
= 0;
// general PAL|NTSC distingiusch stuff
uint8_t top_line
= 1;
uint8_t bottom_line
= 14;
// battery voltages
uint8_t min_voltage
= 0;
uint8_t max_voltage
= 0;
uint8_t scope
[12] = {
5, 5, 0,
25, 5, 0,
5, 10, 0,
25, 10, 0
};
// Flags
uint8_t COSD_FLAGS_MODES
= 0, COSD_FLAGS_CONFIG
= 0, COSD_FLAGS_RUNTIME
= 0, COSD_DISPLAYMODE
= 0;
// stats for after flight
int16_t max_Altimeter
= 0;
uint8_t min_UBat
= 255;
uint16_t max_GroundSpeed
= 0;
int16_t max_Distance
= 0;
uint16_t max_FlyingTime
= 0;
// flags from last round to check for changes
uint8_t old_MKFlags
= 0;
char* directions
[8] = {"NE", "E ", "SE", "S ", "SW", "W ", "NW", "N "};
//char arrowdir[8] = {218, 217, 224, 223, 222, 221, 220, 219};
/* ##########################################################################
* Different display mode function pointers
* ##########################################################################*/
const char str_1
[] PROGMEM
= "default";
const char str_2
[] PROGMEM
= "minimal";
const char str_3
[] PROGMEM
= " jopl";
#if FCONLY
const displaymode_t fcdisplaymodes
[] PROGMEM
= {
{ osd_fcmode_default
, (char *)str_1
},
{ osd_fcmode_jopl
, (char *)str_3
}
};
int (*osd_fcmode
)(void) = (int(*)(void)) & osd_fcmode_default
;
#else
const displaymode_t ncdisplaymodes
[] PROGMEM
= {
{ osd_ncmode_default
, (char *)str_1
},
{ osd_ncmode_minimal
, (char *)str_2
}
};
int (*osd_ncmode
)(void) = (int(*)(void)) & osd_ncmode_default
;
#endif
#endif
/* ##########################################################################
* Interrupt handler
* ##########################################################################*/
/**
* handler for undefined Interrupts
* if not defined AVR will reset in case any unhandled interrupts occur
*/
ISR
(__vector_default
) {
asm
("nop");
}
#if !(ALLCHARSDEBUG|(WRITECHARS != -1))
/* ##########################################################################
* timer stuff
* ##########################################################################*/
static uint8_t delay_spi
= 0;
/**
* timer kicks in every 1000uS ^= 1ms
*/
ISR
(TIMER0_COMP_vect
) {
if (!timer
--) {
uptime
++;
#if FCONLY
if (debugData.
Analog[12] > 10) {
flytime_fc
++;
}
#endif
timer
= 999;
seconds_since_last_data
++;
}
// in case there is still some spi data to send do it now
// delay to give the slave some time to compute values
if (spi_ready
&& icnt
) {
if (!delay_spi
--) {
delay_spi
= 8;
spi_send_next
();
}
}
}
#endif // ends !(ALLCHARSDEBUG|(WRITECHARS != -1))
/* ##########################################################################
* MAIN
* ##########################################################################*/
int main
(void) {
// set up FLAGS
COSD_FLAGS_MODES
= 0, COSD_FLAGS_CONFIG
= 0, COSD_FLAGS_RUNTIME
= 0;
#if NTSC
COSD_FLAGS_CONFIG
|= COSD_FLAG_NTSC
;
#endif
#if HUD
COSD_FLAGS_MODES
|= COSD_FLAG_HUD
;
#endif
#if ARTHORIZON
COSD_FLAGS_MODES
|= COSD_FLAG_ARTHORIZON
;
#endif
#if BIGVARIO
COSD_FLAGS_MODES
|= COSD_FLAG_BIGARIO
;
#endif
#if STATS
COSD_FLAGS_MODES
|= COSD_FLAG_STATS
;
#endif
#if WARNINGS
COSD_FLAGS_MODES
|= COSD_FLAG_WARNINGS
;
#endif
#if FCONLY
COSD_FLAGS_CONFIG
|= COSD_FLAG_FCMODE
;
#endif
// set up Atmega162 Ports
DDRA
|= ((1 << PA1
) | (1 << PA2
) | (1 << PA3
) | (1 << PA5
)); // PA1 output (/CS) | PA2 output (SDIN) |PA3 output (SCLK) | PA5 output (RESET)
MAX_CS_HIGH
MAX_SDIN_LOW
MAX_SCLK_LOW
MAX_RESET_HIGH
DDRC
|= ((1 << PC0
) | (1 << PC1
) | (1 << PC2
) | (1 << PC3
)); // PC0 output (LED1 gn) | PC1 output (LED2 rt) | PC2 output (LED3 gn) | PC3 output (LED4 rt)
LED1_OFF
LED2_OFF
LED3_OFF
LED4_OFF
DDRC
&= ~
((1 << PC4
) | (1 << PC5
) | (1 << PC6
) | (1 << PC7
)); // PC4 input(MODE) | PC5 input(SET) | PC6 input SJ1 | PC7 input SJ2
PORTC
|= ((1 << PC4
) | (1 << PC5
) | (1 << PC6
) | (1 << PC7
)); // pullup
// reset the MAX7456 to be sure any undefined states do no harm
MAX_RESET_LOW
MAX_RESET_HIGH
// give the FC/NC and the maxim time to come up
LED4_ON
_delay_ms
(1000);
LED4_OFF
//Pushing NEW chars to the MAX7456
#if (WRITECHARS != -1)
// DISABLE display (VM0)
spi_send_byte
(0x00, 0b00000000);
learn_all_chars_pgm
();
#else
// read out config for NTSC/PAL distinguishing
get_eeprom
(0);
#endif
// Setup Video Mode
if (COSD_FLAGS_CONFIG
& COSD_FLAG_NTSC
) {
// NTSC + enable display immediately (VM0)
spi_send_byte
(0x00, 0b00001000);
bottom_line
= 12;
} else {
// PAL + enable display immediately (VM0)
spi_send_byte
(0x00, 0b01001000);
bottom_line
= 14;
}
/*// clear all display-mem (DMM)
spi_send_byte(0x04, 0b00000100);
// clearing takes 12uS according to maxim so lets wait longer
_delay_us(120);
// 8bit mode
spi_send_byte(0x04, 0b01000000);*/
// clear display memory and set to 8bit mode
clear
();
#if !(ALLCHARSDEBUG|(WRITECHARS != -1))
// init usart
usart1_init
();
#ifdef SERIALDEBUGDRAW
#define USART0ENABLE 1
#endif
#ifdef ANTENNATRACKTEST
#define USART0ENABLE 1
#endif
#ifdef USART0ENABLE
usart0_init
();
usart0_puts
("\x1B[2J\x1B[H");
usart0_puts
("Welcome\r\n");
#endif
// keep serial port clean
usart1_DisableTXD
();
// set up timer
// CTC, Prescaler /64
TCCR0
= (1 << WGM01
) | (0 << WGM00
) | (0 << CS02
) | (1 << CS01
) | (1 << CS00
);
TCNT0
= 0;
OCR0
= 250;
// enable timer output compare interrupt
TIMSK
&= ~
(1 << TOIE0
);
TIMSK
|= (1 << OCIE0
);
// SPI setup
DDRD
|= (1 << PD2
); // PD2 output (INT0)
SpiMasterInit
();
// PPM detection setup
ppm_init
();
// enable interrupts
sei
();
//write_ascii_string(2,7, " CaScAdE");
//write_ascii_string(2,8, "is TESTING his open source");
//write_ascii_string(2,9, "EPi OSD Firmware");
// we are ready
LED3_ON
// clear serial screen
//usart1_puts("\x1B[2J\x1B[H");
COSD_FLAGS_RUNTIME
&= ~COSD_DATARECEIVED
;
#if !FCONLY
usart1_request_nc_uart
();
#endif
while (1) {
// debug ppm
//write_ndigit_number_u(9, 4, ppm, 5, 0);
// in case SPI is ready and there is nothing to send right now
if (!icnt
&& spi_ready
) {
// correct transfer ends with d (done)
if (SPI_buffer.
buffer.
chk == 'd') {
ampere
= SPI_buffer.
data.
ampere;
ampere_wasted
= SPI_buffer.
data.
mah;
s_volt
= SPI_buffer.
data.
volt;
// if this is the first receival we should print the small A
if (!(COSD_FLAGS_RUNTIME
& COSD_FLAG_STROMREC
)) {
clear
();
COSD_FLAGS_RUNTIME
&= ~COSD_ICONS_WRITTEN
;
// update this flag
COSD_FLAGS_RUNTIME
|= COSD_FLAG_STROMREC
;
}
} else {
// update flags
COSD_FLAGS_RUNTIME
&= ~COSD_FLAG_STROMREC
;
}
StartTransfer
(9);
}
if (rxd_buffer_locked
) {
#if FCONLY
if (rxd_buffer
[2] == 'D') { // FC Data
Decode64
();
//debugData = *((str_DebugOut*)pRxData);
memcpy((char*)(&debugData
), (char*)pRxData
, sizeof(str_DebugOut
));
rxd_buffer_locked
= 0;
// init on first data retrival, distinguished by last battery :)
if (!(COSD_FLAGS_RUNTIME
& COSD_OSD_STARTED
)) {
if (debugData.
Analog[9] > 40) {
// fix for min_UBat
min_UBat
= debugData.
Analog[9];
last_UBat
= debugData.
Analog[9];
init_cosd
(last_UBat
);
COSD_FLAGS_RUNTIME
|= COSD_OSD_STARTED
;
}
} else {
osd_fcmode
();
}
seconds_since_last_data
= 0;
} else {
rxd_buffer_locked
= 0;
}
#else
if (rxd_buffer
[2] == 'O') { // NC OSD Data
Decode64
();
//naviData = *((NaviData_t*)pRxData);
memcpy((char*)(&naviData
), (char*)pRxData
, sizeof(NaviData_t
));
rxd_buffer_locked
= 0;
#ifdef SHIFTBYminus45
naviData.
AngleNick = (int8_t)(((int16_t)naviData.
AngleRoll + (int16_t)naviData.
AngleNick) / 2);
naviData.
AngleRoll = (int8_t)(((int16_t)naviData.
AngleRoll - (int16_t)naviData.
AngleNick) / 2);
naviData.
CompassHeading = (naviData.
CompassHeading + (360 - 45)) % 360;
#endif
// init on first data retrival, distinguished by last battery :)
if (!(COSD_FLAGS_RUNTIME
& COSD_OSD_STARTED
)) {
if (naviData.
UBat > 40) {
// fix for min_UBat
min_UBat
= naviData.
UBat;
last_UBat
= naviData.
UBat;
init_cosd
(last_UBat
);
COSD_FLAGS_RUNTIME
|= COSD_OSD_STARTED
;
}
} else {
osd_ncmode
();
}
//seconds_since_last_data = 0;
// } else if (rxd_buffer[2] == 'C') { // 3D and Stick Data
// Decode64();
// memcpy((char*)(&data3d), (char*)pRxData, sizeof(Data3D_t));
// rxd_buffer_locked = 0;
} else {
rxd_buffer_locked
= 0;
}
#endif
// ONLY FOR TESTING
#ifdef ANTENNATRACKTEST
#include <stdlib.h>
//#include <float.h>
//#include <math.h>
//usart0_puts("\x1B[2J\x1B[H");
/*
naviData.HomePositionDeviation.Distance = 23 * 100; // 23m away (cm * 100)
naviData.HomePositionDeviation.Bearing = 35; // 35�
altimeter_offset = 50; // 50m start height
naviData.CurrentPosition.Altitude = (int32_t) ((int32_t)250 * (int32_t)1000); // 250m height (mm * 1000)
*/
static char conv_array
[7];
// should be float
int tanheight
= (naviData.
HomePositionDeviation.
Distance * 100) / (naviData.
CurrentPosition.
Altitude - altimeter_offset
);
// we need math.h and some faster AVR :)
//tanheight = rad2deg(atan(tanheight));
itoa((naviData.
HomePositionDeviation.
Bearing + 180) % 360, conv_array
, 10);
usart0_puts
("Bearing: ");
usart0_puts
(conv_array
);
usart0_puts
("\tHeightangle: ");
itoa(tanheight
, conv_array
, 10);
usart0_puts
(conv_array
);
usart0_puts
("\r\n");
#endif
}
// handle keypress
if (s1_pressed
()) {
config_menu
();
}
// reqest data untill there has been some answer
if (!(COSD_FLAGS_RUNTIME
& COSD_DATARECEIVED
) ||
// or while not in passive mode
(seconds_since_last_data
> 0 && !(COSD_FLAGS_CONFIG
& COSD_FLAG_PASSIVE
))) {
usart1_EnableTXD
();
//usart1_puts_pgm(PSTR("zu alt\r\n"));
#if FCONLY
// request data ever 100ms from FC;
//usart1_request_mk_data(0, 'd', 100);
usart1_puts_pgm
(PSTR
(REQUEST_DBG_DATA
));
#else
// request OSD Data from NC every 100ms
//usart1_request_mk_data(1, 'o', 100);
usart1_puts_pgm
(PSTR
(REQUEST_OSD_DATA
));
// request 3D Data from NC every 100ms
//usart1_request_mk_data(1, 'c', 100);
// _delay_ms(10);
// usart1_puts_pgm(PSTR(REQUEST_3DDATA));
// and disable debug...
//usart1_request_mk_data(0, 'd', 0);
#endif
// reset last time counter
seconds_since_last_data
= 0;
usart1_DisableTXD
();
// do not spam too much
if (!(COSD_FLAGS_RUNTIME
& COSD_DATARECEIVED
)) {
_delay_ms
(100);
}
}
if (SJ1_CLOSED
&& !(COSD_FLAGS_RUNTIME
& COSD_BLANKBYSJ
)) { // we do not want the HUD anymore
if (COSD_FLAGS_MODES
& COSD_FLAG_HUD
) {
clear
();
}
COSD_FLAGS_MODES
&= ~COSD_FLAG_HUD
;
COSD_FLAGS_RUNTIME
|= COSD_BLANKBYSJ
;
} else if (!SJ1_CLOSED
&& (COSD_FLAGS_RUNTIME
& COSD_BLANKBYSJ
)) { // we want the HUD back again
if (!(COSD_FLAGS_MODES
& COSD_FLAG_HUD
)) {
COSD_FLAGS_RUNTIME
&= ~COSD_ICONS_WRITTEN
;
}
COSD_FLAGS_MODES
|= COSD_FLAG_HUD
;
COSD_FLAGS_RUNTIME
&= ~
(COSD_BLANKBYSJ
);
}
}
#else // character flashing...
clear
();
write_all_chars
();
LED1_ON
LED2_ON
LED3_ON
LED4_ON
while (1) {
};
#endif
return 0;
}