ExtWrapMapPoly.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 elements are multiple types, keys are still assumed to be of a single (non-polymorphic) type
//if elements are compound types (i.e., need wrappers), they must be pre-wrapped before invoking this wrapper, because... come on now.
public class ExtWrapMapPoly extends ExternalizableWrapper {
	public ExternalizableWrapper keyType;
	public boolean ordered;
	
	/* serialization */
	
	public ExtWrapMapPoly (HashMap val) {
		this(val, null);
	}
	
	public ExtWrapMapPoly (HashMap val, ExternalizableWrapper keyType) {
		if (val == null) {
			throw new NullPointerException();
		}
		
		this.val = val;
		this.keyType = keyType;
		this.ordered = (val instanceof OrderedHashtable);
	}

	/* deserialization */
	
	public ExtWrapMapPoly () {
		
	}

	public ExtWrapMapPoly (Class keyType) {
		this(keyType, false);
	}
	
	public ExtWrapMapPoly (ExternalizableWrapper keyType) {
		this(keyType, false);
	}
	
	public ExtWrapMapPoly (Class keyType, boolean ordered) {
		this(new ExtWrapBase(keyType), ordered);
	}
	
	public ExtWrapMapPoly (ExternalizableWrapper keyType, boolean ordered) {
		if (keyType == null) {
			throw new NullPointerException();
		}
		
		this.keyType = keyType;
		this.ordered = ordered;
	}
	
	public ExternalizableWrapper clone (Object val) {
		return new ExtWrapMapPoly((HashMap)val, keyType);
	}
	
	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, new ExtWrapTagged(), 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, new ExtWrapTagged(elem));			
		}		
	}

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

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