196 lines · 6.0 KB · Java
RawDownload
/* * Copyright (c) 2000, 2006, Oracle and/or its affiliates. All rights reserved. * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. * * This code 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.  Oracle designates this * particular file as subject to the "Classpath" exception as provided * by Oracle in the LICENSE file that accompanied this code. * * This code 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 in the LICENSE file that * accompanied this code). * * 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA * or visit www.oracle.com if you need additional information or have any * questions. */ package javax.imageio.stream; import java.io.IOException;import java.io.OutputStream; /** * An implementation of <code>ImageOutputStream</code> that writes its * output to a regular <code>OutputStream</code>.  A memory buffer is * used to cache at least the data between the discard position and * the current write position.  The only constructor takes an * <code>OutputStream</code>, so this class may not be used for * read/modify/write operations.  Reading can occur only on parts of * the stream that have already been written to the cache and not * yet flushed. * */public class MemoryCacheImageOutputStream extends ImageOutputStreamImpl {     private OutputStream stream;     private MemoryCache cache = new MemoryCache();     /**     * Constructs a <code>MemoryCacheImageOutputStream</code> that will write     * to a given <code>OutputStream</code>.     *     * @param stream an <code>OutputStream</code> to write to.     *     * @exception IllegalArgumentException if <code>stream</code> is     * <code>null</code>.     */    public MemoryCacheImageOutputStream(OutputStream stream) {        if (stream == null) {            throw new IllegalArgumentException("stream == null!");        }        this.stream = stream;    }     public int read() throws IOException {        checkClosed();         bitOffset = 0;         int val = cache.read(streamPos);        if (val != -1) {            ++streamPos;        }        return val;    }     public int read(byte[] b, int off, int len) throws IOException {        checkClosed();         if (b == null) {            throw new NullPointerException("b == null!");        }        // Fix 4467608: read([B,I,I) works incorrectly if len<=0        if (off < 0 || len < 0 || off + len > b.length || off + len < 0) {            throw new IndexOutOfBoundsException                ("off < 0 || len < 0 || off+len > b.length || off+len < 0!");        }         bitOffset = 0;         if (len == 0) {            return 0;        }         // check if we're already at/past EOF i.e.        // no more bytes left to read from cache        long bytesLeftInCache = cache.getLength() - streamPos;        if (bytesLeftInCache <= 0) {            return -1; // EOF        }         // guaranteed by now that bytesLeftInCache > 0 && len > 0        // and so the rest of the error checking is done by cache.read()        // NOTE that alot of error checking is duplicated        len = (int)Math.min(bytesLeftInCache, (long)len);        cache.read(b, off, len, streamPos);        streamPos += len;        return len;    }     public void write(int b) throws IOException {        flushBits(); // this will call checkClosed() for us        cache.write(b, streamPos);        ++streamPos;    }     public void write(byte[] b, int off, int len) throws IOException {        flushBits(); // this will call checkClosed() for us        cache.write(b, off, len, streamPos);        streamPos += len;    }     public long length() {        try {            checkClosed();            return cache.getLength();        } catch (IOException e) {            return -1L;        }    }     /**     * Returns <code>true</code> since this     * <code>ImageOutputStream</code> caches data in order to allow     * seeking backwards.     *     * @return <code>true</code>.     *     * @see #isCachedMemory     * @see #isCachedFile     */    public boolean isCached() {        return true;    }     /**     * Returns <code>false</code> since this     * <code>ImageOutputStream</code> does not maintain a file cache.     *     * @return <code>false</code>.     *     * @see #isCached     * @see #isCachedMemory     */    public boolean isCachedFile() {        return false;    }     /**     * Returns <code>true</code> since this     * <code>ImageOutputStream</code> maintains a main memory cache.     *     * @return <code>true</code>.     *     * @see #isCached     * @see #isCachedFile     */    public boolean isCachedMemory() {        return true;    }     /**     * Closes this <code>MemoryCacheImageOutputStream</code>.  All     * pending data is flushed to the output, and the cache     * is released.  The destination <code>OutputStream</code>     * is not closed.     */    public void close() throws IOException {        long length = cache.getLength();        seek(length);        flushBefore(length);        super.close();        cache.reset();        cache = null;        stream = null;    }     public void flushBefore(long pos) throws IOException {        long oFlushedPos = flushedPos;        super.flushBefore(pos); // this will call checkClosed() for us         long flushBytes = flushedPos - oFlushedPos;        cache.writeToStream(stream, oFlushedPos, flushBytes);        cache.disposeBefore(flushedPos);        stream.flush();    }}