1 package org.djutils.serialization.serializers;
2
3 import java.lang.reflect.Constructor;
4 import java.lang.reflect.InvocationTargetException;
5 import java.util.HashMap;
6 import java.util.Map;
7
8 import org.djunits.unit.SIUnit;
9 import org.djunits.unit.Unit;
10 import org.djunits.unit.scale.IdentityScale;
11 import org.djunits.unit.util.UnitRuntimeException;
12 import org.djunits.value.ValueRuntimeException;
13 import org.djunits.value.storage.StorageType;
14 import org.djunits.value.vfloat.matrix.FloatSIMatrix;
15 import org.djunits.value.vfloat.matrix.base.FloatMatrix;
16 import org.djunits.value.vfloat.matrix.data.FloatMatrixData;
17 import org.djunits.value.vfloat.scalar.base.FloatScalar;
18 import org.djunits.value.vfloat.vector.base.FloatVector;
19 import org.djutils.serialization.EndianUtil;
20 import org.djutils.serialization.FieldTypes;
21 import org.djutils.serialization.SerializationException;
22
23
24
25
26
27
28
29
30
31
32
33
34
35 public class FloatMatrixSerializer<U extends Unit<U>, S extends FloatScalar<U, S>, V extends FloatVector<U, S, V>,
36 M extends FloatMatrix<U, S, V, M>> extends ArrayOrMatrixWithUnitSerializer<U, M>
37 {
38
39 private static final Map<Unit<?>, Constructor<? extends FloatMatrix<?, ?, ?, ?>>> CACHE = new HashMap<>();
40
41
42 public FloatMatrixSerializer()
43 {
44 super(FieldTypes.FLOAT_32_UNIT_MATRIX, "Djunits_FloatMatrix", 2);
45 }
46
47 @Override
48 public int size(final M afm)
49 {
50 return 4 + 4 + 2 + 4 * afm.rows() * afm.cols();
51 }
52
53 @Override
54 public int getElementSize()
55 {
56 return 4;
57 }
58
59 @Override
60 public void serialize(final M afm, final byte[] buffer, final Pointer pointer, final EndianUtil endianUtil)
61 throws SerializationException
62 {
63 endianUtil.encodeInt(afm.rows(), buffer, pointer.getAndIncrement(4));
64 endianUtil.encodeInt(afm.cols(), buffer, pointer.getAndIncrement(4));
65 encodeUnit(afm.getDisplayUnit(), buffer, pointer, endianUtil);
66 for (int i = 0; i < afm.rows(); i++)
67 {
68 for (int j = 0; j < afm.cols(); j++)
69 {
70 endianUtil.encodeFloat(afm.get(i, j).getSI(), buffer, pointer.getAndIncrement(4));
71 }
72 }
73 }
74
75 @Override
76 public M deSerialize(final byte[] buffer, final Pointer pointer, final EndianUtil endianUtil) throws SerializationException
77 {
78 int height = endianUtil.decodeInt(buffer, pointer.getAndIncrement(4));
79 int width = endianUtil.decodeInt(buffer, pointer.getAndIncrement(4));
80 Unit<? extends Unit<?>> unit = getUnit(buffer, pointer, endianUtil);
81 float[][] array = new float[height][width];
82 for (int i = 0; i < height; i++)
83 {
84 for (int j = 0; j < width; j++)
85 {
86 array[i][j] = endianUtil.decodeFloat(buffer, pointer.getAndIncrement(4));
87 }
88 }
89 try
90 {
91 FloatMatrixData fvd = FloatMatrixData.instantiate(array, IdentityScale.SCALE, StorageType.DENSE);
92 return instantiateAnonymous(fvd, unit);
93 }
94 catch (ValueRuntimeException exception)
95 {
96 throw new SerializationException(exception);
97 }
98 }
99
100
101
102
103
104
105
106
107
108
109
110
111
112 @SuppressWarnings("unchecked")
113 public static <U extends Unit<U>, S extends FloatScalar<U, S>, V extends FloatVector<U, S, V>,
114 M extends FloatMatrix<U, S, V, M>> M instantiateAnonymous(final FloatMatrixData data, final Unit<?> unit)
115 {
116 try
117 {
118 Constructor<? extends FloatMatrix<?, ?, ?, ?>> matrixConstructor = CACHE.get(unit);
119 if (matrixConstructor == null)
120 {
121 if (!unit.getClass().getSimpleName().endsWith("Unit"))
122 {
123 throw new ClassNotFoundException("Unit " + unit.getClass().getSimpleName()
124 + " name does noet end with 'Unit'. Cannot find corresponding scalar");
125 }
126 Class<? extends FloatMatrix<?, ?, ?, ?>> matrixClass;
127 if (unit instanceof SIUnit)
128 {
129 matrixClass = FloatSIMatrix.class;
130 }
131 else
132 {
133 matrixClass = (Class<FloatMatrix<?, ?, ?, ?>>) Class.forName("org.djunits.value.vfloat.matrix.Float"
134 + unit.getClass().getSimpleName().replace("Unit", "") + "Matrix");
135 }
136 matrixConstructor = matrixClass.getDeclaredConstructor(FloatMatrixData.class, unit.getClass());
137 CACHE.put(unit, matrixConstructor);
138 }
139 return (M) matrixConstructor.newInstance(data, unit);
140 }
141 catch (ClassNotFoundException | NoSuchMethodException | SecurityException | InstantiationException
142 | IllegalAccessException | IllegalArgumentException | InvocationTargetException exception)
143 {
144 throw new UnitRuntimeException(
145 "Cannot instantiate FloatMatrix of unit " + unit.toString() + ". Reason: " + exception.getMessage());
146 }
147 }
148
149 }