| /** |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You 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.apache.activemq.util; |
| |
| import java.io.DataOutput; |
| import java.io.IOException; |
| import java.io.OutputStream; |
| import java.io.UTFDataFormatException; |
| |
| /** |
| * Optimized ByteArrayOutputStream |
| * |
| * |
| */ |
| public final class DataByteArrayOutputStream extends OutputStream implements DataOutput { |
| private static final int DEFAULT_SIZE = 2048; |
| private byte buf[]; |
| private int pos; |
| |
| /** |
| * Creates a new byte array output stream, with a buffer capacity of the |
| * specified size, in bytes. |
| * |
| * @param size the initial size. |
| * @exception IllegalArgumentException if size is negative. |
| */ |
| public DataByteArrayOutputStream(int size) { |
| if (size < 0) { |
| throw new IllegalArgumentException("Invalid size: " + size); |
| } |
| buf = new byte[size]; |
| } |
| |
| /** |
| * Creates a new byte array output stream. |
| */ |
| public DataByteArrayOutputStream() { |
| this(DEFAULT_SIZE); |
| } |
| |
| /** |
| * start using a fresh byte array |
| * |
| * @param size |
| */ |
| public void restart(int size) { |
| buf = new byte[size]; |
| pos = 0; |
| } |
| |
| /** |
| * start using a fresh byte array |
| */ |
| public void restart() { |
| restart(DEFAULT_SIZE); |
| } |
| |
| /** |
| * Get a ByteSequence from the stream |
| * |
| * @return the byte sequence |
| */ |
| public ByteSequence toByteSequence() { |
| return new ByteSequence(buf, 0, pos); |
| } |
| |
| /** |
| * Writes the specified byte to this byte array output stream. |
| * |
| * @param b the byte to be written. |
| */ |
| public void write(int b) { |
| int newcount = pos + 1; |
| ensureEnoughBuffer(newcount); |
| buf[pos] = (byte)b; |
| pos = newcount; |
| } |
| |
| /** |
| * Writes <code>len</code> bytes from the specified byte array starting at |
| * offset <code>off</code> to this byte array output stream. |
| * |
| * @param b the data. |
| * @param off the start offset in the data. |
| * @param len the number of bytes to write. |
| */ |
| public void write(byte b[], int off, int len) { |
| if (len == 0) { |
| return; |
| } |
| int newcount = pos + len; |
| ensureEnoughBuffer(newcount); |
| System.arraycopy(b, off, buf, pos, len); |
| pos = newcount; |
| } |
| |
| /** |
| * @return the underlying byte[] buffer |
| */ |
| public byte[] getData() { |
| return buf; |
| } |
| |
| /** |
| * reset the output stream |
| */ |
| public void reset() { |
| pos = 0; |
| } |
| |
| /** |
| * Set the current position for writing |
| * |
| * @param offset |
| */ |
| public void position(int offset) { |
| ensureEnoughBuffer(offset); |
| pos = offset; |
| } |
| |
| public int size() { |
| return pos; |
| } |
| |
| public void writeBoolean(boolean v) { |
| ensureEnoughBuffer(pos + 1); |
| buf[pos++] = (byte)(v ? 1 : 0); |
| } |
| |
| public void writeByte(int v) { |
| ensureEnoughBuffer(pos + 1); |
| buf[pos++] = (byte)(v >>> 0); |
| } |
| |
| public void writeShort(int v) { |
| ensureEnoughBuffer(pos + 2); |
| buf[pos++] = (byte)(v >>> 8); |
| buf[pos++] = (byte)(v >>> 0); |
| } |
| |
| public void writeChar(int v) { |
| ensureEnoughBuffer(pos + 2); |
| buf[pos++] = (byte)(v >>> 8); |
| buf[pos++] = (byte)(v >>> 0); |
| } |
| |
| public void writeInt(int v) { |
| ensureEnoughBuffer(pos + 4); |
| buf[pos++] = (byte)(v >>> 24); |
| buf[pos++] = (byte)(v >>> 16); |
| buf[pos++] = (byte)(v >>> 8); |
| buf[pos++] = (byte)(v >>> 0); |
| } |
| |
| public void writeLong(long v) { |
| ensureEnoughBuffer(pos + 8); |
| buf[pos++] = (byte)(v >>> 56); |
| buf[pos++] = (byte)(v >>> 48); |
| buf[pos++] = (byte)(v >>> 40); |
| buf[pos++] = (byte)(v >>> 32); |
| buf[pos++] = (byte)(v >>> 24); |
| buf[pos++] = (byte)(v >>> 16); |
| buf[pos++] = (byte)(v >>> 8); |
| buf[pos++] = (byte)(v >>> 0); |
| } |
| |
| public void writeFloat(float v) throws IOException { |
| writeInt(Float.floatToIntBits(v)); |
| } |
| |
| public void writeDouble(double v) throws IOException { |
| writeLong(Double.doubleToLongBits(v)); |
| } |
| |
| public void writeBytes(String s) { |
| int length = s.length(); |
| for (int i = 0; i < length; i++) { |
| write((byte)s.charAt(i)); |
| } |
| } |
| |
| public void writeChars(String s) { |
| int length = s.length(); |
| for (int i = 0; i < length; i++) { |
| int c = s.charAt(i); |
| write((c >>> 8) & 0xFF); |
| write((c >>> 0) & 0xFF); |
| } |
| } |
| |
| public void writeUTF(String text) throws IOException { |
| long encodedsize = MarshallingSupport.countUTFBytes(text); |
| if (encodedsize > 65535) { |
| throw new UTFDataFormatException("encoded string too long: " + encodedsize + " bytes"); |
| } |
| ensureEnoughBuffer((int)(pos + encodedsize + 2)); |
| writeShort((int)encodedsize); |
| |
| byte[] buffer = new byte[(int)encodedsize]; |
| MarshallingSupport.writeUTFBytesToBuffer(text, (int) encodedsize, buf, pos); |
| pos += encodedsize; |
| } |
| |
| private void ensureEnoughBuffer(int newcount) { |
| if (newcount > buf.length) { |
| byte newbuf[] = new byte[Math.max(buf.length << 1, newcount)]; |
| System.arraycopy(buf, 0, newbuf, 0, pos); |
| buf = newbuf; |
| } |
| } |
| } |