tfl is efa based

git-svn-id: https://public-transport-enabler.googlecode.com/svn/trunk@291 0924bc21-9374-b0fa-ee44-9ff1593b38f0
This commit is contained in:
andreas.schildbach 2010-10-15 18:40:52 +00:00
parent 51200b3b72
commit 6ddd09f16c
3 changed files with 139 additions and 174 deletions

View file

@ -29,6 +29,7 @@ import java.util.HashMap;
import java.util.LinkedList; import java.util.LinkedList;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.TimeZone;
import java.util.regex.Matcher; import java.util.regex.Matcher;
import java.util.regex.Pattern; import java.util.regex.Pattern;
@ -57,6 +58,11 @@ public abstract class AbstractEfaProvider implements NetworkProvider
{ {
private static final String DEFAULT_ENCODING = "ISO-8859-1"; private static final String DEFAULT_ENCODING = "ISO-8859-1";
protected TimeZone timeZone()
{
return TimeZone.getTimeZone("Europe/Berlin");
}
protected abstract String autocompleteUri(final CharSequence constraint); protected abstract String autocompleteUri(final CharSequence constraint);
public List<Location> autocompleteStations(final CharSequence constraint) throws IOException public List<Location> autocompleteStations(final CharSequence constraint) throws IOException
@ -353,6 +359,8 @@ public abstract class AbstractEfaProvider implements NetworkProvider
return 'I' + str; return 'I' + str;
if (type.equals("OIC")) // ÖBB-InterCity if (type.equals("OIC")) // ÖBB-InterCity
return 'I' + str; return 'I' + str;
if (type.equals("HT")) // First Hull Trains, GB
return 'I' + str;
if (type.equals("IR")) // Interregio if (type.equals("IR")) // Interregio
return 'R' + str; return 'R' + str;
@ -530,11 +538,55 @@ public abstract class AbstractEfaProvider implements NetworkProvider
return 'R' + str; return 'R' + str;
if (type.equals("VIAMO")) if (type.equals("VIAMO"))
return 'R' + str; return 'R' + str;
if (type.equals("SE")) // Southeastern, GB
return 'R' + str;
if (type.equals("SW")) // South West Trains, GB
return 'R' + str;
if (type.equals("SN")) // Southern, GB
return 'R' + str;
if (type.equals("NT")) // Northern Rail, GB
return 'R' + str;
if (type.equals("CH")) // Chiltern Railways, GB
return 'R' + str;
if (type.equals("EA")) // National Express East Anglia, GB
return 'R' + str;
if (type.equals("FC")) // First Capital Connect, GB
return 'R' + str;
if (type.equals("GW")) // First Great Western, GB
return 'R' + str;
if (type.equals("XC")) // Cross Country, GB, evtl. auch highspeed?
return 'R' + str;
if (type.equals("HC")) // Heathrow Connect, GB
return 'R' + str;
if (type.equals("HX")) // Heathrow Express, GB
return 'R' + str;
if (type.equals("GX")) // Gatwick Express, GB
return 'R' + str;
if (type.equals("C2C")) // c2c, GB
return 'R' + str;
if (type.equals("LM")) // London Midland, GB
return 'R' + str;
if (type.equals("EM")) // East Midlands Trains, GB
return 'R' + str;
if (type.equals("VT")) // Virgin Trains, GB, evtl. auch highspeed?
return 'R' + str;
if (type.equals("SR")) // ScotRail, GB, evtl. auch long-distance?
return 'R' + str;
if (type.equals("AW")) // Arriva Trains Wales, GB
return 'R' + str;
if (type.equals("WS")) // Wrexham & Shropshire, GB
return 'R' + str;
if (type.equals("TP")) // First TransPennine Express, GB, evtl. auch long-distance?
return 'R' + str;
if (type.equals("GC")) // Grand Central, GB
return 'R' + str;
if (type.equals("BSB")) // Breisgau-S-Bahn if (type.equals("BSB")) // Breisgau-S-Bahn
return 'S' + str; return 'S' + str;
if (type.equals("RER")) // Réseau Express Régional, Frankreich if (type.equals("RER")) // Réseau Express Régional, Frankreich
return 'S' + str; return 'S' + str;
if (type.equals("LO")) // London Overground, GB
return 'S' + str;
if (P_LINE_U.matcher(type).matches()) if (P_LINE_U.matcher(type).matches())
return 'U' + str; return 'U' + str;
@ -609,6 +661,7 @@ public abstract class AbstractEfaProvider implements NetworkProvider
final String location = normalizeLocationName(pp.nextText()); final String location = normalizeLocationName(pp.nextText());
final Calendar departureTime = new GregorianCalendar(); final Calendar departureTime = new GregorianCalendar();
departureTime.setTimeZone(timeZone());
final List<Departure> departures = new ArrayList<Departure>(8); final List<Departure> departures = new ArrayList<Departure>(8);
XmlPullUtil.jumpToStartTag(pp, null, "itdDepartureList"); XmlPullUtil.jumpToStartTag(pp, null, "itdDepartureList");
@ -843,6 +896,8 @@ public abstract class AbstractEfaProvider implements NetworkProvider
} }
final Calendar departureTime = new GregorianCalendar(), arrivalTime = new GregorianCalendar(); final Calendar departureTime = new GregorianCalendar(), arrivalTime = new GregorianCalendar();
departureTime.setTimeZone(timeZone());
arrivalTime.setTimeZone(timeZone());
final List<Connection> connections = new ArrayList<Connection>(); final List<Connection> connections = new ArrayList<Connection>();
if (XmlPullUtil.jumpToStartTag(pp, null, "itdRouteList")) if (XmlPullUtil.jumpToStartTag(pp, null, "itdRouteList"))

