com.flybotix.hfr.codex.encode.CompressedEncoder Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of highfrequencyrobots Show documentation

A way for wifi robots and IoT devices to quickly send compressed data arrays across a network.

There is a newer version: 2019.1.11

package com.flybotix.hfr.codex.encode;

import java.nio.ByteBuffer;
import java.util.Arrays;
import java.util.BitSet;

import com.flybotix.hfr.codex.CodexHash;
import com.flybotix.hfr.codex.CodexOf;
import com.flybotix.hfr.codex.Codex;
import com.flybotix.hfr.util.log.ILog;
import com.flybotix.hfr.util.log.Logger;

public class CompressedEncoder  & CodexOf> extends DefaultEncoder {
  private static final ILog mLog = Logger.createLog(CompressedEncoder.class);

  public CompressedEncoder(Class pEnum, IEncoderProperties pProps) {
    super(pEnum, pProps);
  }
  
  public AEncoder createClone() {
    return new CompressedEncoder(super.getEnum(), mProps);
  }

  @Override
  protected boolean isCompressed() {
    return true;
  }

  @Override
  protected Codex decodeImpl(ByteBuffer pData) {
    int length = pData.remaining();
    int initPos = pData.position();

    // reference BitSet.toByteArray() for this calculation
    byte[] bsarray = new byte[mBitSetByteLength];
    pData.get(bsarray);
    BitSet hash = BitSet.valueOf(bsarray);
    hash.set(mLength);
    mLog.debug("Bitset: " + Arrays.toString(bsarray));

    // Usually the offset is 0, but with UDP connections it's non-zero.
    pData.position(bsarray.length + initPos + pData.arrayOffset());
    V[] decoded = mProps.generateEmptyArray((length - bsarray.length) / mProps.sizeOfSingle(), false);
    for(int i = 0; i < decoded.length; i++) {
      decoded[i] = mProps.decodeSingle(pData);
    }

//    Codex result = new Codex(this);
    Codex result = new Codex(getDefaultValue(), super.getEnum());
    int dataidx = 0;
    for(int e = 0; e < mLength; e++) {
      if(hash.get(e)) {
        result.set(e, decoded[dataidx]);
        dataidx++;
      }
    }
    return result;
  }

  @Override
  protected byte[] encodeImpl(Codex pData) {
    CodexHash hash = pData.hash();
    byte[] bsbytes = hash.getBitSet().toByteArray();
    mLog.debug("Bitset: " + Arrays.toString(bsbytes));
    ByteBuffer bb = ByteBuffer.allocate(bsbytes.length + mProps.sizeOfSingle() * hash.getNonNullCount());
    bb.put(bsbytes);
    for(int e = 0; e < mLength; e++) {
      if(pData.get(e) != null) {
        mProps.encodeSingle(bb, pData.get(e));
      }
    }
    return  bb.array();
  }

}