application: Configure the remaining push buttons.
[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 lower right push button.
95     gpio::set_direction(gpio::GPIOA, 1 << 1, gpio::Direction::Input);
96     port::set_af(port::PORTA, 1, 1);
97     port::set_pull(port::PORTA, 1, port::Pull::Up);
98
99     // Configure upper right push button.
100     gpio::set_direction(gpio::GPIOA, 1 << 12, gpio::Direction::Input);
101     port::set_af(port::PORTA, 12, 1);
102     port::set_pull(port::PORTA, 12, port::Pull::Up);
103
104     // Configure lower left push button.
105     gpio::set_direction(gpio::GPIOE, 1 << 24, gpio::Direction::Input);
106     port::set_af(port::PORTE, 24, 1);
107     port::set_pull(port::PORTE, 24, port::Pull::Up);
108
109     // Configure upper left push button.
110     gpio::set_direction(gpio::GPIOE, 1 << 25, gpio::Direction::Input);
111     port::set_af(port::PORTE, 25, 1);
112     port::set_pull(port::PORTE, 25, port::Pull::Up);
113
114     // Configure middle left push button.
115     gpio::set_direction(gpio::GPIOE, 1 << 31, gpio::Direction::Input);
116     port::set_af(port::PORTE, 31, 1);
117     port::set_pull(port::PORTE, 31, port::Pull::Up);
118 }
119
120 #[no_mangle]
121 pub unsafe extern "C" fn _start() -> ! {
122     clock::configure();
123     clock::enable_osc0();
124     systick::init();
125     port::init();
126
127     // Configure pins for I2C0.
128     port::set_af(port::PORTC, 8, 2);
129     port::set_af(port::PORTC, 9, 2);
130
131     i2c::configure(i2c::I2C0);
132
133     nvic::disable_irq(8); // I2C0
134
135     // Configure pin for the display's reset line.
136     gpio::set_direction(gpio::GPIOB, 1 << 16, gpio::Direction::Output);
137     port::set_af(port::PORTB, 16, 1);
138
139     // Configure pin for the MX25L's chip select line.
140     gpio::set_direction(gpio::GPIOD, 1 << 0, gpio::Direction::Output);
141     port::set_af(port::PORTD, 0, 1);
142     gpio::set(gpio::GPIOD, 1 << 0);
143
144     // Configure pins for SPI0.
145     port::set_af(port::PORTD, 1, 2);
146     port::set_af(port::PORTD, 2, 5);
147     port::set_af(port::PORTD, 3, 5);
148
149     spi::configure(spi::SPI0);
150
151     nvic::disable_irq(10); // SPI0
152
153     // Configure pins for UART0.
154     port::set_af(port::PORTE, 20, 4);
155     port::set_af(port::PORTE, 21, 4);
156
157     // Configure pin for the GPS's reset line.
158     gpio::set_direction(gpio::GPIOB, 1 << 1, gpio::Direction::Output);
159     port::set_af(port::PORTB, 1, 1);
160
161     configure_push_buttons();
162
163     enable_interrupts();
164
165     usb_serial::init(0xf055, 0x635d);
166
167     cdc_tx_buf.write(b"\n");
168     cdc_tx_buf.flush();
169
170     let mut display = display::Display::new(gpio::GPIOB, 1 << 16, 0x3c);
171
172     display.init();
173     display.clear();
174
175     let mut screen = screen::Screen::new();
176
177     // Hold GPS in reset while configuring its UART.
178     gpio::clear(gpio::GPIOB, 1);
179     systick::delay_ms(50);
180     uart::configure(uart::UART0);
181     systick::delay_ms(50);
182     gpio::set(gpio::GPIOB, 1);
183
184     nvic::enable_irq(12); // UART0
185
186     let mut gps = gps::Gps::new();
187
188     let mut gps_has_fix = false;
189     let mut gps_has_fix_ticks = 0;
190
191     let mut heart_icon_timer = Timer::new(1000);
192     let mut gps_icon_timer = Timer::new(500);
193
194     let mut prev_tap = gps::TimeAndPos::new();
195
196     loop {
197         let mut tap = gps::TimeAndPos::new();
198         let mut show_time = false;
199         let old_gps_has_fix = gps_has_fix;
200
201         while gps.update(&mut tap) {
202             prev_tap = tap;
203
204             show_time = true;
205
206             gps_has_fix = true;
207             gps_has_fix_ticks = systick::now();
208         }
209
210         // Did GPS fix information expire?
211         if gps_has_fix && systick::has_timeout_ms(gps_has_fix_ticks, 1500) {
212             gps_has_fix = false;
213         }
214
215         if gps_has_fix && !old_gps_has_fix {
216             display.show_icon(display::Icon::SatelliteBody);
217             display.show_icon(display::Icon::SatelliteWave1);
218             display.show_icon(display::Icon::SatelliteWave2);
219         } else if !gps_has_fix {
220             gps_icon_timer.update(|state| {
221                 if state == 1 {
222                     display.show_icon(display::Icon::SatelliteWave1);
223                     2
224                 } else if state == 2 {
225                     display.show_icon(display::Icon::SatelliteWave2);
226                     3
227                 } else if state == 3 {
228                     display.hide_icon(display::Icon::SatelliteBody);
229                     display.hide_icon(display::Icon::SatelliteWave1);
230                     display.hide_icon(display::Icon::SatelliteWave2);
231                     0
232                 } else {
233                     display.show_icon(display::Icon::SatelliteBody);
234                     1
235                 }
236             });
237         }
238
239         if show_time {
240             if let Some(tm) = Time::from_unix_time(prev_tap.unix_time) {
241                 let mut time_s = [b' '; 8];
242                 tm.fmt_time(&mut time_s);
243
244                 screen.clear();
245                 screen.draw_text(&time_s);
246
247                 display.draw(&screen);
248             }
249         }
250
251         heart_icon_timer.update(|state| {
252             if state == 1 {
253                 display.hide_icon(display::Icon::Heart);
254                 0
255             } else {
256                 display.show_icon(display::Icon::Heart);
257                 1
258             }
259         });
260
261         if reset_requested() {
262             nvic::system_reset();
263         }
264     }
265 }