EntityExtractFactory.java

/**
 * 
 */
package com.personalityextractor.entity.extractor;

/**
 * @author semanticvoid
 *
 */
public class EntityExtractFactory {

	public enum Extracter {
		BASELINE, CONSECUTIVE_WORDS, NOUNPHRASE, SENNANOUNPHRASE, PROPERNOUNPHRASE, COMMONNOUNPHRASE;
	}
	
	public static IEntityExtractor produceExtractor(Extracter e) {
		if(e == Extracter.BASELINE) {
			return new BaselineExtractor();
		} else if(e == Extracter.CONSECUTIVE_WORDS) {
			return new ConsecutiveWordsEntityExtractor();
		} else if(e == Extracter.NOUNPHRASE) {
			return new NounPhraseExtractor();
		} else if (e == Extracter.SENNANOUNPHRASE){
			return new SennaNounPhraseExtractor();
		} else if (e == Extracter.PROPERNOUNPHRASE){
			return new ProperNounPhraseExtractor();
		} else if (e == Extracter.COMMONNOUNPHRASE){
			return new CommonNounPhraseExtractor();
		}else {
			// default
			return new BaselineExtractor();
		}
	}
}