TestSerializer1.java

/*
 * Copyright 2015 MiLaboratory.com
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.milaboratory.primitivio.test;

import com.milaboratory.primitivio.PrimitivI;
import com.milaboratory.primitivio.PrimitivO;
import com.milaboratory.primitivio.Serializer;

public class TestSerializer1 implements Serializer<TestClass1> {
    @Override
    public void write(PrimitivO output, TestClass1 object) {
        if (object.getClass() == TestClass1.class)
            output.writeByte((byte) 1);
        else if (object.getClass() == TestSubClass1.class)
            output.writeByte((byte) 2);
        else
            throw new RuntimeException();
        output.writeInt(object.i);
        output.writeUTF(object.k);
        output.writeInt(object.subObjects.length);
        output.writeReference(object);
        for (TestClass1 subObject : object.subObjects)
            output.writeObject(subObject);
    }

    @Override
    public TestClass1 read(PrimitivI input) {
        byte t = input.readByte();
        int j = input.readInt();
        String s = input.readUTF();
        int count = input.readInt();
        TestClass1 instance;
        if (t == 1)
            instance = new TestClass1(j, s, new TestClass1[count]);
        else if (t == 2)
            instance = new TestSubClass1(j, s, new TestClass1[count]);
        else
            throw new RuntimeException();
        input.readReference(instance);
        for (int i = 0; i < count; i++)
            instance.subObjects[i] = input.readObject(TestClass1.class);
        return instance;
    }

    @Override
    public boolean isReference() {
        return true;
    }

    @Override
    public boolean handlesReference() {
        return true;
    }
}