Add fillWith() to FloatArray

Author: carlosdelest

server/src/main/java/org/elasticsearch/common/util/BigArrays.java

@@ -426,6 +426,19 @@ public void set(long index, byte[] buf, int offset, int len) {
             assert index >= 0 && index < size();
             System.arraycopy(buf, offset << 2, array, (int) index << 2, len << 2);
         }
+
+        @Override
+        public void fillWith(StreamInput in) throws IOException {
+            int numBytes = in.readVInt();
+            in.readBytes(array, 0, numBytes);
+        }
+
+        @Override
+        public void writeTo(StreamOutput out) throws IOException {
+            int size = (int) size();
+            out.writeVInt(size * 8);
+            out.write(array, 0, size * Double.BYTES);
+        }
     }
 
     private static class ObjectArrayWrapper<T> extends AbstractArrayWrapper implements ObjectArray<T> {

server/src/main/java/org/elasticsearch/common/util/BigFloatArray.java

@@ -9,10 +9,15 @@
 
 package org.elasticsearch.common.util;
 
+import org.elasticsearch.common.io.stream.StreamInput;
+import org.elasticsearch.common.io.stream.StreamOutput;
+
+import java.io.IOException;
 import java.lang.invoke.MethodHandles;
 import java.lang.invoke.VarHandle;
 import java.nio.ByteOrder;
 
+import static org.elasticsearch.common.util.BigLongArray.writePages;
 import static org.elasticsearch.common.util.PageCacheRecycler.FLOAT_PAGE_SIZE;
 
 /**
@@ -94,4 +99,14 @@ private static int pageIdx(long index) {
     private static int idxInPage(long index) {
         return (int) (index & FLOAT_PAGE_SIZE - 1);
     }
+
+    @Override
+    public void fillWith(StreamInput in) throws IOException {
+        readPages(in);
+    }
+
+    @Override
+    public void writeTo(StreamOutput out) throws IOException {
+        writePages(out, size, pages, Float.BYTES);
+    }
 }

server/src/main/java/org/elasticsearch/common/util/FloatArray.java

@@ -9,18 +9,22 @@
 
 package org.elasticsearch.common.util;
 
+import org.elasticsearch.common.io.stream.StreamInput;
+import org.elasticsearch.common.io.stream.Writeable;
+
+import java.io.IOException;
+
 /**
  * Abstraction of an array of double values.
  */
-public interface FloatArray extends BigArray {
-
+public interface FloatArray extends BigArray, Writeable {
     /**
      * Get an element given its index.
      */
     float get(long index);
 
     /**
-     * Set a value at the given index.
+     * Set a value at the given index
      */
     void set(long index, float value);
 
@@ -33,4 +37,6 @@ public interface FloatArray extends BigArray {
      * Bulk set.
      */
     void set(long index, byte[] buf, int offset, int len);
+
+    void fillWith(StreamInput in) throws IOException;
 }

server/src/main/java/org/elasticsearch/index/mapper/BlockLoader.java

@@ -554,6 +554,13 @@ interface IntBuilder extends Builder {
         IntBuilder appendInt(int value);
     }
 
+    interface DenseVectorBuilder extends Builder {
+        /**
+         * Appends a double to the current entry.
+         */
+        DenseVectorBuilder appendDenseVector(float[] value);
+    }
+
     /**
      * Specialized builder for collecting dense arrays of long values.
      */

test/framework/src/main/java/org/elasticsearch/common/util/MockBigArrays.java

@@ -604,6 +604,11 @@ public void fill(long fromIndex, long toIndex, float value) {
             in.fill(fromIndex, toIndex, value);
         }
 
+        @Override
+        public void fillWith(StreamInput streamInput) throws IOException {
+            in.fillWith(streamInput);
+        }
+
         @Override
         public void set(long index, byte[] buf, int offset, int len) {
             in.set(index, buf, offset, len);
@@ -613,6 +618,11 @@ public void set(long index, byte[] buf, int offset, int len) {
         public Collection<Accountable> getChildResources() {
             return Collections.singleton(Accountables.namedAccountable("delegate", in));
         }
+
+        @Override
+        public void writeTo(StreamOutput out) throws IOException {
+            in.writeTo(out);
+        }
     }
 
     private class DoubleArrayWrapper extends AbstractArrayWrapper implements DoubleArray {