package lemmaextractor.results;


import java.util.ArrayList;

/**
 * Klasa przetrzymująca informacje uzyskane po przetworzeniu zadanego tekstu.
 * Zawiera inforamcje o języku przetwarzania, lematach i nazwach własnych.
 */
public class Result {

    private String language;
    ArrayList<Lemma> lemmas;
    ArrayList<NamedEntity> namedEntities;

    public Result(String lang) {
        language = lang;
        lemmas = new ArrayList<Lemma>();
        namedEntities = new ArrayList<NamedEntity>();

    }

    /**
     * Dodaje lemat.
     */
    public void addLemma(String text, int start, int end) {
        lemmas.add(new Lemma(text, start, end));
    }

    /**
     * Dodaje nazwe własną.
     */
    public void addNamedEntity(String namedEntityForm, String namedEntityLemma, ArrayList<Integer> indices) {
        namedEntities.add(new NamedEntity(namedEntityForm, namedEntityLemma, indices));
    }

    /**
     * Zwraca język dla którego wykonane zostało przetwarzanie.
     *
     * @return język
     */
    public String getLanguage() {
        return language;
    }

    /**
     * Zwraca sekwencję lematów.
     *
     * @return lista obiektów klasy Lemma
     */
    public ArrayList<Lemma> getLemmas() {
        return lemmas;
    }

    /**
     * Zwraca sekwencję nazw własnych.
     *
     * @return Lista obiektów klasy NamedEntity
     */
    public ArrayList<NamedEntity> getNamedEntities() {
        return namedEntities;
    }

    /**
     * Zwraca sekwencję nazw własnych.
     *
     * @return Lista obiektów klasy NamedEntity
     */
    public ArrayList<Lemma> getNamedEntityLemmas(NamedEntity namedEntity) {
        int first = namedEntity.indices.get(0);
        int last = namedEntity.indices.get(namedEntity.indices.size()-1);
     return (ArrayList<Lemma>) lemmas.subList(first, last + 1);
    }

    /**
     * Zwraca początkową pozycję nazwy własnej w tekście.
     *
     * @return początkowa pozycja nazwy własnej w tekście.
     */
    public int getNamedEntityStart(NamedEntity namedEntity) {
        return lemmas.get(namedEntity.indices.get(0)).start;
    }

    /**
     * Zwraca koncową pozycję nazwy własnej w tekście.
     *
     * @return końcowa pozycja nazwy własnej w tekście.
     */
    public int getNamedEntityEnd(NamedEntity namedEntity) {
        return lemmas.get(namedEntity.indices.get(namedEntity.indices.size()-1)).end;
    }
}