dkpro.similarity.experiments.wordchoice.util.WordChoiceProblemFactory.java Source code

Introduction

Here is the source code for dkpro.similarity.experiments.wordchoice.util.WordChoiceProblemFactory.java

Source

/*******************************************************************************
 * Copyright 2013
 * Ubiquitous Knowledge Processing (UKP) Lab
 * Technische Universitt Darmstadt
 *
 * 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 dkpro.similarity.experiments.wordchoice.util;

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

import org.apache.commons.lang.StringUtils;
import org.apache.uima.analysis_engine.AnalysisEngineProcessException;
import org.apache.uima.fit.util.JCasUtil;
import org.apache.uima.jcas.JCas;

import de.tudarmstadt.ukp.dkpro.core.api.segmentation.type.Lemma;
import de.tudarmstadt.ukp.dkpro.core.api.segmentation.type.Sentence;
import dkpro.similarity.type.WordChoiceProblem;

public class WordChoiceProblemFactory {

    public static List<WCProblem> getWordChoiceProblems(JCas jcas) throws AnalysisEngineProcessException {

        List<WCProblem> wcpList = new ArrayList<WCProblem>();

        for (WordChoiceProblem wcp : JCasUtil.select(jcas, WordChoiceProblem.class)) {
            // get the sentences => they represent target and candidates
            List<Sentence> candidateList = JCasUtil.selectCovered(jcas, Sentence.class, wcp);

            if (candidateList.size() != 5) {
                throw new AnalysisEngineProcessException(new Throwable(
                        "Expected five sentence annotations, but only " + candidateList.size() + " were found."));
            }

            Sentence target = candidateList.get(0);

            List<Lemma> targetAnnotationList = JCasUtil.selectCovered(jcas, Lemma.class, target);
            if (targetAnnotationList.size() != 1) {
                System.out.println("'" + target.getCoveredText() + "'");
                throw new AnalysisEngineProcessException(new Throwable(targetAnnotationList.size()
                        + " Lemma annotations for the target found, but expected only one."));
            }
            String targetString = targetAnnotationList.get(0).getValue();

            String cand1 = getCandidate(jcas, candidateList.get(1));
            String cand2 = getCandidate(jcas, candidateList.get(2));
            String cand3 = getCandidate(jcas, candidateList.get(3));
            String cand4 = getCandidate(jcas, candidateList.get(4));

            List<String> candList1 = getCandidateList(jcas, candidateList.get(1));
            List<String> candList2 = getCandidateList(jcas, candidateList.get(2));
            List<String> candList3 = getCandidateList(jcas, candidateList.get(3));
            List<String> candList4 = getCandidateList(jcas, candidateList.get(4));

            wcpList.add(new WCProblem(targetString, cand1, cand2, cand3, cand4, candList1, candList2, candList3,
                    candList4, wcp.getCorrectAnswer()));

        }

        return wcpList;
    }

    private static List<String> getCandidateList(JCas jcas, Sentence candidateAnnotation) {
        List<Lemma> candidateAnnotationList = JCasUtil.selectCovered(jcas,
                de.tudarmstadt.ukp.dkpro.core.api.segmentation.type.Lemma.class, candidateAnnotation);
        return getLemmaList(candidateAnnotationList);

    }

    private static String getCandidate(JCas jcas, Sentence candidateAnnotation) {
        List<String> lemmas = getCandidateList(jcas, candidateAnnotation);
        return StringUtils.join(lemmas, " ");
    }

    public static List<String> getLemmaList(List<Lemma> lemmas) {
        List<String> lemmaList = new ArrayList<String>();
        for (Lemma l : lemmas) {
            lemmaList.add(l.getValue());
        }
        return lemmaList;
    }

}