MtasSpanContainingQuery.java.html 8.29 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>MtasSpanContainingQuery.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">MtasSpanContainingQuery.java</span></div><h1>MtasSpanContainingQuery.java</h1><pre class="source lang-java linenums">package mtas.search.spans;

import java.io.IOException;

import org.apache.lucene.index.IndexReader;
import org.apache.lucene.search.IndexSearcher;
import org.apache.lucene.search.spans.SpanContainingQuery;
import org.apache.lucene.search.spans.SpanWeight;

import mtas.search.spans.util.MtasSpanQuery;

/**
 * The Class MtasSpanContainingQuery.
 */
public class MtasSpanContainingQuery extends MtasSpanQuery {

  /** The base query. */
  private SpanContainingQuery baseQuery;

  /** The big query. */
  private MtasSpanQuery bigQuery;

  /** The small query. */
  private MtasSpanQuery smallQuery;

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

  /**
   * Instantiates a new mtas span containing query.
   *
   * @param q1 the q 1
   * @param q2 the q 2
   */
  public MtasSpanContainingQuery(MtasSpanQuery q1, MtasSpanQuery q2) {
<span class="pc bpc" id="L36" title="2 of 4 branches missed.">    super(q1 != null ? q1.getMinimumWidth() : null,</span>
<span class="pc" id="L37">        q1 != null ? q1.getMaximumWidth() : null);</span>
<span class="pc bpc" id="L38" title="2 of 4 branches missed.">    if (q2 != null &amp;&amp; q2.getMinimumWidth() != null</span>
<span class="fc bfc" id="L39" title="All 2 branches covered.">        &amp;&amp; (this.getMinimumWidth() == null</span>
<span class="pc bpc" id="L40" title="1 of 2 branches missed.">            || this.getMinimumWidth() &lt; q2.getMinimumWidth())) {</span>
<span class="fc" id="L41">      this.setWidth(q2.getMinimumWidth(), this.getMaximumWidth());</span>
    }
<span class="fc" id="L43">    bigQuery = q1;</span>
<span class="fc" id="L44">    smallQuery = q2;</span>
<span class="pc bpc" id="L45" title="2 of 4 branches missed.">    if (bigQuery != null &amp;&amp; bigQuery.getField() != null) {</span>
<span class="fc" id="L46">      field = bigQuery.getField();</span>
<span class="nc bnc" id="L47" title="All 4 branches missed.">    } else if (smallQuery != null &amp;&amp; smallQuery.getField() != null) {</span>
<span class="nc" id="L48">      field = smallQuery.getField();</span>
    } else {
<span class="nc" id="L50">      field = null;</span>
    }
<span class="pc bpc" id="L52" title="3 of 6 branches missed.">    if (field != null &amp;&amp; bigQuery != null &amp;&amp; smallQuery != null) {</span>
<span class="pc bpc" id="L53" title="2 of 4 branches missed.">      if (bigQuery.getField() != null &amp;&amp; smallQuery.getField() != null) {</span>
<span class="fc" id="L54">        baseQuery = new SpanContainingQuery(bigQuery, smallQuery);</span>
      } else {
<span class="nc" id="L56">        baseQuery = null;</span>
      }
    } else {
<span class="nc" id="L59">      baseQuery = null;</span>
    }
<span class="fc" id="L61">  }</span>

  /*
   * (non-Javadoc)
   * 
   * @see org.apache.lucene.search.spans.SpanQuery#getField()
   */
  @Override
  public String getField() {
<span class="fc" id="L70">    return baseQuery.getField();</span>
  }

  /*
   * (non-Javadoc)
   * 
   * @see
   * org.apache.lucene.search.spans.SpanQuery#createWeight(org.apache.lucene.
   * search.IndexSearcher, boolean)
   */
  @Override
  public SpanWeight createWeight(IndexSearcher searcher, boolean needsScores)
      throws IOException {
<span class="fc" id="L83">    return baseQuery.createWeight(searcher, needsScores);</span>
  }

