fix: temp

vibhatha · Aug 1, 2024 · d54ba23 · d54ba23
1 parent 2419c34
commit d54ba23
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/java/c/src/main/java/org/apache/arrow/c/BufferImportTypeVisitor.java b/java/c/src/main/java/org/apache/arrow/c/BufferImportTypeVisitor.java
@@ -28,6 +28,7 @@
 import org.apache.arrow.util.AutoCloseables;
 import org.apache.arrow.util.VisibleForTesting;
 import org.apache.arrow.vector.BaseVariableWidthViewVector;
+import org.apache.arrow.vector.BitVectorHelper;
 import org.apache.arrow.vector.DateDayVector;
 import org.apache.arrow.vector.DateMilliVector;
 import org.apache.arrow.vector.DurationVector;
@@ -118,10 +119,10 @@ private ArrowBuf importFixedBitsWithOffset(ArrowType type, int index, long bitsP
     // Calculate the total capacity needed, including the offset
     final long totalSlots = arrowArrayOffset + fieldNode.getLength();
     final long totalBits = totalSlots * bitsPerSlot;
-    // final long capacity = DataSizeRoundingUtil.divideBy8Ceil(totalBits);
+    final long capacity = DataSizeRoundingUtil.divideBy8Ceil(totalBits);
 
     // Import the buffer with the calculated capacity
-    ArrowBuf buf = importBuffer(type, index, totalBits);
+    ArrowBuf buf = importBuffer(type, index, capacity);
 
     // Calculate the start and end positions in bits
     final long startBit = arrowArrayOffset * bitsPerSlot;
@@ -134,7 +135,17 @@ private ArrowBuf importFixedBitsWithOffset(ArrowType type, int index, long bitsP
 
     if (startByte != endByte) {
       return buf.slice(startByte, endByte - startByte);
-    }
+    } // else {
+      // final ArrowBuf bufCopy = allocator.buffer(buf.capacity());
+      // bufCopy.setZero(0, buf.capacity());
+      //
+      // for (int i = (int) arrowArrayOffset; i < bufCopy.capacity(); i++) {
+      //   if (BitVectorHelper.get(buf, i) == 1) {
+      //     BitVectorHelper.setBit(buf, i);
+      //   }
+      // }
+      // return bufCopy;
+    // }
     return buf;
   }