TestBitstreamWriter.java

package org.jcodec.common.io;
import org.junit.Assert;
import org.junit.Test;

import java.lang.System;
import java.nio.ByteBuffer;

public class TestBitstreamWriter {

    @Test
    public void testWrite() throws Exception {
        ByteBuffer buf = ByteBuffer.allocate(1024);
        BitWriter bw = new BitWriter(buf);
        bw.writeNBit(1, 1);
        bw.writeNBit(12345, 16);
        bw.writeNBit(85412, 15);
        bw.writeNBit(452688, 20);
        bw.writeNBit(25274, 15);
        bw.flush();
        buf.flip();
        Assert.assertArrayEquals(
                new byte[] { (byte) Short.parseShort("10011000", 2), (byte) Short.parseShort("00011100", 2),
                        (byte) Short.parseShort("11001101", 2), (byte) Short.parseShort("10100100", 2),
                        (byte) Short.parseShort("01101110", 2), (byte) Short.parseShort("10000101", 2),
                        (byte) Short.parseShort("00001100", 2), (byte) Short.parseShort("01010111", 2),
                        (byte) Short.parseShort("01000000", 2) }, NIOUtils.toArray(buf));
    }

    @Test
    public void testWrite1Bit() throws Exception {
        ByteBuffer buf = ByteBuffer.allocate(1024);
        BitWriter bw = new BitWriter(buf.duplicate());
        bw.write1Bit(1);
        bw.flush();
        // bw.writeNBit(Integer.MAX_VALUE, 32);
        byte[] dst = new byte[4];
        buf.get(dst);
        for (int i = 0; i < dst.length; i++) {
            System.out.println(String.format("%02x", dst[i] & 0xff));
        }
    }
}