Spelling Suggestions

Code: Select all

<?php
class Speller {

	function __construct($dictfile) {
		$this->dictionary = array();
		$this->file = $dictfile;
		$this->train();
	}

	function train() {
		$f = fopen ($this->file, "r");
	    while ($line= fgets ($f)) {
			$word = split(",", $line);
			$this->dictionary[strtolower($word[0])] = $word[1];
	    }
	    fclose ($f);
	}

	function correct($word) {
	        $word = strtolower($word);
        	if(isset($this->dictionary[$word])) {
        	        return $word;
        	}

       		$edits1 = $edits2 = array();
        	foreach($this->dictionary as $dictWord => $count) {
			if (substr($dictWord, 0, 1) === substr($word, 0, 1)) {
                		$dist = levenshtein($word, $dictWord);
                		if($dist == 1) {
                	        	$edits1[$dictWord] = $count;
                		} else if($dist == 2) {
                		        $edits2[$dictWord] = $count;
                		}
			}
        	}

      		if(count($edits1)) {
        	        arsort($edits1);
                	return key($edits1);
       		} else if(count($edits2)) {
                	arsort($edits2);
                	return key($edits2);
        	}

	        // Nothing better
        	return $word;
	}

}

/**TEST MAIN METHOD**/
//$speller = new Speller('dict.txt');
//print_r($speller->dictionary);
//echo $speller->correct('soni');
//echo $speller->correct('arta');
//echo $speller->correct('mluti');
//echo $speller->correct('canon');

?>