View file

@ -17,214 +17,77 @@
package de.schildbach.pte; package de.schildbach.pte;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Date; import java.util.Date;
import java.util.HashMap; import java.util.TimeZone;
import java.util.List;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import de.schildbach.pte.dto.Departure;
import de.schildbach.pte.dto.GetConnectionDetailsResult;
import de.schildbach.pte.dto.Location; import de.schildbach.pte.dto.Location;
import de.schildbach.pte.dto.NearbyStationsResult;
import de.schildbach.pte.dto.QueryConnectionsResult;
import de.schildbach.pte.dto.QueryDeparturesResult;
import de.schildbach.pte.dto.QueryDeparturesResult.Status;
import de.schildbach.pte.util.Color;
import de.schildbach.pte.util.ParserUtils;
/** /**
* @author Andreas Schildbach * @author Andreas Schildbach
*/ */
public class TflProvider implements NetworkProvider public class TflProvider extends AbstractEfaProvider
{ {
public static final String NETWORK_ID = "journeyplanner.tfl.gov.uk"; public static final String NETWORK_ID = "journeyplanner.tfl.gov.uk";
private static final String API_BASE = "http://journeyplanner.tfl.gov.uk/user/";
@Override
protected TimeZone timeZone()
{
return TimeZone.getTimeZone("Europe/London");
}
public boolean hasCapabilities(final Capability... capabilities) public boolean hasCapabilities(final Capability... capabilities)
{ {
for (final Capability capability : capabilities)
if (capability == Capability.DEPARTURES)
return true;
return false; return false;
} }
public List<Location> autocompleteStations(final CharSequence constraint) throws IOException @Override
protected String autocompleteUri(CharSequence constraint)
{ {
throw new UnsupportedOperationException(); throw new UnsupportedOperationException();
} }
public NearbyStationsResult nearbyStations(final String stationId, final int lat, final int lon, final int maxDistance, final int maxStations) private static final String NEARBY_STATION_URI = API_BASE
throws IOException + "XSLT_DM_REQUEST?outputFormat=XML&coordOutputFormat=WGS84&name_dm=%s&type_dm=stop&itOptionsActive=1&ptOptionsActive=1&useProxFootSearch=1&mergeDep=1&useAllStops=1&mode=direct";
@Override
protected String nearbyStationUri(final String stationId)
{ {
throw new UnsupportedOperationException(); return String.format(NEARBY_STATION_URI, stationId);
} }
public QueryConnectionsResult queryConnections(final Location from, final Location via, final Location to, final Date date, final boolean dep, @Override
final String products, final WalkSpeed walkSpeed) throws IOException protected String nearbyLatLonUri(int lat, int lon)
{ {
throw new UnsupportedOperationException(); return null;
}
public QueryConnectionsResult queryMoreConnections(final String uri) throws IOException
{
throw new UnsupportedOperationException();
}
public GetConnectionDetailsResult getConnectionDetails(final String connectionUri) throws IOException
{
throw new UnsupportedOperationException();
} }
public String departuresQueryUri(final String stationId, final int maxDepartures) public String departuresQueryUri(final String stationId, final int maxDepartures)
{ {
final StringBuilder uri = new StringBuilder(); final StringBuilder uri = new StringBuilder();
uri.append("http://journeyplanner.tfl.gov.uk/user/XSLT_DM_REQUEST?typeInfo_dm=stopID&mode=direct"); uri.append(API_BASE).append("XSLT_DM_REQUEST");
uri.append("&nameInfo_dm=").append(stationId); uri.append("?outputFormat=XML");
uri.append("&coordOutputFormat=WGS84");
uri.append("&type_dm=stop");
uri.append("&name_dm=").append(stationId);
uri.append("&mode=direct");
uri.append("&useRealtime=1");
return uri.toString(); return uri.toString();
} }
private static final Pattern P_DEPARTURES_HEAD_COARSE = Pattern.compile(".*?" // @Override
+ "(?:" // protected String connectionsQueryUri(Location from, Location via, Location to, Date date, boolean dep, String products, WalkSpeed walkSpeed)
+ "<div id=\"main-content\">(.*?)" // head
+ "<table id=\"timetable\">(.*?)</table>.*?" // departures
+ "|(hat diesen Ort nicht gefunden|Geben Sie einen Bahnhof)" // messages
+ ").*?" //
, Pattern.DOTALL);
private static final Pattern P_DEPARTURES_HEAD_FINE = Pattern.compile(".*?" //
+ "<h2>\\s*(?:Selected )?[Dd]epartures from: (.*?)</h2>.*?" // location
, Pattern.DOTALL);
private static final Pattern P_DEPARTURES_COARSE = Pattern.compile("<tr>(<td>\\d.*?)</tr>", Pattern.DOTALL);
private static final Pattern P_DEPARTURES_FINE = Pattern.compile("" //
+ "<td>(\\d{1,2}:\\d{2})</td>" // time
+ "<td><img src=\"assets/images/icon-\\w*.gif\" alt=\"(\\w*)\" hspace=\"1\" /><br />" // lineType
+ "(?:<span class=\"\\w*\">)?(.*?)(?:</span>)?</td>" // line
+ "<td>(.*?)</td>" // destination
, Pattern.DOTALL);
public QueryDeparturesResult queryDepartures(final String uri) throws IOException
{ {
final CharSequence page = ParserUtils.scrape(uri); throw new UnsupportedOperationException();
}
// parse page @Override
final Matcher mHeadCoarse = P_DEPARTURES_HEAD_COARSE.matcher(page); protected String commandLink(String sessionId, String command)
if (mHeadCoarse.matches())
{ {
// messages throw new UnsupportedOperationException();
if (mHeadCoarse.group(3) != null)
return new QueryDeparturesResult(uri, Status.INVALID_STATION);
final Matcher mHeadFine = P_DEPARTURES_HEAD_FINE.matcher(mHeadCoarse.group(1));
if (mHeadFine.matches())
{
final String location = ParserUtils.resolveEntities(mHeadFine.group(1));
final List<Departure> departures = new ArrayList<Departure>(8);
final Date now = new Date();
final Matcher mDepCoarse = P_DEPARTURES_COARSE.matcher(mHeadCoarse.group(2));
while (mDepCoarse.find())
{
final Matcher mDepFine = P_DEPARTURES_FINE.matcher(mDepCoarse.group(1));
if (mDepFine.matches())
{
final Date departureTime = ParserUtils.parseTime(mDepFine.group(1));
final Date departureDateTime = ParserUtils.joinDateTime(now, departureTime);
final String lineType = ParserUtils.resolveEntities(mDepFine.group(2));
final String line = normalizeLine(lineType, ParserUtils.resolveEntities(mDepFine.group(3)));
final String destination = ParserUtils.resolveEntities(mDepFine.group(4));
final Departure dep = new Departure(departureDateTime, line, line != null ? LINES.get(line.charAt(0)) : null, null, 0,
destination);
if (!departures.contains(dep))
departures.add(dep);
}
else
{
throw new IllegalArgumentException("cannot parse '" + mDepCoarse.group(1) + "' on " + uri);
}
}
return new QueryDeparturesResult(uri, 0, location, departures);
}
else
{
throw new IllegalArgumentException("cannot parse '" + mHeadCoarse.group(1) + "' on " + uri);
}
}
else
{
throw new IllegalArgumentException("cannot parse '" + page + "' on " + uri);
}
}
private String normalizeLine(final String type, final String line)
{
final char normalizedType = normalizeType(type);
final String normalizedLine = normalizeLine(line);
if (normalizedType != 0 && normalizedLine != null)
return normalizedType + normalizedLine;
throw new IllegalStateException("cannot normalize type '" + type + "' line '" + line + "'");
}
private char normalizeType(final String type)
{
if (type.equals("Tube"))
return 'U';
if (type.equals("DLR")) // Docklands Light Railway
return 'U';
if (type.equals("Bus"))
return 'B';
if (type.equals("Tram"))
return 'T';
if (type.equals("Rail"))
return '?';
return 0;
}
private static final Pattern P_LINE = Pattern.compile("(.*?) Line");
private static final Pattern P_ROUTE = Pattern.compile("Route (.*?)");
private String normalizeLine(final String line)
{
if (line.length() == 0)
return "";
if (line.equals("Docklands Light Railway"))
return "DLR";
if (line.equals("London Tramlink"))
return "Tramlink";
final Matcher mLine = P_LINE.matcher(line);
if (mLine.matches())
return mLine.group(1);
final Matcher mRoute = P_ROUTE.matcher(line);
if (mRoute.matches())
return mRoute.group(1);
return null;
}
private static final Map<Character, int[]> LINES = new HashMap<Character, int[]>();
static
{
LINES.put('I', new int[] { Color.WHITE, Color.RED, Color.RED });
LINES.put('R', new int[] { Color.GRAY, Color.WHITE });
LINES.put('S', new int[] { Color.parseColor("#006e34"), Color.WHITE });
LINES.put('U', new int[] { Color.parseColor("#003090"), Color.WHITE });
LINES.put('T', new int[] { Color.parseColor("#cc0000"), Color.WHITE });
LINES.put('B', new int[] { Color.parseColor("#993399"), Color.WHITE });
LINES.put('F', new int[] { Color.BLUE, Color.WHITE });
LINES.put('?', new int[] { Color.DKGRAY, Color.WHITE });
}
public int[] lineColors(final String line)
{
return LINES.get(line.charAt(0));
} }
} }

View file

@ -0,0 +1,47 @@
/*
* 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.live;
import org.junit.Test;
import de.schildbach.pte.TflProvider;
import de.schildbach.pte.dto.NearbyStationsResult;
import de.schildbach.pte.dto.QueryDeparturesResult;
/**
* @author Andreas Schildbach
*/
public class TflProviderLiveTest
{
private final TflProvider provider = new TflProvider();
@Test
public void nearbyStation() throws Exception
{
final NearbyStationsResult result = provider.nearbyStations("1000086", 0, 0, 0, 0);
System.out.println(result.stations.size() + " " + result.stations);
}
@Test
public void queryDepartures() throws Exception
{
final QueryDeparturesResult result = provider.queryDepartures(provider.departuresQueryUri("1000086", 0));
System.out.println(result.departures.size() + " " + result.departures);
}
}