--- /dev/null
+package org.simantics.databoard.serialization.impl;\r
+\r
+import gnu.trove.map.hash.TObjectIntHashMap;\r
+\r
+import java.io.DataInput;\r
+import java.io.DataOutput;\r
+import java.io.IOException;\r
+import java.util.ArrayList;\r
+import java.util.List;\r
+\r
+import org.simantics.databoard.Bindings;\r
+import org.simantics.databoard.binding.Binding;\r
+import org.simantics.databoard.binding.VariantBinding;\r
+import org.simantics.databoard.binding.error.BindingException;\r
+import org.simantics.databoard.serialization.Serializer;\r
+import org.simantics.databoard.serialization.Serializer.RecursiveSerializer;\r
+import org.simantics.databoard.serialization.SerializerScheme;\r
+import org.simantics.databoard.type.Datatype;\r
+\r
+/**\r
+ * Serializes Variant instances using bindings from Bindings.getBinding for\r
+ * deserialization. These are assumed to be immutable bindings.\r
+ * \r
+ * @author Toni Kalajainen\r
+ */\r
+public class VariantSerializer extends RecursiveSerializer {\r
+ \r
+ VariantBinding binding;\r
+ Serializer datatypeSerializer; \r
+ SerializerScheme scheme;\r
+ \r
+ public VariantSerializer(VariantBinding binding, SerializerScheme scheme) { \r
+ this.binding = binding; \r
+ this.scheme = scheme;\r
+ }\r
+ \r
+ @Override\r
+ public void finalizeConstruction() {\r
+ Binding dataTypeBinding = Bindings.getBindingUnchecked( Datatype.class );\r
+ this.datatypeSerializer = scheme.getSerializerUnchecked(dataTypeBinding);\r
+ }\r
+\r
+ @Override\r
+ public Object deserialize(DataInput in, List<Object> identities) throws IOException { \r
+ try {\r
+ List<Object> typeIdentities = new ArrayList<Object>(1);\r
+ Datatype type = (Datatype) datatypeSerializer.deserialize(in, typeIdentities);\r
+ Binding valueBinding = Bindings.getBinding(type);\r
+ Serializer valueSerializer = scheme.getSerializerUnchecked(valueBinding);\r
+ assertRemainingBytes(in, valueSerializer.getMinSize());\r
+ Object value = valueSerializer.deserialize(in, identities);\r
+ return binding.create(valueBinding, value);\r
+ } catch (BindingException e) {\r
+ throw new IOException( e ); \r
+ }\r
+ }\r
+ \r
+ @Override\r
+ public void deserializeTo(DataInput in, List<Object> identities, Object obj) throws IOException {\r
+ try {\r
+ List<Object> typeIdentities = new ArrayList<Object>(1); \r
+ Datatype type = (Datatype) datatypeSerializer.deserialize(in, typeIdentities);\r
+ Datatype oldType = binding.getContentType(obj);\r
+ \r
+ if (type.equals(oldType)) {\r
+ Binding valueBinding = binding.getContentBinding(obj);\r
+ Serializer valueSerializer = scheme.getSerializerUnchecked(valueBinding);\r
+ Object component = binding.getContent(obj);\r
+// assertRemainingBytes(in, valueSerializer.getMinSize());\r
+ component = valueSerializer.deserializeToTry(in, identities, component);\r
+ binding.setContent(obj, valueBinding, component);\r
+ } else {\r
+ Binding valueBinding = Bindings.getBinding(type);\r
+ Serializer valueSerializer = scheme.getSerializerUnchecked(valueBinding);\r
+// assertRemainingBytes(in, valueSerializer.getMinSize());\r
+ Object component = valueSerializer.deserialize(in, identities);\r
+ binding.setContent(obj, valueBinding, component);\r
+ }\r
+ } catch (BindingException e) {\r
+ throw new IOException( e ); \r
+ }\r
+ }\r
+\r
+ @Override\r
+ public void skip(DataInput in, List<Object> identities) throws IOException {\r
+ List<Object> typeIdentities = new ArrayList<Object>(1); \r
+ Datatype type = (Datatype) datatypeSerializer.deserialize(in, typeIdentities); \r
+ Binding valueBinding = Bindings.getBinding(type);\r
+ Serializer valueSerializer = scheme.getSerializerUnchecked(valueBinding);\r
+ valueSerializer.skip(in, identities);\r
+ } \r
+ \r
+ @Override\r
+ public void serialize(DataOutput out, TObjectIntHashMap<Object> identities, Object variant) throws IOException {\r
+ try {\r
+ TObjectIntHashMap<Object> typeIdentities = new TObjectIntHashMap<Object>(1);\r
+ Datatype type = binding.getContentType(variant);\r
+ datatypeSerializer.serialize(out, typeIdentities, type); \r
+ \r
+ Binding valueBinding = binding.getContentBinding(variant);\r
+ Object value = binding.getContent(variant, valueBinding);\r
+ \r
+ Serializer valueSerializer = scheme.getSerializerUnchecked(valueBinding);\r
+ valueSerializer.serialize(out, identities, value);\r
+ } catch (BindingException e) {\r
+ throw new IOException( e ); \r
+ }\r
+ }\r
+\r
+ @Override\r
+ public Integer getConstantSize() { \r
+ return null;\r
+ }\r
+\r
+ @Override\r
+ public int getSize(Object variant, TObjectIntHashMap<Object> identities) throws IOException \r
+ {\r
+ try {\r
+ TObjectIntHashMap<Object> typeIdentities = new TObjectIntHashMap<Object>(1);\r
+ Datatype type = binding.getContentType(variant);\r
+ int size = datatypeSerializer.getSize(type, typeIdentities);\r
+ \r
+ Binding valueBinding = binding.getContentBinding(variant);\r
+ Object value = binding.getContent(variant, valueBinding);\r
+ \r
+ Serializer valueSerializer = scheme.getSerializerUnchecked(valueBinding);\r
+ size += valueSerializer.getSize(value, identities);\r
+ return size;\r
+ } catch (BindingException e) {\r
+ throw new IOException( e ); \r
+ }\r
+ }\r
+ \r
+ @Override\r
+ public int getMinSize() {\r
+ return datatypeSerializer.getMinSize();\r
+ }\r
+ \r
+}
\ No newline at end of file