@Override public Block apply(Block block) { if (elementCoercer == null) { return block; } ColumnarArray arrayBlock = toColumnarArray(block); Block elementsBlock = elementCoercer.apply(arrayBlock.getElementsBlock()); boolean[] valueIsNull = new boolean[arrayBlock.getPositionCount()]; int[] offsets = new int[arrayBlock.getPositionCount() + 1]; for (int i = 0; i < arrayBlock.getPositionCount(); i++) { valueIsNull[i] = arrayBlock.isNull(i); offsets[i + 1] = offsets[i] + arrayBlock.getLength(i); } return ArrayBlock.fromElementBlock(arrayBlock.getPositionCount(), Optional.of(valueIsNull), offsets, elementsBlock); } }
private static <T> void assertColumnarArray(Block block, T[] expectedValues) { ColumnarArray columnarArray = toColumnarArray(block); assertEquals(columnarArray.getPositionCount(), expectedValues.length); Block elementsBlock = columnarArray.getElementsBlock(); int elementsPosition = 0; for (int position = 0; position < expectedValues.length; position++) { T expectedArray = expectedValues[position]; assertEquals(columnarArray.isNull(position), expectedArray == null); assertEquals(columnarArray.getLength(position), expectedArray == null ? 0 : Array.getLength(expectedArray)); for (int i = 0; i < columnarArray.getLength(position); i++) { Object expectedElement = Array.get(expectedArray, i); assertBlockPosition(elementsBlock, elementsPosition, expectedElement); elementsPosition++; } } }
private void writeColumnarArray(ColumnarArray columnarArray) { // write nulls and lengths for (int position = 0; position < columnarArray.getPositionCount(); position++) { boolean present = !columnarArray.isNull(position); presentStream.writeBoolean(present); if (present) { nonNullValueCount++; lengthStream.writeLong(columnarArray.getLength(position)); } } // write element values Block elementsBlock = columnarArray.getElementsBlock(); if (elementsBlock.getPositionCount() > 0) { elementWriter.writeBlock(elementsBlock); } }
private void writeColumnarArray(ColumnarArray columnarArray) { // write nulls and lengths for (int position = 0; position < columnarArray.getPositionCount(); position++) { boolean present = !columnarArray.isNull(position); presentStream.writeBoolean(present); if (present) { nonNullValueCount++; lengthStream.writeLong(columnarArray.getLength(position)); } } // write element values Block elementsBlock = columnarArray.getElementsBlock(); if (elementsBlock.getPositionCount() > 0) { elementWriter.writeBlock(elementsBlock); } }