/*

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

package com.sun.xml.internal.bind.v2.util;

import java.io.ByteArrayOutputStream;

import java.io.InputStream;

import java.io.IOException;

import com.sun.xml.internal.bind.v2.runtime.unmarshaller.Base64Data;

/**

 * {@link ByteArrayOutputStream} with access to its raw buffer.

 */

public final class ByteArrayOutputStreamEx extends ByteArrayOutputStream {

    public ByteArrayOutputStreamEx() {

    }

    public ByteArrayOutputStreamEx(int size) {

        super(size);

    }

    public void set(Base64Data dt,String mimeType) {

        dt.set(buf,count,mimeType);

    }

    public byte[] getBuffer() {

        return buf;

    }

    /**

     * Reads the given {@link InputStream} completely into the buffer.

     */

    public void readFrom(InputStream is) throws IOException {

        while(true) {

            if(count==buf.length) {

                // realllocate

                byte[] data = new byte[buf.length*2];

                System.arraycopy(buf,0,data,0,buf.length);

                buf = data;

            }

            int sz = is.read(buf,count,buf.length-count);

            if(sz<0)     return;

            count += sz;

        }

    }

}