ExtWrapMap.java

/*
 * Copyright (C) 2009 JavaRosa
 *
 * 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 org.openrosa.client.jr.core.util.externalizable;

import java.io.IOException;
import java.util.HashMap;
import java.util.Iterator;

import org.openrosa.client.java.io.DataInputStream;
import org.openrosa.client.java.io.DataOutputStream;
import org.openrosa.client.jr.core.util.OrderedHashtable;

//map of objects where key and data are all of single (non-polymorphic) type (key and value can be of separate types)
public class ExtWrapMap extends ExternalizableWrapper {
	public ExternalizableWrapper keyType;
	public ExternalizableWrapper dataType;
	public boolean ordered;
	
	/* serialization */
	
	public ExtWrapMap (HashMap val) {
		this(val, null, null);
	}
	
	public ExtWrapMap (HashMap val, ExternalizableWrapper dataType) {
		this(val, null, dataType);
	}
		
	public ExtWrapMap (HashMap val, ExternalizableWrapper keyType, ExternalizableWrapper dataType) {
		if (val == null) {
			throw new NullPointerException();
		}
		
		this.val = val;
		this.keyType = keyType;
		this.dataType = dataType;
		this.ordered = (val instanceof OrderedHashtable);
	}

	/* deserialization */
	
	public ExtWrapMap () {
		
	}

	public ExtWrapMap (Class keyType, Class dataType) {
		this(keyType, dataType, false);
	}

	public ExtWrapMap (Class keyType, ExternalizableWrapper dataType) {
		this(keyType, dataType, false);
	}
	
	public ExtWrapMap (ExternalizableWrapper keyType, ExternalizableWrapper dataType) {
		this(keyType, dataType, false);
	}
	
	public ExtWrapMap (Class keyType, Class dataType, boolean ordered) {
		this(new ExtWrapBase(keyType), new ExtWrapBase(dataType), ordered);
	}

	public ExtWrapMap (Class keyType, ExternalizableWrapper dataType, boolean ordered) {
		this(new ExtWrapBase(keyType), dataType, ordered);
	}
	
	public ExtWrapMap (ExternalizableWrapper keyType, ExternalizableWrapper dataType, boolean ordered) {
		if (keyType == null || dataType == null) {
			throw new NullPointerException();
		}
		
		this.keyType = keyType;
		this.dataType = dataType;
		this.ordered = ordered;
	}
	
	public ExternalizableWrapper clone (Object val) {
		return new ExtWrapMap((HashMap)val, keyType, dataType);
	}
	
	public void readExternal(DataInputStream in, PrototypeFactory pf) throws IOException, DeserializationException {
		HashMap h = ordered ? new OrderedHashtable() : new HashMap();

		long size = ExtUtil.readNumeric(in);
		for (int i = 0; i < size; i++) {
			Object key = ExtUtil.read(in, keyType, pf);
			Object elem = ExtUtil.read(in, dataType, pf);
			h.put(key, elem);
		}
		
		val = h;
	}

	public void writeExternal(DataOutputStream out) throws IOException {
		HashMap h = (HashMap)val;

		ExtUtil.writeNumeric(out, h.size());
		for (Iterator e = h.keySet().iterator(); e.hasNext(); ) {
			Object key = e.next();
			Object elem = h.get(key);
			
			ExtUtil.write(out, keyType == null ? key : keyType.clone(key));
			ExtUtil.write(out, dataType == null ? elem : dataType.clone(elem));			
		}
	}

	public void metaReadExternal (DataInputStream in, PrototypeFactory pf) throws IOException, DeserializationException {
		ordered = ExtUtil.readBool(in);
		keyType = ExtWrapTagged.readTag(in, pf);
		dataType = ExtWrapTagged.readTag(in, pf);
	}

	public void metaWriteExternal (DataOutputStream out) throws IOException {
		HashMap h = (HashMap)val;
		Object keyTagObj, elemTagObj;
		
		keyTagObj = (keyType == null ? (h.size() == 0 ? new Object() : h.keySet().iterator().next()) : keyType);
		elemTagObj = (dataType == null ? (h.size() == 0 ? new Object() : h.entrySet().iterator().next()) : dataType);
		
		ExtUtil.writeBool(out, ordered);
		ExtWrapTagged.writeTag(out, keyTagObj);
		ExtWrapTagged.writeTag(out, elemTagObj);
	}
}