us.ihmc.scs2.session.mcap.output.MCAPByteBufferDataOutput Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of scs2-session-logger
SCS2 Session Logger
The newest version!
package us.ihmc.scs2.session.mcap.output;

import java.nio.ByteBuffer;
import java.nio.ByteOrder;

public class MCAPByteBufferDataOutput implements MCAPDataOutput
{
   private static final int DEFAULT_INITIAL_CAPACITY = 8192;
   private static final int DEFAULT_GROWTH_FACTOR = 2;
   private static final boolean DEFAULT_DIRECT_BUFFER = false;

   private ByteBuffer buffer;
   private final int growthFactor;
   private final boolean directBuffer;

   public MCAPByteBufferDataOutput()
   {
      this(DEFAULT_INITIAL_CAPACITY, DEFAULT_GROWTH_FACTOR, DEFAULT_DIRECT_BUFFER);
   }

   public MCAPByteBufferDataOutput(int initialCapacity, int growthFactor, boolean directBuffer)
   {
      this.growthFactor = growthFactor;
      this.directBuffer = directBuffer;
      buffer = newBuffer(initialCapacity);
   }

   private ByteBuffer newBuffer(int capacity)
   {
      ByteBuffer newBuffer = directBuffer ? ByteBuffer.allocateDirect(capacity) : ByteBuffer.allocate(capacity);
      newBuffer.order(ByteOrder.LITTLE_ENDIAN);
      return newBuffer;
   }

   @Override
   public long position()
   {
      return buffer.position();
   }

   @Override
   public void putLong(long value)
   {
      ensureCapacity(Long.BYTES);
      buffer.putLong(value);
   }

   @Override
   public void putInt(int value)
   {
      ensureCapacity(Integer.BYTES);
      buffer.putInt(value);
   }

   @Override
   public void putShort(short value)
   {
      ensureCapacity(Short.BYTES);
      buffer.putShort(value);
   }

   @Override
   public void putByte(byte value)
   {
      ensureCapacity(Byte.BYTES);
      buffer.put(value);
   }

   @Override
   public void putBytes(byte[] bytes, int offset, int length)
   {
      ensureCapacity(length);
      buffer.put(bytes, offset, length);
   }

   @Override
   public void putByteBuffer(ByteBuffer byteBuffer)
   {
      ensureCapacity(byteBuffer.remaining());
      buffer.put(byteBuffer);
   }

   @Override
   public void close()
   {
      buffer.flip();
   }

   private void ensureCapacity(int bytesToWrite)
   {
      if (buffer.remaining() < bytesToWrite)
      {
         int newCapacity = Math.max(buffer.capacity() * growthFactor, buffer.position() + bytesToWrite);
         ByteBuffer newBuffer = newBuffer(newCapacity);
         buffer.flip();
         newBuffer.put(buffer);
         buffer = newBuffer;
      }
   }

   public ByteBuffer getBuffer()
   {
      return buffer;
   }
}