progress on SerializePacked
This commit is contained in:
parent
983b4f4238
commit
0ea0d1f909
2 changed files with 137 additions and 4 deletions
|
@ -11,12 +11,119 @@ public final class PackedOutputStream implements WritableByteChannel {
|
||||||
this.inner = output;
|
this.inner = output;
|
||||||
}
|
}
|
||||||
|
|
||||||
public int write(ByteBuffer src) throws IOException {
|
public int write(ByteBuffer inBuf) throws IOException {
|
||||||
this.inner.getWriteBuffer();
|
int length = inBuf.remaining();
|
||||||
|
ByteBuffer out = this.inner.getWriteBuffer();
|
||||||
|
|
||||||
|
ByteBuffer slowBuffer = ByteBuffer.allocate(20);
|
||||||
|
|
||||||
|
int inPtr = inBuf.position();
|
||||||
|
int inEnd = inPtr + length;
|
||||||
|
while (inPtr < inEnd) {
|
||||||
|
|
||||||
|
|
||||||
|
if (out.remaining() < 10) {
|
||||||
|
//# Oops, we're out of space. We need at least 10
|
||||||
|
//# bytes for the fast path, since we don't
|
||||||
|
//# bounds-check on every byte.
|
||||||
|
|
||||||
|
out = slowBuffer;
|
||||||
|
out.rewind();
|
||||||
|
}
|
||||||
|
|
||||||
|
int tagPos = out.position();
|
||||||
|
out.position(tagPos + 1);
|
||||||
|
|
||||||
|
byte curByte;
|
||||||
|
|
||||||
|
curByte = inBuf.get(inPtr);
|
||||||
|
byte bit0 = (curByte != 0) ? (byte)1 : (byte)0;
|
||||||
|
out.put(curByte);
|
||||||
|
out.position(out.position() - bit0);
|
||||||
|
inPtr += 1;
|
||||||
|
|
||||||
|
curByte = inBuf.get(inPtr);
|
||||||
|
byte bit1 = (curByte != 0) ? (byte)1 : (byte)0;
|
||||||
|
out.put(curByte);
|
||||||
|
out.position(out.position() - bit1);
|
||||||
|
inPtr += 1;
|
||||||
|
|
||||||
|
curByte = inBuf.get(inPtr);
|
||||||
|
byte bit2 = (curByte != 0) ? (byte)1 : (byte)0;
|
||||||
|
out.put(curByte);
|
||||||
|
out.position(out.position() - bit2);
|
||||||
|
inPtr += 1;
|
||||||
|
|
||||||
|
curByte = inBuf.get(inPtr);
|
||||||
|
byte bit3 = (curByte != 0) ? (byte)1 : (byte)0;
|
||||||
|
out.put(curByte);
|
||||||
|
out.position(out.position() - bit3);
|
||||||
|
inPtr += 1;
|
||||||
|
|
||||||
|
curByte = inBuf.get(inPtr);
|
||||||
|
byte bit4 = (curByte != 0) ? (byte)1 : (byte)0;
|
||||||
|
out.put(curByte);
|
||||||
|
out.position(out.position() - bit4);
|
||||||
|
inPtr += 1;
|
||||||
|
|
||||||
|
curByte = inBuf.get(inPtr);
|
||||||
|
byte bit5 = (curByte != 0) ? (byte)1 : (byte)0;
|
||||||
|
out.put(curByte);
|
||||||
|
out.position(out.position() - bit5);
|
||||||
|
inPtr += 1;
|
||||||
|
|
||||||
|
curByte = inBuf.get(inPtr);
|
||||||
|
byte bit6 = (curByte != 0) ? (byte)1 : (byte)0;
|
||||||
|
out.put(curByte);
|
||||||
|
out.position(out.position() - bit6);
|
||||||
|
inPtr += 1;
|
||||||
|
|
||||||
|
curByte = inBuf.get(inPtr);
|
||||||
|
byte bit7 = (curByte != 0) ? (byte)1 : (byte)0;
|
||||||
|
out.put(curByte);
|
||||||
|
out.position(out.position() - bit7);
|
||||||
|
inPtr += 1;
|
||||||
|
|
||||||
|
byte tag = (byte)((bit0 << 0) | (bit1 << 1) | (bit2 << 2) | (bit3 << 3) |
|
||||||
|
(bit4 << 4) | (bit5 << 5) | (bit6 << 6) | (bit7 << 7));
|
||||||
|
|
||||||
|
out.put(tagPos, tag);
|
||||||
|
|
||||||
|
if (tag == 0) {
|
||||||
|
//# An all-zero word is followed by a count of
|
||||||
|
//# consecutive zero words (not including the first
|
||||||
|
//# one).
|
||||||
|
|
||||||
|
inBuf.position(inPtr);
|
||||||
|
|
||||||
|
long inWord = inBuf.getLong();
|
||||||
|
int limit = inEnd;
|
||||||
|
if (limit - inPtr > 255) {
|
||||||
|
limit = inPtr + 255;
|
||||||
|
}
|
||||||
|
while(inBuf.position() < limit && inWord == 0) {
|
||||||
|
inWord = inBuf.getLong();
|
||||||
|
}
|
||||||
|
|
||||||
|
out.put((byte)(inPtr - inBuf.position()));
|
||||||
|
inPtr = inBuf.position();
|
||||||
|
} else if (tag == 0xff) {
|
||||||
|
//# An all-nonzero word is followed by a count of
|
||||||
|
//# consecutive uncompressed words, followed by the
|
||||||
|
//# uncompressed words themselves.
|
||||||
|
|
||||||
|
//# Count the number of consecutive words in the input
|
||||||
|
//# which have no more than a single zero-byte. We look
|
||||||
|
//# for at least two zeros because that's the point
|
||||||
|
//# where our compression scheme becomes a net win.
|
||||||
|
|
||||||
|
|
||||||
// TODO
|
// TODO
|
||||||
|
}
|
||||||
|
|
||||||
return 0;
|
|
||||||
|
}
|
||||||
|
return length;
|
||||||
}
|
}
|
||||||
|
|
||||||
public void close() throws IOException {
|
public void close() throws IOException {
|
||||||
|
|
|
@ -0,0 +1,26 @@
|
||||||
|
package org.capnproto;
|
||||||
|
|
||||||
|
import org.scalatest.FunSuite;
|
||||||
|
import org.scalatest.Matchers._;
|
||||||
|
import java.nio.ByteBuffer;
|
||||||
|
|
||||||
|
class SerializePackedSuite extends FunSuite {
|
||||||
|
|
||||||
|
def expectPacksTo(unpacked : Array[Byte], packed : Array[Byte]) {
|
||||||
|
// ----
|
||||||
|
// write
|
||||||
|
|
||||||
|
val bytes = new Array[Byte](packed.length);
|
||||||
|
val writer = new ArrayOutputStream(ByteBuffer.wrap(bytes));
|
||||||
|
val packedOutputStream = new PackedOutputStream (writer);
|
||||||
|
packedOutputStream.write(ByteBuffer.wrap(unpacked));
|
||||||
|
|
||||||
|
(bytes) should equal (packed);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
test("SimplePacking") {
|
||||||
|
expectPacksTo(Array(), Array());
|
||||||
|
//expectPacksTo(Array(0,0,12,0,0,34,0,0), Array(0x24,12,34));
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in a new issue