com.klibisz.elastiknn.storage.BitBuffer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of models Show documentation

Exact and approximate similarity models used in Elastiknn

There is a newer version: 7.17.6.0

package com.klibisz.elastiknn.storage;

public interface BitBuffer {
    void putOne();
    void putZero();
    byte[] toByteArray();

    class IntBuffer implements BitBuffer {

        private final byte[] prefix;
        private int i = 0;
        private int b = 0;

        public IntBuffer(byte[] prefix) {
            this.prefix = prefix;
        }

        public IntBuffer() {
            this.prefix = new byte[0];
        }

        @Override
        public void putOne() {
            this.b += (1 << this.i);
            this.i += 1;
        }

        @Override
        public void putZero() {
            this.i += 1;
        }

        @Override
        public byte[] toByteArray() {
            byte[] barr = UnsafeSerialization.writeInt(b);
            byte[] res = new byte[prefix.length + barr.length];
            System.arraycopy(prefix, 0, res, 0, prefix.length);
            System.arraycopy(barr, 0, res, prefix.length, barr.length);
            return res;
        }
    }

}