common: Store distance travelled in Logger instance.

[gps-watch.git] / src / application / main.rs

diff --git a/src/application/main.rs b/src/application/main.rs
index 8f31ff8015ac29e30c8d4130e5dfe381e9e1809d..dc955ce7af46fb39fc5b37721e4b12c5b7cbcbd3 100644 (file)
--- a/src/application/main.rs
+++ b/src/application/main.rs
@@ -28,8 +28,10 @@
 extern crate common;
 
 mod uart0;
+mod button;
 
 use common::buffer::Buffer;
+use common::ringbuf::Ringbuf;
 use common::clock;
 use common::systick;
 use common::port;
@@ -43,6 +45,9 @@ use common::display;
 use common::gps;
 use common::screen;
 use common::time::Time;
+use common::mx25l::Mx25l;
+use common::shell::Shell;
+use common::logger::Logger;
 
 extern {
     fn enable_interrupts();
@@ -77,7 +82,10 @@ impl Timer {
 }
 
 fn reset_requested() -> bool {
-    (gpio::get(gpio::GPIOA) & (1 << 12)) == 0
+    let pta1 = gpio::get(gpio::GPIOA) & (1 << 1);
+    let pte25 = gpio::get(gpio::GPIOE) & (1 << 25);
+
+    (pta1 | pte25) == 0
 }
 
 #[inline(never)]
@@ -91,10 +99,44 @@ fn panic(_info: &core::panic::PanicInfo) -> ! {
 }
 
 fn configure_push_buttons() {
+    // Configure lower right push button.
+    gpio::set_direction(gpio::GPIOA, 1 << 1, gpio::Direction::Input);
+    port::set_af(port::PORTA, 1, 1);
+    port::set_pull(port::PORTA, 1, port::Pull::Up);
+
     // Configure upper right push button.
     gpio::set_direction(gpio::GPIOA, 1 << 12, gpio::Direction::Input);
     port::set_af(port::PORTA, 12, 1);
     port::set_pull(port::PORTA, 12, port::Pull::Up);
+
+    // Configure lower left push button.
+    gpio::set_direction(gpio::GPIOE, 1 << 24, gpio::Direction::Input);
+    port::set_af(port::PORTE, 24, 1);
+    port::set_pull(port::PORTE, 24, port::Pull::Up);
+
+    // Configure upper left push button.
+    gpio::set_direction(gpio::GPIOE, 1 << 25, gpio::Direction::Input);
+    port::set_af(port::PORTE, 25, 1);
+    port::set_pull(port::PORTE, 25, port::Pull::Up);
+
+    // Configure middle left push button.
+    gpio::set_direction(gpio::GPIOE, 1 << 31, gpio::Direction::Input);
+    port::set_af(port::PORTE, 31, 1);
+    port::set_pull(port::PORTE, 31, port::Pull::Up);
+}
+
+fn uart0_try_read() -> Option<u8> {
+    extern {
+        static mut uart0_rx_buf: Ringbuf;
+    }
+
+    unsafe {
+        if uart0_rx_buf.is_empty() {
+            None
+        } else {
+            Some(uart0_rx_buf.read())
+        }
+    }
 }
 
 #[no_mangle]
@@ -163,6 +205,16 @@ pub unsafe extern "C" fn _start() -> ! {
 
     nvic::enable_irq(12); // UART0
 
+    let mut shell = Shell::new(&mut cdc_tx_buf);
+
+    let mut mx25l = Mx25l::new(gpio::GPIOD, 1 << 0);
+
+    let mut logger = Logger::new(&mut mx25l);
+    logger.init();
+
+    let mut pta12 = button::Button::new(gpio::GPIOA, 1 << 12);
+    let mut is_recording = false;
+
     let mut gps = gps::Gps::new();
 
     let mut gps_has_fix = false;
@@ -176,9 +228,16 @@ pub unsafe extern "C" fn _start() -> ! {
     loop {
         let mut tap = gps::TimeAndPos::new();
         let mut show_time = false;
+        let mut show_distance = false;
         let old_gps_has_fix = gps_has_fix;
 
-        while gps.update(&mut tap) {
+        while gps.update(&mut tap, uart0_try_read) {
+            if is_recording {
+                logger.log(&prev_tap, &tap);
+
+                show_distance = true;
+            }
+
             prev_tap = tap;
 
             show_time = true;
@@ -216,7 +275,17 @@ pub unsafe extern "C" fn _start() -> ! {
             });
         }
 
-        if show_time {
+        if show_distance {
+            let mut distance_m_s = [b' '; 8];
+
+            common::fmt::fmt_u32(&mut distance_m_s,
+                                 logger.total_distance_cm / 100);
+
+            screen.clear();
+            screen.draw_text(&distance_m_s);
+
+            display.draw(&screen);
+        } else if show_time {
             if let Some(tm) = Time::from_unix_time(prev_tap.unix_time) {
                 let mut time_s = [b' '; 8];
                 tm.fmt_time(&mut time_s);
@@ -238,6 +307,18 @@ pub unsafe extern "C" fn _start() -> ! {
             }
         });
 
+        shell.update(&mut logger);
+
+        if pta12.has_been_held_for_ms(1500) {
+            is_recording = !is_recording;
+
+            if is_recording {
+                logger.start_recording(&prev_tap);
+            } else {
+                logger.stop_recording(&prev_tap);
+            }
+        }
+
         if reset_requested() {
             nvic::system_reset();
         }