Mercurial > hg > blitz_stable
diff src/com/go/trove/io/FastCharToByteBuffer.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/FastCharToByteBuffer.java Sat Mar 21 11:00:06 2009 +0000 @@ -0,0 +1,274 @@ +/* ==================================================================== + * Trove - Copyright (c) 1999-2001 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.*; +import java.util.Set; +import java.util.HashSet; + +/****************************************************************************** + * A CharToByteBuffer implementation that converts ISO-8859-1 encoded + * characters faster. To force fast conversion, construct FastCharToByteBuffer + * with ISO-8859-1 as the default encoding. + * + * @author Brian S O'Neill + * @version + * <!--$$Revision: 1.1 $-->, <!--$$JustDate:--> 01/06/14 <!-- $--> + */ +public class FastCharToByteBuffer implements CharToByteBuffer, Serializable { + private static final int TEMP_BUF_LEN = 512; + private static final Set SUPPORTED_ENCODINGS; + + static { + SUPPORTED_ENCODINGS = new HashSet(); + SUPPORTED_ENCODINGS.add("8859-1"); + SUPPORTED_ENCODINGS.add("8859_1"); + SUPPORTED_ENCODINGS.add("iso-8859-1"); + SUPPORTED_ENCODINGS.add("ISO-8859-1"); + SUPPORTED_ENCODINGS.add("iso8859_1"); + SUPPORTED_ENCODINGS.add("ISO8859_1"); + } + + static boolean isSupportedEncoding(String encoding) { + return SUPPORTED_ENCODINGS.contains(encoding); + } + + private ByteBuffer mBuffer; + + private transient byte[] mTempBytes; + private transient char[] mTempChars; + + private CharToByteBuffer mSlowConvertor; + + /** + * @param buffer Buffer that receives the characters converted to bytes. + */ + public FastCharToByteBuffer(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 FastCharToByteBuffer(ByteBuffer buffer, String defaultEncoding) { + mBuffer = buffer; + try { + setEncoding(defaultEncoding); + } + catch (IOException e) { + } + } + + public void setEncoding(String enc) throws IOException { + drain(); + if (isSupportedEncoding(enc)) { + mSlowConvertor = null; + } + else { + mSlowConvertor = new DefaultCharToByteBuffer(mBuffer, enc); + } + } + + public String getEncoding() throws IOException { + return mSlowConvertor != null ? + mSlowConvertor.getEncoding() : "ISO-8859-1"; + } + + public long getBaseByteCount() throws IOException { + drain(); + return mBuffer.getBaseByteCount(); + } + + public long getByteCount() throws IOException { + drain(); + return mBuffer.getByteCount(); + } + + public void writeTo(OutputStream out) throws IOException { + drain(); + mBuffer.writeTo(out); + } + + public void append(byte b) throws IOException { + drain(); + 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(); + mBuffer.append(bytes, offset, length); + } + } + + public void appendSurrogate(ByteData s) throws IOException { + if (s != null) { + drain(); + mBuffer.appendSurrogate(s); + } + } + + public void addCaptureBuffer(ByteBuffer buffer) throws IOException { + drain(); + mBuffer.addCaptureBuffer(buffer); + } + + public void removeCaptureBuffer(ByteBuffer buffer) throws IOException { + drain(); + mBuffer.removeCaptureBuffer(buffer); + } + + public void append(char c) throws IOException { + char[] chars = getTempChars(); + chars[0] = c; + append(chars, 0, 1); + } + + public void append(char[] chars) throws IOException { + append(chars, 0, chars.length); + } + + public void append(char[] chars, int offset, int length) + throws IOException + { + if (mSlowConvertor != null) { + mSlowConvertor.append(chars, offset, length); + return; + } + + appendFast(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 (mSlowConvertor != null) { + mSlowConvertor.append(str, offset, length); + return; + } + + if (length == 0) { + return; + } + + char[] tempChars = getTempChars(); + int bufLen = tempChars.length; + + while (length >= bufLen) { + str.getChars(offset, offset + bufLen, tempChars, 0); + offset += bufLen; + length -= bufLen; + appendFast(tempChars, 0, bufLen); + } + + if (length > 0) { + str.getChars(offset, offset + length, tempChars, 0); + appendFast(tempChars, 0, length); + } + } + + public void reset() throws IOException { + mBuffer.reset(); + } + + public void drain() throws IOException { + if (mSlowConvertor != null) { + mSlowConvertor.drain(); + } + } + + private void appendFast(char[] chars, int offset, int length) + throws IOException + { + byte[] tempBytes = getTempBytes(); + int bufLen = tempBytes.length; + int bi = 0; + int climit = offset + length; + + for (int ci = offset; ci < climit; ci++) { + tempBytes[bi++] = (byte)chars[ci]; + if (bi >= bufLen) { + mBuffer.append(tempBytes, 0, bufLen); + bi = 0; + } + } + + if (bi > 0) { + mBuffer.append(tempBytes, 0, bi); + } + } + + private byte[] getTempBytes() { + if (mTempBytes == null) { + mTempBytes = new byte[TEMP_BUF_LEN]; + } + return mTempBytes; + } + + private char[] getTempChars() { + if (mTempChars == null) { + mTempChars = new char[TEMP_BUF_LEN]; + } + return mTempChars; + } +}