  /*
   * (non-Javadoc)
   * 
   * @see org.apache.lucene.search.Query#toString(java.lang.String)
   */
  @Override
  public String toString(String field) {
<span class="nc" id="L93">    StringBuilder buffer = new StringBuilder();</span>
<span class="nc" id="L94">    buffer.append(this.getClass().getSimpleName());</span>
<span class="nc" id="L95">    buffer.append(&quot;(&quot;);</span>
<span class="nc bnc" id="L96" title="All 2 branches missed.">    buffer.append(bigQuery != null ? bigQuery.toString(field) : &quot;null&quot;);</span>
<span class="nc" id="L97">    buffer.append(&quot;, &quot;);</span>
<span class="nc bnc" id="L98" title="All 2 branches missed.">    buffer.append(smallQuery != null ? smallQuery.toString(field) : &quot;null&quot;);</span>
<span class="nc" id="L99">    buffer.append(&quot;)&quot;);</span>
<span class="nc" id="L100">    return buffer.toString();</span>
  }

  /*
   * (non-Javadoc)
   * 
   * @see mtas.search.spans.util.MtasSpanQuery#rewrite(org.apache.lucene.index.
   * IndexReader)
   */
  @Override
  public MtasSpanQuery rewrite(IndexReader reader) throws IOException {
<span class="fc" id="L111">    MtasSpanQuery newBigQuery = bigQuery.rewrite(reader);</span>
<span class="fc" id="L112">    MtasSpanQuery newSmallQuery = smallQuery.rewrite(reader);</span>
<span class="pc bpc" id="L113" title="4 of 8 branches missed.">    if (newBigQuery == null || newBigQuery instanceof MtasSpanMatchNoneQuery</span>
        || newSmallQuery == null
        || newSmallQuery instanceof MtasSpanMatchNoneQuery) {
<span class="nc" id="L116">      return new MtasSpanMatchNoneQuery(field);</span>
    }
<span class="pc bpc" id="L118" title="1 of 4 branches missed.">    if (!newBigQuery.equals(bigQuery) || !newSmallQuery.equals(smallQuery)) {</span>
<span class="fc" id="L119">      return new MtasSpanContainingQuery(newBigQuery, newSmallQuery)</span>
<span class="fc" id="L120">          .rewrite(reader);</span>
<span class="fc bfc" id="L121" title="All 2 branches covered.">    } else if (newBigQuery.equals(newSmallQuery)) {</span>
<span class="fc" id="L122">      return newBigQuery;</span>
<span class="pc bpc" id="L123" title="1 of 2 branches missed.">    } else if (baseQuery == null) {</span>
<span class="nc" id="L124">      return new MtasSpanMatchNoneQuery(field);</span>
    } else {
<span class="fc" id="L126">      baseQuery = (SpanContainingQuery) baseQuery.rewrite(reader);</span>
<span class="fc" id="L127">      return super.rewrite(reader);</span>
    }
  }

  /*
   * (non-Javadoc)
   * 
   * @see org.apache.lucene.search.Query#equals(java.lang.Object)
   */
  @Override
  public boolean equals(Object obj) {
<span class="fc bfc" id="L138" title="All 2 branches covered.">    if (this == obj)</span>
<span class="fc" id="L139">      return true;</span>
<span class="pc bpc" id="L140" title="1 of 2 branches missed.">    if (obj == null)</span>
<span class="nc" id="L141">      return false;</span>
<span class="pc bpc" id="L142" title="1 of 2 branches missed.">    if (getClass() != obj.getClass())</span>
<span class="nc" id="L143">      return false;</span>
<span class="fc" id="L144">    final MtasSpanContainingQuery that = (MtasSpanContainingQuery) obj;</span>
<span class="fc" id="L145">    return baseQuery.equals(that.baseQuery);</span>
  }

  /*
   * (non-Javadoc)
   * 
   * @see org.apache.lucene.search.Query#hashCode()
   */
  @Override
  public int hashCode() {
<span class="nc" id="L155">    return baseQuery.hashCode();</span>
  }

  /*
   * (non-Javadoc)
   * 
   * @see mtas.search.spans.util.MtasSpanQuery#disableTwoPhaseIterator()
   */
  @Override
  public void disableTwoPhaseIterator() {
<span class="fc" id="L165">    super.disableTwoPhaseIterator();</span>
<span class="fc" id="L166">    bigQuery.disableTwoPhaseIterator();</span>
<span class="fc" id="L167">    smallQuery.disableTwoPhaseIterator();</span>
<span class="fc" id="L168">  }</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>