/* | 
|
 * 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();  | 
|
}  | 
|
}  |