--- android/TrainInfo/src/dk/thoerup/traininfo/provider/XmlDepartureProvider.java	2009/10/02 19:01:53	390
+++ android/TrainInfo/src/dk/thoerup/traininfo/provider/XmlDepartureProvider.java	2010/08/31 05:44:45	1024
@@ -1,8 +1,6 @@
 package dk.thoerup.traininfo.provider;
 
 import java.io.StringReader;
-import java.util.ArrayList;
-import java.util.List;
 
 import javax.xml.parsers.SAXParser;
 import javax.xml.parsers.SAXParserFactory;
@@ -15,56 +13,52 @@
 
 import android.util.Log;
 import dk.thoerup.traininfo.DepartureBean;
+import dk.thoerup.traininfo.DepartureEntry;
 import dk.thoerup.traininfo.util.AndroidTimeoutCache;
 import dk.thoerup.traininfo.util.DownloadUtil;
 import dk.thoerup.traininfo.util.XmlUtil;
 
-public class XmlDepartureProvider extends DefaultHandler implements DepartureProvider {
+public class XmlDepartureProvider implements DepartureProvider {
 
 	final static int CACHE_TIMEOUT = 60*1000;
+		
 	
-	class CacheEntry {
-		public long timestamp;
-		public List<DepartureBean> departures;
-	}
-	
-	
-	AndroidTimeoutCache<Integer,List<DepartureBean>> departureCache = new AndroidTimeoutCache<Integer,List<DepartureBean>>(CACHE_TIMEOUT); 
+	AndroidTimeoutCache<String,DepartureBean> departureCache = new AndroidTimeoutCache<String,DepartureBean>(CACHE_TIMEOUT); 
 	
-	List<DepartureBean> departures;
 		
 	
-	DepartureBean tempDeparture;
+	DepartureEntry tempDeparture;
 	StringBuilder builder = new StringBuilder(512);
 	
+
+	
 	@Override
-	public boolean lookupDepartures(int stationID) {		
-		boolean success;
+	public DepartureBean lookupDepartures(int stationID, boolean arrival) {		
 		
-		departures = departureCache.get(stationID);
+		String key = "" + stationID + ":" + arrival;
+		
+		DepartureBean departures = departureCache.get(key);
 	
 		if (departures == null) {			
-			success = lookupDeparturesWorker(stationID);
+			departures = lookupDeparturesWorker(stationID, arrival);
 			
-			if (success) {			
-				departureCache.put(stationID, departures);
+			if (departures != null) {			
+				departureCache.put(key, departures);
 			}
 			
 		} else {
 			Log.i("XmlDepartureProvider", "cache hit !!!");
-			success = true;
 		}		
 		
-		return success;
+		return departures;
 	}
 	
-	private boolean lookupDeparturesWorker(int stationID) {
-		boolean success = false;
-		departures = new ArrayList<DepartureBean>();
+	private DepartureBean lookupDeparturesWorker(int stationID, boolean arrival) {
+
 		try
 		{	
-
-			String url = XmlUtil.SERVICE_BASE + "/DepartureServlet?format=xml&station=" + stationID;
+			int iArrival = arrival ? 1 : 0;
+			String url = XmlUtil.SERVICE_BASE + "/DepartureServlet?format=xml&station=" + stationID + "&arrival=" + iArrival;
 			Log.i("xmlurl",url);
 			String doc =  DownloadUtil.getContentString(url, 45000, "ISO-8859-1");
 			
@@ -74,66 +68,70 @@
             SAXParser sp = spf.newSAXParser();
             XMLReader xr = sp.getXMLReader();
 
-			xr.setContentHandler(this);
-			xr.setErrorHandler(this);
+            DepartureParser departureParser = new DepartureParser();
+			xr.setContentHandler(departureParser);
+			xr.setErrorHandler(departureParser);
 			xr.parse(source);
-			success = true;
+			
+			return departureParser.getDepartures();
 			
 		} catch (Exception e) {
 			Log.e("XmlDepartureProvider", "looupFunction", e);
-		}
-		return success;
+			return null;
+		}		
 	}
 	
-	@Override
-	public List<DepartureBean> getDepartures(int station) {
-		List<DepartureBean> list = departureCache.get(station);
-		
-		if (list == null) {			
-			list = new ArrayList<DepartureBean>();
-		} 
+
+	class DepartureParser extends DefaultHandler { 
 		
-		return list;
-	}
-	
-	// this can be called several times fore the same text-node if there are many chardata / lines
-	@Override
-	public void characters (char ch[], int start, int length) 
-	{		
-		for (int i= start; i<start+length; i++)
-			builder.append(ch[i]);	
-	}
+		private DepartureBean departures = new DepartureBean(); 
+
+		public DepartureBean getDepartures() {
+			return departures;
+		}
 	
-	@Override
-	public void startElement (String uri, String name, String qName, Attributes atts)throws SAXException
-	{
-		if (name.equalsIgnoreCase("train"))
-			tempDeparture = new DepartureBean();
+		// this can be called several times fore the same text-node if there are many chardata / lines
+		@Override
+		public void characters (char ch[], int start, int length) 
+		{		
+			builder.append(ch, start, length);
+		}
 		
-		builder.setLength(0); //reset StringBuilder
-	}
-	
-	@Override
-	public void endElement (String uri, String name, String qName) throws SAXException
-	{
-		if (name.equals("train")) {
-			departures.add( tempDeparture );
-		} else if (name.equals("time")) {
-			tempDeparture.setTime(builder.toString().trim());
-		} else if (name.equals("updated")) {
-			tempDeparture.setLastUpdate(builder.toString().trim());
-		} else if (name.equals("trainnumber")) {
-			tempDeparture.setTrainNumber(builder.toString().trim());
-		} else if (name.equals("destination")) {
-			tempDeparture.setDestination(builder.toString().trim());
-		} else if (name.equals("origin")) {
-			tempDeparture.setOrigin(builder.toString().trim());
-		} else if (name.equals("location")) {
-			tempDeparture.setLocation(builder.toString().trim());
-		} else if (name.equals("status")) {
-			tempDeparture.setStatus(builder.toString().trim());
-		} else if (name.equals("note")) {
-			tempDeparture.setNote(builder.toString().trim());
-		} 
+		@Override
+		public void startElement (String uri, String name, String qName, Attributes atts)throws SAXException
+		{
+			if (name.equalsIgnoreCase("train"))
+				tempDeparture = new DepartureEntry();
+			
+			builder.setLength(0); //reset StringBuilder
+		}
+		
+		@Override
+		public void endElement (String uri, String name, String qName) throws SAXException
+		{
+			if (name.equals("train")) {			
+				departures.entries.add( tempDeparture );
+			} else if (name.equals("time")) {
+				tempDeparture.setTime(builder.toString().trim());
+			} else if (name.equals("updated")) {
+				tempDeparture.setLastUpdate(builder.toString().trim());
+			} else if (name.equals("trainnumber")) {
+				tempDeparture.setTrainNumber(builder.toString().trim());
+			} else if (name.equals("destination")) {
+				tempDeparture.setDestination(builder.toString().trim());
+			} else if (name.equals("origin")) {
+				tempDeparture.setOrigin(builder.toString().trim());
+			} else if (name.equals("location")) {
+				tempDeparture.setLocation(builder.toString().trim());
+			} else if (name.equals("status")) {
+				tempDeparture.setStatus(builder.toString().trim());
+			} else if (name.equals("note")) {
+				tempDeparture.setNote(builder.toString().trim());
+			} else if (name.equals("type")) {
+				tempDeparture.setType(builder.toString().trim());
+			} else if (name.equals("notification")) {
+				departures.notifications.add( builder.toString().trim() );
+			}
+		}
 	}
 }