org.tinygroup.chinese.rank.SentenceRankDefault Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of org.tinygroup.tinyseg Show documentation

The newest version!

package org.tinygroup.chinese.rank;

import org.tinygroup.chinese.*;
import org.tinygroup.chinese.Character;
import org.tinygroup.logger.Logger;
import org.tinygroup.logger.LoggerFactory;

import java.util.List;

/**
 * Created by luog on 15/4/20.
 */
public class SentenceRankDefault implements SentenceRank {
	private static final Logger LOGGER = LoggerFactory
			.getLogger(SentenceRankDefault.class);
	
    private WordParserManager wordParserManager;

    public void setWordParserManager(WordParserManager wordParserManager) {
        this.wordParserManager = wordParserManager;
    }

    public int rank(List tokenList) {
        int rank = 0;
        for (Token token : tokenList) {
            if(token.getTokenType()!=TokenType.WORD){
                continue;
            }
            String wordString = token.getWord();
            if (wordString.length() == 1) {
                try {
                    Character character = wordParserManager.getCharacter(wordString.charAt(0));
                    if (character.isSingleWord()) {
                        //如果有单字
                        rank += 1;
                        continue;
                    }
                } catch (ParserException e) {
                	LOGGER.errorMessage("评价发生异常:", e);
                }
                rank -= 1;
            } else {
                try {
                    Word word = wordParserManager.getWord(wordString);
                    if (word.getWeight() != 0) {
                        rank += word.getWeight();
                        continue;
                    }

                } catch (ParserException e) {
                	LOGGER.errorMessage("评价发生异常:", e);
                }
                rank += getDefaultWeight(wordString.length());
            }
        }
        return rank;
    }

    private int getDefaultWeight(int length) {
        return length * 2 - 1;
    }
}