org.ansj.recognition.impl.SynonymsRecgnition Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ansj_seg Show documentation

best java chinese word seg !

The newest version!

package org.ansj.recognition.impl;

import org.ansj.domain.Result;
import org.ansj.domain.Term;
import org.ansj.library.SynonymsLibrary;
import org.ansj.recognition.Recognition;
import org.nlpcn.commons.lang.tire.domain.SmartForest;

import java.util.List;

/**
 * 同义词功能
 * 
 * @author Ansj
 *
 */
public class SynonymsRecgnition implements Recognition {

	private static final long serialVersionUID = 5961499108093950130L;

	private SmartForest> synonyms = null;

	public SynonymsRecgnition() {
		this.synonyms = SynonymsLibrary.get();
	}

	public SynonymsRecgnition(String key) {
		this.synonyms = SynonymsLibrary.get(key);
	}
	
	public SynonymsRecgnition(SmartForest> synonyms){
		this.synonyms = synonyms ;
	}

	@Override
	public void recognition(Result result) {
		for (Term term : result) {
			SmartForest> branch = synonyms.getBranch(term.getName());
			if (branch != null && branch.getStatus() > 1) {
				List syns = branch.getParam();
				if (syns != null) {
					term.setSynonyms(syns);
				}
			}
		}
	}

}