FilterStopword.java

/**
 * 
 */
package com.maalaang.omtwitter.text;

import java.util.Set;

import com.maalaang.omtwitter.model.OMTweet;

/**
 * @author Sangwon Park
 *
 */
public class FilterStopword implements TweetFilter {
	
	private Set<String> stopwords = null;
	private int stopwordNum = 0;
	
	private boolean filtered = false;
	
	public FilterStopword(Set<String> stopwords, int stopwordNum) {
		this.stopwords = stopwords;
		this.stopwordNum = stopwordNum;
	}
	
	public void initialize() {
		
	}

	public void next(OMTweet tweet, OMTweetToken[] tokenList) {
		int cnt = 0;
		for (OMTweetToken tok : tokenList) {
			if (stopwords.contains(tok.getText())) {
				if (++cnt > stopwordNum) {
					filtered = false;
					return;
				}
			}
		}
		filtered = true;
	}

	public boolean isFilteredOut() {
		return filtered;
	}

	public void close() {
		
	}
}