]> gitweb.fperrin.net Git - DictionaryPC.git/blobdiff - src/com/hughes/android/dictionary/parser/enwiktionary/FunctionCallbacksDefault.java
Stoplist, more languages...
[DictionaryPC.git] / src / com / hughes / android / dictionary / parser / enwiktionary / FunctionCallbacksDefault.java
index fde94f709f9a692fc53f8bd01958ca180d085b4c..a50e34c6ee3d2a4265eb9b96bc25f20130448baf 100644 (file)
@@ -1,14 +1,31 @@
+// Copyright 2012 Google Inc. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
 package com.hughes.android.dictionary.parser.enwiktionary;
 
 import java.util.Arrays;
+import java.util.Collections;
 import java.util.LinkedHashMap;
 import java.util.LinkedHashSet;
 import java.util.List;
 import java.util.Map;
 import java.util.Set;
+import java.util.concurrent.atomic.AtomicInteger;
 import java.util.logging.Logger;
 
 import com.hughes.android.dictionary.engine.EntryTypeName;
+import com.hughes.android.dictionary.engine.IndexBuilder;
 import com.hughes.android.dictionary.parser.WikiTokenizer;
 import com.hughes.util.ListUtil;
 
@@ -17,6 +34,7 @@ public final class FunctionCallbacksDefault {
   static final Logger LOG = Logger.getLogger(EnWiktionaryXmlParser.class.getName());
   
   static final Map<String,FunctionCallback> DEFAULT = new LinkedHashMap<String, FunctionCallback>();
+  
   static {
     FunctionCallback callback = new TranslationCallback();
     DEFAULT.put("t", callback);
@@ -25,8 +43,6 @@ public final class FunctionCallbacksDefault {
     DEFAULT.put("tø", callback);
     DEFAULT.put("apdx-t", callback);
     
-    DEFAULT.put("qualifier", new QualifierCallback());
-
     callback = new EncodingCallback();
     Set<String> encodings = new LinkedHashSet<String>(Arrays.asList(
         "zh-ts", "zh-tsp",
@@ -38,35 +54,59 @@ public final class FunctionCallbacksDefault {
       DEFAULT.put(encoding, callback);
     }
     
+    callback = new l_term();
+    DEFAULT.put("l", callback);
+    DEFAULT.put("term", callback);
+
     callback = new Gender();
     DEFAULT.put("m", callback);
     DEFAULT.put("f", callback);
     DEFAULT.put("n", callback);
     DEFAULT.put("p", callback);
     DEFAULT.put("g", callback);
-
-    DEFAULT.put("l", new l());
-    DEFAULT.put("italbrac", new italbrac());
-    DEFAULT.put("gloss", new gloss());
-
+    
     callback = new AppendArg0();
-    DEFAULT.put("term", callback);
 
     callback = new Ignore();
     DEFAULT.put("trreq", callback);
     DEFAULT.put("t-image", callback);
     DEFAULT.put("defn", callback);
     DEFAULT.put("rfdef", callback);
+    DEFAULT.put("rfdate", callback);
+    DEFAULT.put("rfex", callback);
+    DEFAULT.put("rfquote", callback);
     DEFAULT.put("attention", callback);
     DEFAULT.put("zh-attention", callback);
 
-    DEFAULT.put("not used", new not_used());
-    DEFAULT.put("form of", new FormOf());
-    DEFAULT.put("wikipedia", new wikipedia());
+
+    callback = new FormOf();
+    DEFAULT.put("form of", callback);
+    DEFAULT.put("conjugation of", callback);
+    DEFAULT.put("participle of", callback);
+    DEFAULT.put("present participle of", callback);
+    DEFAULT.put("past participle of", callback);
+    DEFAULT.put("feminine past participle of", callback);
+    DEFAULT.put("gerund of", callback);
+    DEFAULT.put("feminine of", callback);
+    DEFAULT.put("plural of", callback);
+    DEFAULT.put("feminine plural of", callback);
+    DEFAULT.put("inflected form of", callback);
+    DEFAULT.put("alternative form of", callback);
+    DEFAULT.put("dated form of", callback);
+    DEFAULT.put("apocopic form of", callback);
     
     callback = new InflOrHead();
     DEFAULT.put("infl", callback);
     DEFAULT.put("head", callback);
+    
+    callback = new AppendName();
+    DEFAULT.put("...", callback);
+    
+    DEFAULT.put("qualifier", new QualifierCallback());
+    DEFAULT.put("italbrac", new italbrac());
+    DEFAULT.put("gloss", new gloss());
+    DEFAULT.put("not used", new not_used());
+    DEFAULT.put("wikipedia", new wikipedia());
   }
 
   
@@ -76,24 +116,39 @@ public final class FunctionCallbacksDefault {
         final Map<String, String> namedArgs, final EnWiktionaryXmlParser parser,
         final AppendAndIndexWikiCallback appendAndIndexWikiCallback) {
       
-      appendAndIndexWikiCallback.builder.append(name);
+      if (name != null) {
+        appendAndIndexWikiCallback.builder.append(name);
+      }
       for (int i = 0; i < args.size(); ++i) {
         if (args.get(i).length() > 0) {
           appendAndIndexWikiCallback.builder.append("|");
           appendAndIndexWikiCallback.dispatch(args.get(i), null, null);
         }
       }
-      for (final Map.Entry<String, String> entry : namedArgs.entrySet()) {
-        appendAndIndexWikiCallback.builder.append("|");
-        appendAndIndexWikiCallback.dispatch(entry.getKey(), null, null);
-        appendAndIndexWikiCallback.builder.append("=");
-        appendAndIndexWikiCallback.dispatch(entry.getValue(), null, null);
-      }
+      appendNamedArgs(namedArgs, appendAndIndexWikiCallback);
       return true;
     }
   }
   static NameAndArgs NAME_AND_ARGS = new NameAndArgs();
 
+  static void appendNamedArgs(final Map<String, String> namedArgs,
+      final AppendAndIndexWikiCallback appendAndIndexWikiCallback) {
+    for (final Map.Entry<String, String> entry : namedArgs.entrySet()) {
+      appendAndIndexWikiCallback.builder.append("|");
+      appendAndIndexWikiCallback.dispatch(entry.getKey(), null, null);
+      appendAndIndexWikiCallback.builder.append("=");
+      EntryTypeName entryTypeName = null;
+      IndexBuilder indexBuilder = null;
+      // This doesn't work: we'd need to add to word-forms.
+//      System.out.println(entry.getKey());
+//      if (entry.getKey().equals("tr")) {
+//        entryTypeName = EntryTypeName.WIKTIONARY_TRANSLITERATION;
+//        indexBuilder = appendAndIndexWikiCallback.parser.foreignIndexBuilder;
+//      }
+      appendAndIndexWikiCallback.dispatch(entry.getValue(), indexBuilder, entryTypeName);
+    }
+  }
+
   // ------------------------------------------------------------------
 
   static final class TranslationCallback implements FunctionCallback {
@@ -103,26 +158,53 @@ public final class FunctionCallbacksDefault {
         final AppendAndIndexWikiCallback appendAndIndexWikiCallback) {
 
       final String transliteration = namedArgs.remove("tr");
+      final String alt = namedArgs.remove("alt");
       namedArgs.keySet().removeAll(EnWiktionaryXmlParser.USELESS_WIKI_ARGS);
       if (args.size() < 2) {
         LOG.warning("{{t...}} with wrong args: title=" + parser.title);
         return false;
       }
       final String langCode = ListUtil.get(args, 0);
+      if (!appendAndIndexWikiCallback.langCodeToTCount.containsKey(langCode)) {
+        appendAndIndexWikiCallback.langCodeToTCount.put(langCode, new AtomicInteger());
+      }
+      appendAndIndexWikiCallback.langCodeToTCount.get(langCode).incrementAndGet();
       final String word = ListUtil.get(args, 1);
-      final String gender = ListUtil.get(args, 2);
-      // TODO: deal with second (and third...) gender, and alt.
-      
-      appendAndIndexWikiCallback.dispatch(word, EntryTypeName.WIKTIONARY_TITLE_MULTI);
-      
-      if (gender != null) {
-        appendAndIndexWikiCallback.builder.append(String.format(" {%s}", gender));
+      appendAndIndexWikiCallback.dispatch(alt != null ? alt : word, EntryTypeName.WIKTIONARY_TITLE_MULTI);
+
+      // Genders...
+      if (args.size() > 2) {
+        appendAndIndexWikiCallback.builder.append(" {");
+        for (int i = 2; i < args.size(); ++i) {
+          if (i > 2) {
+            appendAndIndexWikiCallback.builder.append("|");
+          }
+          appendAndIndexWikiCallback.builder.append(args.get(i));
+        }
+        appendAndIndexWikiCallback.builder.append("}");
       }
+
       if (transliteration != null) {
-        appendAndIndexWikiCallback.builder.append(" (tr. ");
+        appendAndIndexWikiCallback.builder.append(" (");
         appendAndIndexWikiCallback.dispatch(transliteration, EntryTypeName.WIKTIONARY_TRANSLITERATION);
         appendAndIndexWikiCallback.builder.append(")");
       }
+      
+      if (alt != null) {
+        // If alt wasn't null, we appended alt instead of the actual word
+        // we're filing under..
+        appendAndIndexWikiCallback.builder.append(" (");
+        appendAndIndexWikiCallback.dispatch(word, EntryTypeName.WIKTIONARY_TITLE_MULTI);
+        appendAndIndexWikiCallback.builder.append(")");
+      }
+
+      // Catch-all for anything else...
+      if (!namedArgs.isEmpty()) {
+        appendAndIndexWikiCallback.builder.append(" {");
+        FunctionCallbacksDefault.appendNamedArgs(namedArgs, appendAndIndexWikiCallback);
+        appendAndIndexWikiCallback.builder.append("}");
+      }
+      
       return true;
     }
     
@@ -156,11 +238,25 @@ public final class FunctionCallbacksDefault {
         final Map<String, String> namedArgs,
         final EnWiktionaryXmlParser parser,
         final AppendAndIndexWikiCallback appendAndIndexWikiCallback) {
-      if (args.size() != 1 || !namedArgs.isEmpty()) {
+      if (!namedArgs.isEmpty()) {
         LOG.warning("weird encoding: " + wikiTokenizer.token());
       }
-      final String wikiText = args.get(0);
-      appendAndIndexWikiCallback.dispatch(wikiText, appendAndIndexWikiCallback.entryTypeName);
+      if (args.size() == 0) {
+        // Things like "{{Jpan}}" exist.
+        return true;
+      }
+      
+      for (int i = 0; i < args.size(); ++i) {
+        if (i > 0) {
+          appendAndIndexWikiCallback.builder.append(", ");
+        }
+        final String arg = args.get(i);
+//        if (arg.equals(parser.title)) {
+//          parser.titleAppended = true;
+//        }
+        appendAndIndexWikiCallback.dispatch(arg, appendAndIndexWikiCallback.entryTypeName);
+      }
+      
       return true;
     }
   }
@@ -188,29 +284,67 @@ public final class FunctionCallbacksDefault {
 
   // ------------------------------------------------------------------
   
-  static final class l implements FunctionCallback {
+  static final class l_term implements FunctionCallback {
     @Override
     public boolean onWikiFunction(final WikiTokenizer wikiTokenizer, final String name, final List<String> args,
         final Map<String, String> namedArgs,
         final EnWiktionaryXmlParser parser,
         final AppendAndIndexWikiCallback appendAndIndexWikiCallback) {
-      // TODO: rewrite this!
-      // encodes text in various langs.
-      // lang is arg 0.
-      // 
+      
+      // for {{l}}, lang is arg 0, but not for {{term}}
+      if (name.equals("term")) {
+        args.add(0, "");
+      }
+      
       final EntryTypeName entryTypeName;
       switch (parser.state) {
       case TRANSLATION_LINE: entryTypeName = EntryTypeName.WIKTIONARY_TRANSLATION_OTHER_TEXT; break;
       case ENGLISH_DEF_OF_FOREIGN: entryTypeName = EntryTypeName.WIKTIONARY_ENGLISH_DEF_WIKI_LINK; break;
       default: throw new IllegalStateException("Invalid enum value: " + parser.state);
       }
+      
       final String langCode = args.get(0);
-      if ("en".equals(langCode)) {
-        appendAndIndexWikiCallback.dispatch(args.get(1), parser.enIndexBuilder, entryTypeName);
+      final IndexBuilder indexBuilder;
+      if ("".equals(langCode)) {
+        indexBuilder = parser.foreignIndexBuilder;
+      } else if ("en".equals(langCode)) {
+        indexBuilder = parser.enIndexBuilder;
       } else {
-        appendAndIndexWikiCallback.dispatch(args.get(1), parser.foreignIndexBuilder, entryTypeName);
+        indexBuilder = parser.foreignIndexBuilder;
       }
-      // TODO: transliteration
+      
+      String displayText = ListUtil.get(args, 2, "");
+      if (displayText.equals("")) {
+        displayText = ListUtil.get(args, 1, null);
+      }
+      
+      if (displayText != null) {
+        appendAndIndexWikiCallback.dispatch(displayText, indexBuilder, entryTypeName);
+      } else {
+        LOG.warning("no display text: " + wikiTokenizer.token());
+      }
+      
+      final String tr = namedArgs.remove("tr");
+      if (tr != null) {
+        appendAndIndexWikiCallback.builder.append(" (");
+        appendAndIndexWikiCallback.dispatch(tr, indexBuilder, EntryTypeName.WIKTIONARY_TRANSLITERATION);
+        appendAndIndexWikiCallback.builder.append(")");
+      }
+      
+      final String gloss = ListUtil.get(args, 3, "");
+      if (!gloss.equals("")) {
+        appendAndIndexWikiCallback.builder.append(" (");
+        appendAndIndexWikiCallback.dispatch(gloss, parser.enIndexBuilder, EntryTypeName.WIKTIONARY_ENGLISH_DEF);
+        appendAndIndexWikiCallback.builder.append(")");
+      }
+      
+      namedArgs.keySet().removeAll(EnWiktionaryXmlParser.USELESS_WIKI_ARGS);
+      if (!namedArgs.isEmpty()) {
+        appendAndIndexWikiCallback.builder.append(" {").append(name);
+        appendNamedArgs(namedArgs, appendAndIndexWikiCallback);
+        appendAndIndexWikiCallback.builder.append("}");
+      }
+
       return true;
     }
   }
@@ -243,9 +377,9 @@ public final class FunctionCallbacksDefault {
       if (args.size() != 1 || !namedArgs.isEmpty()) {
         return false;
       }
-      appendAndIndexWikiCallback.builder.append("[");
+      appendAndIndexWikiCallback.builder.append("(");
       appendAndIndexWikiCallback.dispatch(args.get(0), EntryTypeName.WIKTIONARY_TRANSLATION_OTHER_TEXT);
-      appendAndIndexWikiCallback.builder.append("]");
+      appendAndIndexWikiCallback.builder.append(")");
       return true;
     }
   }
@@ -261,9 +395,9 @@ public final class FunctionCallbacksDefault {
       if (args.size() != 1 || !namedArgs.isEmpty()) {
         return false;
       }
-      appendAndIndexWikiCallback.builder.append("[");
+      appendAndIndexWikiCallback.builder.append("(");
       appendAndIndexWikiCallback.dispatch(args.get(0), EntryTypeName.WIKTIONARY_TRANSLATION_OTHER_TEXT);
-      appendAndIndexWikiCallback.builder.append("]");
+      appendAndIndexWikiCallback.builder.append(")");
       return true;
     }
   }
@@ -293,7 +427,23 @@ public final class FunctionCallbacksDefault {
     }
   }
 
+
+  // ------------------------------------------------------------------
   
+  static final class AppendName implements FunctionCallback {
+    @Override
+    public boolean onWikiFunction(final WikiTokenizer wikiTokenizer, final String name, final List<String> args,
+        final Map<String, String> namedArgs,
+        final EnWiktionaryXmlParser parser,
+        final AppendAndIndexWikiCallback appendAndIndexWikiCallback) {
+      if (!args.isEmpty() || !namedArgs.isEmpty()) {
+        return false;
+      }
+      appendAndIndexWikiCallback.builder.append(name);
+      return true;
+    }
+  }
+
   // --------------------------------------------------------------------
   // --------------------------------------------------------------------
   
@@ -304,6 +454,7 @@ public final class FunctionCallbacksDefault {
         final Map<String, String> namedArgs,
         final EnWiktionaryXmlParser parser,
         final AppendAndIndexWikiCallback appendAndIndexWikiCallback) {
+      parser.entryIsFormOfSomething = true;
       String formName = name;
       if (name.equals("form of")) {
         formName = ListUtil.remove(args, 0, null);
@@ -372,8 +523,6 @@ public final class FunctionCallbacksDefault {
       }
       if (head == null) {
         head = parser.title;
-      } else {
-        head = WikiTokenizer.toPlainText(head);
       }
       parser.titleAppended = true;
       
@@ -401,7 +550,7 @@ public final class FunctionCallbacksDefault {
       }
 
       if (tr != null) {
-        appendAndIndexWikiCallback.builder.append(" (tr. ");
+        appendAndIndexWikiCallback.builder.append(" (");
         appendAndIndexWikiCallback.dispatch(tr, EntryTypeName.WIKTIONARY_TITLE_MULTI);
         appendAndIndexWikiCallback.builder.append(")");
         parser.wordForms.add(tr);