diff src/com/go/trove/io/FileByteBuffer.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/FileByteBuffer.java	Sat Mar 21 11:00:06 2009 +0000
@@ -0,0 +1,265 @@
+/* ====================================================================
+ * 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.OutputStream;
+import java.io.RandomAccessFile;
+import java.io.IOException;
+import java.util.List;
+import java.util.ArrayList;
+
+/******************************************************************************
+ * A ByteBuffer implementation that can read from an open file or can write
+ * to it. This implementation is best suited for temporary byte data that is
+ * too large to hold in memory.
+ *
+ * @author Brian S O'Neill
+ * @version
+ * <!--$$Revision: 1.1 $--> 5 <!-- $$JustDate:--> 00/12/05 <!-- $-->
+ */
+public class FileByteBuffer implements ByteBuffer {
+    private RandomAccessFile mFile;
+    private List mSurrogates;
+    private List mCaptureBuffers;
+
+    /**
+     * Creates a FileByteBuffer on a RandomAccessFile. If the file is opened
+     * read-only, then the append operations will fail.
+     *
+     * @param file The file to use as a buffer.
+     */
+    public FileByteBuffer(RandomAccessFile file) throws IOException {
+        mFile = file;
+        file.seek(0);
+    }
+
+    public long getBaseByteCount() throws IOException {
+        return mFile.length();
+    }
+
+    public long getByteCount() throws IOException {
+        long count = getBaseByteCount();
+        if (mSurrogates == null) {
+            return count;
+        }
+        
+        int size = mSurrogates.size();
+        for (int i=0; i<size; i++) {
+            count += ((Surrogate)mSurrogates.get(i)).mByteData.getByteCount();
+        }
+
+        return count;
+    }
+
+    public void writeTo(OutputStream out) throws IOException {
+        long length = mFile.length();
+        int bufSize;
+        if (length > 4000) {
+            bufSize = 4000;
+        }
+        else {
+            bufSize = (int)length;
+        }
+        
+        byte[] inputBuffer = new byte[bufSize];
+        
+        mFile.seek(0);
+
+        if (mSurrogates != null) {
+            long currentPos = 0;
+            
+            int size = mSurrogates.size();
+            for (int i=0; i<size; i++) {
+                Surrogate s = (Surrogate)mSurrogates.get(i);
+                currentPos = writeTo(inputBuffer, out, currentPos, s.mPos);
+                s.mByteData.writeTo(out);
+            }
+        }
+
+        // Write out the rest of the file.
+        int readAmount;
+        while ((readAmount = mFile.read(inputBuffer, 0, bufSize)) > 0) {
+            out.write(inputBuffer, 0, readAmount);
+        }
+    }
+
+    private long writeTo(byte[] inputBuffer, OutputStream out, 
+                         long fromPos, long toPos) throws IOException {
+        if (toPos == fromPos) {
+            return fromPos;
+        }
+
+        int bufSize = inputBuffer.length;
+        int readAmount;
+
+        while (toPos > fromPos) {
+            int amount;
+            if (bufSize <= (toPos - fromPos)) {
+                amount = bufSize;
+            }
+            else {
+                amount = (int)(toPos - fromPos);
+            }
+
+            while ((readAmount = mFile.read(inputBuffer, 0, amount)) > 0) {
+                out.write(inputBuffer, 0, readAmount);
+                fromPos += readAmount;
+                amount -= readAmount;
+                if (amount <= 0) {
+                    break;
+                }
+            }
+
+            if (readAmount <= 0) {
+                break;
+            }
+        }
+
+        return fromPos;
+    }
+
+    public void append(byte b) throws IOException {
+        List captureBuffers;
+        if ((captureBuffers = mCaptureBuffers) != null) {
+            int size = captureBuffers.size();
+            for (int i=0; i<size; i++) {
+                ((ByteBuffer)captureBuffers.get(i)).append(b);
+            }
+        }
+
+        mFile.write(b);
+    }
+
+    public void append(byte[] bytes) throws IOException {
+        mFile.write(bytes);
+    }
+
+    public void append(byte[] bytes, int offset, int length) 
+        throws IOException 
+    {
+        List captureBuffers;
+        if ((captureBuffers = mCaptureBuffers) != null) {
+            int size = captureBuffers.size();
+            for (int i=0; i<size; i++) {
+                ((ByteBuffer)captureBuffers.get(i)).append
+                    (bytes, offset, length);
+            }
+        }
+
+        mFile.write(bytes, offset, length); 
+    }
+
+    public void appendSurrogate(ByteData s) throws IOException {
+        if (s == null) {
+            return;
+        }
+
+        List captureBuffers;
+        if ((captureBuffers = mCaptureBuffers) != null) {
+            int size = captureBuffers.size();
+            for (int i=0; i<size; i++) {
+                ((ByteBuffer)captureBuffers.get(i)).appendSurrogate(s);
+            }
+        }
+
+        if (mSurrogates == null) {
+            mSurrogates = new ArrayList();
+        }
+
+        mSurrogates.add(new Surrogate(s));
+    }
+
+    public void addCaptureBuffer(ByteBuffer buffer) {
+        List captureBuffers;
+        if ((captureBuffers = mCaptureBuffers) == null) {
+            captureBuffers = mCaptureBuffers = new ArrayList();
+        }
+        captureBuffers.add(buffer);
+    }
+
+    public void removeCaptureBuffer(ByteBuffer buffer) {
+        List captureBuffers;
+        if ((captureBuffers = mCaptureBuffers) != null) {
+            captureBuffers.remove(buffer);
+        }
+    }
+
+    public void reset() throws IOException {
+        List byteDatas;
+        int i, size;
+
+        if ((byteDatas = mSurrogates) != null) {
+            size = byteDatas.size();
+            for (i=0; i<size; i++) {
+                ((ByteData)byteDatas.get(i)).reset();
+            }
+        }
+
+        if ((byteDatas = mCaptureBuffers) != null) {
+            size = byteDatas.size();
+            for (i=0; i<size; i++) {
+                ((ByteData)byteDatas.get(i)).reset();
+            }
+        }
+    }
+
+    private class Surrogate {
+        public final ByteData mByteData;
+        public final long mPos;
+
+        public Surrogate(ByteData data) throws IOException {
+            mByteData = data;
+            mPos = mFile.getFilePointer();
+        }
+    }
+}