Java的PList类库 Blister.7z

/* * Copyright 2011 Daniel Rendall * * 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 uk.co.sromo.blister; import java.io.ByteArrayOutputStream; import java.io.IOException; import java.nio.ByteBuffer; import java.util.HashSet; import java.util.LinkedHashMap; import java.util.Map; import java.util.Set; import java.util.logging.Logger; /** * TODO: Documentation */ public class BinaryPlistEncoder implements BPVisitor { private final static Logger log = Logger.getLogger(BinaryPlistEncoder.class.getSimpleName()); private final static byte[] EMPTY = new byte[0]; private final Map<Integer, byte[]> objectData = new LinkedHashMap<Integer, byte[]>(); // object ref -> bytes for that object private final Set<Integer> collectionTypes = new HashSet<Integer>(); // object refs of collection types which need special treatment private final Map<BPItem, Integer> objectRefs = new LinkedHashMap<BPItem, Integer>(); // BPItem -> object ref int currentObjectRef = 0; public void visit(BPArray item) { if (!objectRefs.containsKey(item)) { // store this with an empty representation so that it gets a reference number int myRef = store(item, EMPTY); // ensure all the children are stored for(BPItem child : item) { child.accept(this); } // so now we have object references for all children. // we can store the array (temporarily) with 4-byte int offsets. ByteBuffer bb = ByteBuffer.allocate(item.size() * 4 + 1); // identify the object type for later... bb.put((byte) BinaryPlist.ARRAY); for(BPItem child : item) { if (!objectRefs.containsKey(child)) { throw new EncodingException("Expected an object ref for " + child + " after visiting array"); } bb.putInt(objectRefs.get(child)); } // now we store the list of item references - note that this is not the true representation of the array objectData.put(myRef, bb.array()); collectionTypes.add(myRef); } } public void visit(BPBoolean item) { if (!objectRefs.containsKey(item)) { byte[] bytes = new byte[] { item.getValue() ? (byte) BinaryPlist.BOOL_TRUE : (byte) BinaryPlist.BOOL_FALSE}; store(item, bytes); } } public void visit(BPData item) { if (!objectRefs.containsKey(item)) { byte[] itemBytes = item.getData(); byte[] itemHeader = getObjectHeader(BinaryPlist.DATA, itemBytes.length); store(item, itemHeader, itemBytes); } } public void visit(BPDate item) { if (!objectRefs.containsKey(item)) { // TODO - implement this properly byte[] bytes = new byte[] {(byte)BinaryPlist.DATE, 0,0,0,0,0,0,0,0}; store(item, bytes); } } public void visit(BPDict item) { if (!objectRefs.containsKey(item)) { // store this with an empty representation so that it gets a reference number int myRef = store(item, EMPTY); // ensure all the children are stored for(BPString key : item.keySet()) { key.accept(this); BPItem value = item.get(key); value.accept(this); } // so now we have object references for all children. // we can store the array (temporarily) with 4-byte int offsets. ByteBuffer bb = ByteBuffer.allocate(item.size() * 8 + 1); // identify the object type for later... bb.put((byte) BinaryPlist.DICT); for(BPString key : item.keySet()) { if (!objectRefs.containsKey(key)) { throw new EncodingException("Expected an object ref for key " + key + " after visiting dict"); } bb.putInt(objectRefs.get(key)); } for(BPString key : item.keySet()) { BPItem value = item.get(key); if (!objectRefs.containsKey(value)) { throw new EncodingException("Expected an object ref for value " + value + " after visiting dict"); } bb.putInt(objectRefs.get(value)); } // now we store the list of item references - note that this is not the true representation of the array objectData.put(myRef, bb.array()); collectionTypes.add(myRef); } } public void visit(BPInt item) { if (!objectRefs.containsKey(item)) { long lValue = item.getLongValue(); if (lValue < 0 || item.getSize() == BPInt.Size.Long) { store(item, new byte[] {(byte) (BinaryPlist.INT | 0x03), (byte) ((lValue >> 56) & 0xff), (byte) ((lValue >> 48) & 0xff), (byte) ((lValue >> 40) & 0xff), (byte) ((lValue >> 32) & 0xff), (byte) ((lValue >> 24) & 0xff), (byte) ((lValue >> 16) & 0xff), (byte) ((lValue >> 8) & 0xff), (byte) (lValue & 0xff) }); } else { int value = item.getValue(); if (value < 256) { store(item, new byte[] {(byte) (BinaryPlist.INT | 0x00), (byte) value}); } else if (value < 65536) { store(item, new byte[] {(byte) (BinaryPlist.INT | 0x01), (byte) ((value >> 8) & 0xff), (byte) (value & 0xff) }); } else { store(item, new byte[] {(byte) (BinaryPlist.INT | 0x02), (byte) ((value >> 24) & 0xff), (byte) ((value >> 16) & 0xff), (byte) ((value >> 8) & 0xff), (byte) (value & 0xff) }); } } } } public void visit(BPNull item) { if (!objectRefs.containsKey(item)) { store(item, new byte[] {(byte) BinaryPlist.NULL}); } } public void visit(BPReal item) { if (!objectRefs.containsKey(item)) { // TODO - implement this properly store(item, new byte[] {(byte) (BinaryPlist.REAL | 0x00), 0}); } } public void visit(BPSet item) { if (!objectRefs.containsKey(item)) { // store this with an empty representation so that it gets a reference number int myRef = store(item, EMPTY); // ensure all the children are stored for(BPItem child : item) { child.accept(this); } // so now we have object references for all children. // we can store the array (temporarily) with 4-byte int offsets. ByteBuffer bb = ByteBuffer.allocate(item.size() * 4 + 1); // identify the object type for later... bb.put((byte) BinaryPlist.SET); for(BPItem child : item) { if (!objectRefs.containsKey(child)) { throw new EncodingException("Expected an object ref for " + child + " after visiting set"); } bb.putInt(objectRefs.get(child)); } // now