All Downloads are FREE. Search and download functionalities are using the official Maven repository.

de.schildbach.pte.AbstractHafasLegacyProvider Maven / Gradle / Ivy

There is a newer version: 2
Show newest version
/*
 * Copyright the original author or authors.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see .
 */

package de.schildbach.pte;

import static com.google.common.base.Preconditions.checkNotNull;
import static com.google.common.base.Preconditions.checkState;

import java.io.BufferedInputStream;
import java.io.ByteArrayInputStream;
import java.io.DataInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.Collections;
import java.util.Date;
import java.util.GregorianCalendar;
import java.util.HashMap;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.atomic.AtomicReference;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.util.zip.GZIPInputStream;

import javax.annotation.Nullable;

import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;
import org.xmlpull.v1.XmlPullParser;
import org.xmlpull.v1.XmlPullParserException;
import org.xmlpull.v1.XmlPullParserFactory;

import com.google.common.base.Charsets;
import com.google.common.base.Strings;

import de.schildbach.pte.dto.Departure;
import de.schildbach.pte.dto.Line;
import de.schildbach.pte.dto.Line.Attr;
import de.schildbach.pte.dto.Location;
import de.schildbach.pte.dto.LocationType;
import de.schildbach.pte.dto.NearbyLocationsResult;
import de.schildbach.pte.dto.Point;
import de.schildbach.pte.dto.Position;
import de.schildbach.pte.dto.Product;
import de.schildbach.pte.dto.QueryDeparturesResult;
import de.schildbach.pte.dto.QueryTripsContext;
import de.schildbach.pte.dto.QueryTripsResult;
import de.schildbach.pte.dto.ResultHeader;
import de.schildbach.pte.dto.StationDepartures;
import de.schildbach.pte.dto.Stop;
import de.schildbach.pte.dto.SuggestLocationsResult;
import de.schildbach.pte.dto.SuggestedLocation;
import de.schildbach.pte.dto.Trip;
import de.schildbach.pte.dto.TripOptions;
import de.schildbach.pte.exception.ParserException;
import de.schildbach.pte.exception.SessionExpiredException;
import de.schildbach.pte.util.HttpClient;
import de.schildbach.pte.util.LittleEndianDataInputStream;
import de.schildbach.pte.util.ParserUtils;
import de.schildbach.pte.util.StringReplaceReader;
import de.schildbach.pte.util.XmlPullUtil;

import okhttp3.HttpUrl;
import okhttp3.ResponseBody;

/**
 * @author Andreas Schildbach
 */
public abstract class AbstractHafasLegacyProvider extends AbstractHafasProvider {
    private static final String REQC_PROD = "hafas";

    protected HttpUrl stationBoardEndpoint;
    protected HttpUrl getStopEndpoint;
    protected HttpUrl queryEndpoint;
    private @Nullable HttpUrl extXmlEndpoint = null;
    protected final String apiLanguage;
    private @Nullable String accessId = null;
    private @Nullable String clientType = "ANDROID";
    private boolean jsonGetStopsUseWeight = true;
    private Charset jsonNearbyLocationsEncoding = Charsets.ISO_8859_1;
    private boolean dominantPlanStopTime = false;
    private boolean useIso8601 = false;
    private boolean stationBoardHasStationTable = true;
    private boolean stationBoardHasLocation = false;
    private boolean stationBoardCanDoEquivs = true;

    @SuppressWarnings("serial")
    private static class Context implements QueryTripsContext {
        public final String laterContext;
        public final String earlierContext;
        public final int sequence;

        public Context(final String laterContext, final String earlierContext, final int sequence) {
            this.laterContext = laterContext;
            this.earlierContext = earlierContext;
            this.sequence = sequence;
        }

        @Override
        public boolean canQueryLater() {
            return laterContext != null;
        }

        @Override
        public boolean canQueryEarlier() {
            return earlierContext != null;
        }
    }

    @SuppressWarnings("serial")
    public static class QueryTripsBinaryContext implements QueryTripsContext {
        public final String ident;
        public final int seqNr;
        public final String ld;
        public final int usedBufferSize;
        private final boolean canQueryMore;

        public QueryTripsBinaryContext(final String ident, final int seqNr, final String ld, final int usedBufferSize,
                final boolean canQueryMore) {
            this.ident = ident;
            this.seqNr = seqNr;
            this.ld = ld;
            this.usedBufferSize = usedBufferSize;
            this.canQueryMore = canQueryMore;
        }

        @Override
        public boolean canQueryLater() {
            return canQueryMore;
        }

        @Override
        public boolean canQueryEarlier() {
            return canQueryMore;
        }
    }

    public AbstractHafasLegacyProvider(final NetworkId network, final HttpUrl apiBase, final String apiLanguage,
            final Product[] productsMap) {
        super(network, productsMap);
        this.stationBoardEndpoint = apiBase.newBuilder().addPathSegment("stboard.exe").build();
        this.getStopEndpoint = apiBase.newBuilder().addPathSegment("ajax-getstop.exe").build();
        this.queryEndpoint = apiBase.newBuilder().addPathSegment("query.exe").build();
        this.apiLanguage = apiLanguage;
    }

    protected AbstractHafasProvider setStationBoardEndpoint(final HttpUrl stationBoardEndpoint) {
        this.stationBoardEndpoint = stationBoardEndpoint;
        return this;
    }

    protected AbstractHafasProvider setGetStopEndpoint(final HttpUrl getStopEndpoint) {
        this.getStopEndpoint = getStopEndpoint;
        return this;
    }

    protected AbstractHafasProvider setQueryEndpoint(final HttpUrl queryEndpoint) {
        this.queryEndpoint = queryEndpoint;
        return this;
    }

    protected AbstractHafasProvider setExtXmlEndpoint(final HttpUrl extXmlEndpoint) {
        this.extXmlEndpoint = extXmlEndpoint;
        return this;
    }

    protected AbstractHafasProvider setAccessId(final String accessId) {
        this.accessId = accessId;
        return this;
    }

    protected AbstractHafasProvider setClientType(final String clientType) {
        this.clientType = clientType;
        return this;
    }

    protected AbstractHafasProvider setDominantPlanStopTime(final boolean dominantPlanStopTime) {
        this.dominantPlanStopTime = dominantPlanStopTime;
        return this;
    }

    protected AbstractHafasProvider setJsonGetStopsUseWeight(final boolean jsonGetStopsUseWeight) {
        this.jsonGetStopsUseWeight = jsonGetStopsUseWeight;
        return this;
    }

    protected AbstractHafasProvider setJsonNearbyLocationsEncoding(final Charset jsonNearbyLocationsEncoding) {
        this.jsonNearbyLocationsEncoding = jsonNearbyLocationsEncoding;
        return this;
    }

    protected AbstractHafasProvider setUseIso8601(final boolean useIso8601) {
        this.useIso8601 = useIso8601;
        return this;
    }

    protected AbstractHafasProvider setStationBoardHasStationTable(final boolean stationBoardHasStationTable) {
        this.stationBoardHasStationTable = stationBoardHasStationTable;
        return this;
    }

    protected AbstractHafasProvider setStationBoardHasLocation(final boolean stationBoardHasLocation) {
        this.stationBoardHasLocation = stationBoardHasLocation;
        return this;
    }

    protected AbstractHafasProvider setStationBoardCanDoEquivs(final boolean canDoEquivs) {
        this.stationBoardCanDoEquivs = canDoEquivs;
        return this;
    }

    private final String wrapReqC(final CharSequence request, final Charset encoding) {
        return "" //
                + "" //
                + request //
                + "";
    }

    private final Location parseStation(final XmlPullParser pp) {
        final String type = pp.getName();
        if ("Station".equals(type)) {
            final String name = XmlPullUtil.attr(pp, "name");
            final String id = XmlPullUtil.attr(pp, "externalStationNr");
            final int x = XmlPullUtil.intAttr(pp, "x");
            final int y = XmlPullUtil.intAttr(pp, "y");
            final Point coord = Point.from1E6(y, x);
            final String[] placeAndName = splitStationName(name);
            return new Location(LocationType.STATION, id, coord, placeAndName[0], placeAndName[1]);
        }
        throw new IllegalStateException("cannot handle: " + type);
    }

    private static final Location parsePoi(final XmlPullParser pp) {
        final String type = pp.getName();
        if ("Poi".equals(type)) {
            String name = XmlPullUtil.attr(pp, "name");
            if (name.equals("unknown"))
                name = null;
            final int x = XmlPullUtil.intAttr(pp, "x");
            final int y = XmlPullUtil.intAttr(pp, "y");
            final Point coord = Point.from1E6(y, x);
            return new Location(LocationType.POI, null, coord, null, name);
        }
        throw new IllegalStateException("cannot handle: " + type);
    }

    private final Location parseAddress(final XmlPullParser pp) {
        final String type = pp.getName();
        if ("Address".equals(type)) {
            String name = XmlPullUtil.attr(pp, "name");
            if (name.equals("unknown"))
                name = null;
            final int x = XmlPullUtil.intAttr(pp, "x");
            final int y = XmlPullUtil.intAttr(pp, "y");
            final Point coord = Point.from1E6(y, x);
            final String[] placeAndName = splitAddress(name);
            return new Location(LocationType.ADDRESS, null, coord, placeAndName[0], placeAndName[1]);
        }
        throw new IllegalStateException("cannot handle: " + type);
    }

    private final Position parsePlatform(final XmlPullParser pp) throws XmlPullParserException, IOException {
        XmlPullUtil.enter(pp, "Platform");
        final String platformText = XmlPullUtil.valueTag(pp, "Text");
        XmlPullUtil.skipExit(pp, "Platform");

        if (platformText == null || platformText.length() == 0)
            return null;
        else
            return parsePosition(platformText);
    }

    @Override
    public SuggestLocationsResult suggestLocations(final CharSequence constraint) throws IOException {
        final HttpUrl.Builder url = getStopEndpoint.newBuilder().addPathSegment(apiLanguage);
        appendJsonGetStopsParameters(url, checkNotNull(constraint), 0);
        return jsonGetStops(url.build());
    }

    protected void appendJsonGetStopsParameters(final HttpUrl.Builder url, final CharSequence constraint,
            final int maxStops) {
        url.addQueryParameter("getstop", "1");
        url.addQueryParameter("REQ0JourneyStopsS0A", "255");
        url.addEncodedQueryParameter("REQ0JourneyStopsS0G",
                ParserUtils.urlEncode(constraint.toString() + "?", requestUrlEncoding));
        if (maxStops > 0)
            url.addQueryParameter("REQ0JourneyStopsB", Integer.toString(maxStops));
        url.addQueryParameter("js", "true");
    }

    private static final Pattern P_AJAX_GET_STOPS_JSON = Pattern
            .compile("SLs\\.sls\\s*=\\s*(.*?);\\s*SLs\\.showSuggestion\\(\\);", Pattern.DOTALL);
    private static final Pattern P_AJAX_GET_STOPS_ID = Pattern.compile(".*?@L=0*(\\d+)@.*?");

    protected final SuggestLocationsResult jsonGetStops(final HttpUrl url) throws IOException {
        final CharSequence page = httpClient.get(url);

        final Matcher mJson = P_AJAX_GET_STOPS_JSON.matcher(page);
        if (mJson.matches()) {
            final String json = mJson.group(1);
            final List locations = new ArrayList<>();

            try {
                final JSONObject head = new JSONObject(json);
                final JSONArray aSuggestions = head.getJSONArray("suggestions");

                for (int i = 0; i < aSuggestions.length(); i++) {
                    final JSONObject suggestion = aSuggestions.optJSONObject(i);
                    if (suggestion != null) {
                        final int type = suggestion.getInt("type");
                        final String value = suggestion.getString("value");
                        final int lat = suggestion.optInt("ycoord");
                        final int lon = suggestion.optInt("xcoord");
                        final int weight = jsonGetStopsUseWeight ? suggestion.getInt("weight") : -i;
                        String localId = null;
                        final Matcher m = P_AJAX_GET_STOPS_ID.matcher(suggestion.getString("id"));
                        if (m.matches())
                            localId = m.group(1);

                        final Location location;

                        if (type == 1) // station
                        {
                            final String[] placeAndName = splitStationName(value);
                            location = new Location(LocationType.STATION, localId, Point.from1E6(lat, lon),
                                    placeAndName[0], placeAndName[1]);
                        } else if (type == 2) // address
                        {
                            final String[] placeAndName = splitAddress(value);
                            location = new Location(LocationType.ADDRESS, null, Point.from1E6(lat, lon),
                                    placeAndName[0], placeAndName[1]);
                        } else if (type == 4) // poi
                        {
                            final String[] placeAndName = splitPOI(value);
                            location = new Location(LocationType.POI, localId, Point.from1E6(lat, lon), placeAndName[0],
                                    placeAndName[1]);
                        } else if (type == 128) // crossing
                        {
                            final String[] placeAndName = splitAddress(value);
                            location = new Location(LocationType.ADDRESS, localId, Point.from1E6(lat, lon),
                                    placeAndName[0], placeAndName[1]);
                        } else if (type == 87) {
                            location = null;
                            // don't know what to do
                        } else {
                            throw new IllegalStateException("unknown type " + type + " on " + url);
                        }

                        if (location != null) {
                            final SuggestedLocation suggestedLocation = new SuggestedLocation(location, weight);
                            locations.add(suggestedLocation);
                        }
                    }
                }

                return new SuggestLocationsResult(new ResultHeader(network, SERVER_PRODUCT), locations);
            } catch (final JSONException x) {
                throw new RuntimeException("cannot parse: '" + json + "' on " + url, x);
            }
        } else {
            throw new RuntimeException("cannot parse: '" + page + "' on " + url);
        }
    }

    @Override
    public QueryDeparturesResult queryDepartures(final String stationId, final @Nullable Date time,
            final int maxDepartures, final boolean equivs) throws IOException {
        checkNotNull(Strings.emptyToNull(stationId));

        final HttpUrl.Builder url = stationBoardEndpoint.newBuilder().addPathSegment(apiLanguage);
        appendXmlStationBoardParameters(url, time, stationId, maxDepartures, equivs, "vs_java3");
        return xmlStationBoard(url.build(), stationId);
    }

