/*
* ====================================================================
* Copyright (c) 2004-2009 TMate Software Ltd. All rights reserved.
*
* This software is licensed as described in the file COPYING, which
* you should have received as part of this distribution. The terms
* are also available at http://svnkit.com/license.html.
* If newer versions of this license are posted there, you may use a
* newer version instead, at your option.
* ====================================================================
*/
package org.tmatesoft.svn.core.internal.util;
import java.io.FilterOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.nio.ByteBuffer;
import java.nio.charset.Charset;
import org.tmatesoft.svn.core.SVNException;
import org.tmatesoft.svn.core.internal.wc.IOExceptionWrapper;
/**
* @author TMate Software Ltd.
* @version 1.3
*/
public class SVNCharsetOutputStream extends FilterOutputStream {
private static final byte[] EMPTY_ARRAY = new byte[0];
private SVNCharsetConvertor myCharsetConvertor;
private ByteBuffer myOutputBuffer;
private boolean myFlushed;
public SVNCharsetOutputStream(OutputStream out, Charset inputCharset, Charset outputCharset) {
super(out);
myCharsetConvertor = new SVNCharsetConvertor(inputCharset.newDecoder(), outputCharset.newEncoder());
myFlushed = false;
}
public void write(int b) throws IOException {
write(new byte[]{(byte) (b & 0xFF)});
}
public void write(byte[] b) throws IOException {
write(b, 0, b.length);
}
public void write(byte[] b, int off, int len) throws IOException {
try {
myOutputBuffer = myCharsetConvertor.convertChunk(b, off, len, myOutputBuffer, false);
myOutputBuffer.flip();
out.write(myOutputBuffer.array(), myOutputBuffer.arrayOffset(), myOutputBuffer.limit());
} catch (SVNException e) {
throw new IOExceptionWrapper(e);
}
}
public void flush() throws IOException {
if (!myFlushed) {
try {
myOutputBuffer = myCharsetConvertor.convertChunk(EMPTY_ARRAY, 0, 0, myOutputBuffer, true);
myOutputBuffer = myCharsetConvertor.flush(myOutputBuffer);
myOutputBuffer.flip();
out.write(myOutputBuffer.array(), myOutputBuffer.arrayOffset(), myOutputBuffer.limit());
} catch (SVNException e) {
throw new IOExceptionWrapper(e);
} finally {
myFlushed = true;
}
}
super.flush();
}
public void close() throws IOException {
flush();
out.close();
}
}