X-Git-Url: https://www.fi.muni.cz/~kas/git//home/kas/public_html/git/?p=bike-lights.git;a=blobdiff_plain;f=firmware%2Fcontrol.c;h=da59972b135c59c2cb2b5678aa4bb578724a327c;hp=57968357f970ad8284cd97f3021cf40eccacd17f;hb=c30006aaf666f7cff3a6ab949c613c2f8cc6163b;hpb=f2a1b61c8c58ba290e073f53f3a0f461da9ce011

diff --git a/firmware/control.c b/firmware/control.c
index 5796835..da59972 100644
--- a/firmware/control.c
+++ b/firmware/control.c
@@ -44,6 +44,12 @@ static pattern_t on3_pattern [] = {
 	PATTERN_END
 };
 
+static pattern_t brake_pattern [] = {
+	{ 4, 0x2 },
+	{ 3, 0x8 },
+	PATTERN_END
+};
+
 static pattern_t normal2_pattern[] = {
 	{ 2, 0x1 },
 	{ 0, 0x1 },
@@ -98,12 +104,29 @@ static pattern_t slow3_pattern[] = {
 	PATTERN_END
 };
 
-static unsigned char dim_mode, towbar_mode;
+static unsigned char dim_mode, towbar_mode, braking;
 
 void init_control()
 {
 	dim_mode = 0;
 	towbar_mode = 0;
+	braking = 0;
+}
+
+void brake_on()
+{
+	braking = 1;
+	gpio_set(0, 1);
+	led_set_pattern(N_PWMLEDS, status_led_pattern_select());
+	led_set_pattern(0, pwmled0_pattern_select());
+}
+
+void brake_off()
+{
+	braking = 0;
+	gpio_set(0, 0);
+	led_set_pattern(N_PWMLEDS, status_led_pattern_select());
+	led_set_pattern(0, pwmled0_pattern_select());
 }
 
 void toggle_dim_mode()
@@ -124,19 +147,19 @@ void set_panic_mode()
 
 pattern_t *pwmled0_pattern_select()
 {
-	// TODO: battery critical
-
-#ifndef TESTING_FW
-	return normal3_pattern;
-#endif
+	if (battery_critical)
+		return on1_pattern;
 
 	if (towbar_mode)
 		return NULL;
 
+	if (braking)
+		return brake_pattern;
+
 	switch (ambient_zone) {
-	case 0: return dim_mode ? NULL : on1_pattern;
-	case 1: return dim_mode ? NULL : slow2_pattern;
-	case 2: return dim_mode ? slow3_pattern : slow2_pattern;
+	case 0: return dim_mode ? NULL : number_pattern(2, 1);
+	case 1: return dim_mode ? slow1_pattern : normal2_pattern;
+	case 2: return dim_mode ? slow2_pattern : normal3_pattern;
 	case 3:
 	default: return dim_mode ? slow3_pattern : normal4_pattern;
 	}
@@ -144,11 +167,13 @@ pattern_t *pwmled0_pattern_select()
 
 pattern_t *pwmled1_pattern_select()
 {
-	// TODO: battery critical
-
 #ifndef TESTING_FW
-	return off_pattern;
+	return NULL;
+#else
+	if (battery_critical)
+		return on1_pattern;
 #endif
+
 	if (towbar_mode) {
 		switch (ambient_zone) {
 		case 0:
@@ -171,11 +196,11 @@ pattern_t *pwmled1_pattern_select()
 
 pattern_t *pwmled2_pattern_select()
 {
-	// TODO: battery critical
-
 #ifndef TESTING_FW
-	return on3_pattern;
+	if (battery_critical)
+		return on1_pattern;
 #endif
+
 	switch (ambient_zone) {
 	case 0: return dim_mode ? on2_pattern : on3_pattern;
 	case 1: return dim_mode ? slow1_pattern : normal2_pattern;
@@ -188,16 +213,20 @@ pattern_t *pwmled2_pattern_select()
 
 pattern_t *status_led_pattern_select()
 {
+	if (braking)
+		return on1_pattern;
+
 	if (buttons_setup_in_progress())
 		return buttons_setup_status0_pattern_select();
 
 	// FIXME: do something sane
-	return number_pattern(1 + ambient_zone, 0);
+	return number_pattern(battery_gauge(), 0);
 }
 
 pattern_t *illumination_led_pattern_select()
 {
-	// todo: battery critical
+	if (battery_critical)
+		return NULL;
 
 	switch (ambient_zone) {
 	case 0: return dim_mode
@@ -207,12 +236,19 @@ pattern_t *illumination_led_pattern_select()
 		? number_pattern(2, 1)
 		: number_pattern(3, 1);
 	case 2: return dim_mode
-		? number_pattern(3, 0)
-		: number_pattern(4, 0);
+		? number_pattern(1, 0)
+		: number_pattern(2, 0);
 	case 3:
 	default: return dim_mode
-		? number_pattern(5, 0)
-		: number_pattern(6, 0);
+		? number_pattern(3, 0)
+		: number_pattern(4, 0);
 	}
 }
 
+pattern_t *laser_pattern_select()
+{
+	if (!dim_mode && ambient_zone <= 1)
+		return number_pattern(2, 1);
+	else
+		return NULL;
+}