mailiverse/gwt/jre/java/io/ByteArrayOutputStream.java

/*
 * Copyright 2010 Google Inc.
 * 
 * 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 java.io;

import core.util.Arrays;

public class ByteArrayOutputStream extends OutputStream {

  protected int count = 0;
  protected byte[] buf;
 
  public ByteArrayOutputStream() {
    this(16);
  }
  
  public ByteArrayOutputStream(int initialSize) {
    buf = new byte[initialSize];
  }
  
  public void write(int b) {
    if (buf.length == count) {
      byte[] newBuf = new byte[buf.length * 3 / 2];
      Arrays.copyFromTo(buf, 0, newBuf, 0, count);
      buf = newBuf;
    }
    
    buf[count++] = (byte)b;
  }
  
  public void write(byte[] ba, int start, int len) {
	    int end = start + len;
	    for (int i = start; i < end; i++) {
	      write(ba[i]);
	    }
	  }
 
  public void write(byte[] ba)
  {
	  write(ba,0, ba.length);
  }
  
  public byte[] toByteArray() {
	  return Arrays.copyOf(buf,count);
  }

  
  public int size() {
    return count;
  }
  
  public void reset () {
	  count = 0;
  }
}