package cz.brmlab.yodaqa.analysis.passage.biotagger;

import org.apache.uima.UimaContext;
import org.apache.uima.analysis_engine.AnalysisEngineProcessException;
import org.apache.uima.cas.CASException;
import org.apache.uima.fit.descriptor.SofaCapability;
import org.apache.uima.fit.util.JCasUtil;
import org.apache.uima.jcas.JCas;
import org.apache.uima.resource.ResourceInitializationException;
import org.slf4j.LoggerFactory;

import cz.brmlab.yodaqa.analysis.ansscore.AnswerFV;
import cz.brmlab.yodaqa.analysis.passage.CandidateGenerator;
import cz.brmlab.yodaqa.model.CandidateAnswer.AF_BIOScore;
import cz.brmlab.yodaqa.model.CandidateAnswer.AF_OriginPsgBIO;
import cz.brmlab.yodaqa.model.SearchResult.AnswerBioMention;
import cz.brmlab.yodaqa.model.SearchResult.Passage;
import cz.brmlab.yodaqa.model.SearchResult.ResultInfo;

import de.tudarmstadt.ukp.dkpro.core.api.segmentation.type.Token;

/**
 * Create CandidateAnswer based on AnswerMention pre-annotations.
 * We split this to a separate annotator and use the AnswerMention
 * intermediate annotations to ease integration with ClearTK. */

@SofaCapability(
	inputSofas = { "Question", "Result", "PickedPassages" },
	outputSofas = { "PickedPassages" }
)

public class CanByAnsBioMention extends CandidateGenerator {
	public CanByAnsBioMention() {
		logger = LoggerFactory.getLogger(CanByAnsBioMention.class);
	}


	public void initialize(UimaContext aContext) throws ResourceInitializationException {
		super.initialize(aContext);
	}

	public void process(JCas jcas) throws AnalysisEngineProcessException {
		JCas resultView, passagesView;
		try {
			resultView = jcas.getView("Result");
			passagesView = jcas.getView("PickedPassages");
		} catch (CASException e) {
			throw new AnalysisEngineProcessException(e);
		}

		ResultInfo ri = JCasUtil.selectSingle(resultView, ResultInfo.class);

		for (AnswerBioMention abm : JCasUtil.select(passagesView, AnswerBioMention.class))
			processMention(passagesView, ri, abm);
	}

	protected void processMention(JCas passagesView, ResultInfo ri, AnswerBioMention abm)
			throws AnalysisEngineProcessException {
		Passage p = JCasUtil.selectCovering(Passage.class, abm).get(0);

		// XXX: we just grab the first token as a base
		Token base = JCasUtil.selectCovered(Token.class, abm).get(0);

		AnswerFV fv = new AnswerFV(ri.getAnsfeatures());
		fv.merge(new AnswerFV(p.getAnsfeatures()));
		fv.setFeature(AF_OriginPsgBIO.class, 1.0);
		fv.setFeature(AF_BIOScore.class, abm.getScore());

		addCandidateAnswer(passagesView, p, base, fv);
	}
}

@@ -2180,122 +2180,8 @@
 );%0A%0A
-%09%09// XXX: we just grab the first token as a base%0A%09%09Token base = JCasUtil.selectCovered(Token.class, abm).get(0);%0A%0A
 %09%09An
@@ -2409,20 +2409,19 @@
 iew, p, 
-base
+abm
 , fv);%0A%09

Create answer from the whole abm, not just first token