Java tutorial
/* * Copyright 2010-2015 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 <http://www.gnu.org/licenses/>. */ 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.io.Reader; import java.nio.charset.Charset; import java.util.ArrayList; import java.util.Calendar; import java.util.Collection; import java.util.Collections; import java.util.Date; import java.util.EnumSet; 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.regex.Matcher; import java.util.regex.Pattern; 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.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; /** * @author Andreas Schildbach */ public abstract class AbstractHafasProvider extends AbstractNetworkProvider { protected final static String SERVER_PRODUCT = "hafas"; private static final String REQC_PROD = "hafas"; protected static final int DEFAULT_MAX_DEPARTURES = 100; protected final String stationBoardEndpoint; protected final String getStopEndpoint; protected final String queryEndpoint; private final int numProductBits; private @Nullable String accessId = null; private @Nullable String clientType = "ANDROID"; private Charset jsonGetStopsEncoding; private boolean jsonGetStopsUseWeight = true; private Charset jsonNearbyLocationsEncoding; private boolean dominantPlanStopTime = false; private boolean useIso8601 = false; private @Nullable String extXmlEndpoint = null; 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; } public boolean canQueryLater() { return laterContext != null; } 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; } public boolean canQueryLater() { return canQueryMore; } public boolean canQueryEarlier() { return canQueryMore; } } public AbstractHafasProvider(final NetworkId network, final String stationBoardEndpoint, final String getStopEndpoint, final String queryEndpoint, final int numProductBits) { this(network, stationBoardEndpoint, getStopEndpoint, queryEndpoint, numProductBits, Charsets.ISO_8859_1); } public AbstractHafasProvider(final NetworkId network, final String stationBoardEndpoint, final String getStopEndpoint, final String queryEndpoint, final int numProductBits, final Charset jsonEncoding) { super(network); this.stationBoardEndpoint = stationBoardEndpoint; this.getStopEndpoint = getStopEndpoint; this.queryEndpoint = queryEndpoint; this.numProductBits = numProductBits; this.jsonGetStopsEncoding = jsonEncoding; this.jsonNearbyLocationsEncoding = jsonEncoding; } protected void setClientType(final String clientType) { this.clientType = clientType; } protected void setAccessId(final String accessId) { this.accessId = accessId; } protected void setDominantPlanStopTime(final boolean dominantPlanStopTime) { this.dominantPlanStopTime = dominantPlanStopTime; } protected void setJsonGetStopsEncoding(final Charset jsonGetStopsEncoding) { this.jsonGetStopsEncoding = jsonGetStopsEncoding; } protected void setJsonGetStopsUseWeight(final boolean jsonGetStopsUseWeight) { this.jsonGetStopsUseWeight = jsonGetStopsUseWeight; } protected void setJsonNearbyLocationsEncoding(final Charset jsonNearbyLocationsEncoding) { this.jsonNearbyLocationsEncoding = jsonNearbyLocationsEncoding; } protected void setUseIso8601(final boolean useIso8601) { this.useIso8601 = useIso8601; } protected void setExtXmlEndpoint(final String extXmlEndpoint) { this.extXmlEndpoint = extXmlEndpoint; } protected void setStationBoardHasStationTable(final boolean stationBoardHasStationTable) { this.stationBoardHasStationTable = stationBoardHasStationTable; } protected void setStationBoardHasLocation(final boolean stationBoardHasLocation) { this.stationBoardHasLocation = stationBoardHasLocation; } protected void setStationBoardCanDoEquivs(final boolean canDoEquivs) { this.stationBoardCanDoEquivs = canDoEquivs; } @Override protected boolean hasCapability(final Capability capability) { return true; } protected final String allProductsString() { final StringBuilder allProducts = new StringBuilder(numProductBits); for (int i = 0; i < numProductBits; i++) allProducts.append('1'); return allProducts.toString(); } protected final int allProductsInt() { return (1 << numProductBits) - 1; } protected Product intToProduct(final int value) { return null; } protected abstract void setProductBits(StringBuilder productBits, Product product); protected static final Pattern P_SPLIT_NAME_FIRST_COMMA = Pattern.compile("([^,]*), (.*)"); protected static final Pattern P_SPLIT_NAME_LAST_COMMA = Pattern.compile("(.*), ([^,]*)"); protected static final Pattern P_SPLIT_NAME_PAREN = Pattern.compile("(.*) \\((.{3,}?)\\)"); protected String[] splitStationName(final String name) { return new String[] { null, name }; } protected String[] splitPOI(final String poi) { return new String[] { null, poi }; } protected String[] splitAddress(final String address) { return new String[] { null, address }; } private final String wrapReqC(final CharSequence request, final Charset encoding) { return "<?xml version=\"1.0\" encoding=\"" + (encoding != null ? encoding.name() : "iso-8859-1") + "\"?>" // + "<ReqC ver=\"1.1\" prod=\"" + REQC_PROD + "\" lang=\"DE\"" + (accessId != null ? " accessId=\"" + accessId + "\"" : "") + ">" // + request // + "</ReqC>"; } 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 String[] placeAndName = splitStationName(name); return new Location(LocationType.STATION, id, y, x, 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"); return new Location(LocationType.POI, null, y, x, 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 String[] placeAndName = splitAddress(name); return new Location(LocationType.ADDRESS, null, y, x, 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); } public SuggestLocationsResult suggestLocations(final CharSequence constraint) throws IOException { final StringBuilder uri = new StringBuilder(getStopEndpoint); appendJsonGetStopsParameters(uri, constraint, 0); return jsonGetStops(uri.toString()); } protected void appendJsonGetStopsParameters(final StringBuilder uri, final CharSequence constraint, final int maxStops) { uri.append("?getstop=1"); uri.append("&REQ0JourneyStopsS0A=255"); uri.append("&REQ0JourneyStopsS0G=") .append(ParserUtils.urlEncode(constraint.toString(), jsonGetStopsEncoding)).append("?"); if (maxStops > 0) uri.append("&REQ0JourneyStopsB=").append(maxStops); uri.append("&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 String uri) throws IOException { final CharSequence page = httpClient.get(uri, null, jsonGetStopsEncoding); final Matcher mJson = P_AJAX_GET_STOPS_JSON.matcher(page); if (mJson.matches()) { final String json = mJson.group(1); final List<SuggestedLocation> locations = new ArrayList<SuggestedLocation>(); 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, lat, lon, placeAndName[0], placeAndName[1]); } else if (type == 2) // address { final String[] placeAndName = splitAddress(value); location = new Location(LocationType.ADDRESS, null, lat, lon, placeAndName[0], placeAndName[1]); } else if (type == 4) // poi { final String[] placeAndName = splitPOI(value); location = new Location(LocationType.POI, localId, lat, lon, placeAndName[0], placeAndName[1]); } else if (type == 128) // crossing { final String[] placeAndName = splitAddress(value); location = new Location(LocationType.ADDRESS, localId, 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 " + uri); } 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 " + uri, x); } } else { throw new RuntimeException("cannot parse: '" + page + "' on " + uri); } } public QueryDeparturesResult queryDepartures(final String stationId, final @Nullable Date time, final int maxDepartures, final boolean equivs) throws IOException { checkNotNull(Strings.emptyToNull(stationId)); final StringBuilder uri = new StringBuilder(stationBoardEndpoint); appendXmlStationBoardParameters(uri, time, stationId, maxDepartures, equivs, "vs_java3"); return xmlStationBoard(uri.toString(), stationId); } protected void appendXmlStationBoardParameters(final StringBuilder uri, final @Nullable Date time, final String stationId, final int maxDepartures, final boolean equivs, final @Nullable String styleSheet) { uri.append("?productsFilter=").append(allProductsString()); uri.append("&boardType=dep"); if (stationBoardCanDoEquivs) uri.append("&disableEquivs=").append(equivs ? "0" : "1"); uri.append("&maxJourneys=").append(maxDepartures > 0 ? maxDepartures : DEFAULT_MAX_DEPARTURES); uri.append("&input=").append(normalizeStationId(stationId)); appendDateTimeParameters(uri, time, "date", "time"); if (clientType != null) uri.append("&clientType=").append(ParserUtils.urlEncode(clientType)); if (styleSheet != null) uri.append("&L=").append(styleSheet); uri.append("&hcount=0"); // prevents showing old departures uri.append("&start=yes"); } protected void appendDateTimeParameters(final StringBuilder uri, 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); uri.append('&').append(dateParamName).append('='); uri.append( ParserUtils.urlEncode(useIso8601 ? String.format(Locale.ENGLISH, "%04d-%02d-%02d", year, month, day) : String.format(Locale.ENGLISH, "%02d.%02d.%02d", day, month, year - 2000))); uri.append('&').append(timeParamName).append('='); uri.append(ParserUtils.urlEncode(String.format(Locale.ENGLISH, "%02d:%02d", hour, minute))); } private static final Pattern P_XML_STATION_BOARD_DELAY = Pattern .compile("(?:-|k\\.A\\.?|cancel|\\+?\\s*(\\d+))"); protected final QueryDeparturesResult xmlStationBoard(final String uri, final String stationId) throws IOException { final String normalizedStationId = normalizeStationId(stationId); StringReplaceReader reader = null; try { // work around unparsable XML reader = new StringReplaceReader( new InputStreamReader(httpClient.getInputStream(uri), Charsets.ISO_8859_1), " & ", " & "); reader.replace("<b>", " "); reader.replace("</b>", " "); reader.replace("<u>", " "); reader.replace("</u>", " "); reader.replace("<br />", " "); reader.replace(" ->", " →"); // right arrow reader.replace(" <-", " ←"); // left arrow reader.replace(" <> ", " ↔ "); // left-right arrow addCustomReplaces(reader); 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 result = 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")) // Your input is not valid return new QueryDeparturesResult(header, QueryDeparturesResult.Status.INVALID_STATION); if (code.equals("H890")) { result.stationDepartures .add(new StationDepartures(new Location(LocationType.STATION, normalizedStationId), Collections.<Departure>emptyList(), null)); return result; } 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 if (!"cancel".equals(eDelay)) { final Calendar plannedTime = new GregorianCalendar(timeZone); plannedTime.clear(); ParserUtils.parseEuropeanTime(plannedTime, fpTime); if (fpDate.length() == 8) ParserUtils.parseGermanDate(plannedTime, fpDate); else if (fpDate.length() == 10) ParserUtils.parseIsoDate(plannedTime, fpDate); else throw new IllegalStateException("cannot parse: '" + fpDate + "'"); 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<Attr> attrs = prodLine.attrs; if (attrs != null) line = newLine(product, prodLine.label, null, attrs.toArray(new Line.Attr[0])); else line = newLine(product, prodLine.label, null); } else { line = prodLine; } 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(result.stationDepartures, location); if (stationDepartures == null) { stationDepartures = new StationDepartures(location, new ArrayList<Departure>(8), null); result.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 : result.stationDepartures) Collections.sort(stationDepartures.departures, Departure.TIME_COMPARATOR); return result; } catch (final XmlPullParserException x) { throw new RuntimeException(x); } finally { if (reader != null) reader.close(); } } private StationDepartures findStationDepartures(final List<StationDepartures> stationDepartures, final Location location) { for (final StationDepartures stationDeparture : stationDepartures) if (stationDeparture.location.equals(location)) return stationDeparture; return null; } protected void addCustomReplaces(final StringReplaceReader reader) { } public QueryTripsResult queryTrips(final Location from, final @Nullable Location via, final Location to, final Date date, final boolean dep, final @Nullable Set<Product> products, final @Nullable Optimize optimize, final @Nullable WalkSpeed walkSpeed, final @Nullable Accessibility accessibility, final @Nullable Set<Option> options) throws IOException { return queryTripsBinary(from, via, to, date, dep, products, walkSpeed, accessibility, options); } 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, final @Nullable Collection<Product> products, final @Nullable WalkSpeed walkSpeed, final @Nullable Accessibility accessibility, final @Nullable Set<Option> options) throws IOException { final ResultHeader header = new ResultHeader(network, SERVER_PRODUCT); if (!from.isIdentified()) { final List<Location> 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<Location> 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<Location> 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); final StringBuilder productsStr = new StringBuilder(numProductBits); if (products != null) { for (int i = 0; i < numProductBits; i++) productsStr.append('0'); for (final Product p : products) setProductBits(productsStr, p); } else { productsStr.append(allProductsString()); } final char bikeChar = (options != null && options.contains(Option.BIKE)) ? '1' : '0'; final StringBuilder conReq = new StringBuilder("<ConReq deliverPolyline=\"1\">"); conReq.append("<Start>").append(locationXml(from)); conReq.append("<Prod prod=\"").append(productsStr).append("\" bike=\"").append(bikeChar) .append("\" couchette=\"0\" direct=\"0\" sleeper=\"0\"/>"); conReq.append("</Start>"); if (via != null) { conReq.append("<Via>").append(locationXml(via)); if (via.type != LocationType.ADDRESS) conReq.append("<Prod prod=\"").append(productsStr).append("\" bike=\"").append(bikeChar) .append("\" couchette=\"0\" direct=\"0\" sleeper=\"0\"/>"); conReq.append("</Via>"); } conReq.append("<Dest>").append(locationXml(to)).append("</Dest>"); conReq.append("<ReqT a=\"").append(dep ? 0 : 1).append("\" date=\"") .append(String.format(Locale.ENGLISH, "%04d.%02d.%02d", c.get(Calendar.YEAR), c.get(Calendar.MONTH) + 1, c.get(Calendar.DAY_OF_MONTH))) .append("\" time=\"").append(String.format(Locale.ENGLISH, "%02d:%02d", c.get(Calendar.HOUR_OF_DAY), c.get(Calendar.MINUTE)) + "\"/>"); conReq.append("<RFlags"); // number of trips backwards conReq.append(" b=\"").append(0).append("\""); // number of trips forwards conReq.append(" f=\"").append(numTripsRequested).append("\""); // percentual extension of change time conReq.append(" chExtension=\"").append(walkSpeed == WalkSpeed.SLOW ? 50 : 0).append("\""); // TODO nrChanges: max number of changes conReq.append(" sMode=\"N\"/>"); conReq.append("</ConReq>"); 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 scrDir=\"").append(later ? 'F' : 'B') .append("\" nrCons=\"").append(numTripsRequested).append("\">"); conScrReq.append("<ConResCtxt>").append(later ? context.laterContext : context.earlierContext) .append("</ConResCtxt>"); conScrReq.append("</ConScrReq>"); 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); Reader reader = null; String firstChars = null; try { final String endpoint = extXmlEndpoint != null ? extXmlEndpoint : queryEndpoint; final InputStream is = httpClient.getInputStream(endpoint, request, null, null); firstChars = HttpClient.peekFirstChars(is); reader = new InputStreamReader(is, Charsets.ISO_8859_1); final XmlPullParserFactory factory = XmlPullParserFactory .newInstance(System.getProperty(XmlPullParserFactory.PROPERTY_NAME), null); final XmlPullParser pp = factory.newPullParser(); pp.setInput(reader); XmlPullUtil.require(pp, "ResC"); final String product = XmlPullUtil.attr(pp, "prod").split(" ")[0]; final ResultHeader header = new ResultHeader(network, SERVER_PRODUCT, product, 0, null); XmlPullUtil.enter(pp, "ResC"); if (XmlPullUtil.test(pp, "Err")) { final String code = XmlPullUtil.attr(pp, "code"); if (code.equals("I3")) // Input: date outside of the timetable period return new QueryTripsResult(header, QueryTripsResult.Status.INVALID_DATE); if (code.equals("F1")) // Spool: Error reading the spoolfile return new QueryTripsResult(header, QueryTripsResult.Status.SERVICE_DOWN); 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"); if (code.equals("K9260")) // Unknown departure station return new QueryTripsResult(header, QueryTripsResult.Status.UNKNOWN_FROM); if (code.equals("K9280")) // Unknown intermediate station return new QueryTripsResult(header, QueryTripsResult.Status.UNKNOWN_VIA); if (code.equals("K9300")) // Unknown arrival station return new QueryTripsResult(header, QueryTripsResult.Status.UNKNOWN_TO); if (code.equals("K9360")) // Date outside of the timetable period return new QueryTripsResult(header, QueryTripsResult.Status.INVALID_DATE); if (code.equals("K9380")) // Dep./Arr./Intermed. or equivalent station defined more that once return new QueryTripsResult(header, QueryTripsResult.Status.TOO_CLOSE); if (code.equals("K895")) // Departure/Arrival are too near return new QueryTripsResult(header, QueryTripsResult.Status.TOO_CLOSE); if (code.equals("K9220")) // Nearby to the given address stations could not be found return new QueryTripsResult(header, QueryTripsResult.Status.UNRESOLVABLE_ADDRESS); if (code.equals("K9240")) // Internal error return new QueryTripsResult(header, QueryTripsResult.Status.SERVICE_DOWN); if (code.equals("K890")) // No connections found return new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS); if (code.equals("K891")) // No route found (try entering an intermediate station) return new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS); if (code.equals("K899")) // An error occurred return new QueryTripsResult(header, QueryTripsResult.Status.SERVICE_DOWN); if (code.equals("K1:890")) // Unsuccessful or incomplete search (direction: forward) return new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS); if (code.equals("K2:890")) // Unsuccessful or incomplete search (direction: backward) return new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS); throw new IllegalStateException("error " + code + " " + XmlPullUtil.attr(pp, "text")); } 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<Trip> trips = new ArrayList<Trip>(); 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"); if (XmlPullUtil.test(pp, "Arr")) XmlPullUtil.next(pp); if (XmlPullUtil.test(pp, "Dep")) XmlPullUtil.next(pp); 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<Trip.Leg> legs = new ArrayList<Trip.Leg>(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); if (XmlPullUtil.test(pp, "Arr")) { XmlPullUtil.enter(pp, "Arr"); XmlPullUtil.skipExit(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<Stop> 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<String, String> 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<Stop>(); 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<Point> path; if (XmlPullUtil.test(pp, "Polyline")) { path = new LinkedList<Point>(); 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(new Point(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"); return new QueryTripsResult(header, null, from, via, to, context, trips); } catch (final XmlPullParserException x) { throw new ParserException("cannot parse xml: " + firstChars, x); } finally { if (reader != null) reader.close(); } } 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<String, String> parseAttributeVariants(final XmlPullParser pp) throws XmlPullParserException, IOException { final Map<String, String> attributeVariants = new HashMap<String, String>(); 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 "<Station externalId=\"" + normalizeStationId(location.id) + "\" />"; else if (location.type == LocationType.POI && location.hasLocation()) return "<Poi type=\"WGS84\" x=\"" + location.lon + "\" y=\"" + location.lat + "\" />"; else if (location.type == LocationType.ADDRESS && location.hasLocation()) return "<Address type=\"WGS84\" x=\"" + location.lon + "\" y=\"" + location.lat + "\" name=\"" + (location.place != null ? location.place + ", " : "") + location.name + "\" />"; else if (location.type == LocationType.COORD && location.hasLocation()) return "<Coord type=\"WGS84\" x=\"" + location.lon + "\" y=\"" + location.lat + "\" />"; 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.hasLocation()) { id.append("@X=").append(location.lon); id.append("@Y=").append(location.lat); id.append("@O=").append(location.name != null ? location.name : String.format(Locale.ENGLISH, "%.6f, %.6f", location.lat / 1E6, location.lon / 1E6)); } 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.hasLocation())) 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 StringBuilder uri, final Location from, final @Nullable Location via, final Location to, final Date date, final boolean dep, final @Nullable Collection<Product> products, final @Nullable Accessibility accessibility, final @Nullable Set<Option> options) { uri.append("?start=Suchen"); uri.append("&REQ0JourneyStopsS0ID=").append(ParserUtils.urlEncode(locationId(from), Charsets.ISO_8859_1)); uri.append("&REQ0JourneyStopsZ0ID=").append(ParserUtils.urlEncode(locationId(to), Charsets.ISO_8859_1)); if (via != null) { // workaround, for there does not seem to be a REQ0JourneyStops1.0ID parameter uri.append("&REQ0JourneyStops1.0A=").append(locationType(via)); if (via.type == LocationType.STATION && via.hasId()) { uri.append("&REQ0JourneyStops1.0L=").append(via.id); } else if (via.hasLocation()) { uri.append("&REQ0JourneyStops1.0X=").append(via.lon); uri.append("&REQ0JourneyStops1.0Y=").append(via.lat); if (via.name == null) uri.append("&REQ0JourneyStops1.0O=") .append(ParserUtils.urlEncode( String.format(Locale.ENGLISH, "%.6f, %.6f", via.lat / 1E6, via.lon / 1E6), Charsets.ISO_8859_1)); } else if (via.name != null) { uri.append("&REQ0JourneyStops1.0G=").append(ParserUtils.urlEncode(via.name, Charsets.ISO_8859_1)); if (via.type != LocationType.ANY) uri.append('!'); } } uri.append("&REQ0HafasSearchForw=").append(dep ? "1" : "0"); appendDateTimeParameters(uri, date, "REQ0JourneyDate", "REQ0JourneyTime"); final StringBuilder productsStr = new StringBuilder(numProductBits); if (products != null) { for (int i = 0; i < numProductBits; i++) productsStr.append('0'); for (final Product p : products) setProductBits(productsStr, p); } else { productsStr.append(allProductsString()); } uri.append("&REQ0JourneyProduct_prod_list_1=").append(productsStr); if (accessibility != null && accessibility != Accessibility.NEUTRAL) { if (accessibility == Accessibility.LIMITED) uri.append("&REQ0AddParamBaimprofile=1"); else if (accessibility == Accessibility.BARRIER_FREE) uri.append("&REQ0AddParamBaimprofile=0"); } if (options != null && options.contains(Option.BIKE)) uri.append("&REQ0JourneyProduct_opt3=1"); appendCommonQueryTripsBinaryParameters(uri); } protected void appendCommonQueryTripsBinaryParameters(final StringBuilder uri) { uri.append("&h2g-direct=11"); if (clientType != null) uri.append("&clientType=").append(ParserUtils.urlEncode(clientType)); } 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, final @Nullable Collection<Product> products, final @Nullable WalkSpeed walkSpeed, final @Nullable Accessibility accessibility, final @Nullable Set<Option> options) throws IOException { final ResultHeader header = new ResultHeader(network, SERVER_PRODUCT); if (!from.isIdentified()) { final List<Location> 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<Location> 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<Location> 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 StringBuilder uri = new StringBuilder(queryEndpoint); appendQueryTripsBinaryParameters(uri, from, via, to, date, dep, products, accessibility, options); return queryTripsBinary(uri.toString(), from, via, to, QUERY_TRIPS_BINARY_BUFFER_SIZE); } protected void appendQueryMoreTripsBinaryParameters(final StringBuilder uri, final QueryTripsBinaryContext context, final boolean later) { uri.append("?seqnr=").append(context.seqNr); uri.append("&ident=").append(context.ident); if (context.ld != null) uri.append("&ld=").append(context.ld); uri.append("&REQ0HafasScrollDir=").append(later ? 1 : 2); appendCommonQueryTripsBinaryParameters(uri); } protected QueryTripsResult queryMoreTripsBinary(final QueryTripsContext contextObj, final boolean later) throws IOException { final QueryTripsBinaryContext context = (QueryTripsBinaryContext) contextObj; final StringBuilder uri = new StringBuilder(queryEndpoint); appendQueryMoreTripsBinaryParameters(uri, context, later); return queryTripsBinary(uri.toString(), 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 String uri, 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! */ LittleEndianDataInputStream is = null; try { final CustomBufferedInputStream bis = new CustomBufferedInputStream(httpClient.getInputStream(uri)); final String firstChars = HttpClient.peekFirstChars(bis); // initialize input stream 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: " + firstChars); final ResultHeader header = new ResultHeader(network, SERVER_PRODUCT, Integer.toString(version), 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) return new QueryTripsResult(header, uri, from, via, to, null, new LinkedList<Trip>()); // 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<Trip> trips = new ArrayList<Trip>(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<Trip.Leg> legs = new ArrayList<Trip.Leg>(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<Line.Attr> lineAttrs = new ArrayList<Line.Attr>(); 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("Vs ")) { 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; 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 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<Stop> 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<Stop>(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(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 fllt 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); final QueryTripsResult result = new QueryTripsResult(header, uri, from, via, to, new QueryTripsBinaryContext(requestId, seqNr, ld, bis.getCount(), canQueryMore), trips); return result; } else if (errorCode == 1) throw new SessionExpiredException(); else if (errorCode == 8) return new QueryTripsResult(header, QueryTripsResult.Status.AMBIGUOUS); else if (errorCode == 207) // H207: Unfortunately your connection request can currently not be processed. return new QueryTripsResult(header, QueryTripsResult.Status.SERVICE_DOWN); else if (errorCode == 887) // H887: Your inquiry was too complex. Please try entering less intermediate stations. return new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS); 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. return new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS); else if (errorCode == 891) // H891: Unfortunately there was no route found. Missing timetable data could be the reason. return new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS); else if (errorCode == 892) // H892: Your inquiry was too complex. Please try entering less intermediate stations. return new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS); else if (errorCode == 899) // H899: there was an unsuccessful or incomplete search due to a timetable change. return new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS); else if (errorCode == 900) // Unsuccessful or incomplete search (timetable change) return new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS); else if (errorCode == 9220) // H9220: Nearby to the given address stations could not be found. return new QueryTripsResult(header, QueryTripsResult.Status.UNRESOLVABLE_ADDRESS); 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. return new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS); else if (errorCode == 9260) // H9260: Unknown departure station return new QueryTripsResult(header, QueryTripsResult.Status.UNKNOWN_FROM); else if (errorCode == 9280) // H9280: Unknown intermediate station return new QueryTripsResult(header, QueryTripsResult.Status.UNKNOWN_VIA); else if (errorCode == 9300) // H9300: Unknown arrival station return new QueryTripsResult(header, QueryTripsResult.Status.UNKNOWN_TO); else if (errorCode == 9320) // The input is incorrect or incomplete return new QueryTripsResult(header, QueryTripsResult.Status.INVALID_DATE); else if (errorCode == 9360) // H9360: Unfortunately your connection request can currently not be processed. return new QueryTripsResult(header, QueryTripsResult.Status.INVALID_DATE); else if (errorCode == 9380) // H9380: Dep./Arr./Intermed. or equivalent station defined more than once return new QueryTripsResult(header, QueryTripsResult.Status.TOO_CLOSE); else if (errorCode == 895) // H895: Departure/Arrival are too near return new QueryTripsResult(header, QueryTripsResult.Status.TOO_CLOSE); else throw new IllegalStateException("error " + errorCode + " on " + uri); } finally { if (is != null) is.close(); } } 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, lat, lon, placeAndName[0], placeAndName[1]); } else if (type == 2) { final String[] placeAndName = splitAddress(name); return new Location(LocationType.ADDRESS, null, lat, lon, placeAndName[0], placeAndName[1]); } else if (type == 3) { final String[] placeAndName = splitPOI(name); return new Location(LocationType.POI, null, 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, lat, lon, placeAndName[0], placeAndName[1]); } finally { stationInputStream.close(); } } } private static final Pattern P_POSITION_PLATFORM = Pattern.compile("Gleis\\s*(.*)\\s*", Pattern.CASE_INSENSITIVE); private Position normalizePosition(final String position) { if (position == null) return null; final Matcher m = P_POSITION_PLATFORM.matcher(position); if (!m.matches()) return parsePosition(position); return parsePosition(m.group(1)); } public NearbyLocationsResult queryNearbyLocations(final EnumSet<LocationType> types, final Location location, final int maxDistance, final int maxLocations) throws IOException { if (location.hasLocation()) return nearbyLocationsByCoordinate(types, location.lat, location.lon, 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 EnumSet<LocationType> types, final int lat, final int lon, final int maxDistance, final int maxLocations) throws IOException { if (types.contains(LocationType.STATION)) { final StringBuilder uri = new StringBuilder(queryEndpoint); appendJsonNearbyStationsParameters(uri, lat, lon, maxDistance, maxLocations); return jsonNearbyLocations(uri.toString()); } else if (types.contains(LocationType.POI)) { final StringBuilder uri = new StringBuilder(queryEndpoint); appendJsonNearbyPOIsParameters(uri, lat, lon, maxDistance, maxLocations); return jsonNearbyLocations(uri.toString()); } else { return new NearbyLocationsResult(null, Collections.<Location>emptyList()); } } protected NearbyLocationsResult nearbyStationsById(final String id, final int maxDistance) throws IOException { final StringBuilder uri = new StringBuilder(stationBoardEndpoint); appendXmlNearbyStationsParameters(uri, id); return xmlNearbyStations(uri.toString()); } protected final void appendXmlNearbyStationsParameters(final StringBuilder uri, final String stationId) { uri.append("?productsFilter=").append(allProductsString()); uri.append("&boardType=dep"); uri.append("&input=").append(normalizeStationId(stationId)); uri.append("&sTI=1&start=yes&hcount=0&L=vs_java3"); if (clientType != null) uri.append("&clientType=").append(ParserUtils.urlEncode(clientType)); } 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("<Err code=\"([^\"]*)\" text=\"([^\"]*)\""); protected final NearbyLocationsResult xmlNearbyStations(final String uri) throws IOException { // scrape page final CharSequence page = httpClient.get(uri); final List<Location> stations = new ArrayList<Location>(); // 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 int parsedLon; final int parsedLat; if (mFine.group(3) != null && mFine.group(4) != null) { parsedLon = Integer.parseInt(mFine.group(3)); parsedLat = Integer.parseInt(mFine.group(4)); } else { parsedLon = 0; parsedLat = 0; } final String[] placeAndName = splitStationName(parsedName); stations.add(new Location(LocationType.STATION, parsedId, parsedLat, parsedLon, placeAndName[0], placeAndName[1])); } else { throw new IllegalArgumentException("cannot parse '" + mCoarse.group(1) + "' on " + uri); } } return new NearbyLocationsResult(null, stations); } protected void appendJsonNearbyStationsParameters(final StringBuilder uri, final int lat, final int lon, final int maxDistance, final int maxStations) { uri.append('y'); uri.append("?performLocating=2&tpl=stop2json"); uri.append("&look_stopclass=").append(allProductsInt()); uri.append("&look_nv=get_stopweight|yes"); // get_shortjson|yes // get_lines|yes // combinemode|2 // density|80 // get_stopweight|yes // get_infotext|yes uri.append("&look_x=").append(lon); uri.append("&look_y=").append(lat); uri.append("&look_maxno=").append(maxStations != 0 ? maxStations : 200); uri.append("&look_maxdist=").append(maxDistance != 0 ? maxDistance : 5000); } protected void appendJsonNearbyPOIsParameters(final StringBuilder uri, final int lat, final int lon, final int maxDistance, final int maxStations) { uri.append('y'); uri.append("?performLocating=4&tpl=poi2json"); uri.append("&look_pois="); // all categories uri.append("&look_x=").append(lon); uri.append("&look_y=").append(lat); uri.append("&look_maxno=").append(maxStations != 0 ? maxStations : 200); uri.append("&look_maxdist=").append(maxDistance != 0 ? maxDistance : 5000); } protected final NearbyLocationsResult jsonNearbyLocations(final String uri) throws IOException { final CharSequence page = httpClient.get(uri, null, jsonNearbyLocationsEncoding); try { final JSONObject head = new JSONObject(page.toString()); final int error = head.getInt("error"); if (error == 0) { final List<Location> locations = new LinkedList<Location>(); 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 stopWeight = stop.optInt("stopweight", -1); if (stopWeight != 0) { final String[] placeAndName = splitStationName(urlname); locations.add(new Location(LocationType.STATION, id, lat, lon, placeAndName[0], placeAndName[1])); } } } 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, lat, lon, placeAndName[0], placeAndName[1])); } } return new NearbyLocationsResult(null, locations); } else 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 " + uri, x); } } protected void setHtmlNearbyStationsPattern(final Pattern htmlNearbyStationsPattern) { this.htmlNearbyStationsPattern = htmlNearbyStationsPattern; } private Pattern htmlNearbyStationsPattern = Pattern.compile("<tr class=\"(zebra[^\"]*)\">(.*?)</tr>", Pattern.DOTALL); private final static Pattern P_NEARBY_FINE_COORDS = Pattern .compile("REQMapRoute0\\.Location0\\.X=(-?\\d+)&(?:amp;)?REQMapRoute0\\.Location0\\.Y=(-?\\d+)&"); private final static Pattern P_NEARBY_FINE_LOCATION = Pattern.compile("[\\?&;]input=(\\d+)&[^\"]*\">([^<]*)<"); protected final NearbyLocationsResult htmlNearbyStations(final String uri) throws IOException { final List<Location> stations = new ArrayList<Location>(); final CharSequence page = httpClient.get(uri); String oldZebra = null; final Matcher mCoarse = htmlNearbyStationsPattern.matcher(page); while (mCoarse.find()) { final String zebra = mCoarse.group(1); if (oldZebra != null && zebra.equals(oldZebra)) throw new IllegalArgumentException("missed row? last:" + zebra); else oldZebra = zebra; final Matcher mFineLocation = P_NEARBY_FINE_LOCATION.matcher(mCoarse.group(2)); if (mFineLocation.find()) { int parsedLon = 0; int parsedLat = 0; final String parsedId = mFineLocation.group(1); final String parsedName = ParserUtils.resolveEntities(mFineLocation.group(2)); final Matcher mFineCoords = P_NEARBY_FINE_COORDS.matcher(mCoarse.group(2)); if (mFineCoords.find()) { parsedLon = Integer.parseInt(mFineCoords.group(1)); parsedLat = Integer.parseInt(mFineCoords.group(2)); } final String[] placeAndName = splitStationName(parsedName); stations.add(new Location(LocationType.STATION, parsedId, parsedLat, parsedLon, placeAndName[0], placeAndName[1])); } else { throw new IllegalArgumentException("cannot parse '" + mCoarse.group(2) + "' on " + uri); } } return new NearbyLocationsResult(null, stations); } 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 ("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)) // Nachtzug Basel-Moskau return Product.HIGH_SPEED_TRAIN; if ("AIR".equals(ucType)) // Generic Flight return Product.HIGH_SPEED_TRAIN; if ("ECB".equals(ucType)) // EC, Verona-Mnchen return Product.HIGH_SPEED_TRAIN; if ("LYN".equals(ucType)) // Dnemark 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 ("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 Espaola, 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 ("EIP".equals(ucType)) // Express Intercity Premium return Product.HIGH_SPEED_TRAIN; if ("INT".equals(ucType)) // Zrich-Brssel - Budapest-Istanbul return Product.HIGH_SPEED_TRAIN; if ("HKX".equals(ucType)) // Hamburg-Koeln-Express 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 ("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)) // Spn vlak, 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)) // Mrkische 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 Bderbahn 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)) // Wrttembergische 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-Lnderbahn-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)) // Stdtebahn Sachsen return Product.REGIONAL_TRAIN; if ("SES".equals(ucType)) // Stdtebahn Sachsen Express return Product.REGIONAL_TRAIN; if ("EVB".equals(ucType)) // Eisenbahnen und Verkehrsbetriebe Elbe-Weser return Product.REGIONAL_TRAIN; if ("STB".equals(ucType)) // Sd-Thringen-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)) // Dllnitzbahn 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)) // Schsisch-Oberlausitzer Eisenbahngesellschaft return Product.REGIONAL_TRAIN; if ("SDG".equals(ucType)) // Schsische 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)) // Hrseltalbahn return Product.REGIONAL_TRAIN; if ("FEG".equals(ucType)) // Freiberger Eisenbahngesellschaft return Product.REGIONAL_TRAIN; if ("NEG".equals(ucType)) // Norddeutsche Eisenbahngesellschaft Niebll return Product.REGIONAL_TRAIN; if ("RBG".equals(ucType)) // Regental Bahnbetriebs GmbH return Product.REGIONAL_TRAIN; if ("MBB".equals(ucType)) // Mecklenburgische Bderbahn 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 Dolnolskie (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-Wrlitzer 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 rgional 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)) // Sdwestdeutsche Verkehrs-AG, Ortenau-S-Bahn return Product.SUBURBAN_TRAIN; if ("RER".equals(ucType)) // Rseau Express Rgional, 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; // 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 "Straenbahn 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-Saarbrcken return Product.BUS; if ("EXB".equals(ucType)) // Expressbus Mnchen-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 Zrich return Product.BUS; // Phone 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 ("FHRE".equals(ucType)) return Product.FERRY; if ("FH".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, eigentlich Autoreisezug 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_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_NORMALIZE_LINE_NUMBER = Pattern.compile("\\d{2,5}"); protected 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)"); 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_NORMALIZE_LINE_NUMBER.matcher(number).matches()) return newLine(null, 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) { if (attrs.length == 0) { return new Line(null, null, product, normalizedName, lineStyle(null, product, normalizedName), comment); } else { final Set<Line.Attr> attrSet = new HashSet<Line.Attr>(); for (final Line.Attr attr : attrs) attrSet.add(attr); return new Line(null, null, product, normalizedName, lineStyle(null, product, normalizedName), attrSet, comment); } } }