X-Git-Url: https://www.fi.muni.cz/~kas/git//home/kas/public_html/git/?p=bike-lights.git;a=blobdiff_plain;f=firmware%2Fbuttons.c;h=ac41c06879b46a01c2cbae7ef19a321b271332a2;hp=467a09ab84f54f910ebba3adb32007c16beb23c3;hb=1d5dfb4d273a61877eb5937f93fbd266b782bc7a;hpb=47e678a90062fa2dbc6cb8f9e5efc81072f5dcc5

diff --git a/firmware/buttons.c b/firmware/buttons.c
index 467a09a..ac41c06 100644
--- a/firmware/buttons.c
+++ b/firmware/buttons.c
@@ -2,11 +2,26 @@
 #include <avr/interrupt.h>
 #include <avr/sleep.h>
 #include <util/delay.h>
+#include <stdlib.h> // for NULL
 
 #include "lights.h"
 
-static uint16_t button_start[N_BUTTONS];
-static unsigned char prev_pin;
+#define WAKEUP_LIMIT	5	// times 100 ms
+#define SHORT_PRESS_MIN 2       // in jiffies (16 Hz ticks)
+#define SHORT_PRESS_MAX 5
+#define LONG_PRESS_MIN  10
+
+static uint16_t button_start[3];
+union {
+	unsigned char all;
+	struct {
+		unsigned char btn1 : 1;
+		unsigned char btn2 : 1;
+		unsigned char brake : 1;
+		unsigned char brake_working : 1;
+		unsigned char brake_reported : 1;
+	};
+} button_state, prev_state;
 
 static unsigned char user_params[MAX_USER_PARAMS] = { 0, 0, 0 };
 static unsigned char user_params_max[MAX_USER_PARAMS] = { 3, 2, 2 };
@@ -20,14 +35,38 @@ static unsigned char user_params_state = 0;
 
 static uint16_t user_params_starttime;
 
-static void toggle_bright_mode()
+static void inline set_status_led(unsigned char n, pattern_t *pattern)
+{
+	led_set_pattern(n + 1, pattern);
+}
+
+unsigned char buttons_setup_in_progress()
 {
-	// TODO
+	if (user_params_state // setup in progress ...
+		// or at least one button is pressed:
+		|| prev_state.btn1
+		|| prev_state.btn2)
+		return 1;
+	return 0;
 }
 
-static void set_panic_mode()
+pattern_t *buttons_setup_status0_pattern_select()
 {
-	// TODO
+	if (user_params_state) // Setup in progress
+		return number_pattern(user_params_state, 1);
+	else
+		return NULL;
+}
+
+pattern_t *buttons_setup_status1_pattern_select()
+{
+	if (user_params_state) // Setup in progress
+		return number_pattern(
+			1 + user_params[user_params_state-1],
+			1
+		);
+	else
+		return NULL;
 }
 
 unsigned char get_user_param(unsigned char param)
@@ -43,7 +82,7 @@ static inline void short_press(unsigned char button)
 
 	if (user_params_state == 0) {
 		if (button == 0)
-			toggle_bright_mode();
+			toggle_dim_mode();
 		else
 			set_panic_mode();
 		return;
@@ -55,7 +94,7 @@ static inline void short_press(unsigned char button)
 		if (user_params[param])
 			user_params[param]--;
 		else
-			user_params[param] = user_params_max[param];
+			user_params[param] = user_params_max[param]-1;
 	} else {
 		user_params[param]++;
 
@@ -64,7 +103,7 @@ static inline void short_press(unsigned char button)
 	}
 	// FIXME: notify somebody about user_params change?
 
-	// set_status_led(2, user_params[param]);
+	set_status_led(1, buttons_setup_status1_pattern_select());
 
 	user_params_starttime = jiffies;
 }
@@ -72,72 +111,203 @@ static inline void short_press(unsigned char button)
 static inline void long_press(unsigned char button)
 {
 	if (button == 0) {
-		power_down();
+		power_down(0);
 		return;
 	}
 
 	// button 1 - cycle through states
 	user_params_state++;
 
-	if (user_params_state >= MAX_USER_PARAMS)
+	if (user_params_state > MAX_USER_PARAMS)
 		user_params_state = 1;
 
-	// set_status_led(1, state);
+	set_status_led(0, buttons_setup_status0_pattern_select());
+	set_status_led(1, buttons_setup_status1_pattern_select());
 	user_params_starttime = jiffies;
 }
 
 void init_buttons()
 {
-	unsigned char i;
+	DDRA &= ~_BV(PA4); // set as input
+	DDRA |= _BV(PA3); // output
+	PORTA &= ~(_BV(PA3) | _BV(PA4));  // PA4 disable internal pull-up
+					// PA3 set to zero
 
-	DDRA &= ~(_BV(PA3) | _BV(PA4));
-	PORTA |=  _BV(PA3) | _BV(PA4);
+	GIMSK &= ~(_BV(PCIE0) | _BV(PCIE1)); // disable pin-change IRQs
+	PCMSK0 = 0; // disable pin-change IRQs on all pins of port A
+	PCMSK1 = 0; // disable pin-change IRQs on all pins of port B
 
 	button_start[0] = 0;
 	button_start[1] = 0;
-	prev_pin = _BV(PA3) | _BV(PA4);
+	button_start[2] = 0;
+
+	prev_state.all    = 0;
+	button_state.all  = 0;
+	user_params_state = 0;
 }
 
