<?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="../.resources/report.css" type="text/css"/><link rel="shortcut icon" href="../.resources/report.gif" type="image/gif"/><title>MtasPayloadEncoder.java</title><link rel="stylesheet" href="../.resources/prettify.css" type="text/css"/><script type="text/javascript" src="../.resources/prettify.js"></script></head><body onload="window['PR_TAB_WIDTH']=4;prettyPrint()"><div class="breadcrumb" id="breadcrumb"><span class="info"><a href="../.sessions.html" class="el_session">Sessions</a></span><a href="../index.html" class="el_report">MTAS</a> > <a href="index.source.html" class="el_package">mtas.codec.payload</a> > <span class="el_source">MtasPayloadEncoder.java</span></div><h1>MtasPayloadEncoder.java</h1><pre class="source lang-java linenums">package mtas.codec.payload;
import java.io.IOException;
import java.util.Arrays;
import mtas.analysis.token.MtasPosition;
import mtas.analysis.token.MtasToken;
import mtas.analysis.token.MtasTokenString;
import org.apache.lucene.util.BytesRef;
/**
* The Class MtasPayloadEncoder.
*/
/**
* The Class MtasPayloadEncoder.
*/
public class MtasPayloadEncoder {
/** The mtas token. */
private MtasToken mtasToken;
/** The byte stream. */
private MtasBitOutputStream byteStream;
/** The encoding flags. */
private int encodingFlags;
/** The encode payload. */
public final static int ENCODE_PAYLOAD = 1;
/** The encode offset. */
public final static int ENCODE_OFFSET = 2;
/** The encode realoffset. */
public final static int ENCODE_REALOFFSET = 4;
/** The encode parent. */
public final static int ENCODE_PARENT = 8;
/** The encode default. */
public final static int ENCODE_DEFAULT = ENCODE_PAYLOAD | ENCODE_OFFSET
| ENCODE_PARENT;
/** The encode all. */
public final static int ENCODE_ALL = ENCODE_PAYLOAD | ENCODE_OFFSET
| ENCODE_REALOFFSET | ENCODE_PARENT;
/**
* Instantiates a new mtas payload encoder.
*
* @param token
* the token
* @param flags
* the flags
*/
<span class="fc" id="L58"> public MtasPayloadEncoder(MtasToken token, int flags) {</span>
<span class="fc" id="L59"> mtasToken = token;</span>
<span class="fc" id="L60"> byteStream = new MtasBitOutputStream();</span>
<span class="fc" id="L61"> encodingFlags = flags;</span>
<span class="fc" id="L62"> }</span>
/**
* Instantiates a new mtas payload encoder.
*
* @param token
* the token
*/
public MtasPayloadEncoder(MtasToken token) {
<span class="nc" id="L71"> this(token, ENCODE_DEFAULT);</span>
<span class="nc" id="L72"> }</span>
/**
* Gets the payload.
*
* @return the payload
* @throws IOException
* Signals that an I/O exception has occurred.
*/
public BytesRef getPayload() throws IOException {
// initial bits - position
<span class="fc bfc" id="L84" title="All 2 branches covered."> if (mtasToken.checkPositionType(MtasPosition.POSITION_SINGLE)) {</span>
<span class="fc" id="L85"> byteStream.writeBit(0);</span>
<span class="fc" id="L86"> byteStream.writeBit(0);</span>
<span class="fc bfc" id="L87" title="All 2 branches covered."> } else if (mtasToken.checkPositionType(MtasPosition.POSITION_RANGE)) {</span>
<span class="fc" id="L88"> byteStream.writeBit(1);</span>
<span class="fc" id="L89"> byteStream.writeBit(0);</span>
<span class="pc bpc" id="L90" title="1 of 2 branches missed."> } else if (mtasToken.checkPositionType(MtasPosition.POSITION_SET)) {</span>
<span class="fc" id="L91"> byteStream.writeBit(0);</span>
<span class="fc" id="L92"> byteStream.writeBit(1);</span>
} else {
<span class="nc" id="L94"> byteStream.writeBit(1);</span>
<span class="nc" id="L95"> byteStream.writeBit(1);</span>
}
// initial bits - offset
<span class="pc bpc" id="L98" title="1 of 2 branches missed."> if ((encodingFlags & ENCODE_OFFSET) == ENCODE_OFFSET</span>
<span class="nc bnc" id="L99" title="All 2 branches missed."> && mtasToken.checkOffset()) {</span>
<span class="nc" id="L100"> byteStream.writeBit(1);</span>
} else {
<span class="fc" id="L102"> byteStream.writeBit(0);</span>
}
// initial bits - realOffset
<span class="pc bpc" id="L105" title="1 of 2 branches missed."> if ((encodingFlags & ENCODE_REALOFFSET) == ENCODE_REALOFFSET</span>
<span class="nc bnc" id="L106" title="All 2 branches missed."> && mtasToken.checkRealOffset()) {</span>
<span class="nc" id="L107"> byteStream.writeBit(1);</span>
} else {
<span class="fc" id="L109"> byteStream.writeBit(0);</span>
}
// initial bits - parentId
<span class="pc bpc" id="L112" title="1 of 2 branches missed."> if ((encodingFlags & ENCODE_PARENT) == ENCODE_PARENT</span>
<span class="fc bfc" id="L113" title="All 2 branches covered."> && mtasToken.checkParentId()) {</span>
<span class="fc" id="L114"> byteStream.writeBit(1);</span>
} else {
<span class="fc" id="L116"> byteStream.writeBit(0);</span>
}
// initial bits - original payload
<span class="pc bpc" id="L119" title="1 of 2 branches missed."> if ((encodingFlags & ENCODE_PAYLOAD) == ENCODE_PAYLOAD</span>
<span class="nc bnc" id="L120" title="All 2 branches missed."> && mtasToken.getPayload() != null) {</span>
<span class="nc" id="L121"> byteStream.writeBit(1);</span>
} else {
<span class="fc" id="L123"> byteStream.writeBit(0);</span>
}
<span class="pc bpc" id="L125" title="1 of 2 branches missed."> if (mtasToken.getType().equals(MtasTokenString.TOKEN_TYPE)) {</span>
<span class="fc" id="L126"> byteStream.writeBit(0);</span>
} else {
// to add other token types later on
<span class="nc" id="L129"> byteStream.writeBit(1);</span>
}
// add id (EliasGammaCoding)
<span class="fc" id="L132"> byteStream.writeEliasGammaCodingNonNegativeInteger(mtasToken.getId());</span>
// add position info (EliasGammaCoding)
<span class="fc bfc" id="L134" title="All 2 branches covered."> if (mtasToken.checkPositionType(MtasPosition.POSITION_SINGLE)) {</span>
// do nothing
<span class="fc bfc" id="L136" title="All 2 branches covered."> } else if (mtasToken.checkPositionType(MtasPosition.POSITION_RANGE)) {</span>
// write length
<span class="fc" id="L138"> byteStream.writeEliasGammaCodingPositiveInteger(</span>
<span class="fc" id="L139"> 1 + mtasToken.getPositionEnd() - mtasToken.getPositionStart());</span>
<span class="pc bpc" id="L140" title="1 of 2 branches missed."> } else if (mtasToken.checkPositionType(MtasPosition.POSITION_SET)) {</span>
// write number of positions
<span class="fc" id="L142"> int[] positionList = mtasToken.getPositions();</span>
<span class="fc" id="L143"> byteStream.writeEliasGammaCodingPositiveInteger(positionList.length);</span>
<span class="fc" id="L144"> int previousPosition = positionList[0];</span>
<span class="fc bfc" id="L145" title="All 2 branches covered."> for (int i = 1; i < positionList.length; i++) {</span>
<span class="fc" id="L146"> byteStream.writeEliasGammaCodingPositiveInteger(</span>
positionList[i] - previousPosition);
<span class="fc" id="L148"> previousPosition = positionList[i];</span>
}
} else {
// do nothing
}
// add offset info (EliasGammaCoding)
<span class="pc bpc" id="L154" title="1 of 2 branches missed."> if ((encodingFlags & ENCODE_OFFSET) == ENCODE_OFFSET</span>
<span class="nc bnc" id="L155" title="All 2 branches missed."> && mtasToken.checkOffset()) {</span>
<span class="nc" id="L156"> byteStream</span>
<span class="nc" id="L157"> .writeEliasGammaCodingNonNegativeInteger(mtasToken.getOffsetStart());</span>
<span class="nc" id="L158"> byteStream.writeEliasGammaCodingPositiveInteger(</span>
<span class="nc" id="L159"> 1 + mtasToken.getOffsetEnd() - mtasToken.getOffsetStart());</span>
}
// add realOffset info (EliasGammaCoding)
<span class="pc bpc" id="L162" title="1 of 2 branches missed."> if ((encodingFlags & ENCODE_REALOFFSET) == ENCODE_REALOFFSET</span>
<span class="nc bnc" id="L163" title="All 2 branches missed."> && mtasToken.checkRealOffset()) {</span>
<span class="nc bnc" id="L164" title="All 2 branches missed."> if ((encodingFlags & ENCODE_OFFSET) == ENCODE_OFFSET</span>
<span class="nc bnc" id="L165" title="All 2 branches missed."> && mtasToken.checkOffset()) {</span>
<span class="nc" id="L166"> byteStream.writeEliasGammaCodingInteger(</span>
<span class="nc" id="L167"> mtasToken.getRealOffsetStart() - mtasToken.getOffsetStart());</span>
<span class="nc" id="L168"> byteStream.writeEliasGammaCodingPositiveInteger(</span>
<span class="nc" id="L169"> 1 + mtasToken.getRealOffsetEnd() - mtasToken.getRealOffsetStart());</span>
} else {
<span class="nc" id="L171"> byteStream.writeEliasGammaCodingNonNegativeInteger(</span>
<span class="nc" id="L172"> mtasToken.getRealOffsetStart());</span>
<span class="nc" id="L173"> byteStream.writeEliasGammaCodingPositiveInteger(</span>
<span class="nc" id="L174"> 1 + mtasToken.getRealOffsetEnd() - mtasToken.getRealOffsetStart());</span>
}
}
// add parent info (EliasGammaCoding)
<span class="pc bpc" id="L178" title="1 of 2 branches missed."> if ((encodingFlags & ENCODE_PARENT) == ENCODE_PARENT</span>
<span class="fc bfc" id="L179" title="All 2 branches covered."> && mtasToken.checkParentId()) {</span>
<span class="fc" id="L180"> byteStream.writeEliasGammaCodingInteger(</span>
<span class="fc" id="L181"> mtasToken.getParentId() - mtasToken.getId());</span>
}
// add minimal number of zero-bits to get round number of bytes
<span class="fc" id="L184"> byteStream.createByte();</span>
// finally add original payload bytes
<span class="pc bpc" id="L186" title="1 of 2 branches missed."> if ((encodingFlags & ENCODE_PAYLOAD) == ENCODE_PAYLOAD</span>
<span class="nc bnc" id="L187" title="All 2 branches missed."> && mtasToken.getPayload() != null) {</span>
<span class="nc" id="L188"> BytesRef payload = mtasToken.getPayload();</span>
<span class="nc" id="L189"> byteStream.write(Arrays.copyOfRange(payload.bytes, payload.offset,</span>
(payload.offset + payload.length)));
}
// construct new payload
<span class="fc" id="L193"> return new BytesRef(byteStream.toByteArray());</span>
}
}
</pre><div class="footer"><span class="right">Created with <a href="http://www.eclemma.org/jacoco">JaCoCo</a> 0.7.5.201505241946</span></div></body></html>