com.joliciel.talismane.languageDetector.DefaultLanguageDetectorProcessor.java Source code

Introduction

Here is the source code for com.joliciel.talismane.languageDetector.DefaultLanguageDetectorProcessor.java

Source

///////////////////////////////////////////////////////////////////////////////
//Copyright (C) 2014 Joliciel Informatique
//
//This file is part of Talismane.
//
//Talismane is free software: you can redistribute it and/or modify
//it under the terms of the GNU Affero General Public License as published by
//the Free Software Foundation, either version 3 of the License, or
//(at your option) any later version.
//
//Talismane is distributed in the hope that it will be useful,
//but WITHOUT ANY WARRANTY; without even the implied warranty of
//MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//GNU Affero General Public License for more details.
//
//You should have received a copy of the GNU Affero General Public License
//along with Talismane.  If not, see <http://www.gnu.org/licenses/>.
//////////////////////////////////////////////////////////////////////////////
package com.joliciel.talismane.languageDetector;

import java.io.IOException;
import java.io.Writer;
import java.util.List;
import java.util.Locale;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import com.joliciel.talismane.utils.LogUtils;
import com.joliciel.talismane.utils.WeightedOutcome;

public class DefaultLanguageDetectorProcessor implements LanguageDetectorProcessor {
    private static final Log LOG = LogFactory.getLog(DefaultLanguageDetectorProcessor.class);

    private Writer out;

    public DefaultLanguageDetectorProcessor(Writer out) {
        this.out = out;
    }

    @Override
    public void onNextText(String text, List<WeightedOutcome<Locale>> results, Writer writer) {
        try {
            if (writer == null)
                writer = out;

            writer.write(text + "\n");
            for (WeightedOutcome<Locale> result : results) {
                writer.write(result.getOutcome().toLanguageTag() + "\t" + result.getWeight() + "\n");
            }
            writer.flush();
        } catch (IOException e) {
            LogUtils.logError(LOG, e);
            throw new RuntimeException(e);
        }
    }

}