MtasSpanPositionSpans.java.html 8.79 KB
<?xml version="1.0" encoding="UTF-8"?><!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd"><html xmlns="http://www.w3.org/1999/xhtml" lang="en"><head><meta http-equiv="Content-Type" content="text/html;charset=UTF-8"/><link rel="stylesheet" href="../jacoco-resources/report.css" type="text/css"/><link rel="shortcut icon" href="../jacoco-resources/report.gif" type="image/gif"/><title>MtasSpanPositionSpans.java</title><link rel="stylesheet" href="../jacoco-resources/prettify.css" type="text/css"/><script type="text/javascript" src="../jacoco-resources/prettify.js"></script></head><body onload="window['PR_TAB_WIDTH']=4;prettyPrint()"><div class="breadcrumb" id="breadcrumb"><span class="info"><a href="../jacoco-sessions.html" class="el_session">Sessions</a></span><a href="../index.html" class="el_report">MTAS</a> &gt; <a href="index.source.html" class="el_package">mtas.search.spans</a> &gt; <span class="el_source">MtasSpanPositionSpans.java</span></div><h1>MtasSpanPositionSpans.java</h1><pre class="source lang-java linenums">package mtas.search.spans;

import java.io.IOException;

import mtas.codec.util.CodecInfo;
import mtas.codec.util.CodecInfo.IndexDoc;
import mtas.search.spans.util.MtasSpans;

import org.apache.lucene.search.TwoPhaseIterator;
import org.apache.lucene.search.spans.SpanCollector;

/**
 * The Class MtasSpanPositionSpans.
 */
public class MtasSpanPositionSpans extends MtasSpans {

  /** The query. */
  private MtasSpanPositionQuery query;

  /** The field. */
  private String field;

  /** The start. */
  private int start;
  
  /** The end. */
  private int end;
  
  /** The min position. */
  private int minPosition;
  
  /** The max position. */
  private int maxPosition;
  
  /** The current start position. */
  private int currentStartPosition;
  
  /** The current end position. */
  private int currentEndPosition;
  
  /** The doc id. */
  private int docId;

  /** The mtas codec info. */
  private CodecInfo mtasCodecInfo;

  /**
   * Instantiates a new mtas span position spans.
   *
   * @param query the query
   * @param mtasCodecInfo the mtas codec info
   * @param field the field
   * @param start the start
   * @param end the end
   */
  public MtasSpanPositionSpans(MtasSpanPositionQuery query,
      CodecInfo mtasCodecInfo, String field, int start, int end) {
<span class="fc" id="L58">    super();</span>
<span class="fc" id="L59">    this.query = query;</span>
<span class="fc" id="L60">    this.mtasCodecInfo = mtasCodecInfo;</span>
<span class="fc" id="L61">    this.field = field;</span>
<span class="fc" id="L62">    this.start = start;</span>
<span class="fc" id="L63">    this.end = end;</span>
<span class="fc" id="L64">    minPosition = NO_MORE_POSITIONS;</span>
<span class="fc" id="L65">    maxPosition = NO_MORE_POSITIONS;</span>
<span class="fc" id="L66">    currentStartPosition = NO_MORE_POSITIONS;</span>
<span class="fc" id="L67">    currentEndPosition = NO_MORE_POSITIONS;</span>
<span class="fc" id="L68">    docId = -1;</span>
<span class="fc" id="L69">  }</span>

  /*
   * (non-Javadoc)
   * 
   * @see org.apache.lucene.search.spans.Spans#nextStartPosition()
   */
  @Override
  public int nextStartPosition() throws IOException {
<span class="fc bfc" id="L78" title="All 2 branches covered.">    if (currentStartPosition &lt; minPosition) {</span>
<span class="fc" id="L79">      currentStartPosition = minPosition;</span>
<span class="fc" id="L80">      currentEndPosition = currentStartPosition + 1;</span>
    } else {
<span class="fc" id="L82">      currentStartPosition++;</span>
<span class="fc" id="L83">      currentEndPosition = currentStartPosition + 1;</span>
<span class="pc bpc" id="L84" title="1 of 2 branches missed.">      if (currentStartPosition &gt; maxPosition) {</span>
<span class="fc" id="L85">        currentStartPosition = NO_MORE_POSITIONS;</span>
<span class="fc" id="L86">        currentEndPosition = NO_MORE_POSITIONS;</span>
      }
    }
<span class="fc" id="L89">    return currentStartPosition;</span>
  }

  /*
   * (non-Javadoc)
   * 
   * @see org.apache.lucene.search.spans.Spans#startPosition()
   */
  @Override
  public int startPosition() {
<span class="fc" id="L99">    return currentStartPosition;</span>
  }

  /*
   * (non-Javadoc)
   * 
   * @see org.apache.lucene.search.spans.Spans#endPosition()
   */
  @Override
  public int endPosition() {
<span class="fc" id="L109">    return currentEndPosition;</span>
  }

  /*
   * (non-Javadoc)
   * 
   * @see org.apache.lucene.search.spans.Spans#width()
   */
  @Override
  public int width() {
<span class="nc" id="L119">    return 1;</span>
  }

