de.gsi.serializer.spi.iobuffer.FieldMultiArrayHelper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of microservice Show documentation
ZeroMQ and REST-based micro-service and IoSerialiser implementation.
There is a newer version: 11.2.7
package de.gsi.serializer.spi.iobuffer;

import de.gsi.dataset.spi.utils.MultiArray;
import de.gsi.dataset.spi.utils.MultiArrayBoolean;
import de.gsi.dataset.spi.utils.MultiArrayByte;
import de.gsi.dataset.spi.utils.MultiArrayChar;
import de.gsi.dataset.spi.utils.MultiArrayDouble;
import de.gsi.dataset.spi.utils.MultiArrayFloat;
import de.gsi.dataset.spi.utils.MultiArrayInt;
import de.gsi.dataset.spi.utils.MultiArrayLong;
import de.gsi.dataset.spi.utils.MultiArrayObject;
import de.gsi.dataset.spi.utils.MultiArrayShort;
import de.gsi.serializer.DataType;
import de.gsi.serializer.FieldDescription;
import de.gsi.serializer.FieldSerialiser;
import de.gsi.serializer.IoClassSerialiser;
import de.gsi.serializer.IoSerialiser;
import de.gsi.serializer.spi.WireDataFieldDescription;

/**
 * helper class to register default serialiser for MultiArray types
 *
 * @author Alexander Krimm
 */
public class FieldMultiArrayHelper {
    private FieldMultiArrayHelper() {
        // utility class
    }

    public static  MultiArray getMultiArray(final IoSerialiser serialiser, final MultiArray dst, final DataType type) {
        final int[] dims = serialiser.getArraySizeDescriptor();
        int n = 1;
        for (int ni : dims) {
            n *= ni;
        }
        switch (type) {
        case BOOL_ARRAY:
            return (MultiArray) MultiArrayBoolean.wrap(serialiser.getBuffer().getBooleanArray(dst == null ? null : (boolean[]) dst.elements(), n), dims);
        case BYTE_ARRAY:
            return (MultiArray) MultiArrayByte.wrap(serialiser.getBuffer().getByteArray(dst == null ? null : (byte[]) dst.elements(), n), dims);
        case SHORT_ARRAY:
            return (MultiArray) MultiArrayShort.wrap(serialiser.getBuffer().getShortArray(dst == null ? null : (short[]) dst.elements(), n), dims);
        case INT_ARRAY:
            return (MultiArray) MultiArrayInt.wrap(serialiser.getBuffer().getIntArray(dst == null ? null : (int[]) dst.elements(), n), dims);
        case LONG_ARRAY:
            return (MultiArray) MultiArrayLong.wrap(serialiser.getBuffer().getLongArray(dst == null ? null : (long[]) dst.elements(), n), dims);
        case FLOAT_ARRAY:
            return (MultiArray) MultiArrayFloat.wrap(serialiser.getBuffer().getFloatArray(dst == null ? null : (float[]) dst.elements(), n), dims);
        case DOUBLE_ARRAY:
            return (MultiArray) MultiArrayDouble.wrap(serialiser.getBuffer().getDoubleArray(dst == null ? null : (double[]) dst.elements(), n), dims);
        case CHAR_ARRAY:
            return (MultiArray) MultiArrayChar.wrap(serialiser.getBuffer().getCharArray(dst == null ? null : (char[]) dst.elements(), n), dims);
        case STRING_ARRAY:
            return (MultiArray) MultiArrayObject.wrap(serialiser.getBuffer().getStringArray(dst == null ? null : (String[]) dst.elements(), n), dims);
        default:
            throw new IllegalStateException("Unexpected value: " + type);
        }
    }

