com.optimaize.langdetect.profiles.LanguageProfileWriter Maven / Gradle / Ivy

package com.optimaize.langdetect.profiles;

import org.jetbrains.annotations.NotNull;

import java.io.*;
import java.nio.charset.Charset;
import java.util.Map;

/**
 * Writes a {@link LanguageProfile} to an output stream or file.
 *
 * All file operations are done with UTF-8.
 *
 * @author François ROLAND
 * @author Fabian Kessler
 */
public class LanguageProfileWriter {

    /**
     * Writes a {@link LanguageProfile} to an OutputStream in UTF-8.
     *
     * @throws java.io.IOException
     */
    public void write(@NotNull LanguageProfile languageProfile, @NotNull OutputStream outputStream) throws IOException {
        try (BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(outputStream, Charset.forName("utf-8")))) {
            writer.write("{\"freq\":{");
            boolean first = true;
            for (Map.Entry entry : languageProfile.iterateGrams()) {
                if (!first) {
                    writer.write(',');
                }
                writer.write('"');
                writer.write(entry.getKey());
                writer.write("\":");
                writer.write(entry.getValue().toString());
                first = false;
            }
            writer.write("},\"n_words\":[");
            first = true;
            for (int i=1; i<=10; i++) {
                long nWord = languageProfile.getNumGramOccurrences(i);
                if (nWord ==0) break;
                if (!first) {
                    writer.write(',');
                }
                writer.write(Long.toString(nWord));
                first = false;
            }
            writer.write("],\"name\":\"");
            writer.write(languageProfile.getLocale().toString());
            writer.write("\"}");
            writer.flush();
        }
    }

    /**
     * Writes a {@link LanguageProfile} to a folder using the language name as the file name.
     *
     * @param fullPath Must be an existing writable directory path.
     * @throws java.io.IOException if such a file name exists already.
     */
    public void writeToDirectory(@NotNull LanguageProfile languageProfile, @NotNull File fullPath) throws IOException {
        if (!fullPath.exists()) {
            throw new IOException("Path does not exist: "+fullPath);
        }
        if (!fullPath.canWrite()) {
            throw new IOException("Path not writable: "+fullPath);
        }
        File file = new File(fullPath.getAbsolutePath()+"/"+languageProfile.getLocale());
        if (file.exists()) {
            throw new IOException("File exists already, refusing to overwrite: "+file);
        }
        try (FileOutputStream output = new FileOutputStream(file)) {
            write(languageProfile, output);
        }
    }

}