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.EnWiktionaryXmlParser;
+import com.hughes.android.dictionary.parser.enwiktionary.EnWiktionaryXmlParser;
import com.hughes.util.Args;
import com.hughes.util.FileUtil;
-/*
-
---maxEntries=100
---dictOut=de-en.dict
---lang1=DE
---lang2=EN
---dictInfo=@dictInfo.txt
-
---input0=/Users/thadh/personal/quickDic/de-en-chemnitz.txt
---input0Name=chemnitz
---input0Charset=UTF8
---input0Format=chemnitz
-
---input1=/Users/thadh/personal/quickDic/dewiktionary-20100326-pages-articles.xml
---input1Name=wiktionary
---input1Format=wiktionary
-
---input2=/Users/thadh/personal/quickDic/de-en-dictcc.txt
---input2Name=dictcc
---input2Charset=Cp1252
---input2Format=dictcc
- */
public class DictionaryBuilder {
public final Dictionary dictionary;
public final List<IndexBuilder> indexBuilders = new ArrayList<IndexBuilder>();
- public DictionaryBuilder(final String dictInfo, final Language lang0, final Language lang1, final String normalizerRules1, final String normalizerRules2) {
- dictionary = new Dictionary(dictInfo);
- indexBuilders.add(new IndexBuilder(this, lang0.getSymbol(), lang0.getSymbol() + "->" + lang1.getSymbol(), lang0, normalizerRules1, false));
- indexBuilders.add(new IndexBuilder(this, lang1.getSymbol(), lang1.getSymbol() + "->" + lang0.getSymbol(), lang1, normalizerRules2, true));
+ 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);
+ 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));
}
void build() {
}
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);
+ }
+
final Map<String,String> keyValueArgs = Args.keyValueArgs(args);
+ if (!keyValueArgs.containsKey("lang1") || !keyValueArgs.containsKey("lang2")) {
+ fatalError("--lang1= and --lang2= must both be specified.");
+ }
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 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)));
}
-
+ if (lang2StoplistFile != null) {
+ lang2Stoplist.addAll(FileUtil.readLines(new File(lang2StoplistFile)));
+ }
+
String normalizerRules1 = keyValueArgs.remove("normalizerRules1");
String normalizerRules2 = keyValueArgs.remove("normalizerRules2");
if (normalizerRules1 == null) {
System.out.println("dictInfo=" + dictInfo);
System.out.println("dictOut=" + dictOutFilename);
- final DictionaryBuilder dictionaryBuilder = new DictionaryBuilder(dictInfo, lang1, lang2, normalizerRules1, normalizerRules2);
+ final DictionaryBuilder dictionaryBuilder = new DictionaryBuilder(dictInfo, lang1, lang2, normalizerRules1, normalizerRules2, lang1Stoplist, lang2Stoplist);
for (int i = 0; i < 100; ++i) {
final String prefix = "input" + i;
fatalError("Must specify human readable name for: " + prefix + "Name");
}
- final EntrySource entrySource = new EntrySource(dictionaryBuilder.dictionary.sources.size(), inputName, dictionaryBuilder.dictionary.pairEntries.size());
+ final EntrySource entrySource = new EntrySource(dictionaryBuilder.dictionary.sources.size(), inputName, 0);
System.out.println("");
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);
+ 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).parseFile(file, entrySource);
} else if ("chemnitz".equals(inputFormat)) {
- new DictFileParser(charset, false, DictFileParser.DOUBLE_COLON, DictFileParser.PIPE, dictionaryBuilder, dictionaryBuilder.indexBuilders.toArray(new IndexBuilder[0]), null).parseFile(file);
+ 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).parseFile(file, entrySource);
} else if ("enwiktionary".equals(inputFormat)) {
- final Pattern langPattern = Pattern.compile(keyValueArgs.remove(prefix + "LangPattern"));
+ 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;
String pageLimit = keyValueArgs.remove(prefix + "PageLimit");
fatalError("Must be 1 or 2: " + prefix + "EnIndex");
}
new EnWiktionaryXmlParser(dictionaryBuilder.indexBuilders.get(enIndex), dictionaryBuilder.indexBuilders.get(1-enIndex),
- langPattern, langCodePattern, enIndex != 0).parse(file, Integer.parseInt(pageLimit));
+ langPattern, langCodePattern, enIndex != 0).parse(file, entrySource, Integer.parseInt(pageLimit));
} else {
fatalError("Invalid or missing input format: " + inputFormat);
}
private static void fatalError(String string) {
System.err.println(string);
+
+
System.exit(1);
}