From 9ab221921ccb9628bd1e3c431f0c40faeb5162c7 Mon Sep 17 00:00:00 2001
From: Birte Kristina Friesel <derf@finalrewind.org>
Date: Wed, 3 Apr 2024 18:25:47 +0200
Subject: annotate train/line number changes along the route

---
 lib/DBInfoscreen/Controller/Stationboard.pm |  8 ++++
 lib/DBInfoscreen/Helper/HAFAS.pm            | 61 +++++++++++++++++++++++++++++
 2 files changed, 69 insertions(+)

(limited to 'lib')

diff --git a/lib/DBInfoscreen/Controller/Stationboard.pm b/lib/DBInfoscreen/Controller/Stationboard.pm
index dd13b82..b0f367e 100644
--- a/lib/DBInfoscreen/Controller/Stationboard.pm
+++ b/lib/DBInfoscreen/Controller/Stationboard.pm
@@ -880,6 +880,10 @@ sub render_train {
 					  = $route->[$i]{local_sched_arr};
 					$departure->{local_sched_dep}
 					  = $route->[$i]{local_sched_dep};
+					$departure->{is_annotated} = $route->[$i]{is_annotated};
+					$departure->{prod_name}    = $route->[$i]{prod_name};
+					$departure->{direction}    = $route->[$i]{direction};
+					$departure->{operator}     = $route->[$i]{operator};
 					last;
 				}
 			}
@@ -1253,6 +1257,10 @@ sub train_details {
 					$res->{local_dt_da}     = $station_info->{local_dt_da};
 					$res->{local_sched_arr} = $station_info->{local_sched_arr};
 					$res->{local_sched_dep} = $station_info->{local_sched_dep};
+					$res->{is_annotated}    = $station_info->{is_annotated};
+					$res->{prod_name}       = $station_info->{prod_name};
+					$res->{direction}       = $station_info->{direction};
+					$res->{operator}        = $station_info->{operator};
 					$res->{platform}        = $station_info->{platform};
 					$res->{scheduled_platform}
 					  = $station_info->{sched_platform};
diff --git a/lib/DBInfoscreen/Helper/HAFAS.pm b/lib/DBInfoscreen/Helper/HAFAS.pm
index 59d8c7b..cdb84f0 100644
--- a/lib/DBInfoscreen/Helper/HAFAS.pm
+++ b/lib/DBInfoscreen/Helper/HAFAS.pm
@@ -105,7 +105,67 @@ sub get_route_p {
 			my $journey = $hafas->result;
 			my @ret;
 			my $station_is_past = 1;
+
+			my $num_names      = 0;
+			my $prev_name      = q{};
+			my $num_directions = 0;
+			my $prev_direction = q{};
+			my $num_operators  = 0;
+			my $prev_operator  = q{};
+
+			for my $stop ( $journey->route ) {
+				my $prod = $stop->prod_dep // $stop->prod_arr;
+				if ( $prod and $prod->name and $prod->name ne $prev_name ) {
+					$num_names++;
+					$prev_name = $prod->name;
+				}
+				if (    $prod
+					and $prod->operator
+					and $prod->operator ne $prev_operator )
+				{
+					$num_operators++;
+					$prev_operator = $prod->operator;
+				}
+				if ( $stop->direction and $stop->direction ne $prev_direction )
+				{
+					$num_directions++;
+					$prev_direction = $stop->direction;
+				}
+			}
+
+			$prev_name      = q{};
+			$prev_direction = q{};
+			$prev_operator  = q{};
+
 			for my $stop ( $journey->route ) {
+
+				my $prod = $stop->prod_dep // $stop->prod_arr;
+				my %annotation;
+				if (    $num_names > 1
+					and $prod
+					and $prod->name
+					and $prod->name ne $prev_name )
+				{
+					$prev_name = $annotation{prod_name} = $prod->name;
+				}
+				if (    $num_operators > 1
+					and $prod
+					and $prod->operator
+					and $prod->operator ne $prev_operator )
+				{
+					$prev_operator = $annotation{operator} = $prod->operator;
+				}
+				if (    $num_directions > 1
+					and $stop->direction
+					and $stop->direction ne $prev_direction )
+				{
+					$prev_direction = $annotation{direction} = $stop->direction;
+				}
+
+				if (%annotation) {
+					$annotation{is_annotated} = 1;
+				}
+
 				push(
 					@ret,
 					{
@@ -129,6 +189,7 @@ sub get_route_p {
 							  and
 							  ( $stop->dep_cancelled or not $stop->sched_dep )
 						),
+						%annotation,
 					}
 				);
 				if (
-- 
cgit v1.2.3