The unified diff between revisions [8f4e93ab..] and [5a7404c7..] is displayed below. It can also be downloaded as a raw diff.

This diff has been restricted to the following files: 'main.c'

#
# old_revision [8f4e93ab4d89edfdbd524b06b83511c6867a9150]
# new_revision [5a7404c7ee458d3eab4b0548e4bc6b6a8b26135f]
#
# patch "main.c"
#  from [495b950dc0db657d6169223890154d27de162224]
#    to [cffbf353c53aca29609c39daadeb325f98328bd5]
#
============================================================
--- main.c	495b950dc0db657d6169223890154d27de162224
+++ main.c	cffbf353c53aca29609c39daadeb325f98328bd5
@@ -1,3 +1,4 @@
+/* main.c */
 
 #include "wmp.h"
 #include "i2c.h"
@@ -5,28 +6,42 @@
 #include "uart.h"
 #include "interrupt.h"
 #include "event.h"
+#include "led.h"
+#include "status.h"
+#include "watchdog.h"
+#include "thrust.h"
 
-#define PINSEL0 (*((volatile unsigned char *) 0xE002C000))
+#define PINSEL0 (*((volatile unsigned int *) 0xE002C000))
+#define PINSEL1 (*((volatile unsigned int *) 0xE002C004))
 #define FP0XDIR (*((volatile unsigned int *) 0x3FFFC000))
 #define FP0XVAL (*((volatile unsigned int *) 0x3FFFC014))
 
 #define SCS (*((volatile unsigned int *) 0xe01fc1a0))
 
+#define BUTTON_PRESSED (!((FP0XVAL) & 0x00010000))
 
 void init_pins(void)
 {
-	PINSEL0 = 0x00000055; /* P0.0 and P0.1 assigned to UART */
+	PINSEL0 = 0x00a88055; /* P0.0 and P0.1 assigned to UART */
 			      /* P0.2 and P0.3 assigned to I2C  */
+	                      /* P0.7 and P0.9 assigned to MAT2.[02] */
+	                      /* P0.10 and P0.11 assigned to CAP1.[01] */
+	PINSEL1 = 0x20000828; /* P0.21 and P0.30 assigned to MAT3.[03] */
+	                      /* P0.17 and P0.18 assigned to CAP1.[23] */
 	SCS = 1;
 	FP0XDIR = 0x04000000; /* P0.26 is an output */
 	FP0XVAL = 0x0;
 }
 
+#ifdef USE_UART
 void reply(char *str)
 {
 	putstr(str);
 	putstr("\r\n");
 }
+#else
+#define reply(x) ((void)0)
+#endif
 
 unsigned int count = 0;
 
@@ -166,12 +181,60 @@ void menu_handler(void);
 
 void menu_handler(void);
 
