MtasMaximumExpandSpanQuery.java.html 12.9 KB

Edit Raw Blame History

<?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>MtasMaximumExpandSpanQuery.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.util</a> &gt; <span class="el_source">MtasMaximumExpandSpanQuery.java</span></div><h1>MtasMaximumExpandSpanQuery.java</h1><pre class="source lang-java linenums">package mtas.search.spans.util;

import java.io.IOException;
import java.lang.reflect.Method;
import java.util.Map;
import java.util.Set;

import org.apache.lucene.codecs.FieldsProducer;
import org.apache.lucene.index.IndexReader;
import org.apache.lucene.index.LeafReader;
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.SpanWeight;
import org.apache.lucene.search.spans.Spans;

import mtas.codec.util.CodecInfo;
import mtas.search.spans.MtasSpanMatchNoneSpans;

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

  /** The query. */
  MtasSpanQuery query;

  /** The minimum left. */
  int minimumLeft;

  /** The maximum left. */
  int maximumLeft;

  /** The minimum right. */
  int minimumRight;

  /** The maximum right. */
  int maximumRight;

  /**
   * Instantiates a new mtas maximum expand span query.
   *
   * @param query the query
   * @param minimumLeft the minimum left
   * @param maximumLeft the maximum left
   * @param minimumRight the minimum right
   * @param maximumRight the maximum right
   */
  public MtasMaximumExpandSpanQuery(MtasSpanQuery query, int minimumLeft,
      int maximumLeft, int minimumRight, int maximumRight) {
<span class="fc" id="L53">    super(null, null);</span>
<span class="fc" id="L54">    this.query = query;</span>
<span class="pc bpc" id="L55" title="4 of 8 branches missed.">    if (minimumLeft &gt; maximumLeft || minimumRight &gt; maximumRight</span>
        || minimumLeft &lt; 0 || minimumRight &lt; 0) {
<span class="nc" id="L57">      throw new IllegalArgumentException();</span>
    }
<span class="fc" id="L59">    this.minimumLeft = minimumLeft;</span>
<span class="fc" id="L60">    this.maximumLeft = maximumLeft;</span>
<span class="fc" id="L61">    this.minimumRight = minimumRight;</span>
<span class="fc" id="L62">    this.maximumRight = maximumRight;</span>
<span class="fc" id="L63">    Integer minimum = query.getMinimumWidth();</span>
<span class="fc" id="L64">    Integer maximum = query.getMaximumWidth();</span>
<span class="fc bfc" id="L65" title="All 2 branches covered.">    if (minimum != null) {</span>
<span class="fc" id="L66">      minimum += minimumLeft + minimumRight;</span>
    }
<span class="fc bfc" id="L68" title="All 2 branches covered.">    if (maximum != null) {</span>
<span class="fc" id="L69">      maximum += maximumLeft + maximumRight;</span>
    }
<span class="fc" id="L71">    setWidth(minimum, maximum);</span>
<span class="fc" id="L72">  }</span>

  /*
   * (non-Javadoc)
   *
   * @see
   * mtas.search.spans.util.MtasSpanQuery#createWeight(org.apache.lucene.search.
   * IndexSearcher, boolean)
   */
  @Override
  public SpanWeight createWeight(IndexSearcher searcher, boolean needsScores)
      throws IOException {
<span class="fc" id="L84">    SpanWeight subWeight = query.createWeight(searcher, needsScores);</span>
<span class="pc bpc" id="L85" title="3 of 4 branches missed.">    if (maximumLeft == 0 &amp;&amp; maximumRight == 0) {</span>
<span class="nc" id="L86">      return subWeight;</span>
    } else {
<span class="fc" id="L88">      return new MtasMaximumExpandWeight(subWeight, searcher, needsScores);</span>
    }
  }

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

  /*
   * (non-Javadoc)
   *
   * @see org.apache.lucene.search.Query#toString(java.lang.String)
   */
  @Override
  public String toString(String field) {
<span class="nc" id="L109">    StringBuilder buffer = new StringBuilder();</span>
<span class="nc" id="L110">    buffer.append(this.getClass().getSimpleName() + &quot;([&quot;);</span>
<span class="nc" id="L111">    buffer.append(query.toString(field) + &quot;][&quot; + minimumLeft + &quot;,&quot; + maximumLeft</span>
        + &quot;][&quot; + minimumRight + &quot;,&quot; + maximumRight + &quot;])&quot;);
<span class="nc" id="L113">    return buffer.toString();</span>
  }

  /*
   * (non-Javadoc)
   *
   * @see org.apache.lucene.search.Query#equals(java.lang.Object)
   */
  @Override
  public boolean equals(Object obj) {
<span class="pc bpc" id="L123" title="1 of 2 branches missed.">    if (this == obj)</span>
<span class="nc" id="L124">      return true;</span>
<span class="pc bpc" id="L125" title="1 of 2 branches missed.">    if (obj == null)</span>
<span class="nc" id="L126">      return false;</span>
<span class="pc bpc" id="L127" title="1 of 2 branches missed.">    if (getClass() != obj.getClass())</span>
<span class="nc" id="L128">      return false;</span>
<span class="fc" id="L129">    final MtasMaximumExpandSpanQuery that = (MtasMaximumExpandSpanQuery) obj;</span>
    boolean isEqual;
<span class="fc" id="L131">    isEqual = query.equals(that.query);</span>
<span class="pc bpc" id="L132" title="1 of 2 branches missed.">    isEqual &amp;= minimumLeft == that.minimumLeft;</span>
<span class="fc bfc" id="L133" title="All 2 branches covered.">    isEqual &amp;= maximumLeft == that.maximumLeft;</span>
<span class="fc bfc" id="L134" title="All 2 branches covered.">    isEqual &amp;= minimumRight == that.minimumRight;</span>
<span class="fc bfc" id="L135" title="All 2 branches covered.">    isEqual &amp;= maximumRight == that.maximumRight;</span>
<span class="fc" id="L136">    return isEqual;</span>
  }

  /*
   * (non-Javadoc)
   *
   * @see org.apache.lucene.search.Query#hashCode()
   */
  @Override
  public int hashCode() {
<span class="nc" id="L146">    int h = Integer.rotateLeft(classHash(), 1);</span>
<span class="nc" id="L147">    h ^= query.hashCode();</span>
<span class="nc" id="L148">    h = Integer.rotateLeft(h, minimumLeft) + minimumLeft;</span>
<span class="nc" id="L149">    h ^= 2;</span>
<span class="nc" id="L150">    h = Integer.rotateLeft(h, maximumLeft) + maximumLeft;</span>
<span class="nc" id="L151">    h ^= 3;</span>
<span class="nc" id="L152">    h = Integer.rotateLeft(h, minimumRight) + minimumRight;</span>
<span class="nc" id="L153">    h ^= 5;</span>
<span class="nc" id="L154">    h = Integer.rotateLeft(h, maximumRight) + maximumRight;</span>
<span class="nc" id="L155">    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="L166">    MtasSpanQuery newQuery = query.rewrite(reader);</span>
<span class="pc bpc" id="L167" title="1 of 4 branches missed.">    if (maximumLeft == 0 &amp;&amp; maximumRight == 0) {</span>
<span class="fc" id="L168">      return newQuery;</span>
<span class="pc bpc" id="L169" title="1 of 2 branches missed.">    } else if (!query.equals(newQuery)) {</span>
<span class="nc" id="L170">      return new MtasMaximumExpandSpanQuery(newQuery, minimumLeft, maximumLeft,</span>
          minimumRight, maximumRight);
    } else {
<span class="fc" id="L173">      return super.rewrite(reader);</span>
    }
  }

  /**
   * The Class MtasMaximumExpandWeight.
   */
  private class MtasMaximumExpandWeight extends MtasSpanWeight {

    /** The Constant METHOD_GET_DELEGATE. */
    private static final String METHOD_GET_DELEGATE = &quot;getDelegate&quot;;

    /** The Constant METHOD_GET_POSTINGS_READER. */
    private static final String METHOD_GET_POSTINGS_READER = &quot;getPostingsReader&quot;;

    /** The sub weight. */
    SpanWeight subWeight;

    /**
     * Instantiates a new mtas maximum expand weight.
     *
     * @param subWeight the sub weight
     * @param searcher the searcher
     * @param needsScores the needs scores
     * @throws IOException Signals that an I/O exception has occurred.
     */
    public MtasMaximumExpandWeight(SpanWeight subWeight, IndexSearcher searcher,
<span class="fc" id="L200">        boolean needsScores) throws IOException {</span>
<span class="pc bpc" id="L201" title="1 of 2 branches missed.">      super(MtasMaximumExpandSpanQuery.this, searcher,</span>
<span class="pc" id="L202">          needsScores ? getTermContexts(subWeight) : null);</span>
<span class="fc" id="L203">      this.subWeight = subWeight;</span>
<span class="fc" id="L204">    }</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="L215">      subWeight.extractTermContexts(contexts);</span>
<span class="nc" id="L216">    }</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 Spans getSpans(LeafReaderContext ctx, Postings requiredPostings)
        throws IOException {
<span class="fc" id="L229">      Spans spans = subWeight.getSpans(ctx, requiredPostings);</span>
<span class="pc bpc" id="L230" title="3 of 4 branches missed.">      if (maximumLeft == 0 &amp;&amp; maximumRight == 0) {</span>
<span class="nc" id="L231">        return spans;</span>
      } else {
        try {
          // get leafreader
<span class="fc" id="L235">          LeafReader r = ctx.reader();</span>
          // get delegate
<span class="fc" id="L237">          Boolean hasMethod = true;</span>
<span class="fc bfc" id="L238" title="All 2 branches covered.">          while (hasMethod) {</span>
<span class="fc" id="L239">            hasMethod = false;</span>
<span class="fc" id="L240">            Method[] methods = r.getClass().getMethods();</span>
<span class="fc bfc" id="L241" title="All 2 branches covered.">            for (Method m : methods) {</span>
<span class="pc bpc" id="L242" title="1 of 2 branches missed.">              if (m.getName().equals(METHOD_GET_DELEGATE)) {</span>
<span class="nc" id="L243">                hasMethod = true;</span>
<span class="nc" id="L244">                r = (LeafReader) m.invoke(r, (Object[]) null);</span>
<span class="nc" id="L245">                break;</span>
              }
            }
<span class="fc" id="L248">          } // get fieldsproducer</span>
<span class="fc" id="L249">          Method fpm = r.getClass().getMethod(METHOD_GET_POSTINGS_READER,</span>
              (Class&lt;?&gt;[]) null);
<span class="fc" id="L251">          FieldsProducer fp = (FieldsProducer) fpm.invoke(r, (Object[]) null);</span>
          // get MtasFieldsProducer using terms
<span class="fc" id="L253">          Terms t = fp.terms(field);</span>
<span class="pc bpc" id="L254" title="1 of 2 branches missed.">          if (t == null) {</span>
<span class="nc" id="L255">            return new MtasSpanMatchNoneSpans();</span>
          } else {
<span class="fc" id="L257">            CodecInfo mtasCodecInfo = CodecInfo.getCodecInfoFromTerms(t);</span>
<span class="fc" id="L258">            return new MtasMaximumExpandSpans(MtasMaximumExpandSpanQuery.this,</span>
<span class="fc" id="L259">                mtasCodecInfo, query.getField(), spans);</span>
          }
<span class="nc" id="L261">        } catch (Exception e) {</span>
<span class="nc" id="L262">          throw new IOException(&quot;Can't get reader&quot;, e);</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="L275">      subWeight.extractTerms(terms);</span>
<span class="nc" id="L276">    }</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>