BasicOutputBuffer.java

// BasicOutputBuffer.java

package org.bson.io;

import java.io.*;

public class BasicOutputBuffer extends OutputBuffer {

    public void write(byte[] b){
        write( b , 0 , b.length );
    }
    
    public void write(byte[] b, int off, int len){
        _ensure( len );
        System.arraycopy( b , off , _buffer , _cur , len );
        _cur += len;
        _size = Math.max( _cur , _size );
    }
    public void write(int b){
        _ensure(1);
        _buffer[_cur++] = (byte)(0xFF&b);
        _size = Math.max( _cur , _size );
    }
    
    public int getPosition(){
        return _cur;
    }
    public void setPosition( int position ){
        _cur = position;
    }
    
    public void seekEnd(){
        _cur = _size;
    }
    public void seekStart(){
        _cur = 0;
    }
    
    /**
     * @return size of data so far
     */
    public int size(){
        return _size;
    }
    
    /**
     * @return bytes written
     */
    public int pipe( OutputStream out )
        throws IOException {
        out.write( _buffer , 0 , _size );
        return _size;
    }

    void _ensure( int more ){
        if ( _cur + more < _buffer.length )
            return;
        byte[] n = new byte[_buffer.length*2];
        System.arraycopy( _buffer , 0 , n , 0 , _size );
        _buffer = n;
    }

    public String asString(){
        return new String( _buffer , 0 , _size );
    }

    public String asString( String encoding )
        throws UnsupportedEncodingException {
        return new String( _buffer , 0 , _size , encoding );
    }


    private int _cur;
    private int _size;
    private byte[] _buffer = new byte[512];
}