mirror of
https://gitlab.com/oeffi/public-transport-enabler.git
synced 2025-07-08 18:48:49 +00:00
HSL: Remove provider for Finland.
This commit is contained in:
Andreas Schildbach
parent
38873cca19
commit
fe40fa2208
4 changed files with 0 additions and 860 deletions
|
|
@ -1,713 +0,0 @@
|
||||||
/*
|
|
||||||
* Copyright the original author or authors.
|
|
||||||
*
|
|
||||||
* This program is free software: you can redistribute it and/or modify
|
|
||||||
* it under the terms of the GNU General Public License as published by
|
|
||||||
* the Free Software Foundation, either version 3 of the License, or
|
|
||||||
* (at your option) any later version.
|
|
||||||
*
|
|
||||||
* This program is distributed in the hope that it will be useful,
|
|
||||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
||||||
* GNU General Public License for more details.
|
|
||||||
*
|
|
||||||
* You should have received a copy of the GNU General Public License
|
|
||||||
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
||||||
*/
|
|
||||||
|
|
||||||
package de.schildbach.pte;
|
|
||||||
|
|
||||||
import java.io.IOException;
|
|
||||||
import java.text.ParsePosition;
|
|
||||||
import java.text.SimpleDateFormat;
|
|
||||||
import java.util.ArrayList;
|
|
||||||
import java.util.Calendar;
|
|
||||||
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.Map;
|
|
||||||
import java.util.Set;
|
|
||||||
import java.util.concurrent.atomic.AtomicReference;
|
|
||||||
|
|
||||||
import javax.annotation.Nullable;
|
|
||||||
|
|
||||||
import org.xmlpull.v1.XmlPullParser;
|
|
||||||
import org.xmlpull.v1.XmlPullParserException;
|
|
||||||
import org.xmlpull.v1.XmlPullParserFactory;
|
|
||||||
|
|
||||||
import com.google.common.base.Joiner;
|
|
||||||
|
|
||||||
import de.schildbach.pte.dto.Departure;
|
|
||||||
import de.schildbach.pte.dto.Line;
|
|
||||||
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.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.Style;
|
|
||||||
import de.schildbach.pte.dto.SuggestLocationsResult;
|
|
||||||
import de.schildbach.pte.dto.SuggestedLocation;
|
|
||||||
import de.schildbach.pte.dto.Trip;
|
|
||||||
import de.schildbach.pte.dto.TripOptions;
|
|
||||||
import de.schildbach.pte.exception.ParserException;
|
|
||||||
import de.schildbach.pte.util.HttpClient;
|
|
||||||
import de.schildbach.pte.util.XmlPullUtil;
|
|
||||||
|
|
||||||
import okhttp3.HttpUrl;
|
|
||||||
import okhttp3.ResponseBody;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @author Mats Sjöberg <mats@sjoberg.fi>
|
|
||||||
*/
|
|
||||||
public class HslProvider extends AbstractNetworkProvider {
|
|
||||||
private static final HttpUrl API_BASE = HttpUrl.parse("https://api.reittiopas.fi/hsl/");
|
|
||||||
private static final String SERVER_PRODUCT = "hsl";
|
|
||||||
private static final String SERVER_VERSION = "1_2_0";
|
|
||||||
private static final int EARLIER_TRIPS_MINUTE_OFFSET = 5;
|
|
||||||
private static final int EARLIER_TRIPS_MINIMUM = 3;
|
|
||||||
|
|
||||||
private final XmlPullParserFactory parserFactory;
|
|
||||||
|
|
||||||
private String user;
|
|
||||||
private String pass;
|
|
||||||
|
|
||||||
public HslProvider(String user, String pass) {
|
|
||||||
super(NetworkId.HSL);
|
|
||||||
|
|
||||||
this.user = user;
|
|
||||||
this.pass = pass;
|
|
||||||
|
|
||||||
try {
|
|
||||||
parserFactory = XmlPullParserFactory.newInstance(System.getProperty(XmlPullParserFactory.PROPERTY_NAME),
|
|
||||||
null);
|
|
||||||
} catch (final XmlPullParserException x) {
|
|
||||||
throw new RuntimeException(x);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
protected boolean hasCapability(final Capability capability) {
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
private HttpUrl.Builder apiUrl(final String request) {
|
|
||||||
final HttpUrl.Builder url = API_BASE.newBuilder();
|
|
||||||
url.addPathSegment(SERVER_VERSION).addPathSegment("");
|
|
||||||
url.addQueryParameter("user", user);
|
|
||||||
url.addQueryParameter("pass", pass);
|
|
||||||
url.addQueryParameter("request", request);
|
|
||||||
url.addQueryParameter("epsg_out", "wgs84");
|
|
||||||
url.addQueryParameter("epsg_in", "wgs84");
|
|
||||||
url.addQueryParameter("format", "xml");
|
|
||||||
return url;
|
|
||||||
}
|
|
||||||
|
|
||||||
private Point coordStrToPoint(final String coordStr) {
|
|
||||||
if (coordStr == null)
|
|
||||||
return null;
|
|
||||||
|
|
||||||
final String[] parts = coordStr.split(",");
|
|
||||||
return Point.fromDouble(Double.parseDouble(parts[1]), Double.parseDouble(parts[0]));
|
|
||||||
}
|
|
||||||
|
|
||||||
private Point xmlCoordsToPoint(final XmlPullParser pp) throws XmlPullParserException, IOException {
|
|
||||||
XmlPullUtil.enter(pp, "coord");
|
|
||||||
String x = xmlValueTag(pp, "x");
|
|
||||||
String y = xmlValueTag(pp, "y");
|
|
||||||
XmlPullUtil.skipExit(pp, "coord");
|
|
||||||
return Point.fromDouble(Double.parseDouble(y), Double.parseDouble(x));
|
|
||||||
}
|
|
||||||
|
|
||||||
private String xmlValueTag(final XmlPullParser pp, final String tagName)
|
|
||||||
throws XmlPullParserException, IOException {
|
|
||||||
XmlPullUtil.skipUntil(pp, tagName);
|
|
||||||
return XmlPullUtil.valueTag(pp, tagName);
|
|
||||||
}
|
|
||||||
|
|
||||||
private String locationToCoords(final Location loc) {
|
|
||||||
return String.format("%2.6f,%2.6f", loc.getLonAsDouble(), loc.getLatAsDouble());
|
|
||||||
}
|
|
||||||
|
|
||||||
private Location queryStop(final String stationId) throws IOException {
|
|
||||||
final HttpUrl.Builder url = apiUrl("stop");
|
|
||||||
url.addQueryParameter("code", stationId);
|
|
||||||
url.addQueryParameter("dep_limit", "1");
|
|
||||||
final AtomicReference<Location> result = new AtomicReference<>();
|
|
||||||
|
|
||||||
final HttpClient.Callback callback = new HttpClient.Callback() {
|
|
||||||
@Override
|
|
||||||
public void onSuccessful(final CharSequence bodyPeek, final ResponseBody body) throws IOException {
|
|
||||||
try {
|
|
||||||
final XmlPullParser pp = parserFactory.newPullParser();
|
|
||||||
pp.setInput(body.charStream());
|
|
||||||
|
|
||||||
XmlPullUtil.enter(pp, "response");
|
|
||||||
XmlPullUtil.enter(pp, "node");
|
|
||||||
|
|
||||||
final String id = xmlValueTag(pp, "code");
|
|
||||||
final String name = xmlValueTag(pp, "name_fi");
|
|
||||||
final Point pt = coordStrToPoint(xmlValueTag(pp, "coords"));
|
|
||||||
result.set(new Location(LocationType.STATION, id, pt, null, name));
|
|
||||||
} catch (final XmlPullParserException x) {
|
|
||||||
throw new ParserException("cannot parse xml: " + bodyPeek, x);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
httpClient.getInputStream(callback, url.build());
|
|
||||||
return result.get();
|
|
||||||
}
|
|
||||||
|
|
||||||
// Determine stations near to given location. At least one of
|
|
||||||
// stationId or lat/lon pair must be present.
|
|
||||||
// NOTE: HSL returns only stops, not other locations, so "types" is not honoured.
|
|
||||||
@Override
|
|
||||||
public NearbyLocationsResult queryNearbyLocations(EnumSet<LocationType> types, Location location, int maxDistance,
|
|
||||||
int maxStations) throws IOException {
|
|
||||||
final HttpUrl.Builder url = apiUrl("stops_area");
|
|
||||||
if (!location.hasCoord()) {
|
|
||||||
if (location.type != LocationType.STATION)
|
|
||||||
throw new IllegalArgumentException("cannot handle: " + location.type);
|
|
||||||
if (!location.hasId())
|
|
||||||
throw new IllegalArgumentException("at least one of stationId or lat/lon " + "must be given");
|
|
||||||
location = queryStop(location.id);
|
|
||||||
}
|
|
||||||
url.addQueryParameter("center_coordinate", locationToCoords(location));
|
|
||||||
url.addQueryParameter("limit", Integer.toString(maxStations));
|
|
||||||
url.addQueryParameter("diameter", Integer.toString(maxDistance * 2));
|
|
||||||
final AtomicReference<NearbyLocationsResult> result = new AtomicReference<>();
|
|
||||||
|
|
||||||
final HttpClient.Callback callback = new HttpClient.Callback() {
|
|
||||||
@Override
|
|
||||||
public void onSuccessful(final CharSequence bodyPeek, final ResponseBody body) throws IOException {
|
|
||||||
try {
|
|
||||||
final XmlPullParser pp = parserFactory.newPullParser();
|
|
||||||
pp.setInput(body.charStream());
|
|
||||||
|
|
||||||
final List<Location> stations = new ArrayList<>();
|
|
||||||
final ResultHeader header = new ResultHeader(network, SERVER_PRODUCT);
|
|
||||||
|
|
||||||
XmlPullUtil.enter(pp, "response");
|
|
||||||
|
|
||||||
while (XmlPullUtil.test(pp, "node")) {
|
|
||||||
XmlPullUtil.enter(pp, "node");
|
|
||||||
|
|
||||||
final String id = xmlValueTag(pp, "code");
|
|
||||||
final String name = xmlValueTag(pp, "name");
|
|
||||||
final Point pt = coordStrToPoint(xmlValueTag(pp, "coords"));
|
|
||||||
final String place = xmlValueTag(pp, "address");
|
|
||||||
|
|
||||||
XmlPullUtil.skipExit(pp, "node");
|
|
||||||
|
|
||||||
stations.add(new Location(LocationType.STATION, id, pt, place, name));
|
|
||||||
}
|
|
||||||
|
|
||||||
result.set(new NearbyLocationsResult(header, stations));
|
|
||||||
} catch (final XmlPullParserException x) {
|
|
||||||
throw new ParserException("cannot parse xml: " + bodyPeek, x);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
httpClient.getInputStream(callback, url.build());
|
|
||||||
return result.get();
|
|
||||||
}
|
|
||||||
|
|
||||||
private Line newLine(String code, int type, String message) {
|
|
||||||
String label = code.substring(1, 5).trim().replaceAll("^0+", "");
|
|
||||||
Product product = Product.BUS;
|
|
||||||
char acode = code.substring(0, 1).charAt(0);
|
|
||||||
|
|
||||||
int color = 0xFF193695;
|
|
||||||
|
|
||||||
if (label.equals("1300M") || type == 6) {
|
|
||||||
label = "Metro";
|
|
||||||
product = Product.SUBWAY;
|
|
||||||
color = 0xFFfb6500;
|
|
||||||
} else if (label.equals("1019") || type == 7) {
|
|
||||||
label = "Ferry";
|
|
||||||
product = Product.FERRY;
|
|
||||||
} else if (type == 2) {
|
|
||||||
product = Product.TRAM;
|
|
||||||
color = 0xFF00ab67;
|
|
||||||
} else if (type == 12 || (type == 0 && acode == '3')) {
|
|
||||||
product = Product.REGIONAL_TRAIN;
|
|
||||||
color = 0xFF2cbe2c;
|
|
||||||
if ((label.charAt(0) == '1' || label.charAt(0) == '2') && label.length() == 2)
|
|
||||||
label = label.substring(1) + "-train";
|
|
||||||
}
|
|
||||||
|
|
||||||
Style style = new Style(color, Style.deriveForegroundColor(color));
|
|
||||||
|
|
||||||
return new Line(code, network.toString(), product, label, style, message);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Get departures at a given station, probably live
|
|
||||||
@Override
|
|
||||||
public QueryDeparturesResult queryDepartures(String stationId, @Nullable Date queryDate, final int maxDepartures,
|
|
||||||
boolean equivs) throws IOException {
|
|
||||||
final HttpUrl.Builder url = apiUrl("stop");
|
|
||||||
url.addQueryParameter("code", stationId);
|
|
||||||
if (queryDate != null) {
|
|
||||||
url.addQueryParameter("date", new SimpleDateFormat("yyyyMMdd").format(queryDate));
|
|
||||||
url.addQueryParameter("time", new SimpleDateFormat("HHmm").format(queryDate));
|
|
||||||
}
|
|
||||||
url.addQueryParameter("dep_limit", Integer.toString(maxDepartures));
|
|
||||||
final AtomicReference<QueryDeparturesResult> result = new AtomicReference<>();
|
|
||||||
|
|
||||||
final HttpClient.Callback callback = new HttpClient.Callback() {
|
|
||||||
@Override
|
|
||||||
public void onSuccessful(final CharSequence bodyPeek, final ResponseBody body) throws IOException {
|
|
||||||
try {
|
|
||||||
final XmlPullParser pp = parserFactory.newPullParser();
|
|
||||||
pp.setInput(body.charStream());
|
|
||||||
|
|
||||||
XmlPullUtil.enter(pp, "response");
|
|
||||||
XmlPullUtil.enter(pp, "node");
|
|
||||||
|
|
||||||
// FIXME: id is never used!?
|
|
||||||
final String id = xmlValueTag(pp, "code");
|
|
||||||
final String name = xmlValueTag(pp, "name_fi");
|
|
||||||
|
|
||||||
final Map<String, Line> lines = new HashMap<>();
|
|
||||||
|
|
||||||
XmlPullUtil.skipUntil(pp, "lines");
|
|
||||||
XmlPullUtil.enter(pp, "lines");
|
|
||||||
while (XmlPullUtil.test(pp, "node")) {
|
|
||||||
final String[] parts = XmlPullUtil.valueTag(pp, "node").split(":");
|
|
||||||
lines.put(parts[0], newLine(parts[0], 0, parts[1]));
|
|
||||||
}
|
|
||||||
XmlPullUtil.skipExit(pp, "lines");
|
|
||||||
|
|
||||||
final ResultHeader header = new ResultHeader(network, SERVER_PRODUCT);
|
|
||||||
final QueryDeparturesResult r = new QueryDeparturesResult(header);
|
|
||||||
|
|
||||||
XmlPullUtil.skipUntil(pp, "departures");
|
|
||||||
XmlPullUtil.enter(pp, "departures");
|
|
||||||
|
|
||||||
final List<Departure> departures = new ArrayList<>(maxDepartures);
|
|
||||||
while (XmlPullUtil.test(pp, "node")) {
|
|
||||||
XmlPullUtil.enter(pp, "node");
|
|
||||||
final String code = xmlValueTag(pp, "code");
|
|
||||||
final String time = xmlValueTag(pp, "time");
|
|
||||||
final String date = xmlValueTag(pp, "date");
|
|
||||||
XmlPullUtil.skipExit(pp, "node");
|
|
||||||
|
|
||||||
SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMddHHmm");
|
|
||||||
Date depDate = sdf.parse(date + time, new ParsePosition(0));
|
|
||||||
|
|
||||||
final Line line = lines.get(code);
|
|
||||||
final Location destination = new Location(LocationType.ANY, line.message, null, null);
|
|
||||||
final Departure departure = new Departure(depDate, null, line, null, destination, null, null);
|
|
||||||
departures.add(departure);
|
|
||||||
}
|
|
||||||
|
|
||||||
Location station = new Location(LocationType.STATION, id, null, name);
|
|
||||||
r.stationDepartures.add(new StationDepartures(station, departures, null));
|
|
||||||
result.set(r);
|
|
||||||
} catch (final XmlPullParserException x) {
|
|
||||||
throw new ParserException("cannot parse xml: " + bodyPeek, x);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
httpClient.getInputStream(callback, url.build());
|
|
||||||
return result.get();
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Meant for auto-completion of location names, like in an {@link android.widget.AutoCompleteTextView}
|
|
||||||
*
|
|
||||||
* @param constraint
|
|
||||||
* input by user so far
|
|
||||||
* @return location suggestions
|
|
||||||
* @throws IOException
|
|
||||||
*/
|
|
||||||
@Override
|
|
||||||
public SuggestLocationsResult suggestLocations(CharSequence constraint) throws IOException {
|
|
||||||
final HttpUrl.Builder url = apiUrl("geocode");
|
|
||||||
// Since HSL is picky about the input we clean out any
|
|
||||||
// character that isn't alphabetic, numeral, -, ', /
|
|
||||||
// or a space. Those should be all chars needed for a
|
|
||||||
// name.
|
|
||||||
String constraintStr = constraint.toString().replaceAll("[^\\p{Ll}\\p{Lu}\\p{Lt}\\p{Lo}\\p{Nd}\\d-'/ ]", "");
|
|
||||||
url.addQueryParameter("key", constraintStr);
|
|
||||||
final AtomicReference<SuggestLocationsResult> result = new AtomicReference<>();
|
|
||||||
|
|
||||||
final HttpClient.Callback callback = new HttpClient.Callback() {
|
|
||||||
@Override
|
|
||||||
public void onSuccessful(final CharSequence bodyPeek, final ResponseBody body) throws IOException {
|
|
||||||
try {
|
|
||||||
final ResultHeader header = new ResultHeader(network, SERVER_PRODUCT);
|
|
||||||
final List<SuggestedLocation> locations = new ArrayList<>();
|
|
||||||
|
|
||||||
final XmlPullParser pp = parserFactory.newPullParser();
|
|
||||||
pp.setInput(body.charStream());
|
|
||||||
|
|
||||||
XmlPullUtil.enter(pp, "response");
|
|
||||||
|
|
||||||
int weight = 10000;
|
|
||||||
|
|
||||||
while (XmlPullUtil.test(pp, "node")) {
|
|
||||||
XmlPullUtil.enter(pp, "node");
|
|
||||||
final String locType = xmlValueTag(pp, "locType");
|
|
||||||
String name = xmlValueTag(pp, "name");
|
|
||||||
final Point pt = coordStrToPoint(xmlValueTag(pp, "coords"));
|
|
||||||
|
|
||||||
LocationType type = LocationType.ANY;
|
|
||||||
if (locType.equals("poi"))
|
|
||||||
type = LocationType.POI;
|
|
||||||
if (locType.equals("address"))
|
|
||||||
type = LocationType.ADDRESS;
|
|
||||||
if (locType.equals("stop"))
|
|
||||||
type = LocationType.STATION;
|
|
||||||
|
|
||||||
XmlPullUtil.skipUntil(pp, "details");
|
|
||||||
XmlPullUtil.enter(pp, "details");
|
|
||||||
XmlPullUtil.optSkip(pp, "address");
|
|
||||||
final String id = XmlPullUtil.optValueTag(pp, "code", null);
|
|
||||||
final String shortCode = XmlPullUtil.optValueTag(pp, "shortCode", null);
|
|
||||||
XmlPullUtil.skipExit(pp, "details");
|
|
||||||
|
|
||||||
XmlPullUtil.skipExit(pp, "node");
|
|
||||||
|
|
||||||
if (shortCode != null)
|
|
||||||
name = name + " (" + shortCode + ")";
|
|
||||||
|
|
||||||
locations.add(new SuggestedLocation(new Location(type, id, pt, null, name), weight));
|
|
||||||
weight -= 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
result.set(new SuggestLocationsResult(header, locations));
|
|
||||||
} catch (final XmlPullParserException x) {
|
|
||||||
throw new ParserException("cannot parse xml: " + bodyPeek, x);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
httpClient.getInputStream(callback, url.build());
|
|
||||||
return result.get();
|
|
||||||
}
|
|
||||||
|
|
||||||
@SuppressWarnings("serial")
|
|
||||||
public static class QueryTripsHslContext implements QueryTripsContext {
|
|
||||||
public final String uri;
|
|
||||||
public Date date;
|
|
||||||
public Date prevDate = null;
|
|
||||||
public Date nextDate = null;
|
|
||||||
public final Location from;
|
|
||||||
public final Location via;
|
|
||||||
public final Location to;
|
|
||||||
public List<Trip> trips;
|
|
||||||
|
|
||||||
public QueryTripsHslContext(final String uri, final Location from, final Location via, final Location to,
|
|
||||||
final Date date) {
|
|
||||||
this.uri = uri;
|
|
||||||
this.from = from;
|
|
||||||
this.via = via;
|
|
||||||
this.to = to;
|
|
||||||
this.date = date;
|
|
||||||
this.trips = new ArrayList<>();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public boolean canQueryLater() {
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public boolean canQueryEarlier() {
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Query trips, asking for any ambiguousnesses
|
|
||||||
// NOTE: HSL ignores accessibility
|
|
||||||
@Override
|
|
||||||
public QueryTripsResult queryTrips(Location from, @Nullable Location via, Location to, Date date, boolean dep,
|
|
||||||
@Nullable TripOptions 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);
|
|
||||||
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);
|
|
||||||
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);
|
|
||||||
if (locations.size() > 1)
|
|
||||||
return new QueryTripsResult(header, null, null, locations);
|
|
||||||
to = locations.get(0);
|
|
||||||
}
|
|
||||||
|
|
||||||
final HttpUrl.Builder url = apiUrl("route");
|
|
||||||
|
|
||||||
url.addQueryParameter("from", locationToCoords(from));
|
|
||||||
if (via != null)
|
|
||||||
url.addQueryParameter("via", locationToCoords(via));
|
|
||||||
url.addQueryParameter("to", locationToCoords(to));
|
|
||||||
|
|
||||||
url.addQueryParameter("timetype", dep ? "departure" : "arrival");
|
|
||||||
|
|
||||||
if (options == null)
|
|
||||||
options = new TripOptions();
|
|
||||||
|
|
||||||
if (options.walkSpeed != WalkSpeed.NORMAL)
|
|
||||||
url.addQueryParameter("walk_speed", Integer.toString(options.walkSpeed == WalkSpeed.SLOW ? 30 : 100));
|
|
||||||
url.addQueryParameter("show", "5");
|
|
||||||
|
|
||||||
if (options.products != null && options.products.size() > 0) {
|
|
||||||
List<String> tt = new ArrayList<>();
|
|
||||||
if (options.products.contains(Product.HIGH_SPEED_TRAIN) || options.products.contains(Product.REGIONAL_TRAIN)
|
|
||||||
|| options.products.contains(Product.SUBURBAN_TRAIN))
|
|
||||||
tt.add("train");
|
|
||||||
if (options.products.contains(Product.SUBWAY))
|
|
||||||
tt.add("metro");
|
|
||||||
if (options.products.contains(Product.TRAM))
|
|
||||||
tt.add("tram");
|
|
||||||
if (options.products.contains(Product.BUS))
|
|
||||||
tt.add("bus");
|
|
||||||
if (options.products.contains(Product.FERRY))
|
|
||||||
tt.add("ferry");
|
|
||||||
|
|
||||||
if (tt.size() > 0)
|
|
||||||
url.addQueryParameter("transport_types", Joiner.on("|").join(tt));
|
|
||||||
}
|
|
||||||
|
|
||||||
QueryTripsHslContext context = new QueryTripsHslContext(url.build().toString(), from, via, to, date);
|
|
||||||
|
|
||||||
return queryHslTrips(from, via, to, context, date, true);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Query more trips (e.g. earlier or later)
|
|
||||||
*
|
|
||||||
* @param contextObj
|
|
||||||
* context to query more trips from
|
|
||||||
* @param later
|
|
||||||
* {@code true} to get later trips, {@code false} to get earlier trips
|
|
||||||
* @return result object that contains possible trips
|
|
||||||
* @throws IOException
|
|
||||||
*/
|
|
||||||
@Override
|
|
||||||
public QueryTripsResult queryMoreTrips(QueryTripsContext contextObj, boolean later) throws IOException {
|
|
||||||
final QueryTripsHslContext context = (QueryTripsHslContext) contextObj;
|
|
||||||
|
|
||||||
QueryTripsResult result;
|
|
||||||
|
|
||||||
if (later) {
|
|
||||||
result = queryHslTrips(context.from, context.via, context.to, context, context.nextDate, later);
|
|
||||||
} else {
|
|
||||||
// if we are fetching earlier trips, we have
|
|
||||||
// to do a hack to search backwards in small
|
|
||||||
// steps
|
|
||||||
|
|
||||||
int tries = 0;
|
|
||||||
int contextTrips = context.trips.size();
|
|
||||||
|
|
||||||
final Calendar cal = new GregorianCalendar(timeZone);
|
|
||||||
cal.setTime(context.prevDate);
|
|
||||||
|
|
||||||
do {
|
|
||||||
cal.add(Calendar.MINUTE, -EARLIER_TRIPS_MINUTE_OFFSET);
|
|
||||||
result = queryHslTrips(context.from, context.via, context.to, context, cal.getTime(), later);
|
|
||||||
|
|
||||||
tries += 1;
|
|
||||||
|
|
||||||
// keep trying if we are fetching earlier
|
|
||||||
// trips and the list of trips hasn't grown enough
|
|
||||||
} while ((result.trips.size() - contextTrips < EARLIER_TRIPS_MINIMUM) && tries < 10);
|
|
||||||
}
|
|
||||||
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
private QueryTripsResult queryHslTrips(final Location from, final Location via, final Location to,
|
|
||||||
final QueryTripsHslContext context, Date date, final boolean later) throws IOException {
|
|
||||||
final HttpUrl.Builder url = HttpUrl.parse(context.uri).newBuilder();
|
|
||||||
url.addQueryParameter("date", new SimpleDateFormat("yyyyMMdd").format(date));
|
|
||||||
url.addQueryParameter("time", new SimpleDateFormat("HHmm").format(date));
|
|
||||||
final AtomicReference<QueryTripsResult> result = new AtomicReference<>();
|
|
||||||
|
|
||||||
final HttpClient.Callback callback = new HttpClient.Callback() {
|
|
||||||
@Override
|
|
||||||
public void onSuccessful(final CharSequence bodyPeek, final ResponseBody body) throws IOException {
|
|
||||||
try {
|
|
||||||
final XmlPullParser pp = parserFactory.newPullParser();
|
|
||||||
pp.setInput(body.charStream());
|
|
||||||
|
|
||||||
XmlPullUtil.enter(pp, "response");
|
|
||||||
|
|
||||||
final ResultHeader header = new ResultHeader(network, SERVER_PRODUCT);
|
|
||||||
|
|
||||||
final List<Trip> trips = new ArrayList<>(context.trips);
|
|
||||||
|
|
||||||
// we use this for quick checking if trip already exists
|
|
||||||
Set<String> tripSet = new HashSet<>();
|
|
||||||
for (Trip t : trips)
|
|
||||||
tripSet.add(t.getId());
|
|
||||||
|
|
||||||
int insert = later ? trips.size() : 0;
|
|
||||||
|
|
||||||
while (XmlPullUtil.test(pp, "node")) {
|
|
||||||
XmlPullUtil.enter(pp, "node");
|
|
||||||
|
|
||||||
XmlPullUtil.enter(pp, "node");
|
|
||||||
|
|
||||||
List<Trip.Leg> legs = new ArrayList<>();
|
|
||||||
|
|
||||||
XmlPullUtil.skipUntil(pp, "legs");
|
|
||||||
XmlPullUtil.enter(pp, "legs");
|
|
||||||
int numTransfers = 0;
|
|
||||||
|
|
||||||
while (XmlPullUtil.test(pp, "node")) {
|
|
||||||
XmlPullUtil.enter(pp, "node");
|
|
||||||
|
|
||||||
int distance = Integer.parseInt(xmlValueTag(pp, "length"));
|
|
||||||
String legType = xmlValueTag(pp, "type");
|
|
||||||
String lineCode = XmlPullUtil.optValueTag(pp, "code", null);
|
|
||||||
|
|
||||||
List<Point> path = new ArrayList<>();
|
|
||||||
|
|
||||||
Location departure = null;
|
|
||||||
Date departureTime = null;
|
|
||||||
Stop departureStop = null;
|
|
||||||
|
|
||||||
Location arrival = null;
|
|
||||||
Date arrivalTime = null;
|
|
||||||
|
|
||||||
LinkedList<Stop> stops = new LinkedList<>();
|
|
||||||
|
|
||||||
XmlPullUtil.skipUntil(pp, "locs");
|
|
||||||
XmlPullUtil.enter(pp, "locs");
|
|
||||||
while (XmlPullUtil.test(pp, "node")) {
|
|
||||||
XmlPullUtil.enter(pp, "node");
|
|
||||||
Point pt = xmlCoordsToPoint(pp);
|
|
||||||
|
|
||||||
String arrTime = xmlValueTag(pp, "arrTime");
|
|
||||||
String depTime = xmlValueTag(pp, "depTime");
|
|
||||||
String name = XmlPullUtil.optValueTag(pp, "name", null);
|
|
||||||
String code = XmlPullUtil.optValueTag(pp, "code", null);
|
|
||||||
String shortCode = XmlPullUtil.optValueTag(pp, "shortCode", null);
|
|
||||||
String stopAddress = XmlPullUtil.optValueTag(pp, "stopAddress", null);
|
|
||||||
|
|
||||||
if (name == null) {
|
|
||||||
name = (path.size() == 0 && from != null && from.name != null) ? from.name : null;
|
|
||||||
}
|
|
||||||
|
|
||||||
SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMddHHmm");
|
|
||||||
Date arrDate = sdf.parse(arrTime, new ParsePosition(0));
|
|
||||||
Date depDate = sdf.parse(depTime, new ParsePosition(0));
|
|
||||||
|
|
||||||
LocationType type = LocationType.ANY;
|
|
||||||
if (code != null)
|
|
||||||
type = LocationType.STATION;
|
|
||||||
Location loc = new Location(type, code, pt, stopAddress, name);
|
|
||||||
|
|
||||||
if (path.size() == 0) {
|
|
||||||
departure = loc;
|
|
||||||
departureTime = depDate;
|
|
||||||
if (type == LocationType.STATION)
|
|
||||||
departureStop = new Stop(loc, true, departureTime, null, null, null);
|
|
||||||
|
|
||||||
} else {
|
|
||||||
arrival = loc;
|
|
||||||
arrivalTime = arrDate;
|
|
||||||
if (type == LocationType.STATION) {
|
|
||||||
stops.add(new Stop(loc, arrDate, null, depDate, null));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
path.add(pt);
|
|
||||||
XmlPullUtil.skipExit(pp, "node");
|
|
||||||
}
|
|
||||||
XmlPullUtil.skipExit(pp, "locs");
|
|
||||||
XmlPullUtil.skipExit(pp, "node");
|
|
||||||
|
|
||||||
if (legType.equals("walk")) {
|
|
||||||
// ugly hack to set the name of the last arrival
|
|
||||||
if (arrival != null && arrival.name == null) {
|
|
||||||
arrival = new Location(arrival.type, arrival.id, arrival.coord, arrival.place,
|
|
||||||
to.name);
|
|
||||||
}
|
|
||||||
|
|
||||||
legs.add(new Trip.Individual(Trip.Individual.Type.WALK, departure, departureTime,
|
|
||||||
arrival, arrivalTime, path, distance));
|
|
||||||
} else {
|
|
||||||
Stop arrivalStop = null;
|
|
||||||
if (stops.size() > 0) {
|
|
||||||
Stop last = stops.getLast();
|
|
||||||
arrivalStop = new Stop(last.location, false, last.plannedArrivalTime, null, null,
|
|
||||||
null);
|
|
||||||
stops.removeLast();
|
|
||||||
}
|
|
||||||
|
|
||||||
Line line = null;
|
|
||||||
if (lineCode != null)
|
|
||||||
line = newLine(lineCode, Integer.parseInt(legType), null);
|
|
||||||
|
|
||||||
legs.add(new Trip.Public(line, null, departureStop, arrivalStop, stops, path, null));
|
|
||||||
numTransfers++;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
XmlPullUtil.skipExit(pp, "legs");
|
|
||||||
XmlPullUtil.skipExit(pp, "node");
|
|
||||||
XmlPullUtil.skipExit(pp, "node");
|
|
||||||
|
|
||||||
Trip t = new Trip(null, from, to, legs, null, null, numTransfers - 1);
|
|
||||||
if (!tripSet.contains(t.getId())) {
|
|
||||||
Date thisTime = t.getFirstDepartureTime();
|
|
||||||
while (insert < trips.size() && thisTime.after(trips.get(insert).getFirstDepartureTime()))
|
|
||||||
insert++;
|
|
||||||
|
|
||||||
trips.add(insert++, t);
|
|
||||||
tripSet.add(t.getId());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Date lastDate = trips.get(trips.size() - 1).getFirstDepartureTime();
|
|
||||||
Date firstDate = trips.get(0).getFirstDepartureTime();
|
|
||||||
if (context.nextDate == null || lastDate.after(context.nextDate))
|
|
||||||
context.nextDate = lastDate;
|
|
||||||
if (context.prevDate == null || firstDate.before(context.prevDate))
|
|
||||||
context.prevDate = firstDate;
|
|
||||||
context.trips = trips;
|
|
||||||
|
|
||||||
result.set(new QueryTripsResult(header, url.build().toString(), from, via, to, context, trips));
|
|
||||||
} catch (final XmlPullParserException x) {
|
|
||||||
throw new ParserException("cannot parse xml: " + bodyPeek, x);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
context.date = date;
|
|
||||||
|
|
||||||
httpClient.getInputStream(callback, url.build());
|
|
||||||
return result.get();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
@ -51,9 +51,6 @@ public enum NetworkId {
|
||||||
// Sweden
|
// Sweden
|
||||||
SE,
|
SE,
|
||||||
|
|
||||||
// Finland
|
|
||||||
HSL,
|
|
||||||
|
|
||||||
// Luxembourg
|
// Luxembourg
|
||||||
LU,
|
LU,
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -1,142 +0,0 @@
|
||||||
/*
|
|
||||||
* Copyright the original author or authors.
|
|
||||||
*
|
|
||||||
* This program is free software: you can redistribute it and/or modify
|
|
||||||
* it under the terms of the GNU General Public License as published by
|
|
||||||
* the Free Software Foundation, either version 3 of the License, or
|
|
||||||
* (at your option) any later version.
|
|
||||||
*
|
|
||||||
* This program is distributed in the hope that it will be useful,
|
|
||||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
||||||
* GNU General Public License for more details.
|
|
||||||
*
|
|
||||||
* You should have received a copy of the GNU General Public License
|
|
||||||
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
||||||
*/
|
|
||||||
|
|
||||||
package de.schildbach.pte.live;
|
|
||||||
|
|
||||||
import static org.junit.Assert.assertEquals;
|
|
||||||
import static org.junit.Assert.assertTrue;
|
|
||||||
|
|
||||||
import java.util.Date;
|
|
||||||
import java.util.List;
|
|
||||||
|
|
||||||
import org.junit.Test;
|
|
||||||
|
|
||||||
import de.schildbach.pte.HslProvider;
|
|
||||||
import de.schildbach.pte.dto.Location;
|
|
||||||
import de.schildbach.pte.dto.LocationType;
|
|
||||||
import de.schildbach.pte.dto.NearbyLocationsResult;
|
|
||||||
import de.schildbach.pte.dto.QueryDeparturesResult;
|
|
||||||
import de.schildbach.pte.dto.QueryTripsResult;
|
|
||||||
import de.schildbach.pte.dto.SuggestLocationsResult;
|
|
||||||
import de.schildbach.pte.dto.Trip;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @author Mats Sjöberg <mats@sjoberg.fi>
|
|
||||||
*/
|
|
||||||
public class HslProviderLiveTest extends AbstractProviderLiveTest {
|
|
||||||
public HslProviderLiveTest() {
|
|
||||||
super(new HslProvider(secretProperty("hsl.usertoken"), secretProperty("hsl.passphrase")));
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
public void nearbyLocations() throws Exception {
|
|
||||||
final NearbyLocationsResult result = queryNearbyLocations(null, Location.coord(60174022, 24939222));
|
|
||||||
print(result);
|
|
||||||
assertEquals(NearbyLocationsResult.Status.OK, result.status);
|
|
||||||
assertTrue(result.locations.size() > 0);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
public void nearbyStationsByCoordinate() throws Exception {
|
|
||||||
final NearbyLocationsResult result = queryNearbyStations(Location.coord(60174022, 24939222));
|
|
||||||
print(result);
|
|
||||||
assertEquals(NearbyLocationsResult.Status.OK, result.status);
|
|
||||||
assertTrue(result.locations.size() > 0);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
public void queryDepartures() throws Exception {
|
|
||||||
final QueryDeparturesResult result = queryDepartures("1030424", false);
|
|
||||||
print(result);
|
|
||||||
assertEquals(QueryDeparturesResult.Status.OK, result.status);
|
|
||||||
assertTrue(result.stationDepartures.size() > 0);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
public void suggestLocationsIdentified() throws Exception {
|
|
||||||
final SuggestLocationsResult result = suggestLocations("3029");
|
|
||||||
print(result);
|
|
||||||
assertEquals(SuggestLocationsResult.Status.OK, result.status);
|
|
||||||
assertTrue(result.getLocations().size() == 1);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
public void suggestLocationsIncomplete() throws Exception {
|
|
||||||
final SuggestLocationsResult result = suggestLocations("Kum");
|
|
||||||
print(result);
|
|
||||||
assertEquals(SuggestLocationsResult.Status.OK, result.status);
|
|
||||||
assertTrue(result.getLocations().size() > 0);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
public void suggestLocationsWithUmlaut() throws Exception {
|
|
||||||
final SuggestLocationsResult result = suggestLocations("Jät");
|
|
||||||
print(result);
|
|
||||||
assertEquals(SuggestLocationsResult.Status.OK, result.status);
|
|
||||||
assertTrue(result.getLocations().size() > 0);
|
|
||||||
}
|
|
||||||
|
|
||||||
private void assertTimesInSequence(List<Trip> trips) {
|
|
||||||
for (int i = 1; i < trips.size(); i++) {
|
|
||||||
Date start1 = trips.get(i - 1).getFirstDepartureTime();
|
|
||||||
Date start2 = trips.get(i).getFirstDepartureTime();
|
|
||||||
assertTrue(i + ": " + start1 + " vs " + start2, start1.compareTo(start2) <= 0);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
public void shortTrip() throws Exception {
|
|
||||||
final QueryTripsResult result = queryTrips(
|
|
||||||
new Location(LocationType.STATION, null, "", "Gustaf Hällströmin katu 1"), null,
|
|
||||||
new Location(LocationType.STATION, null, "", "Tyynenmerenkatu 11"), new Date(), true, null);
|
|
||||||
print(result);
|
|
||||||
assertTimesInSequence(result.trips);
|
|
||||||
|
|
||||||
assertEquals(QueryTripsResult.Status.OK, result.status);
|
|
||||||
assertTrue(result.trips.size() > 0);
|
|
||||||
|
|
||||||
assertTrue(result.context.canQueryLater());
|
|
||||||
|
|
||||||
final QueryTripsResult laterResult = queryMoreTrips(result.context, true);
|
|
||||||
print(laterResult);
|
|
||||||
assertEquals(QueryTripsResult.Status.OK, laterResult.status);
|
|
||||||
assertTrue(laterResult.trips.size() > result.trips.size());
|
|
||||||
assertTimesInSequence(laterResult.trips);
|
|
||||||
|
|
||||||
assertTrue(laterResult.context.canQueryLater());
|
|
||||||
|
|
||||||
final QueryTripsResult later2Result = queryMoreTrips(laterResult.context, true);
|
|
||||||
print(later2Result);
|
|
||||||
for (Trip trip : later2Result.trips) {
|
|
||||||
System.out.println("LATER2 " + trip.getFirstDepartureTime() + " " + trip.getId());
|
|
||||||
}
|
|
||||||
assertEquals(QueryTripsResult.Status.OK, later2Result.status);
|
|
||||||
assertTrue(later2Result.trips.size() > laterResult.trips.size());
|
|
||||||
assertTimesInSequence(later2Result.trips);
|
|
||||||
|
|
||||||
assertTrue(later2Result.context.canQueryEarlier());
|
|
||||||
|
|
||||||
final QueryTripsResult earlierResult = queryMoreTrips(later2Result.context, false);
|
|
||||||
print(earlierResult);
|
|
||||||
for (Trip trip : earlierResult.trips) {
|
|
||||||
System.out.println("EARLIER " + trip.getFirstDepartureTime() + " " + trip.getId());
|
|
||||||
}
|
|
||||||
assertEquals(QueryTripsResult.Status.OK, earlierResult.status);
|
|
||||||
assertTrue(earlierResult.trips.size() > later2Result.trips.size());
|
|
||||||
assertTimesInSequence(earlierResult.trips);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
@ -18,5 +18,3 @@ zvv.api_authorization =
|
||||||
dsb.api_authorization =
|
dsb.api_authorization =
|
||||||
se.api_authorization =
|
se.api_authorization =
|
||||||
lu.api_authorization =
|
lu.api_authorization =
|
||||||
hsl.usertoken =
|
|
||||||
hsl.passphrase =
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue