application: Factor out the configure_push_buttons() function.
[gps-watch.git] / src / application / main.rs
1 /*
2  * Copyright (c) 2019 Tilman Sauerbeck (tilman at code-monkey de)
3  *
4  * Permission is hereby granted, free of charge, to any person obtaining
5  * a copy of this software and associated documentation files (the
6  * "Software"), to deal in the Software without restriction, including
7  * without limitation the rights to use, copy, modify, merge, publish,
8  * distribute, sublicense, and/or sell copies of the Software, and to
9  * permit persons to whom the Software is furnished to do so, subject to
10  * the following conditions:
11  *
12  * The above copyright notice and this permission notice shall be
13  * included in all copies or substantial portions of the Software.
14  *
15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
16  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
17  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
18  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
19  * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
20  * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
21  * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
22  */
23
24 #![no_std]
25 #![no_main]
26 #[link(name="libcommon.rlib")]
27
28 extern crate common;
29
30 mod uart0;
31
32 use common::buffer::Buffer;
33 use common::clock;
34 use common::systick;
35 use common::port;
36 use common::gpio;
37 use common::nvic;
38 use common::i2c;
39 use common::spi;
40 use common::uart;
41 use common::usb_serial;
42 use common::display;
43 use common::gps;
44 use common::screen;
45 use common::time::Time;
46
47 extern {
48     fn enable_interrupts();
49
50     static mut cdc_tx_buf: Buffer;
51 }
52
53 struct Timer {
54     state: u32,
55     delay_ms: u32,
56     last_update_ticks: u32,
57 }
58
59 impl Timer {
60     pub fn new(delay_ms: u32) -> Timer {
61         Timer {
62             state: 0,
63             delay_ms: delay_ms,
64             last_update_ticks: systick::now(),
65         }
66     }
67
68     pub fn update<F>(&mut self, func: F)
69         where F: FnOnce(u32) -> u32
70     {
71         if systick::has_timeout_ms(self.last_update_ticks, self.delay_ms) {
72             self.state = func(self.state);
73
74             self.last_update_ticks = systick::now();
75         }
76     }
77 }
78
79 fn reset_requested() -> bool {
80     (gpio::get(gpio::GPIOA) & (1 << 12)) == 0
81 }
82
83 #[inline(never)]
84 #[panic_handler]
85 fn panic(_info: &core::panic::PanicInfo) -> ! {
86     loop {
87         if reset_requested() {
88             nvic::system_reset();
89         }
90     }
91 }
92
93 fn configure_push_buttons() {
94     // Configure upper right push button.
95     gpio::set_direction(gpio::GPIOA, 1 << 12, gpio::Direction::Input);
96     port::set_af(port::PORTA, 12, 1);
97     port::set_pull(port::PORTA, 12, port::Pull::Up);
98 }
99
100 #[no_mangle]
101 pub unsafe extern "C" fn _start() -> ! {
102     clock::configure();
103     clock::enable_osc0();
104     systick::init();
105     port::init();
106
107     // Configure pins for I2C0.
108     port::set_af(port::PORTC, 8, 2);
109     port::set_af(port::PORTC, 9, 2);
110
111     i2c::configure(i2c::I2C0);
112
113     nvic::disable_irq(8); // I2C0
114
115     // Configure pin for the display's reset line.
116     gpio::set_direction(gpio::GPIOB, 1 << 16, gpio::Direction::Output);
117     port::set_af(port::PORTB, 16, 1);
118
119     // Configure pin for the MX25L's chip select line.
120     gpio::set_direction(gpio::GPIOD, 1 << 0, gpio::Direction::Output);
121     port::set_af(port::PORTD, 0, 1);
122     gpio::set(gpio::GPIOD, 1 << 0);
123
124     // Configure pins for SPI0.
125     port::set_af(port::PORTD, 1, 2);
126     port::set_af(port::PORTD, 2, 5);
127     port::set_af(port::PORTD, 3, 5);
128
129     spi::configure(spi::SPI0);
130
131     nvic::disable_irq(10); // SPI0
132
133     // Configure pins for UART0.
134     port::set_af(port::PORTE, 20, 4);
135     port::set_af(port::PORTE, 21, 4);
136
137     // Configure pin for the GPS's reset line.
138     gpio::set_direction(gpio::GPIOB, 1 << 1, gpio::Direction::Output);
139     port::set_af(port::PORTB, 1, 1);
140
141     configure_push_buttons();
142
143     enable_interrupts();
144
145     usb_serial::init(0xf055, 0x635d);
146
147     cdc_tx_buf.write(b"\n");
148     cdc_tx_buf.flush();
149
150     let mut display = display::Display::new(gpio::GPIOB, 1 << 16, 0x3c);
151
152     display.init();
153     display.clear();
154
155     let mut screen = screen::Screen::new();
156
157     // Hold GPS in reset while configuring its UART.
158     gpio::clear(gpio::GPIOB, 1);
159     systick::delay_ms(50);
160     uart::configure(uart::UART0);
161     systick::delay_ms(50);
162     gpio::set(gpio::GPIOB, 1);
163
164     nvic::enable_irq(12); // UART0
165
166     let mut gps = gps::Gps::new();
167
168     let mut gps_has_fix = false;
169     let mut gps_has_fix_ticks = 0;
170
171     let mut heart_icon_timer = Timer::new(1000);
172     let mut gps_icon_timer = Timer::new(500);
173
174     let mut prev_tap = gps::TimeAndPos::new();
175
176     loop {
177         let mut tap = gps::TimeAndPos::new();
178         let mut show_time = false;
179         let old_gps_has_fix = gps_has_fix;
180
181         while gps.update(&mut tap) {
182             prev_tap = tap;
183
184             show_time = true;
185
186             gps_has_fix = true;
187             gps_has_fix_ticks = systick::now();
188         }
189
190         // Did GPS fix information expire?
191         if gps_has_fix && systick::has_timeout_ms(gps_has_fix_ticks, 1500) {
192             gps_has_fix = false;
193         }
194
195         if gps_has_fix && !old_gps_has_fix {
196             display.show_icon(display::Icon::SatelliteBody);
197             display.show_icon(display::Icon::SatelliteWave1);
198             display.show_icon(display::Icon::SatelliteWave2);
199         } else if !gps_has_fix {
200             gps_icon_timer.update(|state| {
201                 if state == 1 {
202                     display.show_icon(display::Icon::SatelliteWave1);
203                     2
204                 } else if state == 2 {
205                     display.show_icon(display::Icon::SatelliteWave2);
206                     3
207                 } else if state == 3 {
208                     display.hide_icon(display::Icon::SatelliteBody);
209                     display.hide_icon(display::Icon::SatelliteWave1);
210                     display.hide_icon(display::Icon::SatelliteWave2);
211                     0
212                 } else {
213                     display.show_icon(display::Icon::SatelliteBody);
214                     1
215                 }
216             });
217         }
218
219         if show_time {
220             if let Some(tm) = Time::from_unix_time(prev_tap.unix_time) {
221                 let mut time_s = [b' '; 8];
222                 tm.fmt_time(&mut time_s);
223
224                 screen.clear();
225                 screen.draw_text(&time_s);
226
227                 display.draw(&screen);
228             }
229         }
230
231         heart_icon_timer.update(|state| {
232             if state == 1 {
233                 display.hide_icon(display::Icon::Heart);
234                 0
235             } else {
236                 display.show_icon(display::Icon::Heart);
237                 1
238             }
239         });
240
241         if reset_requested() {
242             nvic::system_reset();
243         }
244     }
245 }