org.redkale.convert.ext.DoubleArraySimpledCoder Maven / Gradle / Ivy

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package org.redkale.convert.ext;

import java.util.stream.DoubleStream;
import org.redkale.convert.Reader;
import org.redkale.convert.SimpledCoder;
import org.redkale.convert.Writer;

/**
 * double[] 的SimpledCoder实现
 *
 * 
 * 详情见: https://redkale.org
 *
 * @author zhangjx
 * @param  Reader输入的子类型
 * @param  Writer输出的子类型
 */
public final class DoubleArraySimpledCoder extends SimpledCoder {

    public static final DoubleArraySimpledCoder instance = new DoubleArraySimpledCoder();

    @Override
    public void convertTo(W out, double[] values) {
        if (values == null) {
            out.writeNull();
            return;
        }
        if (out.writeArrayB(values.length, DoubleSimpledCoder.instance, values) < 0) {
            boolean flag = false;
            for (double v : values) {
                if (flag) out.writeArrayMark();
                out.writeDouble(v);
                flag = true;
            }
        }
        out.writeArrayE();
    }

    @Override
    public double[] convertFrom(R in) {
        int len = in.readArrayB(null, DoubleSimpledCoder.instance);
        int contentLength = -1;
        if (len == Reader.SIGN_NULL) return null;
        if (len == Reader.SIGN_NOLENBUTBYTES) {
            contentLength = in.readMemberContentLength(null, DoubleSimpledCoder.instance);
            len = Reader.SIGN_NOLENGTH;
        }
        if (len == Reader.SIGN_NOLENGTH) {
            int size = 0;
            double[] data = new double[8];
            int startPosition = in.position();
            while (in.hasNext(startPosition, contentLength)) {
                if (size >= data.length) {
                    double[] newdata = new double[data.length + 4];
                    System.arraycopy(data, 0, newdata, 0, size);
                    data = newdata;
                }
                data[size++] = in.readDouble();
            }
            in.readArrayE();
            double[] newdata = new double[size];
            System.arraycopy(data, 0, newdata, 0, size);
            return newdata;
        } else {
            double[] values = new double[len];
            for (int i = 0; i < values.length; i++) {
                values[i] = in.readDouble();
            }
            in.readArrayE();
            return values;
        }
    }

    public final static class DoubleStreamSimpledCoder extends SimpledCoder {

        public static final DoubleStreamSimpledCoder instance = new DoubleStreamSimpledCoder();

        @Override
        @SuppressWarnings("unchecked")
        public void convertTo(W out, DoubleStream values) {
            if (values == null) {
                out.writeNull();
                return;
            }
            DoubleArraySimpledCoder.instance.convertTo(out, values.toArray());
        }

        @Override
        @SuppressWarnings("unchecked")
        public DoubleStream convertFrom(R in) {
            double[] value = DoubleArraySimpledCoder.instance.convertFrom(in);
            return value == null ? null : DoubleStream.of(value);
        }

    }
}