summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--lib/Travel/Status/DE/DBWagenreihung.pm33
1 files changed, 33 insertions, 0 deletions
diff --git a/lib/Travel/Status/DE/DBWagenreihung.pm b/lib/Travel/Status/DE/DBWagenreihung.pm
index e4f1d6c..2289383 100644
--- a/lib/Travel/Status/DE/DBWagenreihung.pm
+++ b/lib/Travel/Status/DE/DBWagenreihung.pm
@@ -265,6 +265,39 @@ sub train_no {
return $self->{data}{istformation}{zugnummer};
}
+sub train_powertype {
+ my ($self) = @_;
+
+ if ( exists $self->{train_powertype} ) {
+ return $self->{train_powertype};
+ }
+
+ my @wagons = $self->wagons;
+ my %ml = map { $_ => 0 } ( 90 .. 99 );
+
+ for my $wagon (@wagons) {
+
+ if ( not $wagon->uic_id or length( $wagon->uic_id ) != 12 ) {
+ next;
+ }
+
+ my $wagon_type = substr( $wagon->uic_id, 0, 2 );
+ if ( $wagon_type < 90 ) {
+ next;
+ }
+
+ $ml{$wagon_type}++;
+ }
+
+ my @likelihood = reverse sort { $ml{$a} <=> $ml{$b} } keys %ml;
+
+ if ( $ml{ $likelihood[0] } == 0 ) {
+ return $self->{train_powertype} = undef;
+ }
+
+ return $self->{train_powertype} = $likelihood[0];
+}
+
sub train_subtype {
my ($self) = @_;