1 files changed, 51 insertions, 0 deletions
diff --git a/addons/buildSrc/src/main/java/WordListWriter.java b/addons/buildSrc/src/main/java/WordListWriter.java
new file mode 100644
index 000000000..1d3277f2e
--- /dev/null
+++ b/addons/buildSrc/src/main/java/WordListWriter.java
@@ -0,0 +1,51 @@
+import java.io.Closeable;
+import java.io.File;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.io.OutputStreamWriter;
+import java.nio.charset.Charset;
+
+class WordListWriter implements Closeable {
+
+    private final XmlWriter mXmlWriter;
+    private long mWordsWritten;
+
+    WordListWriter(File outputWordsListFile) throws IOException {
+        final File parentFile = outputWordsListFile.getParentFile();
+        if (!parentFile.exists() && !parentFile.mkdirs()) {
+            throw new IllegalArgumentException("Failed to create output folder " + parentFile.getAbsolutePath());
+        }
+        OutputStreamWriter outputWriter = new OutputStreamWriter(new FileOutputStream(outputWordsListFile), Charset.forName("UTF-8"));
+        mXmlWriter = new XmlWriter(outputWriter, true, 0, true);
+        mXmlWriter.writeEntity("wordlist");
+    }
+
+    public static void writeWordWithRuntimeException(WordListWriter writer, String word, int frequency) {
+        try {
+            writer.addEntry(word, frequency);
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    public void addEntry(String word, int frequency) throws IOException {
+        mXmlWriter
+                .writeEntity("w")
+                .writeAttribute("f", Integer.toString(frequency))
+                .writeText(word)
+                .endEntity();
+        mWordsWritten++;
+    }
+
+    @Override
+    public void close() throws IOException {
+        mXmlWriter.endEntity();
+        System.out.println("Wrote " + mWordsWritten + " words.");
+        mXmlWriter.close();
+    }
+
+    private static int calcActualFreq(double wordIndex, double wordsCount) {
+        return Math.min(255, 1 + (int) (255 * (wordsCount - wordIndex) / wordsCount));
+    }
+
+}