+void wait_for_button_pressed(bool target)
+{
+	bool pressed;
+
+	led_set(!target);
+
+	/* Very crude debouncing */
+	timer_delay_ms(100);
+
+	target = target ? TRUE:FALSE;
+
+	do {
+		pressed = BUTTON_PRESSED;
+	} while (pressed != target);
+	led_set(pressed);
+}
+
+void calibrate_escs()
+{
+	wait_for_button_pressed(0);
+
+	putstr("Calibration mode\r\n");
+
+	wait_for_button_pressed(1);
+	wait_for_button_pressed(0);
+
+	set_thrust(0, 1.0);
+	set_thrust(1, 1.0);
+	set_thrust(2, 1.0);
+	set_thrust(3, 1.0);
+	putstr("Max throttle set\r\n");
+
+	wait_for_button_pressed(1);
+	wait_for_button_pressed(0);
+
+	set_thrust(0, 0.0);
+	set_thrust(1, 0.0);
+	set_thrust(2, 0.0);
+	set_thrust(3, 0.0);
+	putstr("Zero throttle set\r\n");
+
+	wait_for_button_pressed(1);
+	wait_for_button_pressed(0);
+
+	putstr("Exit calibration mode\r\n");
+}
+
 int main(void) {
 	init_interrupt();
 	init_uart();
 	init_i2c();
 	init_pins();
 	init_timer();
+	init_status();
 
 	event_register(EVENT_UART_INPUT, menu_handler);
 
@@ -181,22 +244,44 @@ int main(void) {
 
 	putstr("Your entire life has been a mathematical error... a mathematical error I'm about to correct!\r\n");
 
+	if (BUTTON_PRESSED)
+		calibrate_escs();
+
 	putstr("prompt> ");
 
+	timer_delay_ms(1000);
+	if (!wmp_init())
+		putstr("WMP initialisation failed\r\n");
+
+	/* Flight is potentially live after this. */
+	timer_set_period(5*TIMER_MS);
+	wmp_start_zero();
+
+	led_init();
+
+	init_watchdog();
+
+	/* Good luck! */
 	while (1) {
-		FP0XVAL ^= 0x04000000;
+		led_update();
 		event_dispatch();
+		watchdog_check();
 	}
 
 	return 0;
 }
 
+static int power = 0;
+
 void menu_handler(void)
 {
 	int i;
 	char c;
 
 	while (getch(&c)) {
+#if 1
+		continue; /* Yes, let's just ignore UART input now. */
+#endif
 		if (c == 0x0a)
 			continue;
 		putch(c);
@@ -277,13 +362,85 @@ void menu_handler(void)
 			break;
 		case 'P':
 			putstr("Initialising timer... ");
-			timer_set_period(10*TIMER_MS);
+			/* We want a 100Hz loop but two samples per iteration.
+			 * So, we go for 200Hz. */
+			timer_set_period(5*TIMER_MS);
 			reply("done");
+			wmp_start_zero();
 			break;
-		case 'E':
-			event_dispatch();
-			reply("done");
+		case 'W':
+			for (i = 0; i < 4; i++) {
+				putstr("Width ");
+				putint(i);
+				putstr(": ");
+				putint(timer_input(i));
+				if (!timer_valid(i))
+					putstr(" (invalid)");
+				putstr("\r\n");
+			}
+			if (!timer_allvalid()) {
+				putstr("ALL INVALID!\r\n");
+			}
 			break;
+		case '0' & 0xdf:
+			set_thrust(0, 0.0);
+			set_thrust(1, 0.0);
+			set_thrust(2, 0.0);
+			set_thrust(3, 0.0);
+			power = 0;
+			break;
+#if 0
+		case '1' & 0xdf:
+			power--;
+			if (power < 0)
+				power = 15;
+			power = power % 16;
+			putstr("Power setting: ");
+			putint(power);
+			putstr("\r\n");
+			set_thrust(0, ((float)power)/16.0);
+			break;
+		case '2' & 0xdf:
+			power++;
+			power = power % 16;
+			putstr("Power setting: ");
+			putint(power);
+			putstr("\r\n");
+			set_thrust(0, ((float)power)/16.0);
+			break;
+#endif
+#if 0
+		case '1' & 0xdf:
+			set_thrust(0, 0.5);
+			break;
+		case '2' & 0xdf:
+			set_thrust(1, 0.5);
+			break;
+		case '3' & 0xdf:
+			set_thrust(2, 0.5);
+			break;
+		case '4' & 0xdf:
+			set_thrust(3, 0.5);
+			break;
+		case '5' & 0xdf:
+			set_thrust(0, 1.0);
+			break;
+		case '6' & 0xdf:
+			set_thrust(1, 1.0);
+			break;
+		case '7' & 0xdf:
+			set_thrust(2, 1.0);
+			break;
+		case '8' & 0xdf:
+			set_thrust(3, 1.0);
+			break;
+#endif
+		case '9' & 0xdf:
+			timer_set_pwm_invalid(0);
+			timer_set_pwm_invalid(1);
+			timer_set_pwm_invalid(2);
+			timer_set_pwm_invalid(3);
+			break;
 		default:
 			reply("Unrecognised command.");
 			break;