X-Git-Url: http://gitweb.fperrin.net/?p=GpsPrune.git;a=blobdiff_plain;f=src%2Ftim%2Fprune%2Ffunction%2FSearchWikipediaNames.java;fp=src%2Ftim%2Fprune%2Ffunction%2FSearchWikipediaNames.java;h=20b23deecb918ea0499321e98c81ca68a8a65789;hp=0000000000000000000000000000000000000000;hb=ce6f2161b8596f7018d6a76bff79bc9e571f35fd;hpb=2d8cb72e84d5cc1089ce77baf1e34ea3ea2f8465

diff --git a/src/tim/prune/function/SearchWikipediaNames.java b/src/tim/prune/function/SearchWikipediaNames.java
new file mode 100644
index 0000000..20b23de
--- /dev/null
+++ b/src/tim/prune/function/SearchWikipediaNames.java
@@ -0,0 +1,220 @@
+package tim.prune.function;
+
+import java.io.InputStream;
+import java.io.UnsupportedEncodingException;
+import java.net.URL;
+import java.net.URLEncoder;
+import java.util.ArrayList;
+
+import javax.swing.JOptionPane;
+import javax.xml.parsers.SAXParser;
+import javax.xml.parsers.SAXParserFactory;
+
+import tim.prune.App;
+import tim.prune.I18nManager;
+import tim.prune.data.DataPoint;
+import tim.prune.data.Field;
+import tim.prune.data.Latitude;
+import tim.prune.data.Longitude;
+import tim.prune.function.search.GenericDownloaderFunction;
+import tim.prune.function.search.SearchResult;
+
+/**
+ * Function to search Wikipedia for place names
+ */
+public class SearchWikipediaNames extends GenericDownloaderFunction
+{
+	/** search term */
+	private String _searchTerm = null;
+	/** Maximum number of results to get */
+	private static final int MAX_RESULTS = 20;
+	/** Username to use for geonames queries */
+	private static final String GEONAMES_USERNAME = "gpsprune";
+
+	/**
+	 * Constructor
+	 * @param inApp App object
+	 */
+	public SearchWikipediaNames(App inApp) {
+		super(inApp);
+	}
+
+	/**
+	 * @return name key
+	 */
+	public String getNameKey() {
+		return "function.searchwikipedianames";
+	}
+
+	/**
+	 * @param inColNum index of column, 0 or 1
+	 * @return key for this column
+	 */
+	protected String getColumnKey(int inColNum)
+	{
+		if (inColNum == 0) return "dialog.wikipedia.column.name";
+		return null;
+	}
+
+	/**
+	 * Before dialog is shown, need to get search term
+	 */
+	public void begin()
+	{
+		Object search = JOptionPane.showInputDialog(_app.getFrame(),
+			I18nManager.getText("dialog.searchwikipedianames.search"),
+			I18nManager.getText(getNameKey()),
+			JOptionPane.QUESTION_MESSAGE, null, null, "");
+		if (search != null)
+		{
+			_searchTerm = search.toString().toLowerCase();
+			if (!_searchTerm.equals("")) {
+				super.begin();
+			}
+		}
+	}
+
+	/**
+	 * Run method to call geonames in separate thread
+	 */
+	public void run()
+	{
+		_statusLabel.setText(I18nManager.getText("confirm.running"));
+
+		// Replace awkward characters with character equivalents
+		final String searchTerm = encodeSearchTerm(_searchTerm);
+
+		// Firstly try the local language
+		String lang = I18nManager.getText("wikipedia.lang");
+		submitSearch(searchTerm, lang);
+		// If we didn't get anything, try a secondary language
+		if (_trackListModel.isEmpty() && _errorMessage == null && lang.equals("als")) {
+			submitSearch(searchTerm, "de");
+		}
+		// If still nothing then try english
+		if (_trackListModel.isEmpty() && _errorMessage == null && !lang.equals("en")) {
+			submitSearch(searchTerm, "en");
+		}
+
+		// Set status label according to error or "none found", leave blank if ok
+		if (_errorMessage == null && _trackListModel.isEmpty()) {
+			_errorMessage = I18nManager.getText("dialog.wikipedia.nonefound");
+		}
+		_statusLabel.setText(_errorMessage == null ? "" : _errorMessage);
+	}
+
+	/**
+	 * Submit the given search to the server
+	 * @param inSearchTerm search term
+	 * @param inLang language code such as en, de
+	 */
+	private void submitSearch(String inSearchTerm, String inLang)
+	{
+		// System.out.println("Searching for '" + inSearchTerm + "' with lang: " + inLang);
+
+		String urlString = "http://api.geonames.org/wikipediaSearch?title=" + inSearchTerm
+			+ "&maxRows=" + MAX_RESULTS + "&lang=" + inLang + "&username=" + GEONAMES_USERNAME;
+		// Parse the returned XML with a special handler
+		GetWikipediaXmlHandler xmlHandler = new GetWikipediaXmlHandler();
+		InputStream inStream = null;
+
+		try
+		{
+			URL url = new URL(urlString);
+			SAXParser saxParser = SAXParserFactory.newInstance().newSAXParser();
+			inStream = url.openStream();
+			saxParser.parse(inStream, xmlHandler);
+		}
+		catch (Exception e) {
+			_errorMessage = e.getClass().getName() + " - " + e.getMessage();
+		}
+		// Close stream and ignore errors
+		try {
+			inStream.close();
+		} catch (Exception e) {}
+		// Add track list to model
+		ArrayList<SearchResult> trackList = xmlHandler.getTrackList();
+		// TODO: Do a better job of sorting replies by relevance - use three different lists
+		_trackListModel.addTracks(trackList);
+	}
+
+	/**
+	 * Replace tricky characters in the search term and encode the others
+	 * @param inSearchTerm entered search term
+	 * @return modified search term to give to geonames
+	 */
+	private static final String encodeSearchTerm(String inSearchTerm)
+	{
+		if (inSearchTerm != null && inSearchTerm.length() > 0)
+		{
+			// Replace umlauts oe, ue, ae, OE, UE, AE, szlig
+			StringBuilder sb = new StringBuilder();
+			final int numChars = inSearchTerm.length();
+			for (int i=0; i<numChars; i++)
+			{
+				char c = inSearchTerm.charAt(i);
+				switch (c)
+				{
+					// German umlauted vowels, add an "e"
+					case '\u00fc' : sb.append("ue"); break;
+					case '\u00e4' : sb.append("ae"); break;
+					case '\u00f6' : sb.append("oe"); break;
+					// German doppel s
+					case '\u00df' : sb.append("ss"); break;
+					// accented vowels
+					case '\u00e8' : case '\u00e9' :
+					case '\u00ea' : case '\u00eb' : sb.append('e'); break;
+					case '\u00e0' : case '\u00e1' :
+					case '\u00e2' : sb.append('a'); break;
+					case '\u00f2' : case '\u00f3' :
+					case '\u00f4' : sb.append('o'); break;
+					case '\u00ec' : case '\u00ed' :
+					case '\u00ee' : case '\u00ef' : sb.append('i'); break;
+					// cedillas, ny, l bar
+					case '\u00e7' : sb.append('c'); break;
+					case '\u015f' : sb.append('s'); break;
+					case '\u00f1' : sb.append('n'); break;
+					case '\u0142' : sb.append('l'); break;
+					// everything else
+					default  : sb.append(c);
+				}
+			}
+			String searchTerm = inSearchTerm;
+			try {
+				searchTerm = URLEncoder.encode(sb.toString(), "UTF-8");
+			} catch (UnsupportedEncodingException e1) {}
+			// System.out.println("Converted '" + inSearchTerm + "' to '" + searchTerm + "'");
+			return searchTerm;
+		}
+		return "";
+	}
+
+	/**
+	 * Load the selected point(s)
+	 */
+	protected void loadSelected()
+	{
+		// Find the rows selected in the table and get the corresponding coords
+		int numSelected = _trackTable.getSelectedRowCount();
+		if (numSelected < 1) return;
+		int[] rowNums = _trackTable.getSelectedRows();
+		for (int i=0; i<numSelected; i++)
+		{
+			int rowNum = rowNums[i];
+			if (rowNum >= 0 && rowNum < _trackListModel.getRowCount())
+			{
+				String lat = _trackListModel.getTrack(rowNum).getLatitude();
+				String lon = _trackListModel.getTrack(rowNum).getLongitude();
+				if (lat != null && lon != null)
+				{
+					DataPoint point = new DataPoint(new Latitude(lat), new Longitude(lon), null);
+					point.setFieldValue(Field.WAYPT_NAME, _trackListModel.getTrack(rowNum).getTrackName(), false);
+					_app.createPoint(point);
+				}
+			}
+		}
+		// Close the dialog
+		_cancelled = true;
+		_dialog.dispose();
+	}
+}