Home | History | Annotate | Download | only in serializers 
      1 /*
      2  * Copyright (c) 2009-2010 jMonkeyEngine
      3  * All rights reserved.
      4  *
      5  * Redistribution and use in source and binary forms, with or without
      6  * modification, are permitted provided that the following conditions are
      7  * met:
      8  *
      9  * * Redistributions of source code must retain the above copyright
     10  *   notice, this list of conditions and the following disclaimer.
     11  *
     12  * * Redistributions in binary form must reproduce the above copyright
     13  *   notice, this list of conditions and the following disclaimer in the
     14  *   documentation and/or other materials provided with the distribution.
     15  *
     16  * * Neither the name of 'jMonkeyEngine' nor the names of its contributors
     17  *   may be used to endorse or promote products derived from this software
     18  *   without specific prior written permission.
     19  *
     20  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
     21  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     22  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     23  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
     24  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
     25  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
     26  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
     27  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
     28  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
     29  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
     30  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     31  */
     32 
     33 package com.jme3.network.serializing.serializers;
     34 
     35 import com.jme3.network.Message;
     36 import com.jme3.network.message.GZIPCompressedMessage;
     37 import com.jme3.network.serializing.Serializer;
     38 import java.io.ByteArrayInputStream;
     39 import java.io.ByteArrayOutputStream;
     40 import java.io.IOException;
     41 import java.nio.ByteBuffer;
     42 import java.util.zip.GZIPInputStream;
     43 import java.util.zip.GZIPOutputStream;
     44 
     45 /**
     46  * Serializes GZIP messages.
     47  *
     48  * @author Lars Wesselius
     49  */
     50 public class GZIPSerializer extends Serializer {
     51 
     52     @SuppressWarnings("unchecked")
     53     public <T> T readObject(ByteBuffer data, Class<T> c) throws IOException {
     54         try
     55         {
     56             GZIPCompressedMessage result = new GZIPCompressedMessage();
     57 
     58             byte[] byteArray = new byte[data.remaining()];
     59 
     60             data.get(byteArray);
     61 
     62             GZIPInputStream in = new GZIPInputStream(new ByteArrayInputStream(byteArray));
     63             ByteArrayOutputStream out = new ByteArrayOutputStream();
     64 
     65             byte[] tmp = new byte[9012];
     66             int read;
     67 
     68             while (in.available() > 0 && ((read = in.read(tmp)) > 0)) {
     69                 out.write(tmp, 0, read);
     70             }
     71 
     72             result.setMessage((Message)Serializer.readClassAndObject(ByteBuffer.wrap(out.toByteArray())));
     73             return (T)result;
     74         }
     75         catch (Exception e) {
     76             e.printStackTrace();
     77             throw new IOException(e.toString());
     78         }
     79     }
     80 
     81     public void writeObject(ByteBuffer buffer, Object object) throws IOException {
     82         if (!(object instanceof GZIPCompressedMessage)) return;
     83         Message message = ((GZIPCompressedMessage)object).getMessage();
     84 
     85         ByteBuffer tempBuffer = ByteBuffer.allocate(512000);
     86         Serializer.writeClassAndObject(tempBuffer, message);
     87 
     88         ByteArrayOutputStream byteArrayOutput = new ByteArrayOutputStream();
     89         GZIPOutputStream gzipOutput = new GZIPOutputStream(byteArrayOutput);
     90 
     91         gzipOutput.write(tempBuffer.array());
     92         gzipOutput.flush();
     93         gzipOutput.finish();
     94         gzipOutput.close();
     95 
     96         buffer.put(byteArrayOutput.toByteArray());
     97     }
     98 }
     99