Unverified Commit 42e9b67f authored by Birte Kristina Friesel's avatar Birte Kristina Friesel
Browse files

Allow setting mot bitmasks directly

parent b990b22a

lib/Travel/Routing/DE/HAFAS.pm

+12 −2
Original line number Diff line number Diff line
@@ -237,14 +237,24 @@ sub mot_mask { 
	if ( my @mots = @{ $self->{exclusive_mots} // [] } ) {

		$mot_mask = 0;

		for my $mot (@mots) {

			if ( exists $mot_pos{$mot} ) {

				$mot_mask |= 1 << $mot_pos{$mot};

			}

			elsif ( $mot =~ m{ ^ \d+ $ }x ) {

				$mot_mask |= 1 << $mot;

			}

		}

	}



	if ( my @mots = @{ $self->{excluded_mots} // [] } ) {

		for my $mot (@mots) {

			if ( exists $mot_pos{$mot} ) {

				$mot_mask &= ~( 1 << $mot_pos{$mot} );

			}

			elsif ( $mot =~ m{ ^ \d+ $ }x ) {

				$mot_mask &= ~( 1 << $mot );

			}

		}

	}



	return $mot_mask;