diff --git a/src/de/schildbach/pte/NetworkId.java b/src/de/schildbach/pte/NetworkId.java
index d0a3186e..cf160d63 100644
--- a/src/de/schildbach/pte/NetworkId.java
+++ b/src/de/schildbach/pte/NetworkId.java
@@ -44,7 +44,7 @@ public enum NetworkId
DSB,
// Sweden
- SE,
+ SE, STOCKHOLM,
// Norway
NRI,
diff --git a/src/de/schildbach/pte/StockholmProvider.java b/src/de/schildbach/pte/StockholmProvider.java
new file mode 100644
index 00000000..49fd2ac0
--- /dev/null
+++ b/src/de/schildbach/pte/StockholmProvider.java
@@ -0,0 +1,207 @@
+/*
+ * Copyright 2010, 2011 the original author or authors.
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see .
+ */
+
+package de.schildbach.pte;
+
+import java.io.IOException;
+import java.util.List;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+import de.schildbach.pte.dto.Location;
+import de.schildbach.pte.dto.LocationType;
+import de.schildbach.pte.dto.NearbyStationsResult;
+import de.schildbach.pte.dto.QueryDeparturesResult;
+import de.schildbach.pte.util.ParserUtils;
+
+/**
+ * @author Andreas Schildbach
+ */
+public class StockholmProvider extends AbstractHafasProvider
+{
+ public static final NetworkId NETWORK_ID = NetworkId.STOCKHOLM;
+ private static final String API_BASE = "http://reseplanerare.trafiken.nu/bin/";
+
+ public StockholmProvider()
+ {
+ super(API_BASE + "query.exe/sn", 7, null);
+ }
+
+ public NetworkId id()
+ {
+ return NETWORK_ID;
+ }
+
+ public boolean hasCapabilities(final Capability... capabilities)
+ {
+ for (final Capability capability : capabilities)
+ if (capability == Capability.AUTOCOMPLETE_ONE_LINE || capability == Capability.DEPARTURES || capability == Capability.CONNECTIONS)
+ return true;
+
+ return false;
+ }
+
+ @Override
+ protected void setProductBits(final StringBuilder productBits, final char product)
+ {
+ if (product == 'I')
+ {
+ }
+ else if (product == 'R')
+ {
+ productBits.setCharAt(2, '1'); // Lokalbanor
+ }
+ else if (product == 'S')
+ {
+ productBits.setCharAt(0, '1'); // Pendeltåg
+ }
+ else if (product == 'U')
+ {
+ productBits.setCharAt(1, '1'); // Tunnelbana
+ }
+ else if (product == 'T')
+ {
+ }
+ else if (product == 'B')
+ {
+ productBits.setCharAt(3, '1'); // Bussar
+ productBits.setCharAt(4, '1'); // Flygbussar
+ }
+ else if (product == 'P')
+ {
+ }
+ else if (product == 'F')
+ {
+ productBits.setCharAt(6, '1'); // Waxholmsbåtar
+ }
+ else if (product == 'C')
+ {
+ }
+ else
+ {
+ throw new IllegalArgumentException("cannot handle: " + product);
+ }
+ }
+
+ private static final Pattern P_SPLIT_NAME_PAREN = Pattern.compile("(.*?) \\((.{4,}?)\\)");
+
+ @Override
+ protected String[] splitPlaceAndName(final String name)
+ {
+ final Matcher mParen = P_SPLIT_NAME_PAREN.matcher(name);
+ if (mParen.matches())
+ return new String[] { mParen.group(2), mParen.group(1) };
+
+ return super.splitPlaceAndName(name);
+ }
+
+ public NearbyStationsResult queryNearbyStations(final Location location, final int maxDistance, final int maxStations) throws IOException
+ {
+ final StringBuilder uri = new StringBuilder(API_BASE);
+
+ if (location.hasLocation())
+ {
+ uri.append("query.exe/sny");
+ uri.append("?performLocating=2&tpl=stop2json");
+ uri.append("&look_maxno=").append(maxStations != 0 ? maxStations : 200);
+ uri.append("&look_maxdist=").append(maxDistance != 0 ? maxDistance : 5000);
+ uri.append("&look_stopclass=").append(allProductsInt());
+ uri.append("&look_x=").append(location.lon);
+ uri.append("&look_y=").append(location.lat);
+
+ return jsonNearbyStations(uri.toString());
+ }
+ else if (location.type == LocationType.STATION && location.hasId())
+ {
+ uri.append("stboard.exe/sn");
+ uri.append("?productsFilter=").append(allProductsString());
+ uri.append("&boardType=dep");
+ uri.append("&input=").append(location.id);
+ uri.append("&sTI=1&start=yes&hcount=0&L=vs_java3");
+
+ return xmlNearbyStations(uri.toString());
+ }
+ else
+ {
+ throw new IllegalArgumentException("cannot handle: " + location.toDebugString());
+ }
+ }
+
+ public QueryDeparturesResult queryDepartures(final int stationId, final int maxDepartures, final boolean equivs) throws IOException
+ {
+ final StringBuilder uri = new StringBuilder();
+ uri.append(API_BASE).append("stboard.exe/sn");
+ uri.append("?productsFilter=").append(allProductsString());
+ uri.append("&boardType=dep");
+ uri.append("&disableEquivs=yes"); // don't use nearby stations
+ uri.append("&maxJourneys=50"); // ignore maxDepartures because result contains other stations
+ uri.append("&start=yes");
+ uri.append("&L=vs_java3");
+ uri.append("&input=").append(stationId);
+
+ return xmlQueryDepartures(uri.toString(), stationId);
+ }
+
+ private static final String AUTOCOMPLETE_URI = API_BASE
+ + "ajax-getstop.exe/sny?start=1&tpl=suggest2json&REQ0JourneyStopsS0A=7&getstop=1&noSession=yes&REQ0JourneyStopsB=12&REQ0JourneyStopsS0G=&S=%s";
+ private static final String ENCODING = "ISO-8859-1";
+
+ public List autocompleteStations(final CharSequence constraint) throws IOException
+ {
+ final String uri = String.format(AUTOCOMPLETE_URI, ParserUtils.urlEncode(constraint.toString(), ENCODING));
+
+ return jsonGetStops(uri);
+ }
+
+ @Override
+ protected char normalizeType(final String type)
+ {
+ final String ucType = type.toUpperCase();
+
+ if ("TRAIN".equals(ucType))
+ return 'R';
+ if ("NÄRTRAFIKEN".equals(ucType))
+ return 'R';
+ if ("LOKALTÅG".equals(ucType))
+ return 'R';
+
+ if ("PENDELTÅG".equals(ucType))
+ return 'S';
+
+ if ("METRO".equals(ucType))
+ return 'U';
+ if ("TUNNELBANA".equals(ucType))
+ return 'U';
+
+ if ("TRAM".equals(ucType))
+ return 'T';
+
+ if ("BUS".equals(ucType))
+ return 'B';
+ if ("BUSS".equals(ucType))
+ return 'B';
+ if ("FLYG".equals(ucType))
+ return 'B';
+
+ if ("SHIP".equals(ucType))
+ return 'F';
+ if ("BÅT".equals(ucType))
+ return 'F';
+
+ return 0;
+ }
+}
diff --git a/test/de/schildbach/pte/live/StockholmProviderLiveTest.java b/test/de/schildbach/pte/live/StockholmProviderLiveTest.java
new file mode 100644
index 00000000..0f2955b3
--- /dev/null
+++ b/test/de/schildbach/pte/live/StockholmProviderLiveTest.java
@@ -0,0 +1,90 @@
+/*
+ * Copyright 2010, 2011 the original author or authors.
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see .
+ */
+
+package de.schildbach.pte.live;
+
+import java.util.Date;
+import java.util.List;
+
+import org.junit.Test;
+
+import de.schildbach.pte.StockholmProvider;
+import de.schildbach.pte.NetworkProvider.WalkSpeed;
+import de.schildbach.pte.dto.Location;
+import de.schildbach.pte.dto.LocationType;
+import de.schildbach.pte.dto.NearbyStationsResult;
+import de.schildbach.pte.dto.QueryConnectionsResult;
+import de.schildbach.pte.dto.QueryDeparturesResult;
+
+/**
+ * @author Andreas Schildbach
+ */
+public class StockholmProviderLiveTest
+{
+ private final StockholmProvider provider = new StockholmProvider();
+ private static final String ALL_PRODUCTS = "IRSUTBFC";
+
+ @Test
+ public void nearbyStations() throws Exception
+ {
+ final NearbyStationsResult result = provider.queryNearbyStations(new Location(LocationType.STATION, 311109529), 0, 0);
+
+ System.out.println(result.stations.size() + " " + result.stations);
+ }
+
+ @Test
+ public void nearbyStationsByCoordinate() throws Exception
+ {
+ final NearbyStationsResult result = provider.queryNearbyStations(new Location(LocationType.ADDRESS, 59329897, 18072281), 0, 0);
+
+ System.out.println(result.stations.size() + " " + result.stations);
+ }
+
+ @Test
+ public void queryDepartures() throws Exception
+ {
+ final QueryDeparturesResult result = provider.queryDepartures(311109529, 0, false);
+
+ System.out.println(result.stationDepartures);
+ }
+
+ @Test
+ public void autocomplete() throws Exception
+ {
+ final List autocompletes = provider.autocompleteStations("Luleå Airport");
+
+ list(autocompletes);
+ }
+
+ private void list(final List autocompletes)
+ {
+ System.out.print(autocompletes.size() + " ");
+ for (final Location autocomplete : autocompletes)
+ System.out.print(autocomplete.toDebugString() + " ");
+ System.out.println();
+ }
+
+ @Test
+ public void shortConnection() throws Exception
+ {
+ final QueryConnectionsResult result = provider.queryConnections(new Location(LocationType.STATION, 200101051, "Stockholm", "T-Centralen"),
+ null, new Location(LocationType.STATION, 200101221, "Stockholm", "Abrahamsberg"), new Date(), true, ALL_PRODUCTS, WalkSpeed.NORMAL);
+ System.out.println(result);
+ final QueryConnectionsResult moreResult = provider.queryMoreConnections(result.context);
+ System.out.println(moreResult);
+ }
+}