summaryrefslogtreecommitdiff
path: root/lib/Travel
diff options
context:
space:
mode:
authorDaniel Friesel <derf@finalrewind.org>2016-05-10 11:00:43 +0200
committerDaniel Friesel <derf@finalrewind.org>2016-05-10 11:00:43 +0200
commite194d57f27bf322037ec75c1d6b9770eda677aa7 (patch)
tree9d928007f1f94c1030f8f02d808370129afc14a2 /lib/Travel
parentfca56c178d1e2c121152459813f2f300cf1f9010 (diff)
add get_stations_by_location method
Diffstat (limited to 'lib/Travel')
-rw-r--r--lib/Travel/Status/DE/IRIS/Stations.pm25
1 files changed, 25 insertions, 0 deletions
diff --git a/lib/Travel/Status/DE/IRIS/Stations.pm b/lib/Travel/Status/DE/IRIS/Stations.pm
index c9863ac..54245ca 100644
--- a/lib/Travel/Status/DE/IRIS/Stations.pm
+++ b/lib/Travel/Status/DE/IRIS/Stations.pm
@@ -5,11 +5,16 @@ use warnings;
use 5.014;
use utf8;
+use Geo::Distance;
use List::Util qw(min);
use List::UtilsBy qw(uniq_by);
use List::MoreUtils qw(firstval pairwise);
use Text::LevenshteinXS qw(distance);
+# TODO Geo::Distance is kinda deprecated, it is recommended to use GIS::Distance
+# instead. However, since GIS::Distance is not packaged for Debian, I'll stick
+# with Geo::Distance for now (which works fine enough here)
+
# TODO switch to Text::Levenshtein::XS once AUR/Debian packages become available
our $VERSION = '1.05';
@@ -8487,6 +8492,26 @@ sub get_station {
return get_station_by_name($name);
}
+sub get_stations_by_location {
+ my ($lon, $lat, $num_matches) = @_;
+
+ $num_matches //= 10;
+
+ my $geo = Geo::Distance->new();
+
+ # we only use geolocations inside germany.
+ # For these, this fast preprocessing step will let through all
+ # coordinates inside a 60km radius (and a bunch which are farther out)
+ my @candidates = grep { $#{$_} >= 4 and abs($_->[3] - $lon) < 1 and abs($_->[4] - $lat) < 1 } @stations;
+ my @distances = map { $geo->distance('kilometer', $lon, $lat, $_->[3], $_->[4]) } @candidates;
+ my @station_map = pairwise { [ $a, $b ] } @candidates, @distances;
+
+ @station_map = sort { $a->[1] <=> $b->[1] } @station_map;
+ splice(@station_map, 10);
+
+ return @station_map;
+}
+
sub get_station_by_name {
my ($name) = @_;