Write MMTF to an OutputStream

abradle · abradle · commit 0b42061fdcaa · 2016-08-16T12:08:00.000-07:00
diff --git a/biojava-structure/src/main/java/org/biojava/nbio/structure/io/mmtf/MmtfActions.java b/biojava-structure/src/main/java/org/biojava/nbio/structure/io/mmtf/MmtfActions.java
@@ -1,6 +1,8 @@
 package org.biojava.nbio.structure.io.mmtf;
 
+import java.io.ByteArrayOutputStream;
 import java.io.IOException;
+import java.io.OutputStream;
 import java.nio.file.Path;
 
 import org.biojava.nbio.structure.Structure;
@@ -46,6 +48,24 @@ public static void writeToFile(Structure structure, Path path) throws IOExceptio
 		// Now write this data to file
 		WriterUtils.writeDataToFile(writerToEncoder, path);
 	}
+	
+	/**
+	 * Write a Structure object to an {@link OutputStream}
+	 * @param structure the Structure to write
+	 * @return the {@link OutputStream} of the MMTF structure
+	 * @throws IOException an error transferring the byte[]
+	 */
+	public static OutputStream writeToOutputStream(Structure structure) throws IOException{
+		// Set up this writer
+		AdapterToStructureData writerToEncoder = new AdapterToStructureData();
+		// Get the writer - this is what people implement
+		new MmtfStructureWriter(structure, writerToEncoder);
+		// Now write this data to file
+		byte[] outputBytes = WriterUtils.getDataAsByteArr(writerToEncoder);
+		ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream(outputBytes.length);
+		byteArrayOutputStream.write(outputBytes,0,outputBytes.length);
+		return byteArrayOutputStream;
+	}
 
 	
 	/**
