]> gitweb.fperrin.net Git - DictionaryPC.git/blobdiff - src/com/hughes/android/dictionary/engine/DictionaryBuilderMain.java
Move test data, fix DictFileParser, fix splitter, fix crash during
[DictionaryPC.git] / src / com / hughes / android / dictionary / engine / DictionaryBuilderMain.java
index 833e5e97c4fe0a02e1431a1a24f91824e45824c9..36564eacce3ef918459e6c906849d9833c9af7a6 100644 (file)
@@ -19,7 +19,9 @@ import java.io.PrintWriter;
 import java.io.RandomAccessFile;
 import java.util.ArrayList;
 import java.util.Collections;
+import java.util.LinkedHashMap;
 import java.util.List;
+import java.util.Map;
 
 import junit.framework.TestCase;
 
@@ -28,116 +30,83 @@ public class DictionaryBuilderMain extends TestCase {
   static final String INPUTS = "../DictionaryData/inputs/";
   static final String STOPLISTS = "../DictionaryData/inputs/stoplists/";
   static final String OUTPUTS = "../DictionaryData/outputs/";
-  
-  static class Lang {
-    final String nameRegex;
-    final String isoCode;
-    final String wikiSplit;
-    final String stoplistFile;
-    public Lang(String nameRegex, String code, final String wikiSplit, final String stoplistFile) {
-      this.nameRegex = nameRegex;
-      this.isoCode = code;
-      this.wikiSplit = wikiSplit;
-      this.stoplistFile = stoplistFile;
-    }
-  }
-  
-  
+    
   public static void main(final String[] args) throws Exception {
-
-    Lang[] langs1 = new Lang[] { 
-        new Lang("^English$", "EN", null, "en.txt"),
-    };
-    Lang[] langs2 = new Lang[] { 
-//        new Lang("^.*Italian.*$", "IT", "italian.data", "it.txt"),
-//        new Lang("^.*French.*$", "FR", "french.data", "empty.txt"),
-//        new Lang("^.*Spanish.*$", "ES", "spanish.data", "es.txt"),
-//        new Lang("^.*Greek.*$", "EL", "greek.data", "el.txt"),
-//        new Lang("^.*Japanese.*$", "JA", "japanese.data", "empty.txt"),
-//        new Lang("^.*Chinese.*$|^.*Mandarin.*$", "ZH", "mandarin.data", "empty.txt"),
-        new Lang("^.*Afrikaans.*$", "AF", "afrikaans.data", "empty.txt"),
-        new Lang("^.*Arabic.*$", "AR", "".data, "empty.txt"),
-        new Lang("^.*Hebrew.*$", "HE"),
-        new Lang("^.*Hindi.*$", "HI"),
-        new Lang("^.*Icelandic.*$", "IS"),
-        new Lang("^.*Irish.*$", "GA"),
-        new Lang("^.*Korean.*$", "KO"),
-        new Lang("^.*Maori.*$", "MI"),
-        new Lang("^.*Norwegian.*$", "NO"),
-        new Lang("^.*Persian.*$", "FA"),
-        new Lang("^.*Portuguese.*$", "PT"),
-        new Lang("^.*Romanian.*$", "RO"),
-        new Lang("^.*Russian.*$", "RU"),
-        new Lang("^.*Sanskrit.*$", "SA"),
-        new Lang("^.*Serbian.*$", "SR"),
-        new Lang("^.*Swedish.*$", "SV"),
-        new Lang("^.*Tajik.*$", "TG"),
-        new Lang("^.*Thai.*$", "TH"),
-        new Lang("^.*Tibetan.*$", "BO"),
-        new Lang("^.*Turkish.*$", "TR"),
-        new Lang("^.*Ukranian.*$", "UK"),
-        new Lang("^.*Vietnamese.*$", "VI"),
-        new Lang("^.*Welsh.*$", "CY"),
-        new Lang("^.*Zulu.*$", "ZU"),
-        new Lang("^.*Croation.*$", "HR"),
-        new Lang("^.*Czech.*$", "CS"),
-        new Lang("^.*Dutch.*$", "NL"),
-        new Lang("^.*Finnish.*$", "FI"),
-        /*
-        new Lang("^German$", "DE"),
-        new Lang("^Armenian$", "HY"),
-        new Lang("^English$", "EN"),
-        new Lang("^Kurdish$", "KU"),
-        new Lang("^Lithuanian$", "LT"),
-        new Lang("^Malay$", "MS"),
-        new Lang("^Mongolian$", "MN"),
-        new Lang("^Somali$", "SO"),
-        new Lang("^Sudanese$", "SU"),
-        new Lang("^Yiddish$", "YI"),
-        */
-    };
     
-    for (final Lang lang1 : langs1) {
-      for (final Lang lang2 : langs2) {
-        if (lang1.nameRegex.equals(lang2.nameRegex)) {
-          continue;
-        }
+    final Map<String,String> isoToWikiName = new LinkedHashMap<String, String>(Language.isoCodeToWikiName);
+    isoToWikiName.remove("EN");
+    isoToWikiName.remove("DE");
+
+    final Map<String,String>  isoToDedication = new LinkedHashMap<String, String>();
+    isoToDedication.put("AF", "Afrikaans dictionary dedicated to Heiko and Mariëtte Horn.");
+    isoToDedication.put("HR", "Croation dictionary dedicated to Ines Viskic and Miro Kresonja.");
+    isoToDedication.put("NL", "Dutch dictionary dedicated to Mike LeBeau.");
+    // German handled in file.
+    isoToDedication.put("EL", "Greek dictionary dedicated to Noah Egge.");
+    isoToDedication.put("IT", "Italian dictionary dedicated to Carolina Tropini, my favorite stardust in the whole universe!  Ti amo!");
+    isoToDedication.put("JA", "Japanese dictionary dedicated to Akane Watanabe.");
+    isoToDedication.put("KO", "Korean dictionary dedicated to Ande Elwood--fall fashion und Fernsehturms!");
+    isoToDedication.put("PT", "Portuguese dictionary dedicated to Carlos Melo, one Tough Mudder.");
+    isoToDedication.put("RO", "Romanian dictionary dedicated to Radu Teodorescu.");
+    isoToDedication.put("RU", "Russian dictionary dedicated to Maxim Aronin--best friend always!.");
+    isoToDedication.put("SR", "Serbian dictionary dedicated to Filip Crnogorac--thanks for the honey.");
+    isoToDedication.put("ES", "Spanish dictionary made especially for Carolina Tropini! <3 XoXoXXXXX!");
+    isoToDedication.put("SV", "Swedish dictionary dedicated to Kajsa Palmblad--björn kramar!");
+
+    final Map<String,String>  isoToStoplist = new LinkedHashMap<String, String>();
+    isoToStoplist.put("DE", "de.txt");
+    isoToStoplist.put("EN", "en.txt");
+    isoToStoplist.put("ES", "es.txt");
+    isoToStoplist.put("IT", "it.txt");
+    isoToStoplist.put("FR", "fr.txt");
+
+    final Map<String,String>  isoToRegex = new LinkedHashMap<String, String>();
+    isoToRegex.put("ZH", ".*Chinese.*|.*Mandarin.*|.*Cantonese.*");
+    
+    boolean go = false; 
+    isoToWikiName.clear();
+    for (final String foreignIso : isoToWikiName.keySet()) {
+      if (foreignIso.equals("GA")) {
+        go = true;
+      }
+      if (!go) {
+        continue;
+      }
+
+        final String dictFile = String.format(OUTPUTS + "/EN-%s_enwiktionary.quickdic", foreignIso);
+        System.out.println("building dictFile: " + dictFile);
         
-        int enIndex = -1;
-        Lang nonEnglish = null;
-        if (lang2.isoCode.equals("EN")) {
-          enIndex = 2;
-          nonEnglish = lang1;
+        if (!isoToStoplist.containsKey(foreignIso)) {
+          isoToStoplist.put(foreignIso, "empty.txt");
         }
-        if (lang1.isoCode.equals("EN")) {
-          enIndex = 1;
-          nonEnglish = lang2;
+        if (!isoToDedication.containsKey(foreignIso)) {
+          isoToDedication.put(foreignIso, "");
         }
-        assert nonEnglish != null;
-
-        final String dictFile = String.format(OUTPUTS + "/%s-%s_enwiktionary.quickdic", lang1.isoCode, lang2.isoCode);
-        System.out.println("building dictFile: " + dictFile);
+        if (!isoToRegex.containsKey(foreignIso)) {
+          isoToRegex.put(foreignIso, ".*" + isoToWikiName.get(foreignIso) + ".*");
+        }
+  
         DictionaryBuilder.main(new String[] {
             String.format("--dictOut=%s", dictFile),
-            String.format("--lang1=%s", lang1.isoCode),
-            String.format("--lang2=%s", lang2.isoCode),
-            String.format("--lang1Stoplist=%s", STOPLISTS + lang1.stoplistFile),
-            String.format("--lang2Stoplist=%s", STOPLISTS + lang2.stoplistFile),
-            String.format("--dictInfo=(EN)Wikitionary-based %s-%s dictionary", lang1.isoCode, lang2.isoCode),
-
-            "--input2=" + INPUTS + "enWikiSplit/" + nonEnglish.wikiSplit,
-            "--input2Name=enwiktionary." + nonEnglish.wikiSplit,
+            String.format("--lang1=EN"),
+            String.format("--lang2=%s", foreignIso),
+            String.format("--lang1Stoplist=%s", STOPLISTS + isoToStoplist.get("EN")),
+            String.format("--lang2Stoplist=%s", STOPLISTS + isoToStoplist.get(foreignIso)),
+            String.format("--dictInfo=(EN)Wikitionary-based EN-%s dictionary.  %s", foreignIso, isoToDedication.get(foreignIso)),
+
+            "--input2=" + INPUTS + "enWikiSplit/" + foreignIso + ".data",
+            "--input2Name=enwiktionary." + foreignIso,
             "--input2Format=enwiktionary",
-            "--input2LangPattern=" + nonEnglish.nameRegex,
-            "--input2LangCodePattern=" + nonEnglish.isoCode.toLowerCase(),
-            "--input2EnIndex=" + enIndex,
+            "--input2LangPattern=" + isoToRegex.get(foreignIso),
+            "--input2LangCodePattern=" + foreignIso.toLowerCase(),
+            "--input2EnIndex=2",
 
-            "--input3=" + INPUTS + "enWikiSplit/english.data",
+            "--input3=" + INPUTS + "enWikiSplit/EN.data",
             "--input3Name=enwiktionary.english",
             "--input3Format=enwiktionary",
-            "--input3LangPattern=" + nonEnglish.nameRegex,
-            "--input3LangCodePattern=" + (enIndex == 1 ? lang2.isoCode : lang1.isoCode).toLowerCase(),
-            "--input3EnIndex=" + enIndex,
+            "--input3LangPattern=" + isoToRegex.get(foreignIso),
+            "--input3LangCodePattern=" + foreignIso.toLowerCase(),
+            "--input3EnIndex=2",
 
         });
         
@@ -153,36 +122,33 @@ public class DictionaryBuilderMain extends TestCase {
         textOut.close();
         raf.close();
 
-      }  // langs2
-    }  // langs1
+    }  // foreignIso
 
     DictionaryBuilder.main(new String[] {
-        "--dictOut=" + OUTPUTS + "DE-EN_all_free.quickdic",
+        "--dictOut=" + OUTPUTS + "DE-EN_chemnitz_enwiktionary",
         "--lang1=DE",
         "--lang2=EN",
-        "--dictInfo=@" + INPUTS + "de-en_all_free.info",
+        "--dictInfo=@" + INPUTS + "de-en_chemnitz_enwiktionary.info",
 
         "--input1=" + INPUTS + "de-en_chemnitz.txt",
         "--input1Name=chemnitz",
         "--input1Charset=UTF8",
         "--input1Format=chemnitz",
-    });
-
-    DictionaryBuilder.main(new String[] {
-        "--dictOut=" + OUTPUTS + "de-en_all.quickdic",
-        "--lang1=DE",
-        "--lang2=EN",
-        "--dictInfo=@" + INPUTS + "de-en_all.info",
-
-        "--input2=" + INPUTS + "de-en_chemnitz.txt",
-        "--input2Name=dictcc",
-        "--input2Charset=UTF8",
-        "--input2Format=chemnitz",
-
-        "--input3=" + INPUTS + "/NONFREE/de-en_dictcc.txt",
-        "--input3Name=dictcc",
-        "--input3Charset=UTF8",
-        "--input3Format=dictcc",
+        
+        "--input2=" + INPUTS + "enWikiSplit/DE.data",
+        "--input2Name=enwiktionary.DE",
+        "--input2Format=enwiktionary",
+        "--input2LangPattern=" + isoToRegex.get("DE"),
+        "--input2LangCodePattern=de",
+        "--input2EnIndex=2",
+
+        "--input3=" + INPUTS + "enWikiSplit/EN.data",
+        "--input3Name=enwiktionary.english",
+        "--input3Format=enwiktionary",
+        "--input3LangPattern=" + isoToRegex.get("DE"),
+        "--input3LangCodePattern=de",
+        "--input3EnIndex=2",
+        
     });
 
   }