-void timer_check_buttons()
+void susp_buttons()
 {
-	unsigned char pin = PINA & (_BV(PA3) | _BV(PA4));
-
-	handle_button(0, pin & _BV(PA3), prev_pin & _BV(PA3));
-	handle_button(1, pin & _BV(PA4), prev_pin & _BV(PA4));
+	DDRA &= ~(_BV(PA3) | _BV(PA4)); // set as input
+	PORTA |= _BV(PA4);  // enable internal pull-up
+	PORTA &= ~_BV(PA3); // disable pull-up
 
-	prev_pin = pin;
+	GIMSK &= ~_BV(PCIE0);
+	GIMSK |= _BV(PCIE1);
 
-	if (jiffies - user_params_starttime > 500) {
-		user_params_state = 0;
-		// set_status_led(0, idle);
-		// set_status_led(1, idle);
-	}
+	PCMSK0 = _BV(PCINT4);
+		// disable pin-change IRQs on all pins except PA4
+	PCMSK1 = 0; // disable pin-change IRQs on all pins of port B
 }
 
 static void handle_button(unsigned char button, unsigned char cur,
 	unsigned char prev)
 {
-	// BEWARE: pins are at _zero_ when pressed!
-	if (!cur && prev) {                   // --- just pressed ---
-		button_start[i] = jiffies;
-		// set_status_led(button, 0);
+	if (cur && !prev) {                   // --- just pressed ---
+		button_start[button] = jiffies;
+		set_status_led(button, NULL);
 
-	} else if (!cur && !prev) {           // --- is still pressed ---
-		uint16_t duration = jiffies - button_start[i];
+	} else if (cur && prev) {           // --- is still pressed ---
+		uint16_t duration = jiffies - button_start[button];
 
-		if (duration > 80) {
-			// set_status_led(button, 1); // acknowledge long press
+		if (duration >= LONG_PRESS_MIN) {
+			set_status_led(button, on1_pattern);
+				// acknowledge long press
 		}
-	} else if (cur && !prev) {            // --- just released ---
-		uint16_t duration = jiffies - button_start[i];
+	} else if (!cur && prev) {            // --- just released ---
+		uint16_t duration = jiffies - button_start[button];
 
-		if (duration > 6 && duration < 30)
+		if (duration >= SHORT_PRESS_MIN && duration < SHORT_PRESS_MAX) {
 			short_press(button);
-		} else if (duration > 80) {
-			// set_status_led(button, 0);
+		} else if (duration > LONG_PRESS_MIN) {
+			set_status_led(button, NULL);
 			long_press(button);
 		}
 		// ignore other button-press durations
 	}
 }
 
+static void handle_brake(unsigned char cur, unsigned char prev)
+{
+	uint16_t duration;
+
+	if (cur && !prev) {                   // --- just pressed ---
+		button_start[2] = jiffies;
+		return;
+	} else if (!cur && prev) {            // --- just released ---
+		button_start[2] = jiffies;
+		return;
+	}
+				              // --- no change ---
+	duration = jiffies - button_start[2];
+
+	if (duration <= 3)
+		return;
+
+	if (cur) {
+		if (button_state.brake_working) {
+			static unsigned int brake_time;
+			if (button_state.brake_reported) {
+				if (brake_time) {
+					brake_time--;
+				} else {
+					brake_off();
+					button_state.brake_working = 0;
+					button_state.brake_reported = 0;
+				}
+			} else {
+				button_state.brake_reported = 1;
+				brake_on();
+				brake_time = 255; // avoid longer than ~16s
+			}
+		}
+	} else {
+		button_state.brake_working = 1;
+		if (button_state.brake_reported) {
+			button_state.brake_reported = 0;
+			brake_off();
+		}
+	}
+
+	button_start[2] = jiffies - 7; // avoid overflow
+}
+
+void timer_check_buttons()
+{
+	handle_button(0, button_state.btn1, prev_state.btn1);
+
+	// when button 1 is on, all other states are unreadable
+	if (button_state.btn1) {
+		prev_state.btn1 = button_state.btn1;
+	} else {
+		handle_button(1, button_state.btn2, prev_state.btn2);
+		handle_brake(button_state.brake, prev_state.brake);
+		prev_state.all = button_state.all;
+	}
+
+	if (user_params_state && jiffies - user_params_starttime > 1000) {
+		user_params_state = 0;
+		set_status_led(0, buttons_setup_status0_pattern_select());
+		set_status_led(1, buttons_setup_status1_pattern_select());
+	}
+}
+
+unsigned char buttons_wait_for_release()
+{
+	uint16_t wake_count = 0;
+	unsigned char pin;
+
+	do {
+		if (wake_count++ > WAKEUP_LIMIT)
+			gpio_set(0, 1); // inform the user
+
+		_delay_ms(100);
+
+		pin = PINA & _BV(PA4);
+	} while (!(pin & _BV(PA4)));
+
+	gpio_set(0, 0);
+
+	return wake_count > WAKEUP_LIMIT;
+}
+
+void button_adc(uint16_t adcval)
+{
+#if 0
+	static unsigned char count=250;
+#endif
+	PORTA &= ~_BV(PA3); // disable +5V to the hall probe
+		// adc.c will re-enable it again
+
+	button_state.btn1 = 0;
+	button_state.btn2 = 0;
+	button_state.brake = 0;
+
+	if (adcval == 0x3FF) {
+		button_state.brake = 1;
+	} else if (adcval > 0x240) {
+		button_state.brake = 1;
+		button_state.btn2 = 1;
+	} else if (adcval > 0x180) {
+		// nothing
+	} else if (adcval > 0xc0) {
+		button_state.btn2 = 1;
+	} else {
+		button_state.btn1 = 1;
+	}
+
+#if 0
+	if (--count)
+		return;
+
+	count = 250;
+	log_word(adcval);
+	log_flush();
+#endif
+}
+
+ISR(PCINT_vect)
+{
+	// empty - let it wake us from sleep, but do nothing else
+}
+