+// Copyright 2011 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.engine;
import java.io.File;
import java.io.RandomAccessFile;
import java.nio.charset.Charset;
import java.util.ArrayList;
+import java.util.LinkedHashSet;
import java.util.List;
import java.util.Map;
+import java.util.Set;
+import java.util.regex.Pattern;
+
+import javax.xml.parsers.ParserConfigurationException;
+import org.xml.sax.SAXException;
+
+import com.hughes.android.dictionary.parser.DictFileParser;
+import com.hughes.android.dictionary.parser.Parser;
+import com.hughes.android.dictionary.parser.wiktionary.EnForeignParser;
+import com.hughes.android.dictionary.parser.wiktionary.EnToTranslationParser;
+import com.hughes.android.dictionary.parser.wiktionary.EnTranslationToTranslationParser;
+import com.hughes.android.dictionary.parser.wiktionary.WholeSectionToHtmlParser;
import com.hughes.util.Args;
import com.hughes.util.FileUtil;
-/*
+public class DictionaryBuilder {
---maxEntries=100
---dictOut=de-en.dict
---lang1=DE
---lang2=EN
---dictInfo=@dictInfo.txt
+ public final Dictionary dictionary;
+ public final List<IndexBuilder> indexBuilders = new ArrayList<IndexBuilder>();
---input0=/Users/thadh/personal/quickDic/de-en-chemnitz.txt
---input0Name=chemnitz
---input0Charset=UTF8
---input0Format=chemnitz
+ public DictionaryBuilder(final String dictInfoString, final Language lang0, final Language lang1, final String normalizerRules1, final String normalizerRules2, final Set<String> lang1Stoplist, final Set<String> lang2Stoplist) {
+ dictionary = new Dictionary(dictInfoString);
+ if (lang1 != null) {
+ indexBuilders.add(new IndexBuilder(this, lang0.getIsoCode(), lang0.getIsoCode() + "->" + lang1.getIsoCode(), lang0, normalizerRules1, lang1Stoplist, false));
+ indexBuilders.add(new IndexBuilder(this, lang1.getIsoCode(), lang1.getIsoCode() + "->" + lang0.getIsoCode(), lang1, normalizerRules2, lang2Stoplist, true));
+ } else {
+ indexBuilders.add(new IndexBuilder(this, lang0.getIsoCode(), lang0.getIsoCode(), lang0, normalizerRules1, lang1Stoplist, false));
+ }
+ }
---input1=/Users/thadh/personal/quickDic/dewiktionary-20100326-pages-articles.xml
---input1Name=wiktionary
---input1Format=wiktionary
+ void build() {
+ for (final IndexBuilder indexBuilder : indexBuilders) {
+ indexBuilder.build();
+ dictionary.indices.add(indexBuilder.index);
+ }
+ }
---input2=/Users/thadh/personal/quickDic/de-en-dictcc.txt
---input2Name=dictcc
---input2Charset=Cp1252
---input2Format=dictcc
- */
+ public static void main(final String[] args) throws IOException, ParserConfigurationException, SAXException {
+ System.out.println("Running with arguments:");
+ for (final String arg : args) {
+ System.out.println(arg);
+ }
-public class DictionaryBuilder {
-
- final Dictionary dictionary;
-
- final List<EntryData> entryDatas = new ArrayList<EntryData>();
-
- final List<IndexBuilder> indexBuilders = new ArrayList<IndexBuilder>();
-
- public DictionaryBuilder(final String dictInfo, final Language lang0, final Language lang1) {
- dictionary = new Dictionary(dictInfo);
- indexBuilders.add(new IndexBuilder(this, lang0.getSymbol(), lang0.getSymbol() + "->" + lang1.getSymbol(), lang0, false));
- indexBuilders.add(new IndexBuilder(this, lang1.getSymbol(), lang1.getSymbol() + "->" + lang0.getSymbol(), lang1, true));
- }
-
- void build() {
- for (final IndexBuilder indexBuilder : indexBuilders) {
- indexBuilder.build();
- dictionary.indices.add(indexBuilder.index);
- }
- }
-
- public static void main(final String[] args) throws IOException {
- final Map<String,String> keyValueArgs = Args.keyValueArgs(args);
-
- final Language lang1 = Language.lookup(keyValueArgs.remove("lang1"));
- final Language lang2 = Language.lookup(keyValueArgs.remove("lang2"));
- if (lang1 == null || lang2 == null) {
- fatalError("--lang1= and --lang2= must both be specified.");
- }
-
- final String dictOutFilename = keyValueArgs.remove("dictOut");
- if (dictOutFilename == null) {
- fatalError("--dictOut= must be specified.");
- }
-
- String dictInfo = keyValueArgs.remove("dictInfo");
- if (dictInfo == null) {
- fatalError("--dictInfo= must be specified.");
- }
- if (dictInfo.startsWith("@")) {
- dictInfo = FileUtil.readToString(new File(dictInfo.substring(1)));
- }
-
- final String printFile = keyValueArgs.remove("print");
-
- System.out.println("lang1=" + lang1);
- System.out.println("lang2=" + lang2);
- System.out.println("dictInfo=" + dictInfo);
- System.out.println("dictOut=" + dictOutFilename);
-
- final DictionaryBuilder dictionaryBuilder = new DictionaryBuilder(dictInfo, lang1, lang2);
-
- for (int i = 0; i < 100; ++i) {
- final String prefix = "input" + i;
- if (keyValueArgs.containsKey(prefix)) {
- final File file = new File(keyValueArgs.remove(prefix));
- System.out.println("Processing: " + file);
- String charsetName = keyValueArgs.remove(prefix + "Charset");
- if (charsetName == null) {
- charsetName = "UTF8";
+ final Map<String,String> keyValueArgs = Args.keyValueArgs(args);
+
+ if (!keyValueArgs.containsKey("lang1")) {
+ fatalError("--lang1= must be specified.");
}
- final Charset charset = Charset.forName(charsetName);
- String inputName = keyValueArgs.remove(prefix + "Name");
- if (inputName == null) {
- fatalError("Must specify human readable name for: " + prefix + "Name");
+ final Language lang1 = Language.lookup(keyValueArgs.remove("lang1"));
+ final Language lang2;
+ if (keyValueArgs.containsKey("lang2")) {
+ lang2 = Language.lookup(keyValueArgs.remove("lang2"));
+ } else {
+ lang2 = null;
}
- String inputFormat = keyValueArgs.remove(prefix + "Format");
- if ("dictcc".equals(inputFormat)) {
- new DictFileParser(charset, false, DictFileParser.TAB, null, dictionaryBuilder, dictionaryBuilder.indexBuilders.toArray(new IndexBuilder[0]), null).parseFile(file);
- } else if ("chemnitz".equals(inputFormat)) {
- new DictFileParser(charset, false, DictFileParser.DOUBLE_COLON, DictFileParser.PIPE, dictionaryBuilder, dictionaryBuilder.indexBuilders.toArray(new IndexBuilder[0]), null).parseFile(file);
- } else if ("wiktionary".equals(inputFormat)) {
- throw new RuntimeException();
-// new WiktionaryXmlParser(dict).parse(file);
- } else {
- fatalError("Invalid or missing input format: " + inputFormat);
+ final Set<String> lang1Stoplist = new LinkedHashSet<String>();
+ final Set<String> lang2Stoplist = new LinkedHashSet<String>();
+ final String lang1StoplistFile = keyValueArgs.remove("lang1Stoplist");
+ final String lang2StoplistFile = keyValueArgs.remove("lang2Stoplist");
+ if (lang1StoplistFile != null) {
+ lang1Stoplist.addAll(FileUtil.readLines(new File(lang1StoplistFile)));
}
-
- final EntrySource entrySource = new EntrySource(dictionaryBuilder.dictionary.sources.size(), inputName);
- dictionaryBuilder.dictionary.sources.add(entrySource);
- System.out.println("Done: " + file + "\n\n");
- }
- }
-
- dictionaryBuilder.build();
-
- if (printFile != null) {
- final PrintStream out = new PrintStream(new File(printFile));
- dictionaryBuilder.dictionary.print(out);
- out.close();
+ if (lang2StoplistFile != null) {
+ lang2Stoplist.addAll(FileUtil.readLines(new File(lang2StoplistFile)));
+ }
+
+ String normalizerRules1 = keyValueArgs.remove("normalizerRules1");
+ String normalizerRules2 = keyValueArgs.remove("normalizerRules2");
+ if (normalizerRules1 == null) {
+ normalizerRules1 = lang1.getDefaultNormalizerRules();
+ }
+ if (normalizerRules2 == null) {
+ normalizerRules2 = lang2 == null ? null : lang2.getDefaultNormalizerRules();
+ }
+
+ final String dictOutFilename = keyValueArgs.remove("dictOut");
+ if (dictOutFilename == null) {
+ fatalError("--dictOut= must be specified.");
+ }
+
+ String dictInfo = keyValueArgs.remove("dictInfo");
+ if (dictInfo == null) {
+ fatalError("--dictInfo= must be specified.");
+ }
+ if (dictInfo.startsWith("@")) {
+ dictInfo = FileUtil.readToString(new File(dictInfo.substring(1)));
+ }
+
+ final String printFile = keyValueArgs.remove("print");
+
+ System.out.println("lang1=" + lang1);
+ System.out.println("lang2=" + lang2);
+ System.out.println("normalizerRules1=" + normalizerRules1);
+ System.out.println("normalizerRules2=" + normalizerRules2);
+ System.out.println("dictInfo=" + dictInfo);
+ System.out.println("dictOut=" + dictOutFilename);
+
+ final DictionaryBuilder dictionaryBuilder = new DictionaryBuilder(dictInfo, lang1, lang2, normalizerRules1, normalizerRules2, lang1Stoplist, lang2Stoplist);
+
+ for (int i = 0; i < 100; ++i) {
+ final String prefix = "input" + i;
+ if (keyValueArgs.containsKey(prefix)) {
+ final File file = new File(keyValueArgs.remove(prefix));
+ System.out.println("Processing: " + file);
+ String charsetName = keyValueArgs.remove(prefix + "Charset");
+ if (charsetName == null) {
+ charsetName = "UTF8";
+ }
+ final Charset charset = Charset.forName(charsetName);
+ String inputName = keyValueArgs.remove(prefix + "Name");
+ if (inputName == null) {
+ fatalError("Must specify human readable name for: " + prefix + "Name");
+ }
+ String pageLimitString = keyValueArgs.remove(prefix + "PageLimit");
+ if (pageLimitString == null) {
+ pageLimitString = "-1";
+ }
+ final int pageLimit = Integer.parseInt(pageLimitString);
+
+ final EntrySource entrySource = new EntrySource(dictionaryBuilder.dictionary.sources.size(), inputName, 0);
+ System.out.println("");
+
+ String inputFormat = keyValueArgs.remove(prefix + "Format");
+ if ("tab_separated".equals(inputFormat)) {
+ final boolean flipColumns = "true".equals(keyValueArgs.remove(prefix + "FlipColumns"));
+ new DictFileParser(charset, flipColumns, DictFileParser.TAB, null, dictionaryBuilder, dictionaryBuilder.indexBuilders.toArray(new IndexBuilder[0]), null).parse(file, entrySource, pageLimit);
+ } else if ("chemnitz".equals(inputFormat)) {
+ final boolean flipColumns = "true".equals(keyValueArgs.remove(prefix + "FlipColumns"));
+ new DictFileParser(charset, flipColumns, DictFileParser.DOUBLE_COLON, DictFileParser.PIPE, dictionaryBuilder, dictionaryBuilder.indexBuilders.toArray(new IndexBuilder[0]), null).parse(file, entrySource, pageLimit);
+ } else if ("enwiktionary".equals(inputFormat)) {
+ final String type = keyValueArgs.remove(prefix + "WiktionaryType");
+ final Pattern langPattern = Pattern.compile(keyValueArgs.remove(prefix + "LangPattern"), Pattern.CASE_INSENSITIVE);
+ final Pattern langCodePattern = Pattern.compile(keyValueArgs.remove(prefix + "LangCodePattern"));
+ final int enIndex = Integer.parseInt(keyValueArgs.remove(prefix + "EnIndex")) - 1;
+
+ if (enIndex < 0 || enIndex >= 2) {
+ fatalError("Must be 1 or 2: " + prefix + "EnIndex");
+ }
+ final Parser parser;
+ if ("EnToTranslation".equals(type)) {
+ parser = new EnToTranslationParser(dictionaryBuilder.indexBuilders.get(enIndex), dictionaryBuilder.indexBuilders.get(1-enIndex),
+ langPattern, langCodePattern, enIndex != 0);
+ } else if ("EnForeign".equals(type)) {
+ parser = new EnForeignParser(dictionaryBuilder.indexBuilders.get(enIndex), dictionaryBuilder.indexBuilders.get(1-enIndex),
+ langPattern, langCodePattern, enIndex != 0);
+ } else if ("EnEnglish".equals(type)) {
+ parser = new EnForeignParser(dictionaryBuilder.indexBuilders.get(enIndex), dictionaryBuilder.indexBuilders.get(enIndex),
+ langPattern, langCodePattern, true);
+ } else {
+ fatalError("Invalid WiktionaryType (use EnToTranslation or EnForeign or EnEnglish): " + type);
+ return;
+ }
+ parser.parse(file, entrySource, pageLimit);
+ } else if (EnTranslationToTranslationParser.NAME.equals(inputFormat)) {
+ final String code1 = keyValueArgs.remove(prefix + "LangPattern1");
+ final String code2 = keyValueArgs.remove(prefix + "LangPattern2");
+ if (code1 == null || code2 == null) {
+ fatalError("Must specify LangPattern1 and LangPattern2.");
+ return;
+ }
+ final Pattern codePattern1 = Pattern.compile(code1, Pattern.CASE_INSENSITIVE);
+ final Pattern codePattern2 = Pattern.compile(code2, Pattern.CASE_INSENSITIVE);
+ new EnTranslationToTranslationParser(dictionaryBuilder.indexBuilders, new Pattern[] {codePattern1, codePattern2}).parse(file, entrySource, pageLimit);
+ } else if (WholeSectionToHtmlParser.NAME.equals(inputFormat)) {
+ final int titleIndex = Integer.parseInt(keyValueArgs.remove(prefix + "TitleIndex")) - 1;
+ final String wiktionaryLang = keyValueArgs.remove(prefix + "WiktionaryLang");
+ final String webUrlTemplate = keyValueArgs.remove(prefix + "WebUrlTemplate");
+ String skipLang = keyValueArgs.remove(prefix + "SkipLang");
+ if (skipLang == null) skipLang = "";
+ new WholeSectionToHtmlParser(dictionaryBuilder.indexBuilders.get(titleIndex), null, wiktionaryLang, skipLang, webUrlTemplate).parse(file, entrySource, pageLimit);
+ } else {
+ fatalError("Invalid or missing input format: " + inputFormat);
+ }
+
+ dictionaryBuilder.dictionary.sources.add(entrySource);
+ System.out.println("Done: " + file + "\n\n");
+ }
+ }
+
+ dictionaryBuilder.build();
+ // Drop indexBuilders to free RAM
+ dictionaryBuilder.indexBuilders.clear();
+
+ if (printFile != null) {
+ final PrintStream out = new PrintStream(new File(printFile));
+ dictionaryBuilder.dictionary.print(out);
+ out.close();
+ }
+
+ System.out.println("Writing dictionary to: " + dictOutFilename);
+ final RandomAccessFile dictOut = new RandomAccessFile(dictOutFilename, "rw");
+ dictOut.setLength(0);
+ dictionaryBuilder.dictionary.write(dictOut);
+ dictOut.close();
+
+ if (!keyValueArgs.isEmpty()) {
+ System.err.println("WARNING: couldn't parse arguments: " + keyValueArgs);
+ System.exit(1);
+ }
+
}
-
- System.out.println("Writing dictionary to: " + dictOutFilename);
- final RandomAccessFile dictOut = new RandomAccessFile(dictOutFilename, "rw");
- dictOut.setLength(0);
- dictionaryBuilder.dictionary.write(dictOut);
- dictOut.close();
-
- if (!keyValueArgs.isEmpty()) {
- System.err.println("WARNING: couldn't parse arguments: " + keyValueArgs);
- System.exit(1);
+
+ private static void fatalError(String string) {
+ System.err.println(string);
+
+
+ System.exit(1);
}
-
- }
-
- private static void fatalError(String string) {
- System.err.println(string);
- System.exit(1);
- }
-
+
}