01: /*
02: * Licensed to the Apache Software Foundation (ASF) under one
03: * or more contributor license agreements. See the NOTICE file
04: * distributed with this work for additional information
05: * regarding copyright ownership. The ASF licenses this file
06: * to you under the Apache License, Version 2.0 (the
07: * "License"); you may not use this file except in compliance
08: * with the License. You may obtain a copy of the License at
09: *
10: * http://www.apache.org/licenses/LICENSE-2.0
11: *
12: * Unless required by applicable law or agreed to in writing,
13: * software distributed under the License is distributed on an
14: * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15: * KIND, either express or implied. See the License for the
16: * specific language governing permissions and limitations
17: * under the License.
18: *
19: */
20: package org.apache.mina.example.imagine.step1.codec;
21:
22: import org.apache.mina.filter.codec.ProtocolEncoderOutput;
23: import org.apache.mina.filter.codec.ProtocolEncoderAdapter;
24: import org.apache.mina.common.IoSession;
25: import org.apache.mina.common.IoBuffer;
26: import org.apache.mina.example.imagine.step1.ImageResponse;
27:
28: import javax.imageio.ImageIO;
29: import java.awt.image.BufferedImage;
30: import java.io.ByteArrayOutputStream;
31: import java.io.IOException;
32:
33: /**
34: * an encoder for {@link ImageResponse} objects
35: *
36: * @author The Apache MINA Project (dev@mina.apache.org)
37: * @version $Rev$, $Date$
38: */
39: public class ImageResponseEncoder extends ProtocolEncoderAdapter {
40:
41: public void encode(IoSession session, Object message,
42: ProtocolEncoderOutput out) throws Exception {
43: ImageResponse imageResponse = (ImageResponse) message;
44: byte[] bytes1 = getBytes(imageResponse.getImage1());
45: byte[] bytes2 = getBytes(imageResponse.getImage2());
46: int capacity = bytes1.length + bytes2.length + 8;
47: IoBuffer buffer = IoBuffer.allocate(capacity, false);
48: buffer.setAutoExpand(true);
49: buffer.putInt(bytes1.length);
50: buffer.put(bytes1);
51: buffer.putInt(bytes2.length);
52: buffer.put(bytes2);
53: buffer.flip();
54: out.write(buffer);
55: }
56:
57: private byte[] getBytes(BufferedImage image) throws IOException {
58: ByteArrayOutputStream baos = new ByteArrayOutputStream();
59: ImageIO.write(image, "PNG", baos);
60: return baos.toByteArray();
61: }
62: }
|