Compare commits
5 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| e185b38222 | |||
| c13411065a | |||
| cef6faa05f | |||
| e758edd45a | |||
| 59c13dc46a |
+1
-1
@@ -14,7 +14,7 @@ zip = "8.3"
|
||||
csv = "1.4"
|
||||
|
||||
[profile.dev]
|
||||
opt-level = 1
|
||||
opt-level = 3
|
||||
|
||||
[profile.release]
|
||||
opt-level = 3
|
||||
|
||||
@@ -0,0 +1,132 @@
|
||||
use gtfs_structures::{Agency, Calendar, CalendarDate, Route, Stop};
|
||||
use serde::{de::DeserializeOwned};
|
||||
use std::{collections::HashMap, fs::File};
|
||||
use zip::ZipArchive;
|
||||
|
||||
// The main GTFS struct. This is similar to (but not exactly) gtfs-structures::Gtfs because we don't need everything
|
||||
#[derive(Debug)]
|
||||
pub struct Gtfs {
|
||||
/// All agencies. They can not be read by `agency_id`, as it is not a required field
|
||||
pub agencies: Vec<Agency>,
|
||||
/// All Calendar by `service_id`
|
||||
pub calendar: HashMap<String, Calendar>,
|
||||
/// All calendar dates grouped by service_id
|
||||
pub calendar_dates: HashMap<String, Vec<CalendarDate>>,
|
||||
/// All routes by `route_id`
|
||||
pub routes: HashMap<String, Route>,
|
||||
/// All stop by `stop_id`.
|
||||
pub stops: HashMap<String, Stop>,
|
||||
}
|
||||
|
||||
// Utility function to load all records in a dataset
|
||||
fn load_all<V>(_: &V) -> bool { true }
|
||||
|
||||
// Loads a vector of the selected type
|
||||
fn load_vector<T: serde::de::DeserializeOwned>(
|
||||
destination: &mut Vec<T>,
|
||||
zip_reader: &mut ZipArchive<File>,
|
||||
table_name: &str,
|
||||
) {
|
||||
let file_reader = zip_reader.by_name(table_name).unwrap();
|
||||
let mut rdr = csv::Reader::from_reader(file_reader);
|
||||
|
||||
for row in rdr.deserialize() {
|
||||
let record: T = row.unwrap();
|
||||
destination.push(record);
|
||||
}
|
||||
}
|
||||
|
||||
// Loads a HashMap of the selected type, using the provided index function as the key
|
||||
fn load_map<'a, V: DeserializeOwned>(
|
||||
destination: &mut HashMap<String, V>,
|
||||
zip_reader: &mut ZipArchive<File>,
|
||||
table_name: &str,
|
||||
index: fn(&V) -> String,
|
||||
accept: fn(&V) -> bool,
|
||||
) {
|
||||
let file_reader = zip_reader.by_name(table_name).unwrap();
|
||||
let mut rdr = csv::Reader::from_reader(file_reader);
|
||||
|
||||
for row in rdr.deserialize() {
|
||||
let record: V = row.unwrap();
|
||||
if accept(&record) {
|
||||
let idx: String = index(&record);
|
||||
destination.insert(idx, record);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Loads a HashMap of a vector of the selected type, using the provided index function as the key
|
||||
// And a predicate as a filter
|
||||
fn load_vector_map<'a, V: DeserializeOwned + Clone>(
|
||||
destination: &mut HashMap<String, Vec<V>>,
|
||||
zip_reader: &mut ZipArchive<File>,
|
||||
table_name: &str,
|
||||
index: fn(&V) -> String,
|
||||
accept: fn(&V) -> bool,
|
||||
) {
|
||||
let file_reader = zip_reader.by_name(table_name).unwrap();
|
||||
let mut rdr = csv::Reader::from_reader(file_reader);
|
||||
|
||||
for row in rdr.deserialize() {
|
||||
let record: V = row.unwrap();
|
||||
if accept(&record) {
|
||||
let idx: String = index(&record);
|
||||
destination.entry(idx).or_insert_with(Vec::new).push(record);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub fn init(src_file: &str, routes: Vec<String>) -> Gtfs {
|
||||
// Open zip file
|
||||
let mut zip_reader = zip::ZipArchive::new(File::open(src_file).unwrap()).unwrap();
|
||||
|
||||
let mut gtfs: Gtfs = Gtfs {
|
||||
agencies: Vec::new(),
|
||||
calendar: HashMap::new(),
|
||||
calendar_dates: HashMap::new(),
|
||||
routes: HashMap::new(),
|
||||
stops: HashMap::new(),
|
||||
};
|
||||
|
||||
// Agencies
|
||||
load_vector(&mut gtfs.agencies, &mut zip_reader, "agency.txt");
|
||||
|
||||
// Calendars
|
||||
load_map(
|
||||
&mut gtfs.calendar,
|
||||
&mut zip_reader,
|
||||
"calendar.txt",
|
||||
|c: &Calendar| String::from(&c.id),
|
||||
load_all
|
||||
);
|
||||
|
||||
// Calendar Dates
|
||||
load_vector_map(
|
||||
&mut gtfs.calendar_dates,
|
||||
&mut zip_reader,
|
||||
"calendar_dates.txt",
|
||||
|d: &CalendarDate| String::from(&d.service_id),
|
||||
load_all
|
||||
);
|
||||
|
||||
// Stops
|
||||
load_map(&mut gtfs.stops,
|
||||
&mut zip_reader,
|
||||
"stops.txt",
|
||||
|s: &Stop| {String::from(&s.id)},
|
||||
load_all
|
||||
);
|
||||
|
||||
// Routes
|
||||
let accept_filter: fn(&Route) -> bool = (| rs: Vec<String>, r: &Route | {rs.contains(&r.short_name.unwrap())}).curry(routes);
|
||||
load_map(
|
||||
&mut gtfs.routes,
|
||||
&mut zip_reader,
|
||||
"routes.txt",
|
||||
|r: &Route| String::from(&r.id),
|
||||
accept_filter
|
||||
);
|
||||
|
||||
return gtfs;
|
||||
}
|
||||
+33
-26
@@ -1,6 +1,6 @@
|
||||
use log::debug;
|
||||
use gtfs_structures::{Calendar, CalendarDate, RawStopTime, RawTrip, Route, Stop};
|
||||
use serde::de::DeserializeOwned;
|
||||
use serde::de::{DeserializeOwned};
|
||||
use std::{
|
||||
collections::{HashMap, HashSet},
|
||||
fs::File,
|
||||
@@ -14,14 +14,14 @@ use crate::gtfs::{
|
||||
};
|
||||
|
||||
trait Filter<T> {
|
||||
fn accept(&self, v: &T) -> bool;
|
||||
fn accept(&self, v: &T) -> Option<bool>;
|
||||
}
|
||||
|
||||
// No filter on loaded records
|
||||
struct LoadAll {}
|
||||
impl<T> Filter<T> for LoadAll {
|
||||
fn accept(&self, _: &T) -> bool {
|
||||
return true;
|
||||
fn accept(&self, _: &T) -> Option<bool> {
|
||||
return Some(true);
|
||||
}
|
||||
}
|
||||
|
||||
@@ -29,9 +29,9 @@ struct LoadRoutes<'a> {
|
||||
routes: &'a HashSet<String>,
|
||||
}
|
||||
impl Filter<Route> for LoadRoutes<'_> {
|
||||
fn accept(&self, r: &Route) -> bool {
|
||||
fn accept(&self, r: &Route) -> Option<bool> {
|
||||
let short_name = &r.short_name;
|
||||
return short_name.is_some() && self.routes.contains(short_name.as_ref().unwrap());
|
||||
return Some(short_name.is_some() && self.routes.contains(short_name.as_ref()?));
|
||||
}
|
||||
}
|
||||
|
||||
@@ -39,9 +39,9 @@ struct LoadStops<'a> {
|
||||
stops: &'a HashSet<String>,
|
||||
}
|
||||
impl Filter<Stop> for LoadStops<'_> {
|
||||
fn accept(&self, s: &Stop) -> bool {
|
||||
fn accept(&self, s: &Stop) -> Option<bool> {
|
||||
let stop_code = &s.code;
|
||||
return stop_code.is_some() && self.stops.contains(s.code.as_ref().unwrap());
|
||||
return Some(stop_code.is_some() && self.stops.contains(s.code.as_ref()?));
|
||||
}
|
||||
}
|
||||
|
||||
@@ -49,9 +49,9 @@ struct LoadTrips<'a> {
|
||||
route_ids: &'a HashSet<String>,
|
||||
}
|
||||
impl Filter<RawTrip> for LoadTrips<'_> {
|
||||
fn accept(&self, t: &RawTrip) -> bool {
|
||||
fn accept(&self, t: &RawTrip) -> Option<bool> {
|
||||
let route_id = &t.route_id;
|
||||
return self.route_ids.contains(route_id);
|
||||
return Some(self.route_ids.contains(route_id));
|
||||
}
|
||||
}
|
||||
|
||||
@@ -60,8 +60,8 @@ struct LoadStopTimes<'a> {
|
||||
stop_ids: &'a HashSet<String>,
|
||||
}
|
||||
impl Filter<RawStopTime> for LoadStopTimes<'_> {
|
||||
fn accept(&self, st: &RawStopTime) -> bool {
|
||||
return self.stop_ids.contains(&st.stop_id) && self.trip_ids.contains(&st.trip_id);
|
||||
fn accept(&self, st: &RawStopTime) -> Option<bool> {
|
||||
return Some(self.stop_ids.contains(&st.stop_id) && self.trip_ids.contains(&st.trip_id));
|
||||
}
|
||||
}
|
||||
|
||||
@@ -70,14 +70,15 @@ fn load_vector<T: serde::de::DeserializeOwned>(
|
||||
destination: &mut Vec<T>,
|
||||
zip_reader: &mut ZipArchive<File>,
|
||||
table_name: &str,
|
||||
) {
|
||||
let file_reader = zip_reader.by_name(table_name).unwrap();
|
||||
let mut rdr = csv::Reader::from_reader(file_reader);
|
||||
) -> Option<bool> {
|
||||
let file_reader = zip_reader.by_name(table_name);
|
||||
let mut rdr = csv::Reader::from_reader(file_reader.ok()?);
|
||||
|
||||
for row in rdr.deserialize() {
|
||||
let record: T = row.unwrap();
|
||||
let record: T = row.ok()?;
|
||||
destination.push(record);
|
||||
}
|
||||
return Some(true);
|
||||
}
|
||||
|
||||
// Loads a HashMap of the selected type, using the provided index function as the key
|
||||
@@ -87,27 +88,30 @@ fn load_map<K, V, IndexFn, FilterT>(
|
||||
table_name: &str,
|
||||
index: IndexFn,
|
||||
filter: FilterT,
|
||||
) where
|
||||
) -> Option<bool>
|
||||
where
|
||||
K: Eq + Hash,
|
||||
V: DeserializeOwned,
|
||||
IndexFn: Fn(&V) -> K,
|
||||
FilterT: Filter<V>,
|
||||
FilterT: Filter<V>
|
||||
{
|
||||
let file_reader = zip_reader.by_name(table_name).unwrap();
|
||||
let file_reader = (zip_reader.by_name(table_name)).ok()?;
|
||||
let mut rdr = csv::Reader::from_reader(file_reader);
|
||||
|
||||
for row in rdr.deserialize() {
|
||||
if row.is_ok() {
|
||||
let record: V = row.unwrap();
|
||||
if filter.accept(&record) {
|
||||
let record: V = row.ok()?;
|
||||
let accepted = filter.accept(&record);
|
||||
if accepted.is_some() && accepted? {
|
||||
let idx: K = index(&record);
|
||||
destination.insert(idx, record);
|
||||
}
|
||||
} else {
|
||||
print!("Row failed to deserialize row {:#?}", row.err());
|
||||
panic!();
|
||||
return None;
|
||||
}
|
||||
}
|
||||
return Some(true);
|
||||
}
|
||||
|
||||
// Loads a HashMap of a vector of the selected type, using the provided index function as the key
|
||||
@@ -118,22 +122,25 @@ fn load_vector_map<'a, K, V, IndexFn, FilterT>(
|
||||
table_name: &str,
|
||||
index: IndexFn,
|
||||
filter: FilterT,
|
||||
) where
|
||||
) -> Option<bool>
|
||||
where
|
||||
K: Eq + Hash,
|
||||
V: DeserializeOwned,
|
||||
IndexFn: Fn(&V) -> K,
|
||||
FilterT: Filter<V>,
|
||||
{
|
||||
let file_reader = zip_reader.by_name(table_name).unwrap();
|
||||
let file_reader = zip_reader.by_name(table_name).ok()?;
|
||||
let mut rdr = csv::Reader::from_reader(file_reader);
|
||||
|
||||
for row in rdr.deserialize() {
|
||||
let record: V = row.unwrap();
|
||||
if filter.accept(&record) {
|
||||
let record: V = row.ok()?;
|
||||
let accepted = filter.accept(&record);
|
||||
if accepted.is_some() && accepted? {
|
||||
let idx = index(&record);
|
||||
destination.entry(idx).or_insert_with(Vec::new).push(record);
|
||||
}
|
||||
}
|
||||
return Some(true)
|
||||
}
|
||||
|
||||
pub fn load_gtfs(
|
||||
|
||||
+15
-13
@@ -2,11 +2,11 @@ mod arrival;
|
||||
mod loader;
|
||||
mod utils;
|
||||
pub mod structs;
|
||||
use chrono::{DateTime, Local, NaiveTime, Timelike};
|
||||
use chrono::{DateTime, Local, Timelike};
|
||||
use log::{debug};
|
||||
use sdl3::sys::pixels::SDL_ArrayOrder;
|
||||
use std::{
|
||||
collections::{HashMap, HashSet},
|
||||
fs::File,
|
||||
collections::{HashMap, HashSet}, fs::File, io::Error
|
||||
};
|
||||
use gtfs_structures::{Exception, RawTrip};
|
||||
|
||||
@@ -15,7 +15,7 @@ use crate::gtfs::{loader::load_gtfs, structs::{Arrival, Gtfs, Preferences}};
|
||||
|
||||
impl Gtfs {
|
||||
|
||||
pub fn get_next_arrivals_for(&self, target_datetime: &DateTime<Local>) -> Vec<Arrival<'_>> {
|
||||
pub fn get_next_arrivals_for(&self, target_datetime: &DateTime<Local>) -> Option<Vec<Arrival<'_>>> {
|
||||
let naive_target = target_datetime.naive_local();
|
||||
let target_date = naive_target.date();
|
||||
|
||||
@@ -66,31 +66,33 @@ impl Gtfs {
|
||||
|
||||
for (_id, stop_time) in self.stop_times.iter() {
|
||||
if trips.contains_key(&stop_time.trip_id) {
|
||||
let stop_timestamp = stop_time.departure_time.or(stop_time.arrival_time).unwrap();
|
||||
debug!("Stop timestamp {} current timestamp {}", stop_timestamp, current_timestamp);
|
||||
let stop_timestamp = stop_time.departure_time.or(stop_time.arrival_time)?;
|
||||
let trip= &self.trips.get(&stop_time.trip_id).unwrap();
|
||||
if current_timestamp < stop_timestamp.into() {
|
||||
let arrival: Arrival = Arrival {
|
||||
route: self.routes.get(&trip.route_id).unwrap(),
|
||||
stop: self.stops.get(&stop_time.stop_id).unwrap(),
|
||||
route: self.routes.get(&self.trips.get(&stop_time.trip_id)?.route_id)?,
|
||||
stop: self.stops.get(&stop_time.stop_id)?,
|
||||
stop_time: stop_time,
|
||||
trip: &trip,
|
||||
departure_time: NaiveTime::from_num_seconds_from_midnight_opt(stop_timestamp, 0).unwrap()
|
||||
departure_time: stop_timestamp.into()
|
||||
};
|
||||
debug!("Arrival to {:#?} for trip ID {:#?}.", arrival.trip.trip_headsign.as_ref().unwrap(), arrival.trip.id);
|
||||
arrivals.push(arrival);
|
||||
}
|
||||
}
|
||||
}
|
||||
arrivals.sort();
|
||||
debug!("Found {} arrivals", arrivals.len());
|
||||
|
||||
return arrivals;
|
||||
return Some(arrivals);
|
||||
}
|
||||
|
||||
|
||||
/// Load a GTFS structure from a zip file
|
||||
pub fn load(src_file: &str, prefs: &Preferences) -> Gtfs {
|
||||
pub fn load(src_file: &str, prefs: &Preferences) -> Result<Gtfs, Error> {
|
||||
// Open zip file
|
||||
let mut zip_reader = zip::ZipArchive::new(File::open(src_file).unwrap()).unwrap();
|
||||
let zip_file = File::open(src_file)?;
|
||||
let mut zip_reader = zip::ZipArchive::new(zip_file)?;
|
||||
|
||||
let mut gtfs: Gtfs = Gtfs {
|
||||
agencies: Vec::new(),
|
||||
@@ -105,6 +107,6 @@ impl Gtfs {
|
||||
|
||||
load_gtfs(&mut gtfs, &mut zip_reader, &prefs.route_numbers, &prefs.stop_codes);
|
||||
|
||||
return gtfs;
|
||||
return Ok(gtfs);
|
||||
}
|
||||
}
|
||||
|
||||
+1
-2
@@ -1,6 +1,5 @@
|
||||
use std::collections::{HashMap, HashSet};
|
||||
|
||||
use chrono::NaiveTime;
|
||||
use gtfs_structures::{Agency, Calendar, CalendarDate, RawStopTime, RawTrip, Route, Stop};
|
||||
|
||||
// This is to store the preferences for the GTFS(-R) side of the code.
|
||||
@@ -32,7 +31,7 @@ pub struct Gtfs {
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct Arrival<'a> {
|
||||
pub departure_time: NaiveTime,
|
||||
pub departure_time: u32,
|
||||
pub route: &'a Route,
|
||||
pub stop: &'a Stop,
|
||||
pub stop_time: &'a RawStopTime,
|
||||
|
||||
+2
-2
@@ -1,7 +1,7 @@
|
||||
use std::collections::HashSet;
|
||||
use crate::gtfs::Gtfs;
|
||||
|
||||
pub fn stop_ids_from_codes(gtfs: &Gtfs, stop_codes: &HashSet<String>) -> HashSet<String> {
|
||||
pub(crate) fn stop_ids_from_codes(gtfs: &Gtfs, stop_codes: &HashSet<String>) -> HashSet<String> {
|
||||
let mut ids: HashSet<String> = HashSet::new();
|
||||
|
||||
for stop in >fs.stops {
|
||||
@@ -13,7 +13,7 @@ pub fn stop_ids_from_codes(gtfs: &Gtfs, stop_codes: &HashSet<String>) -> HashSet
|
||||
return ids;
|
||||
}
|
||||
|
||||
pub fn route_ids_from_numbers(gtfs: &Gtfs, route_numbers: &HashSet<String>) -> HashSet<String> {
|
||||
pub(crate) fn route_ids_from_numbers(gtfs: &Gtfs, route_numbers: &HashSet<String>) -> HashSet<String> {
|
||||
let mut ids: HashSet<String> = HashSet::new();
|
||||
|
||||
for route in >fs.routes {
|
||||
|
||||
+25
-15
@@ -1,13 +1,15 @@
|
||||
mod gtfs;
|
||||
mod renderer;
|
||||
use std::{collections::{HashSet, btree_map::Entry}, ops::Add, process, rc::Rc, thread::Builder, time::SystemTime};
|
||||
use std::{collections::HashSet, ops::Add, process, thread::Builder, time::SystemTime};
|
||||
use chrono::{DateTime, Duration, Local, NaiveTime};
|
||||
use log::{Metadata, Record, debug, error, info};
|
||||
use log::{Metadata, Record, error, info};
|
||||
use sdl3::event::Event;
|
||||
use crate::{gtfs::structs::{Arrival, Gtfs}, renderer::structs::{DisplayData, DisplayEntry, Screen}};
|
||||
|
||||
const SRC_FILE: &str = "/home/nahuel/Downloads/GTFS_Realtime.zip";
|
||||
const NUM_ARRIVALS: usize = 4;
|
||||
const UPDATE_INTERVAL_SECONDS: u64 = 62;
|
||||
|
||||
|
||||
// Custom Event to signal data refresh
|
||||
#[derive(Debug)]
|
||||
@@ -15,41 +17,41 @@ struct RefreshDataEvent {
|
||||
}
|
||||
|
||||
|
||||
fn refresh_schedule<'a>(gtfs: &'a Gtfs, screen: &mut Screen<'a>) -> Vec<Arrival<'a>> {
|
||||
fn refresh_schedule<'a>(gtfs: &'a Gtfs, screen : &mut Screen<'a>) -> Option<Vec<Arrival<'a>>> {
|
||||
let current_timestamp = SystemTime::now();
|
||||
let datetime: DateTime<Local> = current_timestamp.clone().into();
|
||||
let mut next_arrivals: Vec<Arrival<'_>> = gtfs.get_next_arrivals_for(&datetime);
|
||||
let mut next_arrivals: Vec<Arrival<'_>> = gtfs.get_next_arrivals_for(&datetime)?;
|
||||
|
||||
if next_arrivals.len() < NUM_ARRIVALS {
|
||||
// If we don't have enough entries today, look for arrivals tomorrow.
|
||||
let mut tomorrow: DateTime<Local> = datetime.clone();
|
||||
tomorrow = tomorrow.with_time(NaiveTime::from_hms_opt(0, 0, 0).unwrap()).unwrap();
|
||||
tomorrow = tomorrow.add(Duration::days(1));
|
||||
next_arrivals.append(&mut gtfs.get_next_arrivals_for(&tomorrow));
|
||||
next_arrivals.append(&mut gtfs.get_next_arrivals_for(&tomorrow)?);
|
||||
}
|
||||
|
||||
next_arrivals.sort();
|
||||
|
||||
// Create the DisplayData structure to render the information to screen
|
||||
let current_time = Local::now().time();
|
||||
let mut display_data: DisplayData = DisplayData {
|
||||
lines: Vec::<DisplayEntry>::new(),
|
||||
status: None
|
||||
};
|
||||
|
||||
display_data.lines.extend(next_arrivals.iter().map(|arrival| -> DisplayEntry {
|
||||
|
||||
DisplayEntry {
|
||||
destination: arrival.stop_time.stop_headsign.clone()
|
||||
.or(arrival.trip.trip_headsign.clone()
|
||||
.or(Option::Some(String::from("Unknown")
|
||||
))).unwrap(),
|
||||
route: arrival.route.short_name.clone().or(arrival.route.long_name.clone()).unwrap(),
|
||||
due_in: (arrival.departure_time - current_time).num_minutes().try_into().unwrap()
|
||||
departure_time: arrival.departure_time,
|
||||
}
|
||||
}));
|
||||
|
||||
screen.update_information(&display_data);
|
||||
return next_arrivals;
|
||||
return Some(next_arrivals);
|
||||
}
|
||||
|
||||
|
||||
@@ -71,13 +73,17 @@ fn main() {
|
||||
fn flush(&self) {}
|
||||
}
|
||||
|
||||
log::set_logger(&MY_LOGGER).unwrap();
|
||||
let logger = log::set_logger(&MY_LOGGER);
|
||||
if logger.is_err() {
|
||||
print!("Error setting up the main logger:{:#?}", logger.err());
|
||||
process::exit(-1);
|
||||
}
|
||||
log::set_max_level(log::LevelFilter::Trace);
|
||||
|
||||
// Create preferences structures from config
|
||||
let gtfs_prefs = gtfs::structs::Preferences {
|
||||
route_numbers: HashSet::from([String::from("15A"), String::from("F1"), String::from("F2"), String::from("F3")]),
|
||||
stop_codes: HashSet::from([String::from("1117")])
|
||||
stop_codes: HashSet::from([String::from("1114")])
|
||||
};
|
||||
|
||||
let screen_prefs = renderer::structs::Prefs {
|
||||
@@ -88,7 +94,13 @@ fn main() {
|
||||
|
||||
// Init GTFS static info
|
||||
info!("Loading GTFS data...");
|
||||
let gtfs = Gtfs::load(SRC_FILE, >fs_prefs);
|
||||
let res = Gtfs::load(SRC_FILE, >fs_prefs);
|
||||
|
||||
if let Err(e) = res {
|
||||
error!("Error loading GTFS data: {}", e);
|
||||
process::exit(-1);
|
||||
}
|
||||
let gtfs = res.unwrap();
|
||||
|
||||
// Init screen
|
||||
info!("Initializing screen...");
|
||||
@@ -105,7 +117,7 @@ fn main() {
|
||||
.name("updater".to_string())
|
||||
.spawn(move || {
|
||||
loop {
|
||||
std::thread::sleep(std::time::Duration::new(60,0));
|
||||
std::thread::sleep(std::time::Duration::new(UPDATE_INTERVAL_SECONDS,0));
|
||||
let event = RefreshDataEvent {};
|
||||
let send_result = event_sender.push_custom_event(event);
|
||||
if send_result.is_err() {
|
||||
@@ -133,9 +145,7 @@ fn main() {
|
||||
// Is the custom event a Refresh Data event?
|
||||
let refresh_data = event.as_user_event_type::<RefreshDataEvent>();
|
||||
if refresh_data.is_some() {
|
||||
debug!("Received user event: {:#?}", refresh_data.unwrap());
|
||||
let _data: Vec<Arrival<'_>> = refresh_schedule(>fs, &mut screen);
|
||||
debug!("-------------------------------- Refresh done.");
|
||||
let _data: Option<Vec<Arrival<'_>>> = refresh_schedule(>fs, &mut screen);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
+61
-29
@@ -1,20 +1,18 @@
|
||||
|
||||
pub mod structs;
|
||||
use std::cmp::min;
|
||||
|
||||
use std::{cmp::min};
|
||||
use log::{error, warn};
|
||||
use sdl3::{Sdl, pixels::Color, rect::Rect};
|
||||
use structs::{Prefs, DisplayData};
|
||||
|
||||
use crate::renderer::structs::Screen;
|
||||
|
||||
const LINE_COUNT: i32 = 6;
|
||||
const LINE_COUNT: i32 = 5;
|
||||
const COLOR_LCD_AMBER : Color = Color::RGB(0xf4, 0xcb, 0x60);
|
||||
const COLOR_LCD_GREEN : Color = Color::RGB(0xb3, 0xff, 0x00);
|
||||
const COLOR_LCD_RED : Color = Color::RGB(0xff, 0x3a, 0x4a);
|
||||
|
||||
//const COLOR_BACKGROUND = pygame.Color(0, 0, 0)
|
||||
const UPDATE_INTERVAL_SECONDS: u32 = 62;
|
||||
const TEXT_SIZE: u32 = 160; // Size of the font in pixels
|
||||
const COLOR_BACKGROUND : Color = Color::RGB(0x0, 0x0, 0x0 );
|
||||
const COLOR_TEXT_BG : Color = Color::RGBA(0x0, 0x0, 0x0, 0x0);
|
||||
const TEXT_SIZE: f32 = 160.0; // Size of the font in pixels
|
||||
|
||||
// Offsets of each part within a line
|
||||
const XOFFSET_ROUTE: u32 = 24;
|
||||
@@ -22,7 +20,6 @@ const XOFFSET_DESTINATION: u32 = 300;
|
||||
const XOFFSEET_TIME_LEFT: u32 = 1606;
|
||||
const INTER_LINE_OVERLAP: u32 = 15;
|
||||
|
||||
|
||||
impl Screen<'_> {
|
||||
|
||||
pub fn get_context(&self) -> &Sdl {
|
||||
@@ -35,45 +32,80 @@ impl Screen<'_> {
|
||||
return COLOR_LCD_RED;
|
||||
}
|
||||
|
||||
fn format_due_for(&self, due_in: i32, departure_time: u32) -> String {
|
||||
if due_in < 60 {
|
||||
return String::from("due");
|
||||
}
|
||||
if due_in < 3600 {
|
||||
return ((due_in / 60) as i32).to_string() + "min";
|
||||
}
|
||||
return format!("{:02}:{:02}", (departure_time / 3600) as i32, ((departure_time % 3600) / 60) as i32);
|
||||
}
|
||||
|
||||
|
||||
pub fn update_information(&mut self, display_data: &DisplayData) {
|
||||
self.do_clear();
|
||||
let num_arrivals: i32 = min(if display_data.status.is_some() {LINE_COUNT - 1} else {LINE_COUNT}, display_data.lines.len().try_into().unwrap());
|
||||
for line in 0..num_arrivals {
|
||||
|
||||
// Print status first
|
||||
if display_data.status.is_some() {
|
||||
// If the update has some information text, show it
|
||||
self.do_print_at(5, display_data.status.as_ref().unwrap(), 0);
|
||||
} else {
|
||||
// Display date and time otherwise
|
||||
self.do_print_at(5, &format!("TODO: DATE/TIME GOES HERE").to_string(), 0);
|
||||
}
|
||||
|
||||
// Then data lines from the bottom up
|
||||
let num_arrivals: i32 = min(LINE_COUNT, display_data.lines.len() as i32);
|
||||
for index in 0..num_arrivals {
|
||||
let line: u32 = ((num_arrivals - 1) - index) as u32;
|
||||
// Compose a line of text with all the information
|
||||
let entry = display_data.lines.get(line as usize).unwrap();
|
||||
let line: u32 = line.try_into().unwrap();
|
||||
let due_in_mins = (entry.due_in / 60) as i32;
|
||||
let arrival_color: Color = self.color_for(due_in_mins);
|
||||
let due_in_mins = (entry.departure_time / 60) as i32;
|
||||
let due_color: Color = self.color_for(due_in_mins);
|
||||
let due_text = self.format_due_for(due_in_mins, entry.departure_time);
|
||||
|
||||
self.color = COLOR_LCD_AMBER;
|
||||
self.do_print_at(line, &entry.route, XOFFSET_ROUTE);
|
||||
self.do_print_at(line, &entry.destination, XOFFSET_DESTINATION);
|
||||
self.color = arrival_color;
|
||||
self.do_print_at(line, &due_in_mins.to_string(), XOFFSEET_TIME_LEFT);
|
||||
self.color = due_color;
|
||||
self.do_print_at(line, &due_text, XOFFSEET_TIME_LEFT);
|
||||
};
|
||||
|
||||
if display_data.status.is_some() {
|
||||
self.do_print_at(5, display_data.status.as_ref().unwrap(), 0);
|
||||
}
|
||||
self.do_update();
|
||||
}
|
||||
|
||||
|
||||
pub fn print(&mut self, line: u32, text: &str) {
|
||||
pub fn _print(&mut self, line: u32, text: &str) {
|
||||
self.do_print_at(line, text, 0);
|
||||
self.do_update();
|
||||
}
|
||||
|
||||
|
||||
fn do_print_at(&mut self, line: u32, text: &str, left: u32) -> u32 {
|
||||
let rendered_text = self.font.render(text).solid(self.color).unwrap();
|
||||
fn do_print_at(&mut self, line: u32, text: &str, left: u32) {
|
||||
if text.len() == 0 {
|
||||
warn!("do_print_at called with a 0-length string");
|
||||
return;
|
||||
}
|
||||
|
||||
let render_result = self.font.render(text).lcd(self.color, COLOR_BACKGROUND);
|
||||
|
||||
if render_result.is_err() {
|
||||
error!("Error rendering text \"{}\": {:#?}", text, render_result.err());
|
||||
return;
|
||||
}
|
||||
|
||||
let rendered_text = render_result.unwrap();
|
||||
let texture_creator = self.canvas.texture_creator();
|
||||
let texture = rendered_text.as_texture(&texture_creator).unwrap();
|
||||
let _= self.canvas.copy(&texture,
|
||||
let texture = rendered_text.as_texture(&texture_creator);
|
||||
|
||||
if texture.is_err() {
|
||||
error!("Error creating texture from rendered text: {:#?}", texture.err());
|
||||
return;
|
||||
}
|
||||
|
||||
let _= self.canvas.copy(&texture.unwrap(),
|
||||
Rect::new(0, 0, rendered_text.width(), rendered_text.height()),
|
||||
Rect::new(left.try_into().unwrap(), (line * (rendered_text.height() - INTER_LINE_OVERLAP)).try_into().unwrap(), rendered_text.width(), rendered_text.height()));
|
||||
return left + rendered_text.width();
|
||||
}
|
||||
|
||||
|
||||
@@ -89,7 +121,7 @@ impl Screen<'_> {
|
||||
|
||||
|
||||
fn do_clear(&mut self) {
|
||||
self.canvas.set_draw_color(Color::BLACK);
|
||||
self.canvas.set_draw_color(COLOR_BACKGROUND);
|
||||
self.canvas.clear();
|
||||
}
|
||||
|
||||
@@ -109,8 +141,8 @@ impl Screen<'_> {
|
||||
|
||||
// Load font
|
||||
let ttf_context = sdl3::ttf::init().unwrap();
|
||||
let font = ttf_context.load_font(&prefs.font_path, 128.0).unwrap();
|
||||
|
||||
let font = ttf_context.load_font(&prefs.font_path, TEXT_SIZE).unwrap();
|
||||
|
||||
let mut screen: Screen = Screen {
|
||||
canvas: Box::new(window.into_canvas()),
|
||||
font: Box::new(font),
|
||||
|
||||
@@ -18,7 +18,7 @@ pub struct Prefs {
|
||||
pub struct DisplayEntry {
|
||||
pub route: String,
|
||||
pub destination: String,
|
||||
pub due_in: i32,
|
||||
pub departure_time: u32,
|
||||
}
|
||||
|
||||
pub struct DisplayData {
|
||||
|
||||
Reference in New Issue
Block a user