diff src/com/go/trove/io/DefaultCharToByteBuffer.java @ 0:3dc0c5604566

Initial checkin of blitz 2.0 fcs - no installer yet.
author Dan Creswell <dan.creswell@gmail.com>
date Sat, 21 Mar 2009 11:00:06 +0000
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/com/go/trove/io/DefaultCharToByteBuffer.java	Sat Mar 21 11:00:06 2009 +0000
@@ -0,0 +1,294 @@
+/* ====================================================================
+ * Trove - Copyright (c) 1999-2000 Walt Disney Internet Group
+ * ====================================================================
+ * The Tea Software License, Version 1.1
+ *
+ * Copyright (c) 2000 Walt Disney Internet Group. All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ *
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in
+ *    the documentation and/or other materials provided with the
+ *    distribution.
+ *
+ * 3. The end-user documentation included with the redistribution,
+ *    if any, must include the following acknowledgment:
+ *       "This product includes software developed by the
+ *        Walt Disney Internet Group (http://opensource.go.com/)."
+ *    Alternately, this acknowledgment may appear in the software itself,
+ *    if and wherever such third-party acknowledgments normally appear.
+ *
+ * 4. The names "Tea", "TeaServlet", "Kettle", "Trove" and "BeanDoc" must
+ *    not be used to endorse or promote products derived from this
+ *    software without prior written permission. For written
+ *    permission, please contact opensource@dig.com.
+ *
+ * 5. Products derived from this software may not be called "Tea",
+ *    "TeaServlet", "Kettle" or "Trove", nor may "Tea", "TeaServlet",
+ *    "Kettle", "Trove" or "BeanDoc" appear in their name, without prior
+ *    written permission of the Walt Disney Internet Group.
+ *
+ * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
+ * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
+ * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED.  IN NO EVENT SHALL THE WALT DISNEY INTERNET GROUP OR ITS
+ * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+ * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+ * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 
+ * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
+ * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ * ====================================================================
+ *
+ * For more information about Tea, please see http://opensource.go.com/.
+ */
+
+package com.go.trove.io;
+
+import java.io.*;
+
+/******************************************************************************
+ * A CharToByteBuffer implementation that wraps a ByteBuffer for storage.
+ * 
+ * @author Brian S O'Neill
+ * @version
+ * <!--$$Revision: 1.1 $-->, <!--$$JustDate:--> 01/05/30 <!-- $-->
+ */
+public class DefaultCharToByteBuffer
+    implements CharToByteBuffer, Serializable
+{
+    private ByteBuffer mBuffer;
+    private transient OutputStreamWriter mConvertor;
+
+    private char[] mChars;
+    private int mCapacity;
+    private int mCursor;
+    
+    private String mDefaultEncoding;
+
+    /**
+     * @param buffer Buffer that receives the characters converted to bytes.
+     */    
+    public DefaultCharToByteBuffer(ByteBuffer buffer) {
+        this(buffer, null);
+    }
+
+    /**
+     * @param buffer Buffer that receives the characters converted to bytes.
+     * @param defaultEncoding Default character encoding to use if setEncoding
+     * is not called.
+     */    
+    public DefaultCharToByteBuffer(ByteBuffer buffer, String defaultEncoding) {
+        mBuffer = buffer;
+        mChars = new char[4000];
+        mCapacity = mChars.length;
+        mDefaultEncoding = defaultEncoding;
+    }
+
+    public void setEncoding(String enc) throws IOException {
+        drain(true);
+        mConvertor = new OutputStreamWriter
+            (new ByteBufferOutputStream(mBuffer), enc);
+    }
+    
+    public String getEncoding() {
+        return (mConvertor == null) ? mDefaultEncoding :
+            mConvertor.getEncoding();
+    }
+    
+    public long getBaseByteCount() throws IOException {
+        return mBuffer.getBaseByteCount();
+    }
+
+    public long getByteCount() throws IOException {
+        drain(true);
+        return mBuffer.getByteCount();
+    }
+    
+    public void writeTo(OutputStream out) throws IOException {
+        drain(true);
+        mBuffer.writeTo(out);
+    }
+    
+    public void append(byte b) throws IOException {
+        drain(true);
+        mBuffer.append(b);
+    }
+    
+    public void append(byte[] bytes) throws IOException {
+        append(bytes, 0, bytes.length);
+    }
+    
+    public void append(byte[] bytes, int offset, int length)
+        throws IOException {
+
+        if (length != 0) {
+            drain(true);
+            mBuffer.append(bytes, offset, length);
+        }
+    }
+    
+    public void appendSurrogate(ByteData s) throws IOException {
+        if (s != null) {
+            drain(true);
+            mBuffer.appendSurrogate(s);
+        }
+    }
+    
+    public void addCaptureBuffer(ByteBuffer buffer) throws IOException {
+        drain(true);
+        mBuffer.addCaptureBuffer(buffer);
+    }
+
+    public void removeCaptureBuffer(ByteBuffer buffer) throws IOException {
+        drain(true);
+        mBuffer.removeCaptureBuffer(buffer);
+    }
+
+    public void append(char c) throws IOException {
+        if (mCursor >= mCapacity) {
+            drain(false);
+        }
+        mChars[mCursor++] = c;
+    }
+    
+    public void append(char[] chars) throws IOException {
+        append(chars, 0, chars.length);
+    }
+    
+    public void append(char[] chars, int offset, int length) 
+        throws IOException 
+    {
+        if (length == 0) {
+            return;
+        }
+
+        int capacity = mCapacity;
+
+        if (length < (capacity - mCursor)) {
+            System.arraycopy(chars, offset, mChars, mCursor, length);
+            mCursor += length;
+            return;
+        }
+
+        // Make room and try again.
+        drain(false);
+
+        if (length < capacity) {
+            System.arraycopy(chars, offset, mChars, mCursor, length);
+            mCursor += length;
+            return;
+        }
+
+        // Write the whole chunk out at once.
+        getConvertor().write(chars, offset, length);
+    }
+    
+    public void append(String str) throws IOException {
+        append(str, 0, str.length());
+    }
+    
+    public void append(String str, int offset, int length) throws IOException {
+        if (length == 0) {
+            return;
+        }
+
+        int capacity = mCapacity;
+        int avail = capacity - mCursor;
+
+        if (length <= avail) {
+            str.getChars(offset, offset + length, mChars, mCursor);
+            mCursor += length;
+            return;
+        }
+
+        // Fill up the rest of the character buffer and drain it.
+        str.getChars(offset, offset + avail, mChars, mCursor);
+        offset += avail;
+        length -= avail;
+        mCursor = capacity;
+        drain(false);
+
+        // Drain chunks that completely fill the character buffer.
+        while (length >= capacity) {
+            str.getChars(offset, offset + capacity, mChars, 0);
+            offset += capacity;
+            length -= capacity;
+            mCursor = capacity;
+            drain(false);
+        }
+
+        // Copy the remainder into the character buffer, but don't drain.
+        if (length > 0) {
+            str.getChars(offset, offset + length, mChars, 0);
+            mCursor = length;
+        }
+    }
+
+    public void reset() throws IOException {
+        mBuffer.reset();
+    }
+
+    public void drain() throws IOException {
+        drain(true);
+    }
+
+    private OutputStreamWriter getConvertor()
+        throws UnsupportedEncodingException
+    {
+        if (mConvertor == null) {
+            if (mDefaultEncoding == null) {
+                mConvertor = new OutputStreamWriter
+                    (new ByteBufferOutputStream(mBuffer));
+            }
+            else {
+                mConvertor = new OutputStreamWriter
+                    (new ByteBufferOutputStream(mBuffer), mDefaultEncoding);
+            }
+        }
+        return mConvertor;
+    }
+
+    private void drain(boolean flush) throws IOException {
+        if (mCursor != 0) {
+            try {
+                getConvertor().write(mChars, 0, mCursor);
+            }
+            finally {
+                mCursor = 0;
+            }
+        }
+
+        if (flush && mConvertor != null) {
+            mConvertor.flush();
+        }
+    }
+
+    private void writeObject(ObjectOutputStream out) throws IOException {
+        out.defaultWriteObject();
+        if (mConvertor == null) {
+            out.writeObject(null);
+        }
+        else {
+            out.writeObject(mConvertor.getEncoding());
+        }
+    }
+
+    private void readObject(ObjectInputStream in)
+        throws IOException, ClassNotFoundException
+    {
+        in.defaultReadObject();
+        String enc = (String)in.readObject();
+        if (enc != null) {
+            mConvertor = new OutputStreamWriter
+                (new ByteBufferOutputStream(mBuffer), enc);
+        }
+    }
+}