/starter/src/cs276/pe1/spell/KGramWithEditAndFreqSpellingCorrector.java

package cs276.pe1.spell;

import java.io.File;
import java.util.List;

import cs276.util.IOUtils;
import cs276.util.StringUtils;
import java.util.*;

import cs276.util.Counter;

public class KGramWithEditAndFreqSpellingCorrector implements SpellingCorrector {
	/** Initializes spelling corrector by indexing kgrams in words from a file */
	
	KGramWithEditDistanceSpellingCorrector KGram;
	KGramSpellingCorrector simpleKGram;
	
	public KGramWithEditAndFreqSpellingCorrector() {
	    KGram = new KGramWithEditDistanceSpellingCorrector();
	    simpleKGram = new KGramSpellingCorrector();
	}

	public List<String> corrections(String word) {
	
	    List<String> guesses = KGram.corrections(word);
	    List<String> newGuesses = new ArrayList<String>();
	    Counter<String> frequencies = new Counter<String>();
	    
	    // get ties
	    if (guesses != null && guesses.size() > 0) {
	        double closestEditDistance = KGram.getEditDistance(word, guesses.get(0));
	        for (String guess : guesses) {
	            if (KGram.getEditDistance(word, guess) == closestEditDistance) {
	                frequencies.setCount(guess, simpleKGram.getOccurrences(guess));
                } else {
                    break;
                }
            }
        }
	        
	    // Use guess frequency as tie-break between guesses with same edit distance;
	    return frequencies.topK(1);
	    
	}

	
}