org.elasticsearch.compute.data.FloatVectorBuilder Maven / Gradle / Ivy

Go to download
/*
 * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
 * or more contributor license agreements. Licensed under the Elastic License
 * 2.0; you may not use this file except in compliance with the Elastic License
 * 2.0.
 */

package org.elasticsearch.compute.data;

import java.util.Arrays;

/**
 * Builder for {@link FloatVector}s that grows as needed.
 * This class is generated. Do not edit it.
 */
final class FloatVectorBuilder extends AbstractVectorBuilder implements FloatVector.Builder {

    private float[] values;

    FloatVectorBuilder(int estimatedSize, BlockFactory blockFactory) {
        super(blockFactory);
        int initialSize = Math.max(estimatedSize, 2);
        adjustBreaker(initialSize);
        values = new float[Math.max(estimatedSize, 2)];
    }

    @Override
    public FloatVectorBuilder appendFloat(float value) {
        ensureCapacity();
        values[valueCount] = value;
        valueCount++;
        return this;
    }

    @Override
    protected int elementSize() {
        return Float.BYTES;
    }

    @Override
    protected int valuesLength() {
        return values.length;
    }

    @Override
    protected void growValuesArray(int newSize) {
        values = Arrays.copyOf(values, newSize);
    }

    @Override
    public FloatVector build() {
        finish();
        FloatVector vector;
        if (valueCount == 1) {
            vector = blockFactory.newConstantFloatBlockWith(values[0], 1, estimatedBytes).asVector();
        } else {
            if (values.length - valueCount > 1024 || valueCount < (values.length / 2)) {
                values = Arrays.copyOf(values, valueCount);
            }
            vector = blockFactory.newFloatArrayVector(values, valueCount, estimatedBytes);
        }
        built();
        return vector;
    }
}