WSOutput.java

/*
 * Copyright 2011-2013 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

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

import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;

public class WSOutput implements Cloneable, Serializable {

    private static final long serialVersionUID = 5688830925667373969L;

    private int lastStart = 0;
    private int lastEnd = 0;
    private List<AnalysisOutput> phrases = new ArrayList<AnalysisOutput>();

    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();
            String next = cnouns.get(i + 1).getWord();
            boolean isOnechar = false;

            // 접두어는 처음 음절에만 온다. 복합명사 분해규칙
            // 처음이 아닌 경우 1글자는 앞 문자와 결합한다.
            if (noun.length() == 1 || next.length() == 1) { // 접두어는 처음 음절에만 온다. 복합명사 분해규칙
                noun += next;
                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);
        candidate.setPhrases(new ArrayList<AnalysisOutput>(phrases));

        return candidate;
    }
}