ExampleSentenceProvider.java

/* LanguageTool, a natural language style checker
 * Copyright (C) 2012 Daniel Naber (http://www.danielnaber.de)
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301
 * USA
 */
package org.languagetool.server;

import org.languagetool.JLanguageTool;
import org.languagetool.Language;
import org.languagetool.rules.IncorrectExample;
import org.languagetool.rules.Rule;
import org.languagetool.rules.patterns.PatternRule;

import java.io.IOException;
import java.util.*;

/**
 * Provides example sentences from the XML rules.
 */
class ExampleSentenceProvider {

  private final int minSentences;
  private final int maxSentences;
  private final Random rnd = new Random(12345);
  private final Map<Language, List<ExampleSentence>> languageToExamples = new HashMap<Language, List<ExampleSentence>>();

  ExampleSentenceProvider(int minSentences, int maxSentences) {
    if (minSentences > maxSentences) {
      throw new IllegalArgumentException("min > max: " + minSentences + " > " + maxSentences);
    }
    this.minSentences = minSentences;
    this.maxSentences = maxSentences;
    for (Language language : Language.REAL_LANGUAGES) {
      try {
        initExampleSentences(language);
      } catch (IOException e) {
        throw new RuntimeException(e);
      }
    }
  }

  private void initExampleSentences(Language language) throws IOException {
    final JLanguageTool lt = new JLanguageTool(language);
    lt.activateDefaultPatternRules();
    final List<Rule> rules = lt.getAllActiveRules();
    final List<ExampleSentence> sentences = new ArrayList<ExampleSentence>();
    for (Rule rule : rules) {
      if (rule instanceof PatternRule && !rule.isDefaultOff()) {
        final List<IncorrectExample> incorrectExamples = rule.getIncorrectExamples();
        for (IncorrectExample incorrectExample : incorrectExamples) {
          final ExampleSentence sentence = new ExampleSentence(incorrectExample.getExample(), rule.getId());
          sentences.add(sentence);
        }
      }
    }
    languageToExamples.put(language, sentences);
  }

  List<ExampleSentence> getRandomSentences(Language lang) {
    final List<ExampleSentence> sentences = new ArrayList<ExampleSentence>(languageToExamples.get(lang));
    final int sentenceCount = rnd.nextInt(Math.max(1, maxSentences - minSentences)) + minSentences;
    Collections.shuffle(sentences, rnd);
    return sentences.subList(0, Math.min(sentences.size(), sentenceCount));
  }

}