159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta/* 259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * Copyright (c) 2009-2010 jMonkeyEngine 359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * All rights reserved. 459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * Redistribution and use in source and binary forms, with or without 659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * modification, are permitted provided that the following conditions are 759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * met: 859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * * Redistributions of source code must retain the above copyright 1059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * notice, this list of conditions and the following disclaimer. 1159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 1259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * * Redistributions in binary form must reproduce the above copyright 1359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * notice, this list of conditions and the following disclaimer in the 1459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * documentation and/or other materials provided with the distribution. 1559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 1659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * * Neither the name of 'jMonkeyEngine' nor the names of its contributors 1759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * may be used to endorse or promote products derived from this software 1859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * without specific prior written permission. 1959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 2059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 2159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 2259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 2359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR 2459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 2559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 2659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 2759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 2859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 2959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 3059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 3159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta */ 3259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 3359b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartapackage com.jme3.network.serializing.serializers; 3459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 3559b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport com.jme3.network.serializing.Serializer; 3659b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport com.jme3.network.serializing.SerializerRegistration; 3759b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.io.IOException; 3859b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.nio.ByteBuffer; 3959b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.util.HashMap; 4059b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.util.Iterator; 4159b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.util.Map; 4259b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.util.Map.Entry; 4359b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.util.Set; 4459b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.util.logging.Level; 4559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 4659b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartapublic class MapSerializer extends Serializer { 4759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 4859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta /* 4959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 5059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Structure: 5159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 5259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta struct Map { 5359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta INT length 5459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta BYTE flags = { 0x01 = all keys have the same type, 5559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 0x02 = all values have the same type } 5659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (flags has 0x01 set) 5759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta SHORT keyType 5859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (flags has 0x02 set) 5959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta SHORT valType 6059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 6159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta struct MapEntry[length] entries { 6259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (flags does not have 0x01 set) 6359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta SHORT keyType 6459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta OBJECT key 6559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 6659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (flags does not have 0x02 set) 6759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta SHORT valType 6859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta OBJECT value 6959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 7059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 7159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 7259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta */ 7359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 7459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta @SuppressWarnings("unchecked") 7559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public <T> T readObject(ByteBuffer data, Class<T> c) throws IOException { 7659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta int length = data.getInt(); 7759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 7859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Map map; 7959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta try { 8059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta map = (Map)c.newInstance(); 8159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } catch (Exception e) { 8259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta log.log(Level.WARNING, "[Serializer][???] Could not determine map type. Using HashMap."); 8359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta map = new HashMap(); 8459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 8559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 8659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (length == 0) return (T)map; 8759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 8859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta int flags = data.get() & 0xff; 8959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta boolean uniqueKeys = (flags & 0x01) == 0; 9059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta boolean uniqueVals = (flags & 0x02) == 0; 9159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 9259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Class keyClazz = null; 9359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Class valClazz = null; 9459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Serializer keySerial = null; 9559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Serializer valSerial = null; 9659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (!uniqueKeys){ 9759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta SerializerRegistration reg = Serializer.readClass(data); 9859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta keyClazz = reg.getType(); 9959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta keySerial = reg.getSerializer(); 10059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 10159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (!uniqueVals){ 10259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta SerializerRegistration reg = Serializer.readClass(data); 10359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta valClazz = reg.getType(); 10459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta valSerial = reg.getSerializer(); 10559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 10659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 10759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta for (int i = 0; i < length; i++){ 10859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Object key; 10959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Object value; 11059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (uniqueKeys){ 11159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta key = Serializer.readClassAndObject(data); 11259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta }else{ 11359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta key = keySerial.readObject(data, keyClazz); 11459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 11559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (uniqueVals){ 11659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta value = Serializer.readClassAndObject(data); 11759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta }else{ 11859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta value = valSerial.readObject(data, valClazz); 11959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 12059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 12159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta map.put(key, value); 12259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 12359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 12459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return (T)map; 12559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 12659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 12759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta @SuppressWarnings("unchecked") 12859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public void writeObject(ByteBuffer buffer, Object object) throws IOException { 12959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Map map = (Map)object; 13059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta int length = map.size(); 13159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 13259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta buffer.putInt(length); 13359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (length == 0) return; 13459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 13559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 13659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Set<Entry> entries = map.entrySet(); 13759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 13859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Iterator<Entry> it = entries.iterator(); 13959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 14059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Entry entry = it.next(); 14159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Class keyClass = entry.getKey().getClass(); 14259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Class valClass = entry.getValue().getClass(); 14359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta while (it.hasNext()) { 14459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta entry = it.next(); 14559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 14659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (entry.getKey().getClass() != keyClass){ 14759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta keyClass = null; 14859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (valClass == null) 14959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta break; 15059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 15159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (entry.getValue().getClass() != valClass){ 15259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta valClass = null; 15359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (keyClass == null) 15459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta break; 15559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 15659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 15759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 15859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta boolean uniqueKeys = keyClass == null; 15959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta boolean uniqueVals = valClass == null; 16059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta int flags = 0; 16159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (!uniqueKeys) flags |= 0x01; 16259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (!uniqueVals) flags |= 0x02; 16359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta buffer.put( (byte) flags ); 16459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 16559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Serializer keySerial = null, valSerial = null; 16659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (!uniqueKeys){ 16759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Serializer.writeClass(buffer, keyClass); 16859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta keySerial = Serializer.getSerializer(keyClass); 16959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 17059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (!uniqueVals){ 17159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Serializer.writeClass(buffer, valClass); 17259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta valSerial = Serializer.getSerializer(valClass); 17359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 17459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 17559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta it = entries.iterator(); 17659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta while (it.hasNext()) { 17759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta entry = it.next(); 17859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (uniqueKeys){ 17959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Serializer.writeClassAndObject(buffer, entry.getKey()); 18059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta }else{ 18159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta keySerial.writeObject(buffer, entry.getKey()); 18259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 18359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (uniqueVals){ 18459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Serializer.writeClassAndObject(buffer, entry.getValue()); 18559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta }else{ 18659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta valSerial.writeObject(buffer, entry.getValue()); 18759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 18859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 18959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 19059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta} 191