    protected void appendXmlStationBoardParameters(final HttpUrl.Builder url, final @Nullable Date time,
            final String stationId, final int maxDepartures, final boolean equivs, final @Nullable String styleSheet) {
        url.addQueryParameter("productsFilter", allProductsString().toString());
        url.addQueryParameter("boardType", "dep");
        if (stationBoardCanDoEquivs)
            url.addQueryParameter("disableEquivs", equivs ? "0" : "1");
        url.addQueryParameter("maxJourneys",
                Integer.toString(maxDepartures > 0 ? maxDepartures : DEFAULT_MAX_DEPARTURES));
        url.addEncodedQueryParameter("input", ParserUtils.urlEncode(normalizeStationId(stationId), requestUrlEncoding));
        appendDateTimeParameters(url, time, "date", "time");
        if (clientType != null)
            url.addEncodedQueryParameter("clientType", ParserUtils.urlEncode(clientType, requestUrlEncoding));
        if (styleSheet != null)
            url.addQueryParameter("L", styleSheet);
        url.addQueryParameter("hcount", "0"); // prevents showing old departures
        url.addQueryParameter("start", "yes");
    }

    protected void appendDateTimeParameters(final HttpUrl.Builder url, final Date time, final String dateParamName,
            final String timeParamName) {
        final Calendar c = new GregorianCalendar(timeZone);
        c.setTime(time);
        final int year = c.get(Calendar.YEAR);
        final int month = c.get(Calendar.MONTH) + 1;
        final int day = c.get(Calendar.DAY_OF_MONTH);
        final int hour = c.get(Calendar.HOUR_OF_DAY);
        final int minute = c.get(Calendar.MINUTE);
        url.addEncodedQueryParameter(dateParamName,
                ParserUtils.urlEncode(
                        useIso8601 ? String.format(Locale.ENGLISH, "%04d-%02d-%02d", year, month, day)
                                : String.format(Locale.ENGLISH, "%02d.%02d.%02d", day, month, year - 2000),
                        requestUrlEncoding));
        url.addEncodedQueryParameter(timeParamName,
                ParserUtils.urlEncode(String.format(Locale.ENGLISH, "%02d:%02d", hour, minute), requestUrlEncoding));
    }

    private static final Pattern P_XML_STATION_BOARD_DELAY = Pattern.compile("(?:-|k\\.A\\.?|cancel|([+-]?\\s*\\d+))");

