package uniqifier;

import java.util.HashSet;
import java.util.Set;

public class WordCollection {
    private Set<String> words;
    private WordNormalizer normalizer;

    public WordCollection(WordNormalizer wn) {
        words = new HashSet();
        normalizer = wn;
    }

    // Breaks up the passed-in string into words, normalizes them using
    // WordNormalizer, then inserts the normalized word into its word map.
    public void BuildFromString(String str) {
        for (String s : str.split("\\s+")) {
            s = normalizer.Normalize(s);
            words.add(s);
        }
    }

    public Set<String> GetWords() {
        return words;
    }
}