@Override public BlockBuilder createFixedSizeBlockBuilder(int positionCount) { return new ShortArrayBlockBuilder(null, positionCount); }
@Override public boolean isNull(int position) { checkReadablePosition(position); return valueIsNull[position]; }
private void checkReadablePosition(int position) { if (position < 0 || position >= getPositionCount()) { throw new IllegalArgumentException("position is not valid"); } } }
@Test public void testBuildingFromShortArrayBlockBuilder() { ShortArrayBlockBuilder blockBuilder = new ShortArrayBlockBuilder(null, 100); populateNullValues(blockBuilder, 100); assertEquals(blockBuilder.build().getEncodingName(), RunLengthBlockEncoding.NAME); }
public ShortArrayBlockBuilder(@Nullable BlockBuilderStatus blockBuilderStatus, int expectedEntries) { this.blockBuilderStatus = blockBuilderStatus; this.initialEntryCount = max(expectedEntries, 1); updateDataSize(); }
@Override public BlockBuilder writeShort(int value) { if (values.length <= positionCount) { growCapacity(); } values[positionCount] = (short) value; hasNonNullValue = true; positionCount++; if (blockBuilderStatus != null) { blockBuilderStatus.addBytes(Byte.BYTES + Short.BYTES); } return this; }
@Override public long getEstimatedDataSizeForStats(int position) { return isNull(position) ? 0 : Short.BYTES; }
private void growCapacity() { int newSize; if (initialized) { newSize = BlockUtil.calculateNewArraySize(values.length); } else { newSize = initialEntryCount; initialized = true; } valueIsNull = Arrays.copyOf(valueIsNull, newSize); values = Arrays.copyOf(values, newSize); updateDataSize(); }
@Override public BlockBuilder appendNull() { if (values.length <= positionCount) { growCapacity(); } valueIsNull[positionCount] = true; hasNullValue = true; positionCount++; if (blockBuilderStatus != null) { blockBuilderStatus.addBytes(Byte.BYTES + Short.BYTES); } return this; }
@Override public long getEstimatedDataSizeForStats(int position) { return isNull(position) ? 0 : Short.BYTES; }
@Override public BlockBuilder newBlockBuilderLike(BlockBuilderStatus blockBuilderStatus) { return new ShortArrayBlockBuilder(blockBuilderStatus, calculateBlockResetSize(positionCount)); }
@Override public short getShort(int position, int offset) { checkReadablePosition(position); if (offset != 0) { throw new IllegalArgumentException("offset must be zero"); } return values[position]; }
@Override public String toString() { StringBuilder sb = new StringBuilder("ShortArrayBlockBuilder{"); sb.append("positionCount=").append(getPositionCount()); sb.append('}'); return sb.toString(); }
public ShortArrayBlockBuilder(@Nullable BlockBuilderStatus blockBuilderStatus, int expectedEntries) { this.blockBuilderStatus = blockBuilderStatus; this.initialEntryCount = max(expectedEntries, 1); updateDataSize(); }
@Override public BlockBuilder writeShort(int value) { if (values.length <= positionCount) { growCapacity(); } values[positionCount] = (short) value; hasNonNullValue = true; positionCount++; if (blockBuilderStatus != null) { blockBuilderStatus.addBytes(Byte.BYTES + Short.BYTES); } return this; }
@Override public BlockBuilder createBlockBuilder(BlockBuilderStatus blockBuilderStatus, int expectedEntries, int expectedBytesPerEntry) { int maxBlockSizeInBytes; if (blockBuilderStatus == null) { maxBlockSizeInBytes = PageBuilderStatus.DEFAULT_MAX_PAGE_SIZE_IN_BYTES; } else { maxBlockSizeInBytes = blockBuilderStatus.getMaxPageSizeInBytes(); } return new ShortArrayBlockBuilder( blockBuilderStatus, Math.min(expectedEntries, maxBlockSizeInBytes / Short.BYTES)); }
@Override public Block getSingleValueBlock(int position) { checkReadablePosition(position); return new ShortArrayBlock( 0, 1, valueIsNull[position] ? new boolean[] {true} : null, new short[] {values[position]}); }
@Override public Block copyRegion(int positionOffset, int length) { checkValidRegion(getPositionCount(), positionOffset, length); if (!hasNonNullValue) { return new RunLengthEncodedBlock(NULL_VALUE_BLOCK, length); } boolean[] newValueIsNull = null; if (hasNullValue) { newValueIsNull = Arrays.copyOfRange(valueIsNull, positionOffset, positionOffset + length); } short[] newValues = Arrays.copyOfRange(values, positionOffset, positionOffset + length); return new ShortArrayBlock(0, length, newValueIsNull, newValues); }
private void growCapacity() { int newSize; if (initialized) { newSize = BlockUtil.calculateNewArraySize(values.length); } else { newSize = initialEntryCount; initialized = true; } valueIsNull = Arrays.copyOf(valueIsNull, newSize); values = Arrays.copyOf(values, newSize); updateDataSize(); }
@Override public BlockBuilder appendNull() { if (values.length <= positionCount) { growCapacity(); } valueIsNull[positionCount] = true; hasNullValue = true; positionCount++; if (blockBuilderStatus != null) { blockBuilderStatus.addBytes(Byte.BYTES + Short.BYTES); } return this; }