    public static  void put(final IoSerialiser serialiser, final String fieldName, final MultiArray value) {
        if (value instanceof MultiArrayDouble) {
            final WireDataFieldDescription fieldHeader = serialiser.putFieldHeader(fieldName, DataType.DOUBLE_ARRAY);
            final int nElements = serialiser.putArraySizeDescriptor(value.getDimensions());
            serialiser.getBuffer().putDoubleArray(((MultiArrayDouble) value).elements(), nElements);
            serialiser.updateDataEndMarker(fieldHeader);
        } else if (value instanceof MultiArrayFloat) {
            final WireDataFieldDescription fieldHeader = serialiser.putFieldHeader(fieldName, DataType.FLOAT_ARRAY);
            final int nElements = serialiser.putArraySizeDescriptor(value.getDimensions());
            serialiser.getBuffer().putFloatArray(((MultiArrayFloat) value).elements(), nElements);
            serialiser.updateDataEndMarker(fieldHeader);
        } else if (value instanceof MultiArrayInt) {
            final WireDataFieldDescription fieldHeader = serialiser.putFieldHeader(fieldName, DataType.INT_ARRAY);
            final int nElements = serialiser.putArraySizeDescriptor(value.getDimensions());
            serialiser.getBuffer().putIntArray(((MultiArrayInt) value).elements(), nElements);
            serialiser.updateDataEndMarker(fieldHeader);
        } else if (value instanceof MultiArrayLong) {
            final WireDataFieldDescription fieldHeader = serialiser.putFieldHeader(fieldName, DataType.LONG_ARRAY);
            final int nElements = serialiser.putArraySizeDescriptor(value.getDimensions());
            serialiser.getBuffer().putLongArray(((MultiArrayLong) value).elements(), nElements);
            serialiser.updateDataEndMarker(fieldHeader);
        } else if (value instanceof MultiArrayShort) {
            final WireDataFieldDescription fieldHeader = serialiser.putFieldHeader(fieldName, DataType.SHORT_ARRAY);
            final int nElements = serialiser.putArraySizeDescriptor(value.getDimensions());
            serialiser.getBuffer().putShortArray(((MultiArrayShort) value).elements(), nElements);
            serialiser.updateDataEndMarker(fieldHeader);
        } else if (value instanceof MultiArrayChar) {
            final WireDataFieldDescription fieldHeader = serialiser.putFieldHeader(fieldName, DataType.CHAR_ARRAY);
            final int nElements = serialiser.putArraySizeDescriptor(value.getDimensions());
            serialiser.getBuffer().putCharArray(((MultiArrayChar) value).elements(), nElements);
            serialiser.updateDataEndMarker(fieldHeader);
        } else if (value instanceof MultiArrayByte) {
            final WireDataFieldDescription fieldHeader = serialiser.putFieldHeader(fieldName, DataType.BYTE_ARRAY);
            final int nElements = serialiser.putArraySizeDescriptor(value.getDimensions());
            serialiser.getBuffer().putByteArray(((MultiArrayByte) value).elements(), nElements);
            serialiser.updateDataEndMarker(fieldHeader);
        } else if (value instanceof MultiArrayObject) {
            final WireDataFieldDescription fieldHeader = serialiser.putFieldHeader(fieldName, DataType.STRING_ARRAY);
            final int nElements = serialiser.putArraySizeDescriptor(value.getDimensions());
            serialiser.getBuffer().putStringArray(((MultiArrayObject) value).elements(), nElements);
            serialiser.updateDataEndMarker(fieldHeader);
        } else {
            throw new IllegalArgumentException("Illegal DataType for MultiArray");
        }
    }

    public static  void put(final IoSerialiser serialiser, final FieldDescription fieldDescription, final MultiArray value) {
        final WireDataFieldDescription fieldHeader = serialiser.putFieldHeader(fieldDescription);
        final int nElements = serialiser.putArraySizeDescriptor(value.getDimensions());
        switch (fieldDescription.getDataType()) {
        case BOOL_ARRAY:
            serialiser.getBuffer().putBooleanArray(((MultiArrayBoolean) value).elements(), nElements);
            break;
        case BYTE_ARRAY:
            serialiser.getBuffer().putByteArray(((MultiArrayByte) value).elements(), nElements);
            break;
        case SHORT_ARRAY:
            serialiser.getBuffer().putShortArray(((MultiArrayShort) value).elements(), nElements);
            break;
        case INT_ARRAY:
            serialiser.getBuffer().putIntArray(((MultiArrayInt) value).elements(), nElements);
            break;
        case LONG_ARRAY:
            serialiser.getBuffer().putLongArray(((MultiArrayLong) value).elements(), nElements);
            break;
        case FLOAT_ARRAY:
            serialiser.getBuffer().putFloatArray(((MultiArrayFloat) value).elements(), nElements);
            break;
        case DOUBLE_ARRAY:
            serialiser.getBuffer().putDoubleArray(((MultiArrayDouble) value).elements(), nElements);
            break;
        case CHAR_ARRAY:
            serialiser.getBuffer().putCharArray(((MultiArrayChar) value).elements(), nElements);
            break;
        case STRING_ARRAY:
            serialiser.getBuffer().putStringArray(((MultiArrayObject) value).elements(), nElements);
            break;
        default:
            throw new IllegalStateException("Unexpected value: " + fieldDescription.getDataType());
        }
        serialiser.updateDataEndMarker(fieldHeader);
    }