  /*
   * (non-Javadoc)
   * 
   * @see
   * org.apache.lucene.search.spans.Spans#collect(org.apache.lucene.search.spans
   * .SpanCollector)
   */
  @Override
  public void collect(SpanCollector collector) throws IOException {
    // don't do anything
<span class="nc" id="L132">  }</span>

  /*
   * (non-Javadoc)
   * 
   * @see org.apache.lucene.search.DocIdSetIterator#docID()
   */
  @Override
  public int docID() {
<span class="fc" id="L141">    return docId;</span>
  }

  /*
   * (non-Javadoc)
   * 
   * @see org.apache.lucene.search.DocIdSetIterator#nextDoc()
   */
  @Override
  public int nextDoc() throws IOException {
    do {
<span class="fc" id="L152">      IndexDoc indexDoc = mtasCodecInfo.getNextDoc(field, docId);</span>
<span class="fc bfc" id="L153" title="All 2 branches covered.">      if (indexDoc != null) {</span>
<span class="fc" id="L154">        docId = indexDoc.docId;</span>
<span class="fc" id="L155">        minPosition = Math.max(start, indexDoc.minPosition);</span>
<span class="fc" id="L156">        maxPosition = Math.min(end, indexDoc.maxPosition);</span>
<span class="fc" id="L157">        currentStartPosition = -1;</span>
<span class="fc" id="L158">        currentEndPosition = -1;</span>
      } else {
<span class="fc" id="L160">        docId = NO_MORE_DOCS;</span>
<span class="fc" id="L161">        minPosition = NO_MORE_POSITIONS;</span>
<span class="fc" id="L162">        maxPosition = NO_MORE_POSITIONS;</span>
<span class="fc" id="L163">        currentStartPosition = NO_MORE_POSITIONS;</span>
<span class="fc" id="L164">        currentEndPosition = NO_MORE_POSITIONS;</span>
      }
<span class="pc bpc" id="L166" title="1 of 4 branches missed.">    } while (docId != NO_MORE_DOCS &amp;&amp; (minPosition &gt; maxPosition));</span>
<span class="fc" id="L167">    return docId;</span>
  }

  /*
   * (non-Javadoc)
   * 
   * @see org.apache.lucene.search.DocIdSetIterator#advance(int)
   */
  @Override
  public int advance(int target) throws IOException {
<span class="nc" id="L177">    int tmpTarget = target - 1;</span>
    do {
<span class="nc" id="L179">      IndexDoc indexDoc = mtasCodecInfo.getNextDoc(field, tmpTarget);</span>
<span class="nc bnc" id="L180" title="All 2 branches missed.">      if (indexDoc != null) {</span>
<span class="nc" id="L181">        docId = indexDoc.docId;</span>
<span class="nc" id="L182">        minPosition = Math.max(start, indexDoc.minPosition);</span>
<span class="nc" id="L183">        maxPosition = Math.min(end, indexDoc.maxPosition);</span>
<span class="nc" id="L184">        currentStartPosition = -1;</span>
<span class="nc" id="L185">        currentEndPosition = -1;</span>
      } else {
<span class="nc" id="L187">        docId = NO_MORE_DOCS;</span>
<span class="nc" id="L188">        minPosition = NO_MORE_POSITIONS;</span>
<span class="nc" id="L189">        maxPosition = NO_MORE_POSITIONS;</span>
<span class="nc" id="L190">        currentStartPosition = NO_MORE_POSITIONS;</span>
<span class="nc" id="L191">        currentEndPosition = NO_MORE_POSITIONS;</span>
      }
<span class="nc" id="L193">      tmpTarget = docId;</span>
<span class="nc bnc" id="L194" title="All 4 branches missed.">    } while (docId != NO_MORE_DOCS &amp;&amp; minPosition &gt; maxPosition);</span>
<span class="nc" id="L195">    return docId;</span>
  }

  /*
   * (non-Javadoc)
   * 
   * @see org.apache.lucene.search.DocIdSetIterator#cost()
   */
  @Override
  public long cost() {
<span class="nc" id="L205">    return mtasCodecInfo.getNumberOfDocs(field);</span>
  }

  /*
   * (non-Javadoc)
   * 
   * @see org.apache.lucene.search.spans.Spans#positionsCost()
   */
  @Override
  public float positionsCost() {
<span class="nc" id="L215">    return 0;</span>
  }

  /* (non-Javadoc)
   * @see mtas.search.spans.util.MtasSpans#asTwoPhaseIterator()
   */
  @Override
  public TwoPhaseIterator asTwoPhaseIterator() {
<span class="nc bnc" id="L223" title="All 2 branches missed.">    if (!query.twoPhaseIteratorAllowed()) {</span>
<span class="nc" id="L224">      return null;</span>
    } else {
      // TODO
<span class="nc" id="L227">      return null;</span>
    }
  }

}
</pre><div class="footer"><span class="right">Created with <a href="http://www.jacoco.org/jacoco">JaCoCo</a> 0.7.9.201702052155</span></div></body></html>