mirror of
https://gitlab.com/oeffi/public-transport-enabler.git
synced 2025-07-14 00:30:31 +00:00

git-svn-id: https://public-transport-enabler.googlecode.com/svn/trunk@45 0924bc21-9374-b0fa-ee44-9ff1593b38f0
730 lines
31 KiB
Java
730 lines
31 KiB
Java
/*
|
|
* Copyright 2010 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 java.io.IOException;
|
|
import java.text.DateFormat;
|
|
import java.text.ParseException;
|
|
import java.text.SimpleDateFormat;
|
|
import java.util.ArrayList;
|
|
import java.util.Calendar;
|
|
import java.util.Date;
|
|
import java.util.GregorianCalendar;
|
|
import java.util.HashMap;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.regex.Matcher;
|
|
import java.util.regex.Pattern;
|
|
|
|
/**
|
|
* @author Andreas Schildbach
|
|
*/
|
|
public final class VbbProvider implements NetworkProvider
|
|
{
|
|
public static final String NETWORK_ID = "mobil.bvg.de";
|
|
|
|
private static final long PARSER_DAY_ROLLOVER_THRESHOLD_MS = 12 * 60 * 60 * 1000;
|
|
private static final long PARSER_DAY_ROLLDOWN_THRESHOLD_MS = 6 * 60 * 60 * 1000;
|
|
|
|
private static final String BVG_BASE_URL = "http://mobil.bvg.de";
|
|
|
|
public boolean hasCapabilities(Capability... capabilities)
|
|
{
|
|
for (final Capability capability : capabilities)
|
|
if (capability == Capability.NEARBY_STATIONS)
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
private static final Pattern P_AUTOCOMPLETE_IS_MAST = Pattern.compile("\\d{6}");
|
|
private static final String AUTOCOMPLETE_NAME_URL = "http://mobil.bvg.de/Fahrinfo/bin/stboard.bin/dox/dox?input=";
|
|
private static final Pattern P_SINGLE_NAME = Pattern.compile(".*Haltestelleninfo.*?<strong>(.*?)</strong>.*", Pattern.DOTALL);
|
|
private static final Pattern P_MULTI_NAME = Pattern.compile("<a href=\\\"/Fahrinfo/bin/stboard.*?\\\">\\s*(.*?)\\s*</a>", Pattern.DOTALL);
|
|
private static final String AUTOCOMPLETE_MASTID_URL = "http://mobil.bvg.de/IstAbfahrtzeiten/index/mobil?input=";
|
|
private static final Pattern P_SINGLE_MASTID = Pattern.compile(".*Ist-Abfahrtzeiten.*?<strong>(.*?)</strong>.*", Pattern.DOTALL);
|
|
|
|
public List<String> autoCompleteStationName(CharSequence constraint) throws IOException
|
|
{
|
|
final List<String> names = new ArrayList<String>();
|
|
|
|
if (P_AUTOCOMPLETE_IS_MAST.matcher(constraint).matches())
|
|
{
|
|
final CharSequence page = ParserUtils.scrape(AUTOCOMPLETE_MASTID_URL + ParserUtils.urlEncode(constraint.toString()));
|
|
|
|
final Matcher mSingle = P_SINGLE_MASTID.matcher(page);
|
|
if (mSingle.matches())
|
|
{
|
|
names.add(ParserUtils.resolveEntities(mSingle.group(1)));
|
|
}
|
|
}
|
|
else
|
|
{
|
|
final CharSequence page = ParserUtils.scrape(AUTOCOMPLETE_NAME_URL + ParserUtils.urlEncode(constraint.toString()));
|
|
|
|
final Matcher mSingle = P_SINGLE_NAME.matcher(page);
|
|
if (mSingle.matches())
|
|
{
|
|
names.add(ParserUtils.resolveEntities(mSingle.group(1)));
|
|
}
|
|
else
|
|
{
|
|
final Matcher mMulti = P_MULTI_NAME.matcher(page);
|
|
while (mMulti.find())
|
|
names.add(ParserUtils.resolveEntities(mMulti.group(1)));
|
|
}
|
|
}
|
|
|
|
return names;
|
|
}
|
|
|
|
public List<Station> nearbyStations(final double lat, final double lon, final int maxDistance, final int maxStations) throws IOException
|
|
{
|
|
throw new UnsupportedOperationException();
|
|
}
|
|
|
|
private static Pattern P_STATION_LOCATION = Pattern.compile("<Station name=\"(.*?)\" x=\"(\\d+)\" y=\"(\\d+)\" type=\"WGS84\"");
|
|
private static Pattern P_STATION_LOCATION_ERROR = Pattern.compile("(No trains in result)|(No Response from Server)");
|
|
|
|
public StationLocationResult stationLocation(final String stationId) throws IOException
|
|
{
|
|
final boolean live = stationId.length() == 6;
|
|
if (live)
|
|
throw new UnsupportedOperationException();
|
|
|
|
final DateFormat DATE_FORMAT = new SimpleDateFormat("yyyyMMdd");
|
|
final DateFormat TIME_FORMAT = new SimpleDateFormat("HH:mm");
|
|
final Date now = new Date();
|
|
final String request = "<?xml version=\"1.0\" encoding=\"iso-8859-1\"?><ReqC lang=\"DE\" prod=\"testsystem\" ver=\"1.1\" accessId=\"VBB-STD\"><STBReq boardType='DEP'><Time>"
|
|
+ TIME_FORMAT.format(now)
|
|
+ "</Time><Period><DateBegin>"
|
|
+ DATE_FORMAT.format(now)
|
|
+ "</DateBegin><DateEnd>"
|
|
+ DATE_FORMAT.format(now) + "</DateEnd></Period><TableStation externalId='" + stationId + "'/></STBReq></ReqC>";
|
|
final String uri = "http://www.vbb-fahrinfo.de/hafas/extxml/extxml.exe/dn";
|
|
|
|
final CharSequence page = ParserUtils.scrape(uri, request);
|
|
|
|
final Matcher mError = P_STATION_LOCATION_ERROR.matcher(page);
|
|
if (mError.find())
|
|
{
|
|
if (mError.group(1) != null)
|
|
return null;
|
|
if (mError.group(2) != null)
|
|
throw new RuntimeException("timeout error");
|
|
}
|
|
|
|
final Matcher m = P_STATION_LOCATION.matcher(page);
|
|
if (m.find())
|
|
{
|
|
final String name = ParserUtils.resolveEntities(m.group(1));
|
|
final double lon = latLonToDouble(Integer.parseInt(m.group(2)));
|
|
final double lat = latLonToDouble(Integer.parseInt(m.group(3)));
|
|
|
|
return new StationLocationResult(lat, lon, name);
|
|
}
|
|
else
|
|
{
|
|
throw new IllegalArgumentException("cannot parse '" + page + "' on " + uri);
|
|
}
|
|
}
|
|
|
|
private static double latLonToDouble(int value)
|
|
{
|
|
return (double) value / 1000000;
|
|
}
|
|
|
|
public static final String STATION_URL_CONNECTION = "http://mobil.bvg.de/Fahrinfo/bin/query.bin/dox";
|
|
|
|
private String connectionsQueryUri(final String from, final String via, final String to, final Date date, final boolean dep)
|
|
{
|
|
final DateFormat DATE_FORMAT = new SimpleDateFormat("dd.MM.yy");
|
|
final DateFormat TIME_FORMAT = new SimpleDateFormat("HH:mm");
|
|
final StringBuilder uri = new StringBuilder();
|
|
|
|
uri.append("http://mobil.bvg.de/Fahrinfo/bin/query.bin/dox");
|
|
uri.append("?REQ0HafasInitialSelection=0");
|
|
uri.append("&REQ0JourneyStopsS0A=255");
|
|
uri.append("&REQ0JourneyStopsS0G=").append(ParserUtils.urlEncode(from));
|
|
if (via != null)
|
|
{
|
|
uri.append("&REQ0JourneyStops1A=1");
|
|
uri.append("&REQ0JourneyStops1G=").append(ParserUtils.urlEncode(via));
|
|
}
|
|
uri.append("&REQ0JourneyStopsZ0A=255");
|
|
uri.append("&REQ0JourneyStopsZ0G=").append(ParserUtils.urlEncode(to));
|
|
uri.append("&REQ0HafasSearchForw=").append(dep ? "1" : "0");
|
|
uri.append("&REQ0JourneyDate=").append(ParserUtils.urlEncode(DATE_FORMAT.format(date)));
|
|
uri.append("&REQ0JourneyTime=").append(ParserUtils.urlEncode(TIME_FORMAT.format(date)));
|
|
uri.append("&start=Suchen");
|
|
|
|
return uri.toString();
|
|
}
|
|
|
|
private static final Pattern P_CHECK_ADDRESS = Pattern.compile("<option.*?>\\s*(.*?)\\s*</option>", Pattern.DOTALL);
|
|
private static final Pattern P_CHECK_FROM = Pattern.compile("Von:");
|
|
private static final Pattern P_CHECK_TO = Pattern.compile("Nach:");
|
|
private static final Pattern P_CHECK_CONNECTIONS_ERROR = Pattern.compile("(zu dicht beieinander)|(keine Verbindung gefunden)");
|
|
|
|
public CheckConnectionsQueryResult checkConnectionsQuery(final String from, final String via, final String to, final Date date, final boolean dep)
|
|
throws IOException
|
|
{
|
|
final String queryUri = connectionsQueryUri(from, via, to, date, dep);
|
|
final CharSequence page = ParserUtils.scrape(queryUri);
|
|
|
|
final Matcher mError = P_CHECK_CONNECTIONS_ERROR.matcher(page);
|
|
if (mError.find())
|
|
{
|
|
if (mError.group(1) != null)
|
|
return CheckConnectionsQueryResult.TOO_CLOSE;
|
|
if (mError.group(2) != null)
|
|
return CheckConnectionsQueryResult.NO_CONNECTIONS;
|
|
}
|
|
|
|
final Matcher mAddress = P_CHECK_ADDRESS.matcher(page);
|
|
|
|
final List<String> addresses = new ArrayList<String>();
|
|
while (mAddress.find())
|
|
{
|
|
final String address = ParserUtils.resolveEntities(mAddress.group(1));
|
|
if (!addresses.contains(address))
|
|
addresses.add(address);
|
|
}
|
|
|
|
if (addresses.isEmpty())
|
|
{
|
|
return new CheckConnectionsQueryResult(CheckConnectionsQueryResult.Status.OK, queryUri, null, null, null);
|
|
}
|
|
else if (P_CHECK_FROM.matcher(page).find())
|
|
{
|
|
if (P_CHECK_TO.matcher(page).find())
|
|
return new CheckConnectionsQueryResult(CheckConnectionsQueryResult.Status.AMBIGUOUS, queryUri, null, addresses, null);
|
|
else
|
|
return new CheckConnectionsQueryResult(CheckConnectionsQueryResult.Status.AMBIGUOUS, queryUri, null, null, addresses);
|
|
}
|
|
else
|
|
{
|
|
return new CheckConnectionsQueryResult(CheckConnectionsQueryResult.Status.AMBIGUOUS, queryUri, addresses, null, null);
|
|
}
|
|
}
|
|
|
|
private static final Pattern P_CONNECTIONS_HEAD = Pattern.compile(
|
|
".*Von: <strong>(.*?)</strong>.*?Nach: <strong>(.*?)</strong>.*?Datum: .., (.*?)<br />.*?"
|
|
+ "(?:<a href=\"(/Fahrinfo/bin/query\\.bin/dox.{1,80}ScrollDir=2)\">.*?)?"
|
|
+ "(?:<a href=\"(/Fahrinfo/bin/query\\.bin/dox.{1,80}ScrollDir=1)\">.*?)?", Pattern.DOTALL);
|
|
private static final Pattern P_CONNECTIONS_COARSE = Pattern.compile("<p class=\"con(?:L|D)\">(.+?)</p>", Pattern.DOTALL);
|
|
private static final Pattern P_CONNECTIONS_FINE = Pattern.compile(".*?<a href=\"(/Fahrinfo/bin/query\\.bin/dox.*?)\">"
|
|
+ "(\\d\\d:\\d\\d)-(\\d\\d:\\d\\d)</a> (?:\\d+ Umst\\.|([\\w\\d ]+)).*?", Pattern.DOTALL);
|
|
|
|
public QueryConnectionsResult queryConnections(final String uri) throws IOException
|
|
{
|
|
final CharSequence page = ParserUtils.scrape(uri);
|
|
|
|
final Matcher mHead = P_CONNECTIONS_HEAD.matcher(page);
|
|
if (mHead.matches())
|
|
{
|
|
final String from = ParserUtils.resolveEntities(mHead.group(1));
|
|
final String to = ParserUtils.resolveEntities(mHead.group(2));
|
|
final Date currentDate = ParserUtils.parseDate(mHead.group(3));
|
|
final String linkEarlier = mHead.group(4) != null ? BVG_BASE_URL + ParserUtils.resolveEntities(mHead.group(4)) : null;
|
|
final String linkLater = mHead.group(5) != null ? BVG_BASE_URL + ParserUtils.resolveEntities(mHead.group(5)) : null;
|
|
final List<Connection> connections = new ArrayList<Connection>();
|
|
|
|
final Matcher mConCoarse = P_CONNECTIONS_COARSE.matcher(page);
|
|
while (mConCoarse.find())
|
|
{
|
|
final Matcher mConFine = P_CONNECTIONS_FINE.matcher(mConCoarse.group(1));
|
|
if (mConFine.matches())
|
|
{
|
|
final String link = BVG_BASE_URL + ParserUtils.resolveEntities(mConFine.group(1));
|
|
Date departureTime = ParserUtils.joinDateTime(currentDate, ParserUtils.parseTime(mConFine.group(2)));
|
|
if (!connections.isEmpty())
|
|
{
|
|
final long diff = ParserUtils.timeDiff(departureTime,
|
|
((Connection.Trip) connections.get(connections.size() - 1).parts.get(0)).departureTime);
|
|
if (diff > PARSER_DAY_ROLLOVER_THRESHOLD_MS)
|
|
departureTime = ParserUtils.addDays(departureTime, -1);
|
|
else if (diff < -PARSER_DAY_ROLLDOWN_THRESHOLD_MS)
|
|
departureTime = ParserUtils.addDays(departureTime, 1);
|
|
}
|
|
Date arrivalTime = ParserUtils.joinDateTime(currentDate, ParserUtils.parseTime(mConFine.group(3)));
|
|
if (departureTime.after(arrivalTime))
|
|
arrivalTime = ParserUtils.addDays(arrivalTime, 1);
|
|
final String line = normalizeLine(ParserUtils.resolveEntities(mConFine.group(4)));
|
|
final Connection connection = new Connection(link, departureTime, arrivalTime, 0, from, 0, to, new ArrayList<Connection.Part>(1));
|
|
connection.parts.add(new Connection.Trip(departureTime, arrivalTime, line, line != null ? LINES.get(line) : null));
|
|
connections.add(connection);
|
|
}
|
|
else
|
|
{
|
|
throw new IllegalArgumentException("cannot parse '" + mConCoarse.group(1) + "' on " + uri);
|
|
}
|
|
}
|
|
|
|
return new QueryConnectionsResult(from, to, currentDate, linkEarlier, linkLater, connections);
|
|
}
|
|
else
|
|
{
|
|
throw new IOException(page.toString());
|
|
}
|
|
}
|
|
|
|
private static final Pattern P_CONNECTION_DETAILS_HEAD = Pattern.compile(".*(?:Datum|Abfahrt): (\\d\\d\\.\\d\\d\\.\\d\\d).*", Pattern.DOTALL);
|
|
private static final Pattern P_CONNECTION_DETAILS_COARSE = Pattern.compile("<p class=\"con\\w\">\n?(.+?)\n?</p>", Pattern.DOTALL);
|
|
static final Pattern P_CONNECTION_DETAILS_FINE = Pattern.compile("(?:<a href=\".*?input=(\\d+).*?\">(?:\\n?<strong>)?" // departureId
|
|
+ "(.+?)(?:</strong>\\n?)?</a>)?.*?" // departure
|
|
+ "(?:" //
|
|
+ "ab (\\d+:\\d+)\n?" // departureTime
|
|
+ "(Gl\\. \\d+)?.*?" // departurePosition
|
|
+ "<strong>\\s*(.*?)\\s*</strong>.*?" // line
|
|
+ "Ri\\. (.*?)[\n\\.]*<.*?" // destination
|
|
+ "an (\\d+:\\d+)\n?" // arrivalTime
|
|
+ "(Gl\\. \\d+)?.*?" // arrivalPosition
|
|
+ "<a href=\".*?input=(\\d+).*?\">\n?" // arrivalId
|
|
+ "<strong>(.*?)</strong>" // arrival
|
|
+ "|" //
|
|
+ "(\\d+) Min\\.[\n\\s]?" // footway
|
|
+ "Fussweg\n?" //
|
|
+ ".*?(?:<a href=\"/Fahrinfo.*?input=(\\d+)\">\n?" // arrivalId
|
|
+ "<strong>(.*?)</strong>|<a href=\"/Stadtplan.*?\">(\\w.*?)</a>).*?" // arrival
|
|
+ ").*?", Pattern.DOTALL);
|
|
|
|
public GetConnectionDetailsResult getConnectionDetails(final String uri) throws IOException
|
|
{
|
|
final CharSequence page = ParserUtils.scrape(uri);
|
|
|
|
final Matcher mHead = P_CONNECTION_DETAILS_HEAD.matcher(page);
|
|
if (mHead.matches())
|
|
{
|
|
final Date currentDate = ParserUtils.parseDate(mHead.group(1));
|
|
final List<Connection.Part> parts = new ArrayList<Connection.Part>(4);
|
|
|
|
Date firstDepartureTime = null;
|
|
String firstDeparture = null;
|
|
int firstDepartureId = 0;
|
|
Date lastArrivalTime = null;
|
|
String lastArrival = null;
|
|
int lastArrivalId = 0;
|
|
|
|
final Matcher mDetCoarse = P_CONNECTION_DETAILS_COARSE.matcher(page);
|
|
while (mDetCoarse.find())
|
|
{
|
|
final Matcher mDetFine = P_CONNECTION_DETAILS_FINE.matcher(mDetCoarse.group(1));
|
|
if (mDetFine.matches())
|
|
{
|
|
int departureId = 0;
|
|
String departure = ParserUtils.resolveEntities(mDetFine.group(2));
|
|
if (departure == null)
|
|
{
|
|
departure = lastArrival;
|
|
departureId = lastArrivalId;
|
|
}
|
|
else
|
|
{
|
|
departureId = Integer.parseInt(mDetFine.group(1));
|
|
}
|
|
|
|
if (departure != null && firstDeparture == null)
|
|
{
|
|
firstDeparture = departure;
|
|
firstDepartureId = departureId;
|
|
}
|
|
|
|
final String min = mDetFine.group(11);
|
|
if (min == null)
|
|
{
|
|
Date departureTime = ParserUtils.joinDateTime(currentDate, ParserUtils.parseTime(mDetFine.group(3)));
|
|
if (lastArrivalTime != null && departureTime.before(lastArrivalTime))
|
|
departureTime = ParserUtils.addDays(departureTime, 1);
|
|
|
|
final String departurePosition = mDetFine.group(4);
|
|
|
|
final String line = normalizeLine(ParserUtils.resolveEntities(mDetFine.group(5)));
|
|
|
|
final String destination = ParserUtils.resolveEntities(mDetFine.group(6));
|
|
|
|
Date arrivalTime = ParserUtils.joinDateTime(currentDate, ParserUtils.parseTime(mDetFine.group(7)));
|
|
if (departureTime.after(arrivalTime))
|
|
arrivalTime = ParserUtils.addDays(arrivalTime, 1);
|
|
|
|
final String arrivalPosition = mDetFine.group(8);
|
|
|
|
final int arrivalId = Integer.parseInt(mDetFine.group(9));
|
|
|
|
final String arrival = ParserUtils.resolveEntities(mDetFine.group(10));
|
|
|
|
parts.add(new Connection.Trip(line, line != null ? LINES.get(line) : null, destination, departureTime, departurePosition,
|
|
departureId, departure, arrivalTime, arrivalPosition, arrivalId, arrival));
|
|
|
|
if (firstDepartureTime == null)
|
|
firstDepartureTime = departureTime;
|
|
|
|
lastArrival = arrival;
|
|
lastArrivalId = arrivalId;
|
|
lastArrivalTime = arrivalTime;
|
|
}
|
|
else
|
|
{
|
|
final int arrivalId = Integer.parseInt(mDetFine.group(12));
|
|
|
|
final String arrival = ParserUtils.resolveEntities(selectNotNull(mDetFine.group(13), mDetFine.group(14)));
|
|
|
|
if (parts.size() > 0 && parts.get(parts.size() - 1) instanceof Connection.Footway)
|
|
{
|
|
final Connection.Footway lastFootway = (Connection.Footway) parts.remove(parts.size() - 1);
|
|
parts.add(new Connection.Footway(lastFootway.min + Integer.parseInt(min), lastFootway.departure, arrival));
|
|
}
|
|
else
|
|
{
|
|
parts.add(new Connection.Footway(Integer.parseInt(min), departure, arrival));
|
|
}
|
|
|
|
lastArrival = arrival;
|
|
lastArrivalId = arrivalId;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
throw new IllegalArgumentException("cannot parse '" + mDetCoarse.group(1) + "' on " + uri);
|
|
}
|
|
}
|
|
|
|
if (firstDepartureTime != null && lastArrivalTime != null)
|
|
return new GetConnectionDetailsResult(currentDate, new Connection(uri, firstDepartureTime, lastArrivalTime, firstDepartureId,
|
|
firstDeparture, lastArrivalId, lastArrival, parts));
|
|
else
|
|
return new GetConnectionDetailsResult(currentDate, null);
|
|
}
|
|
else
|
|
{
|
|
throw new IOException(page.toString());
|
|
}
|
|
}
|
|
|
|
private static String selectNotNull(final String... groups)
|
|
{
|
|
String selected = null;
|
|
for (final String group : groups)
|
|
{
|
|
if (group != null)
|
|
{
|
|
if (selected == null)
|
|
selected = group;
|
|
else
|
|
throw new IllegalStateException("ambiguous");
|
|
}
|
|
}
|
|
return selected;
|
|
}
|
|
|
|
private static final String DEPARTURE_URL_LIVE = "http://mobil.bvg.de/IstAbfahrtzeiten/index/mobil?";
|
|
private static final String DEPARTURE_URL_PLAN = "http://mobil.bvg.de/Fahrinfo/bin/stboard.bin/dox/dox?boardType=dep&start=yes&";
|
|
|
|
public String departuresQueryUri(final String stationId, final int maxDepartures)
|
|
{
|
|
final boolean live = stationId.length() == 6;
|
|
|
|
final StringBuilder uri = new StringBuilder();
|
|
uri.append(live ? DEPARTURE_URL_LIVE : DEPARTURE_URL_PLAN);
|
|
uri.append("input=").append(stationId);
|
|
uri.append("&maxJourneys=").append(maxDepartures != 0 ? maxDepartures : 12);
|
|
return uri.toString();
|
|
}
|
|
|
|
private static final Pattern P_DEPARTURES_HEAD = Pattern.compile(".*<strong>(.*?)</strong>.*Datum:(.*?)<br />.*", Pattern.DOTALL);
|
|
private static final Pattern P_DEPARTURES_COARSE = Pattern.compile(
|
|
"<tr class=\"ivu_table_bg\\d\">\\s*((?:<td class=\"ivu_table_c_dep\">|<td>).+?)\\s*</tr>", Pattern.DOTALL);
|
|
private static final Pattern P_DEPARTURES_LIVE_FINE = Pattern.compile("<td class=\"ivu_table_c_dep\">\\s*(.*?)[\\s\\*]*</td>\\s*" //
|
|
+ "<td class=\"ivu_table_c_line\">\\s*(.*?)\\s*</td>\\s*" //
|
|
+ "<td>.*?<a.*?[^-]>\\s*(.*?)\\s*</a>.*?</td>", Pattern.DOTALL);
|
|
private static final Pattern P_DEPARTURES_PLAN_FINE = Pattern.compile("<td><strong>\\s*(.*?)[\\s\\*]*</strong></td>\\s*" //
|
|
+ "<td>\\s*<strong>\\s*(.*?)[\\s\\*]*</strong>.*?</td>\\s*" //
|
|
+ "<td>\\s*<a.*?>\\s*(.*?)\\s*</a>\\s*</td>", Pattern.DOTALL);
|
|
private static final Pattern P_DEPARTURES_SERVICE_DOWN = Pattern.compile("Wartungsarbeiten");
|
|
private static final Pattern P_DEPARTURES_URI_STATION_ID = Pattern.compile("input=(\\d+)");
|
|
|
|
public QueryDeparturesResult queryDepartures(final String uri) throws IOException
|
|
{
|
|
final CharSequence page = ParserUtils.scrape(uri);
|
|
|
|
final Matcher mStationId = P_DEPARTURES_URI_STATION_ID.matcher(uri);
|
|
if (!mStationId.find())
|
|
throw new IllegalStateException(uri);
|
|
final int stationId = Integer.parseInt(mStationId.group(1));
|
|
|
|
if (P_DEPARTURES_SERVICE_DOWN.matcher(page).find())
|
|
return QueryDeparturesResult.SERVICE_DOWN;
|
|
|
|
// parse page
|
|
final Matcher mHead = P_DEPARTURES_HEAD.matcher(page);
|
|
if (mHead.matches())
|
|
{
|
|
final String location = ParserUtils.resolveEntities(mHead.group(1));
|
|
final Date currentTime = parseDate(mHead.group(2));
|
|
final List<Departure> departures = new ArrayList<Departure>(8);
|
|
|
|
// choose matcher
|
|
final Matcher mDepCoarse = P_DEPARTURES_COARSE.matcher(page);
|
|
while (mDepCoarse.find())
|
|
{
|
|
final boolean live = uri.contains("IstAbfahrtzeiten");
|
|
final Matcher mDepFine = (live ? P_DEPARTURES_LIVE_FINE : P_DEPARTURES_PLAN_FINE).matcher(mDepCoarse.group(1));
|
|
if (mDepFine.matches())
|
|
{
|
|
final Departure dep = parseDeparture(mDepFine, currentTime);
|
|
if (!departures.contains(dep))
|
|
departures.add(dep);
|
|
}
|
|
else
|
|
{
|
|
throw new IllegalArgumentException("cannot parse '" + mDepCoarse.group(1) + "' on " + uri);
|
|
}
|
|
}
|
|
|
|
return new QueryDeparturesResult(uri, stationId, location, currentTime, departures);
|
|
}
|
|
else
|
|
{
|
|
return QueryDeparturesResult.NO_INFO;
|
|
}
|
|
}
|
|
|
|
private static Departure parseDeparture(final Matcher mDep, final Date currentTime)
|
|
{
|
|
// time
|
|
final Calendar current = new GregorianCalendar();
|
|
current.setTime(currentTime);
|
|
final Calendar parsed = new GregorianCalendar();
|
|
parsed.setTime(ParserUtils.parseTime(mDep.group(1)));
|
|
parsed.set(Calendar.YEAR, current.get(Calendar.YEAR));
|
|
parsed.set(Calendar.MONTH, current.get(Calendar.MONTH));
|
|
parsed.set(Calendar.DAY_OF_MONTH, current.get(Calendar.DAY_OF_MONTH));
|
|
if (ParserUtils.timeDiff(parsed.getTime(), currentTime) < -PARSER_DAY_ROLLOVER_THRESHOLD_MS)
|
|
parsed.add(Calendar.DAY_OF_MONTH, 1);
|
|
|
|
// line
|
|
final String line = normalizeLine(ParserUtils.resolveEntities(mDep.group(2)));
|
|
|
|
// destination
|
|
final String destination = ParserUtils.resolveEntities(mDep.group(3));
|
|
|
|
return new Departure(parsed.getTime(), line, line != null ? LINES.get(line) : null, destination);
|
|
}
|
|
|
|
private static final Date parseDate(String str)
|
|
{
|
|
try
|
|
{
|
|
return new SimpleDateFormat("dd.MM.yyyy, HH:mm:ss").parse(str);
|
|
}
|
|
catch (ParseException x1)
|
|
{
|
|
try
|
|
{
|
|
return new SimpleDateFormat("dd.MM.yy").parse(str);
|
|
}
|
|
catch (ParseException x2)
|
|
{
|
|
throw new RuntimeException(x2);
|
|
}
|
|
}
|
|
}
|
|
|
|
private static final Pattern P_NORMALIZE_LINE = Pattern.compile("([A-Za-zÄÖÜäöüß]+)[\\s-]*(.*)");
|
|
private static final Pattern P_NORMALIZE_LINE_SPECIAL_NUMBER = Pattern.compile("\\d{4,}");
|
|
private static final Pattern P_NORMALIZE_LINE_SPECIAL_BUS = Pattern.compile("Bus[A-Z]");
|
|
|
|
private static String normalizeLine(final String line)
|
|
{
|
|
if (line == null || line.length() == 0)
|
|
return null;
|
|
|
|
if (line.startsWith("RE") || line.startsWith("RB") || line.startsWith("NE") || line.startsWith("OE") || line.startsWith("MR")
|
|
|| line.startsWith("PE"))
|
|
return "R" + line;
|
|
if (P_NORMALIZE_LINE_SPECIAL_NUMBER.matcher(line).matches())
|
|
return "R" + line;
|
|
|
|
final Matcher m = P_NORMALIZE_LINE.matcher(line);
|
|
if (m.matches())
|
|
{
|
|
final String type = m.group(1);
|
|
final String number = m.group(2).replace(" ", "");
|
|
|
|
if (type.equals("ICE")) // InterCityExpress
|
|
return "IICE" + number;
|
|
if (type.equals("IC")) // InterCity
|
|
return "IIC" + number;
|
|
if (type.equals("EC")) // EuroCity
|
|
return "IEC" + number;
|
|
if (type.equals("EN")) // EuroNight
|
|
return "IEN" + number;
|
|
if (type.equals("CNL")) // CityNightLine
|
|
return "ICNL" + number;
|
|
if (type.equals("Zug"))
|
|
return "R" + number;
|
|
if (type.equals("D")) // D-Zug?
|
|
return "RD" + number;
|
|
if (type.equals("DNZ")) // unklar, aber vermutlich Russland
|
|
return "RDNZ" + (number.equals("DNZ") ? "" : number);
|
|
if (type.equals("KBS")) // Kursbuchstrecke
|
|
return "RKBS" + number;
|
|
if (type.equals("BKB")) // Buckower Kleinbahn
|
|
return "RBKB" + number;
|
|
if (type.equals("S"))
|
|
return "SS" + number;
|
|
if (type.equals("U"))
|
|
return "UU" + number;
|
|
if (type.equals("Tra") || type.equals("Tram"))
|
|
return "T" + number;
|
|
if (type.equals("Bus"))
|
|
return "B" + number;
|
|
if (P_NORMALIZE_LINE_SPECIAL_BUS.matcher(type).matches()) // workaround for weird scheme BusF/526
|
|
return "B" + line.substring(3);
|
|
if (type.equals("Fäh"))
|
|
return "F" + number;
|
|
if (type.equals("F"))
|
|
return "FF" + number;
|
|
|
|
throw new IllegalStateException("cannot normalize type " + type + " line " + line);
|
|
}
|
|
|
|
throw new IllegalStateException("cannot normalize line " + line);
|
|
}
|
|
|
|
private static final Map<String, int[]> LINES = new HashMap<String, int[]>();
|
|
|
|
static
|
|
{
|
|
LINES.put("I", new int[] { Color.WHITE, Color.RED, Color.RED }); // generic
|
|
LINES.put("R", new int[] { Color.WHITE, Color.RED, Color.RED }); // generic
|
|
LINES.put("S", new int[] { Color.parseColor("#006e34"), Color.WHITE }); // generic
|
|
LINES.put("U", new int[] { Color.parseColor("#003090"), Color.WHITE }); // generic
|
|
|
|
LINES.put("SS1", new int[] { Color.rgb(221, 77, 174), Color.WHITE });
|
|
LINES.put("SS2", new int[] { Color.rgb(16, 132, 73), Color.WHITE });
|
|
LINES.put("SS25", new int[] { Color.rgb(16, 132, 73), Color.WHITE });
|
|
LINES.put("SS3", new int[] { Color.rgb(22, 106, 184), Color.WHITE });
|
|
LINES.put("SS41", new int[] { Color.rgb(162, 63, 48), Color.WHITE });
|
|
LINES.put("SS42", new int[] { Color.rgb(191, 90, 42), Color.WHITE });
|
|
LINES.put("SS45", new int[] { Color.rgb(191, 128, 55), Color.WHITE });
|
|
LINES.put("SS46", new int[] { Color.rgb(191, 128, 55), Color.WHITE });
|
|
LINES.put("SS47", new int[] { Color.rgb(191, 128, 55), Color.WHITE });
|
|
LINES.put("SS5", new int[] { Color.rgb(243, 103, 23), Color.WHITE });
|
|
LINES.put("SS7", new int[] { Color.rgb(119, 96, 176), Color.WHITE });
|
|
LINES.put("SS75", new int[] { Color.rgb(119, 96, 176), Color.WHITE });
|
|
LINES.put("SS8", new int[] { Color.rgb(85, 184, 49), Color.WHITE });
|
|
LINES.put("SS85", new int[] { Color.rgb(85, 184, 49), Color.WHITE });
|
|
LINES.put("SS9", new int[] { Color.rgb(148, 36, 64), Color.WHITE });
|
|
|
|
LINES.put("UU1", new int[] { Color.rgb(84, 131, 47), Color.WHITE });
|
|
LINES.put("UU2", new int[] { Color.rgb(215, 25, 16), Color.WHITE });
|
|
LINES.put("UU3", new int[] { Color.rgb(47, 152, 154), Color.WHITE });
|
|
LINES.put("UU4", new int[] { Color.rgb(255, 233, 42), Color.BLACK });
|
|
LINES.put("UU5", new int[] { Color.rgb(91, 31, 16), Color.WHITE });
|
|
LINES.put("UU55", new int[] { Color.rgb(91, 31, 16), Color.WHITE });
|
|
LINES.put("UU6", new int[] { Color.rgb(127, 57, 115), Color.WHITE });
|
|
LINES.put("UU7", new int[] { Color.rgb(0, 153, 204), Color.WHITE });
|
|
LINES.put("UU8", new int[] { Color.rgb(24, 25, 83), Color.WHITE });
|
|
LINES.put("UU9", new int[] { Color.rgb(255, 90, 34), Color.WHITE });
|
|
|
|
LINES.put("TM1", new int[] { Color.rgb(204, 51, 0), Color.WHITE });
|
|
LINES.put("TM2", new int[] { Color.rgb(116, 192, 67), Color.WHITE });
|
|
LINES.put("TM4", new int[] { Color.rgb(208, 28, 34), Color.WHITE });
|
|
LINES.put("TM5", new int[] { Color.rgb(204, 153, 51), Color.WHITE });
|
|
LINES.put("TM6", new int[] { Color.rgb(0, 0, 255), Color.WHITE });
|
|
LINES.put("TM8", new int[] { Color.rgb(255, 102, 0), Color.WHITE });
|
|
LINES.put("TM10", new int[] { Color.rgb(0, 153, 51), Color.WHITE });
|
|
LINES.put("TM13", new int[] { Color.rgb(51, 153, 102), Color.WHITE });
|
|
LINES.put("TM17", new int[] { Color.rgb(153, 102, 51), Color.WHITE });
|
|
|
|
LINES.put("B12", new int[] { Color.rgb(153, 102, 255), Color.WHITE });
|
|
LINES.put("B16", new int[] { Color.rgb(0, 0, 255), Color.WHITE });
|
|
LINES.put("B18", new int[] { Color.rgb(255, 102, 0), Color.WHITE });
|
|
LINES.put("B21", new int[] { Color.rgb(153, 102, 255), Color.WHITE });
|
|
LINES.put("B27", new int[] { Color.rgb(153, 102, 51), Color.WHITE });
|
|
LINES.put("B37", new int[] { Color.rgb(153, 102, 51), Color.WHITE });
|
|
LINES.put("B50", new int[] { Color.rgb(51, 153, 102), Color.WHITE });
|
|
LINES.put("B60", new int[] { Color.rgb(0, 153, 51), Color.WHITE });
|
|
LINES.put("B61", new int[] { Color.rgb(0, 153, 51), Color.WHITE });
|
|
LINES.put("B62", new int[] { Color.rgb(0, 102, 51), Color.WHITE });
|
|
LINES.put("B63", new int[] { Color.rgb(51, 153, 102), Color.WHITE });
|
|
LINES.put("B67", new int[] { Color.rgb(0, 102, 51), Color.WHITE });
|
|
LINES.put("B68", new int[] { Color.rgb(0, 153, 51), Color.WHITE });
|
|
|
|
LINES.put("FF1", new int[] { Color.BLUE, Color.WHITE }); // Potsdam
|
|
LINES.put("FF10", new int[] { Color.BLUE, Color.WHITE });
|
|
LINES.put("FF11", new int[] { Color.BLUE, Color.WHITE });
|
|
LINES.put("FF12", new int[] { Color.BLUE, Color.WHITE });
|
|
LINES.put("FF21", new int[] { Color.BLUE, Color.WHITE });
|
|
LINES.put("FF23", new int[] { Color.BLUE, Color.WHITE });
|
|
LINES.put("FF24", new int[] { Color.BLUE, Color.WHITE });
|
|
|
|
// Regional lines Brandenburg:
|
|
LINES.put("RRE1", new int[] { Color.parseColor("#EE1C23"), Color.WHITE });
|
|
LINES.put("RRE2", new int[] { Color.parseColor("#FFD403"), Color.BLACK });
|
|
LINES.put("RRE3", new int[] { Color.parseColor("#F57921"), Color.WHITE });
|
|
LINES.put("RRE4", new int[] { Color.parseColor("#952D4F"), Color.WHITE });
|
|
LINES.put("RRE5", new int[] { Color.parseColor("#0072BC"), Color.WHITE });
|
|
LINES.put("RRE6", new int[] { Color.parseColor("#DB6EAB"), Color.WHITE });
|
|
LINES.put("RRE7", new int[] { Color.parseColor("#00854A"), Color.WHITE });
|
|
LINES.put("RRE10", new int[] { Color.parseColor("#A7653F"), Color.WHITE });
|
|
LINES.put("RRE11", new int[] { Color.parseColor("#059EDB"), Color.WHITE });
|
|
LINES.put("RRE11", new int[] { Color.parseColor("#EE1C23"), Color.WHITE });
|
|
LINES.put("RRE15", new int[] { Color.parseColor("#FFD403"), Color.BLACK });
|
|
LINES.put("RRE18", new int[] { Color.parseColor("#00A65E"), Color.WHITE });
|
|
LINES.put("RRB10", new int[] { Color.parseColor("#60BB46"), Color.WHITE });
|
|
LINES.put("RRB12", new int[] { Color.parseColor("#A3238E"), Color.WHITE });
|
|
LINES.put("RRB13", new int[] { Color.parseColor("#F68B1F"), Color.WHITE });
|
|
LINES.put("RRB13", new int[] { Color.parseColor("#00A65E"), Color.WHITE });
|
|
LINES.put("RRB14", new int[] { Color.parseColor("#A3238E"), Color.WHITE });
|
|
LINES.put("RRB20", new int[] { Color.parseColor("#00854A"), Color.WHITE });
|
|
LINES.put("RRB21", new int[] { Color.parseColor("#5E6DB3"), Color.WHITE });
|
|
LINES.put("RRB22", new int[] { Color.parseColor("#0087CB"), Color.WHITE });
|
|
LINES.put("ROE25", new int[] { Color.parseColor("#0087CB"), Color.WHITE });
|
|
LINES.put("RNE26", new int[] { Color.parseColor("#00A896"), Color.WHITE });
|
|
LINES.put("RNE27", new int[] { Color.parseColor("#EE1C23"), Color.WHITE });
|
|
LINES.put("RRB30", new int[] { Color.parseColor("#00A65E"), Color.WHITE });
|
|
LINES.put("RRB31", new int[] { Color.parseColor("#60BB46"), Color.WHITE });
|
|
LINES.put("RMR33", new int[] { Color.parseColor("#EE1C23"), Color.WHITE });
|
|
LINES.put("ROE35", new int[] { Color.parseColor("#5E6DB3"), Color.WHITE });
|
|
LINES.put("ROE36", new int[] { Color.parseColor("#A7653F"), Color.WHITE });
|
|
LINES.put("RRB43", new int[] { Color.parseColor("#5E6DB3"), Color.WHITE });
|
|
LINES.put("RRB45", new int[] { Color.parseColor("#FFD403"), Color.BLACK });
|
|
LINES.put("ROE46", new int[] { Color.parseColor("#DB6EAB"), Color.WHITE });
|
|
LINES.put("RMR51", new int[] { Color.parseColor("#DB6EAB"), Color.WHITE });
|
|
LINES.put("RRB51", new int[] { Color.parseColor("#DB6EAB"), Color.WHITE });
|
|
LINES.put("RRB54", new int[] { Color.parseColor("#FFD403"), Color.parseColor("#333333") });
|
|
LINES.put("RRB55", new int[] { Color.parseColor("#F57921"), Color.WHITE });
|
|
LINES.put("ROE60", new int[] { Color.parseColor("#60BB46"), Color.WHITE });
|
|
LINES.put("ROE63", new int[] { Color.parseColor("#FFD403"), Color.BLACK });
|
|
LINES.put("ROE65", new int[] { Color.parseColor("#0072BC"), Color.WHITE });
|
|
LINES.put("RRB66", new int[] { Color.parseColor("#60BB46"), Color.WHITE });
|
|
LINES.put("RPE70", new int[] { Color.parseColor("#FFD403"), Color.BLACK });
|
|
LINES.put("RPE73", new int[] { Color.parseColor("#00A896"), Color.WHITE });
|
|
LINES.put("RPE74", new int[] { Color.parseColor("#0072BC"), Color.WHITE });
|
|
LINES.put("T89", new int[] { Color.parseColor("#EE1C23"), Color.WHITE });
|
|
LINES.put("RRB91", new int[] { Color.parseColor("#A7653F"), Color.WHITE });
|
|
LINES.put("RRB93", new int[] { Color.parseColor("#A7653F"), Color.WHITE });
|
|
}
|
|
|
|
public int[] lineColors(final String line)
|
|
{
|
|
return LINES.get(line);
|
|
}
|
|
}
|