MonitoredOutputStream.java

/*
 * Copyright © 2009-2011 Rebecca G. Bettencourt / Kreative Software
 * <p>
 * The contents of this file are subject to the Mozilla Public License
 * Version 1.1 (the "License"); you may not use this file except in
 * compliance with the License. You may obtain a copy of the License at
 * <a href="http://www.mozilla.org/MPL/">http://www.mozilla.org/MPL/</a>
 * <p>
 * Software distributed under the License is distributed on an "AS IS"
 * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
 * License for the specific language governing rights and limitations
 * under the License.
 * <p>
 * Alternatively, the contents of this file may be used under the terms
 * of the GNU Lesser General Public License (the "LGPL License"), in which
 * case the provisions of LGPL License are applicable instead of those
 * above. If you wish to allow use of your version of this file only
 * under the terms of the LGPL License and not to allow others to use
 * your version of this file under the MPL, indicate your decision by
 * deleting the provisions above and replace them with the notice and
 * other provisions required by the LGPL License. If you do not delete
 * the provisions above, a recipient may use your version of this file
 * under either the MPL or the LGPL License.
 * @since PowerPaint 1.0
 * @author Rebecca G. Bettencourt, Kreative Software
 */

package com.kreative.paint.io;

import java.io.FilterOutputStream;
import java.io.IOException;
import java.io.OutputStream;

public class MonitoredOutputStream extends FilterOutputStream {
	private Monitor mon;
	private boolean nested;
	
	public MonitoredOutputStream(Monitor mon, OutputStream out, boolean nested) {
		super(out);
		this.mon = mon;
		this.nested = nested;
		if (!nested) mon.startMonitoring();
	}
	
	public void close() throws IOException {
		super.close();
		if (!nested) mon.stopMonitoring();
	}
	
	public void flush() throws IOException {
		super.flush();
	}
	
	public void write(byte[] b) throws IOException {
		super.write(b);
		mon.setValue(mon.getValue() + b.length);
	}
	
	public void write(byte[] b, int off, int len) throws IOException {
		super.write(b, off, len);
		mon.setValue(mon.getValue() + len);
	}
	
	public void write(int b) throws IOException {
		super.write(b);
		mon.setValue(mon.getValue() + 1);
	}
}