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

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.apache.lucene.index.IndexReader;
import org.apache.lucene.index.LeafReaderContext;
import org.apache.lucene.index.Term;
import org.apache.lucene.index.TermContext;
import org.apache.lucene.index.Terms;
import org.apache.lucene.search.IndexSearcher;
import org.apache.lucene.search.spans.SpanQuery;
import org.apache.lucene.search.spans.SpanWeight;
import org.apache.lucene.search.spans.Spans;
import mtas.search.spans.util.MtasSpanQuery;
import mtas.search.spans.util.MtasSpanWeight;
import mtas.search.spans.util.MtasSpans;

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

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

  /** The q 1. */
  private SpanQuery q1;

  /** The q 2. */
  private SpanQuery q2;

  /**
   * Instantiates a new mtas span intersecting query.
   *
   * @param q1 the q 1
   * @param q2 the q 2
   */
  public MtasSpanIntersectingQuery(MtasSpanQuery q1, MtasSpanQuery q2) {
<span class="pc bpc" id="L43" title="2 of 4 branches missed.">    super(q1 != null ? q1.getMinimumWidth() : null,</span>
<span class="pc" id="L44">        q1 != null ? q1.getMaximumWidth() : null);</span>
<span class="pc bpc" id="L45" title="2 of 4 branches missed.">    if (q1 != null &amp;&amp; (field = q1.getField()) != null) {</span>
<span class="pc bpc" id="L46" title="3 of 6 branches missed.">      if (q2 != null &amp;&amp; q2.getField() != null &amp;&amp; !q2.getField().equals(field)) {</span>
<span class="nc" id="L47">        throw new IllegalArgumentException(&quot;Clauses must have same field.&quot;);</span>
      }
<span class="nc bnc" id="L49" title="All 2 branches missed.">    } else if (q2 != null) {</span>
<span class="nc" id="L50">      field = q2.getField();</span>
    } else {
<span class="nc" id="L52">      field = null;</span>
    }
<span class="fc" id="L54">    this.q1 = q1;</span>
<span class="fc" id="L55">    this.q2 = q2;</span>
<span class="fc" id="L56">  }</span>

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

  /*
   * (non-Javadoc)
   * 
   * @see
   * org.apache.lucene.search.spans.SpanQuery#createWeight(org.apache.lucene.
   * search.IndexSearcher, boolean)
   */
  @Override
  public MtasSpanWeight createWeight(IndexSearcher searcher,
      boolean needsScores) throws IOException {
<span class="pc bpc" id="L78" title="2 of 4 branches missed.">    if (q1 == null || q2 == null) {</span>
<span class="nc" id="L79">      return null;</span>
    } else {
<span class="fc" id="L81">      MtasSpanIntersectingQueryWeight w1 = new MtasSpanIntersectingQueryWeight(</span>
<span class="fc" id="L82">          q1.createWeight(searcher, needsScores));</span>
<span class="fc" id="L83">      MtasSpanIntersectingQueryWeight w2 = new MtasSpanIntersectingQueryWeight(</span>
<span class="fc" id="L84">          q2.createWeight(searcher, needsScores));</span>
      // subWeights
<span class="fc" id="L86">      List&lt;MtasSpanIntersectingQueryWeight&gt; subWeights = new ArrayList&lt;&gt;();</span>
<span class="fc" id="L87">      subWeights.add(w1);</span>
<span class="fc" id="L88">      subWeights.add(w2);</span>
      // return
<span class="pc bpc" id="L90" title="1 of 2 branches missed.">      return new SpanIntersectingWeight(w1, w2, searcher,</span>
<span class="pc" id="L91">          needsScores ? getTermContexts(subWeights) : null);</span>
    }
  }

  /**
   * Gets the term contexts.
   *
   * @param items the items
   * @return the term contexts
   */
  protected Map&lt;Term, TermContext&gt; getTermContexts(
      List&lt;MtasSpanIntersectingQueryWeight&gt; items) {
<span class="nc" id="L103">    List&lt;SpanWeight&gt; weights = new ArrayList&lt;&gt;();</span>
<span class="nc bnc" id="L104" title="All 2 branches missed.">    for (MtasSpanIntersectingQueryWeight item : items) {</span>
<span class="nc" id="L105">      weights.add(item.spanWeight);</span>
<span class="nc" id="L106">    }</span>
<span class="nc" id="L107">    return getTermContexts(weights);</span>
  }

  /*
   * (non-Javadoc)
   * 
   * @see org.apache.lucene.search.Query#toString(java.lang.String)
   */
  @Override
  public String toString(String field) {
<span class="nc" id="L117">    StringBuilder buffer = new StringBuilder();</span>
<span class="nc" id="L118">    buffer.append(this.getClass().getSimpleName() + &quot;([&quot;);</span>
<span class="nc bnc" id="L119" title="All 2 branches missed.">    if (q1 != null) {</span>
<span class="nc" id="L120">      buffer.append(q1.toString(q1.getField()));</span>
    } else {
<span class="nc" id="L122">      buffer.append(&quot;null&quot;);</span>
    }
<span class="nc" id="L124">    buffer.append(&quot;,&quot;);</span>
<span class="nc bnc" id="L125" title="All 2 branches missed.">    if (q2 != null) {</span>
<span class="nc" id="L126">      buffer.append(q2.toString(q2.getField()));</span>
    } else {
<span class="nc" id="L128">      buffer.append(&quot;null&quot;);</span>
    }
<span class="nc" id="L130">    buffer.append(&quot;])&quot;);</span>
<span class="nc" id="L131">    return buffer.toString();</span>
  }

  /*
   * (non-Javadoc)
   * 
   * @see org.apache.lucene.search.Query#equals(java.lang.Object)
   */
  @Override
  public boolean equals(Object obj) {
<span class="fc bfc" id="L141" title="All 2 branches covered.">    if (this == obj)</span>
<span class="fc" id="L142">      return true;</span>
<span class="pc bpc" id="L143" title="1 of 2 branches missed.">    if (obj == null)</span>
<span class="nc" id="L144">      return false;</span>
<span class="pc bpc" id="L145" title="1 of 2 branches missed.">    if (getClass() != obj.getClass())</span>
<span class="nc" id="L146">      return false;</span>
<span class="fc" id="L147">    final MtasSpanIntersectingQuery other = (MtasSpanIntersectingQuery) obj;</span>
<span class="pc bpc" id="L148" title="1 of 4 branches missed.">    return q1.equals(other.q1) &amp;&amp; q2.equals(other.q2);</span>
  }

  /*
   * (non-Javadoc)
   * 
   * @see org.apache.lucene.search.Query#hashCode()
   */
  @Override
  public int hashCode() {
<span class="nc" id="L158">    int h = Integer.rotateLeft(classHash(), 1);</span>
<span class="nc" id="L159">    h ^= q1.hashCode();</span>
<span class="nc" id="L160">    h = Integer.rotateLeft(h, 1);</span>
<span class="nc" id="L161">    h ^= q2.hashCode();</span>
<span class="nc" id="L162">    return h;</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="L173">    MtasSpanQuery newQ1 = (MtasSpanQuery) q1.rewrite(reader);</span>
<span class="fc" id="L174">    MtasSpanQuery newQ2 = (MtasSpanQuery) q2.rewrite(reader);</span>
<span class="pc bpc" id="L175" title="1 of 4 branches missed.">    if (!newQ1.equals(q1) || !newQ2.equals(q2)) {</span>
<span class="fc" id="L176">      return new MtasSpanIntersectingQuery(newQ1, newQ2).rewrite(reader);</span>
<span class="pc bpc" id="L177" title="1 of 2 branches missed.">    } else if (newQ1.equals(newQ2)) {</span>
<span class="nc" id="L178">      return newQ1;</span>
    } else {

      boolean returnNone;
<span class="pc bpc" id="L182" title="1 of 2 branches missed.">      returnNone = newQ1.getMaximumWidth() != null</span>
<span class="pc bnc" id="L183" title="All 2 branches missed.">          &amp;&amp; newQ1.getMaximumWidth() == 0;</span>
<span class="pc bpc" id="L184" title="1 of 2 branches missed.">      returnNone |= newQ2.getMaximumWidth() != null</span>
<span class="pc bpc" id="L185" title="1 of 2 branches missed.">          &amp;&amp; newQ2.getMaximumWidth() == 0;</span>
<span class="pc bpc" id="L186" title="1 of 2 branches missed.">      if (returnNone) {</span>
<span class="nc" id="L187">        return new MtasSpanMatchNoneQuery(this.getField());</span>
      } else {
<span class="fc" id="L189">        return super.rewrite(reader);</span>
      }
    }
  }

  /**
   * The Class SpanIntersectingWeight.
   */
  protected class SpanIntersectingWeight extends MtasSpanWeight {

    /** The w 1. */
    MtasSpanIntersectingQueryWeight w1;

    /** The w 2. */
    MtasSpanIntersectingQueryWeight w2;

    /**
     * Instantiates a new span intersecting weight.
     *
     * @param w1 the w 1
     * @param w2 the w 2
     * @param searcher the searcher
     * @param terms the terms
     * @throws IOException Signals that an I/O exception has occurred.
     */
    public SpanIntersectingWeight(MtasSpanIntersectingQueryWeight w1,
        MtasSpanIntersectingQueryWeight w2, IndexSearcher searcher,
<span class="fc" id="L216">        Map&lt;Term, TermContext&gt; terms) throws IOException {</span>
<span class="fc" id="L217">      super(MtasSpanIntersectingQuery.this, searcher, terms);</span>
<span class="fc" id="L218">      this.w1 = w1;</span>
<span class="fc" id="L219">      this.w2 = w2;</span>
<span class="fc" id="L220">    }</span>

    /*
     * (non-Javadoc)
     * 
     * @see
     * org.apache.lucene.search.spans.SpanWeight#extractTermContexts(java.util.
     * Map)
     */
    @Override
    public void extractTermContexts(Map&lt;Term, TermContext&gt; contexts) {
<span class="nc" id="L231">      w1.spanWeight.extractTermContexts(contexts);</span>
<span class="nc" id="L232">      w2.spanWeight.extractTermContexts(contexts);</span>
<span class="nc" id="L233">    }</span>

    /*
     * (non-Javadoc)
     * 
     * @see
     * org.apache.lucene.search.spans.SpanWeight#getSpans(org.apache.lucene.
     * index.LeafReaderContext,
     * org.apache.lucene.search.spans.SpanWeight.Postings)
     */
    @Override
    public MtasSpans getSpans(LeafReaderContext context,
        Postings requiredPostings) throws IOException {
<span class="fc" id="L246">      Terms terms = context.reader().terms(field);</span>
<span class="pc bpc" id="L247" title="1 of 2 branches missed.">      if (terms == null) {</span>
<span class="nc" id="L248">        return null; // field does not exist</span>
      }
<span class="fc" id="L250">      MtasSpanIntersectingQuerySpans s1 = new MtasSpanIntersectingQuerySpans(</span>
<span class="fc" id="L251">          w1.spanWeight.getSpans(context, requiredPostings));</span>
<span class="fc" id="L252">      MtasSpanIntersectingQuerySpans s2 = new MtasSpanIntersectingQuerySpans(</span>
<span class="fc" id="L253">          w2.spanWeight.getSpans(context, requiredPostings));</span>
<span class="fc" id="L254">      return new MtasSpanIntersectingSpans(s1, s2);</span>
    }

    /*
     * (non-Javadoc)
     * 
     * @see org.apache.lucene.search.Weight#extractTerms(java.util.Set)
     */
    @Override
    public void extractTerms(Set&lt;Term&gt; terms) {
<span class="nc" id="L264">      w1.spanWeight.extractTerms(terms);</span>
<span class="nc" id="L265">      w2.spanWeight.extractTerms(terms);</span>
<span class="nc" id="L266">    }</span>

  }

  /**
   * The Class MtasSpanIntersectingQuerySpans.
   */
  protected static class MtasSpanIntersectingQuerySpans {

    /** The spans. */
    public Spans spans;

    /**
     * Instantiates a new mtas span intersecting query spans.
     *
     * @param spans the spans
     */
<span class="fc" id="L283">    public MtasSpanIntersectingQuerySpans(Spans spans) {</span>
<span class="pc bpc" id="L284" title="1 of 2 branches missed.">      this.spans = spans != null ? spans : new MtasSpanMatchNoneSpans();</span>
<span class="fc" id="L285">    }</span>

  }

  /**
   * The Class MtasSpanIntersectingQueryWeight.
   */
  private static class MtasSpanIntersectingQueryWeight {

    /** The span weight. */
    public SpanWeight spanWeight;

    /**
     * Instantiates a new mtas span intersecting query weight.
     *
     * @param spanWeight the span weight
     */
<span class="fc" id="L302">    public MtasSpanIntersectingQueryWeight(SpanWeight spanWeight) {</span>
<span class="fc" id="L303">      this.spanWeight = spanWeight;</span>
<span class="fc" id="L304">    }</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>