bytebuffer.java.svn-base

来自「cqME :java framework for TCK test.」· SVN-BASE 代码 · 共 82 行

/*
 * $Id$
 *
 * Copyright 1996-2007 Sun Microsystems, Inc. All Rights Reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER
 *
 * This program 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.
 *
 * This program 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 at /legal/license.txt).
 *
 * 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 Sun Microsystems, Inc., 4150 Network Circle, Santa
 * Clara, CA 95054 or visit www.sun.com if you need additional
 * information or have any questions.
 *
 */

package com.sun.cldc.communication.http.util;

import java.io.*;

public final class ByteBuffer {

    private byte[] myBuf;
    private int currentOffset=0;
    private boolean verboseIsOn=false;

    public ByteBuffer(int init_capacity) {
        myBuf = new byte[init_capacity];
        verboseIsOn = false;
    }

    public void setVerbose(boolean val) {
        verboseIsOn = val;
    }

    public void write(DataInputStream from, int len) throws IOException {
        check(len);
        from.readFully(myBuf, currentOffset, len);
        currentOffset += len;
    }

    public int dataLength() {
        return currentOffset;
    }

    public byte[] dataCopy() {
        byte bb[] = new byte[currentOffset];
        System.arraycopy(myBuf, 0, bb, 0, currentOffset);
        return bb;
    }

    public byte[] data() {
        return myBuf;
    }

    public void reset() {
        currentOffset = 0;
    }


    private void check(int len) {
        int cap = currentOffset + len; // needed capacity
        if (myBuf.length < cap) {
            if (verboseIsOn) System.out.println("Reallocating buffer: " + myBuf.length + " --> " + cap);
            byte[] temp = new byte[cap];
            System.arraycopy(myBuf, 0, temp, 0, currentOffset);
            myBuf = temp;
        }
    }        
}