support any output stream

Author: bosborn

src/main/java/mil/nga/sf/util/ByteWriter.java

@@ -2,8 +2,11 @@
 
 import java.io.ByteArrayOutputStream;
 import java.io.IOException;
+import java.io.OutputStream;
 import java.nio.ByteBuffer;
 import java.nio.ByteOrder;
+import java.util.logging.Level;
+import java.util.logging.Logger;
 
 /**
  * Write a byte array
@@ -12,10 +15,16 @@
  */
 public class ByteWriter {
 
+	/**
+	 * Logger
+	 */
+	private static final Logger logger = Logger
+			.getLogger(ByteWriter.class.getName());
+
 	/**
 	 * Output stream to write bytes to
 	 */
-	private final ByteArrayOutputStream os = new ByteArrayOutputStream();
+	private final OutputStream outputStream;
 
 	/**
 	 * Byte order
@@ -26,15 +35,57 @@ public class ByteWriter {
 	 * Constructor
 	 */
 	public ByteWriter() {
+		outputStream = new ByteArrayOutputStream();
+	}
+
+	/**
+	 * Constructor
+	 * 
+	 * @param outputStream
+	 *            output stream
+	 * 
+	 * @since 2.0.3
+	 */
+	public ByteWriter(OutputStream outputStream) {
+		this.outputStream = outputStream;
+	}
+
+	/**
+	 * Get the output stream
+	 * 
+	 * @return output stream
+	 * 
+	 * @since 2.0.3
+	 */
+	public OutputStream getOutputStream() {
+		return outputStream;
+	}
+
+	/**
+	 * Get the output stream
+	 * 
+	 * @return output stream
+	 * 
+	 * @since 2.0.3
+	 */
+	public ByteArrayOutputStream getByteArrayOutputStream() {
+		if (!(outputStream instanceof ByteArrayOutputStream)) {
+			throw new SFException(
+					"Output Stream is not a ByteArrayOutputStream: "
+							+ outputStream.getClass().getName());
+		}
+		return (ByteArrayOutputStream) outputStream;
 	}
 
 	/**
 	 * Close the byte writer
 	 */
 	public void close() {
 		try {
-			os.close();
+			outputStream.close();
 		} catch (IOException e) {
+			logger.log(Level.WARNING,
+					"Failed to close byte writer output stream", e);
 		}
 	}
 
@@ -63,7 +114,7 @@ public void setByteOrder(ByteOrder byteOrder) {
 	 * @return written bytes
 	 */
 	public byte[] getBytes() {
-		return os.toByteArray();
+		return getByteArrayOutputStream().toByteArray();
 	}
 
 	/**
@@ -72,7 +123,7 @@ public byte[] getBytes() {
 	 * @return bytes written
 	 */
 	public int size() {
-		return os.size();
+		return getByteArrayOutputStream().size();
 	}
 
 	/**
@@ -85,17 +136,19 @@ public int size() {
 	 */
 	public void writeString(String value) throws IOException {
 		byte[] valueBytes = value.getBytes();
-		os.write(valueBytes);
+		outputStream.write(valueBytes);
 	}
 
 	/**
 	 * Write a byte
 	 * 
 	 * @param value
 	 *            byte
+	 * @throws IOException
+	 *             upon error
 	 */
-	public void writeByte(byte value) {
-		os.write(value);
+	public void writeByte(byte value) throws IOException {
+		outputStream.write(value);
 	}
 
 	/**
@@ -112,7 +165,7 @@ public void writeInt(int value) throws IOException {
 				.putInt(value);
 		byteBuffer.flip();
 		byteBuffer.get(valueBytes);
-		os.write(valueBytes);
+		outputStream.write(valueBytes);
 	}
 
 	/**
@@ -129,7 +182,7 @@ public void writeDouble(double value) throws IOException {
 				.putDouble(value);
 		byteBuffer.flip();
 		byteBuffer.get(valueBytes);
-		os.write(valueBytes);
+		outputStream.write(valueBytes);
 	}
 
 }