--- /dev/null
+package com.hughes.android.dictionary.parser;
+
+import java.io.File;
+import java.io.IOException;
+
+import com.hughes.android.dictionary.engine.EntrySource;
+
+public interface Parser {
+
+ void parse(final File file, final EntrySource entrySource, final int pageLimit) throws IOException;
+
+}
--- /dev/null
+// 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.wiktionary;
+
+import java.io.BufferedInputStream;
+import java.io.DataInputStream;
+import java.io.EOFException;
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.IOException;
+import java.util.LinkedHashMap;
+import java.util.LinkedHashSet;
+import java.util.List;
+import java.util.Map;
+import java.util.Set;
+import java.util.SortedMap;
+import java.util.TreeMap;
+import java.util.concurrent.atomic.AtomicInteger;
+import java.util.logging.Level;
+import java.util.logging.Logger;
+import java.util.regex.Pattern;
+
+import com.hughes.android.dictionary.engine.EntrySource;
+import com.hughes.android.dictionary.engine.EntryTypeName;
+import com.hughes.android.dictionary.engine.IndexBuilder;
+import com.hughes.android.dictionary.engine.IndexedEntry;
+import com.hughes.android.dictionary.parser.Parser;
+import com.hughes.android.dictionary.parser.WikiTokenizer;
+import com.hughes.util.EnumUtil;
+
+public abstract class AbstractWiktionaryParser implements Parser {
+
+ static final Logger LOG = Logger.getLogger("WiktionaryParser");
+
+ final SortedMap<String, AtomicInteger> counters = new TreeMap<String, AtomicInteger>();
+ final Set<String> pairsAdded = new LinkedHashSet<String>();
+
+ EntrySource entrySource;
+ String title;
+
+
+ abstract void parseSection(final String heading, final String text);
+
+ @Override
+ public void parse(final File file, final EntrySource entrySource, final int pageLimit) throws IOException {
+ this.entrySource = entrySource;
+ int pageCount = 0;
+ final DataInputStream dis = new DataInputStream(new BufferedInputStream(new FileInputStream(file)));
+ try {
+ while (true) {
+ if (pageLimit >= 0 && pageCount >= pageLimit) {
+ return;
+ }
+
+ try {
+ title = dis.readUTF();
+ } catch (EOFException e) {
+ LOG.log(Level.INFO, "EOF reading split.");
+ dis.close();
+ return;
+ }
+ final String heading = dis.readUTF();
+ final int bytesLength = dis.readInt();
+ final byte[] bytes = new byte[bytesLength];
+ dis.readFully(bytes);
+ final String text = new String(bytes, "UTF8");
+
+ parseSection(heading, text);
+
+ ++pageCount;
+ if (pageCount % 1000 == 0) {
+ LOG.info("pageCount=" + pageCount);
+ }
+ }
+ } finally {
+ LOG.info("***COUNTERS***");
+ for (final Map.Entry<String, AtomicInteger> entry : counters.entrySet()) {
+ LOG.info(entry.getKey() + ": " + entry.getValue());
+ }
+ }
+ }
+
+ static final Pattern whitespace = Pattern.compile("\\s+");
+ static String trim(final String s) {
+ return whitespace.matcher(s).replaceAll(" ").trim();
+ }
+
+ public void incrementCount(final String string) {
+ AtomicInteger counter = counters.get(string);
+ if (counter != null) {
+ counter = new AtomicInteger();
+ counters.put(string, counter);
+ }
+ counter.incrementAndGet();
+ }
+
+
+ // -------------------------------------------------------------------------
+
+ static final class AppendAndIndexWikiCallback<T extends AbstractWiktionaryParser> implements WikiTokenizer.Callback {
+
+ final T parser;
+ StringBuilder builder;
+ IndexedEntry indexedEntry;
+ IndexBuilder indexBuilder;
+ final Map<String,FunctionCallback<T>> functionCallbacks = new LinkedHashMap<String, FunctionCallback<T>>();
+
+ boolean entryTypeNameSticks = false;
+ EntryTypeName entryTypeName = null;
+
+ final Map<String,AtomicInteger> langCodeToTCount = new LinkedHashMap<String, AtomicInteger>();
+
+ final NameAndArgs<T> nameAndArgs = new NameAndArgs<T>();
+
+ public AppendAndIndexWikiCallback(final T parser) {
+ this.parser = parser;
+ }
+
+ public void reset(final StringBuilder builder, final IndexedEntry indexedEntry) {
+ this.builder = builder;
+ this.indexedEntry = indexedEntry;
+ this.indexBuilder = null;
+ entryTypeName = null;
+ entryTypeNameSticks = false;
+ }
+
+ public void dispatch(final String wikiText, final IndexBuilder indexBuilder, final EntryTypeName entryTypeName) {
+ final IndexBuilder oldIndexBuilder = this.indexBuilder;
+ final EntryTypeName oldEntryTypeName = this.entryTypeName;
+ this.indexBuilder = indexBuilder;
+ if (!entryTypeNameSticks) {
+ this.entryTypeName = EnumUtil.min(entryTypeName, this.entryTypeName);
+ }
+ if (entryTypeName == null) this.entryTypeName = null;
+ WikiTokenizer.dispatch(wikiText, false, this);
+ this.indexBuilder = oldIndexBuilder;
+ this.entryTypeName = oldEntryTypeName;
+ }
+
+ public void dispatch(final String wikiText, final EntryTypeName entryTypeName) {
+ dispatch(wikiText, this.indexBuilder, entryTypeName);
+ }
+
+ @Override
+ public void onPlainText(final String plainText) {
+ // The only non-recursive callback. Just appends to the builder, and indexes.
+ builder.append(plainText);
+ if (indexBuilder != null && entryTypeName != null && indexedEntry != null) {
+ indexBuilder.addEntryWithString(indexedEntry, plainText, entryTypeName);
+ }
+ }
+
+ @Override
+ public void onWikiLink(WikiTokenizer wikiTokenizer) {
+ final String text = wikiTokenizer.wikiLinkText();
+ @SuppressWarnings("unused")
+ final String link = wikiTokenizer.wikiLinkDest();
+ dispatch(text, entryTypeName);
+ }
+
+ @Override
+ public void onFunction(
+ final WikiTokenizer wikiTokenizer,
+ final String name,
+ final List<String> args,
+ final Map<String, String> namedArgs) {
+
+ FunctionCallback<T> functionCallback = functionCallbacks.get(name);
+ if (functionCallback == null || !functionCallback.onWikiFunction(wikiTokenizer, name, args, namedArgs, parser, this)) {
+ // Default function handling:
+// namedArgs.keySet().removeAll(EnWiktionaryXmlParser.USELESS_WIKI_ARGS);
+ final boolean single = args.isEmpty() && namedArgs.isEmpty();
+ builder.append(single ? "{" : "{{");
+
+ final IndexBuilder oldIndexBuilder = indexBuilder;
+ indexBuilder = null;
+ nameAndArgs.onWikiFunction(wikiTokenizer, name, args, namedArgs, parser, this);
+ indexBuilder = oldIndexBuilder;
+
+ builder.append(single ? "}" : "}}");
+ }
+ }
+
+ @Override
+ public void onHtml(WikiTokenizer wikiTokenizer) {
+ // Unindexed for now.
+ builder.append(wikiTokenizer.token());
+ }
+
+ @Override
+ public void onMarkup(WikiTokenizer wikiTokenizer) {
+ // Do nothing.
+ }
+
+ @Override
+ public final void onComment(WikiTokenizer wikiTokenizer) {
+ // Do nothing.
+ }
+
+ @Override
+ public final void onNewline(WikiTokenizer wikiTokenizer) {
+ assert false;
+ }
+
+ @Override
+ public final void onHeading(WikiTokenizer wikiTokenizer) {
+ assert false;
+ }
+
+ @Override
+ public final void onListItem(WikiTokenizer wikiTokenizer) {
+ assert false;
+ }
+ }
+
+ // --------------------------------------------------------------------
+
+ static final class NameAndArgs<T extends AbstractWiktionaryParser> implements FunctionCallback<T> {
+ @Override
+ public boolean onWikiFunction(final WikiTokenizer wikiTokenizer, final String name, final List<String> args,
+ final Map<String, String> namedArgs, final T parser,
+ final AppendAndIndexWikiCallback<T> appendAndIndexWikiCallback) {
+
+ 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);
+ }
+ }
+ appendNamedArgs(namedArgs, appendAndIndexWikiCallback);
+ return true;
+ }
+ }
+ static NameAndArgs<AbstractWiktionaryParser> NAME_AND_ARGS = new NameAndArgs<AbstractWiktionaryParser>();
+
+ 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);
+ }
+ }
+
+
+
+}
+++ /dev/null
-// 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.wiktionary;
-
-import java.util.LinkedHashMap;
-import java.util.List;
-import java.util.Map;
-import java.util.concurrent.atomic.AtomicInteger;
-
-import com.hughes.android.dictionary.engine.EntryTypeName;
-import com.hughes.android.dictionary.engine.IndexBuilder;
-import com.hughes.android.dictionary.engine.IndexedEntry;
-import com.hughes.android.dictionary.parser.WikiTokenizer;
-import com.hughes.util.EnumUtil;
-
-final class AppendAndIndexWikiCallback implements WikiTokenizer.Callback {
-
- final EnWiktionaryXmlParser parser;
- StringBuilder builder;
- IndexedEntry indexedEntry;
- IndexBuilder indexBuilder;
- final Map<String,FunctionCallback> functionCallbacks = new LinkedHashMap<String, FunctionCallback>();
-
- boolean entryTypeNameSticks = false;
- EntryTypeName entryTypeName = null;
-
- final Map<String,AtomicInteger> langCodeToTCount = new LinkedHashMap<String, AtomicInteger>();
-
- public AppendAndIndexWikiCallback(final EnWiktionaryXmlParser parser) {
- this.parser = parser;
- }
-
- public void reset(final StringBuilder builder, final IndexedEntry indexedEntry) {
- this.builder = builder;
- this.indexedEntry = indexedEntry;
- this.indexBuilder = null;
- entryTypeName = null;
- entryTypeNameSticks = false;
- }
-
- public void dispatch(final String wikiText, final IndexBuilder indexBuilder, final EntryTypeName entryTypeName) {
- final IndexBuilder oldIndexBuilder = this.indexBuilder;
- final EntryTypeName oldEntryTypeName = this.entryTypeName;
- this.indexBuilder = indexBuilder;
- if (!entryTypeNameSticks) {
- this.entryTypeName = EnumUtil.min(entryTypeName, this.entryTypeName);
- }
- if (entryTypeName == null) this.entryTypeName = null;
- WikiTokenizer.dispatch(wikiText, false, this);
- this.indexBuilder = oldIndexBuilder;
- this.entryTypeName = oldEntryTypeName;
- }
-
- public void dispatch(final String wikiText, final EntryTypeName entryTypeName) {
- dispatch(wikiText, this.indexBuilder, entryTypeName);
- }
-
-
- @Override
- public void onPlainText(final String plainText) {
- // The only non-recursive callback. Just appends to the builder, and indexes.
- builder.append(plainText);
- if (indexBuilder != null && entryTypeName != null && indexedEntry != null) {
- indexBuilder.addEntryWithString(indexedEntry, plainText, entryTypeName);
- }
- }
-
- @Override
- public void onWikiLink(WikiTokenizer wikiTokenizer) {
- final String text = wikiTokenizer.wikiLinkText();
- final String link = wikiTokenizer.wikiLinkDest();
- if (link != null) {
- if (link.contains("#English")) {
- dispatch(text, parser.enIndexBuilder, EntryTypeName.WIKTIONARY_ENGLISH_DEF_WIKI_LINK);
- } else if (link.contains("#") && parser.langPattern.matcher(link).find()) {
- dispatch(text, parser.foreignIndexBuilder, EntryTypeName.WIKTIONARY_ENGLISH_DEF_OTHER_LANG);
- } else if (link.equals("plural")) {
- builder.append(text);
- } else {
- //LOG.warning("Special link: " + englishTokenizer.token());
- dispatch(text, EntryTypeName.WIKTIONARY_ENGLISH_DEF_WIKI_LINK);
- }
- } else {
- // link == null
- final EntryTypeName entryTypeName;
- switch (parser.state) {
- case TRANSLATION_LINE:
- entryTypeName = EntryTypeName.WIKTIONARY_TRANSLATION_WIKI_TEXT;
- break;
- case ENGLISH_DEF_OF_FOREIGN:
- entryTypeName = EntryTypeName.WIKTIONARY_ENGLISH_DEF_WIKI_LINK;
- break;
- default:
- throw new IllegalStateException("Invalid enum value: " + parser.state);
- }
- dispatch(text, entryTypeName);
- }
- }
-
- @Override
- public void onFunction(
- final WikiTokenizer wikiTokenizer,
- final String name,
- final List<String> args,
- final Map<String, String> namedArgs) {
-
- FunctionCallback functionCallback = functionCallbacks.get(name);
- if (functionCallback == null || !functionCallback.onWikiFunction(wikiTokenizer, name, args, namedArgs, parser, this)) {
- // Default function handling:
- namedArgs.keySet().removeAll(EnWiktionaryXmlParser.USELESS_WIKI_ARGS);
- final boolean single = args.isEmpty() && namedArgs.isEmpty();
- builder.append(single ? "{" : "{{");
-
- final IndexBuilder oldIndexBuilder = indexBuilder;
- indexBuilder = null;
- FunctionCallbacksDefault.NAME_AND_ARGS.onWikiFunction(wikiTokenizer, name, args, namedArgs, parser, this);
- indexBuilder = oldIndexBuilder;
-
- builder.append(single ? "}" : "}}");
- }
- }
-
- @Override
- public void onHtml(WikiTokenizer wikiTokenizer) {
- // Unindexed for now.
- builder.append(wikiTokenizer.token());
- }
-
- @Override
- public void onMarkup(WikiTokenizer wikiTokenizer) {
- // Do nothing.
- }
-
- @Override
- public void onComment(WikiTokenizer wikiTokenizer) {
- // Do nothing.
- }
-
- @Override
- public void onNewline(WikiTokenizer wikiTokenizer) {
- assert false;
- }
-
- @Override
- public void onHeading(WikiTokenizer wikiTokenizer) {
- assert false;
- }
-
- @Override
- public void onListItem(WikiTokenizer wikiTokenizer) {
- assert false;
- }
-
-}
\ No newline at end of file
--- /dev/null
+// 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.wiktionary;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.Collection;
+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.regex.Pattern;
+
+import com.hughes.android.dictionary.engine.EntryTypeName;
+import com.hughes.android.dictionary.engine.IndexBuilder;
+import com.hughes.android.dictionary.engine.IndexedEntry;
+import com.hughes.android.dictionary.engine.PairEntry;
+import com.hughes.android.dictionary.engine.PairEntry.Pair;
+import com.hughes.android.dictionary.parser.WikiTokenizer;
+import com.hughes.util.ListUtil;
+
+public abstract class EnParser extends AbstractWiktionaryParser {
+
+ // TODO: process {{ttbc}} lines
+
+ static final Pattern partOfSpeechHeader = Pattern.compile(
+ "Noun|Verb|Adjective|Adverb|Pronoun|Conjunction|Interjection|" +
+ "Preposition|Proper noun|Article|Prepositional phrase|Acronym|" +
+ "Abbreviation|Initialism|Contraction|Prefix|Suffix|Symbol|Letter|" +
+ "Ligature|Idiom|Phrase|\\{\\{acronym\\}\\}|\\{\\{initialism\\}\\}|" +
+ "\\{\\{abbreviation\\}\\}|" +
+ // These are @deprecated:
+ "Noun form|Verb form|Adjective form|Nominal phrase|Noun phrase|" +
+ "Verb phrase|Transitive verb|Intransitive verb|Reflexive verb|" +
+ // These are extras I found:
+ "Determiner|Numeral|Number|Cardinal number|Ordinal number|Proverb|" +
+ "Particle|Interjection|Pronominal adverb" +
+ "Han character|Hanzi|Hanja|Kanji|Katakana character|Syllable");
+
+ // Might only want to remove "lang" if it's equal to "zh", for example.
+ static final Set<String> USELESS_WIKI_ARGS = new LinkedHashSet<String>(
+ Arrays.asList(
+ "lang",
+ "sc",
+ "sort",
+ "cat",
+ "cat2",
+ "xs",
+ "nodot"));
+
+ static boolean isIgnorableTitle(final String title) {
+ return title.startsWith("Wiktionary:") ||
+ title.startsWith("Template:") ||
+ title.startsWith("Appendix:") ||
+ title.startsWith("Category:") ||
+ title.startsWith("Index:") ||
+ title.startsWith("MediaWiki:") ||
+ title.startsWith("TransWiki:") ||
+ title.startsWith("Citations:") ||
+ title.startsWith("Concordance:") ||
+ title.startsWith("Help:");
+ }
+
+ final IndexBuilder enIndexBuilder;
+ final IndexBuilder foreignIndexBuilder;
+ final Pattern langPattern;
+ final Pattern langCodePattern;
+ final boolean swap;
+
+ // State used while parsing.
+ enum State {
+ TRANSLATION_LINE,
+ ENGLISH_DEF_OF_FOREIGN,
+ ENGLISH_EXAMPLE,
+ FOREIGN_EXAMPLE,
+ }
+ State state = null;
+
+ public boolean entryIsFormOfSomething = false;
+ final Collection<String> wordForms = new ArrayList<String>();
+ boolean titleAppended = false;
+
+
+ final AppendAndIndexWikiCallback<EnParser> appendAndIndexWikiCallback = new AppendAndIndexWikiCallback<EnParser>(this);
+ {
+ appendAndIndexWikiCallback.functionCallbacks.putAll(FunctionCallbacks.DEFAULT);
+ }
+
+ EnParser(final IndexBuilder enIndexBuilder, final IndexBuilder otherIndexBuilder, final Pattern langPattern, final Pattern langCodePattern, final boolean swap) {
+ this.enIndexBuilder = enIndexBuilder;
+ this.foreignIndexBuilder = otherIndexBuilder;
+ this.langPattern = langPattern;
+ this.langCodePattern = langCodePattern;
+ this.swap = swap;
+ }
+
+ // --------------------------------------------------------------------
+
+ static final class EnToTranslationParser extends EnParser {
+
+ EnToTranslationParser(final IndexBuilder enIndexBuilder,
+ final IndexBuilder otherIndexBuilder, final Pattern langPattern,
+ final Pattern langCodePattern, final boolean swap) {
+ super(enIndexBuilder, otherIndexBuilder, langPattern, langCodePattern, swap);
+ }
+
+ @Override
+ void parseSection(String heading, String text) {
+ if (isIgnorableTitle(title)) {
+ return;
+ }
+ heading = heading.replaceAll("=", "").trim();
+ if (!heading.contains("English")) {
+ return;
+ }
+
+ String pos = null;
+ int posDepth = -1;
+
+ final WikiTokenizer wikiTokenizer = new WikiTokenizer(text);
+ while (wikiTokenizer.nextToken() != null) {
+
+ if (wikiTokenizer.isHeading()) {
+ final String headerName = wikiTokenizer.headingWikiText();
+
+ if (wikiTokenizer.headingDepth() <= posDepth) {
+ pos = null;
+ posDepth = -1;
+ }
+
+ if (partOfSpeechHeader.matcher(headerName).matches()) {
+ posDepth = wikiTokenizer.headingDepth();
+ pos = wikiTokenizer.headingWikiText();
+ // TODO: if we're inside the POS section, we should handle the first title line...
+
+ } else if (headerName.equals("Translations")) {
+ if (pos == null) {
+ LOG.info("Translations without POS (but using anyway): " + title);
+ }
+ doTranslations(wikiTokenizer, pos);
+ } else if (headerName.equals("Pronunciation")) {
+ //doPronunciation(wikiLineReader);
+ }
+ } else if (wikiTokenizer.isFunction()) {
+ final String name = wikiTokenizer.functionName();
+ if (name.equals("head") && pos == null) {
+ LOG.warning("{{head}} without POS: " + title);
+ }
+ }
+ }
+ }
+
+ private void doTranslations(final WikiTokenizer wikiTokenizer, final String pos) {
+ if (title.equals("absolutely")) {
+ //System.out.println();
+ }
+
+ String topLevelLang = null;
+ String sense = null;
+ boolean done = false;
+ while (wikiTokenizer.nextToken() != null) {
+ if (wikiTokenizer.isHeading()) {
+ wikiTokenizer.returnToLineStart();
+ return;
+ }
+ if (done) {
+ continue;
+ }
+
+ // Check whether we care about this line:
+
+ if (wikiTokenizer.isFunction()) {
+ final String functionName = wikiTokenizer.functionName();
+ final List<String> positionArgs = wikiTokenizer.functionPositionArgs();
+
+ if (functionName.equals("trans-top")) {
+ sense = null;
+ if (wikiTokenizer.functionPositionArgs().size() >= 1) {
+ sense = positionArgs.get(0);
+ // TODO: could emphasize words in [[brackets]] inside sense.
+ sense = WikiTokenizer.toPlainText(sense);
+ //LOG.info("Sense: " + sense);
+ }
+ } else if (functionName.equals("trans-bottom")) {
+ sense = null;
+ } else if (functionName.equals("trans-mid")) {
+ } else if (functionName.equals("trans-see")) {
+ incrementCount("WARNING:trans-see");
+ } else if (functionName.startsWith("picdic")) {
+ } else if (functionName.startsWith("checktrans")) {
+ done = true;
+ } else if (functionName.startsWith("ttbc")) {
+ wikiTokenizer.nextLine();
+ // TODO: would be great to handle ttbc
+ // TODO: Check this: done = true;
+ } else {
+ LOG.warning("Unexpected translation wikifunction: " + wikiTokenizer.token() + ", title=" + title);
+ }
+ } else if (wikiTokenizer.isListItem()) {
+ final String line = wikiTokenizer.listItemWikiText();
+ // This line could produce an output...
+
+// if (line.contains("ich hoan dich gear")) {
+// //System.out.println();
+// }
+
+ // First strip the language and check whether it matches.
+ // And hold onto it for sub-lines.
+ final int colonIndex = line.indexOf(":");
+ if (colonIndex == -1) {
+ continue;
+ }
+
+ final String lang = trim(WikiTokenizer.toPlainText(line.substring(0, colonIndex)));
+ incrementCount("tCount:" + lang);
+ final boolean appendLang;
+ if (wikiTokenizer.listItemPrefix().length() == 1) {
+ topLevelLang = lang;
+ final boolean thisFind = langPattern.matcher(lang).find();
+ if (!thisFind) {
+ continue;
+ }
+ appendLang = !langPattern.matcher(lang).matches();
+ } else if (topLevelLang == null) {
+ continue;
+ } else {
+ // Two-level -- the only way we won't append is if this second level matches exactly.
+ if (!langPattern.matcher(lang).matches() && !langPattern.matcher(topLevelLang).find()) {
+ continue;
+ }
+ appendLang = !langPattern.matcher(lang).matches();
+ }
+
+ String rest = line.substring(colonIndex + 1).trim();
+ if (rest.length() > 0) {
+ doTranslationLine(line, appendLang ? lang : null, pos, sense, rest);
+ }
+
+ } else if (wikiTokenizer.remainderStartsWith("''See''")) {
+ wikiTokenizer.nextLine();
+ incrementCount("WARNING: ''See''" );
+ LOG.fine("Skipping See line: " + wikiTokenizer.token());
+ } else if (wikiTokenizer.isWikiLink()) {
+ final String wikiLink = wikiTokenizer.wikiLinkText();
+ if (wikiLink.contains(":") && wikiLink.contains(title)) {
+ } else if (wikiLink.contains("Category:")) {
+ } else {
+ incrementCount("WARNING: Unexpected wikiLink" );
+ LOG.warning("Unexpected wikiLink: " + wikiTokenizer.token() + ", title=" + title);
+ }
+ } else if (wikiTokenizer.isNewline() || wikiTokenizer.isMarkup() || wikiTokenizer.isComment()) {
+ } else {
+ final String token = wikiTokenizer.token();
+ if (token.equals("----")) {
+ } else {
+ LOG.warning("Unexpected translation token: " + wikiTokenizer.token() + ", title=" + title);
+ incrementCount("WARNING: Unexpected translation token" );
+ }
+ }
+
+ }
+ }
+
+ private void doTranslationLine(final String line, final String lang, final String pos, final String sense, final String rest) {
+ state = State.TRANSLATION_LINE;
+ // Good chance we'll actually file this one...
+ final PairEntry pairEntry = new PairEntry(entrySource);
+ final IndexedEntry indexedEntry = new IndexedEntry(pairEntry);
+
+ final StringBuilder foreignText = new StringBuilder();
+ appendAndIndexWikiCallback.reset(foreignText, indexedEntry);
+ appendAndIndexWikiCallback.dispatch(rest, foreignIndexBuilder, EntryTypeName.WIKTIONARY_TRANSLATION_OTHER_TEXT);
+
+ if (foreignText.length() == 0) {
+ LOG.warning("Empty foreignText: " + line);
+ incrementCount("WARNING: Empty foreignText" );
+ return;
+ }
+
+ if (lang != null) {
+ foreignText.insert(0, String.format("(%s) ", lang));
+ }
+
+ StringBuilder englishText = new StringBuilder();
+
+ englishText.append(title);
+ if (sense != null) {
+ englishText.append(" (").append(sense).append(")");
+ enIndexBuilder.addEntryWithString(indexedEntry, sense, EntryTypeName.WIKTIONARY_TRANSLATION_SENSE);
+ }
+ if (pos != null) {
+ englishText.append(" (").append(pos.toLowerCase()).append(")");
+ }
+ enIndexBuilder.addEntryWithString(indexedEntry, title, EntryTypeName.WIKTIONARY_TITLE_MULTI);
+
+ final Pair pair = new Pair(trim(englishText.toString()), trim(foreignText.toString()), swap);
+ pairEntry.pairs.add(pair);
+ if (!pairsAdded.add(pair.toString())) {
+ LOG.warning("Duplicate pair: " + pair.toString());
+ incrementCount("WARNING: Duplicate pair" );
+ }
+ }
+ } // EnToTranslationParser
+
+ // -----------------------------------------------------------------------
+
+
+ static final class ForeignParser extends EnParser {
+
+ ForeignParser(final IndexBuilder enIndexBuilder,
+ final IndexBuilder otherIndexBuilder, final Pattern langPattern,
+ final Pattern langCodePattern, final boolean swap) {
+ super(enIndexBuilder, otherIndexBuilder, langPattern, langCodePattern, swap);
+ }
+
+ @Override
+ void parseSection(String heading, String text) {
+ if (isIgnorableTitle(title)) {
+ return;
+ }
+ final String lang = heading.replaceAll("=", "").trim();
+ if (!langPattern.matcher(lang).find()){
+ return;
+ }
+
+ final WikiTokenizer wikiTokenizer = new WikiTokenizer(text);
+ while (wikiTokenizer.nextToken() != null) {
+ if (wikiTokenizer.isHeading()) {
+ final String headingName = wikiTokenizer.headingWikiText();
+ if (headingName.equals("Translations")) {
+ LOG.warning("Translations not in English section: " + title);
+ } else if (headingName.equals("Pronunciation")) {
+ //doPronunciation(wikiLineReader);
+ } else if (partOfSpeechHeader.matcher(headingName).matches()) {
+ doForeignPartOfSpeech(lang, headingName, wikiTokenizer.headingDepth(), wikiTokenizer);
+ }
+ } else {
+ // It's not a heading.
+ // TODO: optimization: skip to next heading.
+ }
+ }
+ }
+
+ static final class ListSection {
+ final String firstPrefix;
+ final String firstLine;
+ final List<String> nextPrefixes = new ArrayList<String>();
+ final List<String> nextLines = new ArrayList<String>();
+
+ public ListSection(String firstPrefix, String firstLine) {
+ this.firstPrefix = firstPrefix;
+ this.firstLine = firstLine;
+ }
+
+ @Override
+ public String toString() {
+ return firstPrefix + firstLine + "{ " + nextPrefixes + "}";
+ }
+ }
+
+ int foreignCount = 0;
+ private void doForeignPartOfSpeech(final String lang, String posHeading, final int posDepth, WikiTokenizer wikiTokenizer) {
+ if (++foreignCount % 1000 == 0) {
+ LOG.info("***" + lang + ", " + title + ", pos=" + posHeading + ", foreignCount=" + foreignCount);
+ }
+ if (title.equals("6")) {
+ System.out.println();
+ }
+
+ final StringBuilder foreignBuilder = new StringBuilder();
+ final List<ListSection> listSections = new ArrayList<ListSection>();
+
+ appendAndIndexWikiCallback.reset(foreignBuilder, null);
+ this.state = State.ENGLISH_DEF_OF_FOREIGN; // TODO: this is wrong, need new category....
+ titleAppended = false;
+ wordForms.clear();
+
+ try {
+
+ ListSection lastListSection = null;
+
+ int currentHeadingDepth = posDepth;
+ while (wikiTokenizer.nextToken() != null) {
+ if (wikiTokenizer.isHeading()) {
+ currentHeadingDepth = wikiTokenizer.headingDepth();
+
+ if (currentHeadingDepth <= posDepth) {
+ wikiTokenizer.returnToLineStart();
+ return;
+ }
+ }
+
+ if (currentHeadingDepth > posDepth) {
+ // TODO: deal with other neat info sections
+ continue;
+ }
+
+ if (wikiTokenizer.isFunction()) {
+ final String name = wikiTokenizer.functionName();
+ final List<String> args = wikiTokenizer.functionPositionArgs();
+ final Map<String,String> namedArgs = wikiTokenizer.functionNamedArgs();
+ // First line is generally a repeat of the title with some extra information.
+ // We need to build up the left side (foreign text, tokens) separately from the
+ // right side (English). The left-side may get paired with multiple right sides.
+ // The left side should get filed under every form of the word in question (singular, plural).
+
+ // For verbs, the conjugation comes later on in a deeper section.
+ // Ideally, we'd want to file every English entry with the verb
+ // under every verb form coming from the conjugation.
+ // Ie. under "fa": see: "make :: fare" and "do :: fare"
+ // But then where should we put the conjugation table?
+ // I think just under fare. But then we need a way to link to the entry (actually the row, since entries doesn't show up!)
+ // for the conjugation table from "fa".
+ // Would like to be able to link to a lang#token.
+
+ appendAndIndexWikiCallback.onFunction(wikiTokenizer, name, args, namedArgs);
+
+ } else if (wikiTokenizer.isListItem()) {
+ final String prefix = wikiTokenizer.listItemPrefix();
+ if (lastListSection != null &&
+ prefix.startsWith(lastListSection.firstPrefix) &&
+ prefix.length() > lastListSection.firstPrefix.length()) {
+ lastListSection.nextPrefixes.add(prefix);
+ lastListSection.nextLines.add(wikiTokenizer.listItemWikiText());
+ } else {
+ lastListSection = new ListSection(prefix, wikiTokenizer.listItemWikiText());
+ listSections.add(lastListSection);
+ }
+ } else if (lastListSection != null) {
+ // Don't append anything after the lists, because there's crap.
+ } else if (wikiTokenizer.isWikiLink()) {
+ // Unindexed!
+ foreignBuilder.append(wikiTokenizer.wikiLinkText());
+
+ } else if (wikiTokenizer.isPlainText()) {
+ // Unindexed!
+ foreignBuilder.append(wikiTokenizer.token());
+
+ } else if (wikiTokenizer.isMarkup() || wikiTokenizer.isNewline() || wikiTokenizer.isComment()) {
+ // Do nothing.
+ } else {
+ LOG.warning("Unexpected token: " + wikiTokenizer.token());
+ }
+ }
+
+ } finally {
+ // Here's where we exit.
+ // Should we make an entry even if there are no foreign list items?
+ String foreign = foreignBuilder.toString().trim();
+ if (!titleAppended && !foreign.toLowerCase().startsWith(title.toLowerCase())) {
+ foreign = String.format("%s %s", title, foreign);
+ }
+ if (!langPattern.matcher(lang).matches()) {
+ foreign = String.format("(%s) %s", lang, foreign);
+ }
+ for (final ListSection listSection : listSections) {
+ doForeignListSection(foreign, title, wordForms, listSection);
+ }
+ }
+ }
+
+
+
+ private void doForeignListSection(final String foreignText, String title, final Collection<String> forms, final ListSection listSection) {
+ state = State.ENGLISH_DEF_OF_FOREIGN;
+ final String prefix = listSection.firstPrefix;
+ if (prefix.length() > 1) {
+ // Could just get looser and say that any prefix longer than first is a sublist.
+ LOG.warning("Prefix too long: " + listSection);
+ return;
+ }
+
+ final PairEntry pairEntry = new PairEntry(entrySource);
+ final IndexedEntry indexedEntry = new IndexedEntry(pairEntry);
+
+ entryIsFormOfSomething = false;
+ final StringBuilder englishBuilder = new StringBuilder();
+ final String mainLine = listSection.firstLine;
+ appendAndIndexWikiCallback.reset(englishBuilder, indexedEntry);
+ appendAndIndexWikiCallback.dispatch(mainLine, enIndexBuilder, EntryTypeName.WIKTIONARY_ENGLISH_DEF);
+
+ final String english = trim(englishBuilder.toString());
+ if (english.length() > 0) {
+ final Pair pair = new Pair(english, trim(foreignText), this.swap);
+ pairEntry.pairs.add(pair);
+ foreignIndexBuilder.addEntryWithString(indexedEntry, title, entryIsFormOfSomething ? EntryTypeName.WIKTIONARY_IS_FORM_OF_SOMETHING_ELSE : EntryTypeName.WIKTIONARY_TITLE_MULTI);
+ for (final String form : forms) {
+ foreignIndexBuilder.addEntryWithString(indexedEntry, form, EntryTypeName.WIKTIONARY_INFLECTED_FORM_MULTI);
+ }
+ }
+
+ // Do examples.
+ String lastForeign = null;
+ for (int i = 0; i < listSection.nextPrefixes.size(); ++i) {
+ final String nextPrefix = listSection.nextPrefixes.get(i);
+ final String nextLine = listSection.nextLines.get(i);
+
+ // TODO: This splitting is not sensitive to wiki code.
+ int dash = nextLine.indexOf("—");
+ int mdashLen = 7;
+ if (dash == -1) {
+ dash = nextLine.indexOf("—");
+ mdashLen = 1;
+ }
+ if (dash == -1) {
+ dash = nextLine.indexOf(" - ");
+ mdashLen = 3;
+ }
+
+ if ((nextPrefix.equals("#:") || nextPrefix.equals("##:")) && dash != -1) {
+ final String foreignEx = nextLine.substring(0, dash);
+ final String englishEx = nextLine.substring(dash + mdashLen);
+ final Pair pair = new Pair(formatAndIndexExampleString(englishEx, enIndexBuilder, indexedEntry), formatAndIndexExampleString(foreignEx, foreignIndexBuilder, indexedEntry), swap);
+ if (pair.lang1 != "--" && pair.lang1 != "--") {
+ pairEntry.pairs.add(pair);
+ }
+ lastForeign = null;
+ } else if (nextPrefix.equals("#:") || nextPrefix.equals("##:")){
+ final Pair pair = new Pair("--", formatAndIndexExampleString(nextLine, null, indexedEntry), swap);
+ lastForeign = nextLine;
+ if (pair.lang1 != "--" && pair.lang1 != "--") {
+ pairEntry.pairs.add(pair);
+ }
+ } else if (nextPrefix.equals("#::") || nextPrefix.equals("#**")) {
+ if (lastForeign != null && pairEntry.pairs.size() > 0) {
+ pairEntry.pairs.remove(pairEntry.pairs.size() - 1);
+ final Pair pair = new Pair(formatAndIndexExampleString(nextLine, enIndexBuilder, indexedEntry), formatAndIndexExampleString(lastForeign, foreignIndexBuilder, indexedEntry), swap);
+ if (pair.lang1 != "--" || pair.lang2 != "--") {
+ pairEntry.pairs.add(pair);
+ }
+ lastForeign = null;
+ } else {
+ LOG.warning("TODO: English example with no foreign: " + title + ", " + nextLine);
+ final Pair pair = new Pair("--", formatAndIndexExampleString(nextLine, null, indexedEntry), swap);
+ if (pair.lang1 != "--" || pair.lang2 != "--") {
+ pairEntry.pairs.add(pair);
+ }
+ }
+ } else if (nextPrefix.equals("#*")) {
+ // Can't really index these.
+ final Pair pair = new Pair("--", formatAndIndexExampleString(nextLine, null, indexedEntry), swap);
+ lastForeign = nextLine;
+ if (pair.lang1 != "--" || pair.lang2 != "--") {
+ pairEntry.pairs.add(pair);
+ }
+ } else if (nextPrefix.equals("#::*") || nextPrefix.equals("##") || nextPrefix.equals("#*:") || nextPrefix.equals("#:*") || true) {
+ final Pair pair = new Pair("--", formatAndIndexExampleString(nextLine, null, indexedEntry), swap);
+ if (pair.lang1 != "--" || pair.lang2 != "--") {
+ pairEntry.pairs.add(pair);
+ }
+// } else {
+// assert false;
+ }
+ }
+ }
+
+ private String formatAndIndexExampleString(final String example, final IndexBuilder indexBuilder, final IndexedEntry indexedEntry) {
+ // TODO:
+// if (wikiTokenizer.token().equals("'''")) {
+// insideTripleQuotes = !insideTripleQuotes;
+// }
+ final StringBuilder builder = new StringBuilder();
+ appendAndIndexWikiCallback.reset(builder, indexedEntry);
+ appendAndIndexWikiCallback.entryTypeName = EntryTypeName.WIKTIONARY_EXAMPLE;
+ appendAndIndexWikiCallback.entryTypeNameSticks = true;
+ try {
+ // TODO: this is a hack needed because we don't safely split on the dash.
+ appendAndIndexWikiCallback.dispatch(example, indexBuilder, EntryTypeName.WIKTIONARY_EXAMPLE);
+ } catch (AssertionError e) {
+ return "--";
+ }
+ final String result = trim(builder.toString());
+ return result.length() > 0 ? result : "--";
+ }
+
+
+ private void itConjAre(List<String> args, Map<String, String> namedArgs) {
+ final String base = args.get(0);
+ final String aux = args.get(1);
+
+ putIfMissing(namedArgs, "inf", base + "are");
+ putIfMissing(namedArgs, "aux", aux);
+ putIfMissing(namedArgs, "ger", base + "ando");
+ putIfMissing(namedArgs, "presp", base + "ante");
+ putIfMissing(namedArgs, "pastp", base + "ato");
+ // Present
+ putIfMissing(namedArgs, "pres1s", base + "o");
+ putIfMissing(namedArgs, "pres2s", base + "i");
+ putIfMissing(namedArgs, "pres3s", base + "a");
+ putIfMissing(namedArgs, "pres1p", base + "iamo");
+ putIfMissing(namedArgs, "pres2p", base + "ate");
+ putIfMissing(namedArgs, "pres3p", base + "ano");
+ // Imperfect
+ putIfMissing(namedArgs, "imperf1s", base + "avo");
+ putIfMissing(namedArgs, "imperf2s", base + "avi");
+ putIfMissing(namedArgs, "imperf3s", base + "ava");
+ putIfMissing(namedArgs, "imperf1p", base + "avamo");
+ putIfMissing(namedArgs, "imperf2p", base + "avate");
+ putIfMissing(namedArgs, "imperf3p", base + "avano");
+ // Passato remoto
+ putIfMissing(namedArgs, "prem1s", base + "ai");
+ putIfMissing(namedArgs, "prem2s", base + "asti");
+ putIfMissing(namedArgs, "prem3s", base + "ò");
+ putIfMissing(namedArgs, "prem1p", base + "ammo");
+ putIfMissing(namedArgs, "prem2p", base + "aste");
+ putIfMissing(namedArgs, "prem3p", base + "arono");
+ // Future
+ putIfMissing(namedArgs, "fut1s", base + "erò");
+ putIfMissing(namedArgs, "fut2s", base + "erai");
+ putIfMissing(namedArgs, "fut3s", base + "erà");
+ putIfMissing(namedArgs, "fut1p", base + "eremo");
+ putIfMissing(namedArgs, "fut2p", base + "erete");
+ putIfMissing(namedArgs, "fut3p", base + "eranno");
+ // Conditional
+ putIfMissing(namedArgs, "cond1s", base + "erei");
+ putIfMissing(namedArgs, "cond2s", base + "eresti");
+ putIfMissing(namedArgs, "cond3s", base + "erebbe");
+ putIfMissing(namedArgs, "cond1p", base + "eremmo");
+ putIfMissing(namedArgs, "cond2p", base + "ereste");
+ putIfMissing(namedArgs, "cond3p", base + "erebbero");
+ // Subjunctive / congiuntivo
+ putIfMissing(namedArgs, "sub123s", base + "i");
+ putIfMissing(namedArgs, "sub1p", base + "iamo");
+ putIfMissing(namedArgs, "sub2p", base + "iate");
+ putIfMissing(namedArgs, "sub3p", base + "ino");
+ // Imperfect subjunctive
+ putIfMissing(namedArgs, "impsub12s", base + "assi");
+ putIfMissing(namedArgs, "impsub3s", base + "asse");
+ putIfMissing(namedArgs, "impsub1p", base + "assimo");
+ putIfMissing(namedArgs, "impsub2p", base + "aste");
+ putIfMissing(namedArgs, "impsub3p", base + "assero");
+ // Imperative
+ putIfMissing(namedArgs, "imp2s", base + "a");
+ putIfMissing(namedArgs, "imp3s", base + "i");
+ putIfMissing(namedArgs, "imp1p", base + "iamo");
+ putIfMissing(namedArgs, "imp2p", base + "ate");
+ putIfMissing(namedArgs, "imp3p", base + "ino");
+
+
+ itConj(args, namedArgs);
+ }
+
+
+ private void itConj(List<String> args, Map<String, String> namedArgs) {
+ // TODO Auto-generated method stub
+
+ }
+
+
+ private static void putIfMissing(final Map<String, String> namedArgs, final String key,
+ final String value) {
+ final String oldValue = namedArgs.get(key);
+ if (oldValue == null || oldValue.length() == 0) {
+ namedArgs.put(key, value);
+ }
+ }
+
+ // TODO: check how ='' and =| are manifested....
+ // TODO: get this right in -are
+ private static void putOrNullify(final Map<String, String> namedArgs, final String key,
+ final String value) {
+ final String oldValue = namedArgs.get(key);
+ if (oldValue == null/* || oldValue.length() == 0*/) {
+ namedArgs.put(key, value);
+ } else {
+ if (oldValue.equals("''")) {
+ namedArgs.put(key, "");
+ }
+ }
+ }
+
+ } // ForeignParser
+
+ // -----------------------------------------------------------------------
+
+ static class FunctionCallbacks {
+
+ static final Map<String,FunctionCallback<EnParser>> DEFAULT = new LinkedHashMap<String, FunctionCallback<EnParser>>();
+
+ static {
+ FunctionCallback<EnParser> callback = new TranslationCallback();
+ DEFAULT.put("t", callback);
+ DEFAULT.put("t+", callback);
+ DEFAULT.put("t-", callback);
+ DEFAULT.put("tø", callback);
+ DEFAULT.put("apdx-t", callback);
+
+ callback = new EncodingCallback();
+ Set<String> encodings = new LinkedHashSet<String>(Arrays.asList(
+ "zh-ts", "zh-tsp",
+ "sd-Arab", "ku-Arab", "Arab", "unicode", "Laoo", "ur-Arab", "Thai",
+ "fa-Arab", "Khmr", "Cyrl", "IPAchar", "ug-Arab", "ko-inline",
+ "Jpan", "Kore", "Hebr", "rfscript", "Beng", "Mong", "Knda", "Cyrs",
+ "yue-tsj", "Mlym", "Tfng", "Grek", "yue-yue-j"));
+ for (final String encoding : encodings) {
+ 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);
+
+ callback = new AppendArg0();
+
+ 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);
+
+
+ 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());
+ }
+
+ static final NameAndArgs<EnParser> NAME_AND_ARGS = new NameAndArgs<EnParser>();
+
+ // ------------------------------------------------------------------
+
+ static final class TranslationCallback implements FunctionCallback<EnParser> {
+ @Override
+ public boolean onWikiFunction(final WikiTokenizer wikiTokenizer, final String name, final List<String> args,
+ final Map<String, String> namedArgs, final EnParser parser,
+ final AppendAndIndexWikiCallback<EnParser> appendAndIndexWikiCallback) {
+
+ final String transliteration = namedArgs.remove("tr");
+ final String alt = namedArgs.remove("alt");
+ namedArgs.keySet().removeAll(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);
+ 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(" (");
+ 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(" {");
+ appendNamedArgs(namedArgs, appendAndIndexWikiCallback);
+ appendAndIndexWikiCallback.builder.append("}");
+ }
+
+ return true;
+ }
+ }
+
+ // ------------------------------------------------------------------
+
+ static final class QualifierCallback implements FunctionCallback<EnParser> {
+ @Override
+ public boolean onWikiFunction(final WikiTokenizer wikiTokenizer, final String name, final List<String> args,
+ final Map<String, String> namedArgs,
+ final EnParser parser,
+ final AppendAndIndexWikiCallback<EnParser> appendAndIndexWikiCallback) {
+ if (args.size() != 1 || !namedArgs.isEmpty()) {
+ LOG.warning("weird qualifier: ");
+ return false;
+ }
+ String qualifier = args.get(0);
+ appendAndIndexWikiCallback.builder.append("(");
+ appendAndIndexWikiCallback.dispatch(qualifier, null);
+ appendAndIndexWikiCallback.builder.append(")");
+ return true;
+ }
+ }
+
+ // ------------------------------------------------------------------
+
+ static final class EncodingCallback implements FunctionCallback<EnParser> {
+ @Override
+ public boolean onWikiFunction(final WikiTokenizer wikiTokenizer, final String name, final List<String> args,
+ final Map<String, String> namedArgs,
+ final EnParser parser,
+ final AppendAndIndexWikiCallback<EnParser> appendAndIndexWikiCallback) {
+ if (!namedArgs.isEmpty()) {
+ LOG.warning("weird encoding: " + wikiTokenizer.token());
+ }
+ 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;
+ }
+ }
+
+ // ------------------------------------------------------------------
+
+ static final class Gender implements FunctionCallback<EnParser> {
+ @Override
+ public boolean onWikiFunction(final WikiTokenizer wikiTokenizer, final String name, final List<String> args,
+ final Map<String, String> namedArgs,
+ final EnParser parser,
+ final AppendAndIndexWikiCallback<EnParser> appendAndIndexWikiCallback) {
+ if (!namedArgs.isEmpty()) {
+ return false;
+ }
+ appendAndIndexWikiCallback.builder.append("{");
+ appendAndIndexWikiCallback.builder.append(name);
+ for (int i = 0; i < args.size(); ++i) {
+ appendAndIndexWikiCallback.builder.append("|").append(args.get(i));
+ }
+ appendAndIndexWikiCallback.builder.append("}");
+ return true;
+ }
+ }
+
+ // ------------------------------------------------------------------
+
+ static final class l_term implements FunctionCallback<EnParser> {
+ @Override
+ public boolean onWikiFunction(final WikiTokenizer wikiTokenizer, final String name, final List<String> args,
+ final Map<String, String> namedArgs,
+ final EnParser parser,
+ final AppendAndIndexWikiCallback<EnParser> appendAndIndexWikiCallback) {
+
+ // 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);
+ final IndexBuilder indexBuilder;
+ if ("".equals(langCode)) {
+ indexBuilder = parser.foreignIndexBuilder;
+ } else if ("en".equals(langCode)) {
+ indexBuilder = parser.enIndexBuilder;
+ } else {
+ indexBuilder = parser.foreignIndexBuilder;
+ }
+
+ 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(USELESS_WIKI_ARGS);
+ if (!namedArgs.isEmpty()) {
+ appendAndIndexWikiCallback.builder.append(" {").append(name);
+ appendNamedArgs(namedArgs, appendAndIndexWikiCallback);
+ appendAndIndexWikiCallback.builder.append("}");
+ }
+
+ return true;
+ }
+ }
+
+ // ------------------------------------------------------------------
+
+ static final class AppendArg0 implements FunctionCallback<EnParser> {
+ @Override
+ public boolean onWikiFunction(final WikiTokenizer wikiTokenizer, final String name, final List<String> args,
+ final Map<String, String> namedArgs,
+ final EnParser parser,
+ final AppendAndIndexWikiCallback<EnParser> appendAndIndexWikiCallback) {
+ if (args.size() != 1 || !namedArgs.isEmpty()) {
+ return false;
+ }
+ appendAndIndexWikiCallback.dispatch(args.get(0), EntryTypeName.WIKTIONARY_TRANSLATION_OTHER_TEXT);
+ // TODO: transliteration
+ return true;
+ }
+ }
+
+ // ------------------------------------------------------------------
+
+ static final class italbrac implements FunctionCallback<EnParser> {
+ @Override
+ public boolean onWikiFunction(final WikiTokenizer wikiTokenizer, final String name, final List<String> args,
+ final Map<String, String> namedArgs,
+ final EnParser parser,
+ final AppendAndIndexWikiCallback<EnParser> appendAndIndexWikiCallback) {
+ if (args.size() != 1 || !namedArgs.isEmpty()) {
+ return false;
+ }
+ appendAndIndexWikiCallback.builder.append("(");
+ appendAndIndexWikiCallback.dispatch(args.get(0), EntryTypeName.WIKTIONARY_TRANSLATION_OTHER_TEXT);
+ appendAndIndexWikiCallback.builder.append(")");
+ return true;
+ }
+ }
+
+ // ------------------------------------------------------------------
+
+ static final class gloss implements FunctionCallback<EnParser> {
+ @Override
+ public boolean onWikiFunction(final WikiTokenizer wikiTokenizer, final String name, final List<String> args,
+ final Map<String, String> namedArgs,
+ final EnParser parser,
+ final AppendAndIndexWikiCallback<EnParser> appendAndIndexWikiCallback) {
+ if (args.size() != 1 || !namedArgs.isEmpty()) {
+ return false;
+ }
+ appendAndIndexWikiCallback.builder.append("(");
+ appendAndIndexWikiCallback.dispatch(args.get(0), EntryTypeName.WIKTIONARY_TRANSLATION_OTHER_TEXT);
+ appendAndIndexWikiCallback.builder.append(")");
+ return true;
+ }
+ }
+
+ // ------------------------------------------------------------------
+
+ static final class Ignore implements FunctionCallback<EnParser> {
+ @Override
+ public boolean onWikiFunction(final WikiTokenizer wikiTokenizer, final String name, final List<String> args,
+ final Map<String, String> namedArgs,
+ final EnParser parser,
+ final AppendAndIndexWikiCallback<EnParser> appendAndIndexWikiCallback) {
+ return true;
+ }
+ }
+
+ // ------------------------------------------------------------------
+
+ static final class not_used implements FunctionCallback<EnParser> {
+ @Override
+ public boolean onWikiFunction(final WikiTokenizer wikiTokenizer, final String name, final List<String> args,
+ final Map<String, String> namedArgs,
+ final EnParser parser,
+ final AppendAndIndexWikiCallback<EnParser> appendAndIndexWikiCallback) {
+ appendAndIndexWikiCallback.builder.append("(not used)");
+ return true;
+ }
+ }
+
+
+ // ------------------------------------------------------------------
+
+ static final class AppendName implements FunctionCallback<EnParser> {
+ @Override
+ public boolean onWikiFunction(final WikiTokenizer wikiTokenizer, final String name, final List<String> args,
+ final Map<String, String> namedArgs,
+ final EnParser parser,
+ final AppendAndIndexWikiCallback<EnParser> appendAndIndexWikiCallback) {
+ if (!args.isEmpty() || !namedArgs.isEmpty()) {
+ return false;
+ }
+ appendAndIndexWikiCallback.builder.append(name);
+ return true;
+ }
+ }
+
+ // --------------------------------------------------------------------
+ // --------------------------------------------------------------------
+
+
+ static final class FormOf implements FunctionCallback<EnParser> {
+ @Override
+ public boolean onWikiFunction(final WikiTokenizer wikiTokenizer, final String name, final List<String> args,
+ final Map<String, String> namedArgs,
+ final EnParser parser,
+ final AppendAndIndexWikiCallback<EnParser> appendAndIndexWikiCallback) {
+ parser.entryIsFormOfSomething = true;
+ String formName = name;
+ if (name.equals("form of")) {
+ formName = ListUtil.remove(args, 0, null);
+ }
+ if (formName == null) {
+ LOG.warning("Missing form name: " + parser.title);
+ formName = "form of";
+ }
+ String baseForm = ListUtil.get(args, 1, "");
+ if ("".equals(baseForm)) {
+ baseForm = ListUtil.get(args, 0, null);
+ ListUtil.remove(args, 1, "");
+ } else {
+ ListUtil.remove(args, 0, null);
+ }
+ namedArgs.keySet().removeAll(USELESS_WIKI_ARGS);
+
+ appendAndIndexWikiCallback.builder.append("{");
+ NAME_AND_ARGS.onWikiFunction(wikiTokenizer, formName, args, namedArgs, parser, appendAndIndexWikiCallback);
+ appendAndIndexWikiCallback.builder.append("}");
+ if (baseForm != null && appendAndIndexWikiCallback.indexedEntry != null) {
+ parser.foreignIndexBuilder.addEntryWithString(appendAndIndexWikiCallback.indexedEntry, baseForm, EntryTypeName.WIKTIONARY_BASE_FORM_MULTI);
+ } else {
+ // null baseForm happens in Danish.
+ LOG.warning("Null baseform: " + parser.title);
+ }
+ return true;
+ }
+ }
+
+ static final FormOf FORM_OF = new FormOf();
+
+
+ // --------------------------------------------------------------------
+ // --------------------------------------------------------------------
+
+ static final class wikipedia implements FunctionCallback<EnParser> {
+ @Override
+ public boolean onWikiFunction(final WikiTokenizer wikiTokenizer, final String name, final List<String> args,
+ final Map<String, String> namedArgs,
+ final EnParser parser,
+ final AppendAndIndexWikiCallback<EnParser> appendAndIndexWikiCallback) {
+ namedArgs.remove("lang");
+ if (args.size() > 1 || !namedArgs.isEmpty()) {
+ // Unindexed!
+ return false;
+ } else if (args.size() == 1) {
+ return false;
+ } else {
+ return true;
+ }
+ }
+ }
+
+ static final class InflOrHead implements FunctionCallback<EnParser> {
+ @Override
+ public boolean onWikiFunction(final WikiTokenizer wikiTokenizer, final String name, final List<String> args,
+ final Map<String, String> namedArgs,
+ final EnParser parser,
+ final AppendAndIndexWikiCallback<EnParser> appendAndIndexWikiCallback) {
+ // See: http://en.wiktionary.org/wiki/Template:infl
+ final String langCode = ListUtil.get(args, 0);
+ String head = namedArgs.remove("head");
+ if (head == null) {
+ head = namedArgs.remove("title"); // Bug
+ }
+ if (head == null) {
+ head = parser.title;
+ }
+ parser.titleAppended = true;
+
+ namedArgs.keySet().removeAll(USELESS_WIKI_ARGS);
+
+ final String tr = namedArgs.remove("tr");
+ String g = namedArgs.remove("g");
+ if (g == null) {
+ g = namedArgs.remove("gender");
+ }
+ final String g2 = namedArgs.remove("g2");
+ final String g3 = namedArgs.remove("g3");
+
+ appendAndIndexWikiCallback.dispatch(head, EntryTypeName.WIKTIONARY_TITLE_MULTI);
+
+ if (g != null) {
+ appendAndIndexWikiCallback.builder.append(" {").append(g);
+ if (g2 != null) {
+ appendAndIndexWikiCallback.builder.append("|").append(g2);
+ }
+ if (g3 != null) {
+ appendAndIndexWikiCallback.builder.append("|").append(g3);
+ }
+ appendAndIndexWikiCallback.builder.append("}");
+ }
+
+ if (tr != null) {
+ appendAndIndexWikiCallback.builder.append(" (");
+ appendAndIndexWikiCallback.dispatch(tr, EntryTypeName.WIKTIONARY_TITLE_MULTI);
+ appendAndIndexWikiCallback.builder.append(")");
+ parser.wordForms.add(tr);
+ }
+
+ final String pos = ListUtil.get(args, 1);
+ if (pos != null) {
+ appendAndIndexWikiCallback.builder.append(" (").append(pos).append(")");
+ }
+ for (int i = 2; i < args.size(); i += 2) {
+ final String inflName = ListUtil.get(args, i);
+ final String inflValue = ListUtil.get(args, i + 1);
+ appendAndIndexWikiCallback.builder.append(", ");
+ appendAndIndexWikiCallback.dispatch(inflName, null, null);
+ if (inflValue != null && inflValue.length() > 0) {
+ appendAndIndexWikiCallback.builder.append(": ");
+ appendAndIndexWikiCallback.dispatch(inflValue, null, null);
+ parser.wordForms.add(inflValue);
+ }
+ }
+ for (final String key : namedArgs.keySet()) {
+ final String value = WikiTokenizer.toPlainText(namedArgs.get(key));
+ appendAndIndexWikiCallback.builder.append(" ");
+ appendAndIndexWikiCallback.dispatch(key, null, null);
+ appendAndIndexWikiCallback.builder.append("=");
+ appendAndIndexWikiCallback.dispatch(value, null, null);
+ parser.wordForms.add(value);
+ }
+ return true;
+ }
+ }
+
+
+ static {
+ DEFAULT.put("it-noun", new it_noun());
+ }
+ static final class it_noun implements FunctionCallback<EnParser> {
+ @Override
+ public boolean onWikiFunction(final WikiTokenizer wikiTokenizer, final String name, final List<String> args,
+ final Map<String, String> namedArgs,
+ final EnParser parser,
+ final AppendAndIndexWikiCallback<EnParser> appendAndIndexWikiCallback) {
+ parser.titleAppended = true;
+ final String base = ListUtil.get(args, 0);
+ final String gender = ListUtil.get(args, 1);
+ final String singular = base + ListUtil.get(args, 2, null);
+ final String plural = base + ListUtil.get(args, 3, null);
+ appendAndIndexWikiCallback.builder.append(" ");
+ appendAndIndexWikiCallback.dispatch(singular, null, null);
+ appendAndIndexWikiCallback.builder.append(" {").append(gender).append("}, ");
+ appendAndIndexWikiCallback.dispatch(plural, null, null);
+ appendAndIndexWikiCallback.builder.append(" {pl}");
+ parser.wordForms.add(singular);
+ parser.wordForms.add(plural);
+ if (!namedArgs.isEmpty() || args.size() > 4) {
+ LOG.warning("Invalid it-noun: " + wikiTokenizer.token());
+ }
+ return true;
+ }
+ }
+
+ static {
+ DEFAULT.put("it-proper noun", new it_proper_noun());
+ }
+ static final class it_proper_noun implements FunctionCallback<EnParser> {
+ @Override
+ public boolean onWikiFunction(final WikiTokenizer wikiTokenizer, final String name, final List<String> args,
+ final Map<String, String> namedArgs,
+ final EnParser parser,
+ final AppendAndIndexWikiCallback<EnParser> appendAndIndexWikiCallback) {
+ return false;
+ }
+ }
+
+ }
+
+
+}
+++ /dev/null
-// 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.parser.wiktionary;
-
-import java.io.BufferedInputStream;
-import java.io.DataInputStream;
-import java.io.EOFException;
-import java.io.File;
-import java.io.FileInputStream;
-import java.io.IOException;
-import java.util.ArrayList;
-import java.util.Arrays;
-import java.util.Collection;
-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.Level;
-import java.util.logging.Logger;
-import java.util.regex.Pattern;
-
-import com.hughes.android.dictionary.engine.EntrySource;
-import com.hughes.android.dictionary.engine.EntryTypeName;
-import com.hughes.android.dictionary.engine.IndexBuilder;
-import com.hughes.android.dictionary.engine.IndexedEntry;
-import com.hughes.android.dictionary.engine.PairEntry;
-import com.hughes.android.dictionary.engine.PairEntry.Pair;
-import com.hughes.android.dictionary.parser.WikiTokenizer;
-
-public class EnWiktionaryXmlParser {
-
- static final Logger LOG = Logger.getLogger(EnWiktionaryXmlParser.class.getName());
-
- // TODO: process {{ttbc}} lines
-
- static final Pattern partOfSpeechHeader = Pattern.compile(
- "Noun|Verb|Adjective|Adverb|Pronoun|Conjunction|Interjection|" +
- "Preposition|Proper noun|Article|Prepositional phrase|Acronym|" +
- "Abbreviation|Initialism|Contraction|Prefix|Suffix|Symbol|Letter|" +
- "Ligature|Idiom|Phrase|\\{\\{acronym\\}\\}|\\{\\{initialism\\}\\}|" +
- "\\{\\{abbreviation\\}\\}|" +
- // These are @deprecated:
- "Noun form|Verb form|Adjective form|Nominal phrase|Noun phrase|" +
- "Verb phrase|Transitive verb|Intransitive verb|Reflexive verb|" +
- // These are extras I found:
- "Determiner|Numeral|Number|Cardinal number|Ordinal number|Proverb|" +
- "Particle|Interjection|Pronominal adverb" +
- "Han character|Hanzi|Hanja|Kanji|Katakana character|Syllable");
-
- EntrySource entrySource;
- final IndexBuilder enIndexBuilder;
- final IndexBuilder foreignIndexBuilder;
- final Pattern langPattern;
- final Pattern langCodePattern;
- final boolean swap;
-
- // State used while parsing.
- enum State {
- TRANSLATION_LINE,
- ENGLISH_DEF_OF_FOREIGN,
- ENGLISH_EXAMPLE,
- FOREIGN_EXAMPLE,
- }
- State state = null;
- String title;
-
- public EnWiktionaryXmlParser(final IndexBuilder enIndexBuilder, final IndexBuilder otherIndexBuilder, final Pattern langPattern, final Pattern langCodePattern, final boolean swap) {
- this.enIndexBuilder = enIndexBuilder;
- this.foreignIndexBuilder = otherIndexBuilder;
- this.langPattern = langPattern;
- this.langCodePattern = langCodePattern;
- this.swap = swap;
- }
-
-
- public void parse(final File file, final EntrySource entrySource, final int pageLimit) throws IOException {
- this.entrySource = entrySource;
- int pageCount = 0;
- final DataInputStream dis = new DataInputStream(new BufferedInputStream(new FileInputStream(file)));
- try {
- while (true) {
- if (pageLimit >= 0 && pageCount >= pageLimit) {
- return;
- }
-
- try {
- title = dis.readUTF();
- } catch (EOFException e) {
- LOG.log(Level.INFO, "EOF reading split.");
- dis.close();
- return;
- }
- final String heading = dis.readUTF();
- final int bytesLength = dis.readInt();
- final byte[] bytes = new byte[bytesLength];
- dis.readFully(bytes);
- final String text = new String(bytes, "UTF8");
-
- parseSection(heading, text);
-
- ++pageCount;
- if (pageCount % 1000 == 0) {
- LOG.info("pageCount=" + pageCount);
- }
- }
- } finally {
- System.out.println("lang Counts: " + appendAndIndexWikiCallback.langCodeToTCount);
- appendAndIndexWikiCallback.langCodeToTCount.keySet().removeAll(WiktionaryLangs.isoCodeToWikiName.keySet());
- System.out.println("unused Counts: " + appendAndIndexWikiCallback.langCodeToTCount);
- System.out.println("lang Counts: " + langNameToTCount);
- langNameToTCount.keySet().removeAll(WiktionaryLangs.isoCodeToWikiName.values());
- System.out.println("unknown counts: " + langNameToTCount);
- }
- }
-
- private void parseSection(String heading, final String text) {
- if (title.startsWith("Wiktionary:") ||
- title.startsWith("Template:") ||
- title.startsWith("Appendix:") ||
- title.startsWith("Category:") ||
- title.startsWith("Index:") ||
- title.startsWith("MediaWiki:") ||
- title.startsWith("TransWiki:") ||
- title.startsWith("Citations:") ||
- title.startsWith("Concordance:") ||
- title.startsWith("Help:")) {
- return;
- }
-
- heading = heading.replaceAll("=", "").trim();
- if (heading.equals("English")) {
- doEnglishWord(text);
- } else if (langPattern.matcher(heading).find()){
- doForeignWord(heading, text);
- }
-
- } // endPage()
-
- // -------------------------------------------------------------------------
-
- private void doEnglishWord(String text) {
-
- String pos = null;
- int posDepth = -1;
-
- final WikiTokenizer wikiTokenizer = new WikiTokenizer(text);
- while (wikiTokenizer.nextToken() != null) {
-
- if (wikiTokenizer.isHeading()) {
- final String headerName = wikiTokenizer.headingWikiText();
-
- if (wikiTokenizer.headingDepth() <= posDepth) {
- pos = null;
- posDepth = -1;
- }
-
- if (partOfSpeechHeader.matcher(headerName).matches()) {
- posDepth = wikiTokenizer.headingDepth();
- pos = wikiTokenizer.headingWikiText();
- // TODO: if we're inside the POS section, we should handle the first title line...
-
- } else if (headerName.equals("Translations")) {
- if (pos == null) {
- LOG.info("Translations without POS (but using anyway): " + title);
- }
- doTranslations(wikiTokenizer, pos);
- } else if (headerName.equals("Pronunciation")) {
- //doPronunciation(wikiLineReader);
- }
- } else if (wikiTokenizer.isFunction()) {
- final String name = wikiTokenizer.functionName();
- if (name.equals("head") && pos == null) {
- LOG.warning("{{head}} without POS: " + title);
- }
- }
- }
- }
-
- final AppendAndIndexWikiCallback appendAndIndexWikiCallback = new AppendAndIndexWikiCallback(this);
- {
- appendAndIndexWikiCallback.functionCallbacks.putAll(FunctionCallbacksDefault.DEFAULT);
- }
-
- final Map<String,AtomicInteger> langNameToTCount = new LinkedHashMap<String, AtomicInteger>();
-
- private void doTranslations(final WikiTokenizer wikiTokenizer, final String pos) {
- if (title.equals("absolutely")) {
- //System.out.println();
- }
-
- String topLevelLang = null;
- String sense = null;
- boolean done = false;
- while (wikiTokenizer.nextToken() != null) {
- if (wikiTokenizer.isHeading()) {
- wikiTokenizer.returnToLineStart();
- return;
- }
- if (done) {
- continue;
- }
-
- // Check whether we care about this line:
-
- if (wikiTokenizer.isFunction()) {
- final String functionName = wikiTokenizer.functionName();
- final List<String> positionArgs = wikiTokenizer.functionPositionArgs();
-
- if (functionName.equals("trans-top")) {
- sense = null;
- if (wikiTokenizer.functionPositionArgs().size() >= 1) {
- sense = positionArgs.get(0);
- // TODO: could emphasize words in [[brackets]] inside sense.
- sense = WikiTokenizer.toPlainText(sense);
- //LOG.info("Sense: " + sense);
- }
- } else if (functionName.equals("trans-bottom")) {
- sense = null;
- } else if (functionName.equals("trans-mid")) {
- } else if (functionName.equals("trans-see")) {
- // TODO: would also be nice...
- } else if (functionName.startsWith("picdic")) {
- } else if (functionName.startsWith("checktrans")) {
- done = true;
- } else if (functionName.startsWith("ttbc")) {
- wikiTokenizer.nextLine();
- // TODO: would be great to handle ttbc
- // TODO: Check this: done = true;
- } else {
- LOG.warning("Unexpected translation wikifunction: " + wikiTokenizer.token() + ", title=" + title);
- }
- } else if (wikiTokenizer.isListItem()) {
- final String line = wikiTokenizer.listItemWikiText();
- // This line could produce an output...
-
-// if (line.contains("ich hoan dich gear")) {
-// //System.out.println();
-// }
-
- // First strip the language and check whether it matches.
- // And hold onto it for sub-lines.
- final int colonIndex = line.indexOf(":");
- if (colonIndex == -1) {
- continue;
- }
-
- final String lang = trim(WikiTokenizer.toPlainText(line.substring(0, colonIndex)));
- if (!langNameToTCount.containsKey(lang)) {
- langNameToTCount.put(lang, new AtomicInteger());
- }
- langNameToTCount.get(lang).incrementAndGet();
- final boolean appendLang;
- if (wikiTokenizer.listItemPrefix().length() == 1) {
- topLevelLang = lang;
- final boolean thisFind = langPattern.matcher(lang).find();
- if (!thisFind) {
- continue;
- }
- appendLang = !langPattern.matcher(lang).matches();
- } else if (topLevelLang == null) {
- continue;
- } else {
- // Two-level -- the only way we won't append is if this second level matches exactly.
- if (!langPattern.matcher(lang).matches() && !langPattern.matcher(topLevelLang).find()) {
- continue;
- }
- appendLang = !langPattern.matcher(lang).matches();
- }
-
- String rest = line.substring(colonIndex + 1).trim();
- if (rest.length() > 0) {
- doTranslationLine(line, appendLang ? lang : null, pos, sense, rest);
- }
-
- } else if (wikiTokenizer.remainderStartsWith("''See''")) {
- wikiTokenizer.nextLine();
- LOG.fine("Skipping See line: " + wikiTokenizer.token());
- } else if (wikiTokenizer.isWikiLink()) {
- final String wikiLink = wikiTokenizer.wikiLinkText();
- if (wikiLink.contains(":") && wikiLink.contains(title)) {
- } else if (wikiLink.contains("Category:")) {
- } else {
- LOG.warning("Unexpected wikiLink: " + wikiTokenizer.token() + ", title=" + title);
- }
- } else if (wikiTokenizer.isNewline() || wikiTokenizer.isMarkup() || wikiTokenizer.isComment()) {
- } else {
- final String token = wikiTokenizer.token();
- if (token.equals("----")) {
- } else {
- LOG.warning("Unexpected translation token: " + wikiTokenizer.token() + ", title=" + title);
- }
- }
-
- }
- }
-
- private void doTranslationLine(final String line, final String lang, final String pos, final String sense, final String rest) {
- state = State.TRANSLATION_LINE;
- // Good chance we'll actually file this one...
- final PairEntry pairEntry = new PairEntry(entrySource);
- final IndexedEntry indexedEntry = new IndexedEntry(pairEntry);
-
- final StringBuilder foreignText = new StringBuilder();
- appendAndIndexWikiCallback.reset(foreignText, indexedEntry);
- appendAndIndexWikiCallback.dispatch(rest, foreignIndexBuilder, EntryTypeName.WIKTIONARY_TRANSLATION_OTHER_TEXT);
-
- if (foreignText.length() == 0) {
- LOG.warning("Empty foreignText: " + line);
- return;
- }
-
- if (lang != null) {
- foreignText.insert(0, String.format("(%s) ", lang));
- }
-
- StringBuilder englishText = new StringBuilder();
-
- englishText.append(title);
- if (sense != null) {
- englishText.append(" (").append(sense).append(")");
- enIndexBuilder.addEntryWithString(indexedEntry, sense, EntryTypeName.WIKTIONARY_TRANSLATION_SENSE);
- }
- if (pos != null) {
- englishText.append(" (").append(pos.toLowerCase()).append(")");
- }
- enIndexBuilder.addEntryWithString(indexedEntry, title, EntryTypeName.WIKTIONARY_TITLE_MULTI);
-
- final Pair pair = new Pair(trim(englishText.toString()), trim(foreignText.toString()), swap);
- pairEntry.pairs.add(pair);
- if (!pairsAdded.add(pair.toString())) {
- LOG.warning("Duplicate pair: " + pair.toString());
- }
- }
-
-
- Set<String> pairsAdded = new LinkedHashSet<String>();
-
- // -------------------------------------------------------------------------
-
- private void doForeignWord(final String lang, final String text) {
- final WikiTokenizer wikiTokenizer = new WikiTokenizer(text);
- while (wikiTokenizer.nextToken() != null) {
- if (wikiTokenizer.isHeading()) {
- final String headingName = wikiTokenizer.headingWikiText();
- if (headingName.equals("Translations")) {
- LOG.warning("Translations not in English section: " + title);
- } else if (headingName.equals("Pronunciation")) {
- //doPronunciation(wikiLineReader);
- } else if (partOfSpeechHeader.matcher(headingName).matches()) {
- doForeignPartOfSpeech(lang, headingName, wikiTokenizer.headingDepth(), wikiTokenizer);
- }
- } else {
- }
- }
- }
-
- static final class ListSection {
- final String firstPrefix;
- final String firstLine;
- final List<String> nextPrefixes = new ArrayList<String>();
- final List<String> nextLines = new ArrayList<String>();
-
- public ListSection(String firstPrefix, String firstLine) {
- this.firstPrefix = firstPrefix;
- this.firstLine = firstLine;
- }
-
- @Override
- public String toString() {
- return firstPrefix + firstLine + "{ " + nextPrefixes + "}";
- }
- }
-
-
- int foreignCount = 0;
- final Collection<String> wordForms = new ArrayList<String>();
- boolean titleAppended = false;
-
- private void doForeignPartOfSpeech(final String lang, String posHeading, final int posDepth, WikiTokenizer wikiTokenizer) {
- if (++foreignCount % 1000 == 0) {
- LOG.info("***" + lang + ", " + title + ", pos=" + posHeading + ", foreignCount=" + foreignCount);
- }
- if (title.equals("6")) {
- System.out.println();
- }
-
- final StringBuilder foreignBuilder = new StringBuilder();
- final List<ListSection> listSections = new ArrayList<ListSection>();
-
- appendAndIndexWikiCallback.reset(foreignBuilder, null);
- this.state = State.ENGLISH_DEF_OF_FOREIGN; // TODO: this is wrong, need new category....
- titleAppended = false;
- wordForms.clear();
-
- try {
-
- ListSection lastListSection = null;
-
- int currentHeadingDepth = posDepth;
- while (wikiTokenizer.nextToken() != null) {
- if (wikiTokenizer.isHeading()) {
- currentHeadingDepth = wikiTokenizer.headingDepth();
-
- if (currentHeadingDepth <= posDepth) {
- wikiTokenizer.returnToLineStart();
- return;
- }
- }
-
- if (currentHeadingDepth > posDepth) {
- // TODO: deal with other neat info sections
- continue;
- }
-
- if (wikiTokenizer.isFunction()) {
- final String name = wikiTokenizer.functionName();
- final List<String> args = wikiTokenizer.functionPositionArgs();
- final Map<String,String> namedArgs = wikiTokenizer.functionNamedArgs();
- // First line is generally a repeat of the title with some extra information.
- // We need to build up the left side (foreign text, tokens) separately from the
- // right side (English). The left-side may get paired with multiple right sides.
- // The left side should get filed under every form of the word in question (singular, plural).
-
- // For verbs, the conjugation comes later on in a deeper section.
- // Ideally, we'd want to file every English entry with the verb
- // under every verb form coming from the conjugation.
- // Ie. under "fa": see: "make :: fare" and "do :: fare"
- // But then where should we put the conjugation table?
- // I think just under fare. But then we need a way to link to the entry (actually the row, since entries doesn't show up!)
- // for the conjugation table from "fa".
- // Would like to be able to link to a lang#token.
-
- appendAndIndexWikiCallback.onFunction(wikiTokenizer, name, args, namedArgs);
-
- } else if (wikiTokenizer.isListItem()) {
- final String prefix = wikiTokenizer.listItemPrefix();
- if (lastListSection != null &&
- prefix.startsWith(lastListSection.firstPrefix) &&
- prefix.length() > lastListSection.firstPrefix.length()) {
- lastListSection.nextPrefixes.add(prefix);
- lastListSection.nextLines.add(wikiTokenizer.listItemWikiText());
- } else {
- lastListSection = new ListSection(prefix, wikiTokenizer.listItemWikiText());
- listSections.add(lastListSection);
- }
- } else if (lastListSection != null) {
- // Don't append anything after the lists, because there's crap.
- } else if (wikiTokenizer.isWikiLink()) {
- // Unindexed!
- foreignBuilder.append(wikiTokenizer.wikiLinkText());
-
- } else if (wikiTokenizer.isPlainText()) {
- // Unindexed!
- foreignBuilder.append(wikiTokenizer.token());
-
- } else if (wikiTokenizer.isMarkup() || wikiTokenizer.isNewline() || wikiTokenizer.isComment()) {
- // Do nothing.
- } else {
- LOG.warning("Unexpected token: " + wikiTokenizer.token());
- }
- }
-
- } finally {
- // Here's where we exit.
- // Should we make an entry even if there are no foreign list items?
- String foreign = foreignBuilder.toString().trim();
- if (!titleAppended && !foreign.toLowerCase().startsWith(title.toLowerCase())) {
- foreign = String.format("%s %s", title, foreign);
- }
- if (!langPattern.matcher(lang).matches()) {
- foreign = String.format("(%s) %s", lang, foreign);
- }
- for (final ListSection listSection : listSections) {
- doForeignListSection(foreign, title, wordForms, listSection);
- }
- }
- }
-
-
- // Might only want to remove "lang" if it's equal to "zh", for example.
- static final Set<String> USELESS_WIKI_ARGS = new LinkedHashSet<String>(
- Arrays.asList(
- "lang",
- "sc",
- "sort",
- "cat",
- "xs",
- "nodot"));
-
- public boolean entryIsFormOfSomething = false;
-
- private void doForeignListSection(final String foreignText, String title, final Collection<String> forms, final ListSection listSection) {
- state = State.ENGLISH_DEF_OF_FOREIGN;
- final String prefix = listSection.firstPrefix;
- if (prefix.length() > 1) {
- // Could just get looser and say that any prefix longer than first is a sublist.
- LOG.warning("Prefix too long: " + listSection);
- return;
- }
-
- final PairEntry pairEntry = new PairEntry(entrySource);
- final IndexedEntry indexedEntry = new IndexedEntry(pairEntry);
-
- entryIsFormOfSomething = false;
- final StringBuilder englishBuilder = new StringBuilder();
- final String mainLine = listSection.firstLine;
- appendAndIndexWikiCallback.reset(englishBuilder, indexedEntry);
- appendAndIndexWikiCallback.dispatch(mainLine, enIndexBuilder, EntryTypeName.WIKTIONARY_ENGLISH_DEF);
-
- final String english = trim(englishBuilder.toString());
- if (english.length() > 0) {
- final Pair pair = new Pair(english, trim(foreignText), this.swap);
- pairEntry.pairs.add(pair);
- foreignIndexBuilder.addEntryWithString(indexedEntry, title, entryIsFormOfSomething ? EntryTypeName.WIKTIONARY_IS_FORM_OF_SOMETHING_ELSE : EntryTypeName.WIKTIONARY_TITLE_MULTI);
- for (final String form : forms) {
- foreignIndexBuilder.addEntryWithString(indexedEntry, form, EntryTypeName.WIKTIONARY_INFLECTED_FORM_MULTI);
- }
- }
-
- // Do examples.
- String lastForeign = null;
- for (int i = 0; i < listSection.nextPrefixes.size(); ++i) {
- final String nextPrefix = listSection.nextPrefixes.get(i);
- final String nextLine = listSection.nextLines.get(i);
-
- // TODO: This splitting is not sensitive to wiki code.
- int dash = nextLine.indexOf("—");
- int mdashLen = 7;
- if (dash == -1) {
- dash = nextLine.indexOf("—");
- mdashLen = 1;
- }
- if (dash == -1) {
- dash = nextLine.indexOf(" - ");
- mdashLen = 3;
- }
-
- if ((nextPrefix.equals("#:") || nextPrefix.equals("##:")) && dash != -1) {
- final String foreignEx = nextLine.substring(0, dash);
- final String englishEx = nextLine.substring(dash + mdashLen);
- final Pair pair = new Pair(formatAndIndexExampleString(englishEx, enIndexBuilder, indexedEntry), formatAndIndexExampleString(foreignEx, foreignIndexBuilder, indexedEntry), swap);
- if (pair.lang1 != "--" && pair.lang1 != "--") {
- pairEntry.pairs.add(pair);
- }
- lastForeign = null;
- } else if (nextPrefix.equals("#:") || nextPrefix.equals("##:")){
- final Pair pair = new Pair("--", formatAndIndexExampleString(nextLine, null, indexedEntry), swap);
- lastForeign = nextLine;
- if (pair.lang1 != "--" && pair.lang1 != "--") {
- pairEntry.pairs.add(pair);
- }
- } else if (nextPrefix.equals("#::") || nextPrefix.equals("#**")) {
- if (lastForeign != null && pairEntry.pairs.size() > 0) {
- pairEntry.pairs.remove(pairEntry.pairs.size() - 1);
- final Pair pair = new Pair(formatAndIndexExampleString(nextLine, enIndexBuilder, indexedEntry), formatAndIndexExampleString(lastForeign, foreignIndexBuilder, indexedEntry), swap);
- if (pair.lang1 != "--" || pair.lang2 != "--") {
- pairEntry.pairs.add(pair);
- }
- lastForeign = null;
- } else {
- LOG.warning("TODO: English example with no foreign: " + title + ", " + nextLine);
- final Pair pair = new Pair("--", formatAndIndexExampleString(nextLine, null, indexedEntry), swap);
- if (pair.lang1 != "--" || pair.lang2 != "--") {
- pairEntry.pairs.add(pair);
- }
- }
- } else if (nextPrefix.equals("#*")) {
- // Can't really index these.
- final Pair pair = new Pair("--", formatAndIndexExampleString(nextLine, null, indexedEntry), swap);
- lastForeign = nextLine;
- if (pair.lang1 != "--" || pair.lang2 != "--") {
- pairEntry.pairs.add(pair);
- }
- } else if (nextPrefix.equals("#::*") || nextPrefix.equals("##") || nextPrefix.equals("#*:") || nextPrefix.equals("#:*") || true) {
- final Pair pair = new Pair("--", formatAndIndexExampleString(nextLine, null, indexedEntry), swap);
- if (pair.lang1 != "--" || pair.lang2 != "--") {
- pairEntry.pairs.add(pair);
- }
-// } else {
-// assert false;
- }
- }
- }
-
- private String formatAndIndexExampleString(final String example, final IndexBuilder indexBuilder, final IndexedEntry indexedEntry) {
- // TODO:
-// if (wikiTokenizer.token().equals("'''")) {
-// insideTripleQuotes = !insideTripleQuotes;
-// }
- final StringBuilder builder = new StringBuilder();
- appendAndIndexWikiCallback.reset(builder, indexedEntry);
- appendAndIndexWikiCallback.entryTypeName = EntryTypeName.WIKTIONARY_EXAMPLE;
- appendAndIndexWikiCallback.entryTypeNameSticks = true;
- try {
- // TODO: this is a hack needed because we don't safely split on the dash.
- appendAndIndexWikiCallback.dispatch(example, indexBuilder, EntryTypeName.WIKTIONARY_EXAMPLE);
- } catch (AssertionError e) {
- return "--";
- }
- final String result = trim(builder.toString());
- return result.length() > 0 ? result : "--";
- }
-
-
- private void itConjAre(List<String> args, Map<String, String> namedArgs) {
- final String base = args.get(0);
- final String aux = args.get(1);
-
- putIfMissing(namedArgs, "inf", base + "are");
- putIfMissing(namedArgs, "aux", aux);
- putIfMissing(namedArgs, "ger", base + "ando");
- putIfMissing(namedArgs, "presp", base + "ante");
- putIfMissing(namedArgs, "pastp", base + "ato");
- // Present
- putIfMissing(namedArgs, "pres1s", base + "o");
- putIfMissing(namedArgs, "pres2s", base + "i");
- putIfMissing(namedArgs, "pres3s", base + "a");
- putIfMissing(namedArgs, "pres1p", base + "iamo");
- putIfMissing(namedArgs, "pres2p", base + "ate");
- putIfMissing(namedArgs, "pres3p", base + "ano");
- // Imperfect
- putIfMissing(namedArgs, "imperf1s", base + "avo");
- putIfMissing(namedArgs, "imperf2s", base + "avi");
- putIfMissing(namedArgs, "imperf3s", base + "ava");
- putIfMissing(namedArgs, "imperf1p", base + "avamo");
- putIfMissing(namedArgs, "imperf2p", base + "avate");
- putIfMissing(namedArgs, "imperf3p", base + "avano");
- // Passato remoto
- putIfMissing(namedArgs, "prem1s", base + "ai");
- putIfMissing(namedArgs, "prem2s", base + "asti");
- putIfMissing(namedArgs, "prem3s", base + "ò");
- putIfMissing(namedArgs, "prem1p", base + "ammo");
- putIfMissing(namedArgs, "prem2p", base + "aste");
- putIfMissing(namedArgs, "prem3p", base + "arono");
- // Future
- putIfMissing(namedArgs, "fut1s", base + "erò");
- putIfMissing(namedArgs, "fut2s", base + "erai");
- putIfMissing(namedArgs, "fut3s", base + "erà");
- putIfMissing(namedArgs, "fut1p", base + "eremo");
- putIfMissing(namedArgs, "fut2p", base + "erete");
- putIfMissing(namedArgs, "fut3p", base + "eranno");
- // Conditional
- putIfMissing(namedArgs, "cond1s", base + "erei");
- putIfMissing(namedArgs, "cond2s", base + "eresti");
- putIfMissing(namedArgs, "cond3s", base + "erebbe");
- putIfMissing(namedArgs, "cond1p", base + "eremmo");
- putIfMissing(namedArgs, "cond2p", base + "ereste");
- putIfMissing(namedArgs, "cond3p", base + "erebbero");
- // Subjunctive / congiuntivo
- putIfMissing(namedArgs, "sub123s", base + "i");
- putIfMissing(namedArgs, "sub1p", base + "iamo");
- putIfMissing(namedArgs, "sub2p", base + "iate");
- putIfMissing(namedArgs, "sub3p", base + "ino");
- // Imperfect subjunctive
- putIfMissing(namedArgs, "impsub12s", base + "assi");
- putIfMissing(namedArgs, "impsub3s", base + "asse");
- putIfMissing(namedArgs, "impsub1p", base + "assimo");
- putIfMissing(namedArgs, "impsub2p", base + "aste");
- putIfMissing(namedArgs, "impsub3p", base + "assero");
- // Imperative
- putIfMissing(namedArgs, "imp2s", base + "a");
- putIfMissing(namedArgs, "imp3s", base + "i");
- putIfMissing(namedArgs, "imp1p", base + "iamo");
- putIfMissing(namedArgs, "imp2p", base + "ate");
- putIfMissing(namedArgs, "imp3p", base + "ino");
-
-
- itConj(args, namedArgs);
- }
-
-
- private void itConj(List<String> args, Map<String, String> namedArgs) {
- // TODO Auto-generated method stub
-
- }
-
-
- private static void putIfMissing(final Map<String, String> namedArgs, final String key,
- final String value) {
- final String oldValue = namedArgs.get(key);
- if (oldValue == null || oldValue.length() == 0) {
- namedArgs.put(key, value);
- }
- }
-
- // TODO: check how ='' and =| are manifested....
- // TODO: get this right in -are
- private static void putOrNullify(final Map<String, String> namedArgs, final String key,
- final String value) {
- final String oldValue = namedArgs.get(key);
- if (oldValue == null/* || oldValue.length() == 0*/) {
- namedArgs.put(key, value);
- } else {
- if (oldValue.equals("''")) {
- namedArgs.put(key, "");
- }
- }
- }
-
- static final Pattern whitespace = Pattern.compile("\\s+");
- static String trim(final String s) {
- return whitespace.matcher(s).replaceAll(" ").trim();
- }
-
-
-}
import java.util.Map;
import com.hughes.android.dictionary.parser.WikiTokenizer;
+import com.hughes.android.dictionary.parser.wiktionary.AbstractWiktionaryParser.AppendAndIndexWikiCallback;
-public interface FunctionCallback {
+public interface FunctionCallback<T extends AbstractWiktionaryParser> {
boolean onWikiFunction(
final WikiTokenizer tokenizer,
final String name,
final List<String> args,
final Map<String,String> namedArgs,
- final EnWiktionaryXmlParser parser,
- final AppendAndIndexWikiCallback appendAndIndexWikiCallback);
+ final T parser,
+ final AppendAndIndexWikiCallback<T> appendAndIndexWikiCallback);
}
+++ /dev/null
-// 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.wiktionary;
-
-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;
-
-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);
- DEFAULT.put("t+", callback);
- DEFAULT.put("t-", callback);
- DEFAULT.put("tø", callback);
- DEFAULT.put("apdx-t", callback);
-
- callback = new EncodingCallback();
- Set<String> encodings = new LinkedHashSet<String>(Arrays.asList(
- "zh-ts", "zh-tsp",
- "sd-Arab", "ku-Arab", "Arab", "unicode", "Laoo", "ur-Arab", "Thai",
- "fa-Arab", "Khmr", "Cyrl", "IPAchar", "ug-Arab", "ko-inline",
- "Jpan", "Kore", "Hebr", "rfscript", "Beng", "Mong", "Knda", "Cyrs",
- "yue-tsj", "Mlym", "Tfng", "Grek", "yue-yue-j"));
- for (final String encoding : encodings) {
- 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);
-
- callback = new AppendArg0();
-
- 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);
-
-
- 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());
- }
-
-
- static final class NameAndArgs 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 (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);
- }
- }
- 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 {
- @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) {
-
- 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);
- 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(" (");
- 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;
- }
-
- }
-
- // ------------------------------------------------------------------
-
- static final class QualifierCallback 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.size() != 1 || !namedArgs.isEmpty()) {
- LOG.warning("weird qualifier: ");
- return false;
- }
- String qualifier = args.get(0);
- appendAndIndexWikiCallback.builder.append("(");
- appendAndIndexWikiCallback.dispatch(qualifier, null);
- appendAndIndexWikiCallback.builder.append(")");
- return true;
- }
- }
-
- // ------------------------------------------------------------------
-
- static final class EncodingCallback 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 (!namedArgs.isEmpty()) {
- LOG.warning("weird encoding: " + wikiTokenizer.token());
- }
- 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;
- }
- }
-
- // ------------------------------------------------------------------
-
- static final class Gender 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 (!namedArgs.isEmpty()) {
- return false;
- }
- appendAndIndexWikiCallback.builder.append("{");
- appendAndIndexWikiCallback.builder.append(name);
- for (int i = 0; i < args.size(); ++i) {
- appendAndIndexWikiCallback.builder.append("|").append(args.get(i));
- }
- appendAndIndexWikiCallback.builder.append("}");
- return true;
- }
- }
-
- // ------------------------------------------------------------------
-
- 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) {
-
- // 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);
- final IndexBuilder indexBuilder;
- if ("".equals(langCode)) {
- indexBuilder = parser.foreignIndexBuilder;
- } else if ("en".equals(langCode)) {
- indexBuilder = parser.enIndexBuilder;
- } else {
- indexBuilder = parser.foreignIndexBuilder;
- }
-
- 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;
- }
- }
-
- // ------------------------------------------------------------------
-
- static final class AppendArg0 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.size() != 1 || !namedArgs.isEmpty()) {
- return false;
- }
- appendAndIndexWikiCallback.dispatch(args.get(0), EntryTypeName.WIKTIONARY_TRANSLATION_OTHER_TEXT);
- // TODO: transliteration
- return true;
- }
- }
-
- // ------------------------------------------------------------------
-
- static final class italbrac 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.size() != 1 || !namedArgs.isEmpty()) {
- return false;
- }
- appendAndIndexWikiCallback.builder.append("(");
- appendAndIndexWikiCallback.dispatch(args.get(0), EntryTypeName.WIKTIONARY_TRANSLATION_OTHER_TEXT);
- appendAndIndexWikiCallback.builder.append(")");
- return true;
- }
- }
-
- // ------------------------------------------------------------------
-
- static final class gloss 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.size() != 1 || !namedArgs.isEmpty()) {
- return false;
- }
- appendAndIndexWikiCallback.builder.append("(");
- appendAndIndexWikiCallback.dispatch(args.get(0), EntryTypeName.WIKTIONARY_TRANSLATION_OTHER_TEXT);
- appendAndIndexWikiCallback.builder.append(")");
- return true;
- }
- }
-
- // ------------------------------------------------------------------
-
- static final class Ignore 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) {
- return true;
- }
- }
-
- // ------------------------------------------------------------------
-
- static final class not_used 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) {
- appendAndIndexWikiCallback.builder.append("(not used)");
- return true;
- }
- }
-
-
- // ------------------------------------------------------------------
-
- 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;
- }
- }
-
- // --------------------------------------------------------------------
- // --------------------------------------------------------------------
-
-
- static final class FormOf 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) {
- parser.entryIsFormOfSomething = true;
- String formName = name;
- if (name.equals("form of")) {
- formName = ListUtil.remove(args, 0, null);
- }
- if (formName == null) {
- LOG.warning("Missing form name: " + parser.title);
- formName = "form of";
- }
- String baseForm = ListUtil.get(args, 1, "");
- if ("".equals(baseForm)) {
- baseForm = ListUtil.get(args, 0, null);
- ListUtil.remove(args, 1, "");
- } else {
- ListUtil.remove(args, 0, null);
- }
- namedArgs.keySet().removeAll(EnWiktionaryXmlParser.USELESS_WIKI_ARGS);
-
- appendAndIndexWikiCallback.builder.append("{");
- NAME_AND_ARGS.onWikiFunction(wikiTokenizer, formName, args, namedArgs, parser, appendAndIndexWikiCallback);
- appendAndIndexWikiCallback.builder.append("}");
- if (baseForm != null && appendAndIndexWikiCallback.indexedEntry != null) {
- parser.foreignIndexBuilder.addEntryWithString(appendAndIndexWikiCallback.indexedEntry, baseForm, EntryTypeName.WIKTIONARY_BASE_FORM_MULTI);
- } else {
- // null baseForm happens in Danish.
- LOG.warning("Null baseform: " + parser.title);
- }
- return true;
- }
- }
-
- static final FormOf FORM_OF = new FormOf();
-
-
- // --------------------------------------------------------------------
- // --------------------------------------------------------------------
-
- static final class wikipedia 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) {
- namedArgs.remove("lang");
- if (args.size() > 1 || !namedArgs.isEmpty()) {
- // Unindexed!
- return false;
- } else if (args.size() == 1) {
- return false;
- } else {
- return true;
- }
- }
- }
-
- static final class InflOrHead 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) {
- // See: http://en.wiktionary.org/wiki/Template:infl
- final String langCode = ListUtil.get(args, 0);
- String head = namedArgs.remove("head");
- if (head == null) {
- head = namedArgs.remove("title"); // Bug
- }
- if (head == null) {
- head = parser.title;
- }
- parser.titleAppended = true;
-
- namedArgs.keySet().removeAll(EnWiktionaryXmlParser.USELESS_WIKI_ARGS);
-
- final String tr = namedArgs.remove("tr");
- String g = namedArgs.remove("g");
- if (g == null) {
- g = namedArgs.remove("gender");
- }
- final String g2 = namedArgs.remove("g2");
- final String g3 = namedArgs.remove("g3");
-
- appendAndIndexWikiCallback.dispatch(head, EntryTypeName.WIKTIONARY_TITLE_MULTI);
-
- if (g != null) {
- appendAndIndexWikiCallback.builder.append(" {").append(g);
- if (g2 != null) {
- appendAndIndexWikiCallback.builder.append("|").append(g2);
- }
- if (g3 != null) {
- appendAndIndexWikiCallback.builder.append("|").append(g3);
- }
- appendAndIndexWikiCallback.builder.append("}");
- }
-
- if (tr != null) {
- appendAndIndexWikiCallback.builder.append(" (");
- appendAndIndexWikiCallback.dispatch(tr, EntryTypeName.WIKTIONARY_TITLE_MULTI);
- appendAndIndexWikiCallback.builder.append(")");
- parser.wordForms.add(tr);
- }
-
- final String pos = ListUtil.get(args, 1);
- if (pos != null) {
- appendAndIndexWikiCallback.builder.append(" (").append(pos).append(")");
- }
- for (int i = 2; i < args.size(); i += 2) {
- final String inflName = ListUtil.get(args, i);
- final String inflValue = ListUtil.get(args, i + 1);
- appendAndIndexWikiCallback.builder.append(", ");
- appendAndIndexWikiCallback.dispatch(inflName, null, null);
- if (inflValue != null && inflValue.length() > 0) {
- appendAndIndexWikiCallback.builder.append(": ");
- appendAndIndexWikiCallback.dispatch(inflValue, null, null);
- parser.wordForms.add(inflValue);
- }
- }
- for (final String key : namedArgs.keySet()) {
- final String value = WikiTokenizer.toPlainText(namedArgs.get(key));
- appendAndIndexWikiCallback.builder.append(" ");
- appendAndIndexWikiCallback.dispatch(key, null, null);
- appendAndIndexWikiCallback.builder.append("=");
- appendAndIndexWikiCallback.dispatch(value, null, null);
- parser.wordForms.add(value);
- }
- return true;
- }
- }
-
-
- static {
- DEFAULT.put("it-noun", new it_noun());
- }
- static final class it_noun 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) {
- parser.titleAppended = true;
- final String base = ListUtil.get(args, 0);
- final String gender = ListUtil.get(args, 1);
- final String singular = base + ListUtil.get(args, 2, null);
- final String plural = base + ListUtil.get(args, 3, null);
- appendAndIndexWikiCallback.builder.append(" ");
- appendAndIndexWikiCallback.dispatch(singular, null, null);
- appendAndIndexWikiCallback.builder.append(" {").append(gender).append("}, ");
- appendAndIndexWikiCallback.dispatch(plural, null, null);
- appendAndIndexWikiCallback.builder.append(" {pl}");
- parser.wordForms.add(singular);
- parser.wordForms.add(plural);
- if (!namedArgs.isEmpty() || args.size() > 4) {
- LOG.warning("Invalid it-noun: " + wikiTokenizer.token());
- }
- return true;
- }
- }
-
- static {
- DEFAULT.put("it-proper noun", new it_proper_noun());
- }
- static final class it_proper_noun 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) {
- return false;
- }
- }
-
-}
projects / DictionaryPC.git / commitdiff