com.jfirer.fse.serializer.array.DoubleArraySerializer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of Fse Show documentation
a high speed serilaize library
The newest version!
package com.jfirer.fse.serializer.array;

import com.jfirer.fse.*;

public class DoubleArraySerializer extends CycleFlagSerializer implements FseSerializer
{
    private final boolean primitive;

    public DoubleArraySerializer(boolean primitive)
    {
        this.primitive = primitive;
    }

    @Override
    public void init(Class type, SerializerFactory serializerFactory)
    {
    }

    @Override
    public void writeToBytes(Object o, int classIndex, InternalByteArray byteArray, FseContext fseContext, int depth)
    {
        byteArray.writeVarInt(classIndex);
        writeElement(o, byteArray);
    }

    private void writeElement(Object o, InternalByteArray byteArray)
    {
        if (primitive)
        {
            double[] array = (double[]) o;
            byteArray.writePositive(array.length );
            for (double i : array)
            {
                byteArray.writeDouble(i);
            }
        }
        else
        {
            Double[] array = (Double[]) o;
            byteArray.writePositive(array.length );
            for (Double each : array)
            {
                if (each == null)
                {
                    byteArray.put((byte) 0);
                }
                else
                {
                    byteArray.put((byte) 1);
                    byteArray.writeDouble(each);
                }
            }
        }
    }

    @Override
    public Object readBytes(InternalByteArray byteArray, FseContext fseContext)
    {
        int len = byteArray.readPositive();
        return readElement(byteArray, len);
    }

    private Object readElement(InternalByteArray byteArray, int len)
    {
        if (primitive)
        {
            double[] array = new double[len];
            for (int i = 0; i < len; i++)
            {
                array[i] = byteArray.readDouble();
            }
            return array;
        }
        else
        {
            Double[] array = new Double[len];
            for (int i = 0; i < len; i++)
            {
                if (byteArray.get() == 0)
                {
                    array[i] = null;
                }
                else
                {
                    array[i] = Double.valueOf(byteArray.readDouble());
                }
            }
            return array;
        }
    }

}