    protected final QueryDeparturesResult xmlStationBoard(final HttpUrl url, final String stationId)
            throws IOException {
        final String normalizedStationId = normalizeStationId(stationId);
        final AtomicReference result = new AtomicReference<>();

        httpClient.getInputStream(new HttpClient.Callback() {

            @Override
            public void onSuccessful(final CharSequence bodyPeek, final ResponseBody body) throws IOException {
                StringReplaceReader reader = null;
                String firstChars = null;

                // work around unparsable XML
                reader = new StringReplaceReader(body.charStream(), " & ", " & ");
                reader.replace("", " ");
                reader.replace("", " ");
                reader.replace("", " ");
                reader.replace("", " ");
                reader.replace("", " ");
                reader.replace("", " ");
                reader.replace("
", " "); reader.replace(" ->", " →"); // right arrow reader.replace(" <-", " ←"); // left arrow reader.replace(" <> ", " ↔ "); // left-right arrow addCustomReplaces(reader); try { final XmlPullParserFactory factory = XmlPullParserFactory .newInstance(System.getProperty(XmlPullParserFactory.PROPERTY_NAME), null); final XmlPullParser pp = factory.newPullParser(); pp.setInput(reader); pp.nextTag(); final ResultHeader header = new ResultHeader(network, SERVER_PRODUCT); final QueryDeparturesResult r = new QueryDeparturesResult(header); if (XmlPullUtil.test(pp, "Err")) { final String code = XmlPullUtil.attr(pp, "code"); final String text = XmlPullUtil.attr(pp, "text"); if (code.equals("H730")) { result.set(new QueryDeparturesResult(header, QueryDeparturesResult.Status.INVALID_STATION)); return; } if (code.equals("H890")) { r.stationDepartures .add(new StationDepartures(new Location(LocationType.STATION, normalizedStationId), Collections. emptyList(), null)); result.set(r); return; } throw new IllegalArgumentException("unknown error " + code + ", " + text); } String[] stationPlaceAndName = null; if (stationBoardHasStationTable) XmlPullUtil.enter(pp, "StationTable"); else checkState(!XmlPullUtil.test(pp, "StationTable")); if (stationBoardHasLocation) { XmlPullUtil.require(pp, "St"); final String evaId = XmlPullUtil.attr(pp, "evaId"); if (evaId != null) { if (!evaId.equals(normalizedStationId)) throw new IllegalStateException( "stationId: " + normalizedStationId + ", evaId: " + evaId); final String name = XmlPullUtil.attr(pp, "name"); if (name != null) stationPlaceAndName = splitStationName(name.trim()); } XmlPullUtil.requireSkip(pp, "St"); } else { checkState(!XmlPullUtil.test(pp, "St")); } while (XmlPullUtil.test(pp, "Journey")) { final String fpTime = XmlPullUtil.attr(pp, "fpTime"); final String fpDate = XmlPullUtil.attr(pp, "fpDate"); final String delay = XmlPullUtil.attr(pp, "delay"); final String eDelay = XmlPullUtil.optAttr(pp, "e_delay", null); final String platform = XmlPullUtil.optAttr(pp, "platform", null); // TODO newpl final String targetLoc = XmlPullUtil.optAttr(pp, "targetLoc", null); // TODO hafasname final String dirnr = XmlPullUtil.optAttr(pp, "dirnr", null); final String prod = XmlPullUtil.attr(pp, "prod"); final String classStr = XmlPullUtil.optAttr(pp, "class", null); final String dir = XmlPullUtil.optAttr(pp, "dir", null); final String capacityStr = XmlPullUtil.optAttr(pp, "capacity", null); final String depStation = XmlPullUtil.optAttr(pp, "depStation", null); final String delayReason = XmlPullUtil.optAttr(pp, "delayReason", null); // TODO is_reachable // TODO disableTrainInfo // TODO lineFG/lineBG (ZVV) final String administration = normalizeLineAdministration( XmlPullUtil.optAttr(pp, "administration", null)); if (!"cancel".equals(delay) && !"cancel".equals(eDelay)) { final Calendar plannedTime = new GregorianCalendar(timeZone); plannedTime.clear(); parseXmlStationBoardDate(plannedTime, fpDate); parseXmlStationBoardTime(plannedTime, fpTime); final Calendar predictedTime; if (eDelay != null) { predictedTime = new GregorianCalendar(timeZone); predictedTime.setTimeInMillis(plannedTime.getTimeInMillis()); predictedTime.add(Calendar.MINUTE, Integer.parseInt(eDelay)); } else if (delay != null) { final Matcher m = P_XML_STATION_BOARD_DELAY.matcher(delay); if (m.matches()) { if (m.group(1) != null) { predictedTime = new GregorianCalendar(timeZone); predictedTime.setTimeInMillis(plannedTime.getTimeInMillis()); predictedTime.add(Calendar.MINUTE, Integer.parseInt(m.group(1))); } else { predictedTime = null; } } else { throw new RuntimeException("cannot parse delay: '" + delay + "'"); } } else { predictedTime = null; } final Position position = parsePosition(ParserUtils.resolveEntities(platform)); final String destinationName; if (dir != null) destinationName = dir.trim(); else if (targetLoc != null) destinationName = targetLoc.trim(); else destinationName = null; final Location destination; if (dirnr != null) { final String[] destinationPlaceAndName = splitStationName(destinationName); destination = new Location(LocationType.STATION, dirnr, destinationPlaceAndName[0], destinationPlaceAndName[1]); } else { destination = new Location(LocationType.ANY, null, null, destinationName); } final Line prodLine = parseLineAndType(prod); final Line line; if (classStr != null) { final Product product = intToProduct(Integer.parseInt(classStr)); if (product == null) throw new IllegalArgumentException(); // could check for type consistency here final Set attrs = prodLine.attrs; if (attrs != null) line = newLine(administration, product, prodLine.label, null, attrs.toArray(new Line.Attr[0])); else line = newLine(administration, product, prodLine.label, null); } else { final Set attrs = prodLine.attrs; if (attrs != null) line = newLine(administration, prodLine.product, prodLine.label, null, attrs.toArray(new Line.Attr[0])); else line = newLine(administration, prodLine.product, prodLine.label, null); } final int[] capacity; if (capacityStr != null && !"0|0".equals(capacityStr)) { final String[] capacityParts = capacityStr.split("\\|"); capacity = new int[] { Integer.parseInt(capacityParts[0]), Integer.parseInt(capacityParts[1]) }; } else { capacity = null; } final String message; if (delayReason != null) { final String msg = delayReason.trim(); message = msg.length() > 0 ? msg : null; } else { message = null; } final Departure departure = new Departure(plannedTime.getTime(), predictedTime != null ? predictedTime.getTime() : null, line, position, destination, capacity, message); final Location location; if (!stationBoardCanDoEquivs || depStation == null) { location = new Location(LocationType.STATION, normalizedStationId, stationPlaceAndName != null ? stationPlaceAndName[0] : null, stationPlaceAndName != null ? stationPlaceAndName[1] : null); } else { final String[] depPlaceAndName = splitStationName(depStation); location = new Location(LocationType.STATION, null, depPlaceAndName[0], depPlaceAndName[1]); } StationDepartures stationDepartures = findStationDepartures(r.stationDepartures, location); if (stationDepartures == null) { stationDepartures = new StationDepartures(location, new ArrayList(8), null); r.stationDepartures.add(stationDepartures); } stationDepartures.departures.add(departure); } XmlPullUtil.requireSkip(pp, "Journey"); } if (stationBoardHasStationTable) XmlPullUtil.exit(pp, "StationTable"); XmlPullUtil.requireEndDocument(pp); // sort departures for (final StationDepartures stationDepartures : r.stationDepartures) Collections.sort(stationDepartures.departures, Departure.TIME_COMPARATOR); result.set(r); } catch (final XmlPullParserException x) { throw new ParserException("cannot parse xml: " + firstChars, x); } } }, url); return result.get(); } protected void parseXmlStationBoardDate(final Calendar calendar, final String dateStr) { if (dateStr.length() == 8) ParserUtils.parseGermanDate(calendar, dateStr); else if (dateStr.length() == 10) ParserUtils.parseIsoDate(calendar, dateStr); else throw new IllegalStateException("cannot parse: '" + dateStr + "'"); } protected void parseXmlStationBoardTime(final Calendar calendar, final String timeStr) { ParserUtils.parseEuropeanTime(calendar, timeStr); } protected void addCustomReplaces(final StringReplaceReader reader) { } @Override public QueryTripsResult queryTrips(final Location from, final @Nullable Location via, final Location to, final Date date, final boolean dep, final @Nullable TripOptions options) throws IOException { return queryTripsBinary(from, via, to, date, dep, options); } @Override public QueryTripsResult queryMoreTrips(final QueryTripsContext context, final boolean later) throws IOException { return queryMoreTripsBinary(context, later); } protected final QueryTripsResult queryTripsXml(Location from, @Nullable Location via, Location to, final Date date, final boolean dep, @Nullable TripOptions options) throws IOException { final ResultHeader header = new ResultHeader(network, SERVER_PRODUCT); if (!from.isIdentified()) { final List locations = suggestLocations(from.name).getLocations(); if (locations.isEmpty()) return new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS); // TODO if (locations.size() > 1) return new QueryTripsResult(header, locations, null, null); from = locations.get(0); } if (via != null && !via.isIdentified()) { final List locations = suggestLocations(via.name).getLocations(); if (locations.isEmpty()) return new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS); // TODO if (locations.size() > 1) return new QueryTripsResult(header, null, locations, null); via = locations.get(0); } if (!to.isIdentified()) { final List locations = suggestLocations(to.name).getLocations(); if (locations.isEmpty()) return new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS); // TODO if (locations.size() > 1) return new QueryTripsResult(header, null, null, locations); to = locations.get(0); } final Calendar c = new GregorianCalendar(timeZone); c.setTime(date); if (options == null) options = new TripOptions(); final CharSequence productsStr; if (options.products != null) productsStr = productsString(options.products); else productsStr = allProductsString(); final char bikeChar = (options.flags != null && options.flags.contains(TripFlag.BIKE)) ? '1' : '0'; final StringBuilder conReq = new StringBuilder(""); conReq.append("").append(locationXml(from)); conReq.append(""); conReq.append(""); if (via != null) { conReq.append("").append(locationXml(via)); if (via.type != LocationType.ADDRESS) conReq.append(""); conReq.append(""); } conReq.append("").append(locationXml(to)).append(""); conReq.append(""); conReq.append(""); conReq.append(""); return queryTripsXml(null, true, conReq, from, via, to); } protected final QueryTripsResult queryMoreTripsXml(final QueryTripsContext contextObj, final boolean later) throws IOException { final Context context = (Context) contextObj; final StringBuilder conScrReq = new StringBuilder(""); conScrReq.append("").append(later ? context.laterContext : context.earlierContext) .append(""); conScrReq.append(""); return queryTripsXml(context, later, conScrReq, null, null, null); } private QueryTripsResult queryTripsXml(final Context previousContext, final boolean later, final CharSequence conReq, final Location from, final @Nullable Location via, final Location to) throws IOException { final String request = wrapReqC(conReq, null); final HttpUrl endpoint = extXmlEndpoint != null ? extXmlEndpoint : queryEndpoint.newBuilder().addPathSegment(apiLanguage).build(); final AtomicReference result = new AtomicReference<>(); httpClient.getInputStream(new HttpClient.Callback() { @Override public void onSuccessful(final CharSequence bodyPeek, final ResponseBody body) throws IOException { try { final XmlPullParserFactory factory = XmlPullParserFactory .newInstance(System.getProperty(XmlPullParserFactory.PROPERTY_NAME), null); final XmlPullParser pp = factory.newPullParser(); pp.setInput(body.charStream()); XmlPullUtil.require(pp, "ResC"); final String product = XmlPullUtil.attr(pp, "prod").split(" ")[0]; final ResultHeader header = new ResultHeader(network, SERVER_PRODUCT, product, null, 0, null); XmlPullUtil.enter(pp, "ResC"); if (XmlPullUtil.test(pp, "Err")) { final String code = XmlPullUtil.attr(pp, "code"); if (code.equals("I3")) { result.set(new QueryTripsResult(header, QueryTripsResult.Status.INVALID_DATE)); return; } if (code.equals("F1")) { result.set(new QueryTripsResult(header, QueryTripsResult.Status.SERVICE_DOWN)); return; } throw new IllegalStateException("error " + code + " " + XmlPullUtil.attr(pp, "text")); } XmlPullUtil.enter(pp, "ConRes"); if (XmlPullUtil.test(pp, "Err")) { final String code = XmlPullUtil.attr(pp, "code"); log.debug("Hafas error: {}", code); if (code.equals("K9260")) { // Unknown departure station result.set(new QueryTripsResult(header, QueryTripsResult.Status.UNKNOWN_FROM)); return; } if (code.equals("K9280")) { // Unknown intermediate station result.set(new QueryTripsResult(header, QueryTripsResult.Status.UNKNOWN_VIA)); return; } if (code.equals("K9300")) { // Unknown arrival station result.set(new QueryTripsResult(header, QueryTripsResult.Status.UNKNOWN_TO)); return; } if (code.equals("K9360")) { // Date outside of the timetable period result.set(new QueryTripsResult(header, QueryTripsResult.Status.INVALID_DATE)); return; } if (code.equals("K9380")) { // Dep./Arr./Intermed. or equivalent station defined more than once result.set(new QueryTripsResult(header, QueryTripsResult.Status.TOO_CLOSE)); return; } if (code.equals("K895")) { // Departure/Arrival are too near result.set(new QueryTripsResult(header, QueryTripsResult.Status.TOO_CLOSE)); return; } if (code.equals("K9220")) { // Nearby to the given address stations could not be found result.set(new QueryTripsResult(header, QueryTripsResult.Status.UNRESOLVABLE_ADDRESS)); return; } if (code.equals("K9240")) { // Internal error result.set(new QueryTripsResult(header, QueryTripsResult.Status.SERVICE_DOWN)); return; } if (code.equals("K890")) { // No connections found result.set(new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS)); return; } if (code.equals("K891")) { // No route found (try entering an intermediate station) result.set(new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS)); return; } if (code.equals("K899")) { // An error occurred result.set(new QueryTripsResult(header, QueryTripsResult.Status.SERVICE_DOWN)); return; } if (code.equals("K1:890")) { // Unsuccessful or incomplete search (direction: forward) result.set(new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS)); return; } if (code.equals("K2:890")) { // Unsuccessful or incomplete search (direction: backward) result.set(new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS)); return; } throw new IllegalStateException("error " + code + " " + XmlPullUtil.attr(pp, "text")); } // H9380 Dep./Arr./Intermed. or equivalent stations defined more than once // H9360 Error in data field // H9320 The input is incorrect or incomplete // H9300 Unknown arrival station // H9280 Unknown intermediate station // H9260 Unknown departure station // H9250 Part inquiry interrupted // H9240 Unsuccessful search // H9230 An internal error occurred // H9220 Nearby to the given address stations could not be found // H900 Unsuccessful or incomplete search (timetable change) // H892 Inquiry too complex (try entering less intermediate stations) // H891 No route found (try entering an intermediate station) // H890 Unsuccessful search. // H500 Because of too many trains the connection is not complete // H460 One or more stops are passed through multiple times. // H455 Prolonged stop // H410 Display may be incomplete due to change of timetable // H390 Departure/Arrival replaced by an equivalent station // H895 Departure/Arrival are too near // H899 Unsuccessful or incomplete search (timetable change final String c = XmlPullUtil.optValueTag(pp, "ConResCtxt", null); final Context context; if (previousContext == null) context = new Context(c, c, 0); else if (later) context = new Context(c, previousContext.earlierContext, previousContext.sequence + 1); else context = new Context(previousContext.laterContext, c, previousContext.sequence + 1); XmlPullUtil.enter(pp, "ConnectionList"); final List trips = new ArrayList<>(); while (XmlPullUtil.test(pp, "Connection")) { final String id = context.sequence + "/" + XmlPullUtil.attr(pp, "id"); XmlPullUtil.enter(pp, "Connection"); while (pp.getName().equals("RtStateList")) XmlPullUtil.next(pp); XmlPullUtil.enter(pp, "Overview"); final Calendar currentDate = new GregorianCalendar(timeZone); currentDate.clear(); parseDate(currentDate, XmlPullUtil.valueTag(pp, "Date")); XmlPullUtil.enter(pp, "Departure"); XmlPullUtil.enter(pp, "BasicStop"); while (pp.getName().equals("StAttrList")) XmlPullUtil.next(pp); final Location departureLocation = parseLocation(pp); XmlPullUtil.enter(pp, "Dep"); XmlPullUtil.skipExit(pp, "Dep"); final int[] capacity; if (XmlPullUtil.test(pp, "StopPrognosis")) { XmlPullUtil.enter(pp, "StopPrognosis"); XmlPullUtil.optSkip(pp, "Arr"); XmlPullUtil.optSkip(pp, "Dep"); XmlPullUtil.enter(pp, "Status"); XmlPullUtil.skipExit(pp, "Status"); final int capacity1st = Integer.parseInt(XmlPullUtil.optValueTag(pp, "Capacity1st", "0")); final int capacity2nd = Integer.parseInt(XmlPullUtil.optValueTag(pp, "Capacity2nd", "0")); if (capacity1st > 0 || capacity2nd > 0) capacity = new int[] { capacity1st, capacity2nd }; else capacity = null; XmlPullUtil.skipExit(pp, "StopPrognosis"); } else { capacity = null; } XmlPullUtil.skipExit(pp, "BasicStop"); XmlPullUtil.skipExit(pp, "Departure"); XmlPullUtil.enter(pp, "Arrival"); XmlPullUtil.enter(pp, "BasicStop"); while (pp.getName().equals("StAttrList")) XmlPullUtil.next(pp); final Location arrivalLocation = parseLocation(pp); XmlPullUtil.skipExit(pp, "BasicStop"); XmlPullUtil.skipExit(pp, "Arrival"); final int numTransfers = Integer.parseInt(XmlPullUtil.valueTag(pp, "Transfers")); XmlPullUtil.skipExit(pp, "Overview"); final List legs = new ArrayList<>(4); XmlPullUtil.enter(pp, "ConSectionList"); final Calendar time = new GregorianCalendar(timeZone); while (XmlPullUtil.test(pp, "ConSection")) { XmlPullUtil.enter(pp, "ConSection"); // departure XmlPullUtil.enter(pp, "Departure"); XmlPullUtil.enter(pp, "BasicStop"); while (pp.getName().equals("StAttrList")) XmlPullUtil.next(pp); final Location sectionDepartureLocation = parseLocation(pp); XmlPullUtil.optSkip(pp, "Arr"); XmlPullUtil.enter(pp, "Dep"); time.setTimeInMillis(currentDate.getTimeInMillis()); parseTime(time, XmlPullUtil.valueTag(pp, "Time")); final Date departureTime = time.getTime(); final Position departurePos = parsePlatform(pp); XmlPullUtil.skipExit(pp, "Dep"); XmlPullUtil.skipExit(pp, "BasicStop"); XmlPullUtil.skipExit(pp, "Departure"); // journey final Line line; Location destination = null; List intermediateStops = null; final String tag = pp.getName(); if (tag.equals("Journey")) { XmlPullUtil.enter(pp, "Journey"); while (pp.getName().equals("JHandle")) XmlPullUtil.next(pp); XmlPullUtil.enter(pp, "JourneyAttributeList"); boolean wheelchairAccess = false; String name = null; String category = null; String shortCategory = null; while (XmlPullUtil.test(pp, "JourneyAttribute")) { XmlPullUtil.enter(pp, "JourneyAttribute"); XmlPullUtil.require(pp, "Attribute"); final String attrName = XmlPullUtil.attr(pp, "type"); final String code = XmlPullUtil.optAttr(pp, "code", null); XmlPullUtil.enter(pp, "Attribute"); final Map attributeVariants = parseAttributeVariants(pp); XmlPullUtil.skipExit(pp, "Attribute"); XmlPullUtil.skipExit(pp, "JourneyAttribute"); if ("bf".equals(code)) { wheelchairAccess = true; } else if ("NAME".equals(attrName)) { name = attributeVariants.get("NORMAL"); } else if ("CATEGORY".equals(attrName)) { shortCategory = attributeVariants.get("SHORT"); category = attributeVariants.get("NORMAL"); // longCategory = attributeVariants.get("LONG"); } else if ("DIRECTION".equals(attrName)) { final String[] destinationPlaceAndName = splitStationName( attributeVariants.get("NORMAL")); destination = new Location(LocationType.ANY, null, destinationPlaceAndName[0], destinationPlaceAndName[1]); } } XmlPullUtil.skipExit(pp, "JourneyAttributeList"); if (XmlPullUtil.test(pp, "PassList")) { intermediateStops = new LinkedList<>(); XmlPullUtil.enter(pp, "PassList"); while (XmlPullUtil.test(pp, "BasicStop")) { XmlPullUtil.enter(pp, "BasicStop"); while (XmlPullUtil.test(pp, "StAttrList")) XmlPullUtil.next(pp); final Location location = parseLocation(pp); if (location.id != sectionDepartureLocation.id) { Date stopArrivalTime = null; Date stopDepartureTime = null; Position stopArrivalPosition = null; Position stopDeparturePosition = null; if (XmlPullUtil.test(pp, "Arr")) { XmlPullUtil.enter(pp, "Arr"); time.setTimeInMillis(currentDate.getTimeInMillis()); parseTime(time, XmlPullUtil.valueTag(pp, "Time")); stopArrivalTime = time.getTime(); stopArrivalPosition = parsePlatform(pp); XmlPullUtil.skipExit(pp, "Arr"); } if (XmlPullUtil.test(pp, "Dep")) { XmlPullUtil.enter(pp, "Dep"); time.setTimeInMillis(currentDate.getTimeInMillis()); parseTime(time, XmlPullUtil.valueTag(pp, "Time")); stopDepartureTime = time.getTime(); stopDeparturePosition = parsePlatform(pp); XmlPullUtil.skipExit(pp, "Dep"); } intermediateStops.add(new Stop(location, stopArrivalTime, stopArrivalPosition, stopDepartureTime, stopDeparturePosition)); } XmlPullUtil.skipExit(pp, "BasicStop"); } XmlPullUtil.skipExit(pp, "PassList"); } XmlPullUtil.skipExit(pp, "Journey"); if (category == null) category = shortCategory; line = parseLine(category, name, wheelchairAccess); } else if (tag.equals("Walk") || tag.equals("Transfer") || tag.equals("GisRoute")) { XmlPullUtil.enter(pp); XmlPullUtil.enter(pp, "Duration"); XmlPullUtil.skipExit(pp, "Duration"); XmlPullUtil.skipExit(pp); line = null; } else { throw new IllegalStateException("cannot handle: " + pp.getName()); } // polyline final List path; if (XmlPullUtil.test(pp, "Polyline")) { path = new LinkedList<>(); XmlPullUtil.enter(pp, "Polyline"); while (XmlPullUtil.test(pp, "Point")) { final int x = XmlPullUtil.intAttr(pp, "x"); final int y = XmlPullUtil.intAttr(pp, "y"); path.add(Point.from1E6(y, x)); XmlPullUtil.next(pp); } XmlPullUtil.skipExit(pp, "Polyline"); } else { path = null; } // arrival XmlPullUtil.enter(pp, "Arrival"); XmlPullUtil.enter(pp, "BasicStop"); while (pp.getName().equals("StAttrList")) XmlPullUtil.next(pp); final Location sectionArrivalLocation = parseLocation(pp); XmlPullUtil.enter(pp, "Arr"); time.setTimeInMillis(currentDate.getTimeInMillis()); parseTime(time, XmlPullUtil.valueTag(pp, "Time")); final Date arrivalTime = time.getTime(); final Position arrivalPos = parsePlatform(pp); XmlPullUtil.skipExit(pp, "Arr"); XmlPullUtil.skipExit(pp, "BasicStop"); XmlPullUtil.skipExit(pp, "Arrival"); // remove last intermediate if (intermediateStops != null) if (!intermediateStops.isEmpty()) if (!intermediateStops.get(intermediateStops.size() - 1).location .equals(sectionArrivalLocation)) intermediateStops.remove(intermediateStops.size() - 1); XmlPullUtil.skipExit(pp, "ConSection"); if (line != null) { final Stop departure = new Stop(sectionDepartureLocation, true, departureTime, null, departurePos, null); final Stop arrival = new Stop(sectionArrivalLocation, false, arrivalTime, null, arrivalPos, null); legs.add(new Trip.Public(line, destination, departure, arrival, intermediateStops, path, null)); } else { if (legs.size() > 0 && legs.get(legs.size() - 1) instanceof Trip.Individual) { final Trip.Individual lastIndividualLeg = (Trip.Individual) legs .remove(legs.size() - 1); legs.add(new Trip.Individual(Trip.Individual.Type.WALK, lastIndividualLeg.departure, lastIndividualLeg.departureTime, sectionArrivalLocation, arrivalTime, null, 0)); } else { legs.add(new Trip.Individual(Trip.Individual.Type.WALK, sectionDepartureLocation, departureTime, sectionArrivalLocation, arrivalTime, null, 0)); } } } XmlPullUtil.skipExit(pp, "ConSectionList"); XmlPullUtil.skipExit(pp, "Connection"); trips.add(new Trip(id, departureLocation, arrivalLocation, legs, null, capacity, numTransfers)); } XmlPullUtil.skipExit(pp, "ConnectionList"); result.set(new QueryTripsResult(header, null, from, via, to, context, trips)); } catch (final XmlPullParserException x) { throw new ParserException("cannot parse xml: " + bodyPeek, x); } } }, endpoint, request, "application/xml", null); return result.get(); } private final Location parseLocation(final XmlPullParser pp) throws XmlPullParserException, IOException { final Location location; if (pp.getName().equals("Station")) location = parseStation(pp); else if (pp.getName().equals("Poi")) location = parsePoi(pp); else if (pp.getName().equals("Address")) location = parseAddress(pp); else throw new IllegalStateException("cannot parse: " + pp.getName()); XmlPullUtil.next(pp); return location; } private final Map parseAttributeVariants(final XmlPullParser pp) throws XmlPullParserException, IOException { final Map attributeVariants = new HashMap<>(); while (XmlPullUtil.test(pp, "AttributeVariant")) { final String type = XmlPullUtil.attr(pp, "type"); XmlPullUtil.enter(pp, "AttributeVariant"); final String value = XmlPullUtil.optValueTag(pp, "Text", null); XmlPullUtil.skipExit(pp, "AttributeVariant"); attributeVariants.put(type, value); } return attributeVariants; } private static final Pattern P_DATE = Pattern.compile("(\\d{4})(\\d{2})(\\d{2})"); private static final void parseDate(final Calendar calendar, final CharSequence str) { final Matcher m = P_DATE.matcher(str); if (!m.matches()) throw new RuntimeException("cannot parse: '" + str + "'"); calendar.set(Calendar.YEAR, Integer.parseInt(m.group(1))); calendar.set(Calendar.MONTH, Integer.parseInt(m.group(2)) - 1); calendar.set(Calendar.DAY_OF_MONTH, Integer.parseInt(m.group(3))); } private static final Pattern P_TIME = Pattern.compile("(\\d+)d(\\d+):(\\d{2}):(\\d{2})"); private static void parseTime(final Calendar calendar, final CharSequence str) { final Matcher m = P_TIME.matcher(str); if (!m.matches()) throw new IllegalArgumentException("cannot parse: '" + str + "'"); calendar.set(Calendar.HOUR_OF_DAY, Integer.parseInt(m.group(2))); calendar.set(Calendar.MINUTE, Integer.parseInt(m.group(3))); calendar.set(Calendar.SECOND, Integer.parseInt(m.group(4))); calendar.set(Calendar.MILLISECOND, 0); calendar.add(Calendar.DAY_OF_MONTH, Integer.parseInt(m.group(1))); } private static final String locationXml(final Location location) { if (location.type == LocationType.STATION && location.hasId()) return ""; else if (location.type == LocationType.POI && location.hasCoord()) return ""; else if (location.type == LocationType.ADDRESS && location.hasCoord()) return "
"; else if (location.type == LocationType.COORD && location.hasCoord()) return ""; else throw new IllegalArgumentException("cannot handle: " + location); } protected final String locationId(final Location location) { final StringBuilder id = new StringBuilder(); id.append("A=").append(locationType(location)); if (location.type == LocationType.STATION && location.hasId()) { id.append("@L=").append(normalizeStationId(location.id)); } else if (location.hasCoord()) { id.append("@X=").append(location.getLonAs1E6()); id.append("@Y=").append(location.getLatAs1E6()); id.append("@O=").append(location.name != null ? location.name : String.format(Locale.ENGLISH, "%.6f, %.6f", location.getLatAsDouble(), location.getLonAsDouble())); } else if (location.name != null) { id.append("@G=").append(location.name); if (location.type != LocationType.ANY) id.append('!'); } return id.toString(); } protected static final int locationType(final Location location) { final LocationType type = location.type; if (type == LocationType.STATION) return 1; if (type == LocationType.POI) return 4; if (type == LocationType.COORD || (type == LocationType.ADDRESS && location.hasCoord())) return 16; if (type == LocationType.ADDRESS && location.name != null) return 2; if (type == LocationType.ANY) return 255; throw new IllegalArgumentException(location.type.toString()); } protected void appendQueryTripsBinaryParameters(final HttpUrl.Builder url, final Location from, final @Nullable Location via, final Location to, final Date date, final boolean dep, final @Nullable Set products, final @Nullable Accessibility accessibility, final @Nullable Set flags) { url.addQueryParameter("start", "Suchen"); url.addEncodedQueryParameter("REQ0JourneyStopsS0ID", ParserUtils.urlEncode(locationId(from), requestUrlEncoding)); url.addEncodedQueryParameter("REQ0JourneyStopsZ0ID", ParserUtils.urlEncode(locationId(to), requestUrlEncoding)); if (via != null) { // workaround, for there does not seem to be a REQ0JourneyStops1.0ID parameter url.addQueryParameter("REQ0JourneyStops1.0A", Integer.toString(locationType(via))); if (via.type == LocationType.STATION && via.hasId()) { url.addQueryParameter("REQ0JourneyStops1.0L", via.id); } else if (via.hasCoord()) { url.addQueryParameter("REQ0JourneyStops1.0X", Integer.toString(via.getLonAs1E6())); url.addQueryParameter("REQ0JourneyStops1.0Y", Integer.toString(via.getLatAs1E6())); if (via.name == null) url.addQueryParameter("REQ0JourneyStops1.0O", String.format(Locale.ENGLISH, "%.6f, %.6f", via.getLatAsDouble(), via.getLonAsDouble())); } else if (via.name != null) { url.addEncodedQueryParameter("REQ0JourneyStops1.0G", ParserUtils .urlEncode(via.name + (via.type != LocationType.ANY ? "!" : ""), requestUrlEncoding)); } } url.addQueryParameter("REQ0HafasSearchForw", dep ? "1" : "0"); appendDateTimeParameters(url, date, "REQ0JourneyDate", "REQ0JourneyTime"); final CharSequence productsStr; if (products != null) productsStr = productsString(products); else productsStr = allProductsString(); url.addQueryParameter("REQ0JourneyProduct_prod_list_1", productsStr.toString()); if (accessibility != null && accessibility != Accessibility.NEUTRAL) { if (accessibility == Accessibility.LIMITED) url.addQueryParameter("REQ0AddParamBaimprofile", "1"); else if (accessibility == Accessibility.BARRIER_FREE) url.addQueryParameter("REQ0AddParamBaimprofile", "0"); } if (flags != null && flags.contains(TripFlag.BIKE)) url.addQueryParameter("REQ0JourneyProduct_opt3", "1"); appendCommonQueryTripsBinaryParameters(url); } protected void appendCommonQueryTripsBinaryParameters(final HttpUrl.Builder url) { url.addQueryParameter("h2g-direct", "11"); if (clientType != null) url.addEncodedQueryParameter("clientType", ParserUtils.urlEncode(clientType, requestUrlEncoding)); } private final static int QUERY_TRIPS_BINARY_BUFFER_SIZE = 384 * 1024; protected final QueryTripsResult queryTripsBinary(Location from, @Nullable Location via, Location to, final Date date, final boolean dep, @Nullable TripOptions options) throws IOException { final ResultHeader header = new ResultHeader(network, SERVER_PRODUCT); if (!from.isIdentified()) { final List locations = suggestLocations(from.name).getLocations(); if (locations.isEmpty()) return new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS); // TODO if (locations.size() > 1) return new QueryTripsResult(header, locations, null, null); from = locations.get(0); } if (via != null && !via.isIdentified()) { final List locations = suggestLocations(via.name).getLocations(); if (locations.isEmpty()) return new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS); // TODO if (locations.size() > 1) return new QueryTripsResult(header, null, locations, null); via = locations.get(0); } if (!to.isIdentified()) { final List locations = suggestLocations(to.name).getLocations(); if (locations.isEmpty()) return new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS); // TODO if (locations.size() > 1) return new QueryTripsResult(header, null, null, locations); to = locations.get(0); } if (options == null) options = new TripOptions(); final HttpUrl.Builder url = queryEndpoint.newBuilder().addPathSegment(apiLanguage); appendQueryTripsBinaryParameters(url, from, via, to, date, dep, options.products, options.accessibility, options.flags); return queryTripsBinary(url.build(), from, via, to, QUERY_TRIPS_BINARY_BUFFER_SIZE); } protected void appendQueryMoreTripsBinaryParameters(final HttpUrl.Builder url, final QueryTripsBinaryContext context, final boolean later) { url.addQueryParameter("seqnr", Integer.toString(context.seqNr)); url.addQueryParameter("ident", context.ident); if (context.ld != null) url.addQueryParameter("ld", context.ld); url.addQueryParameter("REQ0HafasScrollDir", later ? "1" : "2"); appendCommonQueryTripsBinaryParameters(url); } protected QueryTripsResult queryMoreTripsBinary(final QueryTripsContext contextObj, final boolean later) throws IOException { final QueryTripsBinaryContext context = (QueryTripsBinaryContext) contextObj; final HttpUrl.Builder url = queryEndpoint.newBuilder().addPathSegment(apiLanguage); appendQueryMoreTripsBinaryParameters(url, context, later); return queryTripsBinary(url.build(), null, null, null, QUERY_TRIPS_BINARY_BUFFER_SIZE + context.usedBufferSize); } private class CustomBufferedInputStream extends BufferedInputStream { public CustomBufferedInputStream(final InputStream in) { super(in); } public int getCount() { return count; } } private QueryTripsResult queryTripsBinary(final HttpUrl url, final Location from, final @Nullable Location via, final Location to, final int expectedBufferSize) throws IOException { /* * Many thanks to Malte Starostik and Robert, who helped a lot with analyzing this API! */ final AtomicReference result = new AtomicReference<>(); httpClient.getInputStream(new HttpClient.Callback() { @Override public void onSuccessful(final CharSequence bodyPeek, final ResponseBody body) throws IOException { final CustomBufferedInputStream bis = new CustomBufferedInputStream( new GZIPInputStream(body.byteStream())); // initialize input stream final LittleEndianDataInputStream is = new LittleEndianDataInputStream(bis); is.mark(expectedBufferSize); // quick check of status final int version = is.readShortReverse(); if (version != 6 && version != 5) throw new IllegalStateException("unknown version: " + version + ", first chars: " + bodyPeek); final ResultHeader header = new ResultHeader(network, SERVER_PRODUCT, Integer.toString(version), null, 0, null); // quick seek for pointers is.reset(); is.skipBytes(0x20); final int serviceDaysTablePtr = is.readIntReverse(); final int stringTablePtr = is.readIntReverse(); is.reset(); is.skipBytes(0x36); final int stationTablePtr = is.readIntReverse(); final int commentTablePtr = is.readIntReverse(); is.reset(); is.skipBytes(0x46); final int extensionHeaderPtr = is.readIntReverse(); // read strings final StringTable strings = new StringTable(is, stringTablePtr, serviceDaysTablePtr - stringTablePtr); is.reset(); is.skipBytes(extensionHeaderPtr); // read extension header final int extensionHeaderLength = is.readIntReverse(); if (extensionHeaderLength < 0x2c) throw new IllegalStateException("too short: " + extensionHeaderLength); is.skipBytes(12); final int errorCode = is.readShortReverse(); if (errorCode == 0) { // string encoding is.skipBytes(14); final Charset stringEncoding = Charset.forName(strings.read(is)); strings.setEncoding(stringEncoding); // read number of trips is.reset(); is.skipBytes(30); final int numTrips = is.readShortReverse(); if (numTrips == 0) { result.set(new QueryTripsResult(header, url.toString(), from, via, to, null, new LinkedList())); return; } // read rest of header is.reset(); is.skipBytes(0x02); final Location resDeparture = location(is, strings); final Location resArrival = location(is, strings); is.skipBytes(10); final long resDate = date(is); /* final long resDate30 = */date(is); is.reset(); is.skipBytes(extensionHeaderPtr + 0x8); final int seqNr = is.readShortReverse(); if (seqNr == 0) throw new SessionExpiredException(); else if (seqNr < 0) throw new IllegalStateException("illegal sequence number: " + seqNr); final String requestId = strings.read(is); final int tripDetailsPtr = is.readIntReverse(); if (tripDetailsPtr == 0) throw new IllegalStateException("no connection details"); is.skipBytes(4); final int disruptionsPtr = is.readIntReverse(); is.skipBytes(10); final String ld = strings.read(is); final int attrsOffset = is.readIntReverse(); final int tripAttrsPtr; if (extensionHeaderLength >= 0x30) { if (extensionHeaderLength < 0x32) throw new IllegalArgumentException("too short: " + extensionHeaderLength); is.reset(); is.skipBytes(extensionHeaderPtr + 0x2c); tripAttrsPtr = is.readIntReverse(); } else { tripAttrsPtr = 0; } // determine stops offset is.reset(); is.skipBytes(tripDetailsPtr); final int tripDetailsVersion = is.readShortReverse(); if (tripDetailsVersion != 1) throw new IllegalStateException("unknown trip details version: " + tripDetailsVersion); is.skipBytes(0x02); final int tripDetailsIndexOffset = is.readShortReverse(); final int tripDetailsLegOffset = is.readShortReverse(); final int tripDetailsLegSize = is.readShortReverse(); final int stopsSize = is.readShortReverse(); final int stopsOffset = is.readShortReverse(); // read stations final StationTable stations = new StationTable(is, stationTablePtr, commentTablePtr - stationTablePtr, strings); // read comments final CommentTable comments = new CommentTable(is, commentTablePtr, tripDetailsPtr - commentTablePtr, strings); final List trips = new ArrayList<>(numTrips); // read trips for (int iTrip = 0; iTrip < numTrips; iTrip++) { is.reset(); is.skipBytes(0x4a + iTrip * 12); final int serviceDaysTableOffset = is.readShortReverse(); final int legsOffset = is.readIntReverse(); final int numLegs = is.readShortReverse(); final int numChanges = is.readShortReverse(); /* final long duration = time(is, 0, 0); */is.readShortReverse(); is.reset(); is.skipBytes(serviceDaysTablePtr + serviceDaysTableOffset); /* final String serviceDaysText = */strings.read(is); final int serviceBitBase = is.readShortReverse(); final int serviceBitLength = is.readShortReverse(); int tripDayOffset = serviceBitBase * 8; for (int i = 0; i < serviceBitLength; i++) { int serviceBits = is.read(); if (serviceBits == 0) { tripDayOffset += 8; continue; } while ((serviceBits & 0x80) == 0) { serviceBits = serviceBits << 1; tripDayOffset++; } break; } is.reset(); is.skipBytes(tripDetailsPtr + tripDetailsIndexOffset + iTrip * 2); final int tripDetailsOffset = is.readShortReverse(); is.reset(); is.skipBytes(tripDetailsPtr + tripDetailsOffset); final int realtimeStatus = is.readShortReverse(); /* final short delay = */is.readShortReverse(); /* final int legIndex = */is.readShortReverse(); is.skipBytes(2); // 0xffff /* final int legStatus = */is.readShortReverse(); is.skipBytes(2); // 0x0000 String connectionId = null; if (tripAttrsPtr != 0) { is.reset(); is.skipBytes(tripAttrsPtr + iTrip * 2); final int tripAttrsIndex = is.readShortReverse(); is.reset(); is.skipBytes(attrsOffset + tripAttrsIndex * 4); while (true) { final String key = strings.read(is); if (key == null) break; else if (key.equals("ConnectionId")) connectionId = strings.read(is); else is.skipBytes(2); } } final List legs = new ArrayList<>(numLegs); for (int iLegs = 0; iLegs < numLegs; iLegs++) { is.reset(); is.skipBytes(0x4a + legsOffset + iLegs * 20); final long plannedDepartureTime = time(is, resDate, tripDayOffset); final Location departureLocation = stations.read(is); final long plannedArrivalTime = time(is, resDate, tripDayOffset); final Location arrivalLocation = stations.read(is); final int type = is.readShortReverse(); final String lineName = strings.read(is); final Position plannedDeparturePosition = normalizePosition(strings.read(is)); final Position plannedArrivalPosition = normalizePosition(strings.read(is)); final int legAttrIndex = is.readShortReverse(); final List lineAttrs = new ArrayList<>(); String lineComment = null; boolean lineOnDemand = false; for (final String comment : comments.read(is)) { if (comment.startsWith("bf ")) { lineAttrs.add(Line.Attr.WHEEL_CHAIR_ACCESS); } else if (comment.startsWith("FA ") || comment.startsWith("FB ") || comment.startsWith("FR ")) { lineAttrs.add(Line.Attr.BICYCLE_CARRIAGE); } else if (comment.startsWith("$R ") || comment.startsWith("ga ") || comment.startsWith("ja ") || comment.startsWith("Vs ") || comment.startsWith("mu ") || comment.startsWith("mx ")) { lineOnDemand = true; lineComment = comment.substring(5); } } is.reset(); is.skipBytes(attrsOffset + legAttrIndex * 4); String directionStr = null; int lineClass = 0; String lineCategory = null; String routingType = null; String lineNetwork = null; while (true) { final String key = strings.read(is); if (key == null) break; else if (key.equals("Direction")) directionStr = strings.read(is); else if (key.equals("Class")) lineClass = Integer.parseInt(strings.read(is)); else if (key.equals("Category")) lineCategory = strings.read(is); // else if (key.equals("Operator")) // lineOperator = strings.read(is); else if (key.equals("GisRoutingType")) routingType = strings.read(is); else if (key.equals("AdminCode")) lineNetwork = normalizeLineAdministration(strings.read(is)); else is.skipBytes(2); } if (lineCategory == null && lineName != null) lineCategory = categoryFromName(lineName); is.reset(); is.skipBytes(tripDetailsPtr + tripDetailsOffset + tripDetailsLegOffset + iLegs * tripDetailsLegSize); if (tripDetailsLegSize != 16) throw new IllegalStateException( "unhandled trip details leg size: " + tripDetailsLegSize); final long predictedDepartureTime = time(is, resDate, tripDayOffset); final long predictedArrivalTime = time(is, resDate, tripDayOffset); final Position predictedDeparturePosition = normalizePosition(strings.read(is)); final Position predictedArrivalPosition = normalizePosition(strings.read(is)); final int bits = is.readShortReverse(); final boolean arrivalCancelled = (bits & 0x10) != 0; final boolean departureCancelled = (bits & 0x20) != 0; is.readShort(); final int firstStopIndex = is.readShortReverse(); final int numStops = is.readShortReverse(); is.reset(); is.skipBytes(disruptionsPtr); String disruptionText = null; if (is.readShortReverse() == 1) { is.reset(); is.skipBytes(disruptionsPtr + 2 + iTrip * 2); int disruptionsOffset = is.readShortReverse(); while (disruptionsOffset != 0) { is.reset(); is.skipBytes(disruptionsPtr + disruptionsOffset); strings.read(is); // "0" final int disruptionLeg = is.readShortReverse(); is.skipBytes(2); // bitmaske strings.read(is); // start of line strings.read(is); // end of line strings.read(is); // id /* final String disruptionTitle = */strings.read(is); final String disruptionShortText = ParserUtils.formatHtml(strings.read(is)); disruptionsOffset = is.readShortReverse(); // next if (iLegs == disruptionLeg) { final int disruptionAttrsIndex = is.readShortReverse(); is.reset(); is.skipBytes(attrsOffset + disruptionAttrsIndex * 4); while (true) { final String key = strings.read(is); if (key == null) break; else if (key.equals("Text")) disruptionText = ParserUtils.resolveEntities(strings.read(is)); else is.skipBytes(2); } if (disruptionShortText != null) disruptionText = disruptionShortText; } } } List intermediateStops = null; if (numStops > 0) { is.reset(); is.skipBytes(tripDetailsPtr + stopsOffset + firstStopIndex * stopsSize); if (stopsSize != 26) throw new IllegalStateException("unhandled stops size: " + stopsSize); intermediateStops = new ArrayList<>(numStops); for (int iStop = 0; iStop < numStops; iStop++) { final long plannedStopDepartureTime = time(is, resDate, tripDayOffset); final Date plannedStopDepartureDate = plannedStopDepartureTime != 0 ? new Date(plannedStopDepartureTime) : null; final long plannedStopArrivalTime = time(is, resDate, tripDayOffset); final Date plannedStopArrivalDate = plannedStopArrivalTime != 0 ? new Date(plannedStopArrivalTime) : null; final Position plannedStopDeparturePosition = normalizePosition(strings.read(is)); final Position plannedStopArrivalPosition = normalizePosition(strings.read(is)); is.readInt(); final long predictedStopDepartureTime = time(is, resDate, tripDayOffset); final Date predictedStopDepartureDate = predictedStopDepartureTime != 0 ? new Date(predictedStopDepartureTime) : null; final long predictedStopArrivalTime = time(is, resDate, tripDayOffset); final Date predictedStopArrivalDate = predictedStopArrivalTime != 0 ? new Date(predictedStopArrivalTime) : null; final Position predictedStopDeparturePosition = normalizePosition(strings.read(is)); final Position predictedStopArrivalPosition = normalizePosition(strings.read(is)); final int stopBits = is.readShortReverse(); final boolean stopArrivalCancelled = (stopBits & 0x10) != 0; final boolean stopDepartureCancelled = (stopBits & 0x20) != 0; is.readShort(); final Location stopLocation = stations.read(is); final boolean validPredictedDate = !dominantPlanStopTime || (plannedStopArrivalDate != null && plannedStopDepartureDate != null); final Stop stop = new Stop(stopLocation, plannedStopArrivalDate, validPredictedDate ? predictedStopArrivalDate : null, plannedStopArrivalPosition, predictedStopArrivalPosition, stopArrivalCancelled, plannedStopDepartureDate, validPredictedDate ? predictedStopDepartureDate : null, plannedStopDeparturePosition, predictedStopDeparturePosition, stopDepartureCancelled); intermediateStops.add(stop); } } final Trip.Leg leg; if (type == 1 /* Fussweg */ || type == 3 /* Uebergang */ || type == 4 /* Uebergang */) { final Trip.Individual.Type individualType; if (routingType == null) individualType = type == 1 ? Trip.Individual.Type.WALK : Trip.Individual.Type.TRANSFER; else if ("FOOT".equals(routingType)) individualType = Trip.Individual.Type.WALK; else if ("BIKE".equals(routingType)) individualType = Trip.Individual.Type.BIKE; else if ("CAR".equals(routingType) || "P+R".equals(routingType)) individualType = Trip.Individual.Type.CAR; else throw new IllegalStateException("unknown routingType: " + routingType); final Date departureTime = new Date( predictedDepartureTime != 0 ? predictedDepartureTime : plannedDepartureTime); final Date arrivalTime = new Date( predictedArrivalTime != 0 ? predictedArrivalTime : plannedArrivalTime); final Trip.Leg lastLeg = legs.size() > 0 ? legs.get(legs.size() - 1) : null; if (lastLeg != null && lastLeg instanceof Trip.Individual && ((Trip.Individual) lastLeg).type == individualType) { final Trip.Individual lastIndividualLeg = (Trip.Individual) legs .remove(legs.size() - 1); leg = new Trip.Individual(individualType, lastIndividualLeg.departure, lastIndividualLeg.departureTime, arrivalLocation, arrivalTime, null, 0); } else { leg = new Trip.Individual(individualType, departureLocation, departureTime, arrivalLocation, arrivalTime, null, 0); } } else if (type == 2) { final Product lineProduct; if (lineOnDemand) lineProduct = Product.ON_DEMAND; else if (lineClass != 0) lineProduct = intToProduct(lineClass); else lineProduct = normalizeType(lineCategory); final Line line = newLine(lineNetwork, lineProduct, normalizeLineName(lineName), lineComment, lineAttrs.toArray(new Line.Attr[0])); final Location direction; if (directionStr != null) { final String[] directionPlaceAndName = splitStationName(directionStr); direction = new Location(LocationType.ANY, null, directionPlaceAndName[0], directionPlaceAndName[1]); } else { direction = null; } final Stop departure = new Stop(departureLocation, true, plannedDepartureTime != 0 ? new Date(plannedDepartureTime) : null, predictedDepartureTime != 0 ? new Date(predictedDepartureTime) : null, plannedDeparturePosition, predictedDeparturePosition, departureCancelled); final Stop arrival = new Stop(arrivalLocation, false, plannedArrivalTime != 0 ? new Date(plannedArrivalTime) : null, predictedArrivalTime != 0 ? new Date(predictedArrivalTime) : null, plannedArrivalPosition, predictedArrivalPosition, arrivalCancelled); leg = new Trip.Public(line, direction, departure, arrival, intermediateStops, null, disruptionText); } else { throw new IllegalStateException("unhandled type: " + type); } legs.add(leg); } final Trip trip = new Trip(connectionId, resDeparture, resArrival, legs, null, null, (int) numChanges); if (realtimeStatus != 2) // Verbindung fällt aus trips.add(trip); } // if result is only one single individual leg, don't query for more final boolean canQueryMore = trips.size() != 1 || trips.get(0).legs.size() != 1 || !(trips.get(0).legs.get(0) instanceof Trip.Individual); result.set(new QueryTripsResult(header, url.toString(), from, via, to, new QueryTripsBinaryContext(requestId, seqNr, ld, bis.getCount(), canQueryMore), trips)); } else { log.debug("Hafas error: {}", errorCode); if (errorCode == 1) { throw new SessionExpiredException(); } else if (errorCode == 2) { // F2: Your search results could not be stored internally. throw new SessionExpiredException(); } else if (errorCode == 8) { result.set(new QueryTripsResult(header, QueryTripsResult.Status.AMBIGUOUS)); return; } else if (errorCode == 13) { // IN13: Our booking system is currently being used by too many users at the same // time. result.set(new QueryTripsResult(header, QueryTripsResult.Status.SERVICE_DOWN)); return; } else if (errorCode == 19) { result.set(new QueryTripsResult(header, QueryTripsResult.Status.SERVICE_DOWN)); return; } else if (errorCode == 207) { // H207: Unfortunately your connection request can currently not be processed. result.set(new QueryTripsResult(header, QueryTripsResult.Status.SERVICE_DOWN)); return; } else if (errorCode == 887) { // H887: Your inquiry was too complex. Please try entering less intermediate stations. result.set(new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS)); return; } else if (errorCode == 890) { // H890: No connections have been found that correspond to your request. It is // possible // that the requested service does not operate from or to the places you stated on the // requested date of travel. result.set(new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS)); return; } else if (errorCode == 891) { // H891: Unfortunately there was no route found. Missing timetable data could be the // reason. result.set(new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS)); return; } else if (errorCode == 892) { // H892: Your inquiry was too complex. Please try entering less intermediate stations. result.set(new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS)); return; } else if (errorCode == 899) { // H899: there was an unsuccessful or incomplete search due to a timetable change. result.set(new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS)); return; } else if (errorCode == 900) { // Unsuccessful or incomplete search (timetable change) result.set(new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS)); return; } else if (errorCode == 9220) { // H9220: Nearby to the given address stations could not be found. result.set(new QueryTripsResult(header, QueryTripsResult.Status.UNRESOLVABLE_ADDRESS)); return; } else if (errorCode == 9240) { // H9240: Unfortunately there was no route found. Perhaps your start or destination is // not // served at all or with the selected means of transport on the required date/time. result.set(new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS)); return; } else if (errorCode == 9260) { // H9260: Unknown departure station result.set(new QueryTripsResult(header, QueryTripsResult.Status.UNKNOWN_FROM)); return; } else if (errorCode == 9280) { // H9280: Unknown intermediate station result.set(new QueryTripsResult(header, QueryTripsResult.Status.UNKNOWN_VIA)); return; } else if (errorCode == 9300) { // H9300: Unknown arrival station result.set(new QueryTripsResult(header, QueryTripsResult.Status.UNKNOWN_TO)); return; } else if (errorCode == 9320) { // The input is incorrect or incomplete result.set(new QueryTripsResult(header, QueryTripsResult.Status.INVALID_DATE)); return; } else if (errorCode == 9360) { // H9360: Unfortunately your connection request can currently not be processed. result.set(new QueryTripsResult(header, QueryTripsResult.Status.INVALID_DATE)); return; } else if (errorCode == 9380) { // H9380: Dep./Arr./Intermed. or equivalent station defined more than once result.set(new QueryTripsResult(header, QueryTripsResult.Status.TOO_CLOSE)); return; } else if (errorCode == 895) { // H895: Departure/Arrival are too near result.set(new QueryTripsResult(header, QueryTripsResult.Status.TOO_CLOSE)); return; } else if (errorCode == 65535) { result.set(new QueryTripsResult(header, QueryTripsResult.Status.SERVICE_DOWN)); return; } else { throw new IllegalStateException("error " + errorCode + " on " + url); } } } }, url); return result.get(); } private Location location(final LittleEndianDataInputStream is, final StringTable strings) throws IOException { final String name = strings.read(is); is.readShort(); final int type = is.readShortReverse(); final int lon = is.readIntReverse(); final int lat = is.readIntReverse(); if (type == 1) { final String[] placeAndName = splitStationName(name); return new Location(LocationType.STATION, null, Point.from1E6(lat, lon), placeAndName[0], placeAndName[1]); } else if (type == 2) { final String[] placeAndName = splitAddress(name); return new Location(LocationType.ADDRESS, null, Point.from1E6(lat, lon), placeAndName[0], placeAndName[1]); } else if (type == 3) { final String[] placeAndName = splitPOI(name); return new Location(LocationType.POI, null, Point.from1E6(lat, lon), placeAndName[0], placeAndName[1]); } else { throw new IllegalStateException("unknown type: " + type + " " + name); } } private long date(final LittleEndianDataInputStream is) throws IOException { final int days = is.readShortReverse(); final Calendar date = new GregorianCalendar(timeZone); date.clear(); date.set(Calendar.YEAR, 1980); date.set(Calendar.DAY_OF_YEAR, days); return date.getTimeInMillis(); } private long time(final LittleEndianDataInputStream is, final long baseDate, final int dayOffset) throws IOException { final int value = is.readShortReverse(); if (value == 0xffff) return 0; final int hours = value / 100; final int minutes = value % 100; if (minutes < 0 || minutes > 60) throw new IllegalStateException("minutes out of range: " + minutes); final Calendar time = new GregorianCalendar(timeZone); time.setTimeInMillis(baseDate); if (time.get(Calendar.HOUR) != 0 || time.get(Calendar.MINUTE) != 0) throw new IllegalStateException("baseDate not on date boundary: " + baseDate); time.add(Calendar.DAY_OF_YEAR, dayOffset); time.set(Calendar.HOUR, hours); time.set(Calendar.MINUTE, minutes); return time.getTimeInMillis(); } private static class StringTable { private Charset encoding = Charsets.US_ASCII; private final byte[] table; public StringTable(final DataInputStream is, final int stringTablePtr, final int length) throws IOException { is.reset(); is.skipBytes(stringTablePtr); table = new byte[length]; is.readFully(table); } public void setEncoding(final Charset encoding) { this.encoding = encoding; } public String read(final LittleEndianDataInputStream is) throws IOException { final int pointer = is.readShortReverse(); if (pointer == 0) return null; if (pointer >= table.length) throw new IllegalStateException( "pointer " + pointer + " cannot exceed strings table size " + table.length); final InputStreamReader reader = new InputStreamReader( new ByteArrayInputStream(table, pointer, table.length - pointer), encoding); try { final StringBuilder builder = new StringBuilder(); int c; while ((c = reader.read()) != 0) builder.append((char) c); return builder.toString().trim(); } finally { reader.close(); } } } private static class CommentTable { private final StringTable strings; private final byte[] table; public CommentTable(final DataInputStream is, final int commentTablePtr, final int length, final StringTable strings) throws IOException { is.reset(); is.skipBytes(commentTablePtr); table = new byte[length]; is.readFully(table); this.strings = strings; } public String[] read(final LittleEndianDataInputStream is) throws IOException { final int pointer = is.readShortReverse(); if (pointer >= table.length) throw new IllegalStateException( "pointer " + pointer + " cannot exceed comments table size " + table.length); final LittleEndianDataInputStream commentsInputStream = new LittleEndianDataInputStream( new ByteArrayInputStream(table, pointer, table.length - pointer)); try { final int numComments = commentsInputStream.readShortReverse(); final String[] comments = new String[numComments]; for (int i = 0; i < numComments; i++) comments[i] = strings.read(commentsInputStream); return comments; } finally { commentsInputStream.close(); } } } private class StationTable { private final StringTable strings; private final byte[] table; public StationTable(final DataInputStream is, final int stationTablePtr, final int length, final StringTable strings) throws IOException { is.reset(); is.skipBytes(stationTablePtr); table = new byte[length]; is.readFully(table); this.strings = strings; } private Location read(final LittleEndianDataInputStream is) throws IOException { final int index = is.readShortReverse(); final int ptr = index * 14; if (ptr >= table.length) throw new IllegalStateException( "pointer " + ptr + " cannot exceed stations table size " + table.length); final LittleEndianDataInputStream stationInputStream = new LittleEndianDataInputStream( new ByteArrayInputStream(table, ptr, 14)); try { final String[] placeAndName = splitStationName(strings.read(stationInputStream)); final int id = stationInputStream.readIntReverse(); final int lon = stationInputStream.readIntReverse(); final int lat = stationInputStream.readIntReverse(); return new Location(LocationType.STATION, id != 0 ? Integer.toString(id) : null, Point.from1E6(lat, lon), placeAndName[0], placeAndName[1]); } finally { stationInputStream.close(); } } } @Override public NearbyLocationsResult queryNearbyLocations(final Set types, final Location location, final int maxDistance, final int maxLocations) throws IOException { if (location.hasCoord()) return nearbyLocationsByCoordinate(types, location.coord, maxDistance, maxLocations); else if (location.type == LocationType.STATION && location.hasId()) return nearbyStationsById(location.id, maxDistance); else throw new IllegalArgumentException("cannot handle: " + location); } protected final NearbyLocationsResult nearbyLocationsByCoordinate(final Set types, final Point coord, final int maxDistance, final int maxLocations) throws IOException { if (types.contains(LocationType.STATION)) { final HttpUrl.Builder url = queryEndpoint.newBuilder().addPathSegment(apiLanguage + "y"); appendJsonNearbyStationsParameters(url, coord, maxDistance, maxLocations); return jsonNearbyLocations(url.build()); } else if (types.contains(LocationType.POI)) { final HttpUrl.Builder url = queryEndpoint.newBuilder().addPathSegment(apiLanguage + "y"); appendJsonNearbyPOIsParameters(url, coord, maxDistance, maxLocations); return jsonNearbyLocations(url.build()); } else { return new NearbyLocationsResult(null, Collections. emptyList()); } } protected NearbyLocationsResult nearbyStationsById(final String id, final int maxDistance) throws IOException { final HttpUrl.Builder url = stationBoardEndpoint.newBuilder().addPathSegment(apiLanguage); appendXmlNearbyStationsParameters(url, id); return xmlNearbyStations(url.build()); } protected final void appendXmlNearbyStationsParameters(final HttpUrl.Builder url, final String stationId) { url.addQueryParameter("productsFilter", allProductsString().toString()); url.addQueryParameter("boardType", "dep"); url.addEncodedQueryParameter("input", ParserUtils.urlEncode(normalizeStationId(stationId), requestUrlEncoding)); url.addQueryParameter("sTI", "1"); url.addQueryParameter("start", "yes"); url.addQueryParameter("hcount", "0"); url.addQueryParameter("L", "vs_java3"); if (clientType != null) url.addEncodedQueryParameter("clientType", ParserUtils.urlEncode(clientType, requestUrlEncoding)); } private static final Pattern P_XML_NEARBY_STATIONS_COARSE = Pattern.compile("\\G<\\s*St\\s*(.*?)/?>(?:\n|\\z)", Pattern.DOTALL); private static final Pattern P_XML_NEARBY_STATIONS_FINE = Pattern.compile("" // + "evaId=\"(\\d+)\"\\s*" // id + "name=\"([^\"]+)\".*?" // name + "(?:x=\"(\\d+)\"\\s*)?" // x + "(?:y=\"(\\d+)\"\\s*)?" // y , Pattern.DOTALL); private static final Pattern P_XML_NEARBY_STATIONS_MESSAGES = Pattern .compile(" stations = new ArrayList<>(); // parse page final Matcher mMessage = P_XML_NEARBY_STATIONS_MESSAGES.matcher(page); if (mMessage.find()) { final String code = mMessage.group(1); final String text = mMessage.group(2); if (code.equals("H730")) // Your input is not valid return new NearbyLocationsResult(null, NearbyLocationsResult.Status.INVALID_ID); if (code.equals("H890")) // No trains in result return new NearbyLocationsResult(null, stations); throw new IllegalArgumentException("unknown error " + code + ", " + text); } final Matcher mCoarse = P_XML_NEARBY_STATIONS_COARSE.matcher(page); while (mCoarse.find()) { final Matcher mFine = P_XML_NEARBY_STATIONS_FINE.matcher(mCoarse.group(1)); if (mFine.matches()) { final String parsedId = mFine.group(1); final String parsedName = ParserUtils.resolveEntities(mFine.group(2)).trim(); final Point parsedCoord; if (mFine.group(3) != null && mFine.group(4) != null) parsedCoord = Point.from1E6(Integer.parseInt(mFine.group(4)), Integer.parseInt(mFine.group(3))); else parsedCoord = null; final String[] placeAndName = splitStationName(parsedName); stations.add( new Location(LocationType.STATION, parsedId, parsedCoord, placeAndName[0], placeAndName[1])); } else { throw new IllegalArgumentException("cannot parse '" + mCoarse.group(1) + "' on " + url); } } return new NearbyLocationsResult(null, stations); } protected void appendJsonNearbyStationsParameters(final HttpUrl.Builder url, final Point coord, final int maxDistance, final int maxStations) { url.addQueryParameter("performLocating", "2"); url.addQueryParameter("tpl", "stop2json"); url.addQueryParameter("look_stopclass", Integer.toString(allProductsInt())); url.addQueryParameter("look_nv", "get_stopweight|yes"); // get_shortjson|yes // get_lines|yes // combinemode|2 // density|80 // get_stopweight|yes // get_infotext|yes url.addQueryParameter("look_x", Integer.toString(coord.getLonAs1E6())); url.addQueryParameter("look_y", Integer.toString(coord.getLatAs1E6())); url.addQueryParameter("look_maxno", Integer.toString(maxStations != 0 ? maxStations : 200)); url.addQueryParameter("look_maxdist", Integer.toString(maxDistance != 0 ? maxDistance : 5000)); } protected void appendJsonNearbyPOIsParameters(final HttpUrl.Builder url, final Point coord, final int maxDistance, final int maxStations) { url.addQueryParameter("performLocating", "4"); url.addQueryParameter("tpl", "poi2json"); url.addQueryParameter("look_pois", ""); // all categories url.addQueryParameter("look_x", Integer.toString(coord.getLonAs1E6())); url.addQueryParameter("look_y", Integer.toString(coord.getLatAs1E6())); url.addQueryParameter("look_maxno", Integer.toString(maxStations != 0 ? maxStations : 200)); url.addQueryParameter("look_maxdist", Integer.toString(maxDistance != 0 ? maxDistance : 5000)); } protected final NearbyLocationsResult jsonNearbyLocations(final HttpUrl url) throws IOException { final CharSequence page = httpClient.get(url); try { final JSONObject head = new JSONObject(page.toString()); final int error = head.getInt("error"); if (error == 0) { final List locations = new LinkedList<>(); final JSONArray aStops = head.optJSONArray("stops"); if (aStops != null) { final int nStops = aStops.length(); for (int i = 0; i < nStops; i++) { final JSONObject stop = aStops.optJSONObject(i); final String id = stop.getString("extId"); // final String name = ParserUtils.resolveEntities(stop.getString("name")); final String urlname = ParserUtils.urlDecode(stop.getString("urlname"), jsonNearbyLocationsEncoding); final int lat = stop.getInt("y"); final int lon = stop.getInt("x"); final int prodclass = stop.optInt("prodclass", -1); final int stopWeight = stop.optInt("stopweight", -1); if (stopWeight != 0) { final String[] placeAndName = splitStationName(urlname); final Set products = prodclass != -1 ? intToProducts(prodclass) : null; locations.add(new Location(LocationType.STATION, id, Point.from1E6(lat, lon), placeAndName[0], placeAndName[1], products)); } } } final JSONArray aPOIs = head.optJSONArray("pois"); if (aPOIs != null) { final int nPOIs = aPOIs.length(); for (int i = 0; i < nPOIs; i++) { final JSONObject poi = aPOIs.optJSONObject(i); final String id = poi.getString("extId"); // final String name = ParserUtils.resolveEntities(stop.getString("name")); final String urlname = ParserUtils.urlDecode(poi.getString("urlname"), jsonNearbyLocationsEncoding); final int lat = poi.getInt("y"); final int lon = poi.getInt("x"); final String[] placeAndName = splitPOI(urlname); locations.add(new Location(LocationType.POI, id, Point.from1E6(lat, lon), placeAndName[0], placeAndName[1])); } } return new NearbyLocationsResult(null, locations); } else { log.debug("Hafas error: {}", error); if (error == 2) return new NearbyLocationsResult(null, NearbyLocationsResult.Status.SERVICE_DOWN); else throw new RuntimeException("unknown error: " + error); } } catch (final JSONException x) { x.printStackTrace(); throw new RuntimeException("cannot parse: '" + page + "' on " + url, x); } } private static final Pattern P_LINE_SBAHN = Pattern.compile("SN?\\d*"); private static final Pattern P_LINE_TRAM = Pattern.compile("STR\\w{0,5}"); private static final Pattern P_LINE_BUS = Pattern.compile("BUS\\w{0,5}"); private static final Pattern P_LINE_TAXI = Pattern.compile("TAX\\w{0,5}"); protected Product normalizeType(final String type) { final String ucType = type.toUpperCase(); // Intercity if ("EC".equals(ucType)) // EuroCity return Product.HIGH_SPEED_TRAIN; if ("ECE".equals(ucType)) // EuroCity-Express return Product.HIGH_SPEED_TRAIN; if ("EN".equals(ucType)) // EuroNight return Product.HIGH_SPEED_TRAIN; if ("D".equals(ucType)) // EuroNight, Sitzwagenabteil return Product.HIGH_SPEED_TRAIN; if ("EIC".equals(ucType)) // Ekspres InterCity, Polen return Product.HIGH_SPEED_TRAIN; if ("ICE".equals(ucType)) // InterCityExpress return Product.HIGH_SPEED_TRAIN; if ("IC".equals(ucType)) // InterCity return Product.HIGH_SPEED_TRAIN; if ("ICT".equals(ucType)) // InterCity return Product.HIGH_SPEED_TRAIN; if ("ICN".equals(ucType)) // InterCityNight return Product.HIGH_SPEED_TRAIN; if ("ICD".equals(ucType)) // Intercity direkt Amsterdam-Breda return Product.HIGH_SPEED_TRAIN; if ("CNL".equals(ucType)) // CityNightLine return Product.HIGH_SPEED_TRAIN; if ("MT".equals(ucType)) // Schnee-Express return Product.HIGH_SPEED_TRAIN; if ("OEC".equals(ucType)) // ÖBB-EuroCity return Product.HIGH_SPEED_TRAIN; if ("OIC".equals(ucType)) // ÖBB-InterCity return Product.HIGH_SPEED_TRAIN; if ("RJ".equals(ucType)) // RailJet, Österreichische Bundesbahnen return Product.HIGH_SPEED_TRAIN; if ("WB".equals(ucType)) // westbahn return Product.HIGH_SPEED_TRAIN; if ("THA".equals(ucType)) // Thalys return Product.HIGH_SPEED_TRAIN; if ("TGV".equals(ucType)) // Train à Grande Vitesse return Product.HIGH_SPEED_TRAIN; if ("DNZ".equals(ucType)) // Nacht-Schnellzug return Product.HIGH_SPEED_TRAIN; if ("AIR".equals(ucType)) // Generic Flight return Product.HIGH_SPEED_TRAIN; if ("ECB".equals(ucType)) // EC, Verona-München return Product.HIGH_SPEED_TRAIN; if ("LYN".equals(ucType)) // Dänemark return Product.HIGH_SPEED_TRAIN; if ("NZ".equals(ucType)) // Schweden, Nacht return Product.HIGH_SPEED_TRAIN; if ("INZ".equals(ucType)) // Nacht return Product.HIGH_SPEED_TRAIN; if ("RHI".equals(ucType)) // ICE return Product.HIGH_SPEED_TRAIN; if ("RHT".equals(ucType)) // TGV return Product.HIGH_SPEED_TRAIN; if ("TGD".equals(ucType)) // TGV return Product.HIGH_SPEED_TRAIN; if ("IRX".equals(ucType)) // IC return Product.HIGH_SPEED_TRAIN; if ("EUR".equals(ucType)) // Eurostar return Product.HIGH_SPEED_TRAIN; if ("ES".equals(ucType)) // Eurostar Italia return Product.HIGH_SPEED_TRAIN; if ("EST".equals(ucType)) // Eurostar Frankreich return Product.HIGH_SPEED_TRAIN; if ("EM".equals(ucType)) // Euromed, Barcelona-Alicante, Spanien return Product.HIGH_SPEED_TRAIN; if ("A".equals(ucType)) // Spain, Highspeed return Product.HIGH_SPEED_TRAIN; if ("AVE".equals(ucType)) // Alta Velocidad Española, Spanien return Product.HIGH_SPEED_TRAIN; if ("ARC".equals(ucType)) // Arco (Renfe), Spanien return Product.HIGH_SPEED_TRAIN; if ("ALS".equals(ucType)) // Alaris (Renfe), Spanien return Product.HIGH_SPEED_TRAIN; if ("ATR".equals(ucType)) // Altaria (Renfe), Spanien return Product.REGIONAL_TRAIN; if ("TAL".equals(ucType)) // Talgo, Spanien return Product.HIGH_SPEED_TRAIN; if ("TLG".equals(ucType)) // Spanien, Madrid return Product.HIGH_SPEED_TRAIN; if ("HOT".equals(ucType)) // Spanien, Nacht return Product.HIGH_SPEED_TRAIN; if ("X2".equals(ucType)) // X2000 Neigezug, Schweden return Product.HIGH_SPEED_TRAIN; if ("X".equals(ucType)) // InterConnex return Product.HIGH_SPEED_TRAIN; if ("FYR".equals(ucType)) // Fyra, Amsterdam-Schiphol-Rotterdam return Product.HIGH_SPEED_TRAIN; if ("FYRA".equals(ucType)) // Fyra, Amsterdam-Schiphol-Rotterdam return Product.HIGH_SPEED_TRAIN; if ("SC".equals(ucType)) // SuperCity, Tschechien return Product.HIGH_SPEED_TRAIN; if ("LE".equals(ucType)) // LEO Express, Prag return Product.HIGH_SPEED_TRAIN; if ("FLUG".equals(ucType)) return Product.HIGH_SPEED_TRAIN; if ("TLK".equals(ucType)) // Tanie Linie Kolejowe, Polen return Product.HIGH_SPEED_TRAIN; if ("PKP".equals(ucType)) // Polskie Koleje Państwowe (Polnische Staatsbahnen) return Product.HIGH_SPEED_TRAIN; if ("EIP".equals(ucType)) // Express Intercity Premium return Product.HIGH_SPEED_TRAIN; if ("INT".equals(ucType)) // Zürich-Brüssel - Budapest-Istanbul return Product.HIGH_SPEED_TRAIN; if ("HKX".equals(ucType)) // Hamburg-Koeln-Express return Product.HIGH_SPEED_TRAIN; if ("LOC".equals(ucType)) // Locomore return Product.HIGH_SPEED_TRAIN; if ("NJ".equals(ucType)) // NightJet return Product.HIGH_SPEED_TRAIN; if ("FLX".equals(ucType)) return Product.HIGH_SPEED_TRAIN; if ("RJX".equals(ucType)) // railjet xpress return Product.HIGH_SPEED_TRAIN; if ("ICL".equals(ucType)) // InterCity Lyn, Denmark return Product.HIGH_SPEED_TRAIN; if ("FR".equals(ucType)) // Frecciarossa, Italy return Product.HIGH_SPEED_TRAIN; if ("FA".equals(ucType)) // Frecciargento, Italy return Product.HIGH_SPEED_TRAIN; // Regional if ("ZUG".equals(ucType)) // Generic Train return Product.REGIONAL_TRAIN; if ("R".equals(ucType)) // Generic Regional Train return Product.REGIONAL_TRAIN; if ("DPN".equals(ucType)) // Dritter Personen Nahverkehr return Product.REGIONAL_TRAIN; if ("RB".equals(ucType)) // RegionalBahn return Product.REGIONAL_TRAIN; if ("RE".equals(ucType)) // RegionalExpress return Product.REGIONAL_TRAIN; if ("ER".equals(ucType)) return Product.REGIONAL_TRAIN; if ("DB".equals(ucType)) return Product.REGIONAL_TRAIN; if ("IR".equals(ucType)) // Interregio return Product.REGIONAL_TRAIN; if ("IRE".equals(ucType)) // Interregio Express return Product.REGIONAL_TRAIN; if ("HEX".equals(ucType)) // Harz-Berlin-Express, Veolia return Product.REGIONAL_TRAIN; if ("WFB".equals(ucType)) // Westfalenbahn return Product.REGIONAL_TRAIN; if ("RT".equals(ucType)) // RegioTram return Product.REGIONAL_TRAIN; if ("REX".equals(ucType)) // RegionalExpress, Österreich return Product.REGIONAL_TRAIN; if ("OS".equals(ucType)) // Osobný vlak, Slovakia oder Osobní vlak, Czech Republic return Product.REGIONAL_TRAIN; if ("SP".equals(ucType)) // Spěšný vlak, Czech Republic return Product.REGIONAL_TRAIN; if ("RX".equals(ucType)) // Express, Czech Republic return Product.REGIONAL_TRAIN; if ("EZ".equals(ucType)) // ÖBB ErlebnisBahn return Product.REGIONAL_TRAIN; if ("ARZ".equals(ucType)) // Auto-Reisezug Brig - Iselle di Trasquera return Product.REGIONAL_TRAIN; if ("OE".equals(ucType)) // Ostdeutsche Eisenbahn return Product.REGIONAL_TRAIN; if ("MR".equals(ucType)) // Märkische Regionalbahn return Product.REGIONAL_TRAIN; if ("PE".equals(ucType)) // Prignitzer Eisenbahn GmbH return Product.REGIONAL_TRAIN; if ("NE".equals(ucType)) // NEB Betriebsgesellschaft mbH return Product.REGIONAL_TRAIN; if ("MRB".equals(ucType)) // Mitteldeutsche Regiobahn return Product.REGIONAL_TRAIN; if ("ERB".equals(ucType)) // eurobahn (Keolis Deutschland) return Product.REGIONAL_TRAIN; if ("HLB".equals(ucType)) // Hessische Landesbahn return Product.REGIONAL_TRAIN; if ("VIA".equals(ucType)) return Product.REGIONAL_TRAIN; if ("HSB".equals(ucType)) // Harzer Schmalspurbahnen return Product.REGIONAL_TRAIN; if ("OSB".equals(ucType)) // Ortenau-S-Bahn return Product.REGIONAL_TRAIN; if ("VBG".equals(ucType)) // Vogtlandbahn return Product.REGIONAL_TRAIN; if ("AKN".equals(ucType)) // AKN Eisenbahn AG return Product.REGIONAL_TRAIN; if ("OLA".equals(ucType)) // Ostseeland Verkehr return Product.REGIONAL_TRAIN; if ("UBB".equals(ucType)) // Usedomer Bäderbahn return Product.REGIONAL_TRAIN; if ("PEG".equals(ucType)) // Prignitzer Eisenbahn return Product.REGIONAL_TRAIN; if ("NWB".equals(ucType)) // NordWestBahn return Product.REGIONAL_TRAIN; if ("CAN".equals(ucType)) // cantus Verkehrsgesellschaft return Product.REGIONAL_TRAIN; if ("BRB".equals(ucType)) // ABELLIO Rail return Product.REGIONAL_TRAIN; if ("SBB".equals(ucType)) // Schweizerische Bundesbahnen return Product.REGIONAL_TRAIN; if ("VEC".equals(ucType)) // vectus Verkehrsgesellschaft return Product.REGIONAL_TRAIN; if ("TLX".equals(ucType)) // Trilex (Vogtlandbahn) return Product.REGIONAL_TRAIN; if ("TL".equals(ucType)) // Trilex (Vogtlandbahn) return Product.REGIONAL_TRAIN; if ("HZL".equals(ucType)) // Hohenzollerische Landesbahn return Product.REGIONAL_TRAIN; if ("ABR".equals(ucType)) // Bayerische Regiobahn return Product.REGIONAL_TRAIN; if ("CB".equals(ucType)) // City Bahn Chemnitz return Product.REGIONAL_TRAIN; if ("WEG".equals(ucType)) // Württembergische Eisenbahn-Gesellschaft return Product.REGIONAL_TRAIN; if ("NEB".equals(ucType)) // Niederbarnimer Eisenbahn return Product.REGIONAL_TRAIN; if ("ME".equals(ucType)) // metronom Eisenbahngesellschaft return Product.REGIONAL_TRAIN; if ("MER".equals(ucType)) // metronom regional return Product.REGIONAL_TRAIN; if ("ALX".equals(ucType)) // Arriva-Länderbahn-Express return Product.REGIONAL_TRAIN; if ("EB".equals(ucType)) // Erfurter Bahn return Product.REGIONAL_TRAIN; if ("EBX".equals(ucType)) // Erfurter Bahn return Product.REGIONAL_TRAIN; if ("VEN".equals(ucType)) // Rhenus Veniro return Product.REGIONAL_TRAIN; if ("BOB".equals(ucType)) // Bayerische Oberlandbahn return Product.REGIONAL_TRAIN; if ("SBS".equals(ucType)) // Städtebahn Sachsen return Product.REGIONAL_TRAIN; if ("SES".equals(ucType)) // Städtebahn Sachsen Express return Product.REGIONAL_TRAIN; if ("EVB".equals(ucType)) // Eisenbahnen und Verkehrsbetriebe Elbe-Weser return Product.REGIONAL_TRAIN; if ("STB".equals(ucType)) // Süd-Thüringen-Bahn return Product.REGIONAL_TRAIN; if ("STX".equals(ucType)) // Süd-Thüringen-Bahn return Product.REGIONAL_TRAIN; if ("AG".equals(ucType)) // Ingolstadt-Landshut return Product.REGIONAL_TRAIN; if ("PRE".equals(ucType)) // Pressnitztalbahn return Product.REGIONAL_TRAIN; if ("DBG".equals(ucType)) // Döllnitzbahn GmbH return Product.REGIONAL_TRAIN; if ("SHB".equals(ucType)) // Schleswig-Holstein-Bahn return Product.REGIONAL_TRAIN; if ("NOB".equals(ucType)) // Nord-Ostsee-Bahn return Product.REGIONAL_TRAIN; if ("RTB".equals(ucType)) // Rurtalbahn return Product.REGIONAL_TRAIN; if ("BLB".equals(ucType)) // Berchtesgadener Land Bahn return Product.REGIONAL_TRAIN; if ("NBE".equals(ucType)) // Nordbahn Eisenbahngesellschaft return Product.REGIONAL_TRAIN; if ("SOE".equals(ucType)) // Sächsisch-Oberlausitzer Eisenbahngesellschaft return Product.REGIONAL_TRAIN; if ("SDG".equals(ucType)) // Sächsische Dampfeisenbahngesellschaft return Product.REGIONAL_TRAIN; if ("VE".equals(ucType)) // Lutherstadt Wittenberg return Product.REGIONAL_TRAIN; if ("DAB".equals(ucType)) // Daadetalbahn return Product.REGIONAL_TRAIN; if ("WTB".equals(ucType)) // Wutachtalbahn e.V. return Product.REGIONAL_TRAIN; if ("BE".equals(ucType)) // Grensland-Express return Product.REGIONAL_TRAIN; if ("ARR".equals(ucType)) // Ostfriesland return Product.REGIONAL_TRAIN; if ("HTB".equals(ucType)) // Hörseltalbahn return Product.REGIONAL_TRAIN; if ("FEG".equals(ucType)) // Freiberger Eisenbahngesellschaft return Product.REGIONAL_TRAIN; if ("NEG".equals(ucType)) // Norddeutsche Eisenbahngesellschaft Niebüll return Product.REGIONAL_TRAIN; if ("RBG".equals(ucType)) // Regental Bahnbetriebs GmbH return Product.REGIONAL_TRAIN; if ("MBB".equals(ucType)) // Mecklenburgische Bäderbahn Molli return Product.REGIONAL_TRAIN; if ("VEB".equals(ucType)) // Vulkan-Eifel-Bahn Betriebsgesellschaft return Product.REGIONAL_TRAIN; if ("LEO".equals(ucType)) // Chiemgauer Lokalbahn return Product.REGIONAL_TRAIN; if ("VX".equals(ucType)) // Vogtland Express return Product.REGIONAL_TRAIN; if ("MSB".equals(ucType)) // Mainschleifenbahn return Product.REGIONAL_TRAIN; if ("P".equals(ucType)) // Kasbachtalbahn return Product.REGIONAL_TRAIN; if ("ÖBA".equals(ucType)) // Öchsle-Bahn Betriebsgesellschaft return Product.REGIONAL_TRAIN; if ("KTB".equals(ucType)) // Kandertalbahn return Product.REGIONAL_TRAIN; if ("ERX".equals(ucType)) // erixx return Product.REGIONAL_TRAIN; if ("ATZ".equals(ucType)) // Autotunnelzug return Product.REGIONAL_TRAIN; if ("ATB".equals(ucType)) // Autoschleuse Tauernbahn return Product.REGIONAL_TRAIN; if ("CAT".equals(ucType)) // City Airport Train return Product.REGIONAL_TRAIN; if ("EXTRA".equals(ucType) || "EXT".equals(ucType)) // Extrazug return Product.REGIONAL_TRAIN; if ("KD".equals(ucType)) // Koleje Dolnośląskie (Niederschlesische Eisenbahn) return Product.REGIONAL_TRAIN; if ("KM".equals(ucType)) // Koleje Mazowieckie return Product.REGIONAL_TRAIN; if ("EX".equals(ucType)) // Polen return Product.REGIONAL_TRAIN; if ("PCC".equals(ucType)) // PCC Rail, Polen return Product.REGIONAL_TRAIN; if ("ZR".equals(ucType)) // ZSR (Slovakian Republic Railways) return Product.REGIONAL_TRAIN; if ("RNV".equals(ucType)) // Rhein-Neckar-Verkehr GmbH return Product.REGIONAL_TRAIN; if ("DWE".equals(ucType)) // Dessau-Wörlitzer Eisenbahn return Product.REGIONAL_TRAIN; if ("BKB".equals(ucType)) // Buckower Kleinbahn return Product.REGIONAL_TRAIN; if ("GEX".equals(ucType)) // Glacier Express return Product.REGIONAL_TRAIN; if ("M".equals(ucType)) // Meridian return Product.REGIONAL_TRAIN; if ("WBA".equals(ucType)) // Waldbahn return Product.REGIONAL_TRAIN; if ("BEX".equals(ucType)) // Bernina Express return Product.REGIONAL_TRAIN; if ("VAE".equals(ucType)) // Voralpen-Express return Product.REGIONAL_TRAIN; if ("OPB".equals(ucType)) // oberpfalzbahn return Product.REGIONAL_TRAIN; if ("OPX".equals(ucType)) // oberpfalz-express return Product.REGIONAL_TRAIN; if ("TER".equals(ucType)) // Transport express régional return Product.REGIONAL_TRAIN; if ("ENO".equals(ucType)) return Product.REGIONAL_TRAIN; if ("THU".equals(ucType)) // Thurbo AG return Product.REGIONAL_TRAIN; if ("GW".equals(ucType)) // gwtr.cz return Product.REGIONAL_TRAIN; if ("SE".equals(ucType)) // ABELLIO Rail Mitteldeutschland GmbH return Product.REGIONAL_TRAIN; if ("UEX".equals(ucType)) // Slovenia return Product.REGIONAL_TRAIN; if ("KW".equals(ucType)) // Koleje Wielkopolskie return Product.REGIONAL_TRAIN; if ("KS".equals(ucType)) // Koleje Śląskie return Product.REGIONAL_TRAIN; if ("KML".equals(ucType)) // Koleje Malopolskie return Product.REGIONAL_TRAIN; // Suburban Trains if (P_LINE_SBAHN.matcher(ucType).matches()) // Generic (Night) S-Bahn return Product.SUBURBAN_TRAIN; if ("S-BAHN".equals(ucType)) return Product.SUBURBAN_TRAIN; if ("BSB".equals(ucType)) // Breisgau S-Bahn return Product.SUBURBAN_TRAIN; if ("SWE".equals(ucType)) // Südwestdeutsche Verkehrs-AG, Ortenau-S-Bahn return Product.SUBURBAN_TRAIN; if ("RER".equals(ucType)) // Réseau Express Régional, Frankreich return Product.SUBURBAN_TRAIN; if ("WKD".equals(ucType)) // Warszawska Kolej Dojazdowa (Warsaw Suburban Railway) return Product.SUBURBAN_TRAIN; if ("SKM".equals(ucType)) // Szybka Kolej Miejska Tricity return Product.SUBURBAN_TRAIN; if ("SKW".equals(ucType)) // Szybka Kolej Miejska Warschau return Product.SUBURBAN_TRAIN; if ("LKA".equals(ucType)) // Łódzka Kolej Aglomeracyjna return Product.SUBURBAN_TRAIN; // Subway if ("U".equals(ucType)) // Generic U-Bahn return Product.SUBWAY; if ("MET".equals(ucType)) return Product.SUBWAY; if ("METRO".equals(ucType)) return Product.SUBWAY; // Tram if (P_LINE_TRAM.matcher(ucType).matches()) // Generic Tram return Product.TRAM; if ("NFT".equals(ucType)) // Niederflur-Tram return Product.TRAM; if ("TRAM".equals(ucType)) return Product.TRAM; if ("TRA".equals(ucType)) return Product.TRAM; if ("WLB".equals(ucType)) // Wiener Lokalbahnen return Product.TRAM; if ("STRWLB".equals(ucType)) // Wiener Lokalbahnen return Product.TRAM; if ("SCHW-B".equals(ucType)) // Schwebebahn, gilt als "Straßenbahn besonderer Bauart" return Product.TRAM; // Bus if (P_LINE_BUS.matcher(ucType).matches()) // Generic Bus return Product.BUS; if ("NFB".equals(ucType)) // Niederflur-Bus return Product.BUS; if ("SEV".equals(ucType)) // Schienen-Ersatz-Verkehr return Product.BUS; if ("BUSSEV".equals(ucType)) // Schienen-Ersatz-Verkehr return Product.BUS; if ("BSV".equals(ucType)) // Bus SEV return Product.BUS; if ("FB".equals(ucType)) // Fernbus? Luxemburg-Saarbrücken return Product.BUS; if ("EXB".equals(ucType)) // Expressbus München-Prag? return Product.BUS; if ("ICB".equals(ucType)) // ÖBB ICBus return Product.BUS; if ("TRO".equals(ucType)) // Trolleybus return Product.BUS; if ("RFB".equals(ucType)) // Rufbus return Product.BUS; if ("RUF".equals(ucType)) // Rufbus return Product.BUS; if (P_LINE_TAXI.matcher(ucType).matches()) // Generic Taxi return Product.BUS; if ("RFT".equals(ucType)) // Ruftaxi return Product.BUS; if ("LT".equals(ucType)) // Linien-Taxi return Product.BUS; if ("NB".equals(ucType)) // Nachtbus Zürich return Product.BUS; if ("POSTBUS".equals(ucType)) return Product.BUS; // Phone if ("RUFBUS".equals(ucType)) return Product.ON_DEMAND; if (ucType.startsWith("AST")) // Anruf-Sammel-Taxi return Product.ON_DEMAND; if (ucType.startsWith("ALT")) // Anruf-Linien-Taxi return Product.ON_DEMAND; if (ucType.startsWith("BUXI")) // Bus-Taxi (Schweiz) return Product.ON_DEMAND; if ("TB".equals(ucType)) // Taxi-Bus? return Product.ON_DEMAND; // Ferry if ("SCHIFF".equals(ucType)) return Product.FERRY; if ("FÄHRE".equals(ucType)) return Product.FERRY; if ("FÄH".equals(ucType)) return Product.FERRY; if ("FAE".equals(ucType)) return Product.FERRY; if ("SCH".equals(ucType)) // Schiff return Product.FERRY; if ("AS".equals(ucType)) // SyltShuttle return Product.FERRY; if ("AZS".equals(ucType)) // Autozug Sylt Shuttle return Product.FERRY; if ("KAT".equals(ucType)) // Katamaran, e.g. Friedrichshafen - Konstanz return Product.FERRY; if ("BAT".equals(ucType)) // Boots Anlege Terminal? return Product.FERRY; if ("BAV".equals(ucType)) // Boots Anlege? return Product.FERRY; // Cable Car if ("SEILBAHN".equals(ucType)) return Product.CABLECAR; if ("SB".equals(ucType)) // Seilbahn return Product.CABLECAR; if ("ZAHNR".equals(ucType)) // Zahnradbahn, u.a. Zugspitzbahn return Product.CABLECAR; if ("GB".equals(ucType)) // Gondelbahn return Product.CABLECAR; if ("LB".equals(ucType)) // Luftseilbahn return Product.CABLECAR; if ("FUN".equals(ucType)) // Funiculaire (Standseilbahn) return Product.CABLECAR; if ("SL".equals(ucType)) // Sessel-Lift return Product.CABLECAR; // Unknown product if ("E".equals(ucType)) return null; throw new IllegalStateException("cannot normalize type '" + type + "'"); } private static final Pattern P_NORMALIZE_LINE_NAME_BUS = Pattern.compile("bus\\s+(.*)", Pattern.CASE_INSENSITIVE); protected static final Pattern P_NORMALIZE_LINE = Pattern .compile("([A-Za-zßÄÅäáàâåéèêíìîÖöóòôÜüúùûØ/]+)[\\s-]*([^#]*).*"); protected String normalizeLineName(final String lineName) { final Matcher mBus = P_NORMALIZE_LINE_NAME_BUS.matcher(lineName); if (mBus.matches()) return mBus.group(1); final Matcher m = P_NORMALIZE_LINE.matcher(lineName); if (m.matches()) return m.group(1) + m.group(2); return lineName; } private static final Pattern P_NORMALIZE_LINE_ADMINISTRATION = Pattern.compile("([^_]*)_*"); private final String normalizeLineAdministration(final String administration) { if (administration == null) return null; final Matcher m = P_NORMALIZE_LINE_ADMINISTRATION.matcher(administration); if (m.find()) return m.group(1); else return administration; } private static final Pattern P_CATEGORY_FROM_NAME = Pattern.compile("([A-Za-zßÄÅäáàâåéèêíìîÖöóòôÜüúùûØ]+).*"); protected final String categoryFromName(final String lineName) { final Matcher m = P_CATEGORY_FROM_NAME.matcher(lineName); if (m.matches()) return m.group(1); else return lineName; } private static final Pattern P_NORMALIZE_LINE_BUS = Pattern.compile("(?:Bus|BUS)\\s*(.*)"); private static final Pattern P_NORMALIZE_LINE_TRAM = Pattern.compile("(?:Tram|Tra|Str|STR)\\s*(.*)"); protected Line parseLine(final String type, final String normalizedName, final boolean wheelchairAccess) { if (normalizedName != null) { final Matcher mBus = P_NORMALIZE_LINE_BUS.matcher(normalizedName); if (mBus.matches()) return newLine(Product.BUS, mBus.group(1), null); final Matcher mTram = P_NORMALIZE_LINE_TRAM.matcher(normalizedName); if (mTram.matches()) return newLine(Product.TRAM, mTram.group(1), null); } final Product normalizedType = normalizeType(type); final Line.Attr[] attrs; if (wheelchairAccess) attrs = new Line.Attr[] { Line.Attr.WHEEL_CHAIR_ACCESS }; else attrs = new Line.Attr[0]; if (normalizedName != null) { final Matcher m = P_NORMALIZE_LINE.matcher(normalizedName); final String strippedLine = m.matches() ? m.group(1) + m.group(2) : normalizedName; return newLine(normalizedType, strippedLine, null, attrs); } else { return newLine(normalizedType, null, null, attrs); } } protected static final Pattern P_NORMALIZE_LINE_AND_TYPE = Pattern.compile("([^#]*)#(.*)"); private static final Pattern P_LINE_NUMBER = Pattern.compile("\\d{2,5}"); private static final Pattern P_LINE_SUBWAY = Pattern.compile("U\\d{1,2}"); private static final Pattern P_LINE_RUSSIA = Pattern.compile( "\\d{3}(?:AJ|BJ|CJ|DJ|EJ|FJ|GJ|IJ|KJ|LJ|NJ|MJ|OJ|RJ|SJ|TJ|UJ|VJ|ZJ|CH|KH|ZH|EI|JA|JI|MZ|SH|SZ|PC|Y|YJ)"); protected Line parseLineAndType(final String lineAndType) { final Matcher mLineAndType = P_NORMALIZE_LINE_AND_TYPE.matcher(lineAndType); if (mLineAndType.matches()) { final String number = mLineAndType.group(1); final String type = mLineAndType.group(2); if (type.length() == 0) { if (number.length() == 0) return newLine(null, null, null); if (P_LINE_NUMBER.matcher(number).matches()) return newLine(null, number, null); if (P_LINE_SUBWAY.matcher(number).matches()) return newLine(Product.SUBWAY, number, null); if (P_LINE_RUSSIA.matcher(number).matches()) return newLine(Product.REGIONAL_TRAIN, number, null); } else { final Product normalizedType = normalizeType(type); if (normalizedType != null) { if (normalizedType == Product.BUS) { final Matcher mBus = P_NORMALIZE_LINE_BUS.matcher(number); if (mBus.matches()) return newLine(Product.BUS, mBus.group(1), null); } if (normalizedType == Product.TRAM) { final Matcher mTram = P_NORMALIZE_LINE_TRAM.matcher(number); if (mTram.matches()) return newLine(Product.TRAM, mTram.group(1), null); } } return newLine(normalizedType, number.replaceAll("\\s+", ""), null); } throw new IllegalStateException( "cannot normalize type '" + type + "' number '" + number + "' line#type '" + lineAndType + "'"); } throw new IllegalStateException("cannot normalize line#type '" + lineAndType + "'"); } protected Line newLine(final Product product, final String normalizedName, final String comment, final Line.Attr... attrs) { return newLine(null, product, normalizedName, comment, attrs); } protected Line newLine(final String network, final Product product, final String normalizedName, final String comment, final Line.Attr... attrs) { if (attrs.length == 0) { return new Line(null, network, product, normalizedName, lineStyle(network, product, normalizedName), comment); } else { final Set attrSet = new HashSet<>(); for (final Line.Attr attr : attrs) attrSet.add(attr); return new Line(null, network, product, normalizedName, lineStyle(network, product, normalizedName), attrSet, comment); } } }




© 2015 - 2024 Weber Informatics LLC | Privacy Policy