root/misc/red5/encoder-pipe-r3603.diff

src/org/red5/io/flv/impl/FLVWriter.java

@@ -21 +21 @@
 
 import java.io.File;
 import java.io.FileOutputStream;
+import java.io.OutputStream;
+import java.nio.ByteBuffer;
+import java.lang.Process;
 import java.io.IOException;
 import java.io.RandomAccessFile;
 import java.nio.channels.FileChannel;
@@ -115 +118 @@
          * need direct access to file to append full duration metadata
          */
         private File file;
+
+        /**
+         * custom encoder pipe
+         */
+        private Process encoder;
         
         /**
          * Creates writer implementation with given file output stream and last tag
@@ -154 +162 @@
                 channel = this.fos.getChannel();
                 out = IoBuffer.allocate(1024);
                 out.setAutoExpand(true);                
+                try {
+                        // TODO make this configurable
+                        // and pass file-name and some meta-info along
+                        this.encoder = Runtime.getRuntime().exec(new String[] {"/home/rgareus/data/coding/inoutC2/misc/red5/myenc"});
+                } catch (IOException e) {
+                        log.error("IO error on FLVpipe", e);
+                } 
         }
         
         /**
@@ -185 +200 @@
                 out.putInt(0);
 
                 out.flip();
-
+                
+                this.encpipe(out.buf());
                 channel.write(out.buf());
         }
 
+        /**
+         *  write bytebuffer to encoder pipeline
+         */
+        private void encpipe(ByteBuffer bb) {
+                ByteBuffer bc = bb.duplicate();
+                byte[] dst= new byte[bc.remaining()];
+                bc.get(dst,0,dst.length);
+                try {
+                        this.encoder.getOutputStream().write(dst);
+                } catch (IOException e) { ; }
+        }
+
         /** {@inheritDoc}
          */
         public IStreamableFile getFile() {
@@ -260 +288 @@
                 out.putInt(0x00);
 
                 out.flip();
+
+                this.encpipe(out.buf());
                 bytesWritten += channel.write(out.buf());
 
                 IoBuffer bodyBuf = tag.getBody();
+
+                this.encpipe(bodyBuf.buf());
                 bytesWritten += channel.write(bodyBuf.buf());
 
                 if (audioCodecId == -1 && tag.getDataType() == ITag.TYPE_AUDIO) {
@@ -280 +312 @@
                 out.clear();
                 out.putInt(tag.getBodySize() + 11);
                 out.flip();
+
+                this.encpipe(out.buf());
                 bytesWritten += channel.write(out.buf());
+                log.debug("written: {}", bytesWritten);
 
                 return false;
         }
@@ -296 +331 @@
         public void close() {
                 RandomAccessFile appender = null;
                 try {
+                        this.encoder.getOutputStream().close(); 
                         if (metaPosition > 0) {
                                 long oldPos = channel.position();
                                 try {