--- /dev/null
+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();
+ }
+}
projects / GpsPrune.git / blobdiff