WSOutput.java

package org.apache.lucene.analysis.kr.morph;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

public class WSOutput  implements Cloneable {

	private int lastStart = 0;
	
	private int lastEnd = 0;	
	
	private List<AnalysisOutput> phrases = new ArrayList();
	
	public WSOutput() {
		
	}
	
	public WSOutput(AnalysisOutput o) {
		addPhrase(o);
	}
	
	public int getLastStart() {
		return lastStart;
	}

	public void setLastStart(int start) {
		this.lastStart = start;
	}

	public int getLastEnd() {
		return lastEnd;
	}

	public void setLastEnd(int end) {
		this.lastStart = end;
	}
	

	public List<AnalysisOutput> getPhrases() {
		return phrases;
	}

	public void removeLast() {
				
		if(this.phrases.size()==0) return;
		
		AnalysisOutput o = this.phrases.remove(this.phrases.size()-1);
		
		if(this.phrases.size()==0) {
			
			this.lastStart = 0;
			this.lastEnd = 0;
			
		} else {
			
			this.lastEnd -= o.getSource().length();
			
			if(this.phrases.size()>1) {
				AnalysisOutput o1 = this.phrases.get(this.phrases.size()-1);
				this.lastStart = lastEnd-o1.getSource().length();
			} else {
				this.lastStart = 0;
			}
			
		}
		
	}
	
	public void addPhrase(AnalysisOutput o) {

		this.lastStart = this.lastEnd;
		this.lastEnd += o.getSource().length();
		
		if(o.getCNounList().size()==0)
			this.phrases.add(o);
		else
			addCompounds(o);

	}
	
	private void addCompounds(AnalysisOutput o) {
		
		List<CompoundEntry> cnouns = o.getCNounList();
			
		String source = o.getSource();		
		int rmstemlen = 0;
		
//		for(int i=0;i<cnouns.size();i++) {
//			System.out.println(cnouns.get(i).getWord());
//		}
		for(int i=0;i<cnouns.size()-1;i++) {
			
			String noun = cnouns.get(i).getWord();			
			boolean isOnechar = false;
		
			// 접두어는 처음 음절에만 온다. 복합명사 분해규칙
			// 처음이 아닌 경우 1글자는 앞 문자와 결합한다.
			if(cnouns.get(i).getWord().length()==1 ||
					cnouns.get(i+1).getWord().length()==1) { // 접두어는 처음 음절에만 온다. 복합명사 분해규칙
				noun += cnouns.get(i+1).getWord();			
				isOnechar = true;
			}
			
			if(isOnechar && i>=cnouns.size()-2) break;
						
			int score = AnalysisOutput.SCORE_CORRECT;
			if(!cnouns.get(i).isExist()) score=AnalysisOutput.SCORE_CANDIDATE;
			
			AnalysisOutput o1 = new AnalysisOutput(noun, null, null, 
					PatternConstants.POS_NOUN, PatternConstants.PTN_N, score);
			
			o1.setSource(noun);
			
			if(isOnechar) {
				o1.addCNoun(cnouns.get(i));
				o1.addCNoun(cnouns.get(i+1));
			}
		
			if(source.length()>noun.length())
				source = source.substring(noun.length());
			
			this.phrases.add(o1);
			cnouns.remove(cnouns.get(0));
			i--;
			
			if(isOnechar) {
				cnouns.remove(cnouns.get(0));
			}

		}
		
		o.setStem(o.getStem().substring(o.getSource().length()-source.length()));
		o.setSource(source);
		if(cnouns.size()==1) cnouns.remove(0);
	
		this.phrases.add(o);

	}
	
	public void setPhrases(List<AnalysisOutput> phrases) {
		this.phrases = phrases;
	}
	
	public WSOutput clone() throws CloneNotSupportedException {
				
		WSOutput candidate = (WSOutput)super.clone();
		
		candidate.setLastStart(lastStart);
		
		candidate.setLastEnd(lastEnd);
		
		List list = new ArrayList();
		list.addAll(phrases);
		candidate.setPhrases(list);
		
		return candidate;
	}
}