FusionDictionary.java revision f2789819bd005b5b0581e8439601b5501306327d
1bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard/*
2bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard * Copyright (C) 2011 The Android Open Source Project
3bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard *
4bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard * Licensed under the Apache License, Version 2.0 (the "License"); you may not
5bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard * use this file except in compliance with the License. You may obtain a copy of
6bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard * the License at
7bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard *
8bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard * http://www.apache.org/licenses/LICENSE-2.0
9bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard *
10bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard * Unless required by applicable law or agreed to in writing, software
11bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
12bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
13bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard * License for the specific language governing permissions and limitations under
14bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard * the License.
15bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard */
16bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
17cdc51fc6afc7fd374c5c9eeb0539cae5cf1de724Tom Ouyangpackage com.android.inputmethod.latin.makedict;
18bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
19bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalardimport java.util.ArrayList;
20bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalardimport java.util.Arrays;
21bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalardimport java.util.Collections;
22c734c2aca1830643d169fd292e0c9d4d9306af5aJean Chalardimport java.util.HashMap;
23bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalardimport java.util.Iterator;
24bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalardimport java.util.LinkedList;
25bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
26bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard/**
27bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard * A dictionary that can fusion heads and tails of words for more compression.
28bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard */
29bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalardpublic class FusionDictionary implements Iterable<Word> {
30bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
3147db0be7cbdb8abafc18c1e49b71f6dac0d46994Jean Chalard    private static final boolean DBG = MakedictLog.DBG;
3247db0be7cbdb8abafc18c1e49b71f6dac0d46994Jean Chalard
33bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    /**
34bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * A node of the dictionary, containing several CharGroups.
35bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     *
36bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * A node is but an ordered array of CharGroups, which essentially contain all the
37bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * real information.
38bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * This class also contains fields to cache size and address, to help with binary
39bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * generation.
40bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     */
41bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    public static class Node {
42bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        ArrayList<CharGroup> mData;
43bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        // To help with binary generation
44bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        int mCachedSize;
45bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        int mCachedAddress;
46bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        public Node() {
47bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            mData = new ArrayList<CharGroup>();
48bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            mCachedSize = Integer.MIN_VALUE;
49bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            mCachedAddress = Integer.MIN_VALUE;
50bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        }
51bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        public Node(ArrayList<CharGroup> data) {
52bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            mData = data;
53bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            mCachedSize = Integer.MIN_VALUE;
54bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            mCachedAddress = Integer.MIN_VALUE;
55bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        }
56bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    }
57bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
58bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    /**
59bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * A string with a frequency.
60bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     *
61bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * This represents an "attribute", that is either a bigram or a shortcut.
62bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     */
63bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    public static class WeightedString {
6454e84a00fc032ba566cbda41feafa71de77e1c43Jean Chalard        public final String mWord;
6554e84a00fc032ba566cbda41feafa71de77e1c43Jean Chalard        public int mFrequency;
66bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        public WeightedString(String word, int frequency) {
67bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            mWord = word;
68bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            mFrequency = frequency;
69bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        }
709f0ea52a5db9710df6bef4672d8e193c48451df0Ken Wakasa
719f0ea52a5db9710df6bef4672d8e193c48451df0Ken Wakasa        @Override
729f0ea52a5db9710df6bef4672d8e193c48451df0Ken Wakasa        public int hashCode() {
739f0ea52a5db9710df6bef4672d8e193c48451df0Ken Wakasa            return Arrays.hashCode(new Object[] { mWord, mFrequency });
749f0ea52a5db9710df6bef4672d8e193c48451df0Ken Wakasa        }
759f0ea52a5db9710df6bef4672d8e193c48451df0Ken Wakasa
769f0ea52a5db9710df6bef4672d8e193c48451df0Ken Wakasa        @Override
779f0ea52a5db9710df6bef4672d8e193c48451df0Ken Wakasa        public boolean equals(Object o) {
789f0ea52a5db9710df6bef4672d8e193c48451df0Ken Wakasa            if (o == this) return true;
799f0ea52a5db9710df6bef4672d8e193c48451df0Ken Wakasa            if (!(o instanceof WeightedString)) return false;
809f0ea52a5db9710df6bef4672d8e193c48451df0Ken Wakasa            WeightedString w = (WeightedString)o;
819f0ea52a5db9710df6bef4672d8e193c48451df0Ken Wakasa            return mWord.equals(w.mWord) && mFrequency == w.mFrequency;
829f0ea52a5db9710df6bef4672d8e193c48451df0Ken Wakasa        }
83bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    }
84bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
85bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    /**
86eec2e51e2cbc9e69739187557846a439ed74325eJean Chalard     * A group of characters, with a frequency, shortcut targets, bigrams, and children.
87bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     *
88bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * This is the central class of the in-memory representation. A CharGroup is what can
89bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * be seen as a traditional "trie node", except it can hold several characters at the
90bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * same time. A CharGroup essentially represents one or several characters in the middle
91bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * of the trie trie; as such, it can be a terminal, and it can have children.
92bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * In this in-memory representation, whether the CharGroup is a terminal or not is represented
93bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * in the frequency, where NOT_A_TERMINAL (= -1) means this is not a terminal and any other
94bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * value is the frequency of this terminal. A terminal may have non-null shortcuts and/or
95bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * bigrams, but a non-terminal may not. Moreover, children, if present, are null.
96bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     */
97bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    public static class CharGroup {
98bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        public static final int NOT_A_TERMINAL = -1;
99bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        final int mChars[];
1007cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang        ArrayList<WeightedString> mShortcutTargets;
1017cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang        ArrayList<WeightedString> mBigrams;
1027cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang        int mFrequency; // NOT_A_TERMINAL == mFrequency indicates this is not a terminal.
103bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        Node mChildren;
10472b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard        boolean mIsNotAWord; // Only a shortcut
10572b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard        boolean mIsBlacklistEntry;
106bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        // The two following members to help with binary generation
107bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        int mCachedSize;
108bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        int mCachedAddress;
109bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
110eec2e51e2cbc9e69739187557846a439ed74325eJean Chalard        public CharGroup(final int[] chars, final ArrayList<WeightedString> shortcutTargets,
11172b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                final ArrayList<WeightedString> bigrams, final int frequency,
11272b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                final boolean isNotAWord, final boolean isBlacklistEntry) {
113bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            mChars = chars;
114bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            mFrequency = frequency;
115eec2e51e2cbc9e69739187557846a439ed74325eJean Chalard            mShortcutTargets = shortcutTargets;
116bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            mBigrams = bigrams;
117bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            mChildren = null;
11872b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard            mIsNotAWord = isNotAWord;
11972b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard            mIsBlacklistEntry = isBlacklistEntry;
120bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        }
121bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
122eec2e51e2cbc9e69739187557846a439ed74325eJean Chalard        public CharGroup(final int[] chars, final ArrayList<WeightedString> shortcutTargets,
12372b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                final ArrayList<WeightedString> bigrams, final int frequency,
12472b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                final boolean isNotAWord, final boolean isBlacklistEntry, final Node children) {
125bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            mChars = chars;
126bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            mFrequency = frequency;
127eec2e51e2cbc9e69739187557846a439ed74325eJean Chalard            mShortcutTargets = shortcutTargets;
128bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            mBigrams = bigrams;
129bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            mChildren = children;
13072b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard            mIsNotAWord = isNotAWord;
13172b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard            mIsBlacklistEntry = isBlacklistEntry;
132bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        }
133bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
134bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        public void addChild(CharGroup n) {
135bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            if (null == mChildren) {
136bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                mChildren = new Node();
137bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            }
138bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            mChildren.mData.add(n);
139bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        }
140bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
141bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        public boolean isTerminal() {
142bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            return NOT_A_TERMINAL != mFrequency;
143bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        }
144bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
145bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        public boolean hasSeveralChars() {
146bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            assert(mChars.length > 0);
147bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            return 1 < mChars.length;
148bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        }
1497cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang
1507cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang        /**
1517cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang         * Adds a word to the bigram list. Updates the frequency if the word already
1527cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang         * exists.
1537cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang         */
1547cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang        public void addBigram(final String word, final int frequency) {
1557cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang            if (mBigrams == null) {
1567cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                mBigrams = new ArrayList<WeightedString>();
1577cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang            }
1587cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang            WeightedString bigram = getBigram(word);
1597cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang            if (bigram != null) {
1607cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                bigram.mFrequency = frequency;
1617cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang            } else {
1627cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                bigram = new WeightedString(word, frequency);
1637cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                mBigrams.add(bigram);
1647cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang            }
1657cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang        }
1667cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang
1677cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang        /**
1687cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang         * Gets the shortcut target for the given word. Returns null if the word is not in the
1697cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang         * shortcut list.
1707cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang         */
1717cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang        public WeightedString getShortcut(final String word) {
17247db0be7cbdb8abafc18c1e49b71f6dac0d46994Jean Chalard            // TODO: Don't do a linear search
1737cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang            if (mShortcutTargets != null) {
1747cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                final int size = mShortcutTargets.size();
1757cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                for (int i = 0; i < size; ++i) {
1767cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                    WeightedString shortcut = mShortcutTargets.get(i);
1777cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                    if (shortcut.mWord.equals(word)) {
1787cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                        return shortcut;
1797cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                    }
1807cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                }
1817cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang            }
1827cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang            return null;
1837cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang        }
1847cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang
1857cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang        /**
1867cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang         * Gets the bigram for the given word.
1877cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang         * Returns null if the word is not in the bigrams list.
1887cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang         */
1897cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang        public WeightedString getBigram(final String word) {
19047db0be7cbdb8abafc18c1e49b71f6dac0d46994Jean Chalard            // TODO: Don't do a linear search
1917cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang            if (mBigrams != null) {
1927cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                final int size = mBigrams.size();
1937cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                for (int i = 0; i < size; ++i) {
1947cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                    WeightedString bigram = mBigrams.get(i);
1957cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                    if (bigram.mWord.equals(word)) {
1967cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                        return bigram;
1977cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                    }
1987cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                }
1997cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang            }
2007cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang            return null;
2017cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang        }
2027cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang
2037cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang        /**
2047cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang         * Updates the CharGroup with the given properties. Adds the shortcut and bigram lists to
2057cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang         * the existing ones if any. Note: unigram, bigram, and shortcut frequencies are only
2067cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang         * updated if they are higher than the existing ones.
2077cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang         */
20872b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard        public void update(final int frequency, final ArrayList<WeightedString> shortcutTargets,
20972b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                final ArrayList<WeightedString> bigrams,
21072b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                final boolean isNotAWord, final boolean isBlacklistEntry) {
2117cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang            if (frequency > mFrequency) {
2127cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                mFrequency = frequency;
2137cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang            }
2147cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang            if (shortcutTargets != null) {
2157cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                if (mShortcutTargets == null) {
2167cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                    mShortcutTargets = shortcutTargets;
2177cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                } else {
2187cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                    final int size = shortcutTargets.size();
2197cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                    for (int i = 0; i < size; ++i) {
2207cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                        final WeightedString shortcut = shortcutTargets.get(i);
2217cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                        final WeightedString existingShortcut = getShortcut(shortcut.mWord);
2227cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                        if (existingShortcut == null) {
2237cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                            mShortcutTargets.add(shortcut);
2247cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                        } else if (existingShortcut.mFrequency < shortcut.mFrequency) {
2257cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                            existingShortcut.mFrequency = shortcut.mFrequency;
2267cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                        }
2277cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                    }
2287cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                }
2297cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang            }
2307cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang            if (bigrams != null) {
2317cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                if (mBigrams == null) {
2327cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                    mBigrams = bigrams;
2337cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                } else {
2347cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                    final int size = bigrams.size();
2357cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                    for (int i = 0; i < size; ++i) {
2367cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                        final WeightedString bigram = bigrams.get(i);
2377cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                        final WeightedString existingBigram = getBigram(bigram.mWord);
2387cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                        if (existingBigram == null) {
2397cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                            mBigrams.add(bigram);
2407cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                        } else if (existingBigram.mFrequency < bigram.mFrequency) {
2417cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                            existingBigram.mFrequency = bigram.mFrequency;
2427cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                        }
2437cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                    }
2447cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                }
2457cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang            }
24672b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard            mIsNotAWord = isNotAWord;
24772b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard            mIsBlacklistEntry = isBlacklistEntry;
2487cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang        }
249bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    }
250bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
251bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    /**
252bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * Options global to the dictionary.
253bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     *
254bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * There are no options at the moment, so this class is empty.
255bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     */
256bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    public static class DictionaryOptions {
257f420df28233c26e555d203185fb292e83b94b8c3Jean Chalard        public final boolean mGermanUmlautProcessing;
258f420df28233c26e555d203185fb292e83b94b8c3Jean Chalard        public final boolean mFrenchLigatureProcessing;
259f420df28233c26e555d203185fb292e83b94b8c3Jean Chalard        public final HashMap<String, String> mAttributes;
260f420df28233c26e555d203185fb292e83b94b8c3Jean Chalard        public DictionaryOptions(final HashMap<String, String> attributes,
261f420df28233c26e555d203185fb292e83b94b8c3Jean Chalard                final boolean germanUmlautProcessing, final boolean frenchLigatureProcessing) {
262c734c2aca1830643d169fd292e0c9d4d9306af5aJean Chalard            mAttributes = attributes;
263f420df28233c26e555d203185fb292e83b94b8c3Jean Chalard            mGermanUmlautProcessing = germanUmlautProcessing;
264f420df28233c26e555d203185fb292e83b94b8c3Jean Chalard            mFrenchLigatureProcessing = frenchLigatureProcessing;
265c734c2aca1830643d169fd292e0c9d4d9306af5aJean Chalard        }
266bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    }
267bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
268bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    public final DictionaryOptions mOptions;
269bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    public final Node mRoot;
270bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
271bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    public FusionDictionary(final Node root, final DictionaryOptions options) {
272bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        mRoot = root;
273bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        mOptions = options;
274bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    }
275bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
276c734c2aca1830643d169fd292e0c9d4d9306af5aJean Chalard    public void addOptionAttribute(final String key, final String value) {
277c734c2aca1830643d169fd292e0c9d4d9306af5aJean Chalard        mOptions.mAttributes.put(key, value);
278c734c2aca1830643d169fd292e0c9d4d9306af5aJean Chalard    }
279c734c2aca1830643d169fd292e0c9d4d9306af5aJean Chalard
280bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    /**
281bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * Helper method to convert a String to an int array.
282bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     */
28347db0be7cbdb8abafc18c1e49b71f6dac0d46994Jean Chalard    static private int[] getCodePoints(final String word) {
28447db0be7cbdb8abafc18c1e49b71f6dac0d46994Jean Chalard        // TODO: this is a copy-paste of the contents of StringUtils.toCodePointArray,
28547db0be7cbdb8abafc18c1e49b71f6dac0d46994Jean Chalard        // which is not visible from the makedict package. Factor this code.
28647db0be7cbdb8abafc18c1e49b71f6dac0d46994Jean Chalard        final char[] characters = word.toCharArray();
28747db0be7cbdb8abafc18c1e49b71f6dac0d46994Jean Chalard        final int length = characters.length;
28847db0be7cbdb8abafc18c1e49b71f6dac0d46994Jean Chalard        final int[] codePoints = new int[Character.codePointCount(characters, 0, length)];
28947db0be7cbdb8abafc18c1e49b71f6dac0d46994Jean Chalard        int codePoint = Character.codePointAt(characters, 0);
29047db0be7cbdb8abafc18c1e49b71f6dac0d46994Jean Chalard        int dsti = 0;
29147db0be7cbdb8abafc18c1e49b71f6dac0d46994Jean Chalard        for (int srci = Character.charCount(codePoint);
29247db0be7cbdb8abafc18c1e49b71f6dac0d46994Jean Chalard                srci < length; srci += Character.charCount(codePoint), ++dsti) {
29347db0be7cbdb8abafc18c1e49b71f6dac0d46994Jean Chalard            codePoints[dsti] = codePoint;
29447db0be7cbdb8abafc18c1e49b71f6dac0d46994Jean Chalard            codePoint = Character.codePointAt(characters, srci);
295c599f2e9d6ab839f38183aa178684ff0e94178a3Jean Chalard        }
29647db0be7cbdb8abafc18c1e49b71f6dac0d46994Jean Chalard        codePoints[dsti] = codePoint;
29747db0be7cbdb8abafc18c1e49b71f6dac0d46994Jean Chalard        return codePoints;
298c599f2e9d6ab839f38183aa178684ff0e94178a3Jean Chalard    }
299c599f2e9d6ab839f38183aa178684ff0e94178a3Jean Chalard
300c599f2e9d6ab839f38183aa178684ff0e94178a3Jean Chalard    /**
301bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * Helper method to add a word as a string.
302bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     *
303bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * This method adds a word to the dictionary with the given frequency. Optional
304bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * lists of bigrams and shortcuts can be passed here. For each word inside,
305bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * they will be added to the dictionary as necessary.
306bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     *
307bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * @param word the word to add.
308bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * @param frequency the frequency of the word, in the range [0..255].
309eec2e51e2cbc9e69739187557846a439ed74325eJean Chalard     * @param shortcutTargets a list of shortcut targets for this word, or null.
31072b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard     * @param isNotAWord true if this should not be considered a word (e.g. shortcut only)
311bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     */
312eec2e51e2cbc9e69739187557846a439ed74325eJean Chalard    public void add(final String word, final int frequency,
31372b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard            final ArrayList<WeightedString> shortcutTargets, final boolean isNotAWord) {
31472b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard        add(getCodePoints(word), frequency, shortcutTargets, isNotAWord,
31572b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                false /* isBlacklistEntry */);
31672b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard    }
31772b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard
31872b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard    /**
31972b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard     * Helper method to add a blacklist entry as a string.
32072b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard     *
32172b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard     * @param word the word to add as a blacklist entry.
32272b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard     * @param shortcutTargets a list of shortcut targets for this word, or null.
32372b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard     * @param isNotAWord true if this is not a word for spellcheking purposes (shortcut only or so)
32472b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard     */
32572b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard    public void addBlacklistEntry(final String word,
32672b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard            final ArrayList<WeightedString> shortcutTargets, final boolean isNotAWord) {
32772b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard        add(getCodePoints(word), 0, shortcutTargets, isNotAWord, true /* isBlacklistEntry */);
328bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    }
329bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
330bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    /**
331bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * Sanity check for a node.
332bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     *
333bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * This method checks that all CharGroups in a node are ordered as expected.
334bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * If they are, nothing happens. If they aren't, an exception is thrown.
335bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     */
336bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    private void checkStack(Node node) {
337bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        ArrayList<CharGroup> stack = node.mData;
338bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        int lastValue = -1;
339bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        for (int i = 0; i < stack.size(); ++i) {
340bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            int currentValue = stack.get(i).mChars[0];
341bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            if (currentValue <= lastValue)
342bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                throw new RuntimeException("Invalid stack");
343bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            else
344bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                lastValue = currentValue;
345bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        }
346bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    }
347bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
348bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    /**
3497cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang     * Helper method to add a new bigram to the dictionary.
3507cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang     *
3517cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang     * @param word1 the previous word of the context
3527cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang     * @param word2 the next word of the context
3537cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang     * @param frequency the bigram frequency
3547cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang     */
3557cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang    public void setBigram(final String word1, final String word2, final int frequency) {
3567cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang        CharGroup charGroup = findWordInTree(mRoot, word1);
3577cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang        if (charGroup != null) {
3587cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang            final CharGroup charGroup2 = findWordInTree(mRoot, word2);
3597cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang            if (charGroup2 == null) {
36072b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                add(getCodePoints(word2), 0, null, false /* isNotAWord */,
36172b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                        false /* isBlacklistEntry */);
3627cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang            }
3637cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang            charGroup.addBigram(word2, frequency);
3647cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang        } else {
3657cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang            throw new RuntimeException("First word of bigram not found");
3667cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang        }
3677cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang    }
3687cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang
3697cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang    /**
370bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * Add a word to this dictionary.
371bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     *
37244c64f46a143623dd793facd889c8d6eab5e230cJean Chalard     * The shortcuts, if any, have to be in the dictionary already. If they aren't,
373bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * an exception is thrown.
374bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     *
375bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * @param word the word, as an int array.
376bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * @param frequency the frequency of the word, in the range [0..255].
377eec2e51e2cbc9e69739187557846a439ed74325eJean Chalard     * @param shortcutTargets an optional list of shortcut targets for this word (null if none).
37872b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard     * @param isNotAWord true if this is not a word for spellcheking purposes (shortcut only or so)
37972b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard     * @param isBlacklistEntry true if this is a blacklisted word, false otherwise
380bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     */
381eec2e51e2cbc9e69739187557846a439ed74325eJean Chalard    private void add(final int[] word, final int frequency,
38272b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard            final ArrayList<WeightedString> shortcutTargets,
38372b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard            final boolean isNotAWord, final boolean isBlacklistEntry) {
384bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        assert(frequency >= 0 && frequency <= 255);
385bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        Node currentNode = mRoot;
386bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        int charIndex = 0;
387bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
388bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        CharGroup currentGroup = null;
389bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        int differentCharIndex = 0; // Set by the loop to the index of the char that differs
390bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        int nodeIndex = findIndexOfChar(mRoot, word[charIndex]);
391bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        while (CHARACTER_NOT_FOUND != nodeIndex) {
392bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            currentGroup = currentNode.mData.get(nodeIndex);
393cad25fc8a754d6f145bc846f17f270220b15c055Jean Chalard            differentCharIndex = compareArrays(currentGroup.mChars, word, charIndex);
394bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            if (ARRAYS_ARE_EQUAL != differentCharIndex
395bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    && differentCharIndex < currentGroup.mChars.length) break;
396bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            if (null == currentGroup.mChildren) break;
397bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            charIndex += currentGroup.mChars.length;
398bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            if (charIndex >= word.length) break;
399bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            currentNode = currentGroup.mChildren;
400bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            nodeIndex = findIndexOfChar(currentNode, word[charIndex]);
401bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        }
402bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
403bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        if (-1 == nodeIndex) {
404bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            // No node at this point to accept the word. Create one.
405bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            final int insertionIndex = findInsertionIndex(currentNode, word[charIndex]);
406bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            final CharGroup newGroup = new CharGroup(
407eec2e51e2cbc9e69739187557846a439ed74325eJean Chalard                    Arrays.copyOfRange(word, charIndex, word.length),
40872b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                    shortcutTargets, null /* bigrams */, frequency, isNotAWord, isBlacklistEntry);
409bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            currentNode.mData.add(insertionIndex, newGroup);
41047db0be7cbdb8abafc18c1e49b71f6dac0d46994Jean Chalard            if (DBG) checkStack(currentNode);
411bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        } else {
412bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            // There is a word with a common prefix.
413bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            if (differentCharIndex == currentGroup.mChars.length) {
414bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                if (charIndex + differentCharIndex >= word.length) {
415bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    // The new word is a prefix of an existing word, but the node on which it
4167cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                    // should end already exists as is. Since the old CharNode was not a terminal,
4177cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                    // make it one by filling in its frequency and other attributes
41872b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                    currentGroup.update(frequency, shortcutTargets, null, isNotAWord,
41972b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                            isBlacklistEntry);
420bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                } else {
421bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    // The new word matches the full old word and extends past it.
422bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    // We only have to create a new node and add it to the end of this.
423bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    final CharGroup newNode = new CharGroup(
424bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                            Arrays.copyOfRange(word, charIndex + differentCharIndex, word.length),
42572b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                                    shortcutTargets, null /* bigrams */, frequency, isNotAWord,
42672b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                                    isBlacklistEntry);
427bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    currentGroup.mChildren = new Node();
428bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    currentGroup.mChildren.mData.add(newNode);
429bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                }
430bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            } else {
431bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                if (0 == differentCharIndex) {
4327cfe20efbeb4a94b15291aee95d0559ae2449c45Tom Ouyang                    // Exact same word. Update the frequency if higher. This will also add the
43344c64f46a143623dd793facd889c8d6eab5e230cJean Chalard                    // new shortcuts to the existing shortcut list if it already exists.
43472b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                    currentGroup.update(frequency, shortcutTargets, null,
43572b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                            currentGroup.mIsNotAWord && isNotAWord,
43672b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                            currentGroup.mIsBlacklistEntry || isBlacklistEntry);
437bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                } else {
438bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    // Partial prefix match only. We have to replace the current node with a node
439bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    // containing the current prefix and create two new ones for the tails.
440bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    Node newChildren = new Node();
441bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    final CharGroup newOldWord = new CharGroup(
442bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                            Arrays.copyOfRange(currentGroup.mChars, differentCharIndex,
443eec2e51e2cbc9e69739187557846a439ed74325eJean Chalard                                    currentGroup.mChars.length), currentGroup.mShortcutTargets,
44472b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                            currentGroup.mBigrams, currentGroup.mFrequency,
44572b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                            currentGroup.mIsNotAWord, currentGroup.mIsBlacklistEntry,
44672b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                            currentGroup.mChildren);
447bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    newChildren.mData.add(newOldWord);
448bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
449bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    final CharGroup newParent;
450bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    if (charIndex + differentCharIndex >= word.length) {
451bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                        newParent = new CharGroup(
452bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                                Arrays.copyOfRange(currentGroup.mChars, 0, differentCharIndex),
45372b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                                shortcutTargets, null /* bigrams */, frequency,
45472b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                                isNotAWord, isBlacklistEntry, newChildren);
455bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    } else {
456bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                        newParent = new CharGroup(
457bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                                Arrays.copyOfRange(currentGroup.mChars, 0, differentCharIndex),
45872b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                                null /* shortcutTargets */, null /* bigrams */, -1,
45972b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                                false /* isNotAWord */, false /* isBlacklistEntry */, newChildren);
46044c64f46a143623dd793facd889c8d6eab5e230cJean Chalard                        final CharGroup newWord = new CharGroup(Arrays.copyOfRange(word,
46144c64f46a143623dd793facd889c8d6eab5e230cJean Chalard                                charIndex + differentCharIndex, word.length),
46272b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                                shortcutTargets, null /* bigrams */, frequency,
46372b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                                isNotAWord, isBlacklistEntry);
464bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                        final int addIndex = word[charIndex + differentCharIndex]
465bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                                > currentGroup.mChars[differentCharIndex] ? 1 : 0;
466bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                        newChildren.mData.add(addIndex, newWord);
467bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    }
468bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    currentNode.mData.set(nodeIndex, newParent);
469bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                }
47047db0be7cbdb8abafc18c1e49b71f6dac0d46994Jean Chalard                if (DBG) checkStack(currentNode);
471bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            }
472bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        }
473bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    }
474bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
47593ebf74bae44728e0d5f7e738ea28376187a876eTadashi G. Takaoka    private static int ARRAYS_ARE_EQUAL = 0;
47693ebf74bae44728e0d5f7e738ea28376187a876eTadashi G. Takaoka
477bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    /**
478bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * Custom comparison of two int arrays taken to contain character codes.
479bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     *
480bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * This method compares the two arrays passed as an argument in a lexicographic way,
481bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * with an offset in the dst string.
482bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * This method does NOT test for the first character. It is taken to be equal.
483bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * I repeat: this method starts the comparison at 1 <> dstOffset + 1.
484bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * The index where the strings differ is returned. ARRAYS_ARE_EQUAL = 0 is returned if the
485bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * strings are equal. This works BECAUSE we don't look at the first character.
486bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     *
487bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * @param src the left-hand side string of the comparison.
488bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * @param dst the right-hand side string of the comparison.
489bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * @param dstOffset the offset in the right-hand side string.
490bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * @return the index at which the strings differ, or ARRAYS_ARE_EQUAL = 0 if they don't.
491bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     */
492bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    private static int compareArrays(final int[] src, final int[] dst, int dstOffset) {
493bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        // We do NOT test the first char, because we come from a method that already
494bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        // tested it.
495bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        for (int i = 1; i < src.length; ++i) {
496bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            if (dstOffset + i >= dst.length) return i;
497bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            if (src[i] != dst[dstOffset + i]) return i;
498bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        }
499bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        if (dst.length > src.length) return src.length;
500bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        return ARRAYS_ARE_EQUAL;
501bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    }
502bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
503bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    /**
504bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * Helper class that compares and sorts two chargroups according to their
505bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * first element only. I repeat: ONLY the first element is considered, the rest
506bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * is ignored.
507bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * This comparator imposes orderings that are inconsistent with equals.
508bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     */
5092aa02b84a4fcfaf5554c278d2b25cf9414eecf8bKen Wakasa    static private class CharGroupComparator implements java.util.Comparator<CharGroup> {
51093ebf74bae44728e0d5f7e738ea28376187a876eTadashi G. Takaoka        @Override
5112aa02b84a4fcfaf5554c278d2b25cf9414eecf8bKen Wakasa        public int compare(CharGroup c1, CharGroup c2) {
512bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            if (c1.mChars[0] == c2.mChars[0]) return 0;
513bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            return c1.mChars[0] < c2.mChars[0] ? -1 : 1;
514bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        }
515bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    }
516bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    final static private CharGroupComparator CHARGROUP_COMPARATOR = new CharGroupComparator();
517bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
518bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    /**
519bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * Finds the insertion index of a character within a node.
520bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     */
521bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    private static int findInsertionIndex(final Node node, int character) {
5222aa02b84a4fcfaf5554c278d2b25cf9414eecf8bKen Wakasa        final ArrayList<CharGroup> data = node.mData;
52344c64f46a143623dd793facd889c8d6eab5e230cJean Chalard        final CharGroup reference = new CharGroup(new int[] { character },
52472b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                null /* shortcutTargets */, null /* bigrams */, 0, false /* isNotAWord */,
52572b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                false /* isBlacklistEntry */);
526bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        int result = Collections.binarySearch(data, reference, CHARGROUP_COMPARATOR);
527bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        return result >= 0 ? result : -result - 1;
528bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    }
529bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
53093ebf74bae44728e0d5f7e738ea28376187a876eTadashi G. Takaoka    private static int CHARACTER_NOT_FOUND = -1;
53193ebf74bae44728e0d5f7e738ea28376187a876eTadashi G. Takaoka
532bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    /**
533bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * Find the index of a char in a node, if it exists.
534bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     *
535bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * @param node the node to search in.
536bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * @param character the character to search for.
537bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * @return the position of the character if it's there, or CHARACTER_NOT_FOUND = -1 else.
538bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     */
539bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    private static int findIndexOfChar(final Node node, int character) {
540bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        final int insertionIndex = findInsertionIndex(node, character);
541bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        if (node.mData.size() <= insertionIndex) return CHARACTER_NOT_FOUND;
542bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        return character == node.mData.get(insertionIndex).mChars[0] ? insertionIndex
543bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                : CHARACTER_NOT_FOUND;
544bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    }
545bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
546bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    /**
547bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * Helper method to find a word in a given branch.
548bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     */
549bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    public static CharGroup findWordInTree(Node node, final String s) {
550bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        int index = 0;
55112efad3d15147f255f6e01600c40e9fdb1224d84Jean Chalard        final StringBuilder checker = DBG ? new StringBuilder() : null;
552bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
553bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        CharGroup currentGroup;
554bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        do {
555bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            int indexOfGroup = findIndexOfChar(node, s.codePointAt(index));
556bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            if (CHARACTER_NOT_FOUND == indexOfGroup) return null;
557bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            currentGroup = node.mData.get(indexOfGroup);
55866f338983bb9cb04a0d94a4729330b1c8ff01c93Yuichiro Hanada
55966f338983bb9cb04a0d94a4729330b1c8ff01c93Yuichiro Hanada            if (s.length() - index < currentGroup.mChars.length) return null;
56066f338983bb9cb04a0d94a4729330b1c8ff01c93Yuichiro Hanada            int newIndex = index;
56166f338983bb9cb04a0d94a4729330b1c8ff01c93Yuichiro Hanada            while (newIndex < s.length() && newIndex - index < currentGroup.mChars.length) {
56266f338983bb9cb04a0d94a4729330b1c8ff01c93Yuichiro Hanada                if (currentGroup.mChars[newIndex - index] != s.codePointAt(newIndex)) return null;
56366f338983bb9cb04a0d94a4729330b1c8ff01c93Yuichiro Hanada                newIndex++;
56466f338983bb9cb04a0d94a4729330b1c8ff01c93Yuichiro Hanada            }
56566f338983bb9cb04a0d94a4729330b1c8ff01c93Yuichiro Hanada            index = newIndex;
56666f338983bb9cb04a0d94a4729330b1c8ff01c93Yuichiro Hanada
56712efad3d15147f255f6e01600c40e9fdb1224d84Jean Chalard            if (DBG) checker.append(new String(currentGroup.mChars, 0, currentGroup.mChars.length));
568bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            if (index < s.length()) {
569bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                node = currentGroup.mChildren;
570bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            }
571bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        } while (null != node && index < s.length());
572bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
57366f338983bb9cb04a0d94a4729330b1c8ff01c93Yuichiro Hanada        if (index < s.length()) return null;
5740d35c159fefd7591c2ab9d5037c32d1804024197Yuichiro Hanada        if (!currentGroup.isTerminal()) return null;
57512efad3d15147f255f6e01600c40e9fdb1224d84Jean Chalard        if (DBG && !s.equals(checker.toString())) return null;
576bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        return currentGroup;
577bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    }
578bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
579bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    /**
580eddfbb68dcff55c85b3d5b82d406f543bd038825Jean Chalard     * Helper method to find out whether a word is in the dict or not.
581eddfbb68dcff55c85b3d5b82d406f543bd038825Jean Chalard     */
582eddfbb68dcff55c85b3d5b82d406f543bd038825Jean Chalard    public boolean hasWord(final String s) {
583eddfbb68dcff55c85b3d5b82d406f543bd038825Jean Chalard        if (null == s || "".equals(s)) {
584eddfbb68dcff55c85b3d5b82d406f543bd038825Jean Chalard            throw new RuntimeException("Can't search for a null or empty string");
585eddfbb68dcff55c85b3d5b82d406f543bd038825Jean Chalard        }
586eddfbb68dcff55c85b3d5b82d406f543bd038825Jean Chalard        return null != findWordInTree(mRoot, s);
587eddfbb68dcff55c85b3d5b82d406f543bd038825Jean Chalard    }
588eddfbb68dcff55c85b3d5b82d406f543bd038825Jean Chalard
589eddfbb68dcff55c85b3d5b82d406f543bd038825Jean Chalard    /**
590bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * Recursively count the number of character groups in a given branch of the trie.
591bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     *
592bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * @param node the parent node.
593bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * @return the number of char groups in all the branch under this node.
594bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     */
595bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    public static int countCharGroups(final Node node) {
596bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        final int nodeSize = node.mData.size();
597bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        int size = nodeSize;
598bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        for (int i = nodeSize - 1; i >= 0; --i) {
599bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            CharGroup group = node.mData.get(i);
600bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            if (null != group.mChildren)
601bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                size += countCharGroups(group.mChildren);
602bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        }
603bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        return size;
604bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    }
605bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
606bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    /**
607bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * Recursively count the number of nodes in a given branch of the trie.
608bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     *
609bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * @param node the node to count.
61020a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard     * @return the number of nodes in this branch.
611bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     */
612bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    public static int countNodes(final Node node) {
613bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        int size = 1;
614bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        for (int i = node.mData.size() - 1; i >= 0; --i) {
615bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            CharGroup group = node.mData.get(i);
616bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            if (null != group.mChildren)
617bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                size += countNodes(group.mChildren);
618bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        }
619bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        return size;
620bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    }
621bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
62220a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard    // Recursively find out whether there are any bigrams.
62320a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard    // This can be pretty expensive especially if there aren't any (we return as soon
62420a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard    // as we find one, so it's much cheaper if there are bigrams)
62520a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard    private static boolean hasBigramsInternal(final Node node) {
62620a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard        if (null == node) return false;
62720a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard        for (int i = node.mData.size() - 1; i >= 0; --i) {
62820a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard            CharGroup group = node.mData.get(i);
62920a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard            if (null != group.mBigrams) return true;
63020a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard            if (hasBigramsInternal(group.mChildren)) return true;
63120a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard        }
63220a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard        return false;
63320a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard    }
63420a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard
63520a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard    /**
63620a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard     * Finds out whether there are any bigrams in this dictionary.
63720a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard     *
63820a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard     * @return true if there is any bigram, false otherwise.
63920a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard     */
64020a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard    // TODO: this is expensive especially for large dictionaries without any bigram.
64120a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard    // The up side is, this is always accurate and correct and uses no memory. We should
64220a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard    // find a more efficient way of doing this, without compromising too much on memory
64320a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard    // and ease of use.
64420a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard    public boolean hasBigrams() {
64520a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard        return hasBigramsInternal(mRoot);
64620a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard    }
64720a6dea1cabfd8822824f7dca828d898e5b91cbcJean Chalard
648bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    // Historically, the tails of the words were going to be merged to save space.
649bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    // However, that would prevent the code to search for a specific address in log(n)
650bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    // time so this was abandoned.
651bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    // The code is still of interest as it does add some compression to any dictionary
652bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    // that has no need for attributes. Implementations that does not read attributes should be
653bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    // able to read a dictionary with merged tails.
654bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    // Also, the following code does support frequencies, as in, it will only merges
655bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    // tails that share the same frequency. Though it would result in the above loss of
656bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    // performance while searching by address, it is still technically possible to merge
657bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    // tails that contain attributes, but this code does not take that into account - it does
658bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    // not compare attributes and will merge terminals with different attributes regardless.
659bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    public void mergeTails() {
660bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        MakedictLog.i("Do not merge tails");
661bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        return;
662bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
663bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//        MakedictLog.i("Merging nodes. Number of nodes : " + countNodes(root));
664bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//        MakedictLog.i("Number of groups : " + countCharGroups(root));
665bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//
666bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//        final HashMap<String, ArrayList<Node>> repository =
667bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//                  new HashMap<String, ArrayList<Node>>();
668bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//        mergeTailsInner(repository, root);
669bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//
670bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//        MakedictLog.i("Number of different pseudohashes : " + repository.size());
671bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//        int size = 0;
672bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//        for (ArrayList<Node> a : repository.values()) {
673bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//            size += a.size();
674bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//        }
675bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//        MakedictLog.i("Number of nodes after merge : " + (1 + size));
676bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//        MakedictLog.i("Recursively seen nodes : " + countNodes(root));
677bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    }
678bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
679bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    // The following methods are used by the deactivated mergeTails()
680bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//   private static boolean isEqual(Node a, Node b) {
681bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//       if (null == a && null == b) return true;
682bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//       if (null == a || null == b) return false;
683bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//       if (a.data.size() != b.data.size()) return false;
684bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//       final int size = a.data.size();
685bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//       for (int i = size - 1; i >= 0; --i) {
686bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//           CharGroup aGroup = a.data.get(i);
687bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//           CharGroup bGroup = b.data.get(i);
688bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//           if (aGroup.frequency != bGroup.frequency) return false;
689bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//           if (aGroup.alternates == null && bGroup.alternates != null) return false;
690bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//           if (aGroup.alternates != null && !aGroup.equals(bGroup.alternates)) return false;
691bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//           if (!Arrays.equals(aGroup.chars, bGroup.chars)) return false;
692bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//           if (!isEqual(aGroup.children, bGroup.children)) return false;
693bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//       }
694bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//       return true;
695bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//   }
696bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
697bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//   static private HashMap<String, ArrayList<Node>> mergeTailsInner(
698bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//           final HashMap<String, ArrayList<Node>> map, final Node node) {
699bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//       final ArrayList<CharGroup> branches = node.data;
700bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//       final int nodeSize = branches.size();
701bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//       for (int i = 0; i < nodeSize; ++i) {
702bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//           CharGroup group = branches.get(i);
703bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//           if (null != group.children) {
704bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//               String pseudoHash = getPseudoHash(group.children);
705bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//               ArrayList<Node> similarList = map.get(pseudoHash);
706bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//               if (null == similarList) {
707bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//                   similarList = new ArrayList<Node>();
708bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//                   map.put(pseudoHash, similarList);
709bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//               }
710bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//               boolean merged = false;
711bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//               for (Node similar : similarList) {
712bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//                   if (isEqual(group.children, similar)) {
713bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//                       group.children = similar;
714bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//                       merged = true;
715bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//                       break;
716bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//                   }
717bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//               }
718bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//               if (!merged) {
719bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//                   similarList.add(group.children);
720bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//               }
721bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//               mergeTailsInner(map, group.children);
722bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//           }
723bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//       }
724bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//       return map;
725bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//   }
726bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
727bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//  private static String getPseudoHash(final Node node) {
728bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//      StringBuilder s = new StringBuilder();
729bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//      for (CharGroup g : node.data) {
730bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//          s.append(g.frequency);
731f2789819bd005b5b0581e8439601b5501306327dKen Wakasa//          for (int ch : g.chars) {
732bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//              s.append(Character.toChars(ch));
733bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//          }
734bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//      }
735bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//      return s.toString();
736bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard//  }
737bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
738bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    /**
739bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * Iterator to walk through a dictionary.
740bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     *
741bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * This is purely for convenience.
742bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     */
743bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    public static class DictionaryIterator implements Iterator<Word> {
744bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
745bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        private static class Position {
746bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            public Iterator<CharGroup> pos;
747bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            public int length;
748bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            public Position(ArrayList<CharGroup> groups) {
749bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                pos = groups.iterator();
750bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                length = 0;
751bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            }
752bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        }
753bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        final StringBuilder mCurrentString;
754bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        final LinkedList<Position> mPositions;
755bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
756bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        public DictionaryIterator(ArrayList<CharGroup> root) {
757bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            mCurrentString = new StringBuilder();
758bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            mPositions = new LinkedList<Position>();
759bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            final Position rootPos = new Position(root);
760bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            mPositions.add(rootPos);
761bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        }
762bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
763bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        @Override
764bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        public boolean hasNext() {
765bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            for (Position p : mPositions) {
766bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                if (p.pos.hasNext()) {
767bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    return true;
768bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                }
769bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            }
770bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            return false;
771bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        }
772bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
773bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        @Override
774bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        public Word next() {
775bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            Position currentPos = mPositions.getLast();
776bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            mCurrentString.setLength(mCurrentString.length() - currentPos.length);
777bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
778bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            do {
779bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                if (currentPos.pos.hasNext()) {
780bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    final CharGroup currentGroup = currentPos.pos.next();
781bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    currentPos.length = currentGroup.mChars.length;
782bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    for (int i : currentGroup.mChars)
783bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                        mCurrentString.append(Character.toChars(i));
784bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    if (null != currentGroup.mChildren) {
785bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                        currentPos = new Position(currentGroup.mChildren.mData);
786bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                        mPositions.addLast(currentPos);
787bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    }
788bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    if (currentGroup.mFrequency >= 0)
789bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                        return new Word(mCurrentString.toString(), currentGroup.mFrequency,
79072b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                                currentGroup.mShortcutTargets, currentGroup.mBigrams,
79172b1c9394105b6fbc0d8c6ff00f3574ee37a9aaaJean Chalard                                currentGroup.mIsNotAWord, currentGroup.mIsBlacklistEntry);
792bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                } else {
793bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    mPositions.removeLast();
794bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    currentPos = mPositions.getLast();
795bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                    mCurrentString.setLength(mCurrentString.length() - mPositions.getLast().length);
796bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard                }
797f2789819bd005b5b0581e8439601b5501306327dKen Wakasa            } while (true);
798bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        }
799bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
800bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        @Override
801bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        public void remove() {
802bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard            throw new UnsupportedOperationException("Unsupported yet");
803bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        }
804bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
805bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    }
806bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard
807bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    /**
808bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * Method to return an iterator.
809bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     *
810bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * This method enables Java's enhanced for loop. With this you can have a FusionDictionary x
811bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     * and say : for (Word w : x) {}
812bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard     */
813bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    @Override
814bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    public Iterator<Word> iterator() {
815bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard        return new DictionaryIterator(mRoot.mData);
816bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard    }
817bfbbee8c5757aef4a20879547c16af0a4d1bf4c7Jean Chalard}
818