Rev 467 |
Rev 474 |
Go to most recent revision |
Blame |
Compare with Previous |
Last modification |
View Log
| RSS feed
/****************************************************************************
* Copyright (C) 2009 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 *
* 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 <avr/io.h>
#include <avr/interrupt.h>
#include <util/delay.h>
#include <avr/pgmspace.h>
#include "main.h"
#include "max7456_software_spi.h"
#include "usart1.h"
#include "osd_helpers.h"
/* TODO:
* - verifiy correctness of values
* - clean up code :)
*/
#if !(ALLCHARSDEBUG|(WRITECHARS != -1))
// data structs not needed for character flashing
#include "mk-data-structs.h"
/* ##########################################################################
* global definitions and global vars
* ##########################################################################*/
volatile uint16_t setsReceived = 0;
volatile NaviData_t naviData;
volatile DebugOut_t debugData;
// 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;
// remember last time data was received
volatile uint8_t seconds_since_last_data = 0;
// store stats description in progmem to save space
char stats_item_0[] PROGMEM = "max Altitude:";
char stats_item_1[] PROGMEM = "max Speed :";
char stats_item_2[] PROGMEM = "max Distance:";
char stats_item_3[] PROGMEM = "min voltage :";
char stats_item_4[] PROGMEM = "max time :";
char stats_item_5[] PROGMEM = "longitude :";
char stats_item_6[] PROGMEM = "latitude :";
char stats_item_7[] PROGMEM = "max current :";
char* stats_item_pointers[] PROGMEM = {stats_item_0, stats_item_1, stats_item_2,
stats_item_3, stats_item_4, stats_item_5, stats_item_6, stats_item_7};
// store more fixed strings in progmen
char ON[] PROGMEM = "ON ";
char OFF[] PROGMEM = "OFF";
#endif // ends !(ALLCHARSDEBUG|(WRITECHARS != -1))
// 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;
// Flags
uint8_t COSD_FLAGS = 0, COSD_FLAGS2;
/* ##########################################################################
* debounce buttons
* ##########################################################################*/
int s1_pressed() {
if (S1_PRESSED) {
_delay_ms(25);
if (S1_PRESSED) return 1;
}
return 0;
}
int s2_pressed() {
if (S2_PRESSED) {
_delay_ms(25);
if (S2_PRESSED) return 1;
}
return 0;
}
/* ##########################################################################
* 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
* ##########################################################################*/
/* ##########################################################################
* STROM TEST
* ##########################################################################*/
#define INT0_HIGH PORTD |= (1 << PD2);
#define INT0_LOW PORTD &= ~(1 << PD2);
void SpiMasterInit(void) {
volatile char IOReg;
// set PB4(/SS), PB5(MOSI), PB7(SCK) as output
DDRB = (1 << PB4) | (1 << PB5) | (1 << PB7);
PORTB |= (1 << PB4); // pullup SS
// enable SPI Interrupt and SPI in Master Mode with SCK = CK/128
SPCR = (1 << SPIE) | (1 << SPE) | (1 << MSTR) | (1 << SPR0) | (1 << SPR1);
IOReg = SPSR; // clear SPIF bit in SPSR
IOReg = SPDR;
//sei(); // we do it later
}
volatile size_t icnt = 0;
volatile unsigned char * iptr;
volatile unsigned char spi_cmd_buffer[8];
volatile uint8_t spi_ready = 1;
int16_t ampere = 0, max_ampere = 0;
int32_t ampere_wasted = 0;
/**
* SPI interrupt handler
*/
ISR(SPI_STC_vect) {
*iptr++ = SPDR; // safe received byte to buffer
icnt--; // dec length
if (icnt) {
//SPDR = *iptr; // send next byte
spi_ready = 1; // we _should_ send later because the slave needs more time
} else {
SPCR &= ~_BV(SPIE); // deactivate interrupt
INT0_HIGH // transfer is done, slave does not need to listen
}
}
/**
* check if SPI transfer is still busy
*/
int TransferIsBusy(void) {
return SPCR & _BV(SPIE);
}
/**
* start a new transfer of <data> with length <len>
*/
void StartTransfer(unsigned char *data, size_t len) {
INT0_LOW // /SS LOW ^= SS HIGH ^= slave should listen
// set up pointer and length for interrupt handler
iptr = data;
icnt = len;
SPCR |= _BV(SPIE); // enable spi interrupt
SPDR = *iptr; // start transfer by first bye
}
/* ##########################################################################
* / STROM TEST END
* ##########################################################################*/
/**
* timer kicks in every 1000uS ^= 1ms
*/
ISR(TIMER0_OVF_vect) {
OCR0 = 15; // preload
if (!timer--) {
uptime++;
timer = 999;
seconds_since_last_data++;
}
// in case there is still some spi data to send do it now
if (spi_ready && icnt) {
SPDR = *iptr;
}
}
/* ##########################################################################
* A simple config menu for the flags
* ##########################################################################*/
/**
* helper function for menu updating
*/
void config_menu_drawings(uint8_t chosen) {
// clear prevoius _cursor_
write_ascii_string(3, (chosen + 5) % 8, " ");
// draw current _cursor_
write_ascii_string(3, chosen + 5, ">");
if (COSD_FLAGS & COSD_FLAG_HUD) {
write_ascii_string_pgm(23, 5, ON);
} else {
write_ascii_string_pgm(23, 5, OFF);
}
if (COSD_FLAGS & COSD_FLAG_ARTHORIZON) {
write_ascii_string_pgm(23, 6, ON);
} else {
write_ascii_string_pgm(23, 6, OFF);
}
if (COSD_FLAGS & COSD_FLAG_BIGVARIO) {
write_ascii_string_pgm(23, 7, ON);
} else {
write_ascii_string_pgm(23, 7, OFF);
}
if (COSD_FLAGS & COSD_FLAG_STATS) {
write_ascii_string_pgm(23, 8, ON);
} else {
write_ascii_string_pgm(23, 8, OFF);
}
if (COSD_FLAGS & COSD_FLAG_WARNINGS) {
write_ascii_string_pgm(23, 9, ON);
} else {
write_ascii_string_pgm(23, 9, OFF);
}
}
/**
* some sort of clicking response in the menu
*/
void config_menu_doclick(uint8_t chosen, char** menu) {
write_ascii_string(4, chosen + 5, "DONE ");
_delay_ms(1000);
write_ascii_string(4, chosen + 5, menu[chosen]);
}
/**
* a simple config menu tryout
*/
void config_menu(void) {
// disable interrupts (makes the menu more smoothely)
cli();
// clear screen
clear();
char* menu[9] = {"Full HUD",
"Art.Horizon in HUD",
"Big Vario bar",
"Statistics",
"Warnings", // TODO: do it!
"Reset uptime",
"Request OSD-data",
"Disable Debug-data",
"EXIT"};
uint8_t inmenu = 1;
uint8_t chosen = 0;
write_ascii_string(6, 2, "C-OSD Config Menu");
// wait a bit before doing stuff so user has chance to release button
_delay_ms(250);
write_ascii_string(4, 5, menu[0]);
write_ascii_string(4, 6, menu[1]);
write_ascii_string(4, 7, menu[2]);
write_ascii_string(4, 8, menu[3]);
write_ascii_string(4, 9, menu[4]);
write_ascii_string(4, 10, menu[5]);
write_ascii_string(4, 11, menu[6]);
write_ascii_string(4, 12, menu[7]);
write_ascii_string(4, 13, menu[8]);
config_menu_drawings(chosen);
while (inmenu) {
if (s2_pressed()) {
write_ascii_string(3, chosen + 5, " ");
chosen = (chosen + 1) % 9;
write_ascii_string(3, chosen + 5, ">");
_delay_ms(500);
} else if (s1_pressed()) {
switch (chosen) {
case 0: // full HUD
COSD_FLAGS ^= COSD_FLAG_HUD;
config_menu_drawings(chosen);
break;
case 1: // art horizon
COSD_FLAGS ^= COSD_FLAG_ARTHORIZON;
config_menu_drawings(chosen);
break;
case 2: // big vario
COSD_FLAGS ^= COSD_FLAG_BIGVARIO;
config_menu_drawings(chosen);
break;
case 3: // statistics
COSD_FLAGS ^= COSD_FLAG_STATS;
config_menu_drawings(chosen);
break;
case 4: // warnings
COSD_FLAGS ^= COSD_FLAG_WARNINGS;
config_menu_drawings(chosen);
break;
case 5: // reset uptime
uptime = 0;
config_menu_doclick(chosen, menu);
break;
case 6: // re-request OSD data
#if FCONLY
// request data ever 100ms from FC;
usart1_request_mk_data(0, 'd', 100);
#else
// request OSD Data from NC every 100ms
usart1_request_mk_data(1, 'o', 100);
// and disable debug...
usart1_request_mk_data(0, 'd', 0);
#endif
config_menu_doclick(chosen, menu);
break;
case 7: // disable debug data
// disable sending of debug data
// may result in smoother ddata display
usart1_request_mk_data(0, 'd', 0);
config_menu_doclick(chosen, menu);
break;
case 8: // exit
inmenu = 0;
break;
}
_delay_ms(250);
}
}
// clear screen up again
clear();
// update flags to paint display again if needed
COSD_FLAGS &= ~COSD_ICONS_WRITTEN;
// enable interrupts again
sei();
}
/**
* auto config some stuff on startup, currently only battery cells
* TODO: this is testing stuff, strings should go progmem and so on...
*/
void auto_config(uint8_t UBat) {
clear();
write_ascii_string(2, 2, "C-OSD Initialisation");
#if FCONLY
write_ascii_string(2, 3, "FC only Mode");
#else
write_ascii_string(2, 3, "NaviCtrl Mode");
#endif
write_ascii_string(2, 4, BUILDDATE);
uint8_t cellnum = 0;
if (CELL_NUM == -1) {
write_ascii_string(2, 6, "Guessing Number of Cells");
do {
cellnum++;
} while (UBat > ((cellnum * CELL_VOLT_MAX) + 23));
} else {
cellnum = CELL_NUM;
}
min_voltage = cellnum * CELL_VOLT_MIN;
max_voltage = cellnum * CELL_VOLT_MAX;
write_ascii_string(2, 7, "Number of Cells:");
write_ndigit_number_u(21, 7, cellnum, 1, 0);
write_ascii_string(2, 8, "Warn Voltage :");
write_ndigit_number_s_10th(20, 8, min_voltage, 100, 0);
write_ascii_string(2, 9, "Max Voltage :");
write_ndigit_number_s_10th(20, 9, max_voltage, 100, 0);
_delay_ms(200);
clear();
// update flags to paint display again because of clear
COSD_FLAGS &= ~COSD_ICONS_WRITTEN;
}
#endif // ends !(ALLCHARSDEBUG|(WRITECHARS != -1))#endif // ends !(ALLCHARSDEBUG|(WRITECHARS != -1))
/* ##########################################################################
* MAIN
* ##########################################################################*/
int main(void) {
// set up FLAGS, compiler should flatten this one
COSD_FLAGS = (NTSC << 0);
COSD_FLAGS |= (HUD << 1);
COSD_FLAGS |= (ARTHORIZON << 2);
COSD_FLAGS |= (BIGVARIO << 3);
COSD_FLAGS |= (STATS << 4);
COSD_FLAGS |= (WARNINGS << 5);
// set up Atmega162 Ports
DDRA |= (1 << PA1); // PA1 output (/CS)
MAX_CS_HIGH
DDRA |= (1 << PA2); // PA2 output (SDIN)
MAX_SDIN_LOW
DDRA |= (1 << PA3); // PA3 output (SCLK)
MAX_SCLK_LOW
DDRA |= (1 << PA5); // PA5 output (RESET)
MAX_RESET_HIGH
DDRC |= (1 << PC0); // PC0 output (LED1 gn)
LED1_OFF
DDRC |= (1 << PC1); // PC1 output (LED2 rt)
LED2_OFF
DDRC |= (1 << PC2); // PC2 output (LED3 gn)
LED3_OFF
DDRC |= (1 << PC3); // PC3 output (LED4 rt)
LED4_OFF
DDRC &= ~(1 << PC4); // PC4 input (MODE)
PORTC |= (1 << PC4); // pullup
DDRC &= ~(1 << PC5); // PC5 input (SET)
PORTC |= (1 << PC5); // pullup
// set up top and bottom lines
if (COSD_FLAGS & COSD_FLAG_NTSC) {
bottom_line = 12;
} else {
bottom_line = 14;
}
// 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(2000);
LED4_OFF
//Pushing NEW chars to the MAX7456
#if (WRITECHARS != -1)
// DISABLE display (VM0)
spi_send_byte(0x00, 0b00000000);
#include "characters.c"
#endif
// Setup Video Mode
if (COSD_FLAGS & COSD_FLAG_NTSC) {
// NTSC + enable display immediately (VM0)
spi_send_byte(0x00, 0b00001000);
} else {
// PAL + enable display immediately (VM0)
spi_send_byte(0x00, 0b01001000);
}
/*// 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();
// set up timer
TCCR0 |= (1 << CS00) | (1 << CS01); // timer0 prescaler 64
OCR0 = 15; // preload
TIMSK |= (1 << TOIE0); // enable overflow timer0
DDRD |= (1 << PD2); // PD2 output (INT0)
SpiMasterInit();
// enable interrupts
sei();
#endif
//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
#if ALLCHARSDEBUG | (WRITECHARS != -1)
clear();
write_all_chars();
#else
// clear serial screen
//usart1_puts("\x1B[2J\x1B[H");
//usart1_puts("hello world!\r\n");
#if FCONLY
// request data ever 100ms from FC;
usart1_request_mk_data(0, 'd', 100);
#else
// request OSD Data from NC every 100ms
usart1_request_mk_data(1, 'o', 100);
// and disable debug...
usart1_request_mk_data(0, 'd', 0);
#endif
// stats for after flight
int16_t max_Altimeter = 0;
uint8_t min_UBat = 255;
#if !(FCONLY)
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};
#endif
while (1) {
// 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_cmd_buffer[0] == 'd') {
ampere = spi_cmd_buffer[1] << 8;
ampere |= spi_cmd_buffer[2];
ampere_wasted = spi_cmd_buffer[3];
ampere_wasted = ampere_wasted << 8;
ampere_wasted |= spi_cmd_buffer[4];
ampere_wasted = ampere_wasted << 8;
ampere_wasted |= spi_cmd_buffer[5];
ampere_wasted = ampere_wasted << 8;
ampere_wasted |= spi_cmd_buffer[6];
//ampere = *((int16_t *) spi_cmd_buffer + 1);
//ampere_wasted = *((int32_t *) &spi_cmd_buffer + 3);
// if this is the first receival we should print the small A
if (!(COSD_FLAGS2 & COSD_FLAG_STROMREC)) {
COSD_FLAGS &= ~COSD_ICONS_WRITTEN;
// update this flag
COSD_FLAGS2 |= COSD_FLAG_STROMREC;
}
//write_ascii_char(8+4*30, 'v'); // valid
} else {
// update flags
COSD_FLAGS2 &= ~COSD_FLAG_STROMREC;
//write_ascii_char(8+4*30, 'i'); // invalid
}
spi_cmd_buffer[0] = 'A';
spi_cmd_buffer[1] = 'B';
spi_cmd_buffer[2] = 'C';
spi_cmd_buffer[3] = 'D';
spi_cmd_buffer[4] = 'E';
spi_cmd_buffer[5] = 'F';
spi_cmd_buffer[6] = 'G';
StartTransfer((unsigned char*) spi_cmd_buffer, 7);
}
if (rxd_buffer_locked) {
if (COSD_FLAGS & COSD_FLAG_HUD) {
#if FCONLY
if (rxd_buffer[2] == 'D') { // FC Data
Decode64();
debugData = *((DebugOut_t*) pRxData);
// init on first data retrival, distinguished by last battery :)
if (last_UBat == 255) {
// fix for min_UBat
min_UBat = debugData.Analog[9];
auto_config(debugData.Analog[9]);
}
#include "osd_fcmode_default.c"
}
#else
if (rxd_buffer[2] == 'O') { // NC OSD Data
Decode64();
naviData = *((NaviData_t*) pRxData);
// init on first data retrival, distinguished by last battery :)
if (last_UBat == 255) {
// fix for min_UBat
min_UBat = naviData.UBat;
auto_config(naviData.UBat);
}
#include "osd_ncmode_default.c"
}
#endif
}
rxd_buffer_locked = 0;
}
// handle keypress
if (s1_pressed()) {
config_menu();
}
if (seconds_since_last_data > 2) {
#if FCONLY
// request data ever 100ms from FC;
usart1_request_mk_data(0, 'd', 100);
#else
// request OSD Data from NC every 100ms
usart1_request_mk_data(1, 'o', 100);
// and disable debug...
usart1_request_mk_data(0, 'd', 0);
#endif
}
}
#endif
return 0;
}