    /**
     * Registers default serialiser for MultiArray
     *
     * @param serialiser for which the field serialisers should be registered
     */
    public static void register(final IoClassSerialiser serialiser) {
        serialiser.addClassDefinition(new FieldSerialiser<>( //
                (io, obj, field) -> field.getField().set(obj, getMultiArray(io, (MultiArray) field.getField().get(obj), field.getDataType())), // reader
                (io, obj, field) -> getMultiArray(io, (MultiArray) ((field == null) ? obj : field.getField().get(obj)), DataType.DOUBLE_ARRAY), // return
                (io, obj, field) -> put(io, field, (MultiArray) field.getField().get(obj)), // writer
                MultiArrayDouble.class));
        serialiser.addClassDefinition(new FieldSerialiser<>( //
                (io, obj, field) -> field.getField().set(obj, getMultiArray(io, (MultiArray) field.getField().get(obj), field.getDataType())), // reader
                (io, obj, field) -> getMultiArray(io, (MultiArray) ((field == null) ? obj : field.getField().get(obj)), DataType.FLOAT_ARRAY), // return
                (io, obj, field) -> put(io, field, (MultiArray) field.getField().get(obj)), // writer
                MultiArrayFloat.class));
        serialiser.addClassDefinition(new FieldSerialiser<>( //
                (io, obj, field) -> field.getField().set(obj, getMultiArray(io, (MultiArray) field.getField().get(obj), field.getDataType())), // reader
                (io, obj, field) -> getMultiArray(io, (MultiArray) ((field == null) ? obj : field.getField().get(obj)), DataType.INT_ARRAY), // return
                (io, obj, field) -> put(io, field, (MultiArray) field.getField().get(obj)), // writer
                MultiArrayInt.class));
        serialiser.addClassDefinition(new FieldSerialiser<>( //
                (io, obj, field) -> field.getField().set(obj, getMultiArray(io, (MultiArray) field.getField().get(obj), field.getDataType())), // reader
                (io, obj, field) -> getMultiArray(io, (MultiArray) ((field == null) ? obj : field.getField().get(obj)), DataType.LONG_ARRAY), // return
                (io, obj, field) -> put(io, field, (MultiArray) field.getField().get(obj)), // writer
                MultiArrayLong.class));
        serialiser.addClassDefinition(new FieldSerialiser<>( //
                (io, obj, field) -> field.getField().set(obj, getMultiArray(io, (MultiArray) field.getField().get(obj), field.getDataType())), // reader
                (io, obj, field) -> getMultiArray(io, (MultiArray) ((field == null) ? obj : field.getField().get(obj)), DataType.SHORT_ARRAY), // return
                (io, obj, field) -> put(io, field, (MultiArray) field.getField().get(obj)), // writer
                MultiArrayShort.class));
        serialiser.addClassDefinition(new FieldSerialiser<>( //
                (io, obj, field) -> field.getField().set(obj, getMultiArray(io, (MultiArray) field.getField().get(obj), field.getDataType())), // reader
                (io, obj, field) -> getMultiArray(io, (MultiArray) ((field == null) ? obj : field.getField().get(obj)), DataType.BYTE_ARRAY), // return
                (io, obj, field) -> put(io, field, (MultiArray) field.getField().get(obj)), // writer
                MultiArrayByte.class));
        serialiser.addClassDefinition(new FieldSerialiser<>( //
                (io, obj, field) -> field.getField().set(obj, getMultiArray(io, (MultiArray) field.getField().get(obj), field.getDataType())), // reader
                (io, obj, field) -> getMultiArray(io, (MultiArray) ((field == null) ? obj : field.getField().get(obj)), DataType.CHAR_ARRAY), // return
                (io, obj, field) -> put(io, field, (MultiArray) field.getField().get(obj)), // writer
                MultiArrayChar.class));
        serialiser.addClassDefinition(new FieldSerialiser<>( //
                (io, obj, field) -> field.getField().set(obj, getMultiArray(io, (MultiArray) field.getField().get(obj), field.getDataType())), // reader
                (io, obj, field) -> getMultiArray(io, (MultiArray) ((field == null) ? obj : field.getField().get(obj)), DataType.BOOL_ARRAY), // return
                (io, obj, field) -> put(io, field, (MultiArray) field.getField().get(obj)), // writer
                MultiArrayBoolean.class));
        serialiser.addClassDefinition(new FieldSerialiser<>( //
                (io, obj, field) -> field.getField().set(obj, getMultiArray(io, (MultiArray) field.getField().get(obj), field.getDataType())), // reader
                (io, obj, field) -> getMultiArray(io, (MultiArray) ((field == null) ? obj : field.getField().get(obj)), DataType.STRING_ARRAY), // return
                (io, obj, field) -> put(io, field, (MultiArray) field.getField().get(obj)), // writer
                MultiArrayObject